2004-06-14 Andreas Schwab <schwab@suse.de>
[kopensolaris-gnu/glibc.git] / stdio-common / printf-parsemb.c
1 /* Helper functions for parsing printf format strings.
2    Copyright (C) 1995-1999, 2000, 2002, 2003 Free Software Foundation, Inc.
3    This file is part of th GNU C Library.
4
5    The GNU C Library is free software; you can redistribute it and/or
6    modify it under the terms of the GNU Lesser General Public
7    License as published by the Free Software Foundation; either
8    version 2.1 of the License, or (at your option) any later version.
9
10    The GNU C Library is distributed in the hope that it will be useful,
11    but WITHOUT ANY WARRANTY; without even the implied warranty of
12    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
13    Lesser General Public License for more details.
14
15    You should have received a copy of the GNU Lesser General Public
16    License along with the GNU C Library; if not, write to the Free
17    Software Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA
18    02111-1307 USA.  */
19
20 #include <ctype.h>
21 #include <limits.h>
22 #include <stdlib.h>
23 #include <string.h>
24 #include <sys/param.h>
25 #include <wchar.h>
26 #include <wctype.h>
27
28 #ifndef COMPILE_WPRINTF
29 # define CHAR_T         char
30 # define UCHAR_T        unsigned char
31 # define INT_T          int
32 # define L_(Str)        Str
33 # define ISDIGIT(Ch)    isdigit (Ch)
34 #else
35 # define CHAR_T         wchar_t
36 # define UCHAR_T        unsigned int
37 # define INT_T          wint_t
38 # define L_(Str)        L##Str
39 # define ISDIGIT(Ch)    iswdigit (Ch)
40 #endif
41
42 #include "printf-parse.h"
43
44 #define NDEBUG 1
45 #include <assert.h>
46
47
48
49 /* Find the next spec in FORMAT, or the end of the string.  Returns
50    a pointer into FORMAT, to a '%' or a '\0'.  */
51 const UCHAR_T *
52 #ifdef COMPILE_WPRINTF
53 __find_specwc (const UCHAR_T *format)
54 #else
55 __find_specmb (const UCHAR_T *format, mbstate_t *ps)
56 #endif
57 {
58 #ifdef COMPILE_WPRINTF
59   return (const UCHAR_T *) __wcschrnul ((const CHAR_T *) format, L'%');
60 #else
61   while (*format != L_('\0') && *format != L_('%'))
62     {
63       int len;
64
65       /* Remove any hints of a wrong encoding.  */
66       ps->__count = 0;
67       if (! isascii (*format) && (len = __mbrlen (format, MB_CUR_MAX, ps)) > 0)
68         format += len;
69       else
70         ++format;
71     }
72   return format;
73 #endif
74 }
75
76
77 /* FORMAT must point to a '%' at the beginning of a spec.  Fills in *SPEC
78    with the parsed details.  POSN is the number of arguments already
79    consumed.  At most MAXTYPES - POSN types are filled in TYPES.  Return
80    the number of args consumed by this spec; *MAX_REF_ARG is updated so it
81    remains the highest argument index used.  */
82 size_t
83 attribute_hidden
84 #ifdef COMPILE_WPRINTF
85 __parse_one_specwc (const UCHAR_T *format, size_t posn,
86                     struct printf_spec *spec, size_t *max_ref_arg)
87 #else
88 __parse_one_specmb (const UCHAR_T *format, size_t posn,
89                     struct printf_spec *spec, size_t *max_ref_arg,
90                     mbstate_t *ps)
91 #endif
92 {
93   unsigned int n;
94   size_t nargs = 0;
95
96   /* Skip the '%'.  */
97   ++format;
98
99   /* Clear information structure.  */
100   spec->data_arg = -1;
101   spec->info.alt = 0;
102   spec->info.space = 0;
103   spec->info.left = 0;
104   spec->info.showsign = 0;
105   spec->info.group = 0;
106   spec->info.i18n = 0;
107   spec->info.pad = ' ';
108   spec->info.wide = sizeof (UCHAR_T) > 1;
109
110   /* Test for positional argument.  */
111   if (ISDIGIT (*format))
112     {
113       const UCHAR_T *begin = format;
114
115       n = read_int (&format);
116
117       if (n > 0 && *format == L_('$'))
118         /* Is positional parameter.  */
119         {
120           ++format;             /* Skip the '$'.  */
121           spec->data_arg = n - 1;
122           *max_ref_arg = MAX (*max_ref_arg, n);
123         }
124       else
125         /* Oops; that was actually the width and/or 0 padding flag.
126            Step back and read it again.  */
127         format = begin;
128     }
129
130   /* Check for spec modifiers.  */
131   do
132     {
133       switch (*format)
134         {
135         case L_(' '):
136           /* Output a space in place of a sign, when there is no sign.  */
137           spec->info.space = 1;
138           continue;
139         case L_('+'):
140           /* Always output + or - for numbers.  */
141           spec->info.showsign = 1;
142           continue;
143         case L_('-'):
144           /* Left-justify things.  */
145           spec->info.left = 1;
146           continue;
147         case L_('#'):
148           /* Use the "alternate form":
149              Hex has 0x or 0X, FP always has a decimal point.  */
150           spec->info.alt = 1;
151           continue;
152         case L_('0'):
153           /* Pad with 0s.  */
154           spec->info.pad = '0';
155           continue;
156         case L_('\''):
157           /* Show grouping in numbers if the locale information
158              indicates any.  */
159           spec->info.group = 1;
160           continue;
161         case L_('I'):
162           /* Use the internationalized form of the output.  Currently
163              means to use the `outdigits' of the current locale.  */
164           spec->info.i18n = 1;
165           continue;
166         default:
167           break;
168         }
169       break;
170     }
171   while (*++format);
172
173   if (spec->info.left)
174     spec->info.pad = ' ';
175
176   /* Get the field width.  */
177   spec->width_arg = -1;
178   spec->info.width = 0;
179   if (*format == L_('*'))
180     {
181       /* The field width is given in an argument.
182          A negative field width indicates left justification.  */
183       const UCHAR_T *begin = ++format;
184
185       if (ISDIGIT (*format))
186         {
187           /* The width argument might be found in a positional parameter.  */
188           n = read_int (&format);
189
190           if (n > 0 && *format == L_('$'))
191             {
192               spec->width_arg = n - 1;
193               *max_ref_arg = MAX (*max_ref_arg, n);
194               ++format;         /* Skip '$'.  */
195             }
196         }
197
198       if (spec->width_arg < 0)
199         {
200           /* Not in a positional parameter.  Consume one argument.  */
201           spec->width_arg = posn++;
202           ++nargs;
203           format = begin;       /* Step back and reread.  */
204         }
205     }
206   else if (ISDIGIT (*format))
207     /* Constant width specification.  */
208     spec->info.width = read_int (&format);
209
210   /* Get the precision.  */
211   spec->prec_arg = -1;
212   /* -1 means none given; 0 means explicit 0.  */
213   spec->info.prec = -1;
214   if (*format == L_('.'))
215     {
216       ++format;
217       if (*format == L_('*'))
218         {
219           /* The precision is given in an argument.  */
220           const UCHAR_T *begin = ++format;
221
222           if (ISDIGIT (*format))
223             {
224               n = read_int (&format);
225
226               if (n > 0 && *format == L_('$'))
227                 {
228                   spec->prec_arg = n - 1;
229                   *max_ref_arg = MAX (*max_ref_arg, n);
230                   ++format;
231                 }
232             }
233
234           if (spec->prec_arg < 0)
235             {
236               /* Not in a positional parameter.  */
237               spec->prec_arg = posn++;
238               ++nargs;
239               format = begin;
240             }
241         }
242       else if (ISDIGIT (*format))
243         spec->info.prec = read_int (&format);
244       else
245         /* "%.?" is treated like "%.0?".  */
246         spec->info.prec = 0;
247     }
248
249   /* Check for type modifiers.  */
250   spec->info.is_long_double = 0;
251   spec->info.is_short = 0;
252   spec->info.is_long = 0;
253   spec->info.is_char = 0;
254
255   switch (*format++)
256     {
257     case L_('h'):
258       /* ints are short ints or chars.  */
259       if (*format != L_('h'))
260         spec->info.is_short = 1;
261       else
262         {
263           ++format;
264           spec->info.is_char = 1;
265         }
266       break;
267     case L_('l'):
268       /* ints are long ints.  */
269       spec->info.is_long = 1;
270       if (*format != L_('l'))
271         break;
272       ++format;
273       /* FALLTHROUGH */
274     case L_('L'):
275       /* doubles are long doubles, and ints are long long ints.  */
276     case L_('q'):
277       /* 4.4 uses this for long long.  */
278       spec->info.is_long_double = 1;
279       break;
280     case L_('z'):
281     case L_('Z'):
282       /* ints are size_ts.  */
283       assert (sizeof (size_t) <= sizeof (unsigned long long int));
284 #if LONG_MAX != LONG_LONG_MAX
285       spec->info.is_long_double = sizeof (size_t) > sizeof (unsigned long int);
286 #endif
287       spec->info.is_long = sizeof (size_t) > sizeof (unsigned int);
288       break;
289     case L_('t'):
290       assert (sizeof (ptrdiff_t) <= sizeof (long long int));
291 #if LONG_MAX != LONG_LONG_MAX
292       spec->info.is_long_double = (sizeof (ptrdiff_t) > sizeof (long int));
293 #endif
294       spec->info.is_long = sizeof (ptrdiff_t) > sizeof (int);
295       break;
296     case L_('j'):
297       assert (sizeof (uintmax_t) <= sizeof (unsigned long long int));
298 #if LONG_MAX != LONG_LONG_MAX
299       spec->info.is_long_double = (sizeof (uintmax_t)
300                                    > sizeof (unsigned long int));
301 #endif
302       spec->info.is_long = sizeof (uintmax_t) > sizeof (unsigned int);
303       break;
304     default:
305       /* Not a recognized modifier.  Backup.  */
306       --format;
307       break;
308     }
309
310   /* Get the format specification.  */
311   spec->info.spec = (wchar_t) *format++;
312   if (__builtin_expect (__printf_function_table != NULL, 0)
313       && spec->info.spec <= UCHAR_MAX
314       && __printf_arginfo_table[spec->info.spec] != NULL)
315     /* We don't try to get the types for all arguments if the format
316        uses more than one.  The normal case is covered though.  */
317     spec->ndata_args = (*__printf_arginfo_table[spec->info.spec])
318       (&spec->info, 1, &spec->data_arg_type);
319   else
320     {
321       /* Find the data argument types of a built-in spec.  */
322       spec->ndata_args = 1;
323
324       switch (spec->info.spec)
325         {
326         case L'i':
327         case L'd':
328         case L'u':
329         case L'o':
330         case L'X':
331         case L'x':
332 #if LONG_MAX != LONG_LONG_MAX
333           if (spec->info.is_long_double)
334             spec->data_arg_type = PA_INT|PA_FLAG_LONG_LONG;
335           else
336 #endif
337             if (spec->info.is_long)
338               spec->data_arg_type = PA_INT|PA_FLAG_LONG;
339             else if (spec->info.is_short)
340               spec->data_arg_type = PA_INT|PA_FLAG_SHORT;
341             else if (spec->info.is_char)
342               spec->data_arg_type = PA_CHAR;
343             else
344               spec->data_arg_type = PA_INT;
345           break;
346         case L'e':
347         case L'E':
348         case L'f':
349         case L'F':
350         case L'g':
351         case L'G':
352         case L'a':
353         case L'A':
354           if (spec->info.is_long_double)
355             spec->data_arg_type = PA_DOUBLE|PA_FLAG_LONG_DOUBLE;
356           else
357             spec->data_arg_type = PA_DOUBLE;
358           break;
359         case L'c':
360           spec->data_arg_type = PA_CHAR;
361           break;
362         case L'C':
363           spec->data_arg_type = PA_WCHAR;
364           break;
365         case L's':
366           spec->data_arg_type = PA_STRING;
367           break;
368         case L'S':
369           spec->data_arg_type = PA_WSTRING;
370           break;
371         case L'p':
372           spec->data_arg_type = PA_POINTER;
373           break;
374         case L'n':
375           spec->data_arg_type = PA_INT|PA_FLAG_PTR;
376           break;
377
378         case L'm':
379         default:
380           /* An unknown spec will consume no args.  */
381           spec->ndata_args = 0;
382           break;
383         }
384     }
385
386   if (spec->data_arg == -1 && spec->ndata_args > 0)
387     {
388       /* There are args consumed, but no positional spec.  Use the
389          next sequential arg position.  */
390       spec->data_arg = posn;
391       nargs += spec->ndata_args;
392     }
393
394   if (spec->info.spec == L'\0')
395     /* Format ended before this spec was complete.  */
396     spec->end_of_fmt = spec->next_fmt = format - 1;
397   else
398     {
399       /* Find the next format spec.  */
400       spec->end_of_fmt = format;
401 #ifdef COMPILE_WPRINTF
402       spec->next_fmt = __find_specwc (format);
403 #else
404       spec->next_fmt = __find_specmb (format, ps);
405 #endif
406     }
407
408   return nargs;
409 }