arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-11-19 14:08:11 +00:00
Code Issues Releases Wiki Activity

refactor(core): Use set_int()/get_int() for APP_COMMON_AUTHORIZATION_TYPE.

Browse Source
This commit is contained in:
Andrew Kozlik 2023-02-24 13:30:32 +01:00 committed by Martin Milata
parent 9bb1d1cf8b
commit 77d8af1322
1 changed files with 6 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
core/src/apps/common/authorization.py
View File

@ -17,7 +17,7 @@ APP_COMMON_AUTHORIZATION_TYPE = (
def is_set() -> bool:
return bool(storage_cache.get(APP_COMMON_AUTHORIZATION_TYPE))
return storage_cache.get(APP_COMMON_AUTHORIZATION_TYPE) is not None
def set(auth_message: protobuf.MessageType) -> None:
@ -29,21 +29,17 @@ def set(auth_message: protobuf.MessageType) -> None:
# (because only wire-level messages have wire_type, which we use as identifier)
ensure(auth_message.MESSAGE_WIRE_TYPE is not None)
assert auth_message.MESSAGE_WIRE_TYPE is not None # so that typechecker knows too
storage_cache.set(
APP_COMMON_AUTHORIZATION_TYPE,
auth_message.MESSAGE_WIRE_TYPE.to_bytes(2, "big"),
)
storage_cache.set_int(APP_COMMON_AUTHORIZATION_TYPE, auth_message.MESSAGE_WIRE_TYPE)
storage_cache.set(APP_COMMON_AUTHORIZATION_DATA, buffer)
def get() -> protobuf.MessageType | None:
stored_auth_type = storage_cache.get(APP_COMMON_AUTHORIZATION_TYPE)
stored_auth_type = storage_cache.get_int(APP_COMMON_AUTHORIZATION_TYPE)
if not stored_auth_type:
return None
msg_wire_type = int.from_bytes(stored_auth_type, "big")
buffer = storage_cache.get(APP_COMMON_AUTHORIZATION_DATA, b"")
return protobuf.load_message_buffer(buffer, msg_wire_type)
return protobuf.load_message_buffer(buffer, stored_auth_type)
def is_set_any_session(auth_type: MessageType) -> bool:
@ -53,12 +49,11 @@ def is_set_any_session(auth_type: MessageType) -> bool:
def get_wire_types() -> Iterable[int]:
stored_auth_type = storage_cache.get(APP_COMMON_AUTHORIZATION_TYPE)
stored_auth_type = storage_cache.get_int(APP_COMMON_AUTHORIZATION_TYPE)
if stored_auth_type is None:
return ()
msg_wire_type = int.from_bytes(stored_auth_type, "big")
return WIRE_TYPES.get(msg_wire_type, ())
return WIRE_TYPES.get(stored_auth_type, ())
def clear() -> None: