/*
 * Copyright (C) 2016 Red Hat, Inc.
 *
 * This file is part of LIBTASN1.
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 *
 */

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <libtasn1.h>

int main (int argc, char** argv)
{
	int result = 0, len;
	asn1_node definitions = NULL, node1 = NULL;
	char errorDescription[ASN1_MAX_ERROR_DESCRIPTION_SIZE];
	const char *choicefile = getenv ("ASN1_MSCAT");
	const char *datafile = getenv ("ASN1_SPC_PE_IMAGE_DATA");
	FILE *fp;
	char data[1024];
	int data_size = sizeof (data);
	unsigned etype;
	int cmp;

	if (choicefile == NULL) {
		choicefile = "mscat.asn";
	}

	if (datafile == NULL) {
		datafile = "spc_pe_image_data.der";
	}

	result = asn1_parser2tree(choicefile, &definitions, errorDescription);
	if (result != ASN1_SUCCESS) {
		fprintf(stderr, "error in %d: %s\n", __LINE__, errorDescription);
		exit (1);
	}

	result = asn1_create_element(definitions,
				     "CATALOG.SpcPEImageData",
				     &node1);
	if (result != ASN1_SUCCESS) {
		fprintf(stderr, "error in %d\n", __LINE__);
		exit (1);
	}

	fp = fopen(datafile, "rb");
	if (fp == NULL) {
		fprintf (stderr, "error in %d\n", __LINE__);
		exit(1);
	}
	data_size = fread(data, 1, sizeof(data), fp);
	fclose(fp);

	result = asn1_der_decoding2(&node1,
				    data,
				    &data_size,
				    ASN1_DECODE_FLAG_STRICT_DER,
				    errorDescription);
	if (result != ASN1_SUCCESS) {
		fprintf(stderr, "error in %d: decoding: %s\n", __LINE__, errorDescription);
		exit(1);
	}

	len = 0;
	result = asn1_read_value_type(node1,
				      "flags",
				      NULL,
				      &len,
				      &etype);
	if (result != ASN1_MEM_ERROR) {
		fprintf(stderr, "error in %d: %s\n", __LINE__, errorDescription);
		exit(1);
	}

	if ((size_t)len >= sizeof(data)) {
		fprintf(stderr,
			"error len=%d, sizeof(data)=%zu\n",
			len,
			sizeof(data));
		exit(1);
	}

	result = asn1_read_value(node1, "flags", data, &len);
	if (result != ASN1_SUCCESS) {
		fprintf(stderr, "error in %d: %s\n", __LINE__, errorDescription);
		exit (1);
	}

	etype = 0;
	len = 0;
	result = asn1_read_value_type(node1, "link", NULL, &len, &etype);
	if (result != ASN1_MEM_ERROR)
	{
		fprintf(stderr,
			"error in %d: result is: %s\n",
			__LINE__,
			asn1_strerror(result));
		exit(1);
	}

	if (etype != ASN1_ETYPE_CHOICE) {
		fprintf(stderr,
			"error in %d: The type (%u) doesn't match choice.\n",
			__LINE__,
			etype);
		exit(1);
	}

	if (len != 5) {
		fprintf (stderr, "length doesn't match the expected (got: %d, should be: 20)\n", len);
		exit (1);
	}

	len = sizeof(data);
	result = asn1_read_value(node1, "link", data, &len);
	if (result != ASN1_SUCCESS) {
		fprintf(stderr, "error in %d: %s\n", __LINE__, errorDescription);
		exit (1);
	}

	cmp = memcmp(data, "file", len);
	if (cmp != 0) {
		fprintf(stderr, "contents don't match\n");
		exit(1);
	}

	printf("SUCCESS\n");

	asn1_delete_structure (&node1);
	asn1_delete_structure (&definitions);

	return 0;
}