cryptutils

// util/strtos.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 a signed 8-bit integer */

s8 strtos8(const char *nptr, char **endptr, register int base) {

	register const char *s = nptr; register u8 i = 0, c; bool neg = false;

	for (; isspace(*s); ++s);

	if (*s == '+') { ++s; } else if (*s == '-') { ++s; neg = true; }

	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 > (neg ? -(S8_MIN + 1) : S8_MAX - c) / base) {

			errno = ERANGE; i = neg ? S8_MIN : S8_MAX; goto end;

		}

		i = i * base + c;

	}

end:;

	if (endptr) { *endptr = (char *)s; } return neg ? -i : i;

}

/* Convert a string to a signed 16-bit integer */

s16 strtos16(const char *nptr, char **endptr, register int base) {

	register const char *s = nptr; register u16 i = 0, c; bool neg = false;

	for (; isspace(*s); ++s);

	if (*s == '+') { ++s; } else if (*s == '-') { ++s; neg = true; }

	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 > (neg ? -(S16_MIN + 1) : S16_MAX - c) / base) {

			errno = ERANGE; i = neg ? S16_MIN : S16_MAX; goto end;

		}

		i = i * base + c;

	}

end:;

	if (endptr) { *endptr = (char *)s; } return neg ? -i : i;

}

/* Convert a string to a signed 32-bit integer */

s32 strtos32(const char *nptr, char **endptr, register int base) {

	register const char *s = nptr; register u32 i = 0, c; bool neg = false;

	for (; isspace(*s); ++s);

	if (*s == '+') { ++s; } else if (*s == '-') { ++s; neg = true; }

	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 > (neg ? -(S32_MIN + 1) : S32_MAX - c) / base) {

			errno = ERANGE; i = neg ? S32_MIN : S32_MAX; goto end;

		}

		i = i * base + c;

	}

end:;

	if (endptr) { *endptr = (char *)s; } return neg ? -i : i;

}

/* Convert a string to a signed 64-bit integer */

s64 strtos64(const char *nptr, char **endptr, register int base) {

	register const char *s = nptr; register u64 i = 0, c; bool neg = false;

	for (; isspace(*s); ++s);

	if (*s == '+') { ++s; } else if (*s == '-') { ++s; neg = true; }

	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 > (neg ? -(S64_MIN + 1) : S64_MAX - c) / base) {

			errno = ERANGE; i = neg ? S64_MIN : S64_MAX; goto end;

		}

		i = i * base + c;

	}

end:;

	if (endptr) { *endptr = (char *)s; } return neg ? -i : i;

}