1#include "stdio_impl.h"
 2#include "intscan.h"
 3#include "shgetc.h"
 4#include <inttypes.h>
 5#include <limits.h>
 6#include <wctype.h>
 7#include <wchar.h>
 8
 9/* This read function heavily cheats. It knows:
10 *  (1) len will always be 1
11 *  (2) non-ascii characters don't matter */
12
13static size_t do_read(FILE *f, unsigned char *buf, size_t len)
14{
15	size_t i;
16	const wchar_t *wcs = f->cookie;
17
18	if (!wcs[0]) wcs=L"@";
19	for (i=0; i<f->buf_size && wcs[i]; i++)
20		f->buf[i] = wcs[i] < 128 ? wcs[i] : '@';
21	f->rpos = f->buf;
22	f->rend = f->buf + i;
23	f->cookie = (void *)(wcs+i);
24
25	if (i && len) {
26		*buf = *f->rpos++;
27		return 1;
28	}
29	return 0;
30}
31
32static unsigned long long wcstox(const wchar_t *s, wchar_t **p, int base, unsigned long long lim)
33{
34	wchar_t *t = (wchar_t *)s;
35	unsigned char buf[64];
36	FILE f = {0};
37	f.flags = 0;
38	f.rpos = f.rend = f.buf = buf + 4;
39	f.buf_size = sizeof buf - 4;
40#if defined(__wasilibc_unmodified_upstream) || defined(_REENTRANT)
41	f.lock = -1;
42#endif
43	f.read = do_read;
44	while (iswspace(*t)) t++;
45	f.cookie = (void *)t;
46	shlim(&f, 0);
47	unsigned long long y = __intscan(&f, base, 1, lim);
48	if (p) {
49		size_t cnt = shcnt(&f);
50		*p = cnt ? t + cnt : (wchar_t *)s;
51	}
52	return y;
53}
54
55unsigned long long wcstoull(const wchar_t *restrict s, wchar_t **restrict p, int base)
56{
57	return wcstox(s, p, base, ULLONG_MAX);
58}
59
60long long wcstoll(const wchar_t *restrict s, wchar_t **restrict p, int base)
61{
62	return wcstox(s, p, base, LLONG_MIN);
63}
64
65unsigned long wcstoul(const wchar_t *restrict s, wchar_t **restrict p, int base)
66{
67	return wcstox(s, p, base, ULONG_MAX);
68}
69
70long wcstol(const wchar_t *restrict s, wchar_t **restrict p, int base)
71{
72	return wcstox(s, p, base, 0UL+LONG_MIN);
73}
74
75intmax_t wcstoimax(const wchar_t *restrict s, wchar_t **restrict p, int base)
76{
77	return wcstoll(s, p, base);
78}
79
80uintmax_t wcstoumax(const wchar_t *restrict s, wchar_t **restrict p, int base)
81{
82	return wcstoull(s, p, base);
83}