Momentum-Apps/totp_app.c

#include <gui/gui.h>
#include <input/input.h>
#include <dialogs/dialogs.h>
#include <stdlib.h>
#include <notification/notification.h>
#include <notification/notification_messages.h>
#include <dolphin/dolphin.h>
#include "config/app/config.h"
#include "services/config/config.h"
#include "types/plugin_state.h"
#include "types/token_info.h"
#include "types/plugin_event.h"
#include "types/event_type.h"
#include "types/common.h"
#include "ui/scene_director.h"
#include "ui/constants.h"
#include "ui/common_dialogs.h"
#include "services/crypto/crypto_facade.h"
#include <toolbox/cli/cli_command.h>
#include <cli/cli_main_commands.h>
#include "version.h"
#include <wolfssl/version.h>

struct TotpRenderCallbackContext {
    FuriMutex* mutex;
    PluginState* plugin_state;
};

static void render_callback(Canvas* const canvas, void* const ctx) {
    furi_assert(ctx);
    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* 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 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;
    }

    if(dialog_result == DialogMessageButtonRight) {
        totp_scene_director_activate_scene(plugin_state, TotpSceneAuthentication);
    } else {
        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;
        }

        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;
        }
    }

    return true;
}

static bool on_user_idle(void* context) {
    PluginState* plugin_state = 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;
    }

    return false;
}

static bool totp_plugin_state_init(PluginState* const plugin_state) {
    plugin_state->gui = furi_record_open(RECORD_GUI);
    plugin_state->dialogs_app = furi_record_open(RECORD_DIALOGS);
    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->event_queue = furi_message_queue_alloc(8, sizeof(PluginEvent));

#ifdef TOTP_BADBT_AUTOMATION_ENABLED
    if(plugin_state->automation_method & AutomationMethodBadBt) {
        plugin_state->bt_type_code_worker_context = totp_bt_type_code_worker_init(
            *((uint16_t*)plugin_state->crypto_settings.crypto_verify_data),
            plugin_state->bt_type_code_worker_profile_index);
    } else {
        plugin_state->bt_type_code_worker_context = NULL;
    }
#endif

    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);
    } else {
        plugin_state->idle_timeout_context = NULL;
    }

    return true;
}

static void totp_plugin_state_free(PluginState* plugin_state) {
    if(plugin_state->idle_timeout_context != NULL) {
        idle_timeout_stop(plugin_state->idle_timeout_context);
        idle_timeout_free(plugin_state->idle_timeout_context);
    }

    furi_record_close(RECORD_GUI);
    furi_record_close(RECORD_DIALOGS);

    totp_config_file_close(plugin_state);

    if(plugin_state->crypto_settings.crypto_verify_data != NULL) {
        free(plugin_state->crypto_settings.crypto_verify_data);
    }

#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

    if(plugin_state->event_queue != NULL) {
        furi_message_queue_free(plugin_state->event_queue);
    }

    free(plugin_state);
}

int32_t totp_app() {
    FURI_LOG_I(
        LOGGING_TAG,
        "App version: %" PRIu8 ".%" PRIu8 ".%" PRIu8,
        TOTP_APP_VERSION_MAJOR,
        TOTP_APP_VERSION_MINOR,
        TOTP_APP_VERSION_PATCH);
    FURI_LOG_I(LOGGING_TAG, "WolfSSL version: " LIBWOLFSSL_VERSION_STRING);

    PluginState* plugin_state = malloc(sizeof(PluginState));
    furi_check(plugin_state != NULL);

    if(!totp_plugin_state_init(plugin_state)) {
        FURI_LOG_E(LOGGING_TAG, "App state initialization failed\r\n");
        totp_plugin_state_free(plugin_state);
        return 254;
    }

    if(!totp_activate_initial_scene(plugin_state)) {
        FURI_LOG_E(LOGGING_TAG, "An error ocurred during activating initial scene\r\n");
        totp_plugin_state_free(plugin_state);
        return 253;
    }

    TotpCliContext* cli_context = totp_cli_register_command_handler(plugin_state);

    // 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, &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);

    PluginEvent event;
    bool processing = true;
    while(processing) {
        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);
                }

                processing = totp_scene_director_handle_event(&event, plugin_state);

                furi_mutex_release(main_loop_mutex);
            }
        }

        view_port_update(view_port);
    }

    totp_cli_unregister_command_handler(cli_context);
    totp_scene_director_deactivate_active_scene(plugin_state);

    view_port_enabled_set(view_port, false);
    gui_remove_view_port(plugin_state->gui, view_port);
    view_port_free(view_port);
    furi_mutex_free(main_loop_mutex);
    totp_plugin_state_free(plugin_state);
    return 0;
}