lib/libc/stdio/vfwprintf.c

   1 /*-
   2  * Copyright (c) 1990, 1993
   3  *      The Regents of the University of California.  All rights reserved.
   4  *
   5  * This code is derived from software contributed to Berkeley by
   6  * Chris Torek.
   7  *
   8  * Copyright (c) 2011 The FreeBSD Foundation
   9  * All rights reserved.
  10  * Portions of this software were developed by David Chisnall
  11  * under sponsorship from the FreeBSD Foundation.
  12  *
  13  * Redistribution and use in source and binary forms, with or without
  14  * modification, are permitted provided that the following conditions
  15  * are met:
  16  * 1. Redistributions of source code must retain the above copyright
  17  *    notice, this list of conditions and the following disclaimer.
  18  * 2. Redistributions in binary form must reproduce the above copyright
  19  *    notice, this list of conditions and the following disclaimer in the
  20  *    documentation and/or other materials provided with the distribution.
  21  * 3. Neither the name of the University nor the names of its contributors
  22  *    may be used to endorse or promote products derived from this software
  23  *    without specific prior written permission.
  24  *
  25  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
  26  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  27  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  28  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
  29  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  30  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  31  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  32  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  33  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  34  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  35  * SUCH DAMAGE.
  36  */
  37
  38 #if 0
  39 #if defined(LIBC_SCCS) && !defined(lint)
  40 static char sccsid[] = "@(#)vfprintf.c  8.1 (Berkeley) 6/4/93";
  41 #endif /* LIBC_SCCS and not lint */
  42 #endif
  43 #include <sys/cdefs.h>
  44 __FBSDID("$FreeBSD$");
  45
  46 /*
  47  * Actual wprintf innards.
  48  *
  49  * Avoid making gratuitous changes to this source file; it should be kept
  50  * as close as possible to vfprintf.c for ease of maintenance.
  51  */
  52
  53 #include "namespace.h"
  54 #include <sys/types.h>
  55
  56 #include <ctype.h>
  57 #include <errno.h>
  58 #include <limits.h>
  59 #include <locale.h>
  60 #include <stdarg.h>
  61 #include <stddef.h>
  62 #include <stdint.h>
  63 #include <stdio.h>
  64 #include <stdlib.h>
  65 #include <string.h>
  66 #include <wchar.h>
  67 #include <wctype.h>
  68 #include "un-namespace.h"
  69
  70 #include "libc_private.h"
  71 #include "local.h"
  72 #include "fvwrite.h"
  73 #include "printflocal.h"
  74 #include "xlocale_private.h"
  75
  76 static int      __sprint(FILE *, struct __suio *, locale_t);
  77 static int      __sbprintf(FILE *, locale_t, const wchar_t *, va_list) __noinline;
  78 static wint_t   __xfputwc(wchar_t, FILE *, locale_t);
  79 static wchar_t  *__mbsconv(char *, int);
  80
  81 #define CHAR    wchar_t
  82 #include "printfcommon.h"
  83
  84 struct grouping_state {
  85         wchar_t thousands_sep;  /* locale-specific thousands separator */
  86         const char *grouping;   /* locale-specific numeric grouping rules */
  87         int lead;               /* sig figs before decimal or group sep */
  88         int nseps;              /* number of group separators with ' */
  89         int nrepeats;           /* number of repeats of the last group */
  90 };
  91
  92 static const mbstate_t initial_mbs;
  93
  94 static inline wchar_t
  95 get_decpt(locale_t locale)
  96 {
  97         mbstate_t mbs;
  98         wchar_t decpt;
  99         int nconv;
 100
 101         mbs = initial_mbs;
 102         nconv = mbrtowc(&decpt, localeconv_l(locale)->decimal_point, MB_CUR_MAX, &mbs);
 103         if (nconv == (size_t)-1 || nconv == (size_t)-2)
 104                 decpt = '.';    /* failsafe */
 105         return (decpt);
 106 }
 107
 108 static inline wchar_t
 109 get_thousep(locale_t locale)
 110 {
 111         mbstate_t mbs;
 112         wchar_t thousep;
 113         int nconv;
 114
 115         mbs = initial_mbs;
 116         nconv = mbrtowc(&thousep, localeconv_l(locale)->thousands_sep,
 117             MB_CUR_MAX, &mbs);
 118         if (nconv == (size_t)-1 || nconv == (size_t)-2)
 119                 thousep = '\0';    /* failsafe */
 120         return (thousep);
 121 }
 122
 123 /*
 124  * Initialize the thousands' grouping state in preparation to print a
 125  * number with ndigits digits. This routine returns the total number
 126  * of wide characters that will be printed.
 127  */
 128 static int
 129 grouping_init(struct grouping_state *gs, int ndigits, locale_t locale)
 130 {
 131
 132         gs->grouping = localeconv_l(locale)->grouping;
 133         gs->thousands_sep = get_thousep(locale);
 134
 135         gs->nseps = gs->nrepeats = 0;
 136         gs->lead = ndigits;
 137         while (*gs->grouping != CHAR_MAX) {
 138                 if (gs->lead <= *gs->grouping)
 139                         break;
 140                 gs->lead -= *gs->grouping;
 141                 if (*(gs->grouping+1)) {
 142                         gs->nseps++;
 143                         gs->grouping++;
 144                 } else
 145                         gs->nrepeats++;
 146         }
 147         return (gs->nseps + gs->nrepeats);
 148 }
 149
 150 /*
 151  * Print a number with thousands' separators.
 152  */
 153 static int
 154 grouping_print(struct grouping_state *gs, struct io_state *iop,
 155                const CHAR *cp, const CHAR *ep, locale_t locale)
 156 {
 157         const CHAR *cp0 = cp;
 158
 159         if (io_printandpad(iop, cp, ep, gs->lead, zeroes, locale))
 160                 return (-1);
 161         cp += gs->lead;
 162         while (gs->nseps > 0 || gs->nrepeats > 0) {
 163                 if (gs->nrepeats > 0)
 164                         gs->nrepeats--;
 165                 else {
 166                         gs->grouping--;
 167                         gs->nseps--;
 168                 }
 169                 if (io_print(iop, &gs->thousands_sep, 1, locale))
 170                         return (-1);
 171                 if (io_printandpad(iop, cp, ep, *gs->grouping, zeroes, locale))
 172                         return (-1);
 173                 cp += *gs->grouping;
 174         }
 175         if (cp > ep)
 176                 cp = ep;
 177         return (cp - cp0);
 178 }
 179
 180
 181 /*
 182  * Flush out all the vectors defined by the given uio,
 183  * then reset it so that it can be reused.
 184  *
 185  * XXX The fact that we do this a character at a time and convert to a
 186  * multibyte character sequence even if the destination is a wide
 187  * string eclipses the benefits of buffering.
 188  */
 189 static int
 190 __sprint(FILE *fp, struct __suio *uio, locale_t locale)
 191 {
 192         struct __siov *iov;
 193         wchar_t *p;
 194         int i, len;
 195
 196         iov = uio->uio_iov;
 197         for (; uio->uio_resid != 0; uio->uio_resid -= len, iov++) {
 198                 p = (wchar_t *)iov->iov_base;
 199                 len = iov->iov_len;
 200                 for (i = 0; i < len; i++) {
 201                         if (__xfputwc(p[i], fp, locale) == WEOF)
 202                                 return (-1);
 203                 }
 204         }
 205         uio->uio_iovcnt = 0;
 206         return (0);
 207 }
 208
 209 /*
 210  * Helper function for `fprintf to unbuffered unix file': creates a
 211  * temporary buffer.  We only work on write-only files; this avoids
 212  * worries about ungetc buffers and so forth.
 213  */
 214 static int
 215 __sbprintf(FILE *fp, locale_t locale, const wchar_t *fmt, va_list ap)
 216 {
 217         int ret;
 218         FILE fake;
 219         unsigned char buf[BUFSIZ];
 220
 221         /* XXX This is probably not needed. */
 222         if (prepwrite(fp) != 0)
 223                 return (EOF);
 224
 225         /* copy the important variables */
 226         fake._flags = fp->_flags & ~__SNBF;
 227         fake._file = fp->_file;
 228         fake._cookie = fp->_cookie;
 229         fake._write = fp->_write;
 230         fake._orientation = fp->_orientation;
 231         fake._mbstate = fp->_mbstate;
 232
 233         /* set up the buffer */
 234         fake._bf._base = fake._p = buf;
 235         fake._bf._size = fake._w = sizeof(buf);
 236         fake._lbfsize = 0;      /* not actually used, but Just In Case */
 237
 238         /* do the work, then copy any error status */
 239         ret = __vfwprintf(&fake, locale, fmt, ap);
 240         if (ret >= 0 && __fflush(&fake))
 241                 ret = WEOF;
 242         if (fake._flags & __SERR)
 243                 fp->_flags |= __SERR;
 244         return (ret);
 245 }
 246
 247 /*
 248  * Like __fputwc, but handles fake string (__SSTR) files properly.
 249  * File must already be locked.
 250  */
 251 static wint_t
 252 __xfputwc(wchar_t wc, FILE *fp, locale_t locale)
 253 {
 254         mbstate_t mbs;
 255         char buf[MB_LEN_MAX];
 256         struct __suio uio;
 257         struct __siov iov;
 258         size_t len;
 259
 260         if ((fp->_flags & __SSTR) == 0)
 261                 return (__fputwc(wc, fp, locale));
 262
 263         mbs = initial_mbs;
 264         if ((len = wcrtomb(buf, wc, &mbs)) == (size_t)-1) {
 265                 fp->_flags |= __SERR;
 266                 return (WEOF);
 267         }
 268         uio.uio_iov = &iov;
 269         uio.uio_resid = len;
 270         uio.uio_iovcnt = 1;
 271         iov.iov_base = buf;
 272         iov.iov_len = len;
 273         return (__sfvwrite(fp, &uio) != EOF ? (wint_t)wc : WEOF);
 274 }
 275
 276 /*
 277  * Convert a multibyte character string argument for the %s format to a wide
 278  * string representation. ``prec'' specifies the maximum number of bytes
 279  * to output. If ``prec'' is greater than or equal to zero, we can't assume
 280  * that the multibyte char. string ends in a null character.
 281  */
 282 static wchar_t *
 283 __mbsconv(char *mbsarg, int prec)
 284 {
 285         mbstate_t mbs;
 286         wchar_t *convbuf, *wcp;
 287         const char *p;
 288         size_t insize, nchars, nconv;
 289
 290         if (mbsarg == NULL)
 291                 return (NULL);
 292
 293         /*
 294          * Supplied argument is a multibyte string; convert it to wide
 295          * characters first.
 296          */
 297         if (prec >= 0) {
 298                 /*
 299                  * String is not guaranteed to be NUL-terminated. Find the
 300                  * number of characters to print.
 301                  */
 302                 p = mbsarg;
 303                 insize = nchars = nconv = 0;
 304                 mbs = initial_mbs;
 305                 while (nchars != (size_t)prec) {
 306                         nconv = mbrlen(p, MB_CUR_MAX, &mbs);
 307                         if (nconv == 0 || nconv == (size_t)-1 ||
 308                             nconv == (size_t)-2)
 309                                 break;
 310                         p += nconv;
 311                         nchars++;
 312                         insize += nconv;
 313                 }
 314                 if (nconv == (size_t)-1 || nconv == (size_t)-2)
 315                         return (NULL);
 316         } else {
 317                 insize = strlen(mbsarg);
 318                 nconv = 0;
 319         }
 320
 321         /*
 322          * Allocate buffer for the result and perform the conversion,
 323          * converting at most `size' bytes of the input multibyte string to
 324          * wide characters for printing.
 325          */
 326         convbuf = malloc((insize + 1) * sizeof(*convbuf));
 327         if (convbuf == NULL)
 328                 return (NULL);
 329         wcp = convbuf;
 330         p = mbsarg;
 331         mbs = initial_mbs;
 332         while (insize != 0) {
 333                 nconv = mbrtowc(wcp, p, insize, &mbs);
 334                 if (nconv == 0 || nconv == (size_t)-1 || nconv == (size_t)-2)
 335                         break;
 336                 wcp++;
 337                 p += nconv;
 338                 insize -= nconv;
 339         }
 340         if (nconv == (size_t)-1 || nconv == (size_t)-2) {
 341                 free(convbuf);
 342                 return (NULL);
 343         }
 344         *wcp = L'\0';
 345
 346         return (convbuf);
 347 }
 348
 349 /*
 350  * MT-safe version
 351  */
 352 int
 353 vfwprintf_l(FILE * __restrict fp, locale_t locale,
 354                 const wchar_t * __restrict fmt0, va_list ap)
 355
 356 {
 357         int ret;
 358         FIX_LOCALE(locale);
 359         FLOCKFILE(fp);
 360         /* optimise fprintf(stderr) (and other unbuffered Unix files) */
 361         if ((fp->_flags & (__SNBF|__SWR|__SRW)) == (__SNBF|__SWR) &&
 362             fp->_file >= 0)
 363                 ret = __sbprintf(fp, locale, fmt0, ap);
 364         else
 365                 ret = __vfwprintf(fp, locale, fmt0, ap);
 366         FUNLOCKFILE(fp);
 367         return (ret);
 368 }
 369 int
 370 vfwprintf(FILE * __restrict fp, const wchar_t * __restrict fmt0, va_list ap)
 371 {
 372         return vfwprintf_l(fp, __get_locale(), fmt0, ap);
 373 }
 374
 375 /*
 376  * The size of the buffer we use as scratch space for integer
 377  * conversions, among other things.  We need enough space to
 378  * write a uintmax_t in octal (plus one byte).
 379  */
 380 #if UINTMAX_MAX <= UINT64_MAX
 381 #define BUF     32
 382 #else
 383 #error "BUF must be large enough to format a uintmax_t"
 384 #endif
 385
 386 /*
 387  * Non-MT-safe version
 388  */
 389 int
 390 __vfwprintf(FILE *fp, locale_t locale, const wchar_t *fmt0, va_list ap)
 391 {
 392         wchar_t *fmt;           /* format string */
 393         wchar_t ch;             /* character from fmt */
 394         int n, n2;              /* handy integer (short term usage) */
 395         wchar_t *cp;            /* handy char pointer (short term usage) */
 396         int flags;              /* flags as above */
 397         int ret;                /* return value accumulator */
 398         int width;              /* width from format (%8d), or 0 */
 399         int prec;               /* precision from format; <0 for N/A */
 400         wchar_t sign;           /* sign prefix (' ', '+', '-', or \0) */
 401         struct grouping_state gs; /* thousands' grouping info */
 402 #ifndef NO_FLOATING_POINT
 403         /*
 404          * We can decompose the printed representation of floating
 405          * point numbers into several parts, some of which may be empty:
 406          *
 407          * [+|-| ] [0x|0X] MMM . NNN [e|E|p|P] [+|-] ZZ
 408          *    A       B     ---C---      D       E   F
 409          *
 410          * A:   'sign' holds this value if present; '\0' otherwise
 411          * B:   ox[1] holds the 'x' or 'X'; '\0' if not hexadecimal
 412          * C:   cp points to the string MMMNNN.  Leading and trailing
 413          *      zeros are not in the string and must be added.
 414          * D:   expchar holds this character; '\0' if no exponent, e.g. %f
 415          * F:   at least two digits for decimal, at least one digit for hex
 416          */
 417         wchar_t decimal_point;  /* locale specific decimal point */
 418         int signflag;           /* true if float is negative */
 419         union {                 /* floating point arguments %[aAeEfFgG] */
 420                 double dbl;
 421                 long double ldbl;
 422         } fparg;
 423         int expt;               /* integer value of exponent */
 424         char expchar;           /* exponent character: [eEpP\0] */
 425         char *dtoaend;          /* pointer to end of converted digits */
 426         int expsize;            /* character count for expstr */
 427         int ndig;               /* actual number of digits returned by dtoa */
 428         wchar_t expstr[MAXEXPDIG+2];    /* buffer for exponent string: e+ZZZ */
 429         char *dtoaresult;       /* buffer allocated by dtoa */
 430 #endif
 431         u_long  ulval;          /* integer arguments %[diouxX] */
 432         uintmax_t ujval;        /* %j, %ll, %q, %t, %z integers */
 433         int base;               /* base for [diouxX] conversion */
 434         int dprec;              /* a copy of prec if [diouxX], 0 otherwise */
 435         int realsz;             /* field size expanded by dprec, sign, etc */
 436         int size;               /* size of converted field or string */
 437         int prsize;             /* max size of printed field */
 438         const char *xdigs;      /* digits for [xX] conversion */
 439         struct io_state io;     /* I/O buffering state */
 440         wchar_t buf[BUF];       /* buffer with space for digits of uintmax_t */
 441         wchar_t ox[2];          /* space for 0x hex-prefix */
 442         union arg *argtable;    /* args, built due to positional arg */
 443         union arg statargtable [STATIC_ARG_TBL_SIZE];
 444         int nextarg;            /* 1-based argument index */
 445         va_list orgap;          /* original argument pointer */
 446         wchar_t *convbuf;       /* multibyte to wide conversion result */
 447
 448         static const char xdigs_lower[16] = "0123456789abcdef";
 449         static const char xdigs_upper[16] = "0123456789ABCDEF";
 450
 451         /* BEWARE, these `goto error' on error. */
 452 #define PRINT(ptr, len) do {                    \
 453         if (io_print(&io, (ptr), (len), locale))        \
 454                 goto error; \
 455 } while (0)
 456 #define PAD(howmany, with) { \
 457         if (io_pad(&io, (howmany), (with), locale)) \
 458                 goto error; \
 459 }
 460 #define PRINTANDPAD(p, ep, len, with) { \
 461         if (io_printandpad(&io, (p), (ep), (len), (with), locale)) \
 462                 goto error; \
 463 }
 464 #define FLUSH() { \
 465         if (io_flush(&io, locale)) \
 466                 goto error; \
 467 }
 468
 469         /*
 470          * Get the argument indexed by nextarg.   If the argument table is
 471          * built, use it to get the argument.  If its not, get the next
 472          * argument (and arguments must be gotten sequentially).
 473          */
 474 #define GETARG(type) \
 475         ((argtable != NULL) ? *((type*)(&argtable[nextarg++])) : \
 476             (nextarg++, va_arg(ap, type)))
 477
 478         /*
 479          * To extend shorts properly, we need both signed and unsigned
 480          * argument extraction methods.
 481          */
 482 #define SARG() \
 483         (flags&LONGINT ? GETARG(long) : \
 484             flags&SHORTINT ? (long)(short)GETARG(int) : \
 485             flags&CHARINT ? (long)(signed char)GETARG(int) : \
 486             (long)GETARG(int))
 487 #define UARG() \
 488         (flags&LONGINT ? GETARG(u_long) : \
 489             flags&SHORTINT ? (u_long)(u_short)GETARG(int) : \
 490             flags&CHARINT ? (u_long)(u_char)GETARG(int) : \
 491             (u_long)GETARG(u_int))
 492 #define INTMAX_SIZE     (INTMAXT|SIZET|PTRDIFFT|LLONGINT)
 493 #define SJARG() \
 494         (flags&INTMAXT ? GETARG(intmax_t) : \
 495             flags&SIZET ? (intmax_t)GETARG(ssize_t) : \
 496             flags&PTRDIFFT ? (intmax_t)GETARG(ptrdiff_t) : \
 497             (intmax_t)GETARG(long long))
 498 #define UJARG() \
 499         (flags&INTMAXT ? GETARG(uintmax_t) : \
 500             flags&SIZET ? (uintmax_t)GETARG(size_t) : \
 501             flags&PTRDIFFT ? (uintmax_t)GETARG(ptrdiff_t) : \
 502             (uintmax_t)GETARG(unsigned long long))
 503
 504         /*
 505          * Get * arguments, including the form *nn$.  Preserve the nextarg
 506          * that the argument can be gotten once the type is determined.
 507          */
 508 #define GETASTER(val) \
 509         n2 = 0; \
 510         cp = fmt; \
 511         while (is_digit(*cp)) { \
 512                 n2 = 10 * n2 + to_digit(*cp); \
 513                 cp++; \
 514         } \
 515         if (*cp == '$') { \
 516                 int hold = nextarg; \
 517                 if (argtable == NULL) { \
 518                         argtable = statargtable; \
 519                         if (__find_warguments (fmt0, orgap, &argtable)) { \
 520                                 ret = EOF; \
 521                                 goto error; \
 522                         } \
 523                 } \
 524                 nextarg = n2; \
 525                 val = GETARG (int); \
 526                 nextarg = hold; \
 527                 fmt = ++cp; \
 528         } else { \
 529                 val = GETARG (int); \
 530         }
 531
 532
 533         /* sorry, fwprintf(read_only_file, L"") returns WEOF, not 0 */
 534         if (prepwrite(fp) != 0) {
 535                 errno = EBADF;
 536                 return (EOF);
 537         }
 538
 539         convbuf = NULL;
 540         fmt = (wchar_t *)fmt0;
 541         argtable = NULL;
 542         nextarg = 1;
 543         va_copy(orgap, ap);
 544         io_init(&io, fp);
 545         ret = 0;
 546 #ifndef NO_FLOATING_POINT
 547         decimal_point = get_decpt(locale);
 548 #endif
 549
 550         /*
 551          * Scan the format for conversions (`%' character).
 552          */
 553         for (;;) {
 554                 for (cp = fmt; (ch = *fmt) != '\0' && ch != '%'; fmt++)
 555                         /* void */;
 556                 if ((n = fmt - cp) != 0) {
 557                         if ((unsigned)ret + n > INT_MAX) {
 558                                 ret = EOF;
 559                                 errno = EOVERFLOW;
 560                                 goto error;
 561                         }
 562                         PRINT(cp, n);
 563                         ret += n;
 564                 }
 565                 if (ch == '\0')
 566                         goto done;
 567                 fmt++;          /* skip over '%' */
 568
 569                 flags = 0;
 570                 dprec = 0;
 571                 width = 0;
 572                 prec = -1;
 573                 gs.grouping = NULL;
 574                 sign = '\0';
 575                 ox[1] = '\0';
 576
 577 rflag:          ch = *fmt++;
 578 reswitch:       switch (ch) {
 579                 case ' ':
 580                         /*-
 581                          * ``If the space and + flags both appear, the space
 582                          * flag will be ignored.''
 583                          *      -- ANSI X3J11
 584                          */
 585                         if (!sign)
 586                                 sign = ' ';
 587                         goto rflag;
 588                 case '#':
 589                         flags |= ALT;
 590                         goto rflag;
 591                 case '*':
 592                         /*-
 593                          * ``A negative field width argument is taken as a
 594                          * - flag followed by a positive field width.''
 595                          *      -- ANSI X3J11
 596                          * They don't exclude field widths read from args.
 597                          */
 598                         GETASTER (width);
 599                         if (width >= 0)
 600                                 goto rflag;
 601                         width = -width;
 602                         /* FALLTHROUGH */
 603                 case '-':
 604                         flags |= LADJUST;
 605                         goto rflag;
 606                 case '+':
 607                         sign = '+';
 608                         goto rflag;
 609                 case '\'':
 610                         flags |= GROUPING;
 611                         goto rflag;
 612                 case '.':
 613                         if ((ch = *fmt++) == '*') {
 614                                 GETASTER (prec);
 615                                 goto rflag;
 616                         }
 617                         prec = 0;
 618                         while (is_digit(ch)) {
 619                                 prec = 10 * prec + to_digit(ch);
 620                                 ch = *fmt++;
 621                         }
 622                         goto reswitch;
 623                 case '0':
 624                         /*-
 625                          * ``Note that 0 is taken as a flag, not as the
 626                          * beginning of a field width.''
 627                          *      -- ANSI X3J11
 628                          */
 629                         flags |= ZEROPAD;
 630                         goto rflag;
 631                 case '1': case '2': case '3': case '4':
 632                 case '5': case '6': case '7': case '8': case '9':
 633                         n = 0;
 634                         do {
 635                                 n = 10 * n + to_digit(ch);
 636                                 ch = *fmt++;
 637                         } while (is_digit(ch));
 638                         if (ch == '$') {
 639                                 nextarg = n;
 640                                 if (argtable == NULL) {
 641                                         argtable = statargtable;
 642                                         if (__find_warguments (fmt0, orgap,
 643                                                                &argtable)) {
 644                                                 ret = EOF;
 645                                                 goto error;
 646                                         }
 647                                 }
 648                                 goto rflag;
 649                         }
 650                         width = n;
 651                         goto reswitch;
 652 #ifndef NO_FLOATING_POINT
 653                 case 'L':
 654                         flags |= LONGDBL;
 655                         goto rflag;
 656 #endif
 657                 case 'h':
 658                         if (flags & SHORTINT) {
 659                                 flags &= ~SHORTINT;
 660                                 flags |= CHARINT;
 661                         } else
 662                                 flags |= SHORTINT;
 663                         goto rflag;
 664                 case 'j':
 665                         flags |= INTMAXT;
 666                         goto rflag;
 667                 case 'l':
 668                         if (flags & LONGINT) {
 669                                 flags &= ~LONGINT;
 670                                 flags |= LLONGINT;
 671                         } else
 672                                 flags |= LONGINT;
 673                         goto rflag;
 674                 case 'q':
 675                         flags |= LLONGINT;      /* not necessarily */
 676                         goto rflag;
 677                 case 't':
 678                         flags |= PTRDIFFT;
 679                         goto rflag;
 680                 case 'z':
 681                         flags |= SIZET;
 682                         goto rflag;
 683                 case 'C':
 684                         flags |= LONGINT;
 685                         /*FALLTHROUGH*/
 686                 case 'c':
 687                         if (flags & LONGINT)
 688                                 *(cp = buf) = (wchar_t)GETARG(wint_t);
 689                         else
 690                                 *(cp = buf) = (wchar_t)btowc(GETARG(int));
 691                         size = 1;
 692                         sign = '\0';
 693                         break;
 694                 case 'D':
 695                         flags |= LONGINT;
 696                         /*FALLTHROUGH*/
 697                 case 'd':
 698                 case 'i':
 699                         if (flags & INTMAX_SIZE) {
 700                                 ujval = SJARG();
 701                                 if ((intmax_t)ujval < 0) {
 702                                         ujval = -ujval;
 703                                         sign = '-';
 704                                 }
 705                         } else {
 706                                 ulval = SARG();
 707                                 if ((long)ulval < 0) {
 708                                         ulval = -ulval;
 709                                         sign = '-';
 710                                 }
 711                         }
 712                         base = 10;
 713                         goto number;
 714 #ifndef NO_FLOATING_POINT
 715                 case 'a':
 716                 case 'A':
 717                         if (ch == 'a') {
 718                                 ox[1] = 'x';
 719                                 xdigs = xdigs_lower;
 720                                 expchar = 'p';
 721                         } else {
 722                                 ox[1] = 'X';
 723                                 xdigs = xdigs_upper;
 724                                 expchar = 'P';
 725                         }
 726                         if (prec >= 0)
 727                                 prec++;
 728                         if (flags & LONGDBL) {
 729                                 fparg.ldbl = GETARG(long double);
 730                                 dtoaresult =
 731                                     __hldtoa(fparg.ldbl, xdigs, prec,
 732                                         &expt, &signflag, &dtoaend);
 733                         } else {
 734                                 fparg.dbl = GETARG(double);
 735                                 dtoaresult =
 736                                     __hdtoa(fparg.dbl, xdigs, prec,
 737                                         &expt, &signflag, &dtoaend);
 738                         }
 739                         if (prec < 0)
 740                                 prec = dtoaend - dtoaresult;
 741                         if (expt == INT_MAX)
 742                                 ox[1] = '\0';
 743                         if (convbuf != NULL)
 744                                 free(convbuf);
 745                         ndig = dtoaend - dtoaresult;
 746                         cp = convbuf = __mbsconv(dtoaresult, -1);
 747                         freedtoa(dtoaresult);
 748                         goto fp_common;
 749                 case 'e':
 750                 case 'E':
 751                         expchar = ch;
 752                         if (prec < 0)   /* account for digit before decpt */
 753                                 prec = DEFPREC + 1;
 754                         else
 755                                 prec++;
 756                         goto fp_begin;
 757                 case 'f':
 758                 case 'F':
 759                         expchar = '\0';
 760                         goto fp_begin;
 761                 case 'g':
 762                 case 'G':
 763                         expchar = ch - ('g' - 'e');
 764                         if (prec == 0)
 765                                 prec = 1;
 766 fp_begin:
 767                         if (prec < 0)
 768                                 prec = DEFPREC;
 769                         if (convbuf != NULL)
 770                                 free(convbuf);
 771                         if (flags & LONGDBL) {
 772                                 fparg.ldbl = GETARG(long double);
 773                                 dtoaresult =
 774                                     __ldtoa(&fparg.ldbl, expchar ? 2 : 3, prec,
 775                                     &expt, &signflag, &dtoaend);
 776                         } else {
 777                                 fparg.dbl = GETARG(double);
 778                                 dtoaresult =
 779                                     dtoa(fparg.dbl, expchar ? 2 : 3, prec,
 780                                     &expt, &signflag, &dtoaend);
 781                                 if (expt == 9999)
 782                                         expt = INT_MAX;
 783                         }
 784                         ndig = dtoaend - dtoaresult;
 785                         cp = convbuf = __mbsconv(dtoaresult, -1);
 786                         freedtoa(dtoaresult);
 787 fp_common:
 788                         if (signflag)
 789                                 sign = '-';
 790                         if (expt == INT_MAX) {  /* inf or nan */
 791                                 if (*cp == 'N') {
 792                                         cp = (ch >= 'a') ? L"nan" : L"NAN";
 793                                         sign = '\0';
 794                                 } else
 795                                         cp = (ch >= 'a') ? L"inf" : L"INF";
 796                                 size = 3;
 797                                 flags &= ~ZEROPAD;
 798                                 break;
 799                         }
 800                         flags |= FPT;
 801                         if (ch == 'g' || ch == 'G') {
 802                                 if (expt > -4 && expt <= prec) {
 803                                         /* Make %[gG] smell like %[fF] */
 804                                         expchar = '\0';
 805                                         if (flags & ALT)
 806                                                 prec -= expt;
 807                                         else
 808                                                 prec = ndig - expt;
 809                                         if (prec < 0)
 810                                                 prec = 0;
 811                                 } else {
 812                                         /*
 813                                          * Make %[gG] smell like %[eE], but
 814                                          * trim trailing zeroes if no # flag.
 815                                          */
 816                                         if (!(flags & ALT))
 817                                                 prec = ndig;
 818                                 }
 819                         }
 820                         if (expchar) {
 821                                 expsize = exponent(expstr, expt - 1, expchar);
 822                                 size = expsize + prec;
 823                                 if (prec > 1 || flags & ALT)
 824                                         ++size;
 825                         } else {
 826                                 /* space for digits before decimal point */
 827                                 if (expt > 0)
 828                                         size = expt;
 829                                 else    /* "0" */
 830                                         size = 1;
 831                                 /* space for decimal pt and following digits */
 832                                 if (prec || flags & ALT)
 833                                         size += prec + 1;
 834                                 if ((flags & GROUPING) && expt > 0)
 835                                         size += grouping_init(&gs, expt, locale);
 836                         }
 837                         break;
 838 #endif /* !NO_FLOATING_POINT */
 839                 case 'n':
 840                         /*
 841                          * Assignment-like behavior is specified if the
 842                          * value overflows or is otherwise unrepresentable.
 843                          * C99 says to use `signed char' for %hhn conversions.
 844                          */
 845                         if (flags & LLONGINT)
 846                                 *GETARG(long long *) = ret;
 847                         else if (flags & SIZET)
 848                                 *GETARG(ssize_t *) = (ssize_t)ret;
 849                         else if (flags & PTRDIFFT)
 850                                 *GETARG(ptrdiff_t *) = ret;
 851                         else if (flags & INTMAXT)
 852                                 *GETARG(intmax_t *) = ret;
 853                         else if (flags & LONGINT)
 854                                 *GETARG(long *) = ret;
 855                         else if (flags & SHORTINT)
 856                                 *GETARG(short *) = ret;
 857                         else if (flags & CHARINT)
 858                                 *GETARG(signed char *) = ret;
 859                         else
 860                                 *GETARG(int *) = ret;
 861                         continue;       /* no output */
 862                 case 'O':
 863                         flags |= LONGINT;
 864                         /*FALLTHROUGH*/
 865                 case 'o':
 866                         if (flags & INTMAX_SIZE)
 867                                 ujval = UJARG();
 868                         else
 869                                 ulval = UARG();
 870                         base = 8;
 871                         goto nosign;
 872                 case 'p':
 873                         /*-
 874                          * ``The argument shall be a pointer to void.  The
 875                          * value of the pointer is converted to a sequence
 876                          * of printable characters, in an implementation-
 877                          * defined manner.''
 878                          *      -- ANSI X3J11
 879                          */
 880                         ujval = (uintmax_t)(uintptr_t)GETARG(void *);
 881                         base = 16;
 882                         xdigs = xdigs_lower;
 883                         flags = flags | INTMAXT;
 884                         ox[1] = 'x';
 885                         goto nosign;
 886                 case 'S':
 887                         flags |= LONGINT;
 888                         /*FALLTHROUGH*/
 889                 case 's':
 890                         if (flags & LONGINT) {
 891                                 if ((cp = GETARG(wchar_t *)) == NULL)
 892                                         cp = L"(null)";
 893                         } else {
 894                                 char *mbp;
 895
 896                                 if (convbuf != NULL)
 897                                         free(convbuf);
 898                                 if ((mbp = GETARG(char *)) == NULL)
 899                                         cp = L"(null)";
 900                                 else {
 901                                         convbuf = __mbsconv(mbp, prec);
 902                                         if (convbuf == NULL) {
 903                                                 fp->_flags |= __SERR;
 904                                                 goto error;
 905                                         }
 906                                         cp = convbuf;
 907                                 }
 908                         }
 909                         size = (prec >= 0) ? wcsnlen(cp, prec) : wcslen(cp);
 910                         sign = '\0';
 911                         break;
 912                 case 'U':
 913                         flags |= LONGINT;
 914                         /*FALLTHROUGH*/
 915                 case 'u':
 916                         if (flags & INTMAX_SIZE)
 917                                 ujval = UJARG();
 918                         else
 919                                 ulval = UARG();
 920                         base = 10;
 921                         goto nosign;
 922                 case 'X':
 923                         xdigs = xdigs_upper;
 924                         goto hex;
 925                 case 'x':
 926                         xdigs = xdigs_lower;
 927 hex:
 928                         if (flags & INTMAX_SIZE)
 929                                 ujval = UJARG();
 930                         else
 931                                 ulval = UARG();
 932                         base = 16;
 933                         /* leading 0x/X only if non-zero */
 934                         if (flags & ALT &&
 935                             (flags & INTMAX_SIZE ? ujval != 0 : ulval != 0))
 936                                 ox[1] = ch;
 937
 938                         flags &= ~GROUPING;
 939                         /* unsigned conversions */
 940 nosign:                 sign = '\0';
 941                         /*-
 942                          * ``... diouXx conversions ... if a precision is
 943                          * specified, the 0 flag will be ignored.''
 944                          *      -- ANSI X3J11
 945                          */
 946 number:                 if ((dprec = prec) >= 0)
 947                                 flags &= ~ZEROPAD;
 948
 949                         /*-
 950                          * ``The result of converting a zero value with an
 951                          * explicit precision of zero is no characters.''
 952                          *      -- ANSI X3J11
 953                          *
 954                          * ``The C Standard is clear enough as is.  The call
 955                          * printf("%#.0o", 0) should print 0.''
 956                          *      -- Defect Report #151
 957                          */
 958                         cp = buf + BUF;
 959                         if (flags & INTMAX_SIZE) {
 960                                 if (ujval != 0 || prec != 0 ||
 961                                     (flags & ALT && base == 8))
 962                                         cp = __ujtoa(ujval, cp, base,
 963                                             flags & ALT, xdigs);
 964                         } else {
 965                                 if (ulval != 0 || prec != 0 ||
 966                                     (flags & ALT && base == 8))
 967                                         cp = __ultoa(ulval, cp, base,
 968                                             flags & ALT, xdigs);
 969                         }
 970                         size = buf + BUF - cp;
 971                         if (size > BUF) /* should never happen */
 972                                 abort();
 973                         if ((flags & GROUPING) && size != 0)
 974                                 size += grouping_init(&gs, size, locale);
 975                         break;
 976                 default:        /* "%?" prints ?, unless ? is NUL */
 977                         if (ch == '\0')
 978                                 goto done;
 979                         /* pretend it was %c with argument ch */
 980                         cp = buf;
 981                         *cp = ch;
 982                         size = 1;
 983                         sign = '\0';
 984                         break;
 985                 }
 986
 987                 /*
 988                  * All reasonable formats wind up here.  At this point, `cp'
 989                  * points to a string which (if not flags&LADJUST) should be
 990                  * padded out to `width' places.  If flags&ZEROPAD, it should
 991                  * first be prefixed by any sign or other prefix; otherwise,
 992                  * it should be blank padded before the prefix is emitted.
 993                  * After any left-hand padding and prefixing, emit zeroes
 994                  * required by a decimal [diouxX] precision, then print the
 995                  * string proper, then emit zeroes required by any leftover
 996                  * floating precision; finally, if LADJUST, pad with blanks.
 997                  *
 998                  * Compute actual size, so we know how much to pad.
 999                  * size excludes decimal prec; realsz includes it.
1000                  */
1001                 realsz = dprec > size ? dprec : size;
1002                 if (sign)
1003                         realsz++;
1004                 if (ox[1])
1005                         realsz += 2;
1006
1007                 prsize = width > realsz ? width : realsz;
1008                 if ((unsigned)ret + prsize > INT_MAX) {
1009                         ret = EOF;
1010                         errno = EOVERFLOW;
1011                         goto error;
1012                 }
1013
1014                 /* right-adjusting blank padding */
1015                 if ((flags & (LADJUST|ZEROPAD)) == 0)
1016                         PAD(width - realsz, blanks);
1017
1018                 /* prefix */
1019                 if (sign)
1020                         PRINT(&sign, 1);
1021
1022                 if (ox[1]) {    /* ox[1] is either x, X, or \0 */
1023                         ox[0] = '0';
1024                         PRINT(ox, 2);
1025                 }
1026
1027                 /* right-adjusting zero padding */
1028                 if ((flags & (LADJUST|ZEROPAD)) == ZEROPAD)
1029                         PAD(width - realsz, zeroes);
1030
1031                 /* the string or number proper */
1032 #ifndef NO_FLOATING_POINT
1033                 if ((flags & FPT) == 0) {
1034 #endif
1035                         /* leading zeroes from decimal precision */
1036                         PAD(dprec - size, zeroes);
1037                         if (gs.grouping) {
1038                                 if (grouping_print(&gs, &io, cp, buf+BUF, locale) < 0)
1039                                         goto error;
1040                         } else {
1041                                 PRINT(cp, size);
1042                         }
1043 #ifndef NO_FLOATING_POINT
1044                 } else {        /* glue together f_p fragments */
1045                         if (!expchar) { /* %[fF] or sufficiently short %[gG] */
1046                                 if (expt <= 0) {
1047                                         PRINT(zeroes, 1);
1048                                         if (prec || flags & ALT)
1049                                                 PRINT(&decimal_point, 1);
1050                                         PAD(-expt, zeroes);
1051                                         /* already handled initial 0's */
1052                                         prec += expt;
1053                                 } else {
1054                                         if (gs.grouping) {
1055                                                 n = grouping_print(&gs, &io,
1056                                                     cp, convbuf + ndig, locale);
1057                                                 if (n < 0)
1058                                                         goto error;
1059                                                 cp += n;
1060                                         } else {
1061                                                 PRINTANDPAD(cp, convbuf + ndig,
1062                                                     expt, zeroes);
1063                                                 cp += expt;
1064                                         }
1065                                         if (prec || flags & ALT)
1066                                                 PRINT(&decimal_point, 1);
1067                                 }
1068                                 PRINTANDPAD(cp, convbuf + ndig, prec, zeroes);
1069                         } else {        /* %[eE] or sufficiently long %[gG] */
1070                                 if (prec > 1 || flags & ALT) {
1071                                         buf[0] = *cp++;
1072                                         buf[1] = decimal_point;
1073                                         PRINT(buf, 2);
1074                                         PRINT(cp, ndig-1);
1075                                         PAD(prec - ndig, zeroes);
1076                                 } else  /* XeYYY */
1077                                         PRINT(cp, 1);
1078                                 PRINT(expstr, expsize);
1079                         }
1080                 }
1081 #endif
1082                 /* left-adjusting padding (always blank) */
1083                 if (flags & LADJUST)
1084                         PAD(width - realsz, blanks);
1085
1086                 /* finally, adjust ret */
1087                 ret += prsize;
1088
1089                 FLUSH();        /* copy out the I/O vectors */
1090         }
1091 done:
1092         FLUSH();
1093 error:
1094         va_end(orgap);
1095         if (convbuf != NULL)
1096                 free(convbuf);
1097         if (__sferror(fp))
1098                 ret = EOF;
1099         if ((argtable != NULL) && (argtable != statargtable))
1100                 free (argtable);
1101         return (ret);
1102         /* NOTREACHED */
1103 }