117 lines
3.9 KiB
Python
117 lines
3.9 KiB
Python
|
# standard imports
|
|||
|
|
import logging
|
|||
|
|
import re
|
|||
|
|
import socket
|
|||
|
|
import uuid
|
|||
|
|
import json
|
|||
|
|
|
|||
|
|
logg = logging.getLogger(__file__)
|
|||
|
|
|
|||
|
|
|
|||
|
|
def jsonrpc_request(method, params):
|
|||
|
|
uu = uuid.uuid4()
|
|||
|
|
return {
|
|||
|
|
"jsonrpc": "2.0",
|
|||
|
|
"id": str(uu),
|
|||
|
|
"method": method,
|
|||
|
|
"params": params,
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
class PlatformMiddleware:
|
|||
|
|
|
|||
|
|
# id for the request is not available, meaning we cannot easily short-circuit
|
|||
|
|
# hack workaround
|
|||
|
|
id_seq = -1
|
|||
|
|
re_personal = re.compile('^personal_.*')
|
|||
|
|
ipcaddr = None
|
|||
|
|
|
|||
|
|
|
|||
|
|
def __init__(self, make_request, w3):
|
|||
|
|
self.w3 = w3
|
|||
|
|
self.make_request = make_request
|
|||
|
|
if self.ipcaddr == None:
|
|||
|
|
raise AttributeError('ipcaddr not set')
|
|||
|
|
|
|||
|
|
|
|||
|
|
# TODO: understand what format input params come in
|
|||
|
|
# single entry input gives a tuple on params, wtf...
|
|||
|
|
# dict input comes as [{}] and fails if not passed on as an array
|
|||
|
|
@staticmethod
|
|||
|
|
def _translate_params(params):
|
|||
|
|
#if params.__class__.__name__ == 'tuple':
|
|||
|
|
# r = []
|
|||
|
|
# for p in params:
|
|||
|
|
# r.append(p)
|
|||
|
|
# return r
|
|||
|
|
|
|||
|
|
if params.__class__.__name__ == 'list' and len(params) > 0:
|
|||
|
|
return params[0]
|
|||
|
|
|
|||
|
|
return params
|
|||
|
|
|
|||
|
|
|
|||
|
|
# TODO: DRY
|
|||
|
|
def __call__(self, method, suspect_params):
|
|||
|
|
|
|||
|
|
self.id_seq += 1
|
|||
|
|
logg.debug('in middleware method {} params {} ipcpath {}'.format(method, suspect_params, self.ipcaddr))
|
|||
|
|
|
|||
|
|
if self.re_personal.match(method) != None:
|
|||
|
|
params = PlatformMiddleware._translate_params(suspect_params)
|
|||
|
|
# multiple providers is removed in web3.py 5.12.0
|
|||
|
|
# https://github.com/ethereum/web3.py/issues/1701
|
|||
|
|
# thus we need a workaround to use the same web3 instance
|
|||
|
|
s = socket.socket(family=socket.AF_UNIX, type=socket.SOCK_STREAM, proto=0)
|
|||
|
|
ipc_provider_workaround = s.connect(self.ipcaddr)
|
|||
|
|
|
|||
|
|
logg.info('redirecting method {} params {} original params {}'.format(method, params, suspect_params))
|
|||
|
|
o = jsonrpc_request(method, params[0])
|
|||
|
|
j = json.dumps(o)
|
|||
|
|
logg.debug('send {}'.format(j))
|
|||
|
|
s.send(j.encode('utf-8'))
|
|||
|
|
r = s.recv(4096)
|
|||
|
|
s.close()
|
|||
|
|
logg.debug('got recv {}'.format(str(r)))
|
|||
|
|
jr = json.loads(r)
|
|||
|
|
jr['id'] = self.id_seq
|
|||
|
|
#return str(json.dumps(jr))
|
|||
|
|
return jr
|
|||
|
|
|
|||
|
|
elif method == 'eth_signTransaction':
|
|||
|
|
params = PlatformMiddleware._translate_params(suspect_params)
|
|||
|
|
s = socket.socket(family=socket.AF_UNIX, type=socket.SOCK_STREAM, proto=0)
|
|||
|
|
ipc_provider_workaround = s.connect(self.ipcaddr)
|
|||
|
|
logg.info('redirecting method {} params {} original params {}'.format(method, params, suspect_params))
|
|||
|
|
o = jsonrpc_request(method, params[0])
|
|||
|
|
j = json.dumps(o)
|
|||
|
|
logg.debug('send {}'.format(j))
|
|||
|
|
s.send(j.encode('utf-8'))
|
|||
|
|
r = s.recv(4096)
|
|||
|
|
s.close()
|
|||
|
|
logg.debug('got recv {}'.format(str(r)))
|
|||
|
|
jr = json.loads(r)
|
|||
|
|
jr['id'] = self.id_seq
|
|||
|
|
#return str(json.dumps(jr))
|
|||
|
|
return jr
|
|||
|
|
|
|||
|
|
elif method == 'eth_sign':
|
|||
|
|
params = PlatformMiddleware._translate_params(suspect_params)
|
|||
|
|
s = socket.socket(family=socket.AF_UNIX, type=socket.SOCK_STREAM, proto=0)
|
|||
|
|
ipc_provider_workaround = s.connect(self.ipcaddr)
|
|||
|
|
logg.info('redirecting method {} params {} original params {}'.format(method, params, suspect_params))
|
|||
|
|
o = jsonrpc_request(method, params)
|
|||
|
|
j = json.dumps(o)
|
|||
|
|
logg.debug('send {}'.format(j))
|
|||
|
|
s.send(j.encode('utf-8'))
|
|||
|
|
r = s.recv(4096)
|
|||
|
|
s.close()
|
|||
|
|
logg.debug('got recv {}'.format(str(r)))
|
|||
|
|
jr = json.loads(r)
|
|||
|
|
jr['id'] = self.id_seq
|
|||
|
|
return jr
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
r = self.make_request(method, suspect_params)
|
|||
|
|
return r
|