From 262619461d887c5d3c79854e8cd8fe7939156ac5 Mon Sep 17 00:00:00 2001
From: sangge <2251250136@qq.com>
Date: Thu, 26 Oct 2023 08:20:53 +0800
Subject: [PATCH] feat: update demo code

---
 src/demo.py | 93 +++++++++++++++++++++++++++++------------------------
 1 file changed, 51 insertions(+), 42 deletions(-)

diff --git a/src/demo.py b/src/demo.py
index 7c077b5..0f0646a 100644
--- a/src/demo.py
+++ b/src/demo.py
@@ -1,51 +1,60 @@
 from tpre import *
 import time
 
-# 1
-start_time = time.time()
-pk_a, sk_a = GenerateKeyPair()
-m = b"hello world"
-end_time = time.time()
-elapsed_time = end_time - start_time
-print(f"代码块1运行时间:{elapsed_time}秒")
+for N in range(4,21,4):
+    # N = 10
+    # T = 5
+    T = N // 2
+    print(f"当前门限值: N = {N}, T = {T}")
+    
+    start_total_time = time.time()
+    # 1
+    start_time = time.time()
+    pk_a, sk_a = GenerateKeyPair()
+    m = b"hello world"
+    end_time = time.time()
+    elapsed_time = end_time - start_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
-print(f"代码块2运行时间:{elapsed_time}秒")
+    # 2
+    start_time = time.time()
+    capsule_ct = Encrypt(pk_a, m)
+    end_time = time.time()
+    elapsed_time = end_time - start_time
+    print(f"加密算法运行时间:{elapsed_time}秒")
 
-# 3
-pk_b, sk_b = GenerateKeyPair()
+    # 3
+    pk_b, sk_b = GenerateKeyPair()
 
-N = 10
-T = 5
+    
+    # 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
+    print(f"重加密密钥生成算法运行时间:{elapsed_time}秒")
 
-# 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
-print(f"代码块5运行时间:{elapsed_time}秒")
+    # 7
+    start_time = time.time()
+    cfrag_cts = []
 
-# 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)
+    print(f"重加密算法运行时间:{elapsed_time}秒")
 
-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
-print(f"代码块7运行时间:{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
-print(f"代码块9运行时间:{elapsed_time}秒")
-print(m)
+    # 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
+    end_total_time = time.time()
+    total_time = end_total_time - start_total_time
+    print(f"解密算法运行时间:{elapsed_time}秒")
+    print("成功解密:", m)
+    print(f"算法总运行时间:{total_time}秒")
+    print()