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