94 lines
2.5 KiB
Python
94 lines
2.5 KiB
Python
from tpre import *
|
|
import time
|
|
import openpyxl
|
|
|
|
# 初始化Excel工作簿和工作表
|
|
wb = openpyxl.Workbook()
|
|
ws = wb.active
|
|
ws.title = "算法性能结果"
|
|
headers = [
|
|
"门限值 N",
|
|
"门限值 T",
|
|
"密钥生成运行时间",
|
|
"加密算法运行时间",
|
|
"重加密密钥生成算法运行时间",
|
|
"重加密算法运行时间",
|
|
"解密算法运行时间",
|
|
"算法总运行时间",
|
|
]
|
|
ws.append(headers)
|
|
|
|
|
|
for N in range(4, 21, 2):
|
|
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_key_gen = end_time - start_time
|
|
print(f"密钥生成运行时间:{elapsed_time_key_gen}秒")
|
|
|
|
# ... [中间代码不变]
|
|
# 2
|
|
start_time = time.time()
|
|
capsule_ct = Encrypt(pk_a, m)
|
|
end_time = time.time()
|
|
elapsed_time_enc = end_time - start_time
|
|
print(f"加密算法运行时间:{elapsed_time_enc}秒")
|
|
|
|
# 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_rekey_gen = end_time - start_time
|
|
print(f"重加密密钥生成算法运行时间:{elapsed_time_rekey_gen}秒")
|
|
|
|
# 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()
|
|
re_elapsed_time = (end_time - start_time) / len(rekeys)
|
|
print(f"重加密算法运行时间:{re_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_dec = 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_dec}秒")
|
|
print("成功解密:", m)
|
|
print(f"算法总运行时间:{total_time}秒")
|
|
print()
|
|
|
|
# 将结果保存到Excel
|
|
ws.append(
|
|
[
|
|
N,
|
|
T,
|
|
elapsed_time_key_gen,
|
|
elapsed_time_enc,
|
|
elapsed_time_rekey_gen,
|
|
re_elapsed_time,
|
|
elapsed_time_dec,
|
|
total_time,
|
|
]
|
|
)
|
|
|
|
# 保存Excel文件
|
|
wb.save("结果.xlsx")
|