lib/xstrtol.c

   1 /* A more useful interface to strtol.
   2    Copyright (C) 1995, 1996, 1998 Free Software Foundation, Inc.
   3
   4    This program is free software; you can redistribute it and/or modify
   5    it under the terms of the GNU General Public License as published by
   6    the Free Software Foundation; either version 2, or (at your option)
   7    any later version.
   8
   9    This program is distributed in the hope that it will be useful,
  10    but WITHOUT ANY WARRANTY; without even the implied warranty of
  11    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12    GNU General Public License for more details.
  13
  14    You should have received a copy of the GNU General Public License
  15    along with this program; if not, write to the Free Software Foundation,
  16    Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.  */
  17
  18 /* Written by Jim Meyering. */
  19
  20 #if HAVE_CONFIG_H
  21 # include <config.h>
  22 #endif
  23
  24 #if STDC_HEADERS
  25 # include <stdlib.h>
  26 #endif
  27
  28 #if HAVE_STRING_H
  29 # include <string.h>
  30 #else
  31 # include <strings.h>
  32 # ifndef strchr
  33 #  define strchr index
  34 # endif
  35 #endif
  36
  37 #define NDEBUG
  38 #include <assert.h>
  39
  40 #include <errno.h>
  41 #ifndef errno
  42 extern int errno;
  43 #endif
  44
  45 #if HAVE_LIMITS_H
  46 # include <limits.h>
  47 #endif
  48
  49 #ifndef CHAR_BIT
  50 # define CHAR_BIT 8
  51 #endif
  52
  53 /* The extra casts work around common compiler bugs.  */
  54 #define TYPE_SIGNED(t) (! ((t) 0 < (t) -1))
  55 /* The outer cast is needed to work around a bug in Cray C 5.0.3.0.
  56    It is necessary at least when t == time_t.  */
  57 #define TYPE_MINIMUM(t) ((t) (TYPE_SIGNED (t) \
  58                               ? ~ (t) 0 << (sizeof (t) * CHAR_BIT - 1) : (t) 0))
  59 #define TYPE_MAXIMUM(t) (~ (t) 0 - TYPE_MINIMUM (t))
  60
  61 #ifndef ULONG_MAX
  62 # define ULONG_MAX TYPE_MAXIMUM (unsigned long int)
  63 #endif
  64
  65 #ifndef LONG_MAX
  66 # define LONG_MAX TYPE_MAXIMUM (long int)
  67 #endif
  68
  69 #include "xstrtol.h"
  70
  71 #define BKM_SCALE(x, scale_factor, error_return)                        \
  72       do                                                                \
  73         {                                                               \
  74           if ((x) > (double) __ZLONG_MAX / (scale_factor))              \
  75             return (error_return);                                      \
  76           (x) *= (scale_factor);                                        \
  77         }                                                               \
  78       while (0)
  79
  80 __unsigned long int __strtol ();
  81
  82 /* FIXME: comment.  */
  83
  84 strtol_error
  85 __xstrtol (s, ptr, base, val, valid_suffixes)
  86      const char *s;
  87      char **ptr;
  88      int base;
  89      __unsigned long int *val;
  90      const char *valid_suffixes;
  91 {
  92   char *t_ptr;
  93   char **p;
  94   __unsigned long int tmp;
  95
  96   assert (0 <= base && base <= 36);
  97
  98   p = (ptr ? ptr : &t_ptr);
  99
 100   errno = 0;
 101   tmp = __strtol (s, p, base);
 102   if (errno != 0)
 103     return LONGINT_OVERFLOW;
 104   if (*p == s)
 105     return LONGINT_INVALID;
 106
 107   /* Let valid_suffixes == NULL mean `allow any suffix'.  */
 108   /* FIXME: update all callers except the one in tail.c changing
 109      last parameter NULL to `""'.  */
 110   if (!valid_suffixes)
 111     {
 112       *val = tmp;
 113       return LONGINT_OK;
 114     }
 115
 116   if (**p != '\0')
 117     {
 118       if (!strchr (valid_suffixes, **p))
 119         return LONGINT_INVALID_SUFFIX_CHAR;
 120
 121       switch (**p)
 122         {
 123         case 'b':
 124           BKM_SCALE (tmp, 512, LONGINT_OVERFLOW);
 125           ++(*p);
 126           break;
 127
 128         case 'c':
 129           ++(*p);
 130           break;
 131
 132         case 'B':
 133         case 'k':
 134           BKM_SCALE (tmp, 1024, LONGINT_OVERFLOW);
 135           ++(*p);
 136           break;
 137
 138         case 'm':
 139           BKM_SCALE (tmp, 1024 * 1024, LONGINT_OVERFLOW);
 140           ++(*p);
 141           break;
 142
 143         case 'w':
 144           BKM_SCALE (tmp, 2, LONGINT_OVERFLOW);
 145           ++(*p);
 146           break;
 147
 148         default:
 149           return LONGINT_INVALID_SUFFIX_CHAR;
 150           break;
 151         }
 152     }
 153
 154   *val = tmp;
 155   return LONGINT_OK;
 156 }
 157
 158 #ifdef TESTING_XSTRTO
 159
 160 # include <stdio.h>
 161 # include "error.h"
 162
 163 char *program_name;
 164
 165 int
 166 main (int argc, char** argv)
 167 {
 168   strtol_error s_err;
 169   int i;
 170
 171   program_name = argv[0];
 172   for (i=1; i<argc; i++)
 173     {
 174       char *p;
 175       __unsigned long int val;
 176
 177       s_err = __xstrtol (argv[i], &p, 0, &val, "bckmw");
 178       if (s_err == LONGINT_OK)
 179         {
 180           printf ("%s->%lu (%s)\n", argv[i], val, p);
 181         }
 182       else
 183         {
 184           STRTOL_FATAL_ERROR (argv[i], "arg", s_err);
 185         }
 186     }
 187   exit (0);
 188 }
 189
 190 #endif /* TESTING_XSTRTO */