sangge/tpre-python
2
1
Fork 1
Code Issues 6 Pull Requests Packages Projects 1 Activity

refactor: rename demo to speed_test

Browse Source
This commit is contained in:
sangge 2023-11-17 08:13:13 +08:00
parent ae14db82d9
commit e68bf52b91
2 changed files with 61 additions and 71 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

71
src/demo.py
View File

@ -1,71 +0,0 @@
from tpre import *
import time
# for T in range(2, 20, 2):
N = 10
T = N // 2
# print(f"当前门限值: N = {N}, T = {T}")
start_total_time = time.time()
# 1
start_time = time.time()
pk_a, sk_a = GenerateKeyPair()
# print("pk_a: ", pk_a)
# print("sk_a: ", sk_a)
end_time = time.time()
elapsed_time = end_time - start_time
# print(f"密钥生成运行时间:{elapsed_time}秒")
# 2
start_time = time.time()
m = b"hello world"
capsule_ct = Encrypt(pk_a, m)
capsule = capsule_ct[0]
print("check capsule: ", Checkcapsule(capsule))
capsule = (capsule[0], capsule[1], -1)
print("check capsule: ", Checkcapsule(capsule))
# print("capsule_ct: ", capsule_ct)
end_time = time.time()
elapsed_time = end_time - start_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)
# print("rekeys: ", rekeys)
end_time = time.time()
elapsed_time = end_time - start_time
# print(f"重加密密钥生成算法运行时间:{elapsed_time}秒")
# 7
start_time = time.time()
cfrag_cts = []
for rekey in rekeys:
cfrag_ct = ReEncrypt(rekey, capsule_ct)
# cfrag_ct = ReEncrypt(rekeys[0], capsule_ct)
cfrag_cts.append(cfrag_ct)
# print("cfrag_cts: ", cfrag_cts)
end_time = time.time()
re_elapsed_time = (end_time - start_time) / len(rekeys)
# print(f"重加密算法运行时间:{re_elapsed_time}秒")
# 9
start_time = time.time()
cfrags = mergecfrag(cfrag_cts)
# print("cfrags: ", cfrags)
# m = DecryptFrags(sk_b, pk_b, pk_a, cfrags)
m = DecryptFrags(sk_a, pk_b, pk_a, cfrags)
# print("m = ", m)
end_time = time.time()
elapsed_time = end_time - start_time
end_total_time = time.time()
total_time = end_total_time - start_total_time - re_elapsed_time * len(rekeys)
# print(f"解密算法运行时间:{elapsed_time}秒")
# print("成功解密:", m)
# print(f"算法总运行时间:{total_time}秒")
# print()

61
src/speed_test.py Normal file
View File

@ -0,0 +1,61 @@
from tpre import *
import time
N = 20
T = N // 2
print(f"当前门限值: N = {N}, T = {T}")
total_time = 0
# 1
start_time = time.time()
pk_a, sk_a = GenerateKeyPair()
m = b"hello world"
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("成功解密:", m)
print(f"算法总运行时间:{total_time}秒")
print()