// util/strtou.c
// String conversion source file from libutil
// Copyright (C) 2021, Jakob Wakeling
// MIT Licence

#include "util.h"

#include <ctype.h>
#include <errno.h>

/* Convert a string to an unsigned 8-bit integer */
u8 strtou8(const char *nptr, char **endptr, register int base) {
	register const char *s = nptr; register u8 i = 0, c;
	
	for (; isspace(*s); ++s);
	
	if (*s == '+') { ++s; } else if (*s == '-') { errno = EINVAL; goto end; }
	
	if ((!base || base == 16) && s[0] == '0' && (s[1] == 'x' || s[1] == 'X')) {
		s += 2; base = 16;
	}
	else if (base == 0) { base = s[0] == '0' ? 8 : 10; }
	else if (base < 2 || base > 36) { errno = EINVAL; goto end; }
	
	for (;; ++s) {
		if (*s >= '0' && *s <= '9') { c = *s - '0'; }
		else if (*s >= 'A' && *s <= 'Z') { c = *s - ('A' - 10); }
		else if (*s >= 'a' && *s <= 'z') { c = *s - ('a' - 10); }
		else { break; }
		
		if (c >= base) { break; }
		if (i > (U8_MAX - c) / base) {
			errno = ERANGE; i = U8_MAX; goto end;
		}
		
		i = i * base + c;
	}
	
end:;
	if (endptr) { *endptr = (char *)s; } return i;
}

/* Convert a string to an unsigned 16-bit integer */
u16 strtou16(const char *nptr, char **endptr, register int base) {
	register const char *s = nptr; register u16 i = 0, c;
	
	for (; isspace(*s); ++s);
	
	if (*s == '+') { ++s; } else if (*s == '-') { errno = EINVAL; goto end; }
	
	if ((!base || base == 16) && s[0] == '0' && (s[1] == 'x' || s[1] == 'X')) {
		s += 2; base = 16;
	}
	else if (base == 0) { base = s[0] == '0' ? 8 : 10; }
	else if (base < 2 || base > 36) { errno = EINVAL; goto end; }
	
	for (;; ++s) {
		if (*s >= '0' && *s <= '9') { c = *s - '0'; }
		else if (*s >= 'A' && *s <= 'Z') { c = *s - ('A' - 10); }
		else if (*s >= 'a' && *s <= 'z') { c = *s - ('a' - 10); }
		else { break; }
		
		if (c >= base) { break; }
		if (i > (U16_MAX - c) / base) {
			errno = ERANGE; i = U16_MAX; goto end;
		}
		
		i = i * base + c;
	}
	
end:;
	if (endptr) { *endptr = (char *)s; } return i;
}

/* Convert a string to an unsigned 32-bit integer */
u32 strtou32(const char *nptr, char **endptr, register int base) {
	register const char *s = nptr; register u32 i = 0, c;
	
	for (; isspace(*s); ++s);
	
	if (*s == '+') { ++s; } else if (*s == '-') { errno = EINVAL; goto end; }
	
	if ((!base || base == 16) && s[0] == '0' && (s[1] == 'x' || s[1] == 'X')) {
		s += 2; base = 16;
	}
	else if (base == 0) { base = s[0] == '0' ? 8 : 10; }
	else if (base < 2 || base > 36) { errno = EINVAL; goto end; }
	
	for (;; ++s) {
		if (*s >= '0' && *s <= '9') { c = *s - '0'; }
		else if (*s >= 'A' && *s <= 'Z') { c = *s - ('A' - 10); }
		else if (*s >= 'a' && *s <= 'z') { c = *s - ('a' - 10); }
		else { break; }
		
		if (c >= base) { break; }
		if (i > (U32_MAX - c) / base) {
			errno = ERANGE; i = U32_MAX; goto end;
		}
		
		i = i * base + c;
	}
	
end:;
	if (endptr) { *endptr = (char *)s; } return i;
}

/* Convert a string to an unsigned 64-bit integer */
u64 strtou64(const char *nptr, char **endptr, register int base) {
	register const char *s = nptr; register u64 i = 0, c;
	
	for (; isspace(*s); ++s);
	
	if (*s == '+') { ++s; } else if (*s == '-') { errno = EINVAL; goto end; }
	
	if ((!base || base == 16) && s[0] == '0' && (s[1] == 'x' || s[1] == 'X')) {
		s += 2; base = 16;
	}
	else if (base == 0) { base = s[0] == '0' ? 8 : 10; }
	else if (base < 2 || base > 36) { errno = EINVAL; goto end; }
	
	for (;; ++s) {
		if (*s >= '0' && *s <= '9') { c = *s - '0'; }
		else if (*s >= 'A' && *s <= 'Z') { c = *s - ('A' - 10); }
		else if (*s >= 'a' && *s <= 'z') { c = *s - ('a' - 10); }
		else { break; }
		
		if (c >= base) { break; }
		if (i > (U64_MAX - c) / base) {
			errno = ERANGE; i = U64_MAX; goto end;
		}
		
		i = i * base + c;
	}
	
end:;
	if (endptr) { *endptr = (char *)s; } return i;
}