From f47a9c586e40de66d862d41e78499f566972d290 Mon Sep 17 00:00:00 2001
From: Zhi Guan <guanzhi1980@gmail.com>
Date: Mon, 29 Apr 2024 23:41:05 +0800
Subject: [PATCH] Add SM4 CCM test speed

---
 tests/sm4_ccmtest.c | 33 +++++++++++++++++++++++++++++++++
 1 file changed, 33 insertions(+)

diff --git a/tests/sm4_ccmtest.c b/tests/sm4_ccmtest.c
index fefddce2..f6a8ec89 100644
--- a/tests/sm4_ccmtest.c
+++ b/tests/sm4_ccmtest.c
@@ -12,6 +12,7 @@
 #include <string.h>
 #include <stdlib.h>
 #include <stdint.h>
+#include <time.h>
 #include <gmssl/sm4.h>
 #include <gmssl/hex.h>
 #include <gmssl/rand.h>
@@ -176,10 +177,42 @@ static int test_sm4_ccm_test_vectors(void)
 	return 1;
 }
 
+static int speed_sm4_ccm_encrypt(void)
+{
+	SM4_KEY sm4_key;
+	uint8_t key[16] = {0};
+	uint8_t iv[12];
+	uint8_t aad[16];
+	uint8_t tag[16];
+	uint32_t buf[1024];
+	clock_t begin, end;
+	double seconds;
+	int i;
+
+	sm4_set_encrypt_key(&sm4_key, key);
+
+	for (i = 0; i < 4096; i++) {
+		sm4_ccm_encrypt(&sm4_key, iv, sizeof(iv), aad, sizeof(aad), (uint8_t *)buf, sizeof(buf), (uint8_t *)buf, 16, tag);
+	}
+	begin = clock();
+	for (i = 0; i < 4096; i++) {
+		sm4_ccm_encrypt(&sm4_key, iv, sizeof(iv), aad, sizeof(aad), (uint8_t *)buf, sizeof(buf), (uint8_t *)buf, 16, tag);
+	}
+	end = clock();
+
+	seconds = (double)(end - begin)/ CLOCKS_PER_SEC;
+	fprintf(stderr, "%s: %f MiB per second\n", __FUNCTION__, 16/seconds);
+
+	return 1;
+}
+
 int main(void)
 {
 	if (test_sm4_ccm() != 1) goto err;
 	if (test_sm4_ccm_test_vectors() != 1) goto err;
+#if ENABLE_TEST_SPEED
+	if (speed_sm4_ccm_encrypt() != 1) goto err;
+#endif
 	printf("%s all tests passed\n", __FILE__);
 	return 0;
 err: