Add support for parsing long long unsigned integers if using hex

1 files changed, 53 insertions, 2 deletions

diff --git a/src/badlib.h b/src/badlib.h
index d334fa7..6432562 100644
--- a/src/badlib.h
+++ b/src/badlib.h
@@ -470,13 +470,13 @@ str_from_int(sz num, Arena *a) {
 }
 
 Str
-str_from_hex(sz num, sz zeroes, Arena *a) {
+str_from_hex(u64 num, sz zeroes, Arena *a) {
     char char_map[] = "0123456789abcdef";
     u8 tmp[64];
     zeroes = MIN((sz)sizeof(ptrsize) * 2, zeroes);
     u8 *end = tmp + sizeof(tmp);
     u8 *beg = end;
-    sz t = num > 0 ? num : -num;
+    u64 t = num;
     do {
         *--beg = char_map[t % 16];
     } while (t /= 16);
@@ -634,6 +634,57 @@ str_to_int(Str s) {
     return num;
 }
 
+u64
+str_to_uint(Str s) {
+    u64 num = 0;
+    if (str_has_prefix(s, cstr("0b"))) {
+        // Binary number.
+        s = str_remove_prefix(s, cstr("0b"));
+        while (s.size) {
+            char c = str_next(&s);
+            if (c == '_') {
+                continue;
+            }
+            assert(c == '0' || c == '1');
+            num = num * 2 + (c - '0');
+        }
+    } else if (str_has_prefix(s, cstr("0x"))) {
+        // Hex number.
+        s = str_remove_prefix(s, cstr("0x"));
+        while (s.size) {
+            char c = str_next(&s);
+            if (c == '_') {
+                continue;
+            }
+            assert((c >= '0' && c <= '9') || (c >= 'a' && c <= 'f') ||
+                   (c >= 'A' && c <= 'F'));
+            if (c >= '0' && c <= '9') {
+                num = num * 16 + (c - '0');
+            } else if (c >= 'a' && c <= 'f') {
+                num = num * 16 + (c - 'a' + 10);
+            } else if (c >= 'A' && c <= 'F') {
+                num = num * 16 + (c - 'A' + 10);
+            }
+        }
+    } else {
+        // Decimal number.
+        char c = str_peek(s);
+        assert(c != '-');
+        if (c == '+') {
+            str_next(&s);
+        }
+        while (s.size) {
+            char c = str_next(&s);
+            if (c == '_') {
+                continue;
+            }
+            assert(c >= '0' && c <= '9');
+            num = num * 10 + (c - '0');
+        }
+    }
+    return num;
+}
+
 f64
 str_to_float(Str s) {
     char c = str_peek(s);