/*      $OpenBSD: chachapoly_test.c,v 1.3 2021/12/14 06:27:48 deraadt Exp $  */

/*
 * Copyright (c) 2010,2015 Mike Belopuhov <mikeb@openbsd.org>
 * Copyright (c) 2005 Markus Friedl <markus@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 <sys/types.h>
#include <crypto/chachapoly.h>
#include <err.h>
#include <errno.h>
#include <string.h>
#include <stdlib.h>
#include <stdio.h>

#define MINIMUM(a, b) (((a) < (b)) ? (a) : (b))

int debug = 0;

enum { TST_KEY, TST_IV, TST_AAD, TST_PLAIN, TST_CIPHER, TST_TAG, TST_NUM };

struct {
	char	*data[TST_NUM];
} tests[] = {
	/* Chacha20, counter=1 test vectors */

	/* Test vector from RFC7539 2.4.2 */
	{
		/* key + salt */
		"00 01 02 03 04 05 06 07 08 09 0a 0b 0c 0d 0e 0f "
		"10 11 12 13 14 15 16 17 18 19 1a 1b 1c 1d 1e 1f "
		"00 00 00 00",
		/* iv */
		"00 00 00 4a 00 00 00 00",
		/* aad */
		NULL,
		/* plaintext */
		"4c 61 64 69 65 73 20 61 6e 64 20 47 65 6e 74 6c "
		"65 6d 65 6e 20 6f 66 20 74 68 65 20 63 6c 61 73 "
		"73 20 6f 66 20 27 39 39 3a 20 49 66 20 49 20 63 "
		"6f 75 6c 64 20 6f 66 66 65 72 20 79 6f 75 20 6f "
		"6e 6c 79 20 6f 6e 65 20 74 69 70 20 66 6f 72 20 "
		"74 68 65 20 66 75 74 75 72 65 2c 20 73 75 6e 73 "
		"63 72 65 65 6e 20 77 6f 75 6c 64 20 62 65 20 69 "
		"74 2e",
		/* ciphertext */
		"6e 2e 35 9a 25 68 f9 80 41 ba 07 28 dd 0d 69 81 "
		"e9 7e 7a ec 1d 43 60 c2 0a 27 af cc fd 9f ae 0b "
		"f9 1b 65 c5 52 47 33 ab 8f 59 3d ab cd 62 b3 57 "
		"16 39 d6 24 e6 51 52 ab 8f 53 0c 35 9f 08 61 d8 "
		"07 ca 0d bf 50 0d 6a 61 56 a3 8e 08 8a 22 b6 5e "
		"52 bc 51 4d 16 cc f8 06 81 8c e9 1a b7 79 37 36 "
		"5a f9 0b bf 74 a3 5b e6 b4 0b 8e ed f2 78 5e 42 "
		"87 4d",
		/* tag */
		NULL
	},

	/* Test vector#2 from RFC7539 A.2 */
	{
		/* key + salt */
		"00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 "
		"00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 01 "
		"00 00 00 00",
		/* iv */
		"00 00 00 00 00 00 00 02",
		/* aad */
		NULL,
		/* plaintext */
		"41 6e 79 20 73 75 62 6d 69 73 73 69 6f 6e 20 74 "
		"6f 20 74 68 65 20 49 45 54 46 20 69 6e 74 65 6e "
		"64 65 64 20 62 79 20 74 68 65 20 43 6f 6e 74 72 "
		"69 62 75 74 6f 72 20 66 6f 72 20 70 75 62 6c 69 "
		"63 61 74 69 6f 6e 20 61 73 20 61 6c 6c 20 6f 72 "
		"20 70 61 72 74 20 6f 66 20 61 6e 20 49 45 54 46 "
		"20 49 6e 74 65 72 6e 65 74 2d 44 72 61 66 74 20 "
		"6f 72 20 52 46 43 20 61 6e 64 20 61 6e 79 20 73 "
		"74 61 74 65 6d 65 6e 74 20 6d 61 64 65 20 77 69 "
		"74 68 69 6e 20 74 68 65 20 63 6f 6e 74 65 78 74 "
		"20 6f 66 20 61 6e 20 49 45 54 46 20 61 63 74 69 "
		"76 69 74 79 20 69 73 20 63 6f 6e 73 69 64 65 72 "
		"65 64 20 61 6e 20 22 49 45 54 46 20 43 6f 6e 74 "
		"72 69 62 75 74 69 6f 6e 22 2e 20 53 75 63 68 20 "
		"73 74 61 74 65 6d 65 6e 74 73 20 69 6e 63 6c 75 "
		"64 65 20 6f 72 61 6c 20 73 74 61 74 65 6d 65 6e "
		"74 73 20 69 6e 20 49 45 54 46 20 73 65 73 73 69 "
		"6f 6e 73 2c 20 61 73 20 77 65 6c 6c 20 61 73 20 "
		"77 72 69 74 74 65 6e 20 61 6e 64 20 65 6c 65 63 "
		"74 72 6f 6e 69 63 20 63 6f 6d 6d 75 6e 69 63 61 "
		"74 69 6f 6e 73 20 6d 61 64 65 20 61 74 20 61 6e "
		"79 20 74 69 6d 65 20 6f 72 20 70 6c 61 63 65 2c "
		"20 77 68 69 63 68 20 61 72 65 20 61 64 64 72 65 "
		"73 73 65 64 20 74 6f",
		/* ciphertext */
		"a3 fb f0 7d f3 fa 2f de 4f 37 6c a2 3e 82 73 70 "
		"41 60 5d 9f 4f 4f 57 bd 8c ff 2c 1d 4b 79 55 ec "
		"2a 97 94 8b d3 72 29 15 c8 f3 d3 37 f7 d3 70 05 "
		"0e 9e 96 d6 47 b7 c3 9f 56 e0 31 ca 5e b6 25 0d "
		"40 42 e0 27 85 ec ec fa 4b 4b b5 e8 ea d0 44 0e "
		"20 b6 e8 db 09 d8 81 a7 c6 13 2f 42 0e 52 79 50 "
		"42 bd fa 77 73 d8 a9 05 14 47 b3 29 1c e1 41 1c "
		"68 04 65 55 2a a6 c4 05 b7 76 4d 5e 87 be a8 5a "
		"d0 0f 84 49 ed 8f 72 d0 d6 62 ab 05 26 91 ca 66 "
		"42 4b c8 6d 2d f8 0e a4 1f 43 ab f9 37 d3 25 9d "
		"c4 b2 d0 df b4 8a 6c 91 39 dd d7 f7 69 66 e9 28 "
		"e6 35 55 3b a7 6c 5c 87 9d 7b 35 d4 9e b2 e6 2b "
		"08 71 cd ac 63 89 39 e2 5e 8a 1e 0e f9 d5 28 0f "
		"a8 ca 32 8b 35 1c 3c 76 59 89 cb cf 3d aa 8b 6c "
		"cc 3a af 9f 39 79 c9 2b 37 20 fc 88 dc 95 ed 84 "
		"a1 be 05 9c 64 99 b9 fd a2 36 e7 e8 18 b0 4b 0b "
		"c3 9c 1e 87 6b 19 3b fe 55 69 75 3f 88 12 8c c0 "
		"8a aa 9b 63 d1 a1 6f 80 ef 25 54 d7 18 9c 41 1f "
		"58 69 ca 52 c5 b8 3f a3 6f f2 16 b9 c1 d3 00 62 "
		"be bc fd 2d c5 bc e0 91 19 34 fd a7 9a 86 f6 e6 "
		"98 ce d7 59 c3 ff 9b 64 77 33 8f 3d a4 f9 cd 85 "
		"14 ea 99 82 cc af b3 41 b2 38 4d d9 02 f3 d1 ab "
		"7a c6 1d d2 9c 6f 21 ba 5b 86 2f 37 30 e3 7c fd "
		"c4 fd 80 6c 22 f2 21",
		/* tag */
		NULL
	},

	/* Poly1305 test vectors */

	/* Test vector from RFC7539 2.8.2 */
	{
		/* key + salt */
		"80 81 82 83 84 85 86 87 88 89 8a 8b 8c 8d 8e 8f "
		"90 91 92 93 94 95 96 97 98 99 9a 9b 9c 9d 9e 9f "
		"07 00 00 00",
		/* iv */
		"40 41 42 43 44 45 46 47",
		/* aad */
		"50 51 52 53 c0 c1 c2 c3 c4 c5 c6 c7",
		/* plaintext */
		"4c 61 64 69 65 73 20 61 6e 64 20 47 65 6e 74 6c "
		"65 6d 65 6e 20 6f 66 20 74 68 65 20 63 6c 61 73 "
		"73 20 6f 66 20 27 39 39 3a 20 49 66 20 49 20 63 "
		"6f 75 6c 64 20 6f 66 66 65 72 20 79 6f 75 20 6f "
		"6e 6c 79 20 6f 6e 65 20 74 69 70 20 66 6f 72 20 "
		"74 68 65 20 66 75 74 75 72 65 2c 20 73 75 6e 73 "
		"63 72 65 65 6e 20 77 6f 75 6c 64 20 62 65 20 69 "
		"74 2e",
		/* ciphertext */
		"d3 1a 8d 34 64 8e 60 db 7b 86 af bc 53 ef 7e c2 "
		"a4 ad ed 51 29 6e 08 fe a9 e2 b5 a7 36 ee 62 d6 "
		"3d be a4 5e 8c a9 67 12 82 fa fb 69 da 92 72 8b "
		"1a 71 de 0a 9e 06 0b 29 05 d6 a5 b6 7e cd 3b 36 "
		"92 dd bd 7f 2d 77 8b 8c 98 03 ae e3 28 09 1b 58 "
		"fa b3 24 e4 fa d6 75 94 55 85 80 8b 48 31 d7 bc "
		"3f f4 de f0 8e 4b 7a 9d e5 76 d2 65 86 ce c6 4b "
		"61 16",
		/* tag */
		"1a e1 0b 59 4f 09 e2 6a 7e 90 2e cb d0 60 06 91"
	},

	/* Test vector from RFC7539 Appendix A.5 */
	{
		/* key + salt */
		"1c 92 40 a5 eb 55 d3 8a f3 33 88 86 04 f6 b5 f0 "
		"47 39 17 c1 40 2b 80 09 9d ca 5c bc 20 70 75 c0 "
		"00 00 00 00",
		/* iv */
		"01 02 03 04 05 06 07 08",
		/* aad */
		"f3 33 88 86 00 00 00 00 00 00 4e 91",
		/* plaintext */
		"49 6e 74 65 72 6e 65 74 2d 44 72 61 66 74 73 20 "
		"61 72 65 20 64 72 61 66 74 20 64 6f 63 75 6d 65 "
		"6e 74 73 20 76 61 6c 69 64 20 66 6f 72 20 61 20 "
		"6d 61 78 69 6d 75 6d 20 6f 66 20 73 69 78 20 6d "
		"6f 6e 74 68 73 20 61 6e 64 20 6d 61 79 20 62 65 "
		"20 75 70 64 61 74 65 64 2c 20 72 65 70 6c 61 63 "
		"65 64 2c 20 6f 72 20 6f 62 73 6f 6c 65 74 65 64 "
		"20 62 79 20 6f 74 68 65 72 20 64 6f 63 75 6d 65 "
		"6e 74 73 20 61 74 20 61 6e 79 20 74 69 6d 65 2e "
		"20 49 74 20 69 73 20 69 6e 61 70 70 72 6f 70 72 "
		"69 61 74 65 20 74 6f 20 75 73 65 20 49 6e 74 65 "
		"72 6e 65 74 2d 44 72 61 66 74 73 20 61 73 20 72 "
		"65 66 65 72 65 6e 63 65 20 6d 61 74 65 72 69 61 "
		"6c 20 6f 72 20 74 6f 20 63 69 74 65 20 74 68 65 "
		"6d 20 6f 74 68 65 72 20 74 68 61 6e 20 61 73 20 "
		"2f e2 80 9c 77 6f 72 6b 20 69 6e 20 70 72 6f 67 "
		"72 65 73 73 2e 2f e2 80 9d",
		/* ciphertext */
		"64 a0 86 15 75 86 1a f4 60 f0 62 c7 9b e6 43 bd "
		"5e 80 5c fd 34 5c f3 89 f1 08 67 0a c7 6c 8c b2 "
		"4c 6c fc 18 75 5d 43 ee a0 9e e9 4e 38 2d 26 b0 "
		"bd b7 b7 3c 32 1b 01 00 d4 f0 3b 7f 35 58 94 cf "
		"33 2f 83 0e 71 0b 97 ce 98 c8 a8 4a bd 0b 94 81 "
		"14 ad 17 6e 00 8d 33 bd 60 f9 82 b1 ff 37 c8 55 "
		"97 97 a0 6e f4 f0 ef 61 c1 86 32 4e 2b 35 06 38 "
		"36 06 90 7b 6a 7c 02 b0 f9 f6 15 7b 53 c8 67 e4 "
		"b9 16 6c 76 7b 80 4d 46 a5 9b 52 16 cd e7 a4 e9 "
		"90 40 c5 a4 04 33 22 5e e2 82 a1 b0 a0 6c 52 3e "
		"af 45 34 d7 f8 3f a1 15 5b 00 47 71 8c bc 54 6a "
		"0d 07 2b 04 b3 56 4e ea 1b 42 22 73 f5 48 27 1a "
		"0b b2 31 60 53 fa 76 99 19 55 eb d6 31 59 43 4e "
		"ce bb 4e 46 6d ae 5a 10 73 a6 72 76 27 09 7a 10 "
		"49 e6 17 d9 1d 36 10 94 fa 68 f0 ff 77 98 71 30 "
		"30 5b ea ba 2e da 04 df 99 7b 71 4d 6c 6f 2c 29 "
		"a6 ad 5c b4 02 2b 02 70 9b",
		/* tag */
		"ee ad 9d 67 89 0c bb 22 39 23 36 fe a1 85 1f 38"
	},

	/* Test vector from RFC7634 Appendix A */
	{
		/* key + salt */
		"80 81 82 83 84 85 86 87 88 89 8a 8b 8c 8d 8e 8f "
		"90 91 92 93 94 95 96 97 98 99 9a 9b 9c 9d 9e 9f "
		"a0 a1 a2 a3",
		/* iv */
		"10 11 12 13 14 15 16 17",
		/* aad */
		"01 02 03 04 00 00 00 05",
		/* plaintext */
		"45 00 00 54 a6 f2 00 00 40 01 e7 78 c6 33 64 05 "
		"c0 00 02 05 08 00 5b 7a 3a 08 00 00 55 3b ec 10 "
		"00 07 36 27 08 09 0a 0b 0c 0d 0e 0f 10 11 12 13 "
		"14 15 16 17 18 19 1a 1b 1c 1d 1e 1f 20 21 22 23 "
		"24 25 26 27 28 29 2a 2b 2c 2d 2e 2f 30 31 32 33 "
		"34 35 36 37 01 02 02 04",
		/* ciphertext */
		"24 03 94 28 b9 7f 41 7e 3c 13 75 3a 4f 05 08 7b "
		"67 c3 52 e6 a7 fa b1 b9 82 d4 66 ef 40 7a e5 c6 "
		"14 ee 80 99 d5 28 44 eb 61 aa 95 df ab 4c 02 f7 "
		"2a a7 1e 7c 4c 4f 64 c9 be fe 2f ac c6 38 e8 f3 "
		"cb ec 16 3f ac 46 9b 50 27 73 f6 fb 94 e6 64 da "
		"91 65 b8 28 29 f6 41 e0",
		/* tag */
		"76 aa a8 26 6b 7f b0 f7 b1 1b 36 99 07 e1 ad 43"
	},

	/* Test vector from RFC7634 Appendix B */
	{
		/* key + salt */
		"80 81 82 83 84 85 86 87 88 89 8a 8b 8c 8d 8e 8f "
		"90 91 92 93 94 95 96 97 98 99 9a 9b 9c 9d 9e 9f "
		"a0 a1 a2 a3",
		/* iv */
		"10 11 12 13 14 15 16 17",
		/* aad */
		"c0 c1 c2 c3 c4 c5 c6 c7 d0 d1 d2 d3 d4 d5 d6 d7 "
		"2e 20 25 00 00 00 00 09 00 00 00 45 29 00 00 29",
		/* plaintext */
		"00 00 00 0c 00 00 40 01 00 00 00 0a 00",
		/* ciphertext */
		"61 03 94 70 1f 8d 01 7f 7c 12 92 48 89",
		/* tag */
		"6b 71 bf e2 52 36 ef d7 cd c6 70 66 90 63 15 b2"
	},

	/* Chacha20 test cases from the libcrypto aeadtests.txt */
	{
		/* key + salt */
		"42 90 bc b1 54 17 35 31 f3 14 af 57 f3 be 3b 50 "
		"06 da 37 1e ce 27 2a fa 1b 5d bd d1 10 0a 10 07 "
		"00 00 00 00",
		/* iv */
		"cd 7c f6 7b e3 9c 79 4a",
		/* aad */
		NULL,
		/* plaintext */
		"86 d0 99 74 84 0b de d2 a5 ca",
		/* ciphertext */
		"e3 e4 46 f7 ed e9 a1 9b 62 a4",
		/* tag */
		NULL
	}
};

static void
dochacha(unsigned char *key, size_t klen, unsigned char *iv,
    size_t ivlen, unsigned char *plain, size_t plen)
{
	struct chacha20_ctx ctx;
	uint8_t blk[CHACHA20_BLOCK_LEN];
	int i;

	memset(&ctx, 0, sizeof ctx);

	if (chacha20_setkey(&ctx, key, CHACHA20_KEYSIZE + CHACHA20_SALT))
		errx(1, "chacha20_setkey");
	chacha20_reinit((caddr_t)&ctx, iv);

	while (plen >= CHACHA20_BLOCK_LEN) {
		chacha20_crypt((caddr_t)&ctx, plain);
		plain += CHACHA20_BLOCK_LEN;
		plen -= CHACHA20_BLOCK_LEN;
	}
	if (plen > 0) {
		for (i = 0; i < plen; i++)
			blk[i] = plain[i];
		for (; i < CHACHA20_BLOCK_LEN; i++)
			blk[i] = 0;
		chacha20_crypt((caddr_t)&ctx, blk);
		memcpy(plain, blk, plen);
	}
}

static void
dopoly(const unsigned char *key, size_t klen,
    const unsigned char *iv, size_t ivlen,
    const unsigned char *aad, size_t aadlen,
    const unsigned char *in, unsigned char *out, size_t len)
{
	CHACHA20_POLY1305_CTX ctx;
	uint8_t blk[CHACHA20_BLOCK_LEN];
	uint32_t *p;
	int i;

	Chacha20_Poly1305_Init(&ctx);

	Chacha20_Poly1305_Setkey(&ctx, key, klen);

	Chacha20_Poly1305_Reinit(&ctx, iv, ivlen);

	for (i = 0; i < aadlen; i += POLY1305_BLOCK_LEN) {
		memset(blk, 0, POLY1305_BLOCK_LEN);
		memcpy(blk, aad + i, MINIMUM(aadlen - i, POLY1305_BLOCK_LEN));
		Chacha20_Poly1305_Update(&ctx, blk, POLY1305_BLOCK_LEN);
	}

	for (i = 0; i < len; i += CHACHA20_BLOCK_LEN) {
		int dlen = MINIMUM(len - i, CHACHA20_BLOCK_LEN);
		Chacha20_Poly1305_Update(&ctx, in + i, dlen);
	}

	bzero(blk, sizeof blk);
	p = (uint32_t *)blk;
	*p = htole32(aadlen);
	p = (uint32_t *)blk + 2;
	*p = htole32(len);
	Chacha20_Poly1305_Update(&ctx, blk, POLY1305_BLOCK_LEN);

	Chacha20_Poly1305_Final(out, &ctx);
}

static int
match(unsigned char *a, unsigned char *b, size_t len)
{
	int i;

	if (memcmp(a, b, len) == 0)
		return (1);

	warnx("mismatch");

	for (i = 0; i < len; i++)
		printf("%2.2x", a[i]);
	printf("\n");
	for (i = 0; i < len; i++)
		printf("%2.2x", b[i]);
	printf("\n");

	return (0);
}

static int
run(int num)
{
	int i, fail = 1, len, j, length[TST_NUM];
	u_long val;
	char *ep, *from;
	u_char *p, *data[TST_NUM], tag[POLY1305_TAGLEN];
	u_char *ciphertext;

	for (i = 0; i < TST_NUM; i++)
		data[i] = NULL;
	for (i = 0; i < TST_NUM; i++) {
		from = tests[num].data[i];
		if (debug)
			printf("%s\n", from);
		if (!from) {
			length[i] = 0;
			data[i] = NULL;
			continue;
		}
		len = strlen(from);
		if ((p = malloc(len)) == NULL) {
			warn("malloc");
			goto done;
		}
		errno = 0;
		for (j = 0; j < len; j++) {
			val = strtoul(&from[j*3], &ep, 16);
			p[j] = (u_char)val;
			if (*ep == '\0' || errno)
				break;
		}
		length[i] = j+1;
		data[i] = p;
	}

	if (length[TST_PLAIN] != 0) {
		dochacha(data[TST_KEY], length[TST_KEY], data[TST_IV],
		    length[TST_IV], data[TST_PLAIN], length[TST_PLAIN]);
		fail = !match(data[TST_CIPHER], data[TST_PLAIN],
		    length[TST_PLAIN]);
		printf("%s test vector %d (Chacha20)\n",
		    fail ? "FAILED" : "OK", num);
	}
	if (length[TST_AAD] != 0) {
		dopoly(data[TST_KEY], length[TST_KEY], data[TST_IV],
		    length[TST_IV], data[TST_AAD], length[TST_AAD],
		    data[TST_CIPHER], tag, length[TST_CIPHER]);
		fail = !match(data[TST_TAG], tag, POLY1305_TAGLEN);
		printf("%s test vector %d (Poly1305)\n",
		    fail ? "FAILED" : "OK", num);
	}

 done:
	for (i = 0; i < TST_NUM; i++)
		free(data[i]);
	return (fail);
}

int
main(void)
{
	int i, fail = 0;

	for (i = 0; i < (sizeof(tests) / sizeof(tests[0])); i++)
		fail += run(i);

	return (fail > 0 ? 1 : 0);
}