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

feat(trezorctl): Add firmware get-hash command.

Browse Source
This commit is contained in:
Andrew Kozlik 2022-04-23 00:19:07 +02:00 committed by Martin Milata
parent 106ab65e21
commit e43c14f448
3 changed files with 16 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
python/.changelog.d/2239.added Normal file
View File

@ -0,0 +1 @@
Add firmware get-hash command.

9
python/src/trezorlib/cli/firmware.py
View File

@ -590,3 +590,12 @@ def update(
click.echo("Dry run. Not uploading firmware to device.") click.echo("Dry run. Not uploading firmware to device.")
else: else:
upload_firmware_into_device(client=client, firmware_data=firmware_data) upload_firmware_into_device(client=client, firmware_data=firmware_data)
@cli.command()
@click.argument("hex_challenge", required=False)
@with_client
def get_hash(client: "TrezorClient", hex_challenge: Optional[str]) -> str:
"""Get a hash of the installed firmware combined with the optional challenge."""
challenge = bytes.fromhex(hex_challenge) if hex_challenge else None
return firmware.get_hash(client, challenge).hex()

7
python/src/trezorlib/firmware.py
View File

@ -23,7 +23,7 @@ import construct as c
import ecdsa import ecdsa
from . import cosi, messages from . import cosi, messages
from .tools import session from .tools import expect, session
if TYPE_CHECKING: if TYPE_CHECKING:
from .client import TrezorClient from .client import TrezorClient
@ -518,3 +518,8 @@ def update(
return return
else: else:
raise RuntimeError(f"Unexpected message {resp}") raise RuntimeError(f"Unexpected message {resp}")
@expect(messages.FirmwareHash, field="hash", ret_type=bytes)
def get_hash(client: "TrezorClient", challenge: Optional[bytes]):
return client.call(messages.GetFirmwareHash(challenge=challenge))