/*

 * Copyright (C) 2016, 2017 Red Hat, Inc.

 *

 * Author: Nikos Mavrogiannopoulos

 *

 * This file is part of GnuTLS.

 *

 * GnuTLS 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.

 *

 * GnuTLS 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 GnuTLS; if not, write to the Free Software Foundation,

 * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA

 */

/* Parts copied from GnuTLS example programs. */

#ifdef HAVE_CONFIG_H

#include <config.h>

#endif

#include <stdio.h>

#include <stdlib.h>

#include <string.h>

#include <gnutls/gnutls.h>

#include "utils.h"

/* This checks random art encoding */

static void encode(const char *test_name, const char *type, unsigned key_size, unsigned char *input, unsigned input_size, const char *expected)

{

	int ret;

	gnutls_datum_t out;

	ret = gnutls_random_art(GNUTLS_RANDOM_ART_OPENSSH, type, key_size, input, input_size, &out;;

	if (ret < 0) {

		fail("%s: gnutls_random_art: %s\n", test_name, gnutls_strerror(ret));

		exit(1);

	}

	if (strlen(expected)!=out.size) {

		fail("%s: gnutls_random_art: output has incorrect size (%d, expected %d)\n%s\n", test_name, (int)out.size, (int)strlen(expected), out.data);

		exit(1);

	}

	if (strncasecmp(expected, (char*)out.data, out.size) != 0) {

		fail("%s: gnutls_random_art: output does not match the expected:\n%s\n", test_name, out.data);

		exit(1);

	}

	gnutls_free(out.data);

	return;

}

struct encode_tests_st {

	const char *name;

	unsigned char *raw;

	unsigned raw_size;

	const char *key_type;

	unsigned key_size;

	const char *art;

};

struct encode_tests_st encode_tests[] = {

	{

		.name = "key1",

		.raw = (void*)"\x38\x17\x0c\x08\xcb\x45\x8f\xd4\x87\x9c\x34\xb6\xf6\x08\x29\x4c\x50\x31\x2b\xbb",

		.raw_size = 20,

		.key_type = "RSA",

		.key_size = 2048,

		.art =	"+--[ RSA 2048]----+\n"

			"|.o*++==o         |\n"

			"| + *.===.        |\n"

			"|. * + +.o        |\n"

			"| o . o + .       |\n"

			"|.     + S        |\n"

			"| .     o         |\n"

			"|E                |\n"

			"|                 |\n"

			"|                 |\n"

			"+-----------------+"

	},

	{

		.name = "key2",

		.raw = (void*)"\xf8\xa7\x1c\x08\x76\x47\x2c\x08\x38\x17\x0c\x08\x38\x17\x0c\x08\xcb\x45\x8f\xd4\x87\x9c\xa4\xb6\xf6\xf8\x29\xfc\x50\x3f\x2b\xbb",

		.raw_size = 32,

		.key_type = "RSA",

		.key_size = 3072,

		.art =	"+--[ RSA 3072]----+\n"

			"|@*=*+.o          |\n"

			"|O.B.+* o         |\n"

			"|.* +..o o        |\n"

			"|  . .  +         |\n"

			"|   oo.o S        |\n"

			"|  ..+o.+         |\n"

			"|  .o ..oo .      |\n"

			"|   oo...o+       |\n"

			"|    oE+.o        |\n"

			"+-----------------+"

	},

	{

		.name = "key3",

		.raw = (void*)"\x38\xf7\x0c\x08\xcb\x34\x8a\xd4\xb7\x9c\x34\xb4\xf6\x08\x29\x4c\x50\x3f\x2b\xbb",

		.raw_size = 20,

		.key_type = "ECDSA",

		.key_size = 256,

		.art =	"+--[ECDSA  256]---+\n"

			"|oo.  .           |\n"

			"|o ..o .          |\n"

			"| + +**           |\n"

			"|...+***o         |\n"

			"|. o +=+.S        |\n"

			"|   o   o +       |\n"

			"|  .       o      |\n"

			"|   .             |\n"

			"|  E              |\n"

			"+-----------------+"

	}

};

void doit(void)

{

	unsigned i;

	for (i=0;i

		encode(encode_tests[i].name, encode_tests[i].key_type, encode_tests[i].key_size,

			encode_tests[i].raw, encode_tests[i].raw_size, encode_tests[i].art);

	}

}