fix fgetwc when decoding a character that crosses buffer boundary
[musl] / src / stdio / fgetwc.c
1 #include "stdio_impl.h"
2 #include "locale_impl.h"
3 #include <wchar.h>
4 #include <errno.h>
5
6 static wint_t __fgetwc_unlocked_internal(FILE *f)
7 {
8         mbstate_t st = { 0 };
9         wchar_t wc;
10         int c;
11         unsigned char b;
12         size_t l;
13
14         /* Convert character from buffer if possible */
15         if (f->rpos < f->rend) {
16                 l = mbrtowc(&wc, (void *)f->rpos, f->rend - f->rpos, &st);
17                 if (l+2 >= 2) {
18                         f->rpos += l + !l; /* l==0 means 1 byte, null */
19                         return wc;
20                 }
21                 if (l == -1) {
22                         f->rpos++;
23                         return WEOF;
24                 }
25                 f->rpos = f->rend;
26         } else l = -2;
27
28         /* Convert character byte-by-byte */
29         while (l == -2) {
30                 b = c = getc_unlocked(f);
31                 if (c < 0) {
32                         if (!mbsinit(&st)) errno = EILSEQ;
33                         return WEOF;
34                 }
35                 l = mbrtowc(&wc, (void *)&b, 1, &st);
36                 if (l == -1) return WEOF;
37         }
38
39         return wc;
40 }
41
42 wint_t __fgetwc_unlocked(FILE *f)
43 {
44         locale_t *ploc = &CURRENT_LOCALE, loc = *ploc;
45         if (f->mode <= 0) fwide(f, 1);
46         *ploc = f->locale;
47         wchar_t wc = __fgetwc_unlocked_internal(f);
48         *ploc = loc;
49         return wc;
50 }
51
52 wint_t fgetwc(FILE *f)
53 {
54         wint_t c;
55         FLOCK(f);
56         c = __fgetwc_unlocked(f);
57         FUNLOCK(f);
58         return c;
59 }
60
61 weak_alias(__fgetwc_unlocked, fgetwc_unlocked);
62 weak_alias(__fgetwc_unlocked, getwc_unlocked);