unleashed-firmware/applications/plugins/ibtn_fuzzer/scene/ibtnfuzzer_scene_load_file.c

#include "ibtnfuzzer_scene_load_file.h"
#include "ibtnfuzzer_scene_entrypoint.h"

#define IBUTTON_FUZZER_APP_EXTENSION ".ibtn"
#define IBUTTON_FUZZER_APP_PATH_FOLDER "/ext/ibutton"

bool ibtnfuzzer_load(iBtnFuzzerState* context, const char* file_path) {
    bool result = false;
    Storage* storage = furi_record_open(RECORD_STORAGE);
    FlipperFormat* fff_data_file = flipper_format_file_alloc(storage);
    FuriString* temp_str;
    temp_str = furi_string_alloc();
    do {
        if(!flipper_format_file_open_existing(fff_data_file, file_path)) {
            FURI_LOG_E(TAG, "Error open file %s", file_path);
            furi_string_reset(context->notification_msg);
            furi_string_set(context->notification_msg, "Error open file");
            break;
        }

        // FileType
        if(!flipper_format_read_string(fff_data_file, "Filetype", temp_str)) {
            FURI_LOG_E(TAG, "Missing or incorrect Filetype");
            furi_string_reset(context->notification_msg);
            furi_string_set(context->notification_msg, "Missing or incorrect Filetypes");
            break;
        } else {
            FURI_LOG_I(TAG, "Filetype: %s", furi_string_get_cstr(temp_str));
        }

        // Key type
        if(!flipper_format_read_string(fff_data_file, "Key type", temp_str)) {
            FURI_LOG_E(TAG, "Missing or incorrect Key type");
            furi_string_reset(context->notification_msg);
            furi_string_set(context->notification_msg, "Missing or incorrect Key type");
            break;
        } else {
            FURI_LOG_I(TAG, "Key type: %s", furi_string_get_cstr(temp_str));

            if(context->proto == DS1990) {
                if(strcmp(furi_string_get_cstr(temp_str), "Dallas") != 0) {
                    FURI_LOG_E(TAG, "Unsupported Key type");
                    furi_string_reset(context->notification_msg);
                    furi_string_set(context->notification_msg, "Unsupported Key type");
                    break;
                }
            } else if(context->proto == Cyfral) {
                if(strcmp(furi_string_get_cstr(temp_str), "Cyfral") != 0) {
                    FURI_LOG_E(TAG, "Unsupported Key type");
                    furi_string_reset(context->notification_msg);
                    furi_string_set(context->notification_msg, "Unsupported Key type");
                    break;
                }
            } else {
                if(strcmp(furi_string_get_cstr(temp_str), "Metakom") != 0) {
                    FURI_LOG_E(TAG, "Unsupported Key type");
                    furi_string_reset(context->notification_msg);
                    furi_string_set(context->notification_msg, "Unsupported Key type");
                    break;
                }
            }
        }

        // Data
        if(!flipper_format_read_string(fff_data_file, "Data", context->data_str)) {
            FURI_LOG_E(TAG, "Missing or incorrect Data");
            furi_string_reset(context->notification_msg);
            furi_string_set(context->notification_msg, "Missing or incorrect Key");
            break;
        } else {
            FURI_LOG_I(TAG, "Key: %s", furi_string_get_cstr(context->data_str));

            if(context->proto == DS1990) {
                if(furi_string_size(context->data_str) != 23) {
                    FURI_LOG_E(TAG, "Incorrect Key length");
                    furi_string_reset(context->notification_msg);
                    furi_string_set(context->notification_msg, "Incorrect Key length");
                    break;
                }
            } else if(context->proto == Cyfral) {
                if(furi_string_size(context->data_str) != 5) {
                    FURI_LOG_E(TAG, "Incorrect Key length");
                    furi_string_reset(context->notification_msg);
                    furi_string_set(context->notification_msg, "Incorrect Key length");
                    break;
                }
            } else {
                if(furi_string_size(context->data_str) != 11) {
                    FURI_LOG_E(TAG, "Incorrect Key length");
                    furi_string_reset(context->notification_msg);
                    furi_string_set(context->notification_msg, "Incorrect Key length");
                    break;
                }
            }

            // String to uint8_t
            for(uint8_t i = 0; i < 8; i++) {
                char temp_str2[3];
                temp_str2[0] = furi_string_get_cstr(context->data_str)[i * 3];
                temp_str2[1] = furi_string_get_cstr(context->data_str)[i * 3 + 1];
                temp_str2[2] = '\0';
                context->data[i] = (uint8_t)strtol(temp_str2, NULL, 16);
            }
        }

        result = true;
    } while(0);
    furi_string_free(temp_str);
    flipper_format_free(fff_data_file);
    if(result) {
        FURI_LOG_I(TAG, "Loaded successfully");
        furi_string_reset(context->notification_msg);
        furi_string_set(context->notification_msg, "Source loaded.");
    }
    return result;
}

void ibtnfuzzer_scene_load_file_on_enter(iBtnFuzzerState* context) {
    if(ibtnfuzzer_load_protocol_from_file(context)) {
        context->current_scene = SceneSelectField;
    } else {
        ibtnfuzzer_scene_entrypoint_on_enter(context);
        context->current_scene = SceneEntryPoint;
    }
}

void ibtnfuzzer_scene_load_file_on_exit(iBtnFuzzerState* context) {
    UNUSED(context);
}

void ibtnfuzzer_scene_load_file_on_tick(iBtnFuzzerState* context) {
    UNUSED(context);
}

void ibtnfuzzer_scene_load_file_on_event(iBtnFuzzerEvent event, iBtnFuzzerState* context) {
    if(event.evt_type == EventTypeKey) {
        if(event.input_type == InputTypeShort) {
            switch(event.key) {
            case InputKeyDown:
            case InputKeyUp:
            case InputKeyLeft:
            case InputKeyRight:
            case InputKeyOk:
            case InputKeyBack:
                context->current_scene = SceneEntryPoint;
                break;
            default:
                break;
            }
        }
    }
}

void ibtnfuzzer_scene_load_file_on_draw(Canvas* canvas, iBtnFuzzerState* context) {
    UNUSED(context);
    UNUSED(canvas);
}

bool ibtnfuzzer_load_protocol_from_file(iBtnFuzzerState* context) {
    FuriString* user_file_path;
    user_file_path = furi_string_alloc();
    furi_string_set(user_file_path, IBUTTON_FUZZER_APP_PATH_FOLDER);

    DialogsFileBrowserOptions browser_options;
    dialog_file_browser_set_basic_options(
        &browser_options, IBUTTON_FUZZER_APP_EXTENSION, &I_ibutt_10px);
    browser_options.base_path = IBUTTON_FUZZER_APP_PATH_FOLDER;

    // Input events and views are managed by file_select
    bool res = dialog_file_browser_show(
        context->dialogs, user_file_path, user_file_path, &browser_options);

    if(res) {
        res = ibtnfuzzer_load(context, furi_string_get_cstr(user_file_path));
    }

    furi_string_free(user_file_path);

    return res;
}