From c0494494a75f28861c17efc1f5ae5b050f36b98f Mon Sep 17 00:00:00 2001
From: sangge <2251250136@qq.com>
Date: Tue, 17 Oct 2023 23:00:28 +0800
Subject: [PATCH] feat: finish demo

---
 src/demo.py | 28 ++++++++++++++++++++++++++++
 1 file changed, 28 insertions(+)

diff --git a/src/demo.py b/src/demo.py
index ad5f059..f511597 100644
--- a/src/demo.py
+++ b/src/demo.py
@@ -1,3 +1,31 @@
 from tpre import *
 
+# 1
+pk_a, sk_a = GenerateKeyPair(1, ())
+m = b'hello world'
+m = int.from_bytes(m)
+
+# 2
+capsule_ct = Encrypt(pk_a, m)
+
+# 3
+pk_b, sk_b = GenerateKeyPair(1, ())
+
+N = 20
+T = 10
+
+# 5
+rekeys =  GenerateReKey(sk_a, pk_b, N, T)
+
+# 7
+cfrag_cts = []
+
+for rekey in rekeys:
+    cfrag_ct = ReEncrypt(rekey, capsule_ct)
+    cfrag_cts.append(cfrag_ct)
+    
+# 9
+cfrags = mergecfrag(cfrag_cts)
+m = DecryptFrags(sk_b, pk_b, pk_a, cfrags)
+