python-trezor/trezorlib/transport_bridge.py

89 lines
2.8 KiB
Python
Raw Normal View History

2014-07-26 14:27:28 +00:00
'''BridgeTransport implements transport TREZOR Bridge (aka trezord).'''
import json
2016-05-26 15:20:44 +00:00
import requests
2016-05-05 01:16:17 +00:00
from . import protobuf_json
from . import messages_pb2 as proto
2016-06-27 16:21:40 +00:00
from .transport import TransportV1
2014-07-26 14:27:28 +00:00
TREZORD_HOST = 'https://localback.net:21324'
CONFIG_URL = 'https://wallet.trezor.io/data/config_signed.bin'
2014-07-26 14:27:28 +00:00
def get_error(resp):
return ' (error=%d str=%s)' % (resp.status_code, resp.json()['error'])
2016-06-27 16:21:40 +00:00
class BridgeTransport(TransportV1):
2014-07-26 14:27:28 +00:00
def __init__(self, device, *args, **kwargs):
self.configure()
2014-07-26 14:27:28 +00:00
2016-06-30 14:47:17 +00:00
self.path = device['path']
self.session = None
self.response = None
2016-05-26 15:20:44 +00:00
self.conn = requests.Session()
super(BridgeTransport, self).__init__(device, *args, **kwargs)
@staticmethod
def configure():
r = requests.get(CONFIG_URL, verify=False)
2014-07-26 14:27:28 +00:00
if r.status_code != 200:
raise Exception('Could not fetch config from %s' % CONFIG_URL)
2014-07-27 14:22:24 +00:00
config = r.text
2014-07-26 14:27:28 +00:00
r = requests.post(TREZORD_HOST + '/configure', data=config)
2014-07-26 14:27:28 +00:00
if r.status_code != 200:
raise Exception('trezord: Could not configure' + get_error(r))
2014-07-26 14:27:28 +00:00
@classmethod
def enumerate(cls):
"""
Return a list of available TREZOR devices.
"""
cls.configure()
r = requests.get(TREZORD_HOST + '/enumerate')
2014-07-26 14:27:28 +00:00
if r.status_code != 200:
raise Exception('trezord: Could not enumerate devices' + get_error(r))
2014-07-26 14:27:28 +00:00
enum = r.json()
2016-02-10 15:46:58 +00:00
return enum
2014-07-26 14:27:28 +00:00
def _open(self):
2015-03-05 10:15:53 +00:00
r = self.conn.post(TREZORD_HOST + '/acquire/%s' % self.path)
2014-07-26 14:27:28 +00:00
if r.status_code != 200:
raise Exception('trezord: Could not acquire session' + get_error(r))
2014-07-26 14:27:28 +00:00
resp = r.json()
self.session = resp['session']
def _close(self):
2015-03-05 10:15:53 +00:00
r = self.conn.post(TREZORD_HOST + '/release/%s' % self.session)
2014-07-26 14:27:28 +00:00
if r.status_code != 200:
raise Exception('trezord: Could not release session' + get_error(r))
2014-07-26 14:27:28 +00:00
else:
self.session = None
2016-06-27 16:21:40 +00:00
def _ready_to_read(self):
2014-07-26 14:27:28 +00:00
return self.response != None
2016-06-27 16:21:40 +00:00
def write(self, protobuf_msg):
# Override main 'write' method, HTTP transport cannot be
# splitted to chunks
2014-07-26 14:27:28 +00:00
cls = protobuf_msg.__class__.__name__
msg = protobuf_json.pb2json(protobuf_msg)
2014-07-27 14:22:24 +00:00
payload = '{"type": "%s", "message": %s}' % (cls, json.dumps(msg))
2015-03-05 10:15:53 +00:00
r = self.conn.post(TREZORD_HOST + '/call/%s' % self.session, data=payload)
2014-07-26 14:27:28 +00:00
if r.status_code != 200:
raise Exception('trezord: Could not write message' + get_error(r))
2014-07-26 14:27:28 +00:00
else:
self.response = r.json()
def _read(self):
2016-05-26 15:20:44 +00:00
if self.response is None:
2014-07-26 14:27:28 +00:00
raise Exception('No response stored')
cls = getattr(proto, self.response['type'])
inst = cls()
pb = protobuf_json.json2pb(inst, self.response['message'])
2016-06-27 16:21:40 +00:00
return (0, 'protobuf', pb)