Compressed key support

Test vectors taken from
<http://sourceforge.net/mailarchive/message.php?msg_id=28648286>.

Author: yrro

addrgen.py

@@ -6,15 +6,27 @@
 import base58
 import ctssl
 
-def wif (priv, testnet=False):
+def wif (priv, compressed, testnet=False):
         '''
         Convert raw private key to base58-encoded wallet import format.
 
-        >>> wif (binascii.unhexlify (b'338b7f9c13b44747fdef077898f688411693df40d5f6943ebba30917125934c9'))
-        b'5JCzE8aEchKzGQThaYqKn5bcHvisWThwC2tU3eUB6VVAx1WV5fQ'
+        >>> wif (binascii.unhexlify (b'1111111111111111111111111111111111111111111111111111111111111111'), False)
+        b'5HwoXVkHoRM8sL2KmNRS217n1g8mPPBomrY7yehCuXC1115WWsh'
+        >>> wif (binascii.unhexlify (b'1111111111111111111111111111111111111111111111111111111111111111'), True)
+        b'KwntMbt59tTsj8xqpqYqRRWufyjGunvhSyeMo3NTYpFYzZbXJ5Hp'
+        >>> wif (binascii.unhexlify (b'dddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddd'), False)
+        b'5KVzsHJiUxgvBBgtVS7qBTbbYZpwWM4WQNCCyNSiuFCJzYMxg8H'
+        >>> wif (binascii.unhexlify (b'dddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddd'), True)
+        b'L4ezQvyC6QoBhxB4GVs9fAPhUKtbaXYUn8YTqoeXwbevQq4U92vN'
+        >>> wif (binascii.unhexlify (b'47f7616ea6f9b923076625b4488115de1ef1187f760e65f89eb6f4f7ff04b012'), False)
+        b'5JMys7YfK72cRVTrbwkq5paxU7vgkMypB55KyXEtN5uSnjV7K8Y'
+        >>> wif (binascii.unhexlify (b'47f7616ea6f9b923076625b4488115de1ef1187f760e65f89eb6f4f7ff04b012'), True)
+        b'KydbzBtk6uc7M6dXwEgTEH2sphZxSPbmDSz6kUUHi4eUpSQuhEbq'
         '''
         result = (b'\x80' if not testnet else b'\xef') + (b'\x00' * (32 - len (priv)) + priv)
         assert len (result) == 33
+        if compressed:
+                result += b'\x01'
         h1 = hashlib.sha256 (result)
         h2 = hashlib.sha256 (h1.digest ())
         result += h2.digest ()[:4]
@@ -24,11 +36,25 @@ def addr (pub, testnet=False):
         '''
         Convert raw public key to base58check-encoded address.
 
-        >>> addr (binascii.unhexlify (b'04bbed292cf660fcd6fd29590ea53bbad38603a8b55d93806d12af56c994bae8d1df64d4b52607543c44031b26c401648928f748dd447c736b3c47f61f38477c28'))
-        b'1Br8AVNn7XtvEBHc5hyaXsnPZFVAbLpNiL'
+        >>> addr (binascii.unhexlify (b'044f355bdcb7cc0af728ef3cceb9615d90684bb5b2ca5f859ab0f0b704075871aa385b6b1b8ead809ca67454d9683fcf2ba03456d6fe2c4abe2b07f0fbdbb2f1c1'))
+        b'1MsHWS1BnwMc3tLE8G35UXsS58fKipzB7a'
+        >>> addr (binascii.unhexlify (b'034f355bdcb7cc0af728ef3cceb9615d90684bb5b2ca5f859ab0f0b704075871aa'))
+        b'1Q1pE5vPGEEMqRcVRMbtBK842Y6Pzo6nK9'
+        >>> addr (binascii.unhexlify (b'04ed83704c95d829046f1ac27806211132102c34e9ac7ffa1b71110658e5b9d1bdedc416f5cefc1db0625cd0c75de8192d2b592d7e3b00bcfb4a0e860d880fd1fc'))
+        b'1JyMKvPHkrCQd8jQrqTR1rBsAd1VpRhTiE'
+        >>> addr (binascii.unhexlify (b'02ed83704c95d829046f1ac27806211132102c34e9ac7ffa1b71110658e5b9d1bd'))
+        b'1NKRhS7iYUGTaAfaR5z8BueAJesqaTyc4a'
+        >>> addr (binascii.unhexlify (b'042596957532fc37e40486b910802ff45eeaa924548c0e1c080ef804e523ec3ed3ed0a9004acf927666eee18b7f5e8ad72ff100a3bb710a577256fd7ec81eb1cb3'))
+        b'1PM35qz2uwCDzcUJtiqDSudAaaLrWRw41L'
+        >>> addr (binascii.unhexlify (b'032596957532fc37e40486b910802ff45eeaa924548c0e1c080ef804e523ec3ed3'))
+        b'19ck9VKC6KjGxR9LJg4DNMRc45qFrJguvV'
         '''
-        assert len (pub) == 65 # pad?
-        assert pub[0] == 4
+        if pub[0] == 4:
+                assert len (pub) == 65
+        elif pub[0] == 2 or pub[0] == 3:
+                assert len (pub) == 33
+        else:
+                assert False, 'Unknown public key format: {}'.format (pub[0])
         h3 = hashlib.sha256 (pub)
         h4 = hashlib.new ('ripemd160', h3.digest ())
         result = (b'\x00' if not testnet else b'\x6f') + h4.digest ()
@@ -38,23 +64,24 @@ def addr (pub, testnet=False):
         return base58.encode (result)
 
 def generate (a):
-        with ctssl.EC_KEY () as key:
+        with ctssl.EC_KEY (compressed=not a.uncompressed) as key:
                 priv = key.priv ()
                 pub = key.pub ()
                 with print_lock:
-                        print (wif (priv, testnet=a.testnet).decode ('ascii'), addr (pub, testnet=a.testnet).decode ('ascii'))
+                        print (wif (priv, key.compressed, testnet=a.testnet).decode ('ascii'), addr (pub, testnet=a.testnet).decode ('ascii'))
                         if a.raw:
                                 print ('', binascii.hexlify (priv).decode ('ascii'), binascii.hexlify (pub).decode ('ascii'))
 
 def main ():
         parser = argparse.ArgumentParser (description = 'Generate Bitcoin addresses.')
-        parser.add_argument ('--count', '-c', help='Number of addresses to generate', action='store', default=1)
+        parser.add_argument ('--number', '-n', help='Number of addresses to generate', action='store', default=1)
+        parser.add_argument ('--uncompressed', '-u', help='Use uncompressed format', action='store_true')
         parser.add_argument ('--raw', '-r', help='Display raw private/public key', action='store_true')
         parser.add_argument ('--testnet', '-t', help='Generate testnet address', action='store_true')
         a = parser.parse_args ()
 
         p = multiprocessing.Pool (processes=multiprocessing.cpu_count ())
-        p.map (generate, [a] * int (a.count))
+        p.map (generate, [a] * int (a.number))
 
 if __name__ == '__main__':
         print_lock = multiprocessing.Lock ()

ctssl/__init__.py

@@ -4,12 +4,14 @@
 import ctssl.detail as detail
 
 class EC_KEY:
-        def __init__ (self):
+        def __init__ (self, compressed=True):
+                self.compressed = compressed
                 pass
 
         def __enter__ (self):
                 self.k = detail.ssl.EC_KEY_new_by_curve_name (detail.NID_secp256k1)
                 detail.ssl.EC_KEY_generate_key (self.k)
+                detail.ssl.EC_KEY_set_conv_form (self.k, detail.POINT_CONVERSION_COMPRESSED if self.compressed else detail.POINT_CONVERSION_UNCOMPRESSED)
                 return self
 
         def __exit__ (self, *exc_info):

ctssl/detail.py

@@ -1,6 +1,9 @@
 import ctypes
 import ctypes.util
 
+POINT_CONVERSION_COMPRESSED = 2
+POINT_CONVERSION_UNCOMPRESSED = 4
+
 NID_secp256k1 = 714
 
 ssl = ctypes.cdll.LoadLibrary (ctypes.util.find_library ('ssl'))