Momentum-Firmware/applications/external/multi_fuzzer/scenes/fuzzer_scene_attack.c

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

const NotificationSequence sequence_one_red_50_on_blink_blue = {
    &message_red_255,
    &message_delay_50,
    &message_red_0,
    &message_blink_start_10,
    &message_blink_set_color_blue,
    &message_do_not_reset,
    NULL,
};

static void fuzzer_scene_attack_update_uid(PacsFuzzerApp* app) {
    furi_assert(app);
    furi_assert(app->worker);
    furi_assert(app->attack_view);

    fuzzer_worker_get_current_key(app->worker, app->payload);

    fuzzer_view_attack_set_uid(app->attack_view, app->payload);
}

static void fuzzer_scene_attack_set_state(PacsFuzzerApp* app, FuzzerAttackState state) {
    furi_assert(app);

    scene_manager_set_scene_state(app->scene_manager, FuzzerSceneAttack, state);
    switch(state) {
    case FuzzerAttackStateIdle:
        notification_message(app->notifications, &sequence_blink_stop);
        break;

    case FuzzerAttackStateAttacking:
        notification_message(app->notifications, &sequence_blink_start_blue);
        break;
    case FuzzerAttackStateEmulating:
        notification_message(app->notifications, &sequence_blink_start_blue);
        break;

    case FuzzerAttackStateEnd:
        notification_message(app->notifications, &sequence_blink_stop);
        notification_message(app->notifications, &sequence_single_vibro);
        break;

    case FuzzerAttackStateOff:
        notification_message(app->notifications, &sequence_blink_stop);
        break;

    case FuzzerAttackStatePause:
        notification_message(app->notifications, &sequence_blink_stop);
        break;
    }

    fuzzer_view_update_state(app->attack_view, state);
}

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

    notification_message(app->notifications, &sequence_one_red_50_on_blink_blue);
    fuzzer_scene_attack_update_uid(app);
}

void fuzzer_scene_attack_worker_end_callback(void* context) {
    furi_assert(context);
    PacsFuzzerApp* app = context;
    view_dispatcher_send_custom_event(app->view_dispatcher, FuzzerCustomEventViewAttackEnd);
}

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

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

    fuzzer_view_attack_set_callback(app->attack_view, fuzzer_scene_attack_callback, app);

    fuzzer_worker_set_uid_chaged_callback(
        app->worker, fuzzer_scene_attack_worker_tick_callback, app);

    fuzzer_worker_set_end_callback(app->worker, fuzzer_scene_attack_worker_end_callback, app);

    fuzzer_view_attack_reset_data(
        app->attack_view,
        fuzzer_proto_get_menu_label(app->fuzzer_state.menu_index),
        fuzzer_proto_get_name(app->fuzzer_state.proto_index));

    fuzzer_scene_attack_update_uid(app);

    scene_manager_set_scene_state(app->scene_manager, FuzzerSceneAttack, FuzzerAttackStateIdle);

    view_dispatcher_switch_to_view(app->view_dispatcher, FuzzerViewIDAttack);
}

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

    if(event.type == SceneManagerEventTypeCustom) {
        if(event.event == FuzzerCustomEventViewAttackExit) {
            // Exit
            fuzzer_worker_stop(app->worker);

            fuzzer_scene_attack_set_state(app, FuzzerAttackStateOff);
            if(!scene_manager_previous_scene(app->scene_manager)) {
                scene_manager_stop(app->scene_manager);
                view_dispatcher_stop(app->view_dispatcher);
            }
        } else if(event.event == FuzzerCustomEventViewAttackRunAttack) {
            if(fuzzer_worker_start(
                   app->worker,
                   fuzzer_view_attack_get_time_delay(app->attack_view),
                   fuzzer_view_attack_get_emu_time(app->attack_view))) {
                fuzzer_scene_attack_set_state(app, FuzzerAttackStateAttacking);
            } else {
                // Error?
            }
        } else if(event.event == FuzzerCustomEventViewAttackEmulateCurrent) {
            fuzzer_worker_start_emulate(app->worker);

            fuzzer_scene_attack_set_state(app, FuzzerAttackStateEmulating);
        } else if(event.event == FuzzerCustomEventViewAttackPause) {
            fuzzer_worker_pause(app->worker);

            fuzzer_scene_attack_set_state(app, FuzzerAttackStatePause);
        } else if(event.event == FuzzerCustomEventViewAttackIdle) {
            fuzzer_worker_pause(app->worker);

            fuzzer_scene_attack_set_state(app, FuzzerAttackStateIdle);
        } else if(event.event == FuzzerCustomEventViewAttackNextUid) {
            if(fuzzer_worker_next_key(app->worker)) {
                fuzzer_scene_attack_update_uid(app);
            } else {
                notification_message(app->notifications, &sequence_blink_red_100);
            }
        } else if(event.event == FuzzerCustomEventViewAttackPrevUid) {
            if(fuzzer_worker_previous_key(app->worker)) {
                fuzzer_scene_attack_update_uid(app);
            } else {
                notification_message(app->notifications, &sequence_blink_red_100);
            }
        } else if(event.event == FuzzerCustomEventViewAttackSave) {
            scene_manager_next_scene(app->scene_manager, FuzzerSceneSaveName);
        }
        // Callback from worker
        else if(event.event == FuzzerCustomEventViewAttackEnd) {
            fuzzer_scene_attack_set_state(app, FuzzerAttackStateEnd);
            consumed = true;
        }
    }

    return consumed;
}

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

    // XXX the scene has no descendants, and the return will be processed in on_event
    // fuzzer_worker_stop();

    fuzzer_worker_set_uid_chaged_callback(app->worker, NULL, NULL);
    fuzzer_worker_set_end_callback(app->worker, NULL, NULL);
}