mirror of
https://github.com/status-im/eth2.0-specs.git
synced 2025-01-23 09:01:17 +00:00
54 lines
1.6 KiB
Python
54 lines
1.6 KiB
Python
from py_ecc import bls
|
|
|
|
# Flag to make BLS active or not. Used for testing, do not ignore BLS in production unless you know what you are doing.
|
|
bls_active = True
|
|
|
|
STUB_SIGNATURE = b'\x11' * 96
|
|
STUB_PUBKEY = b'\x22' * 48
|
|
STUB_COORDINATES = bls.api.signature_to_G2(bls.sign(b"", 0, b"\0" * 8))
|
|
|
|
|
|
def only_with_bls(alt_return=None):
|
|
"""
|
|
Decorator factory to make a function only run when BLS is active. Otherwise return the default.
|
|
"""
|
|
def runner(fn):
|
|
def entry(*args, **kw):
|
|
if bls_active:
|
|
return fn(*args, **kw)
|
|
else:
|
|
return alt_return
|
|
return entry
|
|
return runner
|
|
|
|
|
|
@only_with_bls(alt_return=True)
|
|
def Verify(PK, message, signature):
|
|
return bls.verify(message_hash=message, pubkey=PK, signature=signature, domain=b'')
|
|
|
|
|
|
@only_with_bls(alt_return=True)
|
|
def AggregateVerify(PKs, messages, signature):
|
|
return bls.verify_multiple(pubkeys=PKs, message_hashes=messages, signature=signature, domain=b'')
|
|
|
|
|
|
@only_with_bls(alt_return=True)
|
|
def FastAggregateVerify(PKs, message, signature):
|
|
aggregate_pubkey = bls.aggregate_pubkeys(PKs)
|
|
return bls.verify(pubkey=aggregate_pubkey, message_hash=message, signature=signature, domain=b'')
|
|
|
|
|
|
@only_with_bls(alt_return=STUB_SIGNATURE)
|
|
def Aggregate(signatures):
|
|
return bls.aggregate_signatures(signatures)
|
|
|
|
|
|
@only_with_bls(alt_return=STUB_SIGNATURE)
|
|
def Sign(SK, message):
|
|
return bls.sign(message_hash=message, privkey=SK, domain=b'')
|
|
|
|
|
|
@only_with_bls(alt_return=STUB_COORDINATES)
|
|
def bls_signature_to_G2(signature):
|
|
return bls.api.signature_to_G2(signature)
|