1 | /* |
---|
2 | You need to define the following (example): |
---|
3 | |
---|
4 | #define type long |
---|
5 | #define xstrtou(rest) xstrtoul##rest |
---|
6 | #define xstrto(rest) xstrtol##rest |
---|
7 | #define xatou(rest) xatoul##rest |
---|
8 | #define xato(rest) xatol##rest |
---|
9 | #define XSTR_UTYPE_MAX ULONG_MAX |
---|
10 | #define XSTR_TYPE_MAX LONG_MAX |
---|
11 | #define XSTR_TYPE_MIN LONG_MIN |
---|
12 | #define XSTR_STRTOU strtoul |
---|
13 | */ |
---|
14 | |
---|
15 | unsigned type xstrtou(_range_sfx)(const char *numstr, int base, |
---|
16 | unsigned type lower, |
---|
17 | unsigned type upper, |
---|
18 | const struct suffix_mult *suffixes) |
---|
19 | { |
---|
20 | unsigned type r; |
---|
21 | int old_errno; |
---|
22 | char *e; |
---|
23 | |
---|
24 | /* Disallow '-' and any leading whitespace. Speed isn't critical here |
---|
25 | * since we're parsing commandline args. So make sure we get the |
---|
26 | * actual isspace function rather than a lnumstrer macro implementaion. */ |
---|
27 | if (*numstr == '-' || *numstr == '+' || (isspace)(*numstr)) |
---|
28 | goto inval; |
---|
29 | |
---|
30 | /* Since this is a lib function, we're not allowed to reset errno to 0. |
---|
31 | * Doing so could break an app that is deferring checking of errno. |
---|
32 | * So, save the old value so that we can restore it if successful. */ |
---|
33 | old_errno = errno; |
---|
34 | errno = 0; |
---|
35 | r = XSTR_STRTOU(numstr, &e, base); |
---|
36 | /* Do the initial validity check. Note: The standards do not |
---|
37 | * guarantee that errno is set if no digits were found. So we |
---|
38 | * must test for this explicitly. */ |
---|
39 | if (errno || numstr == e) |
---|
40 | goto inval; /* error / no digits / illegal trailing chars */ |
---|
41 | |
---|
42 | errno = old_errno; /* Ok. So restore errno. */ |
---|
43 | |
---|
44 | /* Do optional suffix parsing. Allow 'empty' suffix tables. |
---|
45 | * Note that we also allow nul suffixes with associated multipliers, |
---|
46 | * to allow for scaling of the numstr by some default multiplier. */ |
---|
47 | if (suffixes) { |
---|
48 | while (suffixes->mult) { |
---|
49 | if (strcmp(suffixes->suffix, e) == 0) { |
---|
50 | if (XSTR_UTYPE_MAX / suffixes->mult < r) |
---|
51 | goto range; /* overflow! */ |
---|
52 | r *= suffixes->mult; |
---|
53 | goto chk_range; |
---|
54 | } |
---|
55 | ++suffixes; |
---|
56 | } |
---|
57 | } |
---|
58 | |
---|
59 | /* Note: trailing space is an error. |
---|
60 | It would be easy enough to allow though if desired. */ |
---|
61 | if (*e) |
---|
62 | goto inval; |
---|
63 | chk_range: |
---|
64 | /* Finally, check for range limits. */ |
---|
65 | if (r >= lower && r <= upper) |
---|
66 | return r; |
---|
67 | range: |
---|
68 | bb_error_msg_and_die("number %s is not in %llu..%llu range", |
---|
69 | numstr, (unsigned long long)lower, |
---|
70 | (unsigned long long)upper); |
---|
71 | inval: |
---|
72 | bb_error_msg_and_die("invalid number '%s'", numstr); |
---|
73 | } |
---|
74 | |
---|
75 | unsigned type xstrtou(_range)(const char *numstr, int base, |
---|
76 | unsigned type lower, |
---|
77 | unsigned type upper) |
---|
78 | { |
---|
79 | return xstrtou(_range_sfx)(numstr, base, lower, upper, NULL); |
---|
80 | } |
---|
81 | |
---|
82 | unsigned type xstrtou(_sfx)(const char *numstr, int base, |
---|
83 | const struct suffix_mult *suffixes) |
---|
84 | { |
---|
85 | return xstrtou(_range_sfx)(numstr, base, 0, XSTR_UTYPE_MAX, suffixes); |
---|
86 | } |
---|
87 | |
---|
88 | unsigned type xstrtou()(const char *numstr, int base) |
---|
89 | { |
---|
90 | return xstrtou(_range_sfx)(numstr, base, 0, XSTR_UTYPE_MAX, NULL); |
---|
91 | } |
---|
92 | |
---|
93 | unsigned type xatou(_range_sfx)(const char *numstr, |
---|
94 | unsigned type lower, |
---|
95 | unsigned type upper, |
---|
96 | const struct suffix_mult *suffixes) |
---|
97 | { |
---|
98 | return xstrtou(_range_sfx)(numstr, 10, lower, upper, suffixes); |
---|
99 | } |
---|
100 | |
---|
101 | unsigned type xatou(_range)(const char *numstr, |
---|
102 | unsigned type lower, |
---|
103 | unsigned type upper) |
---|
104 | { |
---|
105 | return xstrtou(_range_sfx)(numstr, 10, lower, upper, NULL); |
---|
106 | } |
---|
107 | |
---|
108 | unsigned type xatou(_sfx)(const char *numstr, |
---|
109 | const struct suffix_mult *suffixes) |
---|
110 | { |
---|
111 | return xstrtou(_range_sfx)(numstr, 10, 0, XSTR_UTYPE_MAX, suffixes); |
---|
112 | } |
---|
113 | |
---|
114 | unsigned type xatou()(const char *numstr) |
---|
115 | { |
---|
116 | return xatou(_sfx)(numstr, NULL); |
---|
117 | } |
---|
118 | |
---|
119 | /* Signed ones */ |
---|
120 | |
---|
121 | type xstrto(_range_sfx)(const char *numstr, int base, |
---|
122 | type lower, |
---|
123 | type upper, |
---|
124 | const struct suffix_mult *suffixes) |
---|
125 | { |
---|
126 | unsigned type u = XSTR_TYPE_MAX; |
---|
127 | type r; |
---|
128 | const char *p = numstr; |
---|
129 | |
---|
130 | if (p[0] == '-') { |
---|
131 | ++p; |
---|
132 | ++u; /* two's complement */ |
---|
133 | } |
---|
134 | |
---|
135 | r = xstrtou(_range_sfx)(p, base, 0, u, suffixes); |
---|
136 | |
---|
137 | if (*numstr == '-') { |
---|
138 | r = -r; |
---|
139 | } |
---|
140 | |
---|
141 | if (r < lower || r > upper) { |
---|
142 | bb_error_msg_and_die("number %s is not in %lld..%lld range", |
---|
143 | numstr, (long long)lower, (long long)upper); |
---|
144 | } |
---|
145 | |
---|
146 | return r; |
---|
147 | } |
---|
148 | |
---|
149 | type xstrto(_range)(const char *numstr, int base, type lower, type upper) |
---|
150 | { |
---|
151 | return xstrto(_range_sfx)(numstr, base, lower, upper, NULL); |
---|
152 | } |
---|
153 | |
---|
154 | type xato(_range_sfx)(const char *numstr, |
---|
155 | type lower, |
---|
156 | type upper, |
---|
157 | const struct suffix_mult *suffixes) |
---|
158 | { |
---|
159 | return xstrto(_range_sfx)(numstr, 10, lower, upper, suffixes); |
---|
160 | } |
---|
161 | |
---|
162 | type xato(_range)(const char *numstr, type lower, type upper) |
---|
163 | { |
---|
164 | return xstrto(_range_sfx)(numstr, 10, lower, upper, NULL); |
---|
165 | } |
---|
166 | |
---|
167 | type xato(_sfx)(const char *numstr, const struct suffix_mult *suffixes) |
---|
168 | { |
---|
169 | return xstrto(_range_sfx)(numstr, 10, XSTR_TYPE_MIN, XSTR_TYPE_MAX, suffixes); |
---|
170 | } |
---|
171 | |
---|
172 | type xato()(const char *numstr) |
---|
173 | { |
---|
174 | return xstrto(_range_sfx)(numstr, 10, XSTR_TYPE_MIN, XSTR_TYPE_MAX, NULL); |
---|
175 | } |
---|
176 | |
---|
177 | #undef type |
---|
178 | #undef xstrtou |
---|
179 | #undef xstrto |
---|
180 | #undef xatou |
---|
181 | #undef xato |
---|
182 | #undef XSTR_UTYPE_MAX |
---|
183 | #undef XSTR_TYPE_MAX |
---|
184 | #undef XSTR_TYPE_MIN |
---|
185 | #undef XSTR_STRTOU |
---|