Nucypher
/
pyUmbral
mirror of https://github.com/nucypher/pyUmbral.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Type hints for decapsulate.

pull/9/head
jMyles 2018-01-03 12:47:15 -05:00
parent cfce67f4cd
commit e651b66f9f
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
umbral/umbral.py
View File

@ -284,7 +284,8 @@ class PRE(object):
else: # if len(reencrypted_keys) == 1:
return ReconstructedCapsule(e_prime=cFrag_0.e1, v_prime=cFrag_0.v1, x=cFrag_0.x)
def decapsulate_reencrypted(self, pub_key, priv_key, capsule, orig_pk, orig_ciphertext, key_length=32):
def decapsulate_reencrypted(self, pub_key: Point, priv_key: BigNum, capsule, orig_pub_key: Point,
orig_ciphertext: Capsule, key_length=32):
"""Derive the same symmetric key"""
xcomp = capsule.x
@ -301,6 +302,6 @@ class PRE(object):
s = orig_ciphertext.bn_sig
h = self.hash_to_bn([e, v])
inv_d = ~d
assert orig_pk * (s * inv_d) == v_prime + (e_prime * h), "Generic Umbral Error"
assert orig_pub_key * (s * inv_d) == v_prime + (e_prime * h), "Generic Umbral Error"
return key