diff --git a/src/network.c b/src/network.c index eb25302d0b..6671bf538d 100644 --- a/src/network.c +++ b/src/network.c @@ -144,3 +144,9 @@ const char *get_displayable_ticker(const uint64_t *chain_id) { } return ticker; } + +bool app_compatible_with_chain_id(const uint64_t *chain_id) { + return ((chainConfig->chainId == *chain_id) || + (chain_is_ethereum_compatible(&chainConfig->chainId) && + chain_is_ethereum_compatible(chain_id))); +} diff --git a/src/network.h b/src/network.h index c6640d554b..4feaf6a2c8 100644 --- a/src/network.h +++ b/src/network.h @@ -7,6 +7,7 @@ const char *get_network_name_from_chain_id(const uint64_t *chain_id); const char *get_network_ticker_from_chain_id(const uint64_t *chain_id); bool chain_is_ethereum_compatible(const uint64_t *chain_id); +bool app_compatible_with_chain_id(const uint64_t *chain_id); uint64_t get_tx_chain_id(void); diff --git a/src_features/provideErc20TokenInformation/cmd_provideTokenInfo.c b/src_features/provideErc20TokenInformation/cmd_provideTokenInfo.c index 5fd546bb44..60ba810cd6 100644 --- a/src_features/provideErc20TokenInformation/cmd_provideTokenInfo.c +++ b/src_features/provideErc20TokenInformation/cmd_provideTokenInfo.c @@ -3,6 +3,7 @@ #include "public_keys.h" #include "common_ui.h" #include "os_io_seproxyhal.h" +#include "network.h" #ifdef HAVE_CONTRACT_NAME_IN_DESCRIPTOR @@ -111,7 +112,7 @@ void handleProvideErc20TokenInformation(uint8_t p1, UNUSED(tx); uint32_t offset = 0; uint8_t tickerLength; - uint32_t chainId; + uint64_t chain_id; uint8_t hash[INT256_LENGTH]; cx_ecfp_public_key_t tokenKey; @@ -141,12 +142,13 @@ void handleProvideErc20TokenInformation(uint8_t p1, memmove(token->address, workBuffer + offset, 20); offset += 20; dataLength -= 20; + // TODO: Handle 64-bit long chain IDs token->decimals = U4BE(workBuffer, offset); offset += 4; dataLength -= 4; - chainId = U4BE(workBuffer, offset); - if ((chainConfig->chainId != ETHEREUM_MAINNET_CHAINID) && (chainConfig->chainId != chainId)) { - PRINTF("ChainId token mismatch: %d vs %d\n", chainConfig->chainId, chainId); + chain_id = U4BE(workBuffer, offset); + if (!app_compatible_with_chain_id(&chain_id)) { + PRINTF("ChainId token mismatch: %d vs %d\n", chainConfig->chainId, chain_id); THROW(0x6A80); } offset += 4; diff --git a/src_features/provideNFTInformation/cmd_provideNFTInfo.c b/src_features/provideNFTInformation/cmd_provideNFTInfo.c index cde32159d6..b8ba152be1 100644 --- a/src_features/provideNFTInformation/cmd_provideNFTInfo.c +++ b/src_features/provideNFTInformation/cmd_provideNFTInfo.c @@ -130,7 +130,7 @@ void handleProvideNFTInformation(uint8_t p1, // this prints raw data, so to have a more meaningful print, display // the buffer before the endianness swap PRINTF("ChainID: %.*H\n", sizeof(chain_id), (workBuffer + offset)); - if (!chain_is_ethereum_compatible(&chain_id)) { + if (!app_compatible_with_chain_id(&chain_id)) { PRINTF("Unsupported chain ID!\n"); THROW(APDU_RESPONSE_INVALID_DATA); } diff --git a/src_features/setPlugin/cmd_setPlugin.c b/src_features/setPlugin/cmd_setPlugin.c index 8c4a67e187..3bd6fc2c98 100644 --- a/src_features/setPlugin/cmd_setPlugin.c +++ b/src_features/setPlugin/cmd_setPlugin.c @@ -159,7 +159,7 @@ void handleSetPlugin(uint8_t p1, // this prints raw data, so to have a more meaningful print, display // the buffer before the endianness swap PRINTF("ChainID: %.*H\n", sizeof(chain_id), (workBuffer + offset)); - if (!chain_is_ethereum_compatible(&chain_id)) { + if (!app_compatible_with_chain_id(&chain_id)) { PRINTF("Unsupported chain ID!\n"); THROW(APDU_RESPONSE_INVALID_DATA); }