/* $OpenBSD: asn1evp.c,v 1.3 2018/11/08 21:37:21 jsing Exp $ */
/*
 * Copyright (c) 2017 Joel Sing <jsing@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 <limits.h>
#include <stdio.h>
#include <string.h>

#include <openssl/asn1.h>

#define TEST_NUM 0x7fffffffL

unsigned char asn1_atios[] = {
	0x30, 0x10, 0x02, 0x04, 0x7f, 0xff, 0xff, 0xff,
	0x04, 0x08, 0x00, 0x01, 0x02, 0x03, 0x04, 0x05,
	0x06, 0x07,
};

unsigned char test_octetstring[] = {
	0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07,
};

static void
hexdump(const unsigned char *buf, size_t len)
{
        size_t i;

        for (i = 1; i <= len; i++)
                fprintf(stderr, " 0x%02hhx,%s", buf[i - 1], i % 8 ? "" : "\n");

        fprintf(stderr, "\n");
}

static int
compare_data(const char *label, const unsigned char *d1, size_t d1_len,
    const unsigned char *d2, size_t d2_len)
{
	if (d1_len != d2_len) {
		fprintf(stderr, "FAIL: got %s with length %zu, want %zu\n",
		    label, d1_len, d2_len);
		return -1;
	}
	if (memcmp(d1, d2, d1_len) != 0) {
		fprintf(stderr, "FAIL: %s differs\n", label);
		fprintf(stderr, "got:\n");
		hexdump(d1, d1_len);
		fprintf(stderr, "want:\n");
		hexdump(d2, d2_len);
		return -1;
	}
	return 0;
}

int
main(int argc, char **argv)
{
	unsigned char data[16];
	long num = TEST_NUM;
	ASN1_TYPE *at = NULL;
	int failed = 1;
	int len;

	if ((at = ASN1_TYPE_new()) == NULL) {
		fprintf(stderr, "FAIL: ASN1_TYPE_new returned NULL\n");
		goto done;
	}

	if (!ASN1_TYPE_set_int_octetstring(at, num, test_octetstring,
	    sizeof(test_octetstring))) {
		fprintf(stderr, "FAIL: ASN1_TYPE_set_int_octetstring failed\n");
		goto done;
	}
	if (at->type != V_ASN1_SEQUENCE) {
		fprintf(stderr, "FAIL: not a V_ASN1_SEQUENCE (%i != %i)\n",
		    at->type, V_ASN1_SEQUENCE);
		goto done;
	}
	if (at->value.sequence->type != V_ASN1_OCTET_STRING) {
		fprintf(stderr, "FAIL: not a V_ASN1_OCTET_STRING (%i != %i)\n",
		    at->type, V_ASN1_OCTET_STRING);
		goto done;
	}
	if (compare_data("sequence", at->value.sequence->data,
	    at->value.sequence->length, asn1_atios, sizeof(asn1_atios)) == -1)
		goto done;

	memset(&data, 0, sizeof(data));
	num = 0;

	if ((len = ASN1_TYPE_get_int_octetstring(at, &num, data,
	    sizeof(data))) < 0) {
		fprintf(stderr, "FAIL: ASN1_TYPE_get_int_octetstring failed\n");
		goto done;
	}
	if (num != TEST_NUM) {
		fprintf(stderr, "FAIL: got num %li, want %li\n", num, TEST_NUM);
		goto done;
	}
	if (compare_data("octet string", data, len,
	    test_octetstring, sizeof(test_octetstring)) == -1)
		goto done;
	if (data[len] != 0) {
		fprintf(stderr, "FAIL: octet string overflowed buffer\n");
		goto done;
	}

	memset(&data, 0, sizeof(data));
	num = 0;

	/* With a limit buffer, the output should be truncated... */
	if ((len = ASN1_TYPE_get_int_octetstring(at, &num, data, 4)) < 0) {
		fprintf(stderr, "FAIL: ASN1_TYPE_get_int_octetstring failed\n");
		goto done;
	}
	if (num != TEST_NUM) {
		fprintf(stderr, "FAIL: got num %li, want %li\n", num, TEST_NUM);
		goto done;
	}
	if (len != sizeof(test_octetstring)) {
		fprintf(stderr, "FAIL: got length mismatch (%i != %zu)\n",
		    len, sizeof(test_octetstring));
		goto done;
	}
	if (compare_data("octet string", data, 4, test_octetstring, 4) == -1)
		goto done;
	if (data[4] != 0) {
		fprintf(stderr, "FAIL: octet string overflowed buffer\n");
		goto done;
	}

	failed = 0;

 done:
	ASN1_TYPE_free(at);

	return failed;
}