1 /* $NetBSD: strsuftoll.c,v 1.9 2011/10/22 22:08:47 christos Exp $ */
2 /*-
3 * Copyright (c) 2001-2002,2004 The NetBSD Foundation, Inc.
4 * All rights reserved.
5 *
6 * This code is derived from software contributed to The NetBSD Foundation
7 * by Luke Mewburn.
8 *
9 * Redistribution and use in source and binary forms, with or without
10 * modification, are permitted provided that the following conditions
11 * are met:
12 * 1. Redistributions of source code must retain the above copyright
13 * notice, this list of conditions and the following disclaimer.
14 * 2. Redistributions in binary form must reproduce the above copyright
15 * notice, this list of conditions and the following disclaimer in the
16 * documentation and/or other materials provided with the distribution.
17 *
18 * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
19 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
20 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
21 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
22 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
23 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
24 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
25 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
26 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
27 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
28 * POSSIBILITY OF SUCH DAMAGE.
29 */
30 /*-
31 * Copyright (c) 1991, 1993, 1994
32 * The Regents of the University of California. All rights reserved.
33 *
34 * This code is derived from software contributed to Berkeley by
35 * Keith Muller of the University of California, San Diego and Lance
36 * Visser of Convex Computer Corporation.
37 *
38 * Redistribution and use in source and binary forms, with or without
39 * modification, are permitted provided that the following conditions
40 * are met:
41 * 1. Redistributions of source code must retain the above copyright
42 * notice, this list of conditions and the following disclaimer.
43 * 2. Redistributions in binary form must reproduce the above copyright
44 * notice, this list of conditions and the following disclaimer in the
45 * documentation and/or other materials provided with the distribution.
46 * 3. Neither the name of the University nor the names of its contributors
47 * may be used to endorse or promote products derived from this software
48 * without specific prior written permission.
49 *
50 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
51 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
52 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
53 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
54 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
55 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
56 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
57 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
58 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
59 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
60 * SUCH DAMAGE.
61 */
62
63 #if HAVE_NBTOOL_CONFIG_H
64 #include "nbtool_config.h"
65 #endif
66
67 #include <sys/cdefs.h>
68
69 #if defined(LIBC_SCCS) && !defined(lint)
70 __RCSID("$NetBSD: strsuftoll.c,v 1.9 2011/10/22 22:08:47 christos Exp $");
71 #endif /* LIBC_SCCS and not lint */
72
73 #ifdef _LIBC
74 #include "namespace.h"
75 #endif
76
77 #if !HAVE_STRSUFTOLL
78
79 #include <sys/types.h>
80 #include <sys/time.h>
81
82 #include <assert.h>
83 #include <ctype.h>
84 #include <err.h>
85 #include <errno.h>
86 #include <limits.h>
87 #include <stdio.h>
88 #include <stdlib.h>
89 #include <string.h>
90
91 #ifdef _LIBC
92 # ifdef __weak_alias
__weak_alias(strsuftoll,_strsuftoll)93 __weak_alias(strsuftoll, _strsuftoll)
94 __weak_alias(strsuftollx, _strsuftollx)
95 # endif
96 #endif /* LIBC */
97
98 /*
99 * Convert an expression of the following forms to a (u)int64_t.
100 * 1) A positive decimal number.
101 * 2) A positive decimal number followed by a b (mult by 512).
102 * 3) A positive decimal number followed by a k (mult by 1024).
103 * 4) A positive decimal number followed by a m (mult by 1048576).
104 * 5) A positive decimal number followed by a g (mult by 1073741824).
105 * 6) A positive decimal number followed by a t (mult by 1099511627776).
106 * 7) A positive decimal number followed by a w (mult by sizeof int)
107 * 8) Two or more positive decimal numbers (with/without k,b or w).
108 * separated by x (also * for backwards compatibility), specifying
109 * the product of the indicated values.
110 * Returns the result upon successful conversion, or exits with an
111 * appropriate error.
112 *
113 */
114 /* LONGLONG */
115 long long
116 strsuftoll(const char *desc, const char *val,
117 long long min, long long max)
118 {
119 long long result;
120 char errbuf[100];
121
122 result = strsuftollx(desc, val, min, max, errbuf, sizeof(errbuf));
123 if (*errbuf != '\0')
124 errx(EXIT_FAILURE, "%s", errbuf);
125 return result;
126 }
127
128 /*
129 * As strsuftoll(), but returns the error message into the provided buffer
130 * rather than exiting with it.
131 */
132 /* LONGLONG */
133 static long long
__strsuftollx(const char * desc,const char * val,long long min,long long max,char * ebuf,size_t ebuflen,size_t depth)134 __strsuftollx(const char *desc, const char *val,
135 long long min, long long max, char *ebuf, size_t ebuflen, size_t depth)
136 {
137 long long num, t;
138 char *expr;
139
140 _DIAGASSERT(desc != NULL);
141 _DIAGASSERT(val != NULL);
142 _DIAGASSERT(ebuf != NULL);
143
144 if (depth > 16) {
145 snprintf(ebuf, ebuflen, "%s: Recursion limit exceeded", desc);
146 return 0;
147 }
148
149 while (isspace((unsigned char)*val)) /* Skip leading space */
150 val++;
151
152 errno = 0;
153 num = strtoll(val, &expr, 10);
154 if (errno == ERANGE)
155 goto erange; /* Overflow */
156
157 if (expr == val) /* No digits */
158 goto badnum;
159
160 switch (*expr) {
161 case 'b':
162 t = num;
163 num *= 512; /* 1 block */
164 if (t > num)
165 goto erange;
166 ++expr;
167 break;
168 case 'k':
169 t = num;
170 num *= 1024; /* 1 kibibyte */
171 if (t > num)
172 goto erange;
173 ++expr;
174 break;
175 case 'm':
176 t = num;
177 num *= 1048576; /* 1 mebibyte */
178 if (t > num)
179 goto erange;
180 ++expr;
181 break;
182 case 'g':
183 t = num;
184 num *= 1073741824; /* 1 gibibyte */
185 if (t > num)
186 goto erange;
187 ++expr;
188 break;
189 case 't':
190 t = num;
191 num *= 1099511627776LL; /* 1 tebibyte */
192 if (t > num)
193 goto erange;
194 ++expr;
195 break;
196 case 'w':
197 t = num;
198 num *= sizeof(int); /* 1 word */
199 if (t > num)
200 goto erange;
201 ++expr;
202 break;
203 }
204
205 switch (*expr) {
206 case '\0':
207 break;
208 case '*': /* Backward compatible */
209 case 'x':
210 t = num;
211 num *= __strsuftollx(desc, expr + 1, min, max, ebuf, ebuflen,
212 depth + 1);
213 if (*ebuf != '\0')
214 return 0;
215 if (t > num) {
216 erange:
217 errno = ERANGE;
218 snprintf(ebuf, ebuflen, "%s: %s", desc, strerror(errno));
219 return 0;
220 }
221 break;
222 default:
223 badnum:
224 snprintf(ebuf, ebuflen, "%s `%s': illegal number", desc, val);
225 return 0;
226 }
227 if (num < min) {
228 /* LONGLONG */
229 snprintf(ebuf, ebuflen, "%s %lld is less than %lld.",
230 desc, (long long)num, (long long)min);
231 return 0;
232 }
233 if (num > max) {
234 /* LONGLONG */
235 snprintf(ebuf, ebuflen, "%s %lld is greater than %lld.",
236 desc, (long long)num, (long long)max);
237 return 0;
238 }
239 *ebuf = '\0';
240 return num;
241 }
242
243 long long
strsuftollx(const char * desc,const char * val,long long min,long long max,char * ebuf,size_t ebuflen)244 strsuftollx(const char *desc, const char *val,
245 long long min, long long max, char *ebuf, size_t ebuflen)
246 {
247 return __strsuftollx(desc, val, min, max, ebuf, ebuflen, 0);
248 }
249 #endif /* !HAVE_STRSUFTOLL */
250