Momentum-Apps/multi_fuzzer/scenes/fuzzer_scene_main.c

#include "../fuzzer_i.h"
#include "../helpers/fuzzer_custom_event.h"

#include "../lib/worker/protocol.h"
#define TAG "Fuzzer main menu"

void fuzzer_scene_main_callback(FuzzerCustomEvent event, void* context) {
    furi_assert(context);
    PacsFuzzerApp* app = context;
    view_dispatcher_send_custom_event(app->view_dispatcher, event);
}

void fuzzer_scene_main_error_popup_callback(void* context) {
    PacsFuzzerApp* app = context;
    notification_message(app->notifications, &sequence_reset_rgb);
    view_dispatcher_send_custom_event(app->view_dispatcher, FuzzerCustomEventViewMainPopupErr);
}

static bool fuzzer_scene_main_load_custom_dict(void* context) {
    furi_assert(context);
    PacsFuzzerApp* app = context;

    FuzzerConsts* consts = app->fuzzer_const;

    furi_string_set_str(app->file_path, consts->custom_dict_folder);

    DialogsFileBrowserOptions browser_options;
    dialog_file_browser_set_basic_options(
        &browser_options, consts->custom_dict_extension, &I_rfid_10px);
    browser_options.base_path = consts->custom_dict_folder;
    browser_options.hide_ext = false;

    bool res =
        dialog_file_browser_show(app->dialogs, app->file_path, app->file_path, &browser_options);

    return res;
}

static bool fuzzer_scene_main_load_key_dialog(void* context) {
    furi_assert(context);
    PacsFuzzerApp* app = context;

    FuzzerConsts* consts = app->fuzzer_const;

    furi_string_set_str(app->file_path, consts->path_key_folder);

    DialogsFileBrowserOptions browser_options;
    dialog_file_browser_set_basic_options(
        &browser_options, consts->key_extension, consts->key_icon);
    browser_options.base_path = consts->path_key_folder;
    browser_options.hide_ext = true;

    bool res =
        dialog_file_browser_show(app->dialogs, app->file_path, app->file_path, &browser_options);

    return res;
}

static void fuzzer_scene_main_show_error(void* context, const char* erre_str) {
    furi_assert(context);
    PacsFuzzerApp* app = context;
    popup_set_header(app->popup, erre_str, 64, 20, AlignCenter, AlignTop);
    notification_message(app->notifications, &sequence_set_red_255);
    notification_message(app->notifications, &sequence_double_vibro);
    view_dispatcher_switch_to_view(app->view_dispatcher, FuzzerViewIDPopup);
}

void fuzzer_scene_main_on_enter(void* context) {
    furi_assert(context);
    PacsFuzzerApp* app = context;

    fuzzer_view_main_set_callback(app->main_view, fuzzer_scene_main_callback, app);

    fuzzer_view_main_update_data(app->main_view, app->fuzzer_state);

    // Setup view
    Popup* popup = app->popup;
    // popup_set_icon(popup, 83, 22, &I_WarningDolphinFlip_45x42);
    popup_set_timeout(popup, 2500);
    popup_set_context(popup, app);
    popup_set_callback(popup, fuzzer_scene_main_error_popup_callback);
    popup_enable_timeout(popup);

    view_dispatcher_switch_to_view(app->view_dispatcher, FuzzerViewIDMain);
}

bool fuzzer_scene_main_on_event(void* context, SceneManagerEvent event) {
    furi_assert(context);
    PacsFuzzerApp* app = context;
    bool consumed = false;

    if(event.type == SceneManagerEventTypeCustom) {
        if(event.event == FuzzerCustomEventViewMainBack) {
            if(!scene_manager_previous_scene(app->scene_manager)) {
                scene_manager_stop(app->scene_manager);
                view_dispatcher_stop(app->view_dispatcher);
            }
            consumed = true;
        } else if(event.event == FuzzerCustomEventViewMainPopupErr) {
            view_dispatcher_switch_to_view(app->view_dispatcher, FuzzerViewIDMain);
            consumed = true;
        } else if(event.event == FuzzerCustomEventViewMainOk) {
            fuzzer_view_main_get_state(app->main_view, &app->fuzzer_state);

            // TODO error logic
            bool loading_ok = false;

            switch(fuzzer_proto_get_attack_id_by_index(app->fuzzer_state.menu_index)) {
            case FuzzerAttackIdDefaultValues:
                loading_ok =
                    fuzzer_worker_init_attack_dict(app->worker, app->fuzzer_state.proto_index);

                if(!loading_ok) {
                    // error
                    fuzzer_scene_main_show_error(app, "Default dictionary\nis empty");
                }
                break;

            case FuzzerAttackIdBFCustomerID:
                // TODO
                app->payload->data_size = fuzzer_proto_get_max_data_size();
                memset(app->payload->data, 0x00, app->payload->data_size);

                if(fuzzer_worker_init_attack_bf_byte(
                       app->worker, app->fuzzer_state.proto_index, app->payload, 0)) {
                    scene_manager_set_scene_state(
                        app->scene_manager,
                        FuzzerSceneFieldEditor,
                        FuzzerFieldEditorStateEditingOff);
                    scene_manager_next_scene(app->scene_manager, FuzzerSceneFieldEditor);

                } else {
                    // error
                }
                break;

            case FuzzerAttackIdLoadFile:
                if(!fuzzer_scene_main_load_key_dialog(app)) {
                    break;
                } else {
                    switch(fuzzer_worker_load_key_from_file(
                        app->worker,
                        &app->fuzzer_state.proto_index,
                        furi_string_get_cstr(app->file_path))) {
                    case FuzzerWorkerLoadKeyStateOk:
                    case FuzzerWorkerLoadKeyStateDifferentProto:
                        scene_manager_set_scene_state(
                            app->scene_manager,
                            FuzzerSceneFieldEditor,
                            FuzzerFieldEditorStateEditingOn);
                        scene_manager_next_scene(app->scene_manager, FuzzerSceneFieldEditor);
                        FURI_LOG_I(TAG, "Load ok");
                        break;

                    case FuzzerWorkerLoadKeyStateBadFile:
                        fuzzer_scene_main_show_error(app, "Cant load\nor broken file");
                        FURI_LOG_E(TAG, "Cant load or broken file");
                        break;

                    case FuzzerWorkerLoadKeyStateUnsuportedProto:
                        fuzzer_scene_main_show_error(app, "Unsupported protocol");
                        FURI_LOG_E(TAG, "Unsupported protocol");
                        break;
                    }
                }
                break;

            case FuzzerAttackIdLoadFileCustomUids:
                if(!fuzzer_scene_main_load_custom_dict(app)) {
                    break;
                } else {
                    loading_ok = fuzzer_worker_init_attack_file_dict(
                        app->worker, app->fuzzer_state.proto_index, app->file_path);
                    if(!loading_ok) {
                        fuzzer_scene_main_show_error(app, "Incorrect key format\nor length");
                        // error
                    }
                }
                break;

            default:
                fuzzer_scene_main_show_error(app, "Unsuported attack");
                break;
            }

            if(loading_ok) {
                scene_manager_next_scene(app->scene_manager, FuzzerSceneAttack);
            }
            consumed = true;
        }
    }

    return consumed;
}

void fuzzer_scene_main_on_exit(void* context) {
    // furi_assert(context);
    // PacsFuzzerApp* app = context;
    UNUSED(context);
}