| 1 | /* Copyright (C) 2005-2019 Free Software Foundation, Inc. | 
|---|
| 2 | This file is part of the GNU C Library. | 
|---|
| 3 | Contributed by Ulrich Drepper <drepper@gnu.org>. | 
|---|
| 4 |  | 
|---|
| 5 | The GNU C Library is free software; you can redistribute it and/or | 
|---|
| 6 | modify it under the terms of the GNU Lesser General Public | 
|---|
| 7 | License as published by the Free Software Foundation; either | 
|---|
| 8 | version 2.1 of the License, or (at your option) any later version. | 
|---|
| 9 |  | 
|---|
| 10 | The GNU C Library is distributed in the hope that it will be useful, | 
|---|
| 11 | but WITHOUT ANY WARRANTY; without even the implied warranty of | 
|---|
| 12 | MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU | 
|---|
| 13 | Lesser General Public License for more details. | 
|---|
| 14 |  | 
|---|
| 15 | You should have received a copy of the GNU Lesser General Public | 
|---|
| 16 | License along with the GNU C Library; if not, see | 
|---|
| 17 | <http://www.gnu.org/licenses/>.  */ | 
|---|
| 18 |  | 
|---|
| 19 | #include <stdarg.h> | 
|---|
| 20 | #include <stdio.h> | 
|---|
| 21 | #include <stdlib.h> | 
|---|
| 22 | #include <string.h> | 
|---|
| 23 | #include <wchar.h> | 
|---|
| 24 | #include <libioP.h> | 
|---|
| 25 |  | 
|---|
| 26 | static int | 
|---|
| 27 | locked_vfxprintf (FILE *fp, const char *fmt, va_list ap) | 
|---|
| 28 | { | 
|---|
| 29 | if (_IO_fwide (fp, 0) <= 0) | 
|---|
| 30 | return __vfprintf_internal (fp, fmt, ap, 0); | 
|---|
| 31 |  | 
|---|
| 32 | /* We must convert the narrow format string to a wide one. | 
|---|
| 33 | Each byte can produce at most one wide character.  */ | 
|---|
| 34 | wchar_t *wfmt; | 
|---|
| 35 | mbstate_t mbstate; | 
|---|
| 36 | int res; | 
|---|
| 37 | int used_malloc = 0; | 
|---|
| 38 | size_t len = strlen (fmt) + 1; | 
|---|
| 39 |  | 
|---|
| 40 | if (__glibc_unlikely (len > SIZE_MAX / sizeof (wchar_t))) | 
|---|
| 41 | { | 
|---|
| 42 | __set_errno (EOVERFLOW); | 
|---|
| 43 | return -1; | 
|---|
| 44 | } | 
|---|
| 45 | if (__libc_use_alloca (len * sizeof (wchar_t))) | 
|---|
| 46 | wfmt = alloca (len * sizeof (wchar_t)); | 
|---|
| 47 | else if ((wfmt = malloc (len * sizeof (wchar_t))) == NULL) | 
|---|
| 48 | return -1; | 
|---|
| 49 | else | 
|---|
| 50 | used_malloc = 1; | 
|---|
| 51 |  | 
|---|
| 52 | memset (&mbstate, 0, sizeof mbstate); | 
|---|
| 53 | res = __mbsrtowcs (wfmt, &fmt, len, &mbstate); | 
|---|
| 54 |  | 
|---|
| 55 | if (res != -1) | 
|---|
| 56 | res = __vfwprintf_internal (fp, wfmt, ap, 0); | 
|---|
| 57 |  | 
|---|
| 58 | if (used_malloc) | 
|---|
| 59 | free (wfmt); | 
|---|
| 60 |  | 
|---|
| 61 | return res; | 
|---|
| 62 | } | 
|---|
| 63 |  | 
|---|
| 64 | int | 
|---|
| 65 | __vfxprintf (FILE *fp, const char *fmt, va_list ap) | 
|---|
| 66 | { | 
|---|
| 67 | if (fp == NULL) | 
|---|
| 68 | fp = stderr; | 
|---|
| 69 | _IO_flockfile (fp); | 
|---|
| 70 | int res = locked_vfxprintf (fp, fmt, ap); | 
|---|
| 71 | _IO_funlockfile (fp); | 
|---|
| 72 | return res; | 
|---|
| 73 | } | 
|---|
| 74 |  | 
|---|
| 75 | int | 
|---|
| 76 | __fxprintf (FILE *fp, const char *fmt, ...) | 
|---|
| 77 | { | 
|---|
| 78 | va_list ap; | 
|---|
| 79 | va_start (ap, fmt); | 
|---|
| 80 | int res = __vfxprintf (fp, fmt, ap); | 
|---|
| 81 | va_end (ap); | 
|---|
| 82 | return res; | 
|---|
| 83 | } | 
|---|
| 84 |  | 
|---|
| 85 | int | 
|---|
| 86 | __fxprintf_nocancel (FILE *fp, const char *fmt, ...) | 
|---|
| 87 | { | 
|---|
| 88 | if (fp == NULL) | 
|---|
| 89 | fp = stderr; | 
|---|
| 90 |  | 
|---|
| 91 | va_list ap; | 
|---|
| 92 | va_start (ap, fmt); | 
|---|
| 93 | _IO_flockfile (fp); | 
|---|
| 94 | int save_flags2 = fp->_flags2; | 
|---|
| 95 | fp->_flags2 |= _IO_FLAGS2_NOTCANCEL; | 
|---|
| 96 |  | 
|---|
| 97 | int res = locked_vfxprintf (fp, fmt, ap); | 
|---|
| 98 |  | 
|---|
| 99 | fp->_flags2 = save_flags2; | 
|---|
| 100 | _IO_funlockfile (fp); | 
|---|
| 101 | va_end (ap); | 
|---|
| 102 | return res; | 
|---|
| 103 | } | 
|---|
| 104 |  | 
|---|