lib/libnetbsd/strsuftoll.c

   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 */
 106 long long
 107 strsuftollx(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 */
 210 long long
 211 strsuftoll(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 }