xref: /OK3568_Linux_fs/u-boot/lib/strto.c (revision 4882a59341e53eb6f0b4789bf948001014eff981)
1*4882a593Smuzhiyun /*
2*4882a593Smuzhiyun  *  linux/lib/vsprintf.c
3*4882a593Smuzhiyun  *
4*4882a593Smuzhiyun  *  Copyright (C) 1991, 1992  Linus Torvalds
5*4882a593Smuzhiyun  */
6*4882a593Smuzhiyun 
7*4882a593Smuzhiyun /* vsprintf.c -- Lars Wirzenius & Linus Torvalds. */
8*4882a593Smuzhiyun /*
9*4882a593Smuzhiyun  * Wirzenius wrote this portably, Torvalds fucked it up :-)
10*4882a593Smuzhiyun  */
11*4882a593Smuzhiyun 
12*4882a593Smuzhiyun #include <common.h>
13*4882a593Smuzhiyun #include <errno.h>
14*4882a593Smuzhiyun #include <linux/ctype.h>
15*4882a593Smuzhiyun 
16*4882a593Smuzhiyun /* from lib/kstrtox.c */
_parse_integer_fixup_radix(const char * s,unsigned int * base)17*4882a593Smuzhiyun static const char *_parse_integer_fixup_radix(const char *s, unsigned int *base)
18*4882a593Smuzhiyun {
19*4882a593Smuzhiyun 	if (*base == 0) {
20*4882a593Smuzhiyun 		if (s[0] == '0') {
21*4882a593Smuzhiyun 			if (tolower(s[1]) == 'x' && isxdigit(s[2]))
22*4882a593Smuzhiyun 				*base = 16;
23*4882a593Smuzhiyun 			else
24*4882a593Smuzhiyun 				*base = 8;
25*4882a593Smuzhiyun 		} else
26*4882a593Smuzhiyun 			*base = 10;
27*4882a593Smuzhiyun 	}
28*4882a593Smuzhiyun 	if (*base == 16 && s[0] == '0' && tolower(s[1]) == 'x')
29*4882a593Smuzhiyun 		s += 2;
30*4882a593Smuzhiyun 	return s;
31*4882a593Smuzhiyun }
32*4882a593Smuzhiyun 
simple_strtoul(const char * cp,char ** endp,unsigned int base)33*4882a593Smuzhiyun unsigned long simple_strtoul(const char *cp, char **endp,
34*4882a593Smuzhiyun 				unsigned int base)
35*4882a593Smuzhiyun {
36*4882a593Smuzhiyun 	unsigned long result = 0;
37*4882a593Smuzhiyun 	unsigned long value;
38*4882a593Smuzhiyun 
39*4882a593Smuzhiyun 	cp = _parse_integer_fixup_radix(cp, &base);
40*4882a593Smuzhiyun 
41*4882a593Smuzhiyun 	while (isxdigit(*cp) && (value = isdigit(*cp) ? *cp-'0' : (islower(*cp)
42*4882a593Smuzhiyun 	    ? toupper(*cp) : *cp)-'A'+10) < base) {
43*4882a593Smuzhiyun 		result = result*base + value;
44*4882a593Smuzhiyun 		cp++;
45*4882a593Smuzhiyun 	}
46*4882a593Smuzhiyun 
47*4882a593Smuzhiyun 	if (endp)
48*4882a593Smuzhiyun 		*endp = (char *)cp;
49*4882a593Smuzhiyun 
50*4882a593Smuzhiyun 	return result;
51*4882a593Smuzhiyun }
52*4882a593Smuzhiyun 
strict_strtoul(const char * cp,unsigned int base,unsigned long * res)53*4882a593Smuzhiyun int strict_strtoul(const char *cp, unsigned int base, unsigned long *res)
54*4882a593Smuzhiyun {
55*4882a593Smuzhiyun 	char *tail;
56*4882a593Smuzhiyun 	unsigned long val;
57*4882a593Smuzhiyun 	size_t len;
58*4882a593Smuzhiyun 
59*4882a593Smuzhiyun 	*res = 0;
60*4882a593Smuzhiyun 	len = strlen(cp);
61*4882a593Smuzhiyun 	if (len == 0)
62*4882a593Smuzhiyun 		return -EINVAL;
63*4882a593Smuzhiyun 
64*4882a593Smuzhiyun 	val = simple_strtoul(cp, &tail, base);
65*4882a593Smuzhiyun 	if (tail == cp)
66*4882a593Smuzhiyun 		return -EINVAL;
67*4882a593Smuzhiyun 
68*4882a593Smuzhiyun 	if ((*tail == '\0') ||
69*4882a593Smuzhiyun 		((len == (size_t)(tail - cp) + 1) && (*tail == '\n'))) {
70*4882a593Smuzhiyun 		*res = val;
71*4882a593Smuzhiyun 		return 0;
72*4882a593Smuzhiyun 	}
73*4882a593Smuzhiyun 
74*4882a593Smuzhiyun 	return -EINVAL;
75*4882a593Smuzhiyun }
76*4882a593Smuzhiyun 
simple_strtol(const char * cp,char ** endp,unsigned int base)77*4882a593Smuzhiyun long simple_strtol(const char *cp, char **endp, unsigned int base)
78*4882a593Smuzhiyun {
79*4882a593Smuzhiyun 	if (*cp == '-')
80*4882a593Smuzhiyun 		return -simple_strtoul(cp + 1, endp, base);
81*4882a593Smuzhiyun 
82*4882a593Smuzhiyun 	return simple_strtoul(cp, endp, base);
83*4882a593Smuzhiyun }
84*4882a593Smuzhiyun 
ustrtoul(const char * cp,char ** endp,unsigned int base)85*4882a593Smuzhiyun unsigned long ustrtoul(const char *cp, char **endp, unsigned int base)
86*4882a593Smuzhiyun {
87*4882a593Smuzhiyun 	unsigned long result = simple_strtoul(cp, endp, base);
88*4882a593Smuzhiyun 	switch (**endp) {
89*4882a593Smuzhiyun 	case 'G':
90*4882a593Smuzhiyun 		result *= 1024;
91*4882a593Smuzhiyun 		/* fall through */
92*4882a593Smuzhiyun 	case 'M':
93*4882a593Smuzhiyun 		result *= 1024;
94*4882a593Smuzhiyun 		/* fall through */
95*4882a593Smuzhiyun 	case 'K':
96*4882a593Smuzhiyun 	case 'k':
97*4882a593Smuzhiyun 		result *= 1024;
98*4882a593Smuzhiyun 		if ((*endp)[1] == 'i') {
99*4882a593Smuzhiyun 			if ((*endp)[2] == 'B')
100*4882a593Smuzhiyun 				(*endp) += 3;
101*4882a593Smuzhiyun 			else
102*4882a593Smuzhiyun 				(*endp) += 2;
103*4882a593Smuzhiyun 		}
104*4882a593Smuzhiyun 	}
105*4882a593Smuzhiyun 	return result;
106*4882a593Smuzhiyun }
107*4882a593Smuzhiyun 
ustrtoull(const char * cp,char ** endp,unsigned int base)108*4882a593Smuzhiyun unsigned long long ustrtoull(const char *cp, char **endp, unsigned int base)
109*4882a593Smuzhiyun {
110*4882a593Smuzhiyun 	unsigned long long result = simple_strtoull(cp, endp, base);
111*4882a593Smuzhiyun 	switch (**endp) {
112*4882a593Smuzhiyun 	case 'G':
113*4882a593Smuzhiyun 		result *= 1024;
114*4882a593Smuzhiyun 		/* fall through */
115*4882a593Smuzhiyun 	case 'M':
116*4882a593Smuzhiyun 		result *= 1024;
117*4882a593Smuzhiyun 		/* fall through */
118*4882a593Smuzhiyun 	case 'K':
119*4882a593Smuzhiyun 	case 'k':
120*4882a593Smuzhiyun 		result *= 1024;
121*4882a593Smuzhiyun 		if ((*endp)[1] == 'i') {
122*4882a593Smuzhiyun 			if ((*endp)[2] == 'B')
123*4882a593Smuzhiyun 				(*endp) += 3;
124*4882a593Smuzhiyun 			else
125*4882a593Smuzhiyun 				(*endp) += 2;
126*4882a593Smuzhiyun 		}
127*4882a593Smuzhiyun 	}
128*4882a593Smuzhiyun 	return result;
129*4882a593Smuzhiyun }
130*4882a593Smuzhiyun 
simple_strtoull(const char * cp,char ** endp,unsigned int base)131*4882a593Smuzhiyun unsigned long long simple_strtoull(const char *cp, char **endp,
132*4882a593Smuzhiyun 					unsigned int base)
133*4882a593Smuzhiyun {
134*4882a593Smuzhiyun 	unsigned long long result = 0, value;
135*4882a593Smuzhiyun 
136*4882a593Smuzhiyun 	cp = _parse_integer_fixup_radix(cp, &base);
137*4882a593Smuzhiyun 
138*4882a593Smuzhiyun 	while (isxdigit(*cp) && (value = isdigit(*cp) ? *cp - '0'
139*4882a593Smuzhiyun 		: (islower(*cp) ? toupper(*cp) : *cp) - 'A' + 10) < base) {
140*4882a593Smuzhiyun 		result = result * base + value;
141*4882a593Smuzhiyun 		cp++;
142*4882a593Smuzhiyun 	}
143*4882a593Smuzhiyun 
144*4882a593Smuzhiyun 	if (endp)
145*4882a593Smuzhiyun 		*endp = (char *) cp;
146*4882a593Smuzhiyun 
147*4882a593Smuzhiyun 	return result;
148*4882a593Smuzhiyun }
149*4882a593Smuzhiyun 
trailing_strtoln(const char * str,const char * end)150*4882a593Smuzhiyun long trailing_strtoln(const char *str, const char *end)
151*4882a593Smuzhiyun {
152*4882a593Smuzhiyun 	const char *p;
153*4882a593Smuzhiyun 
154*4882a593Smuzhiyun 	if (!end)
155*4882a593Smuzhiyun 		end = str + strlen(str);
156*4882a593Smuzhiyun 	if (isdigit(end[-1])) {
157*4882a593Smuzhiyun 		for (p = end - 1; p > str; p--) {
158*4882a593Smuzhiyun 			if (!isdigit(*p))
159*4882a593Smuzhiyun 				return simple_strtoul(p + 1, NULL, 10);
160*4882a593Smuzhiyun 		}
161*4882a593Smuzhiyun 	}
162*4882a593Smuzhiyun 
163*4882a593Smuzhiyun 	return -1;
164*4882a593Smuzhiyun }
165*4882a593Smuzhiyun 
trailing_strtol(const char * str)166*4882a593Smuzhiyun long trailing_strtol(const char *str)
167*4882a593Smuzhiyun {
168*4882a593Smuzhiyun 	return trailing_strtoln(str, NULL);
169*4882a593Smuzhiyun }
170