12061Sjkh/*-
227790Sasami * Copyright (c) 1990, 1993
32061Sjkh *	The Regents of the University of California.  All rights reserved.
42061Sjkh *
515603Smarkm * This code is derived from software contributed to Berkeley by
63197Scsgr * Chris Torek.
720710Sasami *
820710Sasami * Copyright (c) 2011 The FreeBSD Foundation
93197Scsgr * All rights reserved.
102061Sjkh * Portions of this software were developed by David Chisnall
1112483Speter * under sponsorship from the FreeBSD Foundation.
122160Scsgr *
132834Swollman * Redistribution and use in source and binary forms, with or without
142061Sjkh * modification, are permitted provided that the following conditions
152061Sjkh * are met:
162160Scsgr * 1. Redistributions of source code must retain the above copyright
1717308Speter *    notice, this list of conditions and the following disclaimer.
1819320Sadam * 2. Redistributions in binary form must reproduce the above copyright
1927788Sasami *    notice, this list of conditions and the following disclaimer in the
2027788Sasami *    documentation and/or other materials provided with the distribution.
2125980Sasami * 3. Neither the name of the University nor the names of its contributors
221594Srgrimes *    may be used to endorse or promote products derived from this software
2317308Speter *    without specific prior written permission.
2417308Speter *
2517308Speter * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
2617308Speter * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
2717308Speter * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
2817308Speter * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
2917308Speter * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
3019175Sbde * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
3119175Sbde * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
3219175Sbde * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
3319175Sbde * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
3417308Speter * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
3525647Sbde * SUCH DAMAGE.
3617308Speter */
372061Sjkh
382061Sjkh#if defined(LIBC_SCCS) && !defined(lint)
391594Srgrimesstatic char sccsid[] = "@(#)vfprintf.c	8.1 (Berkeley) 6/4/93";
4025313Sbde#endif /* LIBC_SCCS and not lint */
4125313Sbde#include <sys/cdefs.h>
4225313Sbde__FBSDID("$FreeBSD$");
4325313Sbde
4425313Sbde/*
4525313Sbde * Actual printf innards.
4625313Sbde *
4725313Sbde * This code is large and complicated...
487407Srgrimes */
497108Sphk
507108Sphk#include "namespace.h"
517108Sphk#include <sys/types.h>
527407Srgrimes
537407Srgrimes#include <ctype.h>
547407Srgrimes#include <errno.h>
557108Sphk#include <limits.h>
562061Sjkh#include <locale.h>
572061Sjkh#include <stddef.h>
582061Sjkh#include <stdint.h>
5917308Speter#include <stdio.h>
602061Sjkh#include <stdlib.h>
612061Sjkh#include <string.h>
622061Sjkh#include <wchar.h>
632061Sjkh#include <printf.h>
642061Sjkh
653197Scsgr#include <stdarg.h>
662626Scsgr#include "xlocale_private.h"
672626Scsgr#include "un-namespace.h"
682061Sjkh
692061Sjkh#include "libc_private.h"
702061Sjkh#include "local.h"
712061Sjkh#include "fvwrite.h"
722061Sjkh#include "printflocal.h"
732061Sjkh
7419320Sadamstatic int	__sprint(FILE *, struct __suio *, locale_t);
752061Sjkhstatic int	__sbprintf(FILE *, locale_t, const char *, va_list) __printflike(3, 0)
762061Sjkh	__noinline;
772061Sjkhstatic char	*__wcsconv(wchar_t *, int);
782061Sjkh
792061Sjkh#define	CHAR	char
802061Sjkh#include "printfcommon.h"
812061Sjkh
822061Sjkhstruct grouping_state {
832061Sjkh	char *thousands_sep;	/* locale-specific thousands separator */
842061Sjkh	int thousep_len;	/* length of thousands_sep */
852061Sjkh	const char *grouping;	/* locale-specific numeric grouping rules */
862834Swollman	int lead;		/* sig figs before decimal or group sep */
872834Swollman	int nseps;		/* number of group separators with ' */
882834Swollman	int nrepeats;		/* number of repeats of the last group */
892834Swollman};
902834Swollman
912834Swollman/*
921594Srgrimes * Initialize the thousands' grouping state in preparation to print a
934486Sphk * number with ndigits digits. This routine returns the total number
944486Sphk * of bytes that will be needed.
954486Sphk */
964486Sphkstatic int
974486Sphkgrouping_init(struct grouping_state *gs, int ndigits, locale_t loc)
982061Sjkh{
992061Sjkh	struct lconv *locale;
10025979Sjkh
10125979Sjkh	locale = localeconv_l(loc);
10225979Sjkh	gs->grouping = locale->grouping;
10325979Sjkh	gs->thousands_sep = locale->thousands_sep;
1042061Sjkh	gs->thousep_len = strlen(gs->thousands_sep);
10525979Sjkh
1062061Sjkh	gs->nseps = gs->nrepeats = 0;
1072061Sjkh	gs->lead = ndigits;
10817308Speter	while (*gs->grouping != CHAR_MAX) {
1092061Sjkh		if (gs->lead <= *gs->grouping)
1102061Sjkh			break;
1112061Sjkh		gs->lead -= *gs->grouping;
1122061Sjkh		if (*(gs->grouping+1)) {
1132061Sjkh			gs->nseps++;
11412483Speter			gs->grouping++;
11512483Speter		} else
11612483Speter			gs->nrepeats++;
11712483Speter	}
1182061Sjkh	return ((gs->nseps + gs->nrepeats) * gs->thousep_len);
1192061Sjkh}
1208854Srgrimes
1212061Sjkh/*
1222061Sjkh * Print a number with thousands' separators.
12312483Speter */
1242061Sjkhstatic int
12518714Sachegrouping_print(struct grouping_state *gs, struct io_state *iop,
12618714Sache	       const CHAR *cp, const CHAR *ep, locale_t locale)
12718714Sache{
12817308Speter	const CHAR *cp0 = cp;
12917308Speter
13017308Speter	if (io_printandpad(iop, cp, ep, gs->lead, zeroes, locale))
13117308Speter		return (-1);
13221536Sjmacd	cp += gs->lead;
13315603Smarkm	while (gs->nseps > 0 || gs->nrepeats > 0) {
13417308Speter		if (gs->nrepeats > 0)
13517308Speter			gs->nrepeats--;
13617308Speter		else {
13717308Speter			gs->grouping--;
13817308Speter			gs->nseps--;
13917308Speter		}
14017308Speter		if (io_print(iop, gs->thousands_sep, gs->thousep_len, locale))
14117308Speter			return (-1);
14217308Speter		if (io_printandpad(iop, cp, ep, *gs->grouping, zeroes, locale))
14318362Sjkh			return (-1);
14419966Sache		cp += *gs->grouping;
14518362Sjkh	}
14617308Speter	if (cp > ep)
14717308Speter		cp = ep;
14817308Speter	return (cp - cp0);
14917308Speter}
15017308Speter
15117308Speter/*
15216550Sjkh * Flush out all the vectors defined by the given uio,
1532061Sjkh * then reset it so that it can be reused.
15417308Speter */
1552061Sjkhstatic int
15617308Speter__sprint(FILE *fp, struct __suio *uio, locale_t locale)
1572061Sjkh{
15817308Speter	int err;
15917308Speter
16017308Speter	if (uio->uio_resid == 0) {
16117308Speter		uio->uio_iovcnt = 0;
16217308Speter		return (0);
16317308Speter	}
16417466Speter	err = __sfvwrite(fp, uio);
16517308Speter	uio->uio_resid = 0;
16617308Speter	uio->uio_iovcnt = 0;
16717466Speter	return (err);
16817308Speter}
16917308Speter
17017308Speter/*
17117308Speter * Helper function for `fprintf to unbuffered unix file': creates a
17217466Speter * temporary buffer.  We only work on write-only files; this avoids
17317308Speter * worries about ungetc buffers and so forth.
17417308Speter */
17517308Speterstatic int
17617308Speter__sbprintf(FILE *fp, locale_t locale, const char *fmt, va_list ap)
17717308Speter{
17817308Speter	int ret;
17917308Speter	FILE fake = FAKE_FILE;
18017308Speter	unsigned char buf[BUFSIZ];
18117308Speter
18217308Speter	/* XXX This is probably not needed. */
18317308Speter	if (prepwrite(fp) != 0)
18417308Speter		return (EOF);
18517308Speter
18617308Speter	/* copy the important variables */
18717308Speter	fake._flags = fp->_flags & ~__SNBF;
18817308Speter	fake._file = fp->_file;
18917308Speter	fake._cookie = fp->_cookie;
19017308Speter	fake._write = fp->_write;
19117308Speter	fake._orientation = fp->_orientation;
19217308Speter	fake._mbstate = fp->_mbstate;
19317308Speter
19417308Speter	/* set up the buffer */
19517308Speter	fake._bf._base = fake._p = buf;
19617308Speter	fake._bf._size = fake._w = sizeof(buf);
19717308Speter	fake._lbfsize = 0;	/* not actually used, but Just In Case */
19817308Speter
19917308Speter	/* do the work, then copy any error status */
20026510Sjfieber	ret = __vfprintf(&fake, locale, fmt, ap);
20117308Speter	if (ret >= 0 && __fflush(&fake))
20217308Speter		ret = EOF;
20317308Speter	if (fake._flags & __SERR)
20417308Speter		fp->_flags |= __SERR;
20517962Speter	return (ret);
20617308Speter}
20717962Speter
20817962Speter/*
20917962Speter * Convert a wide character string argument for the %ls format to a multibyte
21017962Speter * string representation. If not -1, prec specifies the maximum number of
21117962Speter * bytes to output, and also means that we can't assume that the wide char.
21217962Speter * string ends is null-terminated.
21317962Speter */
21417962Speterstatic char *
21517962Speter__wcsconv(wchar_t *wcsarg, int prec)
21617962Speter{
21717962Speter	static const mbstate_t initial;
21817962Speter	mbstate_t mbs;
21917962Speter	char buf[MB_LEN_MAX];
22017962Speter	wchar_t *p;
22117962Speter	char *convbuf;
22217308Speter	size_t clen, nbytes;
22316550Sjkh
22417308Speter	/* Allocate space for the maximum number of bytes we could output. */
22517308Speter	if (prec < 0) {
22617308Speter		p = wcsarg;
22717308Speter		mbs = initial;
22816550Sjkh		nbytes = wcsrtombs(NULL, (const wchar_t **)&p, 0, &mbs);
22916550Sjkh		if (nbytes == (size_t)-1)
23017308Speter			return (NULL);
23117308Speter	} else {
23219966Sache		/*
23317962Speter		 * Optimisation: if the output precision is small enough,
2342061Sjkh		 * just allocate enough memory for the maximum instead of
23517308Speter		 * scanning the string.
23617308Speter		 */
23717308Speter		if (prec < 128)
23817308Speter			nbytes = prec;
23917308Speter		else {
24017308Speter			nbytes = 0;
24117308Speter			p = wcsarg;
24217308Speter			mbs = initial;
24312483Speter			for (;;) {
24417308Speter				clen = wcrtomb(buf, *p++, &mbs);
24512483Speter				if (clen == 0 || clen == (size_t)-1 ||
24617308Speter				    nbytes + clen > prec)
24712483Speter					break;
2482061Sjkh				nbytes += clen;
24917962Speter			}
25017962Speter		}
25117308Speter	}
25217962Speter	if ((convbuf = malloc(nbytes + 1)) == NULL)
25317962Speter		return (NULL);
25417962Speter
25517962Speter	/* Fill the output buffer. */
25617308Speter	p = wcsarg;
2572061Sjkh	mbs = initial;
25817308Speter	if ((nbytes = wcsrtombs(convbuf, (const wchar_t **)&p,
25917308Speter	    nbytes, &mbs)) == (size_t)-1) {
26017308Speter		free(convbuf);
26117308Speter		return (NULL);
26217308Speter	}
26317308Speter	convbuf[nbytes] = '\0';
26417308Speter	return (convbuf);
2652302Spaul}
2662302Spaul
2672302Spaul/*
2682302Spaul * MT-safe version
2692302Spaul */
27018714Sacheint
27110760Sachevfprintf_l(FILE * __restrict fp, locale_t locale, const char * __restrict fmt0,
27218714Sache		va_list ap)
2732302Spaul{
27410760Sache	int ret;
27518714Sache	FIX_LOCALE(locale);
27610760Sache
27710760Sache	FLOCKFILE(fp);
2782302Spaul	/* optimise fprintf(stderr) (and other unbuffered Unix files) */
2792302Spaul	if ((fp->_flags & (__SNBF|__SWR|__SRW)) == (__SNBF|__SWR) &&
2802302Spaul	    fp->_file >= 0)
2812302Spaul		ret = __sbprintf(fp, locale, fmt0, ap);
28227790Sasami	else
2832302Spaul		ret = __vfprintf(fp, locale, fmt0, ap);
2842302Spaul	FUNLOCKFILE(fp);
28517308Speter	return (ret);
28617308Speter}
28717308Speterint
28817308Spetervfprintf(FILE * __restrict fp, const char * __restrict fmt0, va_list ap)
28917308Speter{
29017308Speter	return vfprintf_l(fp, __get_locale(), fmt0, ap);
29117308Speter}
2922061Sjkh
29317308Speter/*
2942061Sjkh * The size of the buffer we use as scratch space for integer
29517308Speter * conversions, among other things.  We need enough space to
29617308Speter * write a uintmax_t in octal (plus one byte).
29717308Speter */
29817308Speter#if UINTMAX_MAX <= UINT64_MAX
29917308Speter#define	BUF	32
30017308Speter#else
30117308Speter#error "BUF must be large enough to format a uintmax_t"
30217308Speter#endif
30317308Speter
30417308Speter/*
30517308Speter * Non-MT-safe version
30617308Speter */
30717308Speterint
30817308Speter__vfprintf(FILE *fp, locale_t locale, const char *fmt0, va_list ap)
3092061Sjkh{
31017308Speter	char *fmt;		/* format string */
31117308Speter	int ch;			/* character from fmt */
31217308Speter	int n, n2;		/* handy integer (short term usage) */
31317308Speter	char *cp;		/* handy char pointer (short term usage) */
31417308Speter	int flags;		/* flags as above */
31517308Speter	int ret;		/* return value accumulator */
3163626Swollman	int width;		/* width from format (%8d), or 0 */
3173626Swollman	int prec;		/* precision from format; <0 for N/A */
3183626Swollman	char sign;		/* sign prefix (' ', '+', '-', or \0) */
3193626Swollman	struct grouping_state gs; /* thousands' grouping info */
3203626Swollman
3213626Swollman#ifndef NO_FLOATING_POINT
3223626Swollman	/*
3233626Swollman	 * We can decompose the printed representation of floating
3243626Swollman	 * point numbers into several parts, some of which may be empty:
3253626Swollman	 *
3263626Swollman	 * [+|-| ] [0x|0X] MMM . NNN [e|E|p|P] [+|-] ZZ
3277059Sroberto	 *    A       B     ---C---      D       E   F
3283626Swollman	 *
3293626Swollman	 * A:	'sign' holds this value if present; '\0' otherwise
3303626Swollman	 * B:	ox[1] holds the 'x' or 'X'; '\0' if not hexadecimal
3313626Swollman	 * C:	cp points to the string MMMNNN.  Leading and trailing
3323626Swollman	 *	zeros are not in the string and must be added.
3333626Swollman	 * D:	expchar holds this character; '\0' if no exponent, e.g. %f
3343626Swollman	 * F:	at least two digits for decimal, at least one digit for hex
33517308Speter	 */
33617308Speter	char *decimal_point;	/* locale specific decimal point */
33717308Speter	int decpt_len;		/* length of decimal_point */
33817308Speter	int signflag;		/* true if float is negative */
33917308Speter	union {			/* floating point arguments %[aAeEfFgG] */
34017308Speter		double dbl;
34117308Speter		long double ldbl;
34217308Speter	} fparg;
34317308Speter	int expt;		/* integer value of exponent */
34417308Speter	char expchar;		/* exponent character: [eEpP\0] */
3453626Swollman	char *dtoaend;		/* pointer to end of converted digits */
34617308Speter	int expsize;		/* character count for expstr */
34717308Speter	int ndig;		/* actual number of digits returned by dtoa */
34817308Speter	char expstr[MAXEXPDIG+2];	/* buffer for exponent string: e+ZZZ */
34917308Speter	char *dtoaresult;	/* buffer allocated by dtoa */
35017308Speter#endif
35117308Speter	u_long	ulval;		/* integer arguments %[diouxX] */
35217308Speter	uintmax_t ujval;	/* %j, %ll, %q, %t, %z integers */
35317308Speter	int base;		/* base for [diouxX] conversion */
35417308Speter	int dprec;		/* a copy of prec if [diouxX], 0 otherwise */
35517308Speter	int realsz;		/* field size expanded by dprec, sign, etc */
35617308Speter	int size;		/* size of converted field or string */
35717308Speter	int prsize;             /* max size of printed field */
35817820Sjkh	const char *xdigs;     	/* digits for %[xX] conversion */
35917308Speter	struct io_state io;	/* I/O buffering state */
36017820Sjkh	char buf[BUF];		/* buffer with space for digits of uintmax_t */
36117308Speter	char ox[2];		/* space for 0x; ox[1] is either x, X, or \0 */
36217820Sjkh	union arg *argtable;    /* args, built due to positional arg */
36317467Speter	union arg statargtable [STATIC_ARG_TBL_SIZE];
36417308Speter	int nextarg;            /* 1-based argument index */
36517308Speter	va_list orgap;          /* original argument pointer */
36617308Speter	char *convbuf;		/* wide to multibyte conversion result */
36717308Speter	int savserr;
36817308Speter
36917308Speter	static const char xdigs_lower[16] = "0123456789abcdef";
37017308Speter	static const char xdigs_upper[16] = "0123456789ABCDEF";
37117308Speter
37217820Sjkh	/* BEWARE, these `goto error' on error. */
37317820Sjkh#define	PRINT(ptr, len) { \
37417308Speter	if (io_print(&io, (ptr), (len), locale))	\
37517308Speter		goto error; \
37617308Speter}
37717308Speter#define	PAD(howmany, with) { \
37814119Speter	if (io_pad(&io, (howmany), (with), locale)) \
3792061Sjkh		goto error; \
3807130Srgrimes}
3817130Srgrimes#define	PRINTANDPAD(p, ep, len, with) {	\
3827130Srgrimes	if (io_printandpad(&io, (p), (ep), (len), (with), locale)) \
3832061Sjkh		goto error; \
38425783Speter}
38517308Speter#define	FLUSH() { \
3862685Srgrimes	if (io_flush(&io, locale)) \
3876927Snate		goto error; \
38827790Sasami}
38927790Sasami
39027790Sasami	/*
39126504Sjhay	 * Get the argument indexed by nextarg.   If the argument table is
3923197Scsgr	 * built, use it to get the argument.  If its not, get the next
3933197Scsgr	 * argument (and arguments must be gotten sequentially).
39412166Sjkh	 */
39512485Sjkh#define GETARG(type) \
3963197Scsgr	((argtable != NULL) ? *((type*)(&argtable[nextarg++])) : \
39725313Sbde	    (nextarg++, va_arg(ap, type)))
39826152Speter
3992061Sjkh	/*
4002061Sjkh	 * To extend shorts properly, we need both signed and unsigned
4012061Sjkh	 * argument extraction methods.
40216786Snate	 */
4032883Sphk#define	SARG() \
40427790Sasami	(flags&LONGINT ? GETARG(long) : \
40517308Speter	    flags&SHORTINT ? (long)(short)GETARG(int) : \
4067281Srgrimes	    flags&CHARINT ? (long)(signed char)GETARG(int) : \
4073242Spaul	    (long)GETARG(int))
4083242Spaul#define	UARG() \
4097171Sats	(flags&LONGINT ? GETARG(u_long) : \
4102061Sjkh	    flags&SHORTINT ? (u_long)(u_short)GETARG(int) : \
4113213Spst	    flags&CHARINT ? (u_long)(u_char)GETARG(int) : \
41217308Speter	    (u_long)GETARG(u_int))
41317308Speter#define	INTMAX_SIZE	(INTMAXT|SIZET|PTRDIFFT|LLONGINT)
4145749Swollman#define SJARG() \
4155772Swollman	(flags&INTMAXT ? GETARG(intmax_t) : \
41617308Speter	    flags&SIZET ? (intmax_t)GETARG(ssize_t) : \
41717308Speter	    flags&PTRDIFFT ? (intmax_t)GETARG(ptrdiff_t) : \
41826504Sjhay	    (intmax_t)GETARG(long long))
4192061Sjkh#define	UJARG() \
42017308Speter	(flags&INTMAXT ? GETARG(uintmax_t) : \
42117308Speter	    flags&SIZET ? (uintmax_t)GETARG(size_t) : \
42217308Speter	    flags&PTRDIFFT ? (uintmax_t)GETARG(ptrdiff_t) : \
4235366Snate	    (uintmax_t)GETARG(unsigned long long))
42417820Sjkh
42517467Speter	/*
42617820Sjkh	 * Get * arguments, including the form *nn$.  Preserve the nextarg
42717308Speter	 * that the argument can be gotten once the type is determined.
42818674Speter	 */
42918674Speter#define GETASTER(val) \
43017820Sjkh	n2 = 0; \
43117308Speter	cp = fmt; \
43217820Sjkh	while (is_digit(*cp)) { \
43317308Speter		n2 = 10 * n2 + to_digit(*cp); \
43417820Sjkh		cp++; \
43517308Speter	} \
43617820Sjkh	if (*cp == '$') { \
43717308Speter		int hold = nextarg; \
43817820Sjkh		if (argtable == NULL) { \
43917308Speter			argtable = statargtable; \
4405728Swollman			if (__find_arguments (fmt0, orgap, &argtable)) { \
44117820Sjkh				ret = EOF; \
44217308Speter				goto error; \
44321922Sjkh			} \
44421922Sjkh		} \
44518392Speter		nextarg = n2; \
44618392Speter		val = GETARG (int); \
44718392Speter		nextarg = hold; \
44818392Speter		fmt = ++cp; \
44918392Speter	} else { \
45018392Speter		val = GETARG (int); \
4515366Snate	}
45217308Speter
45317308Speter	if (__use_xprintf == 0 && getenv("USE_XPRINTF"))
45417308Speter		__use_xprintf = 1;
4552061Sjkh	if (__use_xprintf > 0)
45624754Sjdp		return (__xvprintf(fp, fmt0, ap));
45724754Sjdp
45824754Sjdp	/* sorry, fprintf(read_only_file, "") returns EOF, not 0 */
45924754Sjdp	if (prepwrite(fp) != 0) {
4608295Srgrimes		errno = EBADF;
46117820Sjkh		return (EOF);
46217308Speter	}
4638295Srgrimes
4648489Srgrimes	savserr = fp->_flags & __SERR;
46517820Sjkh	fp->_flags &= ~__SERR;
46617308Speter
4678489Srgrimes	convbuf = NULL;
4688489Srgrimes	fmt = (char *)fmt0;
46917820Sjkh	argtable = NULL;
47017308Speter	nextarg = 1;
4718489Srgrimes	va_copy(orgap, ap);
4728295Srgrimes	io_init(&io, fp);
47317820Sjkh	ret = 0;
47417308Speter#ifndef NO_FLOATING_POINT
4758295Srgrimes	dtoaresult = NULL;
4762160Scsgr	decimal_point = localeconv_l(locale)->decimal_point;
47717820Sjkh	/* The overwhelmingly common case is decpt_len == 1. */
47817308Speter	decpt_len = (decimal_point[1] == '\0' ? 1 : strlen(decimal_point));
4792160Scsgr#endif
4802279Spaul
48117820Sjkh	/*
48217308Speter	 * Scan the format for conversions (`%' character).
4832279Spaul	 */
48417234Sjraynard	for (;;) {
48517820Sjkh		for (cp = fmt; (ch = *fmt) != '\0' && ch != '%'; fmt++)
48617308Speter			/* void */;
48711772Snate		if ((n = fmt - cp) != 0) {
4883197Scsgr			if ((unsigned)ret + n > INT_MAX) {
48917820Sjkh				ret = EOF;
49017308Speter				errno = EOVERFLOW;
4912626Scsgr				goto error;
4928304Srgrimes			}
49317820Sjkh			PRINT(cp, n);
49417308Speter			ret += n;
4958304Srgrimes		}
4962061Sjkh		if (ch == '\0')
49717308Speter			goto done;
49817308Speter		fmt++;		/* skip over '%' */
49917308Speter
50017308Speter		flags = 0;
50111806Sphk		dprec = 0;
50219175Sbde		width = 0;
50321673Sjkh		prec = -1;
50421536Sjmacd		gs.grouping = NULL;
50519175Sbde		sign = '\0';
50619175Sbde		ox[1] = '\0';
50719175Sbde
50819175Sbderflag:		ch = *fmt++;
50919175Sbdereswitch:	switch (ch) {
51017308Speter		case ' ':
51119175Sbde			/*-
5122061Sjkh			 * ``If the space and + flags both appear, the space
5131594Srgrimes			 * flag will be ignored.''
514			 *	-- ANSI X3J11
515			 */
516			if (!sign)
517				sign = ' ';
518			goto rflag;
519		case '#':
520			flags |= ALT;
521			goto rflag;
522		case '*':
523			/*-
524			 * ``A negative field width argument is taken as a
525			 * - flag followed by a positive field width.''
526			 *	-- ANSI X3J11
527			 * They don't exclude field widths read from args.
528			 */
529			GETASTER (width);
530			if (width >= 0)
531				goto rflag;
532			width = -width;
533			/* FALLTHROUGH */
534		case '-':
535			flags |= LADJUST;
536			goto rflag;
537		case '+':
538			sign = '+';
539			goto rflag;
540		case '\'':
541			flags |= GROUPING;
542			goto rflag;
543		case '.':
544			if ((ch = *fmt++) == '*') {
545				GETASTER (prec);
546				goto rflag;
547			}
548			prec = 0;
549			while (is_digit(ch)) {
550				prec = 10 * prec + to_digit(ch);
551				ch = *fmt++;
552			}
553			goto reswitch;
554		case '0':
555			/*-
556			 * ``Note that 0 is taken as a flag, not as the
557			 * beginning of a field width.''
558			 *	-- ANSI X3J11
559			 */
560			flags |= ZEROPAD;
561			goto rflag;
562		case '1': case '2': case '3': case '4':
563		case '5': case '6': case '7': case '8': case '9':
564			n = 0;
565			do {
566				n = 10 * n + to_digit(ch);
567				ch = *fmt++;
568			} while (is_digit(ch));
569			if (ch == '$') {
570				nextarg = n;
571				if (argtable == NULL) {
572					argtable = statargtable;
573					if (__find_arguments (fmt0, orgap,
574							      &argtable)) {
575						ret = EOF;
576						goto error;
577					}
578				}
579				goto rflag;
580			}
581			width = n;
582			goto reswitch;
583#ifndef NO_FLOATING_POINT
584		case 'L':
585			flags |= LONGDBL;
586			goto rflag;
587#endif
588		case 'h':
589			if (flags & SHORTINT) {
590				flags &= ~SHORTINT;
591				flags |= CHARINT;
592			} else
593				flags |= SHORTINT;
594			goto rflag;
595		case 'j':
596			flags |= INTMAXT;
597			goto rflag;
598		case 'l':
599			if (flags & LONGINT) {
600				flags &= ~LONGINT;
601				flags |= LLONGINT;
602			} else
603				flags |= LONGINT;
604			goto rflag;
605		case 'q':
606			flags |= LLONGINT;	/* not necessarily */
607			goto rflag;
608		case 't':
609			flags |= PTRDIFFT;
610			goto rflag;
611		case 'z':
612			flags |= SIZET;
613			goto rflag;
614		case 'C':
615			flags |= LONGINT;
616			/*FALLTHROUGH*/
617		case 'c':
618			if (flags & LONGINT) {
619				static const mbstate_t initial;
620				mbstate_t mbs;
621				size_t mbseqlen;
622
623				mbs = initial;
624				mbseqlen = wcrtomb(cp = buf,
625				    (wchar_t)GETARG(wint_t), &mbs);
626				if (mbseqlen == (size_t)-1) {
627					fp->_flags |= __SERR;
628					goto error;
629				}
630				size = (int)mbseqlen;
631			} else {
632				*(cp = buf) = GETARG(int);
633				size = 1;
634			}
635			sign = '\0';
636			break;
637		case 'D':
638			flags |= LONGINT;
639			/*FALLTHROUGH*/
640		case 'd':
641		case 'i':
642			if (flags & INTMAX_SIZE) {
643				ujval = SJARG();
644				if ((intmax_t)ujval < 0) {
645					ujval = -ujval;
646					sign = '-';
647				}
648			} else {
649				ulval = SARG();
650				if ((long)ulval < 0) {
651					ulval = -ulval;
652					sign = '-';
653				}
654			}
655			base = 10;
656			goto number;
657#ifndef NO_FLOATING_POINT
658		case 'a':
659		case 'A':
660			if (ch == 'a') {
661				ox[1] = 'x';
662				xdigs = xdigs_lower;
663				expchar = 'p';
664			} else {
665				ox[1] = 'X';
666				xdigs = xdigs_upper;
667				expchar = 'P';
668			}
669			if (prec >= 0)
670				prec++;
671			if (dtoaresult != NULL)
672				freedtoa(dtoaresult);
673			if (flags & LONGDBL) {
674				fparg.ldbl = GETARG(long double);
675				dtoaresult = cp =
676				    __hldtoa(fparg.ldbl, xdigs, prec,
677				    &expt, &signflag, &dtoaend);
678			} else {
679				fparg.dbl = GETARG(double);
680				dtoaresult = cp =
681				    __hdtoa(fparg.dbl, xdigs, prec,
682				    &expt, &signflag, &dtoaend);
683			}
684			if (prec < 0)
685				prec = dtoaend - cp;
686			if (expt == INT_MAX)
687				ox[1] = '\0';
688			goto fp_common;
689		case 'e':
690		case 'E':
691			expchar = ch;
692			if (prec < 0)	/* account for digit before decpt */
693				prec = DEFPREC + 1;
694			else
695				prec++;
696			goto fp_begin;
697		case 'f':
698		case 'F':
699			expchar = '\0';
700			goto fp_begin;
701		case 'g':
702		case 'G':
703			expchar = ch - ('g' - 'e');
704			if (prec == 0)
705				prec = 1;
706fp_begin:
707			if (prec < 0)
708				prec = DEFPREC;
709			if (dtoaresult != NULL)
710				freedtoa(dtoaresult);
711			if (flags & LONGDBL) {
712				fparg.ldbl = GETARG(long double);
713				dtoaresult = cp =
714				    __ldtoa(&fparg.ldbl, expchar ? 2 : 3, prec,
715				    &expt, &signflag, &dtoaend);
716			} else {
717				fparg.dbl = GETARG(double);
718				dtoaresult = cp =
719				    dtoa(fparg.dbl, expchar ? 2 : 3, prec,
720				    &expt, &signflag, &dtoaend);
721				if (expt == 9999)
722					expt = INT_MAX;
723			}
724fp_common:
725			if (signflag)
726				sign = '-';
727			if (expt == INT_MAX) {	/* inf or nan */
728				if (*cp == 'N') {
729					cp = (ch >= 'a') ? "nan" : "NAN";
730					sign = '\0';
731				} else
732					cp = (ch >= 'a') ? "inf" : "INF";
733				size = 3;
734				flags &= ~ZEROPAD;
735				break;
736			}
737			flags |= FPT;
738			ndig = dtoaend - cp;
739			if (ch == 'g' || ch == 'G') {
740				if (expt > -4 && expt <= prec) {
741					/* Make %[gG] smell like %[fF] */
742					expchar = '\0';
743					if (flags & ALT)
744						prec -= expt;
745					else
746						prec = ndig - expt;
747					if (prec < 0)
748						prec = 0;
749				} else {
750					/*
751					 * Make %[gG] smell like %[eE], but
752					 * trim trailing zeroes if no # flag.
753					 */
754					if (!(flags & ALT))
755						prec = ndig;
756				}
757			}
758			if (expchar) {
759				expsize = exponent(expstr, expt - 1, expchar);
760				size = expsize + prec;
761				if (prec > 1 || flags & ALT)
762					size += decpt_len;
763			} else {
764				/* space for digits before decimal point */
765				if (expt > 0)
766					size = expt;
767				else	/* "0" */
768					size = 1;
769				/* space for decimal pt and following digits */
770				if (prec || flags & ALT)
771					size += prec + decpt_len;
772				if ((flags & GROUPING) && expt > 0)
773					size += grouping_init(&gs, expt, locale);
774			}
775			break;
776#endif /* !NO_FLOATING_POINT */
777		case 'n':
778			/*
779			 * Assignment-like behavior is specified if the
780			 * value overflows or is otherwise unrepresentable.
781			 * C99 says to use `signed char' for %hhn conversions.
782			 */
783			if (flags & LLONGINT)
784				*GETARG(long long *) = ret;
785			else if (flags & SIZET)
786				*GETARG(ssize_t *) = (ssize_t)ret;
787			else if (flags & PTRDIFFT)
788				*GETARG(ptrdiff_t *) = ret;
789			else if (flags & INTMAXT)
790				*GETARG(intmax_t *) = ret;
791			else if (flags & LONGINT)
792				*GETARG(long *) = ret;
793			else if (flags & SHORTINT)
794				*GETARG(short *) = ret;
795			else if (flags & CHARINT)
796				*GETARG(signed char *) = ret;
797			else
798				*GETARG(int *) = ret;
799			continue;	/* no output */
800		case 'O':
801			flags |= LONGINT;
802			/*FALLTHROUGH*/
803		case 'o':
804			if (flags & INTMAX_SIZE)
805				ujval = UJARG();
806			else
807				ulval = UARG();
808			base = 8;
809			goto nosign;
810		case 'p':
811			/*-
812			 * ``The argument shall be a pointer to void.  The
813			 * value of the pointer is converted to a sequence
814			 * of printable characters, in an implementation-
815			 * defined manner.''
816			 *	-- ANSI X3J11
817			 */
818			ujval = (uintmax_t)(uintptr_t)GETARG(void *);
819			base = 16;
820			xdigs = xdigs_lower;
821			flags = flags | INTMAXT;
822			ox[1] = 'x';
823			goto nosign;
824		case 'S':
825			flags |= LONGINT;
826			/*FALLTHROUGH*/
827		case 's':
828			if (flags & LONGINT) {
829				wchar_t *wcp;
830
831				if (convbuf != NULL)
832					free(convbuf);
833				if ((wcp = GETARG(wchar_t *)) == NULL)
834					cp = "(null)";
835				else {
836					convbuf = __wcsconv(wcp, prec);
837					if (convbuf == NULL) {
838						fp->_flags |= __SERR;
839						goto error;
840					}
841					cp = convbuf;
842				}
843			} else if ((cp = GETARG(char *)) == NULL)
844				cp = "(null)";
845			size = (prec >= 0) ? strnlen(cp, prec) : strlen(cp);
846			sign = '\0';
847			break;
848		case 'U':
849			flags |= LONGINT;
850			/*FALLTHROUGH*/
851		case 'u':
852			if (flags & INTMAX_SIZE)
853				ujval = UJARG();
854			else
855				ulval = UARG();
856			base = 10;
857			goto nosign;
858		case 'X':
859			xdigs = xdigs_upper;
860			goto hex;
861		case 'x':
862			xdigs = xdigs_lower;
863hex:
864			if (flags & INTMAX_SIZE)
865				ujval = UJARG();
866			else
867				ulval = UARG();
868			base = 16;
869			/* leading 0x/X only if non-zero */
870			if (flags & ALT &&
871			    (flags & INTMAX_SIZE ? ujval != 0 : ulval != 0))
872				ox[1] = ch;
873
874			flags &= ~GROUPING;
875			/* unsigned conversions */
876nosign:			sign = '\0';
877			/*-
878			 * ``... diouXx conversions ... if a precision is
879			 * specified, the 0 flag will be ignored.''
880			 *	-- ANSI X3J11
881			 */
882number:			if ((dprec = prec) >= 0)
883				flags &= ~ZEROPAD;
884
885			/*-
886			 * ``The result of converting a zero value with an
887			 * explicit precision of zero is no characters.''
888			 *	-- ANSI X3J11
889			 *
890			 * ``The C Standard is clear enough as is.  The call
891			 * printf("%#.0o", 0) should print 0.''
892			 *	-- Defect Report #151
893			 */
894			cp = buf + BUF;
895			if (flags & INTMAX_SIZE) {
896				if (ujval != 0 || prec != 0 ||
897				    (flags & ALT && base == 8))
898					cp = __ujtoa(ujval, cp, base,
899					    flags & ALT, xdigs);
900			} else {
901				if (ulval != 0 || prec != 0 ||
902				    (flags & ALT && base == 8))
903					cp = __ultoa(ulval, cp, base,
904					    flags & ALT, xdigs);
905			}
906			size = buf + BUF - cp;
907			if (size > BUF)	/* should never happen */
908				abort();
909			if ((flags & GROUPING) && size != 0)
910				size += grouping_init(&gs, size, locale);
911			break;
912		default:	/* "%?" prints ?, unless ? is NUL */
913			if (ch == '\0')
914				goto done;
915			/* pretend it was %c with argument ch */
916			cp = buf;
917			*cp = ch;
918			size = 1;
919			sign = '\0';
920			break;
921		}
922
923		/*
924		 * All reasonable formats wind up here.  At this point, `cp'
925		 * points to a string which (if not flags&LADJUST) should be
926		 * padded out to `width' places.  If flags&ZEROPAD, it should
927		 * first be prefixed by any sign or other prefix; otherwise,
928		 * it should be blank padded before the prefix is emitted.
929		 * After any left-hand padding and prefixing, emit zeroes
930		 * required by a decimal [diouxX] precision, then print the
931		 * string proper, then emit zeroes required by any leftover
932		 * floating precision; finally, if LADJUST, pad with blanks.
933		 *
934		 * Compute actual size, so we know how much to pad.
935		 * size excludes decimal prec; realsz includes it.
936		 */
937		realsz = dprec > size ? dprec : size;
938		if (sign)
939			realsz++;
940		if (ox[1])
941			realsz += 2;
942
943		prsize = width > realsz ? width : realsz;
944		if ((unsigned)ret + prsize > INT_MAX) {
945			ret = EOF;
946			errno = EOVERFLOW;
947			goto error;
948		}
949
950		/* right-adjusting blank padding */
951		if ((flags & (LADJUST|ZEROPAD)) == 0)
952			PAD(width - realsz, blanks);
953
954		/* prefix */
955		if (sign)
956			PRINT(&sign, 1);
957
958		if (ox[1]) {	/* ox[1] is either x, X, or \0 */
959			ox[0] = '0';
960			PRINT(ox, 2);
961		}
962
963		/* right-adjusting zero padding */
964		if ((flags & (LADJUST|ZEROPAD)) == ZEROPAD)
965			PAD(width - realsz, zeroes);
966
967		/* the string or number proper */
968#ifndef NO_FLOATING_POINT
969		if ((flags & FPT) == 0) {
970#endif
971			/* leading zeroes from decimal precision */
972			PAD(dprec - size, zeroes);
973			if (gs.grouping) {
974				if (grouping_print(&gs, &io, cp, buf+BUF, locale) < 0)
975					goto error;
976			} else {
977				PRINT(cp, size);
978			}
979#ifndef NO_FLOATING_POINT
980		} else {	/* glue together f_p fragments */
981			if (!expchar) {	/* %[fF] or sufficiently short %[gG] */
982				if (expt <= 0) {
983					PRINT(zeroes, 1);
984					if (prec || flags & ALT)
985						PRINT(decimal_point,decpt_len);
986					PAD(-expt, zeroes);
987					/* already handled initial 0's */
988					prec += expt;
989				} else {
990					if (gs.grouping) {
991						n = grouping_print(&gs, &io,
992						    cp, dtoaend, locale);
993						if (n < 0)
994							goto error;
995						cp += n;
996					} else {
997						PRINTANDPAD(cp, dtoaend,
998						    expt, zeroes);
999						cp += expt;
1000					}
1001					if (prec || flags & ALT)
1002						PRINT(decimal_point,decpt_len);
1003				}
1004				PRINTANDPAD(cp, dtoaend, prec, zeroes);
1005			} else {	/* %[eE] or sufficiently long %[gG] */
1006				if (prec > 1 || flags & ALT) {
1007					PRINT(cp++, 1);
1008					PRINT(decimal_point, decpt_len);
1009					PRINT(cp, ndig-1);
1010					PAD(prec - ndig, zeroes);
1011				} else	/* XeYYY */
1012					PRINT(cp, 1);
1013				PRINT(expstr, expsize);
1014			}
1015		}
1016#endif
1017		/* left-adjusting padding (always blank) */
1018		if (flags & LADJUST)
1019			PAD(width - realsz, blanks);
1020
1021		/* finally, adjust ret */
1022		ret += prsize;
1023
1024		FLUSH();	/* copy out the I/O vectors */
1025	}
1026done:
1027	FLUSH();
1028error:
1029	va_end(orgap);
1030#ifndef NO_FLOATING_POINT
1031	if (dtoaresult != NULL)
1032		freedtoa(dtoaresult);
1033#endif
1034	if (convbuf != NULL)
1035		free(convbuf);
1036	if (__sferror(fp))
1037		ret = EOF;
1038	else
1039		fp->_flags |= savserr;
1040	if ((argtable != NULL) && (argtable != statargtable))
1041		free (argtable);
1042	return (ret);
1043	/* NOTREACHED */
1044}
1045
1046