445 lines
14 KiB
C
445 lines
14 KiB
C
#include <string.h>
|
|
#include "eth_plugin_interface.h"
|
|
#include "shared_context.h" // TODO : rewrite as independant code
|
|
#include "eth_plugin_internal.h" // TODO : rewrite as independant code
|
|
|
|
typedef enum {
|
|
|
|
STARKWARE_REGISTER = 0,
|
|
STARKWARE_DEPOSIT_TOKEN,
|
|
STARKWARE_DEPOSIT_ETH,
|
|
STARKWARE_DEPOSIT_CANCEL,
|
|
STARKWARE_DEPOSIT_RECLAIM,
|
|
STARKWARE_WITHDRAW,
|
|
STARKWARE_FULL_WITHDRAW,
|
|
STARKWARE_FREEZE,
|
|
STARKWARE_ESCAPE,
|
|
STARKWARE_VERIFY_ESCAPE
|
|
|
|
} starkwareSelector_t;
|
|
|
|
// register : starkkey (32), drop param 2
|
|
// Registration
|
|
// Contract Name
|
|
// From ETH address
|
|
// Master account
|
|
// deposit token : verify tokenId (32), vaultId (4), quantized Amount (32)
|
|
// Deposit
|
|
// Contract Name
|
|
// Token Account
|
|
// Amount
|
|
// deposit : verify tokenId (32), vaultId (4)
|
|
// Flow similar to deposit
|
|
// deposit cancel, deposit reclaim : tokenId (32) ignored, vaultId(4)
|
|
// full withdrawal, freeze : vaultId (4)
|
|
// Cancel Deposit | Reclaim Deposit | Full Withdrawal | Freeze
|
|
// Contract Name
|
|
// Token Account
|
|
// withdrawal : verify tokenId (32)
|
|
// Withdrawal
|
|
// Contract Name
|
|
// To Eth Address
|
|
// Token Symbol
|
|
// escape : starkkey (32), vaultId (4), verify tokenId (32), quantized Amount (32)
|
|
// Escape
|
|
// Contract Name
|
|
// Amount
|
|
// Master Account
|
|
// Token Account
|
|
// verify escape : escapeProof (ignore)
|
|
// Verify Escape
|
|
// Contract Name
|
|
|
|
static const uint8_t STARKWARE_EXPECTED_DATA_SIZE[] = {
|
|
4 + 32,
|
|
4 + 32 + 32 + 32,
|
|
4 + 32 + 32,
|
|
4 + 32 + 32,
|
|
4 + 32 + 32,
|
|
4 + 32,
|
|
4 + 32,
|
|
4 + 32,
|
|
4 + 32 + 32 + 32 + 32,
|
|
0
|
|
};
|
|
|
|
static const uint8_t STARKWARE_NUM_SCREENS[] = {
|
|
4 - 1,
|
|
4 - 1,
|
|
4 - 1,
|
|
3 - 1,
|
|
3 - 1,
|
|
4 - 1,
|
|
3 - 1,
|
|
3 - 1,
|
|
5 - 1,
|
|
2 - 1
|
|
};
|
|
|
|
typedef struct starkware_parameters_t {
|
|
|
|
uint8_t vaultId[4];
|
|
uint8_t selectorIndex;
|
|
uint8_t starkKey[32];
|
|
uint8_t amount[32];
|
|
uint8_t validToken;
|
|
|
|
} starkware_parameters_t;
|
|
|
|
// TODO : rewrite as independant code
|
|
bool starkware_verify_token_id(uint8_t *tmp32, uint8_t *tokenId) {
|
|
if (quantumSet) {
|
|
tokenDefinition_t *currentToken = NULL;
|
|
if (dataContext.tokenContext.quantumIndex != MAX_TOKEN) {
|
|
currentToken = &tmpCtx.transactionContext.tokens[dataContext.tokenContext.quantumIndex];
|
|
}
|
|
compute_token_id(&global_sha3,
|
|
(currentToken != NULL ? currentToken->address : NULL),
|
|
dataContext.tokenContext.quantum, tmp32);
|
|
if (memcmp(tokenId, tmp32, 32) != 0) {
|
|
PRINTF("Token ID not matching - computed %.*H\n", 32, tmp32);
|
|
PRINTF("Current quantum %.*H\n", 32, dataContext.tokenContext.quantum);
|
|
PRINTF("Requested %.*H\n", 32, tokenId);
|
|
return false;
|
|
}
|
|
}
|
|
else {
|
|
PRINTF("Quantum not set\n");
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
void starkware_print_vault_id(uint32_t vaultId, char *destination) {
|
|
snprintf(destination, 10, "%d", vaultId);
|
|
}
|
|
|
|
void starkware_print_stark_key(uint8_t *starkKey, char *destination) {
|
|
snprintf(destination, 70, "0x%.*H", 32, starkKey);
|
|
}
|
|
|
|
// TODO : rewrite as independant code
|
|
void starkware_print_eth_address(uint8_t *address, char *destination) {
|
|
destination[0] = '0';
|
|
destination[1] = 'x';
|
|
getEthAddressStringFromBinary(address, destination + 2, &global_sha3, chainConfig);
|
|
destination[42] = '\0';
|
|
}
|
|
|
|
// TODO : rewrite as independant code
|
|
void starkware_print_amount(uint8_t *amountData, char *destination, bool forEscape) {
|
|
uint256_t amount, amountPre, quantum;
|
|
uint8_t decimals;
|
|
char *ticker = (char*)PIC(chainConfig->coinName);
|
|
|
|
if ((amountData == NULL) || (forEscape && (dataContext.tokenContext.quantumIndex == MAX_TOKEN))) {
|
|
decimals = WEI_TO_ETHER;
|
|
if (!forEscape) {
|
|
convertUint256BE(tmpContent.txContent.value.value, tmpContent.txContent.value.length, &amountPre);
|
|
}
|
|
else {
|
|
readu256BE(amountData, &amountPre);
|
|
}
|
|
}
|
|
else {
|
|
tokenDefinition_t *token = &tmpCtx.transactionContext.tokens[dataContext.tokenContext.quantumIndex];
|
|
decimals = token->decimals;
|
|
ticker = (char*)token->ticker;
|
|
readu256BE(amountData, &amountPre);
|
|
}
|
|
readu256BE(dataContext.tokenContext.quantum, &quantum);
|
|
mul256(&amountPre, &quantum, &amount);
|
|
tostring256(&amount, 10, (char*)(G_io_apdu_buffer + 100), 100);
|
|
strcpy(destination, ticker);
|
|
adjustDecimals((char*)(G_io_apdu_buffer + 100), strlen((char*)(G_io_apdu_buffer + 100)), destination + strlen(ticker), 50 - strlen(ticker), decimals);
|
|
}
|
|
|
|
// TODO : rewrite as independant code
|
|
void starkware_print_ticker(char *destination) {
|
|
char *ticker = (char*)PIC(chainConfig->coinName);
|
|
|
|
if (dataContext.tokenContext.quantumIndex != MAX_TOKEN) {
|
|
tokenDefinition_t *token = &tmpCtx.transactionContext.tokens[dataContext.tokenContext.quantumIndex];
|
|
ticker = (char*)token->ticker;
|
|
}
|
|
strcpy(destination, ticker);
|
|
}
|
|
|
|
// TODO : rewrite as independant code
|
|
void starkware_get_source_address(char *destination) {
|
|
uint8_t privateKeyData[32];
|
|
cx_ecfp_private_key_t privateKey;
|
|
cx_ecfp_public_key_t publicKey;
|
|
os_perso_derive_node_bip32(CX_CURVE_256K1, tmpCtx.transactionContext.bip32Path,
|
|
tmpCtx.transactionContext.pathLength,
|
|
privateKeyData, NULL);
|
|
cx_ecfp_init_private_key(CX_CURVE_256K1, privateKeyData, 32, &privateKey);
|
|
io_seproxyhal_io_heartbeat();
|
|
cx_ecfp_generate_pair(CX_CURVE_256K1, &publicKey, &privateKey, 1);
|
|
os_memset(&privateKey, 0, sizeof(privateKey));
|
|
os_memset(privateKeyData, 0, sizeof(privateKeyData));
|
|
io_seproxyhal_io_heartbeat();
|
|
destination[0] = '0';
|
|
destination[1] = 'x';
|
|
getEthAddressStringFromKey(&publicKey, destination + 2, &global_sha3, chainConfig);
|
|
destination[42] = '\0';
|
|
}
|
|
|
|
void starkware_plugin_call(int message, void *parameters) {
|
|
|
|
switch(message) {
|
|
case ETH_PLUGIN_INIT_CONTRACT: {
|
|
uint8_t i;
|
|
ethPluginInitContract_t *msg = (ethPluginInitContract_t*)parameters;
|
|
starkware_parameters_t *context = (starkware_parameters_t*)msg->pluginContext;
|
|
PRINTF("starkware plugin init\n");
|
|
for (i=0; i<NUM_STARKWARE_SELECTORS; i++) {
|
|
if (memcmp(PIC(STARKWARE_SELECTORS[i]), msg->selector, SELECTOR_SIZE) == 0) {
|
|
context->selectorIndex = i;
|
|
break;
|
|
}
|
|
}
|
|
if (i == NUM_STARKWARE_SELECTORS) {
|
|
PRINTF("Unknown selector %.*H\n", SELECTOR_SIZE, msg->selector);
|
|
break;
|
|
}
|
|
if (STARKWARE_EXPECTED_DATA_SIZE[context->selectorIndex] != 0) {
|
|
if (msg->dataSize != STARKWARE_EXPECTED_DATA_SIZE[context->selectorIndex]) {
|
|
PRINTF("Unexpected data size for command %d expected %d got %d\n", context->selectorIndex,
|
|
STARKWARE_EXPECTED_DATA_SIZE[context->selectorIndex], msg->dataSize);
|
|
break;
|
|
}
|
|
}
|
|
context->validToken = true;
|
|
msg->result = ETH_PLUGIN_RESULT_OK;
|
|
}
|
|
break;
|
|
|
|
case ETH_PLUGIN_PROVIDE_PARAMETER : {
|
|
ethPluginProvideParameter_t *msg = (ethPluginProvideParameter_t*)parameters;
|
|
starkware_parameters_t *context = (starkware_parameters_t*)msg->pluginContext;
|
|
PRINTF("starkware plugin provide parameter %d %.*H\n", msg->parameterOffset, 32, msg->parameter);
|
|
// Ignore for verify escape
|
|
if (context->selectorIndex == STARKWARE_VERIFY_ESCAPE) {
|
|
msg->result = ETH_PLUGIN_RESULT_OK;
|
|
break;
|
|
}
|
|
switch(msg->parameterOffset) {
|
|
case 4:
|
|
switch(context->selectorIndex) {
|
|
case STARKWARE_REGISTER:
|
|
case STARKWARE_ESCAPE:
|
|
memmove(context->starkKey, msg->parameter, 32);
|
|
break;
|
|
case STARKWARE_DEPOSIT_CANCEL:
|
|
case STARKWARE_DEPOSIT_RECLAIM:
|
|
break;
|
|
case STARKWARE_FULL_WITHDRAW:
|
|
case STARKWARE_FREEZE:
|
|
memmove(context->vaultId, msg->parameter + 32 - 4, 4);
|
|
break;
|
|
case STARKWARE_DEPOSIT_ETH:
|
|
case STARKWARE_DEPOSIT_TOKEN:
|
|
case STARKWARE_WITHDRAW:
|
|
context->validToken = starkware_verify_token_id(context->amount, msg->parameter);
|
|
break;
|
|
default:
|
|
break;
|
|
}
|
|
msg->result = ETH_PLUGIN_RESULT_OK;
|
|
break;
|
|
|
|
case 4 + 32:
|
|
switch(context->selectorIndex) {
|
|
case STARKWARE_DEPOSIT_CANCEL:
|
|
case STARKWARE_DEPOSIT_RECLAIM:
|
|
case STARKWARE_DEPOSIT_ETH:
|
|
case STARKWARE_DEPOSIT_TOKEN:
|
|
case STARKWARE_ESCAPE:
|
|
memmove(context->vaultId, msg->parameter + 32 - 4, 4);
|
|
break;
|
|
default:
|
|
break;
|
|
}
|
|
msg->result = ETH_PLUGIN_RESULT_OK;
|
|
break;
|
|
|
|
case 4 + 32 + 32:
|
|
switch(context->selectorIndex) {
|
|
case STARKWARE_DEPOSIT_TOKEN:
|
|
memmove(context->amount, msg->parameter, 32);
|
|
break;
|
|
case STARKWARE_ESCAPE:
|
|
context->validToken = starkware_verify_token_id(context->amount, msg->parameter);
|
|
break;
|
|
default:
|
|
break;
|
|
}
|
|
msg->result = ETH_PLUGIN_RESULT_OK;
|
|
break;
|
|
|
|
case 4 + 32 + 32 + 32:
|
|
switch(context->selectorIndex) {
|
|
case STARKWARE_ESCAPE:
|
|
memmove(context->amount, msg->parameter, 32);
|
|
break;
|
|
default:
|
|
break;
|
|
}
|
|
msg->result = ETH_PLUGIN_RESULT_OK;
|
|
break;
|
|
|
|
default:
|
|
PRINTF("Unhandled parameter offset\n");
|
|
break;
|
|
}
|
|
}
|
|
break;
|
|
|
|
case ETH_PLUGIN_FINALIZE: {
|
|
ethPluginFinalize_t *msg = (ethPluginFinalize_t*)parameters;
|
|
starkware_parameters_t *context = (starkware_parameters_t*)msg->pluginContext;
|
|
PRINTF("starkware plugin finalize\n");
|
|
if (!context->validToken) {
|
|
msg->result = ETH_PLUGIN_RESULT_FALLBACK;
|
|
}
|
|
else {
|
|
msg->uiType = ETH_UI_TYPE_GENERIC;
|
|
msg->numScreens = STARKWARE_NUM_SCREENS[context->selectorIndex];
|
|
msg->result = ETH_PLUGIN_RESULT_OK;
|
|
}
|
|
}
|
|
break;
|
|
|
|
case ETH_PLUGIN_QUERY_CONTRACT_ID: {
|
|
ethQueryContractID_t *msg = (ethQueryContractID_t*)parameters;
|
|
starkware_parameters_t *context = (starkware_parameters_t*)msg->pluginContext;
|
|
PRINTF("starkware query contract id\n");
|
|
switch(context->selectorIndex) {
|
|
case STARKWARE_REGISTER:
|
|
strcpy(msg->name, "Register");
|
|
break;
|
|
case STARKWARE_DEPOSIT_TOKEN:
|
|
case STARKWARE_DEPOSIT_ETH:
|
|
strcpy(msg->name, "Deposit");
|
|
break;
|
|
case STARKWARE_DEPOSIT_CANCEL:
|
|
strcpy(msg->name, "Cancel Deposit");
|
|
break;
|
|
case STARKWARE_DEPOSIT_RECLAIM:
|
|
strcpy(msg->name, "Reclaim Deposit");
|
|
break;
|
|
case STARKWARE_WITHDRAW:
|
|
strcpy(msg->name, "Withdrawal");
|
|
break;
|
|
case STARKWARE_FULL_WITHDRAW:
|
|
strcpy(msg->name, "Full Withdrawal");
|
|
break;
|
|
case STARKWARE_FREEZE:
|
|
strcpy(msg->name, "Freeze");
|
|
break;
|
|
case STARKWARE_ESCAPE:
|
|
strcpy(msg->name, "Escape");
|
|
break;
|
|
case STARKWARE_VERIFY_ESCAPE:
|
|
strcpy(msg->name, "Verify Escape");
|
|
break;
|
|
default:
|
|
break;
|
|
}
|
|
strcpy(msg->version, "Starkware");
|
|
msg->result = ETH_PLUGIN_RESULT_OK;
|
|
}
|
|
break;
|
|
|
|
case ETH_PLUGIN_QUERY_CONTRACT_UI: {
|
|
ethQueryContractUI_t *msg = (ethQueryContractUI_t*)parameters;
|
|
starkware_parameters_t *context = (starkware_parameters_t*)msg->pluginContext;
|
|
switch(msg->screenIndex) {
|
|
case 0:
|
|
strcpy(msg->title, "Contract Name");
|
|
starkware_print_eth_address(tmpContent.txContent.destination, msg->msg);
|
|
msg->result = ETH_PLUGIN_RESULT_OK;
|
|
break;
|
|
|
|
case 1:
|
|
switch(context->selectorIndex) {
|
|
case STARKWARE_REGISTER:
|
|
strcpy(msg->title, "From ETH Address");
|
|
starkware_get_source_address(msg->msg);
|
|
break;
|
|
case STARKWARE_DEPOSIT_TOKEN:
|
|
case STARKWARE_DEPOSIT_ETH:
|
|
case STARKWARE_DEPOSIT_CANCEL:
|
|
case STARKWARE_DEPOSIT_RECLAIM:
|
|
case STARKWARE_FULL_WITHDRAW:
|
|
case STARKWARE_FREEZE:
|
|
strcpy(msg->title, "Token Account");
|
|
starkware_print_vault_id(U4BE(context->vaultId, 0), msg->msg);
|
|
break;
|
|
case STARKWARE_WITHDRAW:
|
|
strcpy(msg->title, "To ETH Address");
|
|
starkware_get_source_address(msg->msg);
|
|
break;
|
|
case STARKWARE_ESCAPE:
|
|
strcpy(msg->title, "Amount");
|
|
starkware_print_amount(context->amount, msg->msg, true);
|
|
break;
|
|
default:
|
|
PRINTF("Unexpected screen %d for %d\n", msg->screenIndex, context->selectorIndex);
|
|
break;
|
|
}
|
|
msg->result = ETH_PLUGIN_RESULT_OK;
|
|
break;
|
|
|
|
case 2:
|
|
switch(context->selectorIndex) {
|
|
case STARKWARE_REGISTER:
|
|
case STARKWARE_ESCAPE:
|
|
strcpy(msg->title, "Master Account");
|
|
PRINTF("Master account %s\n", msg->msg);
|
|
starkware_print_stark_key(context->starkKey, msg->msg);
|
|
break;
|
|
case STARKWARE_DEPOSIT_TOKEN:
|
|
case STARKWARE_DEPOSIT_ETH:
|
|
strcpy(msg->title, "Amount");
|
|
starkware_print_amount(
|
|
(context->selectorIndex == STARKWARE_DEPOSIT_ETH ? NULL : context->amount),
|
|
msg->msg, false);
|
|
break;
|
|
case STARKWARE_WITHDRAW:
|
|
strcpy(msg->title, "Token Symbol");
|
|
starkware_print_ticker(msg->msg);
|
|
break;
|
|
default:
|
|
PRINTF("Unexpected screen %d for %d\n", msg->screenIndex, context->selectorIndex);
|
|
break;
|
|
}
|
|
msg->result = ETH_PLUGIN_RESULT_OK;
|
|
break;
|
|
|
|
case 3:
|
|
switch(context->selectorIndex) {
|
|
case STARKWARE_ESCAPE:
|
|
strcpy(msg->title, "Token Account");
|
|
starkware_print_vault_id(U4BE(context->vaultId, 0), msg->msg);
|
|
break;
|
|
default:
|
|
PRINTF("Unexpected screen %d for %d\n", msg->screenIndex, context->selectorIndex);
|
|
break;
|
|
}
|
|
msg->result = ETH_PLUGIN_RESULT_OK;
|
|
break;
|
|
|
|
default:
|
|
PRINTF("Unexpected screen %d for %d\n", msg->screenIndex, context->selectorIndex);
|
|
break;
|
|
}
|
|
}
|
|
break;
|
|
|
|
default:
|
|
PRINTF("Unhandled message %d\n", message);
|
|
}
|
|
}
|