Home | History | Annotate | Line # | Download | only in stdio
      1 /*	$NetBSD: vswprintf.c,v 1.6 2013/05/19 21:45:00 christos Exp $	*/
      2 
      3 /*
      4  * Copyright (c) 1997 Todd C. Miller <Todd.Miller (at) 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 
     30 #include <sys/cdefs.h>
     31 #if defined(LIBC_SCCS) && !defined(lint)
     32 #if 0
     33 __FBSDID("$FreeBSD: src/lib/libc/stdio/vswprintf.c,v 1.6 2005/02/21 19:41:44 fjoe Exp $");
     34 #else
     35 __RCSID("$NetBSD: vswprintf.c,v 1.6 2013/05/19 21:45:00 christos Exp $");
     36 #endif
     37 #endif /* LIBC_SCCS and not lint */
     38 
     39 #include "namespace.h"
     40 #include <errno.h>
     41 #include <locale.h>
     42 #include <stdio.h>
     43 #include <stdlib.h>
     44 #include <wchar.h>
     45 #include <stdarg.h>
     46 
     47 #include "reentrant.h"
     48 #include "setlocale_local.h"
     49 #include "local.h"
     50 
     51 __weak_alias(vswprintf_l, _vswprintf_l)
     52 
     53 int
     54 vswprintf_l(wchar_t * __restrict s, size_t n, locale_t loc,
     55     const wchar_t * __restrict fmt, va_list ap)
     56 {
     57 	static const mbstate_t initial;
     58 	mbstate_t mbs;
     59 	FILE f;
     60 	char *mbp;
     61 	int ret, sverrno;
     62 	size_t nwc;
     63 	struct __sfileext fext;
     64 
     65 	if (n == 0) {
     66 		errno = EINVAL;
     67 		return -1;
     68 	}
     69 
     70 	_FILEEXT_SETUP(&f, &fext);
     71 	f._file = -1;
     72 	f._flags = __SWR | __SSTR | __SALC;
     73 	f._bf._base = f._p = malloc(128);
     74 	if (f._bf._base == NULL) {
     75 		errno = ENOMEM;
     76 		return -1;
     77 	}
     78 	f._bf._size = f._w = 127;		/* Leave room for the NUL */
     79 	ret = __vfwprintf_unlocked_l(&f, loc, fmt, ap);
     80 	if (ret < 0) {
     81 		sverrno = errno;
     82 		free(f._bf._base);
     83 		errno = sverrno;
     84 		return -1;
     85 	}
     86 	*f._p = '\0';
     87 	mbp = (char *)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, (void *)&mbp, n, &mbs, loc);
     94 	free(f._bf._base);
     95 	if (nwc == (size_t)-1) {
     96 		errno = EILSEQ;
     97 		return -1;
     98 	}
     99 	if (nwc == n) {
    100 		s[n - 1] = L'\0';
    101 		errno = EOVERFLOW;
    102 		return -1;
    103 	}
    104 
    105 	return ret;
    106 }
    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, _current_locale(), fmt, ap);
    113 }
    114