BTCPrivate
/
trezor-core
mirror of https://github.com/BTCPrivate/trezor-core.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

src/trezor/wire: make sure the writes succeed

This commit is contained in:
Jan Pochyla 2018-02-28 17:17:02 +01:00
parent 8f1ed5edce
commit 8ac915e75f
1 changed files with 12 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
src/trezor/wire/codec_v1.py
View File

@ -1,7 +1,7 @@
from micropython import const
import ustruct
from trezor import io
from trezor import io, ui
from trezor import loop
from trezor import utils
@ -134,8 +134,11 @@ class Writer:
if self.ofs == _REP_LEN:
# we are at the end of the report, flush it
await write
self.iface.write(self.data)
while True:
await write
n = self.iface.write(self.data)
if n == len(self.data):
break
self.ofs = _REP_CONT_DATA
return nwritten
@ -149,5 +152,9 @@ class Writer:
self.data[self.ofs] = 0x00
self.ofs += 1
await loop.select(self.iface.iface_num() | io.POLL_WRITE)
self.iface.write(self.data)
write = loop.select(self.iface.iface_num() | io.POLL_WRITE)
while True:
await write
n = self.iface.write(self.data)
if n == len(self.data):
break