Skip to content

test: verify storage power cycles on full and Bitcoin-only products #71

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Draft
BitHighlander wants to merge 1 commit into
base: audit/owned-emulator-network
Choose a base branch
from
Draft

test: verify storage power cycles on full and Bitcoin-only products #71

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
22 changes: 22 additions & 0 deletions tests/test_storage_version_gate.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -998,6 +998,8 @@ def _create_wallet(self):
label=LABEL, language="english")
c.init_device()
self.assertTrue(c.features.initialized)
from keepkeylib import messages_pb2 as proto
self.bitcoin_only = c.call(proto.GetCoinTable()).num_coins == 2
addr = c.get_address("Bitcoin", BIP44_ADDRESS_N)
finally:
c.close()
Expand Down Expand Up @@ -1063,6 +1065,8 @@ def test_reboot_preserves_the_wallet(self):
# only record which branch the author was standing on.
declared = _define(_read_source("include/keepkey/firmware/storage.h"),
"STORAGE_VERSION")
if self.bitcoin_only:
declared += STORAGE_VERSION_BTC_ONLY_BASE
self.assertEqual(
declared, self.emu.read_u32(off, OFF_VERSION),
"the firmware committed a storage version other than the %d its "
Expand Down Expand Up @@ -1189,6 +1193,24 @@ def test_bitcoin_only_band_refuses_without_wiping(self):
wallet comes back once the stamp is the multi-chain one again.
"""
addr, off = self._create_wallet()
if self.bitcoin_only:
# This product must read its own band; the full-product branch below
# must refuse the same band while preserving its bytes.
declared = _define(_read_source("include/keepkey/firmware/storage.h"),
"STORAGE_VERSION")
self.assertEqual(STORAGE_VERSION_BTC_ONLY_BASE + declared,
self.emu.read_u32(off, OFF_VERSION))
before = self.emu.image()
self.emu.boot()
c = self.emu.client(self.method, pin=PIN)
try:
c.init_device()
self.assertTrue(c.features.initialized)
self.assertEqual(before, self.emu.image())
self.assertEqual(addr, c.get_address("Bitcoin", BIP44_ADDRESS_N))
finally:
c.close()
return
self.assertLess(
self.emu.read_u32(off, OFF_VERSION), STORAGE_VERSION_BTC_ONLY_BASE,
"this emulator already stamps its wallets into the bitcoin-only "
Expand Down