void recovery_word(const char *word) { if (!awaiting_word) { fsm_sendFailure(FailureType_Failure_UnexpectedMessage, "Not in Recovery mode"); layoutHome(); return; } if (word_pos == 0) { // fake word if (strcmp(word, fake_word) != 0) { storage_reset(); fsm_sendFailure(FailureType_Failure_SyntaxError, "Wrong word retyped"); layoutHome(); return; } } else { // real word if (enforce_wordlist) { // check if word is valid const char * const *wl = mnemonic_wordlist(); bool found = false; while (*wl) { if (strcmp(word, *wl) == 0) { found = true; break; } wl++; } if (!found) { storage_reset(); fsm_sendFailure(FailureType_Failure_SyntaxError, "Word not found in a wordlist"); layoutHome(); return; } } strlcpy(words[word_pos - 1], word, sizeof(words[word_pos - 1])); } if (word_index + 1 == 24) { // last one uint32_t i; strlcpy(storage.mnemonic, words[0], sizeof(storage.mnemonic)); for (i = 1; i < word_count; i++) { strlcat(storage.mnemonic, " ", sizeof(storage.mnemonic)); strlcat(storage.mnemonic, words[i], sizeof(storage.mnemonic)); } if (!enforce_wordlist || mnemonic_check(storage.mnemonic)) { storage.has_mnemonic = true; storage_commit(); fsm_sendSuccess("Device recovered"); } else { storage_reset(); fsm_sendFailure(FailureType_Failure_SyntaxError, "Invalid mnemonic, are words in correct order?"); } awaiting_word = false; layoutHome(); } else { word_index++; next_word(); } }
static uint8 at_sreset(char *ret_buf) { struct ther_info *ti = get_ti(); if (ti->mode != CAL_MODE) enter_cal_mode(ti); storage_reset(); return 0; }
/* * storage_load_device() - Load configuration data from usb message to shadow memory * * INPUT * - msg: load device message * OUTPUT * none */ void storage_load_device(LoadDevice *msg) { storage_reset(); shadow_config.storage.has_imported = true; shadow_config.storage.imported = true; if(msg->has_pin > 0) { storage_set_pin(msg->pin); } if(msg->has_passphrase_protection) { shadow_config.storage.has_passphrase_protection = true; shadow_config.storage.passphrase_protection = msg->passphrase_protection; } else { shadow_config.storage.has_passphrase_protection = false; } if(msg->has_node) { shadow_config.storage.has_node = true; shadow_config.storage.has_mnemonic = false; memcpy(&shadow_config.storage.node, &(msg->node), sizeof(HDNodeType)); sessionRootNodeCached = false; memset(&sessionRootNode, 0, sizeof(sessionRootNode)); } else if(msg->has_mnemonic) { shadow_config.storage.has_mnemonic = true; shadow_config.storage.has_node = false; strlcpy(shadow_config.storage.mnemonic, msg->mnemonic, sizeof(shadow_config.storage.mnemonic)); sessionRootNodeCached = false; memset(&sessionRootNode, 0, sizeof(sessionRootNode)); } if(msg->has_language) { storage_set_language(msg->language); } if(msg->has_label) { storage_set_label(msg->label); } }
/* * storage_init() - Validate storage content and copy data to shadow memory * * INPUT * none * OUTPUT * none */ void storage_init(void) { ConfigFlash *stor_config; /* Find storage sector with valid data and set storage_location variable */ if(find_active_storage(&storage_location)) { stor_config = (ConfigFlash *)flash_write_helper(storage_location); } else { /* Set to storage sector1 as default if no sector has been initialized */ storage_location = STORAGE_SECT_DEFAULT; stor_config = (ConfigFlash *)flash_write_helper(storage_location); } /* Reset shadow configuration in RAM */ storage_reset(); /* Verify storage partition is initialized */ if(memcmp((void *)stor_config->meta.magic , STORAGE_MAGIC_STR, STORAGE_MAGIC_LEN) == 0) { /* Clear out stor_config before finding end config node */ memcpy(shadow_config.meta.uuid, (void *)&stor_config->meta.uuid, sizeof(shadow_config.meta.uuid)); data2hex(shadow_config.meta.uuid, sizeof(shadow_config.meta.uuid), shadow_config.meta.uuid_str); if(stor_config->storage.version) { if(stor_config->storage.version <= STORAGE_VERSION) { storage_from_flash(stor_config); } } /* New app with storage version changed! update the storage space */ if(stor_config->storage.version != STORAGE_VERSION) { storage_commit(); } } else { /* Keep storage area cleared */ storage_reset_uuid(); storage_commit(); } }
void fsm_msgWipeDevice(WipeDevice *msg) { (void)msg; layoutDialogSwipe(DIALOG_ICON_QUESTION, "Cancel", "Confirm", NULL, "Do you really want to", "wipe the device?", NULL, "All data will be lost.", NULL, NULL); if (!protectButton(ButtonRequestType_ButtonRequest_WipeDevice, false)) { fsm_sendFailure(FailureType_Failure_ActionCancelled, "Wipe cancelled"); layoutHome(); return; } storage_reset(); storage_reset_uuid(); storage_commit(); // the following does not work on Mac anyway :-/ Linux/Windows are fine, so it is not needed // usbReconnect(); // force re-enumeration because of the serial number change fsm_sendSuccess("Device wiped"); layoutHome(); }
void fsm_msgWipeDevice(WipeDevice *msg) { (void)msg; if(!confirm(ButtonRequestType_ButtonRequest_WipeDevice, "Wipe Device", "Do you want to erase your private keys and settings?")) { fsm_sendFailure(FailureType_Failure_ActionCancelled, "Wipe cancelled"); go_home(); return; } /* Wipe device */ storage_reset(); storage_reset_uuid(); storage_commit(); fsm_sendSuccess("Device wiped"); go_home(); }
void storage_init(void) { storage_reset(); // if magic is ok if (memcmp((void *)FLASH_STORAGE_START, "stor", 4) == 0) { // load uuid memcpy(storage_uuid, (void *)(FLASH_STORAGE_START + 4), sizeof(storage_uuid)); data2hex(storage_uuid, sizeof(storage_uuid), storage_uuid_str); // load storage struct uint32_t version = ((Storage *)(FLASH_STORAGE_START + 4 + sizeof(storage_uuid)))->version; if (version && version <= STORAGE_VERSION) { storage_from_flash(version); } if (version != STORAGE_VERSION) { storage_commit(); } } else { storage_reset_uuid(); storage_commit(); } }
void reset_entropy(const uint8_t *ext_entropy, uint32_t len) { if(!awaiting_entropy) { fsm_sendFailure(FailureType_Failure_UnexpectedMessage, "Not in Reset mode"); return; } SHA256_CTX ctx; sha256_Init(&ctx); sha256_Update(&ctx, int_entropy, 32); sha256_Update(&ctx, ext_entropy, len); sha256_Final(int_entropy, &ctx); const char *temp_mnemonic = mnemonic_from_data(int_entropy, strength / 8); memset(int_entropy, 0, 32); awaiting_entropy = false; /* * Format mnemonic for user review */ uint32_t word_count = 0, current_page = 0, page_count; char *tok; char tokened_mnemonic[TOKENED_MNEMONIC_BUF]; char mnemonic_by_screen[MAX_PAGES][MNEMONIC_BY_SCREEN_BUF]; char formatted_mnemonic[MAX_PAGES][FORMATTED_MNEMONIC_BUF]; char mnemonic_display[FORMATTED_MNEMONIC_BUF]; char formatted_word[MAX_WORD_LEN + ADDITIONAL_WORD_PAD]; strlcpy(tokened_mnemonic, temp_mnemonic, TOKENED_MNEMONIC_BUF); tok = strtok(tokened_mnemonic, " "); while(tok) { snprintf(formatted_word, MAX_WORD_LEN + ADDITIONAL_WORD_PAD, "%lu.%s", (unsigned long)(word_count + 1), tok); /* Check that we have enough room on display to show word */ snprintf(mnemonic_display, FORMATTED_MNEMONIC_BUF, "%s %s", formatted_mnemonic[current_page], formatted_word); if(calc_str_line(get_body_font(), mnemonic_display, BODY_WIDTH) > 3) { page_count++; current_page++; snprintf(mnemonic_display, FORMATTED_MNEMONIC_BUF, "%s %s", formatted_mnemonic[current_page], formatted_word); } strlcpy(formatted_mnemonic[current_page], mnemonic_display, FORMATTED_MNEMONIC_BUF); /* Save mnemonic for each screen */ if(strlen(mnemonic_by_screen[current_page]) == 0) { strlcpy(mnemonic_by_screen[current_page], tok, MNEMONIC_BY_SCREEN_BUF); } else { strlcat(mnemonic_by_screen[current_page], " ", MNEMONIC_BY_SCREEN_BUF); strlcat(mnemonic_by_screen[current_page], tok, MNEMONIC_BY_SCREEN_BUF); } tok = strtok(NULL, " "); word_count++; } /* Have user confirm mnemonic is sets of 12 words */ for(page_count = current_page + 1, current_page = 0; current_page < page_count; current_page++) { char title[MEDIUM_STR_BUF] = "Recovery Sentence"; /* make current screen mnemonic available via debuglink */ strlcpy(current_words, mnemonic_by_screen[current_page], MNEMONIC_BY_SCREEN_BUF); if(page_count > 1) { /* snprintf: 20 + 10 (%d) + 1 (NULL) = 31 */ snprintf(title, MEDIUM_STR_BUF, "Recovery Sentence %lu/%lu", current_page + 1, page_count); } if(!confirm(ButtonRequestType_ButtonRequest_ConfirmWord, title, "%s", formatted_mnemonic[current_page])) { fsm_sendFailure(FailureType_Failure_ActionCancelled, "Reset cancelled"); storage_reset(); go_home(); return; } } /* Save mnemonic */ storage_set_mnemonic(temp_mnemonic); storage_commit(); fsm_sendSuccess("Device reset"); go_home(); }
void recovery_word(const char *word) { if (!awaiting_word) { fsm_sendFailure(FailureType_Failure_UnexpectedMessage, "Not in Recovery mode"); go_home(); return; } if (word_pos == 0) { // fake word if (strcmp(word, fake_word) != 0) { storage_reset(); fsm_sendFailure(FailureType_Failure_SyntaxError, "Wrong word retyped"); go_home(); return; } } else { // real word if (enforce_wordlist) { // check if word is valid const char * const *wl = mnemonic_wordlist(); bool found = false; while (*wl) { if (strcmp(word, *wl) == 0) { found = true; break; } wl++; } if (!found) { storage_reset(); fsm_sendFailure(FailureType_Failure_SyntaxError, "Word not found in a wordlist"); go_home(); return; } } strlcpy(words[word_pos - 1], word, sizeof(words[word_pos - 1])); } if (word_index + 1 == 24) { // last one storage_set_mnemonic_from_words((const char (*)[])words, word_count); if (!enforce_wordlist || mnemonic_check(storage_get_shadow_mnemonic())) { storage_commit(); fsm_sendSuccess("Device recovered"); } else { storage_reset(); fsm_sendFailure(FailureType_Failure_SyntaxError, "Invalid mnemonic, are words in correct order?"); } awaiting_word = false; go_home(); } else { word_index++; next_word(); } }
void reset_entropy(const uint8_t *ext_entropy, uint32_t len) { if (!awaiting_entropy) { fsm_sendFailure(FailureType_Failure_UnexpectedMessage, "Not in Reset mode"); return; } SHA256_CTX ctx; sha256_Init(&ctx); sha256_Update(&ctx, int_entropy, 32); sha256_Update(&ctx, ext_entropy, len); sha256_Final(int_entropy, &ctx); strlcpy(storage.mnemonic, mnemonic_from_data(int_entropy, strength / 8), sizeof(storage.mnemonic)); memset(int_entropy, 0, 32); awaiting_entropy = false; int pass, word_pos, i = 0, j; for (pass = 0; pass < 2; pass++) { i = 0; for (word_pos = 1; word_pos <= (int)strength/32*3; word_pos++) { // copy current_word j = 0; while (storage.mnemonic[i] != ' ' && storage.mnemonic[i] != 0 && j + 1 < (int)sizeof(current_word)) { current_word[j] = storage.mnemonic[i]; i++; j++; } current_word[j] = 0; if (storage.mnemonic[i] != 0) i++; char desc[] = "##th word is:"; if (word_pos < 10) { desc[0] = ' '; } else { desc[0] = '0' + word_pos / 10; } desc[1] = '0' + word_pos % 10; if (word_pos == 1 || word_pos == 21) { desc[2] = 's'; desc[3] = 't'; } else if (word_pos == 2 || word_pos == 22) { desc[2] = 'n'; desc[3] = 'd'; } else if (word_pos == 3 || word_pos == 23) { desc[2] = 'r'; desc[3] = 'd'; } if (word_pos == (int)strength/32*3) { // last word if (pass == 1) { layoutDialogSwipe(DIALOG_ICON_INFO, NULL, "Finish", NULL, "Please check the seed", NULL, (word_pos < 10 ? desc + 1 : desc), NULL, current_word, NULL); } else { layoutDialogSwipe(DIALOG_ICON_INFO, NULL, "Again", NULL, "Write down the seed", NULL, (word_pos < 10 ? desc + 1 : desc), NULL, current_word, NULL); } } else { if (pass == 1) { layoutDialogSwipe(DIALOG_ICON_INFO, NULL, "Next", NULL, "Please check the seed", NULL, (word_pos < 10 ? desc + 1 : desc), NULL, current_word, NULL); } else { layoutDialogSwipe(DIALOG_ICON_INFO, NULL, "Next", NULL, "Write down the seed", NULL, (word_pos < 10 ? desc + 1 : desc), NULL, current_word, NULL); } } if (!protectButton(ButtonRequestType_ButtonRequest_ConfirmWord, true)) { storage_reset(); layoutHome(); return; } } } storage.has_mnemonic = true; storage_commit(); fsm_sendSuccess("Device reset"); layoutHome(); }