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

Have Capsule.get_size return an non-activated capsule size by default

pull/147/head
tuxxy 2018-05-23 04:49:04 -06:00
parent 4efbe3ddc7
commit 12bcfe610e
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
umbral/pre.py
View File

@ -63,7 +63,7 @@ class Capsule(object):
self._attached_cfrags = list()
@classmethod
def get_size(cls, curve: ec.EllipticCurve=None):
def get_size(cls, curve: ec.EllipticCurve=None, activated=False):
"""
Returns the size (in bytes) of a Capsule given the curve.
If no curve is provided, it will use the default curve.
@ -72,7 +72,10 @@ class Capsule(object):
bn_size = CurveBN.get_size(curve)
point_size = Point.get_size(curve)
return (bn_size * 1) + (point_size * 2)
if not activated:
return (bn_size * 1) + (point_size * 2)
else:
return (bn_size * 1) + (point_size * 5)
class NotValid(ValueError):
"""