Merge branch 'x86-urgent-for-linus' of git://git.kernel.org/pub/scm/linux/kernel...
[sfrench/cifs-2.6.git] / lib / kstrtox.c
1 /*
2  * Convert integer string representation to an integer.
3  * If an integer doesn't fit into specified type, -E is returned.
4  *
5  * Integer starts with optional sign.
6  * kstrtou*() functions do not accept sign "-".
7  *
8  * Radix 0 means autodetection: leading "0x" implies radix 16,
9  * leading "0" implies radix 8, otherwise radix is 10.
10  * Autodetection hints work after optional sign, but not before.
11  *
12  * If -E is returned, result is not touched.
13  */
14 #include <linux/ctype.h>
15 #include <linux/errno.h>
16 #include <linux/kernel.h>
17 #include <linux/math64.h>
18 #include <linux/export.h>
19 #include <linux/types.h>
20 #include <asm/uaccess.h>
21 #include "kstrtox.h"
22
23 const char *_parse_integer_fixup_radix(const char *s, unsigned int *base)
24 {
25         if (*base == 0) {
26                 if (s[0] == '0') {
27                         if (_tolower(s[1]) == 'x' && isxdigit(s[2]))
28                                 *base = 16;
29                         else
30                                 *base = 8;
31                 } else
32                         *base = 10;
33         }
34         if (*base == 16 && s[0] == '0' && _tolower(s[1]) == 'x')
35                 s += 2;
36         return s;
37 }
38
39 /*
40  * Convert non-negative integer string representation in explicitly given radix
41  * to an integer.
42  * Return number of characters consumed maybe or-ed with overflow bit.
43  * If overflow occurs, result integer (incorrect) is still returned.
44  *
45  * Don't you dare use this function.
46  */
47 unsigned int _parse_integer(const char *s, unsigned int base, unsigned long long *p)
48 {
49         unsigned long long res;
50         unsigned int rv;
51         int overflow;
52
53         res = 0;
54         rv = 0;
55         overflow = 0;
56         while (*s) {
57                 unsigned int val;
58
59                 if ('0' <= *s && *s <= '9')
60                         val = *s - '0';
61                 else if ('a' <= _tolower(*s) && _tolower(*s) <= 'f')
62                         val = _tolower(*s) - 'a' + 10;
63                 else
64                         break;
65
66                 if (val >= base)
67                         break;
68                 /*
69                  * Check for overflow only if we are within range of
70                  * it in the max base we support (16)
71                  */
72                 if (unlikely(res & (~0ull << 60))) {
73                         if (res > div_u64(ULLONG_MAX - val, base))
74                                 overflow = 1;
75                 }
76                 res = res * base + val;
77                 rv++;
78                 s++;
79         }
80         *p = res;
81         if (overflow)
82                 rv |= KSTRTOX_OVERFLOW;
83         return rv;
84 }
85
86 static int _kstrtoull(const char *s, unsigned int base, unsigned long long *res)
87 {
88         unsigned long long _res;
89         unsigned int rv;
90
91         s = _parse_integer_fixup_radix(s, &base);
92         rv = _parse_integer(s, base, &_res);
93         if (rv & KSTRTOX_OVERFLOW)
94                 return -ERANGE;
95         if (rv == 0)
96                 return -EINVAL;
97         s += rv;
98         if (*s == '\n')
99                 s++;
100         if (*s)
101                 return -EINVAL;
102         *res = _res;
103         return 0;
104 }
105
106 /**
107  * kstrtoull - convert a string to an unsigned long long
108  * @s: The start of the string. The string must be null-terminated, and may also
109  *  include a single newline before its terminating null. The first character
110  *  may also be a plus sign, but not a minus sign.
111  * @base: The number base to use. The maximum supported base is 16. If base is
112  *  given as 0, then the base of the string is automatically detected with the
113  *  conventional semantics - If it begins with 0x the number will be parsed as a
114  *  hexadecimal (case insensitive), if it otherwise begins with 0, it will be
115  *  parsed as an octal number. Otherwise it will be parsed as a decimal.
116  * @res: Where to write the result of the conversion on success.
117  *
118  * Returns 0 on success, -ERANGE on overflow and -EINVAL on parsing error.
119  * Used as a replacement for the obsolete simple_strtoull. Return code must
120  * be checked.
121  */
122 int kstrtoull(const char *s, unsigned int base, unsigned long long *res)
123 {
124         if (s[0] == '+')
125                 s++;
126         return _kstrtoull(s, base, res);
127 }
128 EXPORT_SYMBOL(kstrtoull);
129
130 /**
131  * kstrtoll - convert a string to a long long
132  * @s: The start of the string. The string must be null-terminated, and may also
133  *  include a single newline before its terminating null. The first character
134  *  may also be a plus sign or a minus sign.
135  * @base: The number base to use. The maximum supported base is 16. If base is
136  *  given as 0, then the base of the string is automatically detected with the
137  *  conventional semantics - If it begins with 0x the number will be parsed as a
138  *  hexadecimal (case insensitive), if it otherwise begins with 0, it will be
139  *  parsed as an octal number. Otherwise it will be parsed as a decimal.
140  * @res: Where to write the result of the conversion on success.
141  *
142  * Returns 0 on success, -ERANGE on overflow and -EINVAL on parsing error.
143  * Used as a replacement for the obsolete simple_strtoull. Return code must
144  * be checked.
145  */
146 int kstrtoll(const char *s, unsigned int base, long long *res)
147 {
148         unsigned long long tmp;
149         int rv;
150
151         if (s[0] == '-') {
152                 rv = _kstrtoull(s + 1, base, &tmp);
153                 if (rv < 0)
154                         return rv;
155                 if ((long long)(-tmp) >= 0)
156                         return -ERANGE;
157                 *res = -tmp;
158         } else {
159                 rv = kstrtoull(s, base, &tmp);
160                 if (rv < 0)
161                         return rv;
162                 if ((long long)tmp < 0)
163                         return -ERANGE;
164                 *res = tmp;
165         }
166         return 0;
167 }
168 EXPORT_SYMBOL(kstrtoll);
169
170 /* Internal, do not use. */
171 int _kstrtoul(const char *s, unsigned int base, unsigned long *res)
172 {
173         unsigned long long tmp;
174         int rv;
175
176         rv = kstrtoull(s, base, &tmp);
177         if (rv < 0)
178                 return rv;
179         if (tmp != (unsigned long long)(unsigned long)tmp)
180                 return -ERANGE;
181         *res = tmp;
182         return 0;
183 }
184 EXPORT_SYMBOL(_kstrtoul);
185
186 /* Internal, do not use. */
187 int _kstrtol(const char *s, unsigned int base, long *res)
188 {
189         long long tmp;
190         int rv;
191
192         rv = kstrtoll(s, base, &tmp);
193         if (rv < 0)
194                 return rv;
195         if (tmp != (long long)(long)tmp)
196                 return -ERANGE;
197         *res = tmp;
198         return 0;
199 }
200 EXPORT_SYMBOL(_kstrtol);
201
202 /**
203  * kstrtouint - convert a string to an unsigned int
204  * @s: The start of the string. The string must be null-terminated, and may also
205  *  include a single newline before its terminating null. The first character
206  *  may also be a plus sign, but not a minus sign.
207  * @base: The number base to use. The maximum supported base is 16. If base is
208  *  given as 0, then the base of the string is automatically detected with the
209  *  conventional semantics - If it begins with 0x the number will be parsed as a
210  *  hexadecimal (case insensitive), if it otherwise begins with 0, it will be
211  *  parsed as an octal number. Otherwise it will be parsed as a decimal.
212  * @res: Where to write the result of the conversion on success.
213  *
214  * Returns 0 on success, -ERANGE on overflow and -EINVAL on parsing error.
215  * Used as a replacement for the obsolete simple_strtoull. Return code must
216  * be checked.
217  */
218 int kstrtouint(const char *s, unsigned int base, unsigned int *res)
219 {
220         unsigned long long tmp;
221         int rv;
222
223         rv = kstrtoull(s, base, &tmp);
224         if (rv < 0)
225                 return rv;
226         if (tmp != (unsigned long long)(unsigned int)tmp)
227                 return -ERANGE;
228         *res = tmp;
229         return 0;
230 }
231 EXPORT_SYMBOL(kstrtouint);
232
233 /**
234  * kstrtoint - convert a string to an int
235  * @s: The start of the string. The string must be null-terminated, and may also
236  *  include a single newline before its terminating null. The first character
237  *  may also be a plus sign or a minus sign.
238  * @base: The number base to use. The maximum supported base is 16. If base is
239  *  given as 0, then the base of the string is automatically detected with the
240  *  conventional semantics - If it begins with 0x the number will be parsed as a
241  *  hexadecimal (case insensitive), if it otherwise begins with 0, it will be
242  *  parsed as an octal number. Otherwise it will be parsed as a decimal.
243  * @res: Where to write the result of the conversion on success.
244  *
245  * Returns 0 on success, -ERANGE on overflow and -EINVAL on parsing error.
246  * Used as a replacement for the obsolete simple_strtoull. Return code must
247  * be checked.
248  */
249 int kstrtoint(const char *s, unsigned int base, int *res)
250 {
251         long long tmp;
252         int rv;
253
254         rv = kstrtoll(s, base, &tmp);
255         if (rv < 0)
256                 return rv;
257         if (tmp != (long long)(int)tmp)
258                 return -ERANGE;
259         *res = tmp;
260         return 0;
261 }
262 EXPORT_SYMBOL(kstrtoint);
263
264 int kstrtou16(const char *s, unsigned int base, u16 *res)
265 {
266         unsigned long long tmp;
267         int rv;
268
269         rv = kstrtoull(s, base, &tmp);
270         if (rv < 0)
271                 return rv;
272         if (tmp != (unsigned long long)(u16)tmp)
273                 return -ERANGE;
274         *res = tmp;
275         return 0;
276 }
277 EXPORT_SYMBOL(kstrtou16);
278
279 int kstrtos16(const char *s, unsigned int base, s16 *res)
280 {
281         long long tmp;
282         int rv;
283
284         rv = kstrtoll(s, base, &tmp);
285         if (rv < 0)
286                 return rv;
287         if (tmp != (long long)(s16)tmp)
288                 return -ERANGE;
289         *res = tmp;
290         return 0;
291 }
292 EXPORT_SYMBOL(kstrtos16);
293
294 int kstrtou8(const char *s, unsigned int base, u8 *res)
295 {
296         unsigned long long tmp;
297         int rv;
298
299         rv = kstrtoull(s, base, &tmp);
300         if (rv < 0)
301                 return rv;
302         if (tmp != (unsigned long long)(u8)tmp)
303                 return -ERANGE;
304         *res = tmp;
305         return 0;
306 }
307 EXPORT_SYMBOL(kstrtou8);
308
309 int kstrtos8(const char *s, unsigned int base, s8 *res)
310 {
311         long long tmp;
312         int rv;
313
314         rv = kstrtoll(s, base, &tmp);
315         if (rv < 0)
316                 return rv;
317         if (tmp != (long long)(s8)tmp)
318                 return -ERANGE;
319         *res = tmp;
320         return 0;
321 }
322 EXPORT_SYMBOL(kstrtos8);
323
324 #define kstrto_from_user(f, g, type)                                    \
325 int f(const char __user *s, size_t count, unsigned int base, type *res) \
326 {                                                                       \
327         /* sign, base 2 representation, newline, terminator */          \
328         char buf[1 + sizeof(type) * 8 + 1 + 1];                         \
329                                                                         \
330         count = min(count, sizeof(buf) - 1);                            \
331         if (copy_from_user(buf, s, count))                              \
332                 return -EFAULT;                                         \
333         buf[count] = '\0';                                              \
334         return g(buf, base, res);                                       \
335 }                                                                       \
336 EXPORT_SYMBOL(f)
337
338 kstrto_from_user(kstrtoull_from_user,   kstrtoull,      unsigned long long);
339 kstrto_from_user(kstrtoll_from_user,    kstrtoll,       long long);
340 kstrto_from_user(kstrtoul_from_user,    kstrtoul,       unsigned long);
341 kstrto_from_user(kstrtol_from_user,     kstrtol,        long);
342 kstrto_from_user(kstrtouint_from_user,  kstrtouint,     unsigned int);
343 kstrto_from_user(kstrtoint_from_user,   kstrtoint,      int);
344 kstrto_from_user(kstrtou16_from_user,   kstrtou16,      u16);
345 kstrto_from_user(kstrtos16_from_user,   kstrtos16,      s16);
346 kstrto_from_user(kstrtou8_from_user,    kstrtou8,       u8);
347 kstrto_from_user(kstrtos8_from_user,    kstrtos8,       s8);