arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-11-18 05:28:40 +00:00
Code Issues Releases Wiki Activity

python/debuglink: add debuglink message logging

Browse Source
This commit is contained in:
matejcik 2020-04-24 12:23:53 +02:00 committed by matejcik
parent 6069173252
commit ee32806bb2
1 changed files with 27 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
python/src/trezorlib/debuglink.py
View File

@ -14,6 +14,7 @@
# You should have received a copy of the License along with this library. # 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>. # If not, see <https://www.gnu.org/licenses/lgpl-3.0.html>.
import logging
from collections import namedtuple from collections import namedtuple
from copy import deepcopy from copy import deepcopy
@ -21,13 +22,15 @@ from mnemonic import Mnemonic
from . import mapping, messages as proto, protobuf from . import mapping, messages as proto, protobuf
from .client import TrezorClient from .client import TrezorClient
from .log import DUMP_BYTES
from .tools import expect from .tools import expect
EXPECTED_RESPONSES_CONTEXT_LINES = 3 EXPECTED_RESPONSES_CONTEXT_LINES = 3
LayoutLines = namedtuple("LayoutLines", "lines text") LayoutLines = namedtuple("LayoutLines", "lines text")
LOG = logging.getLogger(__name__)
def layout_lines(lines): def layout_lines(lines):
return LayoutLines(lines, " ".join(lines)) return LayoutLines(lines, " ".join(lines))
@ -45,12 +48,34 @@ class DebugLink:
self.transport.end_session() self.transport.end_session()
def _call(self, msg, nowait=False): def _call(self, msg, nowait=False):
LOG.debug(
"sending message: {}".format(msg.__class__.__name__),
extra={"protobuf": msg},
)
msg_type, msg_bytes = mapping.encode(msg) msg_type, msg_bytes = mapping.encode(msg)
LOG.log(
DUMP_BYTES,
"encoded as type {} ({} bytes): {}".format(
msg_type, len(msg_bytes), msg_bytes.hex()
),
)
self.transport.write(msg_type, msg_bytes) self.transport.write(msg_type, msg_bytes)
if nowait: if nowait:
return None return None
ret_type, ret_bytes = self.transport.read() ret_type, ret_bytes = self.transport.read()
return mapping.decode(ret_type, ret_bytes) LOG.log(
DUMP_BYTES,
"received type {} ({} bytes): {}".format(
msg_type, len(msg_bytes), msg_bytes.hex()
),
)
msg = mapping.decode(ret_type, ret_bytes)
LOG.debug(
"received message: {}".format(msg.__class__.__name__),
extra={"protobuf": msg},
)
return msg
def state(self): def state(self):
return self._call(proto.DebugLinkGetState()) return self._call(proto.DebugLinkGetState())