PyBitmessage/src/highlevelcrypto.py

60 lines
2.3 KiB
Python
Raw Normal View History

2013-01-16 17:52:52 +01:00
import pyelliptic
from pyelliptic import arithmetic as a, OpenSSL
2013-01-16 17:52:52 +01:00
def makeCryptor(privkey):
2014-07-15 00:01:56 +02:00
private_key = a.changebase(privkey, 16, 256, minlen=32)
public_key = pointMult(private_key)
privkey_bin = '\x02\xca\x00\x20' + private_key
pubkey_bin = '\x02\xca\x00\x20' + public_key[1:-32] + '\x00\x20' + public_key[-32:]
cryptor = pyelliptic.ECC(curve='secp256k1',privkey=privkey_bin,pubkey=pubkey_bin)
return cryptor
2013-01-16 17:52:52 +01:00
def hexToPubkey(pubkey):
2014-07-15 00:01:56 +02:00
pubkey_raw = a.changebase(pubkey[2:],16,256,minlen=64)
pubkey_bin = '\x02\xca\x00 '+pubkey_raw[:32]+'\x00 '+pubkey_raw[32:]
return pubkey_bin
2013-01-16 17:52:52 +01:00
def makePubCryptor(pubkey):
2014-07-15 00:01:56 +02:00
pubkey_bin = hexToPubkey(pubkey)
return pyelliptic.ECC(curve='secp256k1',pubkey=pubkey_bin)
2013-01-16 17:52:52 +01:00
# Converts hex private key into hex public key
def privToPub(privkey):
2014-07-15 00:01:56 +02:00
private_key = a.changebase(privkey, 16, 256, minlen=32)
public_key = pointMult(private_key)
return public_key.encode('hex')
2013-01-16 17:52:52 +01:00
# Encrypts message with hex public key
def encrypt(msg,hexPubkey):
2014-07-15 00:01:56 +02:00
return pyelliptic.ECC(curve='secp256k1').encrypt(msg,hexToPubkey(hexPubkey))
2013-01-16 17:52:52 +01:00
# Decrypts message with hex private key
def decrypt(msg,hexPrivkey):
2014-07-15 00:01:56 +02:00
return makeCryptor(hexPrivkey).decrypt(msg)
2013-01-18 23:38:09 +01:00
# Decrypts message with an existing pyelliptic.ECC.ECC object
2013-01-16 17:52:52 +01:00
def decryptFast(msg,cryptor):
2014-07-15 00:01:56 +02:00
return cryptor.decrypt(msg)
2013-01-16 17:52:52 +01:00
# Signs with hex private key
def sign(msg,hexPrivkey):
2014-07-15 00:01:56 +02:00
return makeCryptor(hexPrivkey).sign(msg)
2013-01-16 17:52:52 +01:00
# Verifies with hex public key
def verify(msg,sig,hexPubkey):
2014-08-27 09:14:32 +02:00
try:
return makePubCryptor(hexPubkey).verify(sig,msg)
except:
return False
# Does an EC point multiplication; turns a private key into a public key.
def pointMult(secret):
2014-07-15 00:01:56 +02:00
k = OpenSSL.EC_KEY_new_by_curve_name(OpenSSL.get_curve('secp256k1'))
priv_key = OpenSSL.BN_bin2bn(secret, 32, None)
group = OpenSSL.EC_KEY_get0_group(k)
pub_key = OpenSSL.EC_POINT_new(group)
OpenSSL.EC_POINT_mul(group, pub_key, priv_key, None, None, None)
OpenSSL.EC_KEY_set_private_key(k, priv_key)
OpenSSL.EC_KEY_set_public_key(k, pub_key)
size = OpenSSL.i2o_ECPublicKey(k, None)
mb = OpenSSL.create_string_buffer(size)
OpenSSL.i2o_ECPublicKey(k, OpenSSL.byref(OpenSSL.pointer(mb)))
OpenSSL.EC_POINT_free(pub_key)
OpenSSL.BN_free(priv_key)
OpenSSL.EC_KEY_free(k)
return mb.raw