src/apps/common/storage: add autolock_delay_ms variable

trezor · Apr 3, 2018 · a6e8a37 · a6e8a37
1 parent 04680f4
commit a6e8a37
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/src/apps/common/storage.py b/src/apps/common/storage.py
@@ -21,6 +21,7 @@
 _U2F_COUNTER        = const(0x09)  # int
 _PASSPHRASE_SOURCE  = const(0x0A)  # int
 _UNFINISHED_BACKUP  = const(0x0B)  # bool (0x01 or empty)
+_AUTOLOCK_DELAY_MS  = const(0x0C)  # int
 
 
 def _new_device_id() -> str:
@@ -130,6 +131,20 @@ def set_flags(flags: int) -> None:
         config.set(_APP, _FLAGS, flags.to_bytes(4, 'big'))
 
 
+def get_autolock_delay_ms() -> int:
+    b = config.get(_APP, _AUTOLOCK_DELAY_MS)
+    if b is None:
+        return 10 * 60 * 1000
+    else:
+        return int.from_bytes(b, 'big')
+
+
+def set_autolock_delay_ms(delay_ms: int) -> None:
+    if delay_ms < 60 * 1000:
+        delay_ms = 60 * 1000
+    config.set(_APP, _AUTOLOCK_DELAY_MS, delay_ms.to_bytes(4, 'big'))
+
+
 def next_u2f_counter() -> int:
     b = config.get(_APP, _U2F_COUNTER)
     if b is None: