# This file is part of the TREZOR project. # # Copyright (C) 2012-2016 Marek Palatinus # Copyright (C) 2012-2016 Pavol Rusnak # Copyright (C) 2016 Jochen Hoenicke # # This library is free software: you can redistribute it and/or modify # it under the terms of the GNU Lesser General Public License as published by # the Free Software Foundation, either version 3 of the License, or # (at your option) any later version. # # This library is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # GNU Lesser General Public License for more details. # # You should have received a copy of the GNU Lesser General Public License # along with this library. If not, see . import logging import requests import binascii from io import BytesIO import struct from .. import mapping from .. import messages from .. import protobuf from . import Transport, TransportException LOG = logging.getLogger(__name__) TREZORD_HOST = 'http://127.0.0.1:21325' def get_error(resp): return ' (error=%d str=%s)' % (resp.status_code, resp.json()['error']) class BridgeTransport(Transport): ''' BridgeTransport implements transport through TREZOR Bridge (aka trezord). ''' PATH_PREFIX = 'bridge' HEADERS = {'Origin': 'https://python.trezor.io'} def __init__(self, device): super().__init__() self.device = device self.conn = requests.Session() self.session = None self.response = None def get_path(self): return '%s:%s' % (self.PATH_PREFIX, self.device['path']) @classmethod def enumerate(cls): try: r = requests.post(TREZORD_HOST + '/enumerate', headers=cls.HEADERS) if r.status_code != 200: raise TransportException('trezord: Could not enumerate devices' + get_error(r)) return [BridgeTransport(dev) for dev in r.json()] except: return [] def open(self): r = self.conn.post(TREZORD_HOST + '/acquire/%s/null' % self.device['path'], headers=self.HEADERS) if r.status_code != 200: raise TransportException('trezord: Could not acquire session' + get_error(r)) self.session = r.json()['session'] def close(self): if not self.session: return r = self.conn.post(TREZORD_HOST + '/release/%s' % self.session, headers=self.HEADERS) if r.status_code != 200: raise TransportException('trezord: Could not release session' + get_error(r)) self.session = None def write(self, msg): LOG.debug("sending message: {}".format(msg.__class__.__name__), extra={'protobuf': msg}) data = BytesIO() protobuf.dump_message(data, msg) ser = data.getvalue() header = struct.pack(">HL", mapping.get_type(msg), len(ser)) data = binascii.hexlify(header + ser).decode() r = self.conn.post( TREZORD_HOST + '/call/%s' % self.session, data=data, headers=self.HEADERS) if r.status_code != 200: raise TransportException('trezord: Could not write message' + get_error(r)) self.response = r.text def read(self): if self.response is None: raise TransportException('No response stored') data = binascii.unhexlify(self.response) headerlen = struct.calcsize('>HL') (msg_type, datalen) = struct.unpack('>HL', data[:headerlen]) data = BytesIO(data[headerlen:headerlen + datalen]) msg = protobuf.load_message(data, mapping.get_class(msg_type)) LOG.debug("received message: {}".format(msg.__class__.__name__), extra={'protobuf': msg}) self.response = None return msg TRANSPORT = BridgeTransport