/*

 * Copyright (C) 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, see <http://www.gnu.org/licenses/>.

 *

 */

#ifdef HAVE_CONFIG_H

#include <config.h>

#endif

/* This is a unit test of _gnutls_record_overhead. */

#include <stdio.h>

#include <stdlib.h>

#include <unistd.h>

#include <sys/types.h>

#include <sys/time.h>

#include <errno.h>

#include <stdint.h>

#include <stddef.h>

#include <setjmp.h>

#include <cmocka.h>

#include <gnutls/gnutls.h>

#include "../lib/gnutls_int.h"

#undef _gnutls_debug_log

#undef gnutls_assert

#undef gnutls_assert_val

#define _gnutls_debug_log printf

#define gnutls_assert()

#define gnutls_assert_val(val) val

#include "../lib/algorithms.h"

int _gnutls_record_overhead(const cipher_entry_st * cipher,

			    const mac_entry_st * mac,

			    unsigned max);

#define OVERHEAD(c, m) \

	_gnutls_record_overhead(cipher_to_entry(c), mac_to_entry(m), \

				0)

#define MAX_OVERHEAD(c, m) \

	_gnutls_record_overhead(cipher_to_entry(c), mac_to_entry(m), \

				1)

static void check_aes_gcm(void **glob_state)

{

	const unsigned ov = 16+8;

	/* Under AES-GCM the overhead is constant */

	assert_int_equal(OVERHEAD(GNUTLS_CIPHER_AES_128_GCM, GNUTLS_MAC_AEAD), ov);

	assert_int_equal(MAX_OVERHEAD(GNUTLS_CIPHER_AES_128_GCM, GNUTLS_MAC_AEAD), ov);

}

static void check_aes_sha1_min(void **glob_state)

{

	const unsigned mac = 20;

	const unsigned block = 16;

	assert_int_equal(OVERHEAD(GNUTLS_CIPHER_AES_128_CBC, GNUTLS_MAC_SHA1), 1+mac+block);

}

static void check_aes_sha1_max(void **glob_state)

{

	const unsigned mac = 20;

	const unsigned block = 16;

	assert_int_equal(MAX_OVERHEAD(GNUTLS_CIPHER_AES_128_CBC, GNUTLS_MAC_SHA1), block+mac+block);

}

int main(void)

{

	const struct CMUnitTest tests[] = {

		cmocka_unit_test(check_aes_gcm),

		cmocka_unit_test(check_aes_sha1_min),

		cmocka_unit_test(check_aes_sha1_max)

	};

	return cmocka_run_group_tests(tests, NULL, NULL);

}