arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2025-05-09 18:38:47 +00:00
Code Issues Releases Wiki Activity

apps/fido_u2f: fix worfklow api, optimize

Browse Source
This commit is contained in:
Jan Pochyla 2017-09-06 15:43:37 +02:00
parent 5b84db0f7f
commit 26b5baf657
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/apps/fido_u2f/__init__.py
View File

@ -229,8 +229,9 @@ class Cmd:
async def read_cmd(iface: io.HID) -> Cmd: async def read_cmd(iface: io.HID) -> Cmd:
desc_init = frame_init() desc_init = frame_init()
desc_cont = frame_cont() desc_cont = frame_cont()
read = loop.select(iface.iface_num())
buf, = await loop.select(iface.iface_num()) buf = await read
# log.debug(__name__, 'read init %s', buf) # log.debug(__name__, 'read init %s', buf)
ifrm = overlay_struct(buf, desc_init) ifrm = overlay_struct(buf, desc_init)
@ -252,7 +253,7 @@ async def read_cmd(iface: io.HID) -> Cmd:
data = data[:bcnt] data = data[:bcnt]
while datalen < bcnt: while datalen < bcnt:
buf, = await loop.select(iface.iface_num()) buf = await read
# log.debug(__name__, 'read cont %s', buf) # log.debug(__name__, 'read cont %s', buf)
cfrm = overlay_struct(buf, desc_cont) cfrm = overlay_struct(buf, desc_cont)
@ -444,10 +445,12 @@ class ConfirmState:
def fork(self) -> None: def fork(self) -> None:
self.deadline_ms = utime.ticks_ms() + _CONFIRM_STATE_TIMEOUT_MS self.deadline_ms = utime.ticks_ms() + _CONFIRM_STATE_TIMEOUT_MS
self.task = self.confirm() self.task = self.confirm()
workflow.start(self.task) workflow.onstart(self.task)
loop.schedule_task(self.task)
def kill(self) -> None: def kill(self) -> None:
if self.task is not None: if self.task is not None:
workflow.onclose(self.task)
self.task.close() self.task.close()
async def confirm(self) -> None: async def confirm(self) -> None: