[apple/libc.git] / stdio / FreeBSD / vswprintf.c

/*	$OpenBSD: vasprintf.c,v 1.4 1998/06/21 22:13:47 millert Exp $	*/

/*
 * Copyright (c) 1997 Todd C. Miller <Todd.Miller@courtesan.com>
 * All rights reserved.
 *
 * 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.
 * 3. The name of the author may not be used to endorse or promote products
 *    derived from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED ``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 AUTHOR 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.
 */

#include <sys/cdefs.h>
#if 0
__FBSDID("FreeBSD: src/lib/libc/stdio/vasprintf.c,v 1.16 2002/08/21 16:19:57 mike Exp ");
#endif
__FBSDID("$FreeBSD: src/lib/libc/stdio/vswprintf.c,v 1.7 2008/04/17 22:17:54 jhb Exp $");

#include "xlocale_private.h"

#include <errno.h>
#include <stdio.h>
#include <stdlib.h>
#include <wchar.h>
#include "local.h"

int
vswprintf_l(wchar_t * __restrict s, size_t n, locale_t loc,
    const wchar_t * __restrict fmt, __va_list ap)
{
	static const mbstate_t initial;
	mbstate_t mbs;
	FILE f;
	char *mbp;
	int ret, sverrno;
	size_t nwc;
	struct __sFILEX ext;
	f._extra = &ext;
	INITEXTRA(&f);

	NORMALIZE_LOCALE(loc);
	if (n == 0) {
		errno = EINVAL;
		return (-1);
	}

	f._file = -1;
	f._flags = __SWR | __SSTR | __SALC;
	f._bf._base = f._p = (unsigned char *)malloc(128);
	if (f._bf._base == NULL) {
		errno = ENOMEM;
		return (-1);
	}
	f._bf._size = f._w = 127;		/* Leave room for the NUL */
	f._orientation = 0;
	memset(&f._mbstate, 0, sizeof(mbstate_t));
	ret = __vfwprintf(&f, loc, fmt, ap);
	if (ret < 0) {
		sverrno = errno;
		free(f._bf._base);
		errno = sverrno;
		return (-1);
	}
	*f._p = '\0';
	mbp = f._bf._base;
	/*
	 * XXX Undo the conversion from wide characters to multibyte that
	 * fputwc() did in __vfwprintf().
	 */
	mbs = initial;
	nwc = mbsrtowcs_l(s, (const char **)&mbp, n, &mbs, loc);
	free(f._bf._base);
	if (nwc == (size_t)-1) {
		errno = EILSEQ;
		return (-1);
	}
	if (nwc == n) {
		s[n - 1] = L'\0';
		errno = EOVERFLOW;
		return (-1);
	}

	return (ret);
}

int
vswprintf(wchar_t * __restrict s, size_t n,
    const wchar_t * __restrict fmt, __va_list ap)
{
	return vswprintf_l(s, n, __current_locale(), fmt, ap);
}
Commit	Line	Data
5b2abdfb A	1	/* $OpenBSD: vasprintf.c,v 1.4 1998/06/21 22:13:47 millert Exp $ */
	2
	3	/*
	4	* Copyright (c) 1997 Todd C. Miller <Todd.Miller@courtesan.com>
	5	* All rights reserved.
	6	*
	7	* Redistribution and use in source and binary forms, with or without
	8	* modification, are permitted provided that the following conditions
	9	* are met:
	10	* 1. Redistributions of source code must retain the above copyright
	11	* notice, this list of conditions and the following disclaimer.
	12	* 2. Redistributions in binary form must reproduce the above copyright
	13	* notice, this list of conditions and the following disclaimer in the
	14	* documentation and/or other materials provided with the distribution.
	15	* 3. The name of the author may not be used to endorse or promote products
	16	* derived from this software without specific prior written permission.
	17	*
	18	* THIS SOFTWARE IS PROVIDED ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES,
	19	* INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY
	20	* AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL
	21	* THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
	22	* EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
	23	* PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
	24	* OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
	25	* WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
	26	* OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
	27	* ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
	28	*/
	29
9385eb3d A	30	#include <sys/cdefs.h>
	31	#if 0
	32	__FBSDID("FreeBSD: src/lib/libc/stdio/vasprintf.c,v 1.16 2002/08/21 16:19:57 mike Exp ");
	33	#endif
1f2f436a	34	__FBSDID("$FreeBSD: src/lib/libc/stdio/vswprintf.c,v 1.7 2008/04/17 22:17:54 jhb Exp $");
5b2abdfb	35
ad3c9f2a A	36	#include "xlocale_private.h"
ad3c9f2a A	37
9385eb3d	38	#include <errno.h>
5b2abdfb A	39	#include <stdio.h>
5b2abdfb A	40	#include <stdlib.h>
9385eb3d A	41	#include <wchar.h>
9385eb3d A	42	#include "local.h"
5b2abdfb A	43
5b2abdfb A	44	int
ad3c9f2a A	45	vswprintf_l(wchar_t * __restrict s, size_t n, locale_t loc,
ad3c9f2a A	46	const wchar_t * __restrict fmt, __va_list ap)
5b2abdfb	47	{
3d9156a7 A	48	static const mbstate_t initial;
3d9156a7 A	49	mbstate_t mbs;
5b2abdfb	50	FILE f;
9385eb3d A	51	char *mbp;
9385eb3d A	52	int ret, sverrno;
1f2f436a	53	size_t nwc;
ad3c9f2a A	54	struct __sFILEX ext;
	55	f._extra = &ext;
	56	INITEXTRA(&f);
9385eb3d	57
ad3c9f2a	58	NORMALIZE_LOCALE(loc);
9385eb3d A	59	if (n == 0) {
	60	errno = EINVAL;
	61	return (-1);
	62	}
5b2abdfb A	63
	64	f._file = -1;
	65	f._flags = __SWR \| __SSTR \| __SALC;
	66	f._bf._base = f._p = (unsigned char *)malloc(128);
	67	if (f._bf._base == NULL) {
5b2abdfb A	68	errno = ENOMEM;
	69	return (-1);
	70	}
9385eb3d	71	f._bf._size = f._w = 127; /* Leave room for the NUL */
1f2f436a A	72	f._orientation = 0;
1f2f436a A	73	memset(&f._mbstate, 0, sizeof(mbstate_t));
ad3c9f2a	74	ret = __vfwprintf(&f, loc, fmt, ap);
9385eb3d A	75	if (ret < 0) {
	76	sverrno = errno;
	77	free(f._bf._base);
	78	errno = sverrno;
	79	return (-1);
	80	}
5b2abdfb	81	*f._p = '\0';
9385eb3d	82	mbp = f._bf._base;
9385eb3d A	83	/*
	84	* XXX Undo the conversion from wide characters to multibyte that
	85	* fputwc() did in __vfwprintf().
	86	*/
3d9156a7	87	mbs = initial;
ad3c9f2a	88	nwc = mbsrtowcs_l(s, (const char **)&mbp, n, &mbs, loc);
1f2f436a A	89	free(f._bf._base);
1f2f436a A	90	if (nwc == (size_t)-1) {
9385eb3d A	91	errno = EILSEQ;
9385eb3d A	92	return (-1);
5b2abdfb	93	}
1f2f436a	94	if (nwc == n) {
9385eb3d A	95	s[n - 1] = L'\0';
	96	errno = EOVERFLOW;
	97	return (-1);
	98	}
	99
5b2abdfb A	100	return (ret);
5b2abdfb A	101	}
ad3c9f2a A	102
	103	int
	104	vswprintf(wchar_t * __restrict s, size_t n,
	105	const wchar_t * __restrict fmt, __va_list ap)
	106	{
	107	return vswprintf_l(s, n, __current_locale(), fmt, ap);
	108	}