xref: /freebsd/contrib/bmake/_strtol.h (revision a18b956b73cee784e5c422d20fd0e4dabebd7eee)
1 /* $NetBSD: _strtol.h,v 1.11 2017/07/06 21:08:44 joerg Exp $ */
2 
3 /*-
4  * Copyright (c) 1990, 1993
5  *	The Regents of the University of California.  All rights reserved.
6  *
7  * Redistribution and use in source and binary forms, with or without
8  * modification, are permitted provided that the following conditions
9  * are met:
10  * 1. Redistributions of source code must retain the above copyright
11  *    notice, this list of conditions and the following disclaimer.
12  * 2. Redistributions in binary form must reproduce the above copyright
13  *    notice, this list of conditions and the following disclaimer in the
14  *    documentation and/or other materials provided with the distribution.
15  * 3. Neither the name of the University nor the names of its contributors
16  *    may be used to endorse or promote products derived from this software
17  *    without specific prior written permission.
18  *
19  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
20  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
21  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
22  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
23  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
24  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
25  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
26  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
27  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
28  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
29  * SUCH DAMAGE.
30  *
31  * Original version ID:
32  * NetBSD: src/lib/libc/locale/_wcstol.h,v 1.2 2003/08/07 16:43:03 agc Exp
33  */
34 
35 /*
36  * function template for strtol, strtoll and strtoimax.
37  *
38  * parameters:
39  *	_FUNCNAME : function name
40  *      __INT     : return type
41  *      __INT_MIN : lower limit of the return type
42  *      __INT_MAX : upper limit of the return type
43  */
44 #if defined(_KERNEL) || defined(_STANDALONE) || defined(HAVE_NBTOOL_CONFIG_H) || defined(BCS_ONLY)
45 __INT
46 _FUNCNAME(const char *nptr, char **endptr, int base)
47 #else
48 #include <locale.h>
49 #include "setlocale_local.h"
50 #define INT_FUNCNAME_(pre, name, post)	pre ## name ## post
51 #define INT_FUNCNAME(pre, name, post)	INT_FUNCNAME_(pre, name, post)
52 
53 static __INT
54 INT_FUNCNAME(_int_, _FUNCNAME, _l)(const char *nptr, char **endptr,
55 				   int base, locale_t loc)
56 #endif
57 {
58 	const char *s;
59 	__INT acc, cutoff;
60 	unsigned char c;
61 	int i, neg, any, cutlim;
62 
63 	_DIAGASSERT(nptr != NULL);
64 	/* endptr may be NULL */
65 
66 	/* check base value */
67 	if (base && (base < 2 || base > 36)) {
68 #if !defined(_KERNEL) && !defined(_STANDALONE)
69 		errno = EINVAL;
70 		if (endptr != NULL)
71 			/* LINTED interface specification */
72 			*endptr = __UNCONST(nptr);
73 		return 0;
74 #else
75 		panic("%s: invalid base %d", __func__, base);
76 #endif
77 	}
78 
79 	/*
80 	 * Skip white space and pick up leading +/- sign if any.
81 	 * If base is 0, allow 0x for hex and 0 for octal, else
82 	 * assume decimal; if base is already 16, allow 0x.
83 	 */
84 	s = nptr;
85 #if defined(_KERNEL) || defined(_STANDALONE) || \
86     defined(HAVE_NBTOOL_CONFIG_H) || defined(BCS_ONLY)
87 	do {
88 		c = *s++;
89 	} while (isspace(c));
90 #else
91 	do {
92 		c = *s++;
93 	} while (isspace_l(c, loc));
94 #endif
95 	if (c == '-') {
96 		neg = 1;
97 		c = *s++;
98 	} else {
99 		neg = 0;
100 		if (c == '+')
101 			c = *s++;
102 	}
103 	if ((base == 0 || base == 16) &&
104 	    c == '0' && (*s == 'x' || *s == 'X') &&
105 	    ((s[1] >= '0' && s[1] <= '9') ||
106 	     (s[1] >= 'a' && s[1] <= 'f') ||
107 	     (s[1] >= 'A' && s[1] <= 'F'))) {
108 		c = s[1];
109 		s += 2;
110 		base = 16;
111 #if 0
112 	} else if ((base == 0 || base == 2) &&
113 	    c == '0' && (*s == 'b' || *s == 'B') &&
114 	    (s[1] >= '0' && s[1] <= '1')) {
115 		c = s[1];
116 		s += 2;
117 		base = 2;
118 #endif
119 	} else if (base == 0)
120 		base = (c == '0' ? 8 : 10);
121 
122 	/*
123 	 * Compute the cutoff value between legal numbers and illegal
124 	 * numbers.  That is the largest legal value, divided by the
125 	 * base.  An input number that is greater than this value, if
126 	 * followed by a legal input character, is too big.  One that
127 	 * is equal to this value may be valid or not; the limit
128 	 * between valid and invalid numbers is then based on the last
129 	 * digit.  For instance, if the range for longs is
130 	 * [-2147483648..2147483647] and the input base is 10,
131 	 * cutoff will be set to 214748364 and cutlim to either
132 	 * 7 (neg==0) or 8 (neg==1), meaning that if we have accumulated
133 	 * a value > 214748364, or equal but the next digit is > 7 (or 8),
134 	 * the number is too big, and we will return a range error.
135 	 *
136 	 * Set any if any `digits' consumed; make it negative to indicate
137 	 * overflow.
138 	 */
139 	cutoff = (__INT)(neg ? __INT_MIN : __INT_MAX);
140 	cutlim = (int)(cutoff % base);
141 	cutoff /= base;
142 	if (neg) {
143 		if (cutlim > 0) {
144 			cutlim -= base;
145 			cutoff += 1;
146 		}
147 		cutlim = -cutlim;
148 	}
149 	for (acc = 0, any = 0;; c = *s++) {
150 		if (c >= '0' && c <= '9')
151 			i = c - '0';
152 		else if (c >= 'a' && c <= 'z')
153 			i = (c - 'a') + 10;
154 		else if (c >= 'A' && c <= 'Z')
155 			i = (c - 'A') + 10;
156 		else
157 			break;
158 		if (i >= base)
159 			break;
160 		if (any < 0)
161 			continue;
162 		if (neg) {
163 			if (acc < cutoff || (acc == cutoff && i > cutlim)) {
164 				acc = __INT_MIN;
165 #if !defined(_KERNEL) && !defined(_STANDALONE)
166 				any = -1;
167 				errno = ERANGE;
168 #else
169 				any = 0;
170 				break;
171 #endif
172 			} else {
173 				any = 1;
174 				acc *= base;
175 				acc -= i;
176 			}
177 		} else {
178 			if (acc > cutoff || (acc == cutoff && i > cutlim)) {
179 				acc = __INT_MAX;
180 #if !defined(_KERNEL) && !defined(_STANDALONE)
181 				any = -1;
182 				errno = ERANGE;
183 #else
184 				any = 0;
185 				break;
186 #endif
187 			} else {
188 				any = 1;
189 				acc *= base;
190 				acc += i;
191 			}
192 		}
193 	}
194 	if (endptr != NULL)
195 		/* LINTED interface specification */
196 		*endptr = __UNCONST(any ? s - 1 : nptr);
197 	return(acc);
198 }
199 
200 #if !defined(_KERNEL) && !defined(_STANDALONE) && \
201     !defined(HAVE_NBTOOL_CONFIG_H) && !defined(BCS_ONLY)
202 __INT
203 _FUNCNAME(const char *nptr, char **endptr, int base)
204 {
205 	return INT_FUNCNAME(_int_, _FUNCNAME, _l)(nptr, endptr, base, _current_locale());
206 }
207 
208 __INT
209 INT_FUNCNAME(, _FUNCNAME, _l)(const char *nptr, char **endptr, int base, locale_t loc)
210 {
211 	return INT_FUNCNAME(_int_, _FUNCNAME, _l)(nptr, endptr, base, loc);
212 }
213 #endif
214