From 9ae319211f47d329a5923a85f8a46bccd004dd72 Mon Sep 17 00:00:00 2001 From: Pavol Rusnak Date: Thu, 22 Aug 2019 20:16:18 +0200 Subject: [PATCH] core: fill in Features.features --- core/src/apps/homescreen/__init__.py | 21 ++++++++++++++++++++- 1 file changed, 20 insertions(+), 1 deletion(-) diff --git a/core/src/apps/homescreen/__init__.py b/core/src/apps/homescreen/__init__.py index f80e08fbf..08313f89b 100644 --- a/core/src/apps/homescreen/__init__.py +++ b/core/src/apps/homescreen/__init__.py @@ -1,5 +1,5 @@ from trezor import config, utils, wire -from trezor.messages import MessageType +from trezor.messages import Feature, MessageType from trezor.messages.Features import Features from trezor.messages.Success import Success from trezor.wire import register @@ -36,6 +36,25 @@ def get_features() -> Features: f.no_backup = storage.device.no_backup() f.flags = storage.device.get_flags() f.recovery_mode = storage.recovery.is_in_progress() + if utils.BITCOIN_ONLY: + f.features = [Feature.Bitcoin, Feature.Crypto] + else: + f.features = [ + Feature.Bitcoin, + Feature.Bitcoin_like, + Feature.Binance, + Feature.Cardano, + Feature.Crypto, + Feature.EOS, + Feature.Ethereum, + Feature.Lisk, + Feature.Monero, + Feature.NEM, + Feature.Ripple, + Feature.Stellar, + Feature.Tezos, + Feature.U2F, + ] return f