/*
 * This file and its contents are supplied under the terms of the
 * Common Development and Distribution License ("CDDL"), version 1.0.
 * You may only use this file in accordance with the terms of version
 * 1.0 of the CDDL.
 *
 * A full copy of the text of the CDDL should have accompanied this
 * source.  A copy of the CDDL is also available via the Internet at
 * http://www.illumos.org/license/CDDL.
 */

/*
 * Copyright 2023 Oxide Computer Company
 */

/*
 * Test the implementation of strtonum() and strtonumx()
 */

#include <limits.h>
#include <stdbool.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

bool fail = false;

static void
runtestx(const char *nstr, long long minval, long long maxval,
    int base, long long expval, bool expect_pass)
{
	const char *errstr;
	char key[0x100];
	long long val;

	if (base == -1) {
		val = strtonum(nstr, minval, maxval, &errstr);
		(void) snprintf(key, sizeof (key), "strtonum(%s)", nstr);
	} else {
		val = strtonumx(nstr, minval, maxval, &errstr, base);
		(void) snprintf(key, sizeof (key), "strtonumx(%s, %d)",
		    nstr, base);
	}

	if (expect_pass) {
		if (errstr != NULL) {
			fprintf(stderr, "FAIL %s was rejected: "
			    "%s [%lld,%lld]\n",
			    key, errstr, minval, maxval);
			fail = true;
		} else {
			fprintf(stdout, "PASS %s was accepted: "
			    "[%lld,%lld] = %lld\n",
			    key, minval, maxval, val);
			if (val != expval) {
				fprintf(stderr, "FAIL %s returned wrong value: "
				    "[%lld,%lld] = %lld vs. %lld\n", key,
				    minval, maxval, val, expval);
				fail = true;
			}
		}
	} else {
		if (errstr == NULL) {
			fprintf(stderr, "FAIL %s was accepted: "
			    "[%lld,%lld] = %lld\n",
			    key, minval, maxval, val);
			fail = true;
		} else {
			fprintf(stdout, "PASS %s was rejected: "
			    "%s [%lld,%lld]\n",
			    key, errstr, minval, maxval);
		}
	}
}

static void
runtest(const char *nstr, long long minval, long long maxval, long long expval,
    bool expect_pass)
{
	runtestx(nstr, minval, maxval, -1, expval, expect_pass);
	runtestx(nstr, minval, maxval, 10, expval, expect_pass);
	runtestx(nstr, minval, maxval,  0, expval, expect_pass);
}

int
main(void)
{
	runtest("1", 0, 100, 1, true);
	runtest("0", -3, 7, 0, true);
	runtest("0", 2, 10, 0, false);
	runtest("0", 2, LLONG_MAX, 0, false);
	runtest("-2", 0, LLONG_MAX, 0, false);
	runtest("0", -5, LLONG_MAX, 0, true);
	runtest("-3", -3, LLONG_MAX, -3, true);
	runtest("-2", 10, -1, 0, false);
	runtest("-2", -10, -1, -2, true);
	runtest("-20", -10, -1, 0, false);
	runtest("20", -10, -1, 0, false);

	runtest("-9223372036854775808", LLONG_MIN, LLONG_MAX, LLONG_MIN, true);
	runtest("-9223372036854775809", LLONG_MIN, LLONG_MAX, 0, false);

	runtest("9223372036854775807", LLONG_MIN, LLONG_MAX, LLONG_MAX, true);
	runtest("9223372036854775808", LLONG_MIN, LLONG_MAX, 0, false);

	for (int base = 2; base <= 36; base++)
		runtestx("1", 0, 100, base, 1, true);

	runtestx("1", 0, 100, -2, 0, false);
	runtestx("1", 0, 100, 1, 0, false);
	runtestx("1", 0, 100, 37, 0, false);

	runtestx("0x1234", 0, LLONG_MAX, 16, 0x1234, true);
	runtestx("0x1234", 0, LLONG_MAX, 0, 0x1234, true);
	runtestx("0x1234", 0, LLONG_MAX, 10, 0, false);

	runtestx("AZbc123", 0, LLONG_MAX, 36, 23903176539, true);
	runtestx("AZbc123", 0, LLONG_MAX, 35, 0, false);
	runtestx("AYBC123", 0, LLONG_MAX, 35, 20185422673, true);
	runtestx("AYBC123", 0, LLONG_MAX, 34, 0, false);

	runtestx("01234", 0, LLONG_MAX, 8, 01234, true);
	runtestx("01234", 0, LLONG_MAX, 0, 01234, true);
	runtestx("01234", 0, LLONG_MAX, 10, 1234, true);

	if (fail) {
		printf("\nOverall status: FAIL\n");
		return (EXIT_FAILURE);
	}

	printf("\nOverall status: PASS\n");
	return (EXIT_SUCCESS);
}