mirror of
https://github.com/trezor/trezor-firmware.git
synced 2024-12-20 21:38:26 +00:00
156 lines
4.2 KiB
Python
156 lines
4.2 KiB
Python
from pathlib import Path
|
|
|
|
import construct
|
|
import pytest
|
|
import requests
|
|
|
|
from trezorlib import firmware
|
|
from trezorlib.firmware import (
|
|
LegacyFirmware,
|
|
LegacyV2Firmware,
|
|
VendorFirmware,
|
|
VendorHeader,
|
|
)
|
|
|
|
CORE_FW_VERSION = "2.4.2"
|
|
CORE_FW_FINGERPRINT = "54ccf155510b5292bd17ed748409d0d135112e24e62eb74184639460beecb213"
|
|
LEGACY_FW_VERSION = "1.10.3"
|
|
LEGACY_FW_FINGERPRINT = (
|
|
"bf0cc936a9afbf0a4ae7b727a2817fb69fba432d7230a0ff7b79b4a73b845197"
|
|
)
|
|
|
|
CORE_FW = f"https://data.trezor.io/firmware/2/trezor-{CORE_FW_VERSION}.bin"
|
|
LEGACY_FW = f"https://data.trezor.io/firmware/1/trezor-{LEGACY_FW_VERSION}.bin"
|
|
|
|
HERE = Path(__file__).parent
|
|
|
|
VENDOR_HEADER = (
|
|
HERE.parent.parent
|
|
/ "core"
|
|
/ "embed"
|
|
/ "models"
|
|
/ "T2T1"
|
|
/ "vendorheader"
|
|
/ "vendorheader_satoshilabs_signed_prod.bin"
|
|
)
|
|
|
|
|
|
def _fetch(url: str, version: str) -> bytes:
|
|
path = HERE / f"trezor-{version}.bin"
|
|
if not path.exists():
|
|
r = requests.get(url)
|
|
r.raise_for_status()
|
|
path.write_bytes(r.content)
|
|
return path.read_bytes()
|
|
|
|
|
|
@pytest.fixture()
|
|
def legacy_fw() -> bytes:
|
|
return _fetch(LEGACY_FW, LEGACY_FW_VERSION)
|
|
|
|
|
|
@pytest.fixture()
|
|
def core_fw() -> bytes:
|
|
return _fetch(CORE_FW, CORE_FW_VERSION)
|
|
|
|
|
|
def test_core_basic(core_fw: bytes) -> None:
|
|
fw = VendorFirmware.parse(core_fw)
|
|
fw.verify()
|
|
assert fw.digest().hex() == CORE_FW_FINGERPRINT
|
|
version_str = ".".join(str(x) for x in fw.firmware.header.version)
|
|
assert version_str.startswith(CORE_FW_VERSION)
|
|
assert fw.vendor_header.text == "SatoshiLabs"
|
|
assert fw.build() == core_fw
|
|
|
|
|
|
def test_vendor_header(core_fw: bytes) -> None:
|
|
fw = VendorFirmware.parse(core_fw)
|
|
|
|
vh_data = fw.vendor_header.build()
|
|
assert vh_data in core_fw
|
|
assert vh_data == VENDOR_HEADER.read_bytes()
|
|
|
|
vh = VendorHeader.parse(vh_data)
|
|
assert vh == fw.vendor_header
|
|
vh.verify()
|
|
|
|
with pytest.raises(construct.ConstructError):
|
|
VendorFirmware.parse(vh_data)
|
|
|
|
|
|
def test_core_code_hashes(core_fw: bytes) -> None:
|
|
fw = VendorFirmware.parse(core_fw)
|
|
fw.firmware.header.hashes = []
|
|
assert fw.digest().hex() == CORE_FW_FINGERPRINT
|
|
|
|
|
|
def test_legacy_basic(legacy_fw: bytes) -> None:
|
|
fw = LegacyFirmware.parse(legacy_fw)
|
|
fw.verify()
|
|
assert fw.digest().hex() == LEGACY_FW_FINGERPRINT
|
|
assert fw.build() == legacy_fw
|
|
|
|
|
|
def test_unsigned(legacy_fw: bytes) -> None:
|
|
legacy = LegacyFirmware.parse(legacy_fw)
|
|
|
|
legacy.verify()
|
|
legacy.key_indexes = [0, 0, 0]
|
|
legacy.signatures = [b"", b"", b""]
|
|
|
|
with pytest.raises(firmware.Unsigned):
|
|
legacy.verify()
|
|
|
|
assert legacy.embedded_v2 is not None
|
|
legacy.embedded_v2.verify()
|
|
|
|
legacy.embedded_v2.header.v1_key_indexes = [0, 0, 0]
|
|
legacy.embedded_v2.header.v1_signatures = [b"", b"", b""]
|
|
with pytest.raises(firmware.Unsigned):
|
|
legacy.embedded_v2.verify()
|
|
|
|
|
|
def test_disallow_unsigned(core_fw: bytes) -> None:
|
|
core = VendorFirmware.parse(core_fw)
|
|
core.firmware.header.sigmask = 0
|
|
core.firmware.header.signature = b""
|
|
with pytest.raises(firmware.InvalidSignatureError):
|
|
core.verify()
|
|
|
|
|
|
def test_embedded_v2(legacy_fw: bytes) -> None:
|
|
legacy = LegacyFirmware.parse(legacy_fw)
|
|
assert legacy.embedded_v2 is not None
|
|
legacy.embedded_v2.verify()
|
|
|
|
embedded_data = legacy.embedded_v2.build()
|
|
cutoff_data = legacy_fw[256:]
|
|
assert cutoff_data == embedded_data
|
|
embedded = LegacyV2Firmware.parse(cutoff_data)
|
|
assert embedded == legacy.embedded_v2
|
|
|
|
|
|
def test_integrity_legacy(legacy_fw: bytes) -> None:
|
|
legacy = LegacyFirmware.parse(legacy_fw)
|
|
legacy.verify()
|
|
|
|
modified_data = bytearray(legacy_fw)
|
|
modified_data[-1] ^= 0x01
|
|
modified = LegacyFirmware.parse(modified_data)
|
|
assert modified.digest() != legacy.digest()
|
|
with pytest.raises(firmware.InvalidSignatureError):
|
|
modified.verify()
|
|
|
|
|
|
def test_integrity_core(core_fw: bytes) -> None:
|
|
core = VendorFirmware.parse(core_fw)
|
|
core.verify()
|
|
|
|
modified_data = bytearray(core_fw)
|
|
modified_data[-1] ^= 0x01
|
|
modified = VendorFirmware.parse(modified_data)
|
|
assert modified.digest() != core.digest()
|
|
with pytest.raises(firmware.FirmwareIntegrityError):
|
|
modified.verify()
|