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