Add personal_newAccount handler for socket server

This commit is contained in:
nolash
2020-08-05 21:30:08 +02:00
parent e71b5ef4ea
commit acaff79ad2
3 changed files with 105 additions and 24 deletions

View File

@@ -9,6 +9,8 @@ from eth_keys import KeyAPI
from eth_keys.backends import NativeECCBackend
import sha3
from common import strip_hex_prefix
keyapi = KeyAPI(NativeECCBackend)
logging.basicConfig(level=logging.DEBUG)
@@ -24,7 +26,7 @@ class ReferenceDatabase:
def __init__(self, dbname, **kwargs):
self.conn = psycopg2.connect('dbname='+dbname)
self.conn = psycopg2.connect('dbname=' + dbname)
self.cur = self.conn.cursor()
self.symmetric_key = kwargs.get('symmetric_key')
@@ -36,12 +38,20 @@ class ReferenceDatabase:
return self._decrypt(k, password)
def new(self, address, password=None):
def new(self, password=None):
b = os.urandom(32)
pk = keyapi.PrivateKey(b)
pubk = keyapi.private_key_to_public_key(pk)
address_hex = pubk.to_checksum_address()
address_hex_clean = strip_hex_prefix(address_hex)
logg.debug('address {}'.format(address_hex_clean))
c = self._encrypt(pk.to_bytes(), password)
s = sql.SQL('INSERT INTO ethereum (wallet_address_hex, key_ciphertext) VALUES (%s, %s)')
self.cur.execute(s, [ address, c.decode('utf-8') ])
self.cur.execute(s, [ address_hex_clean, c.decode('utf-8') ])
self.conn.commit()
return address_hex
def _encrypt(self, private_key, password):
@@ -64,7 +74,8 @@ class ReferenceDatabase:
return f.decrypt(c.encode('utf-8'))
def __exit__(self):
self.conn
def __del__(self):
logg.debug('closing database')
self.conn.commit()
self.cur.close()
self.conn.close()

View File

@@ -27,12 +27,12 @@ class ReferenceSigner(Signer):
super(ReferenceSigner, self).__init__(keyGetter)
def signTransaction(self, tx):
def signTransaction(self, tx, password=None):
s = tx.serialize()
h = sha3.keccak_256()
h.update(s)
g = h.digest()
k = keys.PrivateKey(self.keyGetter(tx.sender))
k = keys.PrivateKey(self.keyGetter(tx.sender, password))
z = keys.ecdsa_sign(message_hash=g, private_key=k)
tx.v = (tx.v * 2) + 35 + z[64]
tx.r = z[:32]