1 files changed, 97 insertions, 0 deletions
diff --git a/tests/gnutls_record_overhead.c b/tests/gnutls_record_overhead.c
new file mode 100644
index 0000000000..48fca72a28
--- /dev/null
+++ b/tests/gnutls_record_overhead.c
@@ -0,0 +1,97 @@
+/*
+ * 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);
+}