arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2025-03-30 15:05:43 +00:00
Code Issues Releases Wiki Activity

device: pass u2f counter when running device recovery

Browse Source
This commit is contained in:
matejcik 2019-01-23 11:34:59 +01:00
parent 7d14018f99
commit 3390422ce3
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
trezorlib/device.py
View File

@ -15,6 +15,7 @@
# If not, see <https://www.gnu.org/licenses/lgpl-3.0.html>. # If not, see <https://www.gnu.org/licenses/lgpl-3.0.html>.
import os import os
import time
import warnings import warnings
from . import messages as proto from . import messages as proto
@ -110,6 +111,7 @@ def recover(
input_callback=None, input_callback=None,
type=proto.RecoveryDeviceType.ScrambledWords, type=proto.RecoveryDeviceType.ScrambledWords,
dry_run=False, dry_run=False,
u2f_counter=None,
): ):
if client.features.model == "1" and input_callback is None: if client.features.model == "1" and input_callback is None:
raise RuntimeError("Input callback required for Trezor One") raise RuntimeError("Input callback required for Trezor One")
@ -122,6 +124,9 @@ def recover(
"Device already initialized. Call device.wipe() and try again." "Device already initialized. Call device.wipe() and try again."
) )
if u2f_counter is None:
u2f_counter = int(time.time())
res = client.call( res = client.call(
proto.RecoveryDevice( proto.RecoveryDevice(
word_count=word_count, word_count=word_count,
@ -132,6 +137,7 @@ def recover(
enforce_wordlist=True, enforce_wordlist=True,
type=type, type=type,
dry_run=dry_run, dry_run=dry_run,
u2f_counter=u2f_counter,
) )
) )