#include "stdio_impl.h"
+#include "locale_impl.h"
#include <wchar.h>
#include <limits.h>
#include <ctype.h>
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;
f->flags &= ~F_EOF;
FUNLOCK(f);
+ *ploc = loc;
return c;
}