xref: /dragonfly/lib/libc/stdio/vswprintf.c (revision c03f08f3)
1 /*	$NetBSD: vswprintf.c,v 1.1 2005/05/14 23:51:02 christos Exp $	*/
2 /*	$DragonFly: src/lib/libc/stdio/vswprintf.c,v 1.2 2006/03/02 18:05:30 joerg Exp $ */
3 
4 /*
5  * Copyright (c) 1997 Todd C. Miller <Todd.Miller@courtesan.com>
6  * All rights reserved.
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  * 3. The name of the author may not be used to endorse or promote products
17  *    derived from this software without specific prior written permission.
18  *
19  * THIS SOFTWARE IS PROVIDED ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES,
20  * INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY
21  * AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL
22  * THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
23  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
24  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
25  * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
26  * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
27  * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
28  * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
29  */
30 
31 #include <errno.h>
32 #include <stdio.h>
33 #include <stdarg.h>
34 #include <stdlib.h>
35 #include <string.h>
36 #include <wchar.h>
37 
38 #include "local.h"
39 #include "priv_stdio.h"
40 
41 int
42 vswprintf(wchar_t * __restrict s, size_t n, const wchar_t * __restrict fmt,
43     va_list ap)
44 {
45 	static const mbstate_t initial;
46 	mbstate_t mbs;
47 	FILE f;
48 	char *mbp;
49 	int ret, sverrno;
50 	size_t nwc;
51 
52 	if (n == 0) {
53 		errno = EINVAL;
54 		return (-1);
55 	}
56 
57 	f.pub._fileno = -1;
58 	f.pub._flags = __SWR | __SSTR | __SALC;
59 	f._bf._base = f.pub._p = (unsigned char *)malloc(128);
60 	if (f._bf._base == NULL) {
61 		errno = ENOMEM;
62 		return (-1);
63 	}
64 	f._bf._size = f.pub._w = 127;		/* Leave room for the NUL */
65 	f._up = NULL;
66 	f.fl_mutex = PTHREAD_MUTEX_INITIALIZER;
67 	f.fl_owner = NULL;
68 	f.fl_count = 0;
69 	memset(WCIO_GET(&f), 0, sizeof(struct wchar_io_data));
70 	ret = __vfwprintf_unlocked(&f, fmt, ap);
71 	if (ret < 0) {
72 		sverrno = errno;
73 		free(f._bf._base);
74 		errno = sverrno;
75 		return (-1);
76 	}
77 	*f.pub._p = '\0';
78 	mbp = (char *)f._bf._base;
79 	/*
80 	 * XXX Undo the conversion from wide characters to multibyte that
81 	 * fputwc() did in __vfwprintf().
82 	 */
83 	mbs = initial;
84 	nwc = mbsrtowcs(s, (const char **)&mbp, n, &mbs);
85 	free(f._bf._base);
86 	if (nwc == (size_t)-1) {
87 		errno = EILSEQ;
88 		return (-1);
89 	}
90 	if (nwc == n) {
91 		s[n - 1] = L'\0';
92 		errno = EOVERFLOW;
93 		return (-1);
94 	}
95 
96 	return (ret);
97 }
98