From fe10d3d47c7eeb101fdddea9ba4a2cb145338783 Mon Sep 17 00:00:00 2001 From: Kris Nuttycombe Date: Fri, 10 Dec 2021 12:25:45 -0700 Subject: [PATCH] Ensure that pubkey has the correct length and structure. Co-authored-by: Daira Hopwood --- unified_full_viewing_keys.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/unified_full_viewing_keys.py b/unified_full_viewing_keys.py index 2eca5e0..baa81f9 100755 --- a/unified_full_viewing_keys.py +++ b/unified_full_viewing_keys.py @@ -33,6 +33,8 @@ def main(): privkey = ec.derive_private_key(int.from_bytes(rand.b(32), 'little'), ec.SECP256K1()) pubkey = privkey.public_key() pubkey_bytes = pubkey.public_bytes(Encoding.X962, PublicFormat.CompressedPoint) + assert len(pubkey_bytes) == 33 + assert pubkey_bytes[0] in (0x02, 0x03) t_key_bytes = c + pubkey_bytes else: t_key_bytes = None