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

feat(python/trezorctl): notify user about PIN/passphrase entry in terminal (fixes #3203)

Browse Source
This commit is contained in:
matejcik 2023-10-19 13:27:50 +02:00
parent c9e9c34e06
commit 3c0344304f
2 changed files with 15 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
python/.changelog.d/3203.changed Normal file
View File

@ -0,0 +1 @@
ClickUI: notify user in terminal that they should enter PIN or passphrase on Trezor.

19
python/src/trezorlib/ui.py
View File

@ -90,15 +90,24 @@ class ClickUI:
self, always_prompt: bool = False, passphrase_on_host: bool = False
) -> None:
self.pinmatrix_shown = False
self.prompt_shown = False
self.last_prompt_shown = ""
self.always_prompt = always_prompt
self.passphrase_on_host = passphrase_on_host
def button_request(self, _br: messages.ButtonRequest) -> None:
if not self.prompt_shown:
echo("Please confirm action on your Trezor device.")
def _prompt_for_button(self, br: messages.ButtonRequest) -> str:
if br.code == messages.ButtonRequestType.PassphraseEntry:
return "Please enter passphrase on your Trezor device."
if br.code == messages.ButtonRequestType.PinEntry:
return "Please enter PIN on your Trezor device."
return "Please confirm action on your Trezor device."
def button_request(self, br: messages.ButtonRequest) -> None:
prompt = self._prompt_for_button(br)
if prompt != self.last_prompt_shown:
echo(prompt)
if not self.always_prompt:
self.prompt_shown = True
self.last_prompt_shown = prompt
def get_pin(self, code: Optional[PinMatrixRequestType] = None) -> str:
if code == PIN_CURRENT: