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 * 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 #if 0 38 __FBSDID("FreeBSD: src/lib/libc/stdio/vasprintf.c,v 1.16 2002/08/21 16:19:57 mike Exp "); 39 #endif 40 #include <errno.h> 41 #include <limits.h> 42 #include <stdio.h> 43 #include <stdlib.h> 44 #include <wchar.h> 45 #include "local.h" 46 #include "xlocale_private.h" 47 48 int 49 vswprintf_l(wchar_t * __restrict s, size_t n, locale_t locale, 50 const wchar_t * __restrict fmt, __va_list ap) 51 { 52 static const mbstate_t initial; 53 mbstate_t mbs; 54 FILE f = FAKE_FILE; 55 char *mbp; 56 int ret, sverrno; 57 size_t nwc; 58 FIX_LOCALE(locale); 59 60 if (n == 0) { 61 errno = EINVAL; 62 return (-1); 63 } 64 if (n - 1 > INT_MAX) { 65 errno = EOVERFLOW; 66 *s = L'\0'; 67 return (-1); 68 } 69 70 f._flags = __SWR | __SSTR | __SALC; 71 f._bf._base = f._p = (unsigned char *)malloc(128); 72 if (f._bf._base == NULL) { 73 errno = ENOMEM; 74 *s = L'\0'; 75 return (-1); 76 } 77 f._bf._size = f._w = 127; /* Leave room for the NUL */ 78 ret = __vfwprintf(&f, locale, fmt, ap); 79 if (ret < 0) { 80 sverrno = errno; 81 free(f._bf._base); 82 errno = sverrno; 83 *s = L'\0'; 84 return (-1); 85 } 86 *f._p = '\0'; 87 mbp = f._bf._base; 88 /* 89 * XXX Undo the conversion from wide characters to multibyte that 90 * fputwc() did in __vfwprintf(). 91 */ 92 mbs = initial; 93 nwc = mbsrtowcs_l(s, (const char **)&mbp, n, &mbs, locale); 94 free(f._bf._base); 95 if (nwc == (size_t)-1) { 96 errno = EILSEQ; 97 *s = L'\0'; 98 return (-1); 99 } 100 if (nwc == n) { 101 s[n - 1] = L'\0'; 102 errno = EOVERFLOW; 103 return (-1); 104 } 105 106 return (ret); 107 } 108 int 109 vswprintf(wchar_t * __restrict s, size_t n, const wchar_t * __restrict fmt, 110 __va_list ap) 111 { 112 return vswprintf_l(s, n, __get_locale(), fmt, ap); 113 } 114