d-bis/app-ethereum
4
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

New function to get asset index by address

Browse Source
This commit is contained in:
Alexandre Paillier
2024-06-17 15:20:11 +02:00
parent 6d6f7c740c
commit 51ddfe537d
2 changed files with 19 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
src/manage_asset_info.c
View File

@@ -6,33 +6,35 @@ void forget_known_assets(void) {
tmpCtx.transactionContext.currentAssetIndex = 0; tmpCtx.transactionContext.currentAssetIndex = 0;
} }
static extraInfo_t *get_asset_info(uint8_t index) { static extraInfo_t *get_asset_info(int index) {
if (index >= MAX_ASSETS) { if ((index < 0) || (index >= MAX_ASSETS)) {
return NULL; return NULL;
} }
return &tmpCtx.transactionContext.extraInfo[index]; return &tmpCtx.transactionContext.extraInfo[index];
} }
static bool asset_info_is_set(uint8_t index) { static bool asset_info_is_set(int index) {
if (index >= MAX_ASSETS) { if ((index < 0) || (index >= MAX_ASSETS)) {
return false; return false;
} }
return tmpCtx.transactionContext.assetSet[index]; return tmpCtx.transactionContext.assetSet[index];
} }
extraInfo_t *get_asset_info_by_addr(const uint8_t *contractAddress) { int get_asset_index_by_addr(const uint8_t *addr) {
// Works for ERC-20 & NFT tokens since both structs in the union have the // Works for ERC-20 & NFT tokens since both structs in the union have the
// contract address aligned // contract address aligned
for (uint8_t i = 0; i < MAX_ASSETS; i++) { for (int i = 0; i < MAX_ASSETS; i++) {
extraInfo_t *currentItem = get_asset_info(i); extraInfo_t *asset = get_asset_info(i);
if (asset_info_is_set(i) && if (asset_info_is_set(i) && (memcmp(asset->token.address, addr, ADDRESS_LENGTH) == 0)) {
(memcmp(currentItem->token.address, contractAddress, ADDRESS_LENGTH) == 0)) {
PRINTF("Token found at index %d\n", i); PRINTF("Token found at index %d\n", i);
return currentItem; return i;
} }
} }
return -1;
}
return NULL; extraInfo_t *get_asset_info_by_addr(const uint8_t *addr) {
return get_asset_info(get_asset_index_by_addr(addr));
} }
extraInfo_t *get_current_asset_info(void) { extraInfo_t *get_current_asset_info(void) {

6
src/manage_asset_info.h
View File

@@ -1,8 +1,14 @@
#ifndef MANAGE_ASSET_INFO_H_
#define MANAGE_ASSET_INFO_H_
#include "shared_context.h" #include "shared_context.h"
#include "common_utils.h" #include "common_utils.h"
#include "asset_info.h" #include "asset_info.h"
void forget_known_assets(void); void forget_known_assets(void);
int get_asset_index_by_addr(const uint8_t *addr);
extraInfo_t *get_asset_info_by_addr(const uint8_t *contractAddress); extraInfo_t *get_asset_info_by_addr(const uint8_t *contractAddress);
extraInfo_t *get_current_asset_info(void); extraInfo_t *get_current_asset_info(void);
void validate_current_asset_info(void); void validate_current_asset_info(void);
#endif // MANAGE_ASSET_INFO_H_