def aggregate_priv_keys(private_keys, public_keys, secure):
"""
Aggregates private keys together
"""
if not secure:
sum_keys = sum(pk.value for pk in private_keys) % default_ec.n
else:
if not public_keys:
raise Exception(
"Must include public keys in secure aggregation")
if len(private_keys) != len(public_keys):
raise Exception("Invalid number of keys")
priv_pub_keys = zip(public_keys, private_keys)
priv_pub_keys.sort()
computed_Ts = hash_pks(len(private_keys), public_keys)
n = public_keys[0].value.ec.n
sum_keys = 0
for i, (_, privkey) in enumerate(priv_pub_keys):
sum_keys += privkey.value * computed_Ts[i]
sum_keys %= n
return PrivateKey.from_bytes(sum_keys.to_bytes(32, "big"))
def aggregate_priv_keys(private_keys, public_keys, secure):
"""
Aggregates private keys together
"""
if secure and len(private_keys) != len(public_keys):
raise Exception("Invalid number of keys")
priv_pub_keys = [(public_keys[i], private_keys[i])
for i in range(len(private_keys))]
# Sort by public keys
priv_pub_keys.sort()
computed_Ts = hash_pks(len(private_keys), public_keys)
n = public_keys[0].value.ec.n
sum_keys = 0
for i in range(len(priv_pub_keys)):
addend = priv_pub_keys[i][1].value
if (secure):
addend *= computed_Ts[i]
sum_keys = (sum_keys + addend) % n
return PrivateKey.from_bytes(sum_keys.to_bytes(32, "big"))
def test1():
seed = bytes([
0, 50, 6, 244, 24, 199, 1, 25, 52, 88, 192, 19, 18, 12, 89, 6, 220, 18,
102, 58, 209, 82, 12, 62, 89, 110, 182, 9, 44, 20, 254, 22
])
sk = PrivateKey.from_seed(seed)
pk = sk.get_public_key()
msg = bytes([100, 2, 254, 88, 90, 45, 23])
sig = sk.sign(msg)
sk_bytes = sk.serialize()
pk_bytes = pk.serialize()
sig_bytes = sig.serialize()
sk = PrivateKey.from_bytes(sk_bytes)
pk = PublicKey.from_bytes(pk_bytes)
sig = Signature.from_bytes(sig_bytes)
sig.set_aggregation_info(AggregationInfo.from_msg(pk, msg))
ok = sig.verify()
assert (ok)
seed = bytes([1]) + seed[1:]
sk1 = PrivateKey.from_seed(seed)
seed = bytes([2]) + seed[1:]
sk2 = PrivateKey.from_seed(seed)
pk1 = sk1.get_public_key()
sig1 = sk1.sign(msg)
pk2 = sk2.get_public_key()
sig2 = sk2.sign(msg)
agg_sig = Signature.aggregate([sig1, sig2])
agg_pubkey = PublicKey.aggregate([pk1, pk2])
agg_sig.set_aggregation_info(AggregationInfo.from_msg(agg_pubkey, msg))
assert (agg_sig.verify())
seed = bytes([3]) + seed[1:]
sk3 = PrivateKey.from_seed(seed)
pk3 = sk3.get_public_key()
msg2 = bytes([100, 2, 254, 88, 90, 45, 23])
sig1 = sk1.sign(msg)
sig2 = sk2.sign(msg)
sig3 = sk3.sign(msg2)
agg_sig_l = Signature.aggregate([sig1, sig2])
agg_sig_final = Signature.aggregate([agg_sig_l, sig3])
sig_bytes = agg_sig_final.serialize()
agg_sig_final = Signature.from_bytes(sig_bytes)
a1 = AggregationInfo.from_msg(pk1, msg)
a2 = AggregationInfo.from_msg(pk2, msg)
a3 = AggregationInfo.from_msg(pk3, msg2)
a1a2 = AggregationInfo.merge_infos([a1, a2])
a_final = AggregationInfo.merge_infos([a1a2, a3])
print(a_final)
agg_sig_final.set_aggregation_info(a_final)
ok = agg_sig_final.verify()
ok = agg_sig_l.verify()
agg_sig_final = agg_sig_final.divide_by([agg_sig_l])
ok = agg_sig_final.verify()
agg_sk = PrivateKey.aggregate([sk1, sk2], [pk1, pk2])
agg_sk.sign(msg)
seed = bytes([
1, 50, 6, 244, 24, 199, 1, 25, 52, 88, 192, 19, 18, 12, 89, 6, 220, 18,
102, 58, 209, 82, 12, 62, 89, 110, 182, 9, 44, 20, 254, 22
])
esk = ExtendedPrivateKey.from_seed(seed)
epk = esk.get_extended_public_key()
sk_child = esk.private_child(0).private_child(5)
pk_child = epk.public_child(0).public_child(5)
buffer1 = pk_child.serialize()
buffer2 = sk_child.serialize()
print(len(buffer1), buffer1)
print(len(buffer2), buffer2)
assert (sk_child.get_extended_public_key() == pk_child)
