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