/* * * Licensed under GPLv2, see file LICENSE in this source tree. */ /* You need to define the following (example): #define type long #define xstrtou(rest) xstrtoul##rest #define xstrto(rest) xstrtol##rest #define xatou(rest) xatoul##rest #define xato(rest) xatol##rest #define XSTR_UTYPE_MAX ULONG_MAX #define XSTR_TYPE_MAX LONG_MAX #define XSTR_TYPE_MIN LONG_MIN #define XSTR_STRTOU strtoul */ unsigned type FAST_FUNC xstrtou(_range_sfx)(const char *numstr, int base, unsigned type lower, unsigned type upper, const struct suffix_mult *suffixes) { unsigned type r; int old_errno; char *e; /* Disallow '-' and any leading whitespace. */ if (*numstr == '-' || *numstr == '+' || isspace(*numstr)) goto inval; /* Since this is a lib function, we're not allowed to reset errno to 0. * Doing so could break an app that is deferring checking of errno. * So, save the old value so that we can restore it if successful. */ old_errno = errno; errno = 0; r = XSTR_STRTOU(numstr, &e, base); /* Do the initial validity check. Note: The standards do not * guarantee that errno is set if no digits were found. So we * must test for this explicitly. */ if (errno || numstr == e) goto inval; /* error / no digits / illegal trailing chars */ errno = old_errno; /* Ok. So restore errno. */ /* Do optional suffix parsing. Allow 'empty' suffix tables. * Note that we also allow nul suffixes with associated multipliers, * to allow for scaling of the numstr by some default multiplier. */ if (suffixes) { while (suffixes->mult) { if (strcmp(suffixes->suffix, e) == 0) { if (XSTR_UTYPE_MAX / suffixes->mult < r) goto range; /* overflow! */ r *= suffixes->mult; goto chk_range; } ++suffixes; } } /* Note: trailing space is an error. * It would be easy enough to allow though if desired. */ if (*e) goto inval; chk_range: /* Finally, check for range limits. */ if (r >= lower && r <= upper) return r; range: bb_error_msg_and_die("number %s is not in %llu..%llu range", numstr, (unsigned long long)lower, (unsigned long long)upper); inval: bb_error_msg_and_die("invalid number '%s'", numstr); } unsigned type FAST_FUNC xstrtou(_range)(const char *numstr, int base, unsigned type lower, unsigned type upper) { return xstrtou(_range_sfx)(numstr, base, lower, upper, NULL); } unsigned type FAST_FUNC xstrtou(_sfx)(const char *numstr, int base, const struct suffix_mult *suffixes) { return xstrtou(_range_sfx)(numstr, base, 0, XSTR_UTYPE_MAX, suffixes); } unsigned type FAST_FUNC xstrtou()(const char *numstr, int base) { return xstrtou(_range_sfx)(numstr, base, 0, XSTR_UTYPE_MAX, NULL); } unsigned type FAST_FUNC xatou(_range_sfx)(const char *numstr, unsigned type lower, unsigned type upper, const struct suffix_mult *suffixes) { return xstrtou(_range_sfx)(numstr, 10, lower, upper, suffixes); } unsigned type FAST_FUNC xatou(_range)(const char *numstr, unsigned type lower, unsigned type upper) { return xstrtou(_range_sfx)(numstr, 10, lower, upper, NULL); } unsigned type FAST_FUNC xatou(_sfx)(const char *numstr, const struct suffix_mult *suffixes) { return xstrtou(_range_sfx)(numstr, 10, 0, XSTR_UTYPE_MAX, suffixes); } unsigned type FAST_FUNC xatou()(const char *numstr) { return xatou(_sfx)(numstr, NULL); } /* Signed ones */ type FAST_FUNC xstrto(_range_sfx)(const char *numstr, int base, type lower, type upper, const struct suffix_mult *suffixes) { unsigned type u = XSTR_TYPE_MAX; type r; const char *p = numstr; /* NB: if you'll decide to disallow '+': * at least renice applet needs to allow it */ if (p[0] == '+' || p[0] == '-') { ++p; if (p[0] == '-') ++u; /* = _MIN (01111... + 1 == 10000...) */ } r = xstrtou(_range_sfx)(p, base, 0, u, suffixes); if (*numstr == '-') { r = -r; } if (r < lower || r > upper) { bb_error_msg_and_die("number %s is not in %lld..%lld range", numstr, (long long)lower, (long long)upper); } return r; } type FAST_FUNC xstrto(_range)(const char *numstr, int base, type lower, type upper) { return xstrto(_range_sfx)(numstr, base, lower, upper, NULL); } type FAST_FUNC xstrto()(const char *numstr, int base) { return xstrto(_range_sfx)(numstr, base, XSTR_TYPE_MIN, XSTR_TYPE_MAX, NULL); } type FAST_FUNC xato(_range_sfx)(const char *numstr, type lower, type upper, const struct suffix_mult *suffixes) { return xstrto(_range_sfx)(numstr, 10, lower, upper, suffixes); } type FAST_FUNC xato(_range)(const char *numstr, type lower, type upper) { return xstrto(_range_sfx)(numstr, 10, lower, upper, NULL); } type FAST_FUNC xato(_sfx)(const char *numstr, const struct suffix_mult *suffixes) { return xstrto(_range_sfx)(numstr, 10, XSTR_TYPE_MIN, XSTR_TYPE_MAX, suffixes); } type FAST_FUNC xato()(const char *numstr) { return xstrto(_range_sfx)(numstr, 10, XSTR_TYPE_MIN, XSTR_TYPE_MAX, NULL); } #undef type #undef xstrtou #undef xstrto #undef xatou #undef xato #undef XSTR_UTYPE_MAX #undef XSTR_TYPE_MAX #undef XSTR_TYPE_MIN #undef XSTR_STRTOU