def test_reset_device_skip_backup_break(self): external_entropy = b'zlutoucky kun upel divoke ody' * 2 strength = 128 ret = self.client.call_raw( proto.ResetDevice(display_random=False, strength=strength, passphrase_protection=False, pin_protection=False, language='english', label='test', skip_backup=True)) # Provide entropy assert isinstance(ret, proto.EntropyRequest) ret = self.client.call_raw(proto.EntropyAck(entropy=external_entropy)) assert isinstance(ret, proto.Success) # Check if device is properly initialized resp = self.client.call_raw(proto.Initialize()) assert resp.initialized is True assert resp.needs_backup is True # start Backup workflow ret = self.client.call_raw(proto.BackupDevice()) # send Initialize -> break workflow ret = self.client.call_raw(proto.Initialize()) assert isinstance(resp, proto.Features) # start backup again - should fail ret = self.client.call_raw(proto.BackupDevice()) assert isinstance(ret, proto.Failure)
def test_reset_device_skip_backup_break(self): ret = self.client.call_raw( proto.ResetDevice( display_random=False, strength=self.strength, passphrase_protection=False, pin_protection=False, language="english", label="test", skip_backup=True, ) ) assert isinstance(ret, proto.ButtonRequest) self.client.debug.press_yes() ret = self.client.call_raw(proto.ButtonAck()) # Provide entropy assert isinstance(ret, proto.EntropyRequest) ret = self.client.call_raw(proto.EntropyAck(entropy=self.external_entropy)) assert isinstance(ret, proto.Success) # Check if device is properly initialized ret = self.client.call_raw(proto.Initialize()) assert ret.initialized is True assert ret.needs_backup is True assert ret.unfinished_backup is False assert ret.no_backup is False # start Backup workflow ret = self.client.call_raw(proto.BackupDevice()) # send Initialize -> break workflow ret = self.client.call_raw(proto.Initialize()) assert isinstance(ret, proto.Features) assert ret.initialized is True assert ret.needs_backup is False assert ret.unfinished_backup is True assert ret.no_backup is False # start backup again - should fail ret = self.client.call_raw(proto.BackupDevice()) assert isinstance(ret, proto.Failure) # read Features again ret = self.client.call_raw(proto.Initialize()) assert isinstance(ret, proto.Features) assert ret.initialized is True assert ret.needs_backup is False assert ret.unfinished_backup is True assert ret.no_backup is False
def test_reset_device_skip_backup_break(client): ret = client.call_raw( messages.ResetDevice( display_random=False, strength=STRENGTH, passphrase_protection=False, pin_protection=False, language="en-US", label="test", skip_backup=True, ) ) assert isinstance(ret, messages.ButtonRequest) client.debug.press_yes() ret = client.call_raw(messages.ButtonAck()) # Provide entropy assert isinstance(ret, messages.EntropyRequest) ret = client.call_raw(messages.EntropyAck(entropy=EXTERNAL_ENTROPY)) assert isinstance(ret, messages.Success) # Check if device is properly initialized ret = client.call_raw(messages.Initialize()) assert ret.initialized is True assert ret.needs_backup is True assert ret.unfinished_backup is False assert ret.no_backup is False # start Backup workflow ret = client.call_raw(messages.BackupDevice()) # send Initialize -> break workflow ret = client.call_raw(messages.Initialize()) assert isinstance(ret, messages.Features) assert ret.initialized is True assert ret.needs_backup is False assert ret.unfinished_backup is True assert ret.no_backup is False # start backup again - should fail ret = client.call_raw(messages.BackupDevice()) assert isinstance(ret, messages.Failure) # read Features again ret = client.call_raw(messages.Initialize()) assert isinstance(ret, messages.Features) assert ret.initialized is True assert ret.needs_backup is False assert ret.unfinished_backup is True assert ret.no_backup is False
def test_interrupt_backup_fails(client): device.reset( client, display_random=False, strength=128, passphrase_protection=False, pin_protection=False, label="test", language="english", skip_backup=True, ) assert client.features.initialized is True assert client.features.needs_backup is True assert client.features.unfinished_backup is False assert client.features.no_backup is False # start backup client.call_raw(messages.BackupDevice()) # interupt backup by sending initialize client.init_device() # check that device state is as expected assert client.features.initialized is True assert client.features.needs_backup is False assert client.features.unfinished_backup is True assert client.features.no_backup is False # Second attempt at backup should fail with pytest.raises(TrezorFailure, match=r".*Seed already backed up"): device.backup(client)
def test_reset_device_no_backup(self): ret = self.client.call_raw( proto.ResetDevice( display_random=False, strength=self.strength, passphrase_protection=False, pin_protection=False, language="english", label="test", no_backup=True, ) ) # Provide entropy assert isinstance(ret, proto.EntropyRequest) ret = self.client.call_raw(proto.EntropyAck(entropy=self.external_entropy)) assert isinstance(ret, proto.Success) # Check if device is properly initialized ret = self.client.call_raw(proto.Initialize()) assert ret.initialized is True assert ret.needs_backup is False assert ret.unfinished_backup is False assert ret.no_backup is True # start backup - should fail ret = self.client.call_raw(proto.BackupDevice()) assert isinstance(ret, proto.Failure)
def test_interrupt_backup_fails(client): assert client.features.initialized is True assert client.features.needs_backup is True assert client.features.unfinished_backup is False assert client.features.no_backup is False # start backup client.call_raw(messages.BackupDevice()) # interupt backup by sending initialize client.init_device() # check that device state is as expected assert client.features.initialized is True assert client.features.needs_backup is False assert client.features.unfinished_backup is True assert client.features.no_backup is False # Second attempt at backup should fail with pytest.raises(TrezorFailure, match=r".*Seed already backed up"): device.backup(client)
def test_reset_device_skip_backup(self, client): ret = client.call_raw( proto.ResetDevice( display_random=False, strength=self.strength, passphrase_protection=False, pin_protection=False, language="english", label="test", skip_backup=True, )) assert isinstance(ret, proto.ButtonRequest) client.debug.press_yes() ret = client.call_raw(proto.ButtonAck()) # Provide entropy assert isinstance(ret, proto.EntropyRequest) internal_entropy = client.debug.read_reset_entropy() ret = client.call_raw(proto.EntropyAck(entropy=self.external_entropy)) assert isinstance(ret, proto.Success) # Check if device is properly initialized ret = client.call_raw(proto.Initialize()) assert ret.initialized is True assert ret.needs_backup is True assert ret.unfinished_backup is False assert ret.no_backup is False # Generate mnemonic locally entropy = generate_entropy(self.strength, internal_entropy, self.external_entropy) expected_mnemonic = Mnemonic("english").to_mnemonic(entropy) # start Backup workflow ret = client.call_raw(proto.BackupDevice()) mnemonic = [] for _ in range(self.strength // 32 * 3): assert isinstance(ret, proto.ButtonRequest) mnemonic.append(client.debug.read_reset_word()) client.debug.press_yes() client.call_raw(proto.ButtonAck()) mnemonic = " ".join(mnemonic) # Compare that device generated proper mnemonic for given entropies assert mnemonic == expected_mnemonic mnemonic = [] for _ in range(self.strength // 32 * 3): assert isinstance(ret, proto.ButtonRequest) mnemonic.append(client.debug.read_reset_word()) client.debug.press_yes() ret = client.call_raw(proto.ButtonAck()) assert isinstance(ret, proto.Success) mnemonic = " ".join(mnemonic) # Compare that second pass printed out the same mnemonic once again assert mnemonic == expected_mnemonic # start backup again - should fail ret = client.call_raw(proto.BackupDevice()) assert isinstance(ret, proto.Failure)
def test_initialized_device_backup_fail(self, client): ret = client.call_raw(proto.BackupDevice()) assert isinstance(ret, proto.Failure)
def test_initialized_device_backup_fail(client): ret = client.call_raw(messages.BackupDevice()) assert isinstance(ret, messages.Failure)
def test_initialized_device_backup_fail(self): self.setup_mnemonic_nopin_nopassphrase() ret = self.client.call_raw(proto.BackupDevice()) assert isinstance(ret, proto.Failure)
def test_reset_device_skip_backup(self): external_entropy = b'zlutoucky kun upel divoke ody' * 2 strength = 128 ret = self.client.call_raw( proto.ResetDevice(display_random=False, strength=strength, passphrase_protection=False, pin_protection=False, language='english', label='test', skip_backup=True)) # Provide entropy assert isinstance(ret, proto.EntropyRequest) internal_entropy = self.client.debug.read_reset_entropy() ret = self.client.call_raw(proto.EntropyAck(entropy=external_entropy)) assert isinstance(ret, proto.Success) # Check if device is properly initialized resp = self.client.call_raw(proto.Initialize()) assert resp.initialized is True assert resp.needs_backup is True # Generate mnemonic locally entropy = generate_entropy(strength, internal_entropy, external_entropy) expected_mnemonic = Mnemonic('english').to_mnemonic(entropy) # start Backup workflow ret = self.client.call_raw(proto.BackupDevice()) mnemonic = [] for _ in range(strength // 32 * 3): assert isinstance(ret, proto.ButtonRequest) mnemonic.append(self.client.debug.read_reset_word()) self.client.debug.press_yes() self.client.call_raw(proto.ButtonAck()) mnemonic = ' '.join(mnemonic) # Compare that device generated proper mnemonic for given entropies assert mnemonic == expected_mnemonic mnemonic = [] for _ in range(strength // 32 * 3): assert isinstance(ret, proto.ButtonRequest) mnemonic.append(self.client.debug.read_reset_word()) self.client.debug.press_yes() resp = self.client.call_raw(proto.ButtonAck()) assert isinstance(resp, proto.Success) mnemonic = ' '.join(mnemonic) # Compare that second pass printed out the same mnemonic once again assert mnemonic == expected_mnemonic # start backup again - should fail ret = self.client.call_raw(proto.BackupDevice()) assert isinstance(ret, proto.Failure)