vswprintf.c 2.8 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697
  1. /* $OpenBSD: vswprintf.c,v 1.3 2011/04/28 17:38:46 stsp Exp $ */
  2. /* $NetBSD: vswprintf.c,v 1.1 2005/05/14 23:51:02 christos Exp $ */
  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. #include <sys/cdefs.h>
  30. #include <errno.h>
  31. #include <stdio.h>
  32. #include <stdlib.h>
  33. #include <string.h>
  34. #include <wchar.h>
  35. #include <stdarg.h>
  36. #include "local.h"
  37. int
  38. vswprintf(wchar_t * __restrict s, size_t n, const wchar_t * __restrict fmt,
  39. __va_list ap)
  40. {
  41. mbstate_t mbs;
  42. FILE f;
  43. char *mbp;
  44. int ret, sverrno;
  45. size_t nwc;
  46. struct __sfileext fext;
  47. if (n == 0) {
  48. errno = EINVAL;
  49. return (-1);
  50. }
  51. _FILEEXT_SETUP(&f, &fext);
  52. f._file = -1;
  53. f._flags = __SWR | __SSTR | __SALC;
  54. f._bf._base = f._p = (unsigned char *)malloc(128);
  55. if (f._bf._base == NULL) {
  56. errno = ENOMEM;
  57. return (-1);
  58. }
  59. f._bf._size = f._w = 127; /* Leave room for the NUL */
  60. ret = __vfwprintf(&f, fmt, ap);
  61. if (ret < 0) {
  62. sverrno = errno;
  63. free(f._bf._base);
  64. errno = sverrno;
  65. return (-1);
  66. }
  67. if (ret == 0) {
  68. s[0] = L'\0';
  69. free(f._bf._base);
  70. return (0);
  71. }
  72. *f._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. memset(&mbs, 0x0, sizeof(mbs));
  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. return (ret);
  91. }