Loading...
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 | --- vswprintf.c.orig 2004-11-25 11:38:36.000000000 -0800 +++ vswprintf.c 2005-02-24 15:20:20.000000000 -0800 @@ -33,6 +33,8 @@ #endif __FBSDID("$FreeBSD: src/lib/libc/stdio/vswprintf.c,v 1.5 2004/04/07 09:55:05 tjr Exp $"); +#include "xlocale_private.h" + #include <errno.h> #include <stdio.h> #include <stdlib.h> @@ -49,7 +51,66 @@ struct __sFILEX ext; char *mbp; int ret, sverrno; + size_t conv; + locale_t loc = __current_locale(); + + if (n == 0) { + errno = EINVAL; + return (-1); + } + + f._file = -1; + f._flags = __SWR | __SSTR | __SALC; + f._bf._base = f._p = (unsigned char *)malloc(128); + if (f._bf._base == NULL) { + errno = ENOMEM; + return (-1); + } + f._bf._size = f._w = 127; /* Leave room for the NUL */ + f._extra = &ext; + INITEXTRA(&f); + ret = __vfwprintf(&f, loc, fmt, ap); + if (ret < 0) { + sverrno = errno; + free(f._bf._base); + errno = sverrno; + return (-1); + } + *f._p = '\0'; + mbp = (char *)f._bf._base; + /* + * XXX Undo the conversion from wide characters to multibyte that + * fputwc() did in __vfwprintf(). + */ + mbs = initial; + if ((conv = mbsrtowcs_l(s, (const char **)&mbp, n, &mbs, loc)) == (size_t)-1) { + free(f._bf._base); + errno = EILSEQ; + return (-1); + } + free(f._bf._base); + if (conv >= n) { + s[n - 1] = L'\0'; + errno = EOVERFLOW; + return (-1); + } + + return (ret); +} + +int +vswprintf_l(wchar_t * __restrict s, size_t n, locale_t loc, + const wchar_t * __restrict fmt, __va_list ap) +{ + static const mbstate_t initial; + mbstate_t mbs; + FILE f; + struct __sFILEX ext; + char *mbp; + int ret, sverrno; + size_t conv; + NORMALIZE_LOCALE(loc); if (n == 0) { errno = EINVAL; return (-1); @@ -65,7 +126,7 @@ f._bf._size = f._w = 127; /* Leave room for the NUL */ f._extra = &ext; INITEXTRA(&f); - ret = __vfwprintf(&f, fmt, ap); + ret = __vfwprintf(&f, loc, fmt, ap); if (ret < 0) { sverrno = errno; free(f._bf._base); @@ -73,19 +134,19 @@ return (-1); } *f._p = '\0'; - mbp = f._bf._base; + mbp = (char *)f._bf._base; /* * XXX Undo the conversion from wide characters to multibyte that * fputwc() did in __vfwprintf(). */ mbs = initial; - if (mbsrtowcs(s, (const char **)&mbp, n, &mbs) == (size_t)-1) { + if ((conv = mbsrtowcs_l(s, (const char **)&mbp, n, &mbs, loc)) == (size_t)-1) { free(f._bf._base); errno = EILSEQ; return (-1); } free(f._bf._base); - if (s[n - 1] != L'\0') { + if (conv >= n) { s[n - 1] = L'\0'; errno = EOVERFLOW; return (-1); |