libc-top-half/musl/src/stdlib/wcstol.c

   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
  13 static 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
  32 static 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 = 0;
  39         f.buf = buf + 4;
  40         f.buf_size = sizeof buf - 4;
  41 #if defined(__wasilibc_unmodified_upstream) || defined(_REENTRANT)
  42         f.lock = -1;
  43 #endif
  44         f.read = do_read;
  45         while (iswspace(*t)) t++;
  46         f.cookie = (void *)t;
  47         shlim(&f, 0);
  48         unsigned long long y = __intscan(&f, base, 1, lim);
  49         if (p) {
  50                 size_t cnt = shcnt(&f);
  51                 *p = cnt ? t + cnt : (wchar_t *)s;
  52         }
  53         return y;
  54 }
  55
  56 unsigned long long wcstoull(const wchar_t *restrict s, wchar_t **restrict p, int base)
  57 {
  58         return wcstox(s, p, base, ULLONG_MAX);
  59 }
  60
  61 long long wcstoll(const wchar_t *restrict s, wchar_t **restrict p, int base)
  62 {
  63         return wcstox(s, p, base, LLONG_MIN);
  64 }
  65
  66 unsigned long wcstoul(const wchar_t *restrict s, wchar_t **restrict p, int base)
  67 {
  68         return wcstox(s, p, base, ULONG_MAX);
  69 }
  70
  71 long wcstol(const wchar_t *restrict s, wchar_t **restrict p, int base)
  72 {
  73         return wcstox(s, p, base, 0UL+LONG_MIN);
  74 }
  75
  76 intmax_t wcstoimax(const wchar_t *restrict s, wchar_t **restrict p, int base)
  77 {
  78         return wcstoll(s, p, base);
  79 }
  80
  81 uintmax_t wcstoumax(const wchar_t *restrict s, wchar_t **restrict p, int base)
  82 {
  83         return wcstoull(s, p, base);
  84 }