arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2025-01-11 16:00:57 +00:00
Code Issues Releases Wiki Activity

python: ensure client session counter is never negative

Browse Source
This commit is contained in:
matejcik 2019-10-18 13:31:13 +02:00
parent 3f9b0b7f4f
commit 598e828844
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
python/src/trezorlib/client.py
View File

@ -132,9 +132,9 @@ class TrezorClient:
self.session_counter += 1 self.session_counter += 1
def close(self): def close(self):
if self.session_counter == 1: self.session_counter = max(self.session_counter - 1, 0)
if self.session_counter == 0:
self.transport.end_session() self.transport.end_session()
self.session_counter -= 1
def cancel(self): def cancel(self):
self._raw_write(messages.Cancel()) self._raw_write(messages.Cancel())

4
python/src/trezorlib/transport/protocol.py
View File

@ -93,9 +93,9 @@ class Protocol:
self.session_counter += 1 self.session_counter += 1
def end_session(self) -> None: def end_session(self) -> None:
if self.session_counter == 1: self.session_counter = max(self.session_counter - 1, 0)
if self.session_counter == 0:
self.handle.close() self.handle.close()
self.session_counter -= 1
def read(self) -> protobuf.MessageType: def read(self) -> protobuf.MessageType:
raise NotImplementedError raise NotImplementedError