class Crypto(object):
def __init__(self):
self.G = generator_secp256k1
self._r = self.G.order()
def generate_key_pair(self):
self.private_key = ecdsa.util.randrange(pow(2, 256)) % self._r
self.eck = EC_KEY(number_to_string(self.private_key, self._r))
self.public_key = point_to_ser(self.private_key * self.G, True)
def export_public_key(self):
"""
serialization of public key
"""
return bytes.hex(self.public_key)
def encrypt(self, message, pubkey):
res = self.eck.encrypt_message(message.encode('utf-8'),
bytes.fromhex(pubkey))
return res.decode('utf-8')
def decrypt(self, message):
return self.eck.decrypt_message(message)
def hash(self, text, algorithm='sha224'):
h = hashlib.new(algorithm)
h.update(text.encode('utf-8'))
return h.digest()
class Crypto:
""" Functions related to cryptography """
def __init__(self):
self.G = generator_secp256k1
self._r = self.G.order()
self.private_key, self.eck, self.public_key = None, None, None
def generate_key_pair(self):
""" generate encryption/decryption pair """
self.private_key = ecdsa.util.randrange(self._r)
self.eck = EC_KEY(number_to_string(self.private_key, self._r))
self.public_key = point_to_ser(self.private_key * self.G, True)
def export_private_key(self):
""" Export private key as hex string """
if self.private_key:
return bytes.hex(number_to_string(self.private_key, self._r))
else:
return None
def restore_from_privkey(self, secret_string):
"restore key pair from private key expressed in a hex form"
self.private_key = string_to_number(bytes.fromhex(secret_string))
self.eck = EC_KEY(bytes.fromhex(secret_string))
self.public_key = point_to_ser(self.private_key * self.G, True)
def export_public_key(self):
""" serialization of public key """
return bytes.hex(self.public_key)
def encrypt(self, message, pubkey):
""" encrypt message with pubkey """
try:
res = self.eck.encrypt_message(message.encode('utf-8'),
bytes.fromhex(pubkey))
return res.decode('utf-8')
except Exception as e: # grrr.. bitcoin.py raises 'Exception' :/
raise EncryptError(
"Bitcoin.py raised '{}' during Crypto.encrypt".format(
type(e).__name__), e, pubkey, message) from e
def decrypt(self, message):
""" decrypt message """
try:
return self.eck.decrypt_message(message)
except (InvalidPassword, Exception) as e:
raise DecryptError(
"Bitcoin.py raised '{}' during Crypto.decrypt".format(
type(e).__name__), e, self.private_key, message) from e
@staticmethod
def hash(text):
''' Returns sha256(sha256(text)) as bytes. text may be bytes or str. '''
return Hash(text) # bitcoin.Hash is sha256(sha256(x))
class Crypto(object):
"""
This class used for tasks related to cryptography
"""
def __init__(self):
self.G = generator_secp256k1
self._r = self.G.order()
def generate_key_pair(self):
"generate encryption/decryption pair"
self.private_key = ecdsa.util.randrange(self._r)
self.eck = EC_KEY(number_to_string(self.private_key, self._r))
self.public_key = point_to_ser(self.private_key * self.G, True)
def export_private_key(self):
"Export private key as hex string"
if self.private_key:
return bytes.hex(number_to_string(self.private_key, self._r))
else:
return None
def restore_from_privkey(self, secret_string):
"restore key pair from private key expressed in a hex form"
self.private_key = string_to_number(bytes.fromhex(secret_string))
self.eck = EC_KEY(bytes.fromhex(secret_string))
self.public_key = point_to_ser(self.private_key * self.G, True)
def export_public_key(self):
"""
serialization of public key
"""
return bytes.hex(self.public_key)
def encrypt(self, message, pubkey):
"encrypt message with pubkey"
res = self.eck.encrypt_message(message.encode('utf-8'),
bytes.fromhex(pubkey))
return res.decode('utf-8')
def decrypt(self, message):
"decrypt message"
return self.eck.decrypt_message(message)
def hash(self, text, algorithm='sha224'):
"method for hashing the text"
h = hashlib.new(algorithm)
h.update(text.encode('utf-8'))
return h.digest()
class Crypto_cheater(Crypto):
"""
This class is faking the Crypto. It needs for cheating on encryption decryption phase
"""
def generate_fake_key_pair(self):
self.fake_private_key = ecdsa.util.randrange( pow(2,256) ) %self._r
self.fake_eck = EC_KEY(number_to_string(self.fake_private_key, self._r))
self.fake_public_key = point_to_ser(self.fake_private_key*self.G,True)
def export_fake_public_key(self):
return bytes.hex(self.fake_public_key)
def decrypt(self, message):
try:
return self.eck.decrypt_message(message)
except InvalidPassword:
return self.fake_eck.decrypt_message(message)
