#include "totp_scene_add_new_token.h" #include "../../types/common.h" #include "../../services/ui/constants.h" #include "../scene_director.h" #include "totp_input_text.h" #include "../../types/token_info.h" #include "../../services/list/list.h" #include "../../services/base32/base32.h" #include "../../services/config/config.h" #include "../../services/ui/ui_controls.h" #include "../generate_token/totp_scene_generate_token.h" #define TOKEN_ALGO_LIST_LENGTH 3 char* TOKEN_ALGO_LIST[] = {"SHA1", "SHA256", "SHA512"}; #define TOKEN_DIGITS_LIST_LENGTH 2 char* TOKEN_DIGITS_LIST[] = {"6 digits", "8 digits"}; typedef enum { TokenNameTextBox, TokenSecretTextBox, TokenAlgoSelect, TokenLengthSelect, ConfirmButton, } Control; typedef struct { char* token_name; size_t token_name_length; char* token_secret; size_t token_secret_length; bool saved; Control selected_control; InputTextSceneContext* token_name_input_context; InputTextSceneContext* token_secret_input_context; InputTextSceneState* input_state; uint32_t input_started_at; int16_t current_token_index; int16_t screen_y_offset; TokenHashAlgo algo; TokenDigitsCount digits_count; } SceneState; void totp_scene_add_new_token_init(const PluginState* plugin_state) { UNUSED(plugin_state); } static void on_token_name_user_comitted(InputTextSceneCallbackResult* result) { SceneState* scene_state = result->callback_data; free(scene_state->token_name); scene_state->token_name = result->user_input; scene_state->token_name_length = result->user_input_length; scene_state->input_started_at = 0; free(result); } static void on_token_secret_user_comitted(InputTextSceneCallbackResult* result) { SceneState* scene_state = result->callback_data; free(scene_state->token_secret); scene_state->token_secret = result->user_input; scene_state->token_secret_length = result->user_input_length; scene_state->input_started_at = 0; free(result); } void totp_scene_add_new_token_activate( PluginState* plugin_state, const TokenAddEditSceneContext* context) { SceneState* scene_state = malloc(sizeof(SceneState)); plugin_state->current_scene_state = scene_state; scene_state->token_name = "Name"; scene_state->token_name_length = strlen(scene_state->token_name); scene_state->token_secret = "Secret"; scene_state->token_secret_length = strlen(scene_state->token_secret); scene_state->token_name_input_context = malloc(sizeof(InputTextSceneContext)); scene_state->token_name_input_context->header_text = "Enter token name"; scene_state->token_name_input_context->callback_data = scene_state; scene_state->token_name_input_context->callback = on_token_name_user_comitted; scene_state->token_secret_input_context = malloc(sizeof(InputTextSceneContext)); scene_state->token_secret_input_context->header_text = "Enter token secret"; scene_state->token_secret_input_context->callback_data = scene_state; scene_state->token_secret_input_context->callback = on_token_secret_user_comitted; scene_state->screen_y_offset = 0; scene_state->input_state = NULL; if(context == NULL) { scene_state->current_token_index = -1; } else { scene_state->current_token_index = context->current_token_index; } } void totp_scene_add_new_token_render(Canvas* const canvas, PluginState* plugin_state) { SceneState* scene_state = (SceneState*)plugin_state->current_scene_state; if(scene_state->input_started_at > 0) { totp_input_text_render(canvas, scene_state->input_state); return; } ui_control_text_box_render( canvas, 10 - scene_state->screen_y_offset, scene_state->token_name, scene_state->selected_control == TokenNameTextBox); ui_control_text_box_render( canvas, 27 - scene_state->screen_y_offset, scene_state->token_secret, scene_state->selected_control == TokenSecretTextBox); ui_control_select_render( canvas, 0, 44 - scene_state->screen_y_offset, SCREEN_WIDTH, TOKEN_ALGO_LIST[scene_state->algo], scene_state->selected_control == TokenAlgoSelect); ui_control_select_render( canvas, 0, 63 - scene_state->screen_y_offset, SCREEN_WIDTH, TOKEN_DIGITS_LIST[scene_state->digits_count], scene_state->selected_control == TokenLengthSelect); ui_control_button_render( canvas, SCREEN_WIDTH_CENTER - 24, 85 - scene_state->screen_y_offset, 48, 13, "Confirm", scene_state->selected_control == ConfirmButton); canvas_set_color(canvas, ColorWhite); canvas_draw_box(canvas, 0, 0, SCREEN_WIDTH, 10); canvas_set_color(canvas, ColorBlack); canvas_set_font(canvas, FontPrimary); canvas_draw_str_aligned(canvas, 0, 0, AlignLeft, AlignTop, "Add new token"); canvas_set_font(canvas, FontSecondary); } void update_screen_y_offset(SceneState* scene_state) { if(scene_state->selected_control > TokenAlgoSelect) { scene_state->screen_y_offset = 35; } else { scene_state->screen_y_offset = 0; } } bool totp_scene_add_new_token_handle_event(PluginEvent* const event, PluginState* plugin_state) { if(event->type == EventTypeKey) { SceneState* scene_state = (SceneState*)plugin_state->current_scene_state; if(scene_state->input_started_at > 0 && furi_get_tick() - scene_state->input_started_at > 300) { return totp_input_text_handle_event(event, scene_state->input_state); } if(event->input.type == InputTypeLong && event->input.key == InputKeyBack) { return false; } else if(event->input.type == InputTypePress) { switch(event->input.key) { case InputKeyUp: if(scene_state->selected_control > TokenNameTextBox) { scene_state->selected_control--; update_screen_y_offset(scene_state); } break; case InputKeyDown: if(scene_state->selected_control < ConfirmButton) { scene_state->selected_control++; update_screen_y_offset(scene_state); } break; case InputKeyRight: if(scene_state->selected_control == TokenAlgoSelect) { if(scene_state->algo < SHA512) { scene_state->algo++; } else { scene_state->algo = SHA1; } } else if(scene_state->selected_control == TokenLengthSelect) { if(scene_state->digits_count < TOTP_8_DIGITS) { scene_state->digits_count++; } else { scene_state->digits_count = TOTP_6_DIGITS; } } break; case InputKeyLeft: if(scene_state->selected_control == TokenAlgoSelect) { if(scene_state->algo > SHA1) { scene_state->algo--; } else { scene_state->algo = SHA512; } } else if(scene_state->selected_control == TokenLengthSelect) { if(scene_state->digits_count > TOTP_6_DIGITS) { scene_state->digits_count--; } else { scene_state->digits_count = TOTP_8_DIGITS; } } break; case InputKeyOk: switch(scene_state->selected_control) { case TokenNameTextBox: if(scene_state->input_state != NULL) { totp_input_text_free(scene_state->input_state); } scene_state->input_state = totp_input_text_activate(scene_state->token_name_input_context); scene_state->input_started_at = furi_get_tick(); break; case TokenSecretTextBox: if(scene_state->input_state != NULL) { totp_input_text_free(scene_state->input_state); } scene_state->input_state = totp_input_text_activate(scene_state->token_secret_input_context); scene_state->input_started_at = furi_get_tick(); break; case TokenAlgoSelect: break; case TokenLengthSelect: break; case ConfirmButton: { TokenInfo* tokenInfo = token_info_alloc(); bool token_secret_set = token_info_set_secret( tokenInfo, scene_state->token_secret, scene_state->token_secret_length, &plugin_state->iv[0]); if(token_secret_set) { tokenInfo->name = malloc(scene_state->token_name_length + 1); strlcpy( tokenInfo->name, scene_state->token_name, scene_state->token_name_length + 1); tokenInfo->algo = scene_state->algo; tokenInfo->digits = scene_state->digits_count; if(plugin_state->tokens_list == NULL) { plugin_state->tokens_list = list_init_head(tokenInfo); } else { list_add(plugin_state->tokens_list, tokenInfo); } plugin_state->tokens_count++; totp_config_file_save_new_token(tokenInfo); GenerateTokenSceneContext generate_scene_context = { .current_token_index = plugin_state->tokens_count - 1}; totp_scene_director_activate_scene( plugin_state, TotpSceneGenerateToken, &generate_scene_context); } else { token_info_free(tokenInfo); DialogMessage* message = dialog_message_alloc(); dialog_message_set_buttons(message, "Back", NULL, NULL); dialog_message_set_text( message, "Token secret is invalid", SCREEN_WIDTH_CENTER, SCREEN_HEIGHT_CENTER, AlignCenter, AlignCenter); dialog_message_show(plugin_state->dialogs, message); dialog_message_free(message); scene_state->selected_control = TokenSecretTextBox; update_screen_y_offset(scene_state); } break; } } break; case InputKeyBack: if(scene_state->current_token_index >= 0) { GenerateTokenSceneContext generate_scene_context = { .current_token_index = scene_state->current_token_index}; totp_scene_director_activate_scene( plugin_state, TotpSceneGenerateToken, &generate_scene_context); } else { totp_scene_director_activate_scene(plugin_state, TotpSceneGenerateToken, NULL); } break; } } } return true; } void totp_scene_add_new_token_deactivate(PluginState* plugin_state) { if(plugin_state->current_scene_state == NULL) return; SceneState* scene_state = (SceneState*)plugin_state->current_scene_state; free(scene_state->token_name); free(scene_state->token_secret); free(scene_state->token_name_input_context->header_text); free(scene_state->token_name_input_context); free(scene_state->token_secret_input_context->header_text); free(scene_state->token_secret_input_context); if(scene_state->input_state != NULL) { totp_input_text_free(scene_state->input_state); } free(plugin_state->current_scene_state); plugin_state->current_scene_state = NULL; } void totp_scene_add_new_token_free(const PluginState* plugin_state) { UNUSED(plugin_state); }