/*	$OpenBSD: test-seqnum.c,v 1.1 2024/10/07 12:27:27 tb Exp $ */

/*
 * Copyright (c) 2024 Theo Buehler <tb@openbsd.org>
 *
 * Permission to use, copy, modify, and distribute this software for any
 * purpose with or without fee is hereby granted, provided that the above
 * copyright notice and this permission notice appear in all copies.
 *
 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 */

#include <openssl/asn1.h>
#include <openssl/bn.h>

#include <err.h>
#include <stdio.h>
#include <unistd.h>

#include "extern.h"

#define MAX_DER 25

static const struct seqnum {
	const char *descr;
	const unsigned char der[MAX_DER];
	int der_len;
	int valid;
} seqnum_tests[] = {
	{
		.descr = "0 - smallest acceptable value:",
		.der = {
			0x02, 0x01, 0x00,
		},
		.der_len = 3,
		.valid = 1,
	},
	{
		.descr = "1 - acceptable:",
		.der = {
			0x02, 0x01, 0x01,
		},
		.der_len = 3,
		.valid = 1,
	},
	{
		.descr = "-1 - invalid:",
		.der = {
			0x02, 0x01, 0xff,
		},
		.der_len = 3,
		.valid = 0,
	},
	{
		.descr = "2^159 - 1 - largest acceptable value:",
		.der = {
			0x02, 0x14,
			0x7f, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,
			0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,
			0xff, 0xff, 0xff, 0xff,
		},
		.der_len = 22,
		.valid = 1,
	},
	{
		.descr = "-2^159 - invalid, but fits in 20 octets:",
		.der = {
			0x02, 0x14,
			0x80, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
			0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
			0x00, 0x00, 0x00, 0x00,
		},
		.der_len = 22,
		.valid = 0,
	},
	{
		.descr = "2^159 - smallest inacceptable positive value:",
		.der = {
			0x02, 0x15,
			0x00, 0x80, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
			0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
			0x00, 0x00, 0x00, 0x00, 0x00,
		},
		.der_len = 23,
		.valid = 0,
	},
	{
		.descr = "2^160 - 1 - largest unsigned 20-bit number:",
		.der = {
			0x02, 0x15,
			0x00, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,
			0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,
			0xff, 0xff, 0xff, 0xff, 0xff,
		},
		.der_len = 23,
		.valid = 0,
	},
	{
		.descr = "2^160: too large:",
		.der = {
			0x02, 0x15,
			0x01, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
			0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
			0x00, 0x00, 0x00, 0x00, 0x00,
		},
		.der_len = 23,
		.valid = 0,
	},
};

#define N_SEQNUM_TESTS (sizeof(seqnum_tests) / sizeof(seqnum_tests[0]))

static int
seqnum_testcase(const struct seqnum *test)
{
	ASN1_INTEGER *aint = NULL;
	const BIGNUM *bn;
	const unsigned char *p;
	char *s = NULL;
	int failed = 1;

	p = test->der;
	if ((aint = d2i_ASN1_INTEGER(NULL, &p, test->der_len)) == NULL) {
		fprintf(stderr, "FAIL: %s d2i_ASN1_INTEGER\n", test->descr);
		goto err;
	}

	s = x509_convert_seqnum(__func__, test->descr, aint);

	if (s == NULL && test->valid) {
		fprintf(stderr, "FAIL: %s failed to convert seqnum\n",
		    test->descr);
		goto err;
	}
	if (s != NULL && !test->valid) {
		fprintf(stderr, "FAIL: %s invalid seqnum succeeded\n",
		    test->descr);
		goto err;
	}

	failed = 0;
 err:
	ASN1_INTEGER_free(aint);
	free(s);

	return failed;
}

static int
seqnum_test(void)
{
	size_t i;
	int failed = 0;

	for (i = 0; i < N_SEQNUM_TESTS; i++)
		failed |= seqnum_testcase(&seqnum_tests[i]);

	return failed;
}

time_t
get_current_time(void)
{
	return time(NULL);
}

int experimental, filemode, outformats, verbose;

int
main(void)
{
	int failed = 0;

	failed = seqnum_test();

	if (!failed)
		printf("OK\n");

	return failed;
}