Merge with the version from textutils.
[gnulib.git] / 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 __unsigned long int __strtol ();
72
73 static int
74 bkm_scale (x, scale_factor)
75      __unsigned long int *x;
76      int scale_factor;
77 {
78   /* The cast to `__unsigned long int' before the cast to double is
79      required to work around a bug in SunOS's /bin/cc.  */
80   if (*x > (double) ((__unsigned long int) __ZLONG_MAX) / scale_factor)
81     {
82       return 1;
83     }
84   *x *= scale_factor;
85   return 0;
86 }
87
88 /* FIXME: comment.  */
89
90 strtol_error
91 __xstrtol (s, ptr, base, val, valid_suffixes)
92      const char *s;
93      char **ptr;
94      int base;
95      __unsigned long int *val;
96      const char *valid_suffixes;
97 {
98   char *t_ptr;
99   char **p;
100   __unsigned long int tmp;
101
102   assert (0 <= base && base <= 36);
103
104   p = (ptr ? ptr : &t_ptr);
105
106   errno = 0;
107   tmp = __strtol (s, p, base);
108   if (errno != 0)
109     return LONGINT_OVERFLOW;
110   if (*p == s)
111     return LONGINT_INVALID;
112
113   /* Let valid_suffixes == NULL mean `allow any suffix'.  */
114   /* FIXME: update all callers except the one in tail.c changing
115      last parameter NULL to `""'.  */
116   if (!valid_suffixes)
117     {
118       *val = tmp;
119       return LONGINT_OK;
120     }
121
122   if (**p != '\0')
123     {
124       if (!strchr (valid_suffixes, **p))
125         return LONGINT_INVALID_SUFFIX_CHAR;
126
127       switch (**p)
128         {
129         case 'b':
130           if (bkm_scale (&tmp, 512))
131             return LONGINT_OVERFLOW;
132           ++(*p);
133           break;
134
135         case 'c':
136           ++(*p);
137           break;
138
139         case 'B':
140         case 'k':
141           if (bkm_scale (&tmp, 1024))
142             return LONGINT_OVERFLOW;
143           ++(*p);
144           break;
145
146         case 'm':
147           if (bkm_scale (&tmp, 1024 * 1024))
148             return LONGINT_OVERFLOW;
149           ++(*p);
150           break;
151
152         case 'w':
153           if (bkm_scale (&tmp, 2))
154             return LONGINT_OVERFLOW;
155           ++(*p);
156           break;
157
158         default:
159           return LONGINT_INVALID_SUFFIX_CHAR;
160           break;
161         }
162     }
163
164   *val = tmp;
165   return LONGINT_OK;
166 }
167
168 #ifdef TESTING_XSTRTO
169
170 # include <stdio.h>
171 # include "error.h"
172
173 char *program_name;
174
175 int
176 main (int argc, char** argv)
177 {
178   strtol_error s_err;
179   int i;
180
181   program_name = argv[0];
182   for (i=1; i<argc; i++)
183     {
184       char *p;
185       __unsigned long int val;
186
187       s_err = __xstrtol (argv[i], &p, 0, &val, "bckmw");
188       if (s_err == LONGINT_OK)
189         {
190           printf ("%s->%lu (%s)\n", argv[i], val, p);
191         }
192       else
193         {
194           STRTOL_FATAL_ERROR (argv[i], "arg", s_err);
195         }
196     }
197   exit (0);
198 }
199
200 #endif /* TESTING_XSTRTO */