/*-

 * Copyright (c) 1990, 1993

 *	The Regents of the University of California.  All rights reserved.

 *

 * This code is derived from software contributed to Berkeley by

 * Chris Torek.

 *

 * Redistribution and use in source and binary forms, with or without

 * modification, are permitted provided that the following conditions

 * are met:

 * 1. Redistributions of source code must retain the above copyright

 *    notice, this list of conditions and the following disclaimer.

 * 2. Redistributions in binary form must reproduce the above copyright

 *    notice, this list of conditions and the following disclaimer in the

 *    documentation and/or other materials provided with the distribution.

 * 4. Neither the name of the University nor the names of its contributors

 *    may be used to endorse or promote products derived from this software

 *    without specific prior written permission.

 *

 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND

 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE

 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE

 * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE

 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL

 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS

 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)

 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT

 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY

 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF

 * SUCH DAMAGE.

 * * Portions Copyright (c) 2005-2006 Nokia Corporation and/or its subsidiary(-ies). All rights reserved. 

 */

#if defined(LIBC_SCCS) && !defined(lint)

static char sccsid[] = "@(#)vfprintf.c	8.1 (Berkeley) 6/4/93";

#endif /* LIBC_SCCS and not lint */

#include <sys/cdefs.h>

#ifndef __SYMBIAN32__

__FBSDID("$FreeBSD: src/lib/libc/stdio/vfprintf.c,v 1.69 2005/04/16 22:36:51 das Exp $");

#endif

/*

 * Actual printf innards.

 *

 * This code is large and complicated...

 */

#include "namespace.h"

#include <sys/types.h>

#include <ctype.h>

#include <limits.h>

#include <locale.h>

#include <stddef.h>

#include <stdint.h>

#include <stdio.h>

#include <stdlib.h>

#include <string.h>

#include <wchar.h>

#include <stdarg.h>

#include <math.h>

#include "fpmath.h"

#include "un-namespace.h"

#include "libc_private.h"

#include "local.h"

#include "fvwrite.h"

#include "common_def.h"

#if (defined(__SYMBIAN32__) && (defined(__WINSCW__) || defined(__WINS__)))

#include "libc_wsd_defs.h"

#endif

#include "gdtoaimp.h"

union arg {

	int	intarg;

	u_int	uintarg;

	long	longarg;

	u_long	ulongarg;

	long long longlongarg;

	unsigned long long ulonglongarg;

	ptrdiff_t ptrdiffarg;

	size_t	sizearg;

	intmax_t intmaxarg;

	uintmax_t uintmaxarg;

	void	*pvoidarg;

	char	*pchararg;

	signed char *pschararg;

	short	*pshortarg;

	int	*pintarg;

	long	*plongarg;

	long long *plonglongarg;

	ptrdiff_t *pptrdiffarg;

	size_t	*psizearg;

	intmax_t *pintmaxarg;

#ifndef NO_FLOATING_POINT

	double	doublearg;

	long double longdoublearg;

#endif

	wint_t	wintarg;

	wchar_t	*pwchararg;

};

/*

 * Type ids for argument type table.

 */

enum typeid {

	T_UNUSED, TP_SHORT, T_INT, T_U_INT, TP_INT,

	T_LONG, T_U_LONG, TP_LONG, T_LLONG, T_U_LLONG, TP_LLONG,

	T_PTRDIFFT, TP_PTRDIFFT, T_SIZET, TP_SIZET,

	T_INTMAXT, T_UINTMAXT, TP_INTMAXT, TP_VOID, TP_CHAR, TP_SCHAR,

	T_DOUBLE, T_LONG_DOUBLE, T_WINT, TP_WCHAR

};

static int	__sprint(FILE *, struct __suio *);

static int	__sbprintf(FILE *, const char *, va_list) __printflike(2, 0);

static char	*__ujtoa(uintmax_t, char *, int, int, const char *, int, char,

		    const char *);

static char	*__ultoa(u_long, char *, int, int, const char *, int, char,

		    const char *);

static char	*__wcsconv(wchar_t *, int);

static void	__find_arguments(const char *, va_list, union arg **);

static void	__grow_type_table(int, enum typeid **, int *);

/*

 * Flush out all the vectors defined by the given uio,

 * then reset it so that it can be reused.

 */

static int

__sprint(FILE *fp, struct __suio *uio)

{

	int err;

	if (uio->uio_resid == 0) {

		uio->uio_iovcnt = 0;

		return (0);

	}

	err = __sfvwrite(fp, uio);

	uio->uio_resid = 0;

	uio->uio_iovcnt = 0;

	return (err);

}

/*

 * Helper function for `fprintf to unbuffered unix file': creates a

 * temporary buffer.  We only work on write-only files; this avoids

 * worries about ungetc buffers and so forth.

 */

static int

__sbprintf(FILE *fp, const char *fmt, va_list ap)

{

	int ret;

	FILE fake;

	unsigned char buf[BUFSIZ];

	/* copy the important variables */

	fake._flags = fp->_flags & ~__SNBF;

	fake._file = fp->_file;

	fake._cookie = fp->_cookie;

	fake._write = fp->_write;

	fake._extra = fp->_extra;

	/* set up the buffer */

	fake._bf._base = fake._p = buf;

	fake._bf._size = fake._w = sizeof(buf);

	fake._lbfsize = 0;	/* not actually used, but Just In Case */

	/* do the work, then copy any error status */

	ret = __vfprintf(&fake, fmt, ap);

	if (ret >= 0 && __fflush(&fake))

		ret = EOF;

	if (fake._flags & __SERR)

		fp->_flags |= __SERR;

	return (ret);

}

/*

 * Macros for converting digits to letters and vice versa

 */

#define	to_digit(c)	((c) - '0')

#define is_digit(c)	((unsigned)to_digit(c) <= 9)

#define	to_char(n)	((n) + '0')

/*

 * Convert an unsigned long to ASCII for printf purposes, returning

 * a pointer to the first character of the string representation.

 * Octal numbers can be forced to have a leading zero; hex numbers

 * use the given digits.

 */

static char *

__ultoa(u_long val, char *endp, int base, int octzero, const char *xdigs,

	int needgrp, char thousep, const char *grp)

{

	char *cp = endp;

	long sval;

	int ndig;

	/*

	 * Handle the three cases separately, in the hope of getting

	 * better/faster code.

	 */

	switch (base) {

	case 10:

		if (val < 10) {	/* many numbers are 1 digit */

			*--cp = to_char(val);

			return (cp);

		}

		ndig = 0;

		/*

		 * On many machines, unsigned arithmetic is harder than

		 * signed arithmetic, so we do at most one unsigned mod and

		 * divide; this is sufficient to reduce the range of

		 * the incoming value to where signed arithmetic works.

		 */

		if (val > LONG_MAX) {

			*--cp = to_char(val % 10);

			ndig++;

			sval = val / 10;

		} else

			sval = val;

		do {

			*--cp = to_char(sval % 10);

			ndig++;

			/*

			 * If (*grp == CHAR_MAX) then no more grouping

			 * should be performed.

			 */

			if (needgrp && ndig == *grp && *grp != CHAR_MAX

					&& sval > 9) {

				*--cp = thousep;

				ndig = 0;

				/*

				 * If (*(grp+1) == '\0') then we have to

				 * use *grp character (last grouping rule)

				 * for all next cases

				 */

				if (*(grp+1) != '\0')

					grp++;

			}

			sval /= 10;

		} while (sval != 0);

		break;

	case 8:

		do {

			*--cp = to_char(val & 7);

			val >>= 3;

		} while (val);

		if (octzero && *cp != '0')

			*--cp = '0';

		break;

	case 16:

		do {

			*--cp = xdigs[val & 15];

			val >>= 4;

		} while (val);

		break;

	default:			/* oops */

		abort();

	}

	return (cp);

}

/* Identical to __ultoa, but for intmax_t. */

static char *

__ujtoa(uintmax_t val, char *endp, int base, int octzero, const char *xdigs, 

	int needgrp, char thousep, const char *grp)

{

	char *cp = endp;

	intmax_t sval;

	int ndig;

	/* quick test for small values; __ultoa is typically much faster */

	/* (perhaps instead we should run until small, then call __ultoa?) */

	if (val <= ULONG_MAX)

		return (__ultoa((u_long)val, endp, base, octzero, xdigs,

		    needgrp, thousep, grp));

	switch (base) {

	case 10:

		if (val < 10) {

			*--cp = to_char(val % 10);

			return (cp);

		}

		ndig = 0;

		if (val > INTMAX_MAX) {

			*--cp = to_char(val % 10);

			ndig++;

			sval = val / 10;

		} else

			sval = val;

		do {

			*--cp = to_char(sval % 10);

			ndig++;

			/*

			 * If (*grp == CHAR_MAX) then no more grouping

			 * should be performed.

			 */

			if (needgrp && *grp != CHAR_MAX && ndig == *grp

					&& sval > 9) {

				*--cp = thousep;

				ndig = 0;

				/*

				 * If (*(grp+1) == '\0') then we have to

				 * use *grp character (last grouping rule)

				 * for all next cases

				 */

				if (*(grp+1) != '\0')

					grp++;

			}

			sval /= 10;

		} while (sval != 0);

		break;

	case 8:

		do {

			*--cp = to_char(val & 7);

			val >>= 3;

		} while (val);

		if (octzero && *cp != '0')

			*--cp = '0';

		break;

	case 16:

		do {

			*--cp = xdigs[val & 15];

			val >>= 4;

		} while (val);

		break;

	default:

		abort();

	}

	return (cp);

}

/*

 * Convert a wide character string argument for the %ls format to a multibyte

 * string representation. ``prec'' specifies the maximum number of bytes

 * to output. If ``prec'' is greater than or equal to zero, we can't assume

 * that the wide char. string ends in a null character.

 */

static char *

__wcsconv(wchar_t *wcsarg, int prec)

{

	static const mbstate_t initial = {0,{0}};

	mbstate_t mbs;

	char buf[MB_LEN_MAX];

	wchar_t *p;

	char *convbuf, *mbp;

	size_t clen, nbytes;

	/*

	 * Determine the number of bytes to output and allocate space for

	 * the output.

	 */

	if (prec >= 0) {

		nbytes = 0;

		p = wcsarg;

		mbs = initial;

		for (;;) {

			clen = wcrtomb(buf, *p++, &mbs);

			if (clen == 0 || clen == (size_t)-1 ||

			    nbytes + clen > prec)

				break;

			nbytes += clen;

		}

	} else {

		p = wcsarg;

		mbs = initial;

		nbytes = wcsrtombs(NULL, (const wchar_t **)&p, 0, &mbs);

		if (nbytes == (size_t)-1)

			return (NULL);

	}

	if ((convbuf = malloc(nbytes + 1)) == NULL)

		return (NULL);

	/*

	 * Fill the output buffer with the multibyte representations of as

	 * many wide characters as will fit.

	 */

	mbp = convbuf;

	p = wcsarg;

	mbs = initial;

	while (mbp - convbuf < nbytes) {

		clen = wcrtomb(mbp, *p++, &mbs);

		if (clen == 0 || clen == (size_t)-1)

			break;

		mbp += clen;

	}

	if (clen == (size_t)-1) {

		free(convbuf);

		return (NULL);

	}

	*mbp = '\0';

	return (convbuf);

}

/*

 * MT-safe version

 */

EXPORT_C

int

vfprintf(FILE * __restrict fp, const char * __restrict fmt0, va_list ap)

{

	int ret;

	FLOCKFILE(fp);

	ret = __vfprintf(fp, fmt0, ap);

	FUNLOCKFILE(fp);

	return (ret);

}

#ifndef NO_FLOATING_POINT

#include <float.h>

#include <math.h>

#include "floatio.h"

#ifndef __SYMBIAN32__

#include "gdtoa.h"

#endif //__SYMBIAN32__

#define	DEFPREC		6

static int exponent(char *, int, int);

#endif /* !NO_FLOATING_POINT */

/*

 * The size of the buffer we use as scratch space for integer

 * conversions, among other things.  Technically, we would need the

 * most space for base 10 conversions with thousands' grouping

 * characters between each pair of digits.  100 bytes is a

 * conservative overestimate even for a 128-bit uintmax_t.

 */

#define	BUF	100

#define STATIC_ARG_TBL_SIZE 8           /* Size of static argument table. */

/*

 * Flags used during conversion.

 */

#define	ALT		0x001		/* alternate form */

#define	LADJUST		0x004		/* left adjustment */

#define	LONGDBL		0x008		/* long double */

#define	LONGINT		0x010		/* long integer */

#define	LLONGINT	0x020		/* long long integer */

#define	SHORTINT	0x040		/* short integer */

#define	ZEROPAD		0x080		/* zero (as opposed to blank) pad */

#define	FPT		0x100		/* Floating point number */

#define	GROUPING	0x200		/* use grouping ("'" flag) */

					/* C99 additional size modifiers: */

#define	SIZET		0x400		/* size_t */

#define	PTRDIFFT	0x800		/* ptrdiff_t */

#define	INTMAXT		0x1000		/* intmax_t */

#define	CHARINT		0x2000		/* print char using int format */

/*

 * Non-MT-safe version

 */

int

__vfprintf(FILE *fp, const char *fmt0, va_list ap)

{

	char *fmt;		/* format string */

	int ch;			/* character from fmt */

	int n, n2;		/* handy integer (short term usage) */

	char *cp;		/* handy char pointer (short term usage) */

	struct __siov *iovp;	/* for PRINT macro */

	int flags;		/* flags as above */

	int ret;		/* return value accumulator */

	int width;		/* width from format (%8d), or 0 */

	int prec;		/* precision from format; <0 for N/A */

	char sign;		/* sign prefix (' ', '+', '-', or \0) */

	char thousands_sep;	/* locale specific thousands separator */

	const char *grouping;	/* locale specific numeric grouping rules */

#ifndef NO_FLOATING_POINT

	/*

	 * We can decompose the printed representation of floating

	 * point numbers into several parts, some of which may be empty:

	 *

	 * [+|-| ] [0x|0X] MMM . NNN [e|E|p|P] [+|-] ZZ

	 *    A       B     ---C---      D       E   F

	 *

	 * A:	'sign' holds this value if present; '\0' otherwise

	 * B:	ox[1] holds the 'x' or 'X'; '\0' if not hexadecimal

	 * C:	cp points to the string MMMNNN.  Leading and trailing

	 *	zeros are not in the string and must be added.

	 * D:	expchar holds this character; '\0' if no exponent, e.g. %f

	 * F:	at least two digits for decimal, at least one digit for hex

	 */

	char *decimal_point;	/* locale specific decimal point */

	int signflag;		/* true if float is negative */

	union {			/* floating point arguments %[aAeEfFgG] */

		double dbl;

		long double ldbl;

	} fparg;

	int expt;		/* integer value of exponent */

	char *dtoaend;		/* pointer to end of converted digits */

	int expsize;		/* character count for expstr */

	int lead;		/* sig figs before decimal or group sep */

	int ndig;		/* actual number of digits returned by dtoa */

	char expstr[MAXEXPDIG+2];	/* buffer for exponent string: e+ZZZ */

	char *dtoaresult;	/* buffer allocated by dtoa */

	int nseps;		/* number of group separators with ' */

	int nrepeats;		/* number of repeats of the last group */

#endif

	u_long	ulval;		/* integer arguments %[diouxX] */

	uintmax_t ujval;	/* %j, %ll, %q, %t, %z integers */

	int base;		/* base for [diouxX] conversion */

	int dprec;		/* a copy of prec if [diouxX], 0 otherwise */

	int realsz;		/* field size expanded by dprec, sign, etc */

	int size;		/* size of converted field or string */

	int prsize;             /* max size of printed field */

	const char *xdigs;     	/* digits for %[xX] conversion */

#define NIOV 8

	struct __suio uio;	/* output information: summary */

	struct __siov iov[NIOV];/* ... and individual io vectors */

	char buf[BUF];		/* buffer with space for digits of uintmax_t */

	char ox[2];		/* space for 0x; ox[1] is either x, X, or \0 */

	union arg *argtable;    /* args, built due to positional arg */

	union arg statargtable [STATIC_ARG_TBL_SIZE];

	int nextarg;            /* 1-based argument index */

	va_list orgap;          /* original argument pointer */

	char *convbuf;		/* wide to multibyte conversion result */