Mirror/Momentum-Firmware
1
0
Fork 0
mirror of https://github.com/Next-Flip/Momentum-Firmware.git synced 2026-05-14 21:38:35 -07:00
Code Issues Packages Projects Releases Wiki Activity

Update TOTP

Browse Source 
https://github.com/akopachov/flipper-zero_authenticator
This commit is contained in:
MX
2023-08-05 13:57:55 +03:00
parent 74ffb02b56
commit afa7bd7f79
57 changed files with 1521 additions and 577 deletions
Download Patch File Download Diff File Expand all files Collapse all files

216
applications/external/totp/totp_app.c vendored
View File

@@ -15,60 +15,53 @@
#include "ui/scene_director.h"
#include "ui/constants.h"
#include "ui/common_dialogs.h"
#include "services/crypto/crypto.h"
#include "services/crypto/crypto_facade.h"
#include "cli/cli.h"
#include "version.h"
static void render_callback(Canvas* const canvas, void* ctx) {
struct TotpRenderCallbackContext {
FuriMutex* mutex;
PluginState* plugin_state;
};
static void render_callback(Canvas* const canvas, void* const ctx) {
furi_assert(ctx);
PluginState* plugin_state = ctx;
if(furi_mutex_acquire(plugin_state->mutex, 25) == FuriStatusOk) {
totp_scene_director_render(canvas, plugin_state);
furi_mutex_release(plugin_state->mutex);
const struct TotpRenderCallbackContext* context = ctx;
if(furi_mutex_acquire(context->mutex, 25) == FuriStatusOk) {
totp_scene_director_render(canvas, context->plugin_state);
furi_mutex_release(context->mutex);
}
}
static void input_callback(InputEvent* input_event, FuriMessageQueue* event_queue) {
furi_assert(event_queue);
static void input_callback(InputEvent* const input_event, void* const ctx) {
furi_assert(ctx);
FuriMessageQueue* event_queue = ctx;
PluginEvent event = {.type = EventTypeKey, .input = *input_event};
furi_message_queue_put(event_queue, &event, FuriWaitForever);
}
static bool totp_activate_initial_scene(PluginState* const plugin_state) {
if(plugin_state->crypto_verify_data == NULL) {
DialogMessage* message = dialog_message_alloc();
dialog_message_set_buttons(message, "No", NULL, "Yes");
dialog_message_set_text(
message,
"Would you like to setup PIN?",
SCREEN_WIDTH_CENTER,
SCREEN_HEIGHT_CENTER,
AlignCenter,
AlignCenter);
DialogMessageButton dialog_result =
dialog_message_show(plugin_state->dialogs_app, message);
dialog_message_free(message);
if(dialog_result == DialogMessageButtonRight) {
totp_scene_director_activate_scene(plugin_state, TotpSceneAuthentication);
} else {
CryptoSeedIVResult seed_result = totp_crypto_seed_iv(plugin_state, NULL, 0);
if(seed_result & CryptoSeedIVResultFlagSuccess &&
seed_result & CryptoSeedIVResultFlagNewCryptoVerifyData) {
if(!totp_config_file_update_crypto_signatures(plugin_state)) {
totp_dialogs_config_loading_error(plugin_state);
return false;
}
} else if(seed_result == CryptoSeedIVResultFailed) {
totp_dialogs_config_loading_error(plugin_state);
return false;
}
static bool first_run_init(PluginState* const plugin_state) {
DialogMessage* message = dialog_message_alloc();
dialog_message_set_buttons(message, "No", NULL, "Yes");
dialog_message_set_text(
message,
"Would you like to setup PIN?",
SCREEN_WIDTH_CENTER,
SCREEN_HEIGHT_CENTER,
AlignCenter,
AlignCenter);
DialogMessageButton dialog_result = dialog_message_show(plugin_state->dialogs_app, message);
dialog_message_free(message);
if(!totp_crypto_check_key_slot(plugin_state->crypto_settings.crypto_key_slot)) {
totp_dialogs_config_loading_error(plugin_state);
return false;
}
totp_scene_director_activate_scene(plugin_state, TotpSceneGenerateToken);
}
} else if(plugin_state->pin_set) {
if(dialog_result == DialogMessageButtonRight) {
totp_scene_director_activate_scene(plugin_state, TotpSceneAuthentication);
} else {
CryptoSeedIVResult seed_result = totp_crypto_seed_iv(plugin_state, NULL, 0);
CryptoSeedIVResult seed_result =
totp_crypto_seed_iv(&plugin_state->crypto_settings, NULL, 0);
if(seed_result & CryptoSeedIVResultFlagSuccess &&
seed_result & CryptoSeedIVResultFlagNewCryptoVerifyData) {
if(!totp_config_file_update_crypto_signatures(plugin_state)) {
@@ -80,23 +73,65 @@ static bool totp_activate_initial_scene(PluginState* const plugin_state) {
return false;
}
if(totp_crypto_verify_key(plugin_state)) {
totp_scene_director_activate_scene(plugin_state, TotpSceneGenerateToken);
} else {
FURI_LOG_E(
LOGGING_TAG,
"Digital signature verification failed. Looks like conf file was created on another flipper and can't be used on any other");
DialogMessage* message = dialog_message_alloc();
dialog_message_set_buttons(message, "Exit", NULL, NULL);
dialog_message_set_text(
message,
"Digital signature verification failed",
SCREEN_WIDTH_CENTER,
SCREEN_HEIGHT_CENTER,
AlignCenter,
AlignCenter);
dialog_message_show(plugin_state->dialogs_app, message);
dialog_message_free(message);
totp_scene_director_activate_scene(plugin_state, TotpSceneGenerateToken);
}
return true;
}
static bool pinless_activation(PluginState* const plugin_state) {
CryptoSeedIVResult seed_result = totp_crypto_seed_iv(&plugin_state->crypto_settings, NULL, 0);
if(seed_result & CryptoSeedIVResultFlagSuccess &&
seed_result & CryptoSeedIVResultFlagNewCryptoVerifyData) {
if(!totp_config_file_update_crypto_signatures(plugin_state)) {
totp_dialogs_config_loading_error(plugin_state);
return false;
}
} else if(seed_result == CryptoSeedIVResultFailed) {
totp_dialogs_config_loading_error(plugin_state);
return false;
}
if(totp_crypto_verify_key(&plugin_state->crypto_settings)) {
totp_config_file_ensure_latest_encryption(plugin_state, NULL, 0);
totp_scene_director_activate_scene(plugin_state, TotpSceneGenerateToken);
} else {
FURI_LOG_E(
LOGGING_TAG,
"Digital signature verification failed. Looks like conf file was created on another device and can't be used on any other");
DialogMessage* message = dialog_message_alloc();
dialog_message_set_buttons(message, "Exit", NULL, NULL);
dialog_message_set_text(
message,
"Digital signature verification failed",
SCREEN_WIDTH_CENTER,
SCREEN_HEIGHT_CENTER,
AlignCenter,
AlignCenter);
dialog_message_show(plugin_state->dialogs_app, message);
dialog_message_free(message);
return false;
}
return true;
}
static bool pin_activation(PluginState* const plugin_state) {
totp_scene_director_activate_scene(plugin_state, TotpSceneAuthentication);
return true;
}
static bool totp_activate_initial_scene(PluginState* const plugin_state) {
if(plugin_state->crypto_settings.crypto_verify_data == NULL) {
if(!first_run_init(plugin_state)) {
return false;
}
} else if(plugin_state->crypto_settings.pin_required) {
if(!pin_activation(plugin_state)) {
return false;
}
} else {
if(!pinless_activation(plugin_state)) {
return false;
}
}
@@ -109,6 +144,7 @@ static bool on_user_idle(void* context) {
if(plugin_state->current_scene != TotpSceneAuthentication &&
plugin_state->current_scene != TotpSceneStandby) {
totp_scene_director_activate_scene(plugin_state, TotpSceneAuthentication);
totp_scene_director_force_redraw(plugin_state);
return true;
}
@@ -117,18 +153,17 @@ static bool on_user_idle(void* context) {
static bool totp_plugin_state_init(PluginState* const plugin_state) {
plugin_state->gui = furi_record_open(RECORD_GUI);
plugin_state->notification_app = furi_record_open(RECORD_NOTIFICATION);
plugin_state->dialogs_app = furi_record_open(RECORD_DIALOGS);
memset(&plugin_state->iv[0], 0, TOTP_IV_SIZE);
memset(&plugin_state->crypto_settings.iv[0], 0, CRYPTO_IV_LENGTH);
if(!totp_config_file_load(plugin_state)) {
totp_dialogs_config_loading_error(plugin_state);
return false;
}
plugin_state->mutex = furi_mutex_alloc(FuriMutexTypeNormal);
plugin_state->event_queue = furi_message_queue_alloc(8, sizeof(PluginEvent));
#ifdef TOTP_BADBT_TYPE_ENABLED
#ifdef TOTP_BADBT_AUTOMATION_ENABLED
if(plugin_state->automation_method & AutomationMethodBadBt) {
plugin_state->bt_type_code_worker_context = totp_bt_type_code_worker_init();
} else {
@@ -136,7 +171,7 @@ static bool totp_plugin_state_init(PluginState* const plugin_state) {
}
#endif
if(plugin_state->pin_set) {
if(plugin_state->crypto_settings.pin_required) {
plugin_state->idle_timeout_context =
idle_timeout_alloc(TOTP_AUTO_LOCK_IDLE_TIMEOUT_SEC, &on_user_idle, plugin_state);
idle_timeout_start(plugin_state->idle_timeout_context);
@@ -154,28 +189,35 @@ static void totp_plugin_state_free(PluginState* plugin_state) {
}
furi_record_close(RECORD_GUI);
furi_record_close(RECORD_NOTIFICATION);
furi_record_close(RECORD_DIALOGS);
totp_config_file_close(plugin_state);
if(plugin_state->crypto_verify_data != NULL) {
free(plugin_state->crypto_verify_data);
if(plugin_state->crypto_settings.crypto_verify_data != NULL) {
free(plugin_state->crypto_settings.crypto_verify_data);
}
#ifdef TOTP_BADBT_TYPE_ENABLED
#ifdef TOTP_BADBT_AUTOMATION_ENABLED
if(plugin_state->bt_type_code_worker_context != NULL) {
totp_bt_type_code_worker_free(plugin_state->bt_type_code_worker_context);
plugin_state->bt_type_code_worker_context = NULL;
}
#endif
furi_mutex_free(plugin_state->mutex);
if(plugin_state->event_queue != NULL) {
furi_message_queue_free(plugin_state->event_queue);
}
free(plugin_state);
}
int32_t totp_app() {
FuriMessageQueue* event_queue = furi_message_queue_alloc(8, sizeof(PluginEvent));
FURI_LOG_I(
LOGGING_TAG,
"App version: %" PRIu8 ".%" PRIu8 ".%" PRIu8,
TOTP_APP_VERSION_MAJOR,
TOTP_APP_VERSION_MINOR,
TOTP_APP_VERSION_PATCH);
PluginState* plugin_state = malloc(sizeof(PluginState));
furi_check(plugin_state != NULL);
@@ -185,7 +227,7 @@ int32_t totp_app() {
return 254;
}
TotpCliContext* cli_context = totp_cli_register_command_handler(plugin_state, event_queue);
TotpCliContext* cli_context = totp_cli_register_command_handler(plugin_state);
if(!totp_activate_initial_scene(plugin_state)) {
FURI_LOG_E(LOGGING_TAG, "An error ocurred during activating initial scene\r\n");
@@ -196,10 +238,14 @@ int32_t totp_app() {
// Affecting dolphin level
dolphin_deed(DolphinDeedPluginStart);
FuriMutex* main_loop_mutex = furi_mutex_alloc(FuriMutexTypeNormal);
struct TotpRenderCallbackContext render_context = {
.plugin_state = plugin_state, .mutex = main_loop_mutex};
// Set system callbacks
ViewPort* view_port = view_port_alloc();
view_port_draw_callback_set(view_port, render_callback, plugin_state);
view_port_input_callback_set(view_port, input_callback, event_queue);
view_port_draw_callback_set(view_port, render_callback, &render_context);
view_port_input_callback_set(view_port, input_callback, plugin_state->event_queue);
// Open GUI and register view_port
gui_add_view_port(plugin_state->gui, view_port, GuiLayerFullscreen);
@@ -207,24 +253,24 @@ int32_t totp_app() {
PluginEvent event;
bool processing = true;
while(processing) {
FuriStatus event_status = furi_message_queue_get(event_queue, &event, 100);
if(furi_mutex_acquire(plugin_state->mutex, FuriWaitForever) == FuriStatusOk) {
if(event_status == FuriStatusOk) {
if(furi_message_queue_get(plugin_state->event_queue, &event, FuriWaitForever) ==
FuriStatusOk) {
if(event.type == EventForceCloseApp) {
processing = false;
} else if(event.type == EventForceRedraw) {
processing = true; //-V1048
} else if(furi_mutex_acquire(main_loop_mutex, FuriWaitForever) == FuriStatusOk) {
if(event.type == EventTypeKey && plugin_state->idle_timeout_context != NULL) {
idle_timeout_report_activity(plugin_state->idle_timeout_context);
}
if(event.type == EventForceCloseApp) {
processing = false;
} else {
processing = totp_scene_director_handle_event(&event, plugin_state);
}
}
processing = totp_scene_director_handle_event(&event, plugin_state);
view_port_update(view_port);
furi_mutex_release(plugin_state->mutex);
furi_mutex_release(main_loop_mutex);
}
}
view_port_update(view_port);
}
totp_cli_unregister_command_handler(cli_context);
@@ -233,7 +279,7 @@ int32_t totp_app() {
view_port_enabled_set(view_port, false);
gui_remove_view_port(plugin_state->gui, view_port);
view_port_free(view_port);
furi_message_queue_free(event_queue);
furi_mutex_free(main_loop_mutex);
totp_plugin_state_free(plugin_state);
return 0;
}