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