Refactor libzrtp unit tests

Modified-by: Travis Cross <tc@traviscross.com> Signed-off-by: Travis Cross <tc@traviscross.com>
2026-10-09 21:14:02 +00:00 · 2013-02-11 16:41:45 +00:00
parent 7bca61ff0c
commit 73fabdfea5
7 changed files with 694 additions and 447 deletions
@@ -0,0 +1,122 @@
+/*
+ * libZRTP SDK library, implements the ZRTP secure VoIP protocol.
+ * Copyright (c) 2006-2009 Philip R. Zimmermann.  All rights reserved.
+ * Contact: http://philzimmermann.com
+ * For licensing and other legal details, see the file zrtp_legal.c.
+ * 
+ * Viktor Krykun <v.krikun at zfoneproject.com> 
+ */
+
+#include <setjmp.h>
+#include <stdio.h>
+
+#include "zrtp.h"
+#include "cmockery/cmockery.h"
+
+zrtp_global_t *zrtp;
+
+void setup() {
+	zrtp_status_t s;
+	zrtp_config_t zrtp_config;
+	
+	zrtp_config_defaults(&zrtp_config);
+	
+	s = zrtp_init(&zrtp_config, &zrtp);
+	assert_int_equal(s, zrtp_status_ok);
+}
+
+void teardown() {
+	zrtp_down(zrtp);
+}
+
+static void aes128_ctr_test() {
+	zrtp_cipher_t *cipher = zrtp_comp_find(ZRTP_CC_CIPHER, ZRTP_CIPHER_AES128, zrtp);
+	assert_non_null(cipher);
+	cipher->self_test(cipher, ZRTP_CIPHER_MODE_CTR);
+}
+
+static void aes128_cfb_test() {
+	zrtp_cipher_t *cipher = zrtp_comp_find(ZRTP_CC_CIPHER, ZRTP_CIPHER_AES128, zrtp);
+	assert_non_null(cipher);
+	cipher->self_test(cipher, ZRTP_CIPHER_MODE_CFB);
+}
+
+static void aes256_ctr_test() {
+	zrtp_cipher_t *cipher = zrtp_comp_find(ZRTP_CC_CIPHER, ZRTP_CIPHER_AES256, zrtp);
+	assert_non_null(cipher);
+	cipher->self_test(cipher, ZRTP_CIPHER_MODE_CTR);
+}
+
+static void aes256_cfb_test() {
+	zrtp_cipher_t *cipher = zrtp_comp_find(ZRTP_CC_CIPHER, ZRTP_CIPHER_AES256, zrtp);
+	assert_non_null(cipher);
+	cipher->self_test(cipher, ZRTP_CIPHER_MODE_CFB);
+}
+/*
+static void aes256_cfb_test2() {
+		rtp_aes_cfb_ctx_t *ctx = (zrtp_aes_cfb_ctx_t*)self->start( self,
+																    aes_cfb_test_key,
+																    NULL,
+																    ZRTP_CIPHER_MODE_CFB);
+		if (NULL == ctx) {
+			return zrtp_status_fail;
+		}
+
+		ZRTP_LOG(3, (_ZTU_,"256 bit AES CFB\n"));
+		ZRTP_LOG(3, (_ZTU_, "1st test...\n"));
+
+		zrtp_memcpy(aes_cfb_test_buf2b, aes_cfb_test_buf2a, sizeof(aes_cfb_test_buf2a));
+		zrtp_memcpy(&tmp_iv, aes_cfb_test_iv, sizeof(tmp_iv));
+
+		ZRTP_LOG(3, (_ZTU_, "\tencryption... "));
+
+		self->set_iv(self, ctx, &tmp_iv);
+		err = self->encrypt(self, ctx, aes_cfb_test_buf2b, sizeof(aes_cfb_test_buf2b));
+		if (zrtp_status_ok != err) {
+			ZRTP_LOGC(1, ("ERROR! 256-bit AES CFB encrypt returns error %d\n", err));
+			self->stop(self, ctx);
+			return err;
+		}
+
+		for (i=0; i<16; i++) {
+			if (aes_cfb_test_buf2b[i] != 0x00) {
+				ZRTP_LOGC(1, ("ERROR! 256-bit AES CFB failed on encrypt test\n"));
+				self->stop(self, ctx);
+				return zrtp_status_fail;
+			}
+		}
+		ZRTP_LOGC(3, ("OK\n"));
+
+		ZRTP_LOG(3, (_ZTU_, "\tdecryption... "));
+
+		zrtp_memcpy(&tmp_iv, aes_cfb_test_iv, sizeof(tmp_iv));
+		self->set_iv(self, ctx,  &tmp_iv);
+
+		err = self->decrypt(self, ctx, aes_cfb_test_buf2b, sizeof(aes_cfb_test_buf2b));
+		if (zrtp_status_ok != err) {
+			ZRTP_LOGC(1, ("ERROR! 256-bit AES CFB decrypt returns error %d\n", err));
+			self->stop(self, ctx);
+			return err;
+		}
+		for (i=0; i<sizeof(aes_cfb_test_buf2b); i++) {
+			if (aes_cfb_test_buf2b[i] != aes_cfb_test_buf2a[i]) {
+				ZRTP_LOGC(1, ("ERROR! 256-bit AES CFB failed on decrypt test\n"));
+				self->stop(self, ctx);
+				return zrtp_status_fail;
+			}
+		}
+		self->stop(self, ctx);
+		ZRTP_LOGC(3, ("OK\n"));
+
+}
+*/
+int main(void) {
+	const UnitTest tests[] = {
+		unit_test_setup_teardown(aes128_ctr_test, setup, teardown),
+		unit_test_setup_teardown(aes128_cfb_test, setup, teardown),
+		unit_test_setup_teardown(aes256_ctr_test, setup, teardown),
+		unit_test_setup_teardown(aes256_cfb_test, setup, teardown),
+  	};
+
+	return run_tests(tests);
+}