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