1 /* $OpenBSD: vswprintf.c,v 1.4 2012/12/05 23:20:01 deraadt Exp $ */
2 /* $NetBSD: vswprintf.c,v 1.1 2005/05/14 23:51:02 christos 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 <stdlib.h>
34 #include <string.h>
35 #include <wchar.h>
36 #include <stdarg.h>
37 #include "local.h"
38
39 int
vswprintf(wchar_t * __restrict s,size_t n,const wchar_t * __restrict fmt,__va_list ap)40 vswprintf(wchar_t * __restrict s, size_t n, const wchar_t * __restrict fmt,
41 __va_list ap)
42 {
43 mbstate_t mbs;
44 FILE f;
45 char *mbp;
46 int ret, sverrno;
47 size_t nwc;
48 struct __sfileext fext;
49
50 if (n == 0) {
51 errno = EINVAL;
52 return (-1);
53 }
54
55 _FILEEXT_SETUP(&f, &fext);
56 f._file = -1;
57 f._flags = __SWR | __SSTR | __SALC;
58 f._bf._base = f._p = (unsigned char *)malloc(128);
59 if (f._bf._base == NULL) {
60 errno = ENOMEM;
61 return (-1);
62 }
63 f._bf._size = f._w = 127; /* Leave room for the NUL */
64 ret = __vfwprintf(&f, fmt, ap);
65 if (ret < 0) {
66 sverrno = errno;
67 free(f._bf._base);
68 errno = sverrno;
69 return (-1);
70 }
71 if (ret == 0) {
72 s[0] = L'\0';
73 free(f._bf._base);
74 return (0);
75 }
76 *f._p = '\0';
77 mbp = (char *)f._bf._base;
78 /*
79 * XXX Undo the conversion from wide characters to multibyte that
80 * fputwc() did in __vfwprintf().
81 */
82 bzero(&mbs, sizeof(mbs));
83 nwc = mbsrtowcs(s, (const char **)&mbp, n, &mbs);
84 free(f._bf._base);
85 if (nwc == (size_t)-1) {
86 errno = EILSEQ;
87 return (-1);
88 }
89 if (nwc == n) {
90 s[n - 1] = L'\0';
91 errno = EOVERFLOW;
92 return (-1);
93 }
94
95 return (ret);
96 }
97