1
0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-12-23 06:48:16 +00:00
trezor-firmware/trezorlib/transport/bridge.py
matejcik 54f1599a5a regenerate license headers
This clarifies the intent: the project is licenced under terms
of LGPL version 3 only, but the standard headers cover only "3 or later",
so we had to rewrite them.

In the same step, we removed author information from individual files
in favor of "SatoshiLabs and contributors", and include an AUTHORS
file that lists the contributors.

Apologies to those whose names are missing; please contact us if you wish
to add your info to the AUTHORS file.
2018-06-21 16:49:13 +02:00

109 lines
3.7 KiB
Python

# This file is part of the Trezor project.
#
# Copyright (C) 2012-2018 SatoshiLabs and contributors
#
# This library is free software: you can redistribute it and/or modify
# it under the terms of the GNU Lesser General Public License version 3
# as published by the Free Software Foundation.
#
# 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 License along with this library.
# If not, see <https://www.gnu.org/licenses/lgpl-3.0.html>.
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