all printf variants: fix argument type handling for %c and %lc
[musl] / src / stdio / ungetwc.c
index 913f716..9edf366 100644 (file)
@@ -1,4 +1,5 @@
 #include "stdio_impl.h"
+#include "locale_impl.h"
 #include <wchar.h>
 #include <limits.h>
 #include <ctype.h>
@@ -7,22 +8,20 @@
 wint_t ungetwc(wint_t c, FILE *f)
 {
        unsigned char mbc[MB_LEN_MAX];
-       int l=1;
-
-       if (c == WEOF) return c;
-
-       /* Try conversion early so we can fail without locking if invalid */
-       if (!isascii(c) && (l = wctomb((void *)mbc, c)) < 0)
-               return WEOF;
+       int l;
+       locale_t *ploc = &CURRENT_LOCALE, loc = *ploc;
 
        FLOCK(f);
 
-       f->mode |= f->mode+1;
+       if (f->mode <= 0) fwide(f, 1);
+       *ploc = f->locale;
 
        if (!f->rpos) __toread(f);
-       if (!f->rpos || f->rpos < f->buf - UNGET + l) {
+       if (!f->rpos || c == WEOF || (l = wcrtomb((void *)mbc, c, 0)) < 0 ||
+           f->rpos < f->buf - UNGET + l) {
                FUNLOCK(f);
-               return EOF;
+               *ploc = loc;
+               return WEOF;
        }
 
        if (isascii(c)) *--f->rpos = c;
@@ -31,5 +30,6 @@ wint_t ungetwc(wint_t c, FILE *f)
        f->flags &= ~F_EOF;
 
        FUNLOCK(f);
+       *ploc = loc;
        return c;
 }