From a8ccd00db7403fe6c74667b241a8f690b995547d Mon Sep 17 00:00:00 2001
From: sangge <2251250136@qq.com>
Date: Fri, 17 Nov 2023 16:20:18 +0800
Subject: [PATCH] feat: add lenth test

---
 src/lenth_test.py | 63 +++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 63 insertions(+)
 create mode 100644 src/lenth_test.py

diff --git a/src/lenth_test.py b/src/lenth_test.py
new file mode 100644
index 0000000..8c396f1
--- /dev/null
+++ b/src/lenth_test.py
@@ -0,0 +1,63 @@
+from tpre import *
+import time
+
+N = 20
+T = N // 2
+print(f"当前门限值: N = {N}, T = {T}")
+
+for i in range(1, 10):
+    total_time = 0
+
+    # 1
+    start_time = time.time()
+    pk_a, sk_a = GenerateKeyPair()
+    m = b"hello world" * pow(10, i)
+    print(f"明文长度:{len(m)}")
+    end_time = time.time()
+    elapsed_time = end_time - start_time
+    total_time += elapsed_time
+    print(f"密钥生成运行时间:{elapsed_time}秒")
+
+    # 2
+    start_time = time.time()
+    capsule_ct = Encrypt(pk_a, m)
+    end_time = time.time()
+    elapsed_time = end_time - start_time
+    total_time += elapsed_time
+    print(f"加密算法运行时间:{elapsed_time}秒")
+
+    # 3
+    pk_b, sk_b = GenerateKeyPair()
+
+    # 5
+    start_time = time.time()
+    id_tuple = tuple(range(N))
+    rekeys = GenerateReKey(sk_a, pk_b, N, T, id_tuple)
+    end_time = time.time()
+    elapsed_time = end_time - start_time
+    total_time += elapsed_time
+    print(f"重加密密钥生成算法运行时间:{elapsed_time}秒")
+
+    # 7
+    start_time = time.time()
+    cfrag_cts = []
+
+    for rekey in rekeys:
+        cfrag_ct = ReEncrypt(rekey, capsule_ct)
+        cfrag_cts.append(cfrag_ct)
+    end_time = time.time()
+    elapsed_time = (end_time - start_time) / len(rekeys)
+    total_time += elapsed_time
+    print(f"重加密算法运行时间:{elapsed_time}秒")
+
+    # 9
+    start_time = time.time()
+    cfrags = mergecfrag(cfrag_cts)
+    m = DecryptFrags(sk_b, pk_b, pk_a, cfrags)
+    end_time = time.time()
+    elapsed_time = end_time - start_time
+    total_time += elapsed_time
+    print(f"解密算法运行时间:{elapsed_time}秒")
+    print("成功解密:")
+    print(f"算法总运行时间:{total_time}秒")
+    print()