arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-12-18 20:38:10 +00:00
Code Issues Releases Wiki Activity

src/trezor/wire: clean up after workflows

Browse Source
This commit is contained in:
Jan Pochyla 2018-04-03 20:38:36 +02:00
parent b38d6870eb
commit 04680f4bbf
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/trezor/wire/__init__.py
View File

@ -3,6 +3,7 @@ import protobuf
from trezor import log from trezor import log
from trezor import loop from trezor import loop
from trezor import messages from trezor import messages
from trezor import utils
from trezor import workflow from trezor import workflow
from . import codec_v1 from . import codec_v1
@ -119,12 +120,14 @@ async def session_handler(iface, sid):
except KeyError: except KeyError:
handler, args = unexpected_msg, () handler, args = unexpected_msg, ()
m = utils.unimport_begin()
w = handler(ctx, reader, *args) w = handler(ctx, reader, *args)
try: try:
workflow.onstart(w) workflow.onstart(w)
await w await w
finally: finally:
workflow.onclose(w) workflow.onclose(w)
utils.unimport_end(m)
except UnexpectedMessageError as exc: except UnexpectedMessageError as exc:
# retry with opened reader from the exception # retry with opened reader from the exception