Momentum-Apps/applications/ibutton/ibutton-app.cpp

#include "ibutton-app.h"
#include <stdarg.h>
#include <callback-connector.h>
#include <m-string.h>

const char* iButtonApp::app_folder = "ibutton";
const char* iButtonApp::app_extension = ".ibtn";

void iButtonApp::run(void) {
    iButtonEvent event;
    bool consumed;
    bool exit = false;

    scenes[current_scene]->on_enter(this);

    while(!exit) {
        view.receive_event(&event);

        consumed = scenes[current_scene]->on_event(this, &event);

        if(!consumed) {
            if(event.type == iButtonEvent::Type::EventTypeBack) {
                exit = switch_to_previous_scene();
            }
        }
    };

    scenes[current_scene]->on_exit(this);
}

void iButtonApp::print_key_data(void) {
    uint8_t* key_data = key.get_data();
    switch(key.get_key_type()) {
    case iButtonKeyType::KeyDallas:
        printf(
            "Dallas %02X %02X %02X %02X %02X %02X %02X %02X\r\n",
            key_data[0],
            key_data[1],
            key_data[2],
            key_data[3],
            key_data[4],
            key_data[5],
            key_data[6],
            key_data[7]);
        break;
    case iButtonKeyType::KeyCyfral:
        printf("Cyfral %02X %02X\r\n", key_data[0], key_data[1]);
        break;
    case iButtonKeyType::KeyMetakom:
        printf(
            "Metakom %02X %02X %02X %02X\r\n", key_data[0], key_data[1], key_data[2], key_data[3]);
        break;
    }
}

bool iButtonApp::read_hex_byte(string_t args, uint8_t* byte) {
    char* endptr;
    *byte = strtoul(string_get_cstr(args), &endptr, 16);
    if(*endptr == '\0') {
        return false;
    }
    size_t ws = string_search_char(args, ' ');
    if(ws != 2) {
        return false;
    }
    string_right(args, ws);
    string_strim(args);
    return true;
}

void iButtonApp::cli_cmd_callback(Cli* cli, string_t args, void* context) {
    iButtonApp::Scene scene;
    string_t cmd;
    string_init(cmd);
    if(!string_cmp_str(args, "read")) {
        scene = iButtonApp::Scene::SceneCliRead;
        printf("Reading key ...\r\n");
    } else {
        // Parse write / emulate commands
        size_t ws = string_search_char(args, ' ');
        if(ws == STRING_FAILURE) {
            printf("Incorrect input. Try tm <read | write | emulate> [key_type] [key_data]");
            string_clear(cmd);
            return;
        } else {
            string_set_n(cmd, args, 0, ws);
            string_right(args, ws);
            string_strim(args);
        }
        if(!string_cmp_str(cmd, "write")) {
            scene = iButtonApp::Scene::SceneCliWrite;
        } else if(!string_cmp_str(cmd, "emulate")) {
            scene = iButtonApp::Scene::SceneCliEmulate;
        } else {
            printf("Incorrect input. Try tm <write | emulate> <key_type> <key_data>");
            string_clear(cmd);
            return;
        }
        string_clear(cmd);
        // Parse key type
        string_t key_type;
        string_init(key_type);
        ws = string_search_char(args, ' ');
        string_set_n(key_type, args, 0, ws);
        uint8_t bytes_to_read = 0;
        if(!string_cmp_str(key_type, "0")) {
            key.set_type(iButtonKeyType::KeyDallas);
            bytes_to_read = 8;
        } else if(!string_cmp_str(key_type, "1")) {
            key.set_type(iButtonKeyType::KeyCyfral);
            bytes_to_read = 2;
        } else if(!string_cmp_str(key_type, "2")) {
            key.set_type(iButtonKeyType::KeyMetakom);
            bytes_to_read = 4;
        } else {
            printf("Incorrect key type. Try 0 - KeyDallas, 1 - KeyCyfral, 2 - KeyMetakom");
            string_clear(key_type);
            return;
        }
        string_clear(key_type);
        // Read key data
        string_right(args, 1);
        string_strim(args);
        uint8_t key_data[8] = {};
        uint8_t i = 0;
        bool ret = true;
        while((i < bytes_to_read) && ret) {
            ret = read_hex_byte(args, &key_data[i++]);
        }
        if(i != bytes_to_read) {
            printf("Incorrect key data. Type %d key data hex digits", bytes_to_read);
            return;
        }
        key.set_data(key_data, bytes_to_read);
        if(scene == iButtonApp::Scene::SceneCliWrite) {
            printf("Writing key ");
        } else {
            printf("Emulating key ");
        }
        print_key_data();
    }
    switch_to_next_scene(scene);
    // Wait return event
    iButtonApp::CliEvent result;
    if(osMessageQueueGet(cli_event_result, &result, NULL, osWaitForever) != osOK) {
        printf("Command execution error");
        return;
    }
    // Process return event
    switch(result) {
    case iButtonApp::CliEvent::CliReadSuccess:
    case iButtonApp::CliEvent::CliReadCRCError:
        print_key_data();
        if(result == iButtonApp::CliEvent::CliReadCRCError) {
            printf("Warning: invalid CRC");
        }
        break;
    case iButtonApp::CliEvent::CliReadNotKeyError:
        printf("Read error: not a key");
        break;
    case iButtonApp::CliEvent::CliTimeout:
        printf("Timeout error");
        break;
    case iButtonApp::CliEvent::CliInterrupt:
        printf("Command interrupted");
        break;
    case iButtonApp::CliEvent::CliWriteSuccess:
        printf("Write success");
        break;
    case iButtonApp::CliEvent::CliWriteFail:
        printf("Write fail");
        break;
    default:
        break;
    }
    return;
}

void iButtonApp::cli_send_event(iButtonApp::CliEvent scene) {
    osMessageQueuePut(cli_event_result, &scene, 0, osWaitForever);
}

iButtonApp::iButtonApp() {
    api_hal_power_insomnia_enter();

    cli_event_result = osMessageQueueNew(1, sizeof(iButtonApp::Scene), NULL);
    key_worker = new KeyWorker(&ibutton_gpio);
    sd_ex_api = static_cast<SdCard_Api*>(furi_record_open("sdcard-ex"));
    fs_api = static_cast<FS_Api*>(furi_record_open("sdcard"));
    cli = static_cast<Cli*>(furi_record_open("cli"));
    notification = static_cast<NotificationApp*>(furi_record_open("notification"));

    auto callback = cbc::obtain_connector(this, &iButtonApp::cli_cmd_callback);
    cli_add_command(cli, "tm", callback, cli);

    // we need random
    srand(DWT->CYCCNT);
}

iButtonApp::~iButtonApp() {
    cli_delete_command(cli, "tm");

    furi_record_close("sdcard-ex");
    furi_record_close("sdcard");
    furi_record_close("cli");
    furi_record_close("notification");

    osMessageQueueDelete(cli_event_result);

    for(std::map<Scene, iButtonScene*>::iterator it = scenes.begin(); it != scenes.end(); ++it) {
        delete it->second;
        scenes.erase(it);
    }

    api_hal_power_insomnia_exit();
}

iButtonAppViewManager* iButtonApp::get_view_manager() {
    return &view;
}

void iButtonApp::switch_to_next_scene(Scene next_scene) {
    previous_scenes_list.push_front(current_scene);

    if(next_scene != Scene::SceneExit) {
        scenes[current_scene]->on_exit(this);
        current_scene = next_scene;
        scenes[current_scene]->on_enter(this);
    }
}

void iButtonApp::search_and_switch_to_previous_scene(std::initializer_list<Scene> scenes_list) {
    Scene previous_scene = Scene::SceneStart;
    bool scene_found = false;

    while(!scene_found) {
        previous_scene = get_previous_scene();
        for(Scene element : scenes_list) {
            if(previous_scene == element || previous_scene == Scene::SceneStart) {
                scene_found = true;
                break;
            }
        }
    }

    scenes[current_scene]->on_exit(this);
    current_scene = previous_scene;
    scenes[current_scene]->on_enter(this);
}

bool iButtonApp::switch_to_previous_scene(uint8_t count) {
    Scene previous_scene = Scene::SceneStart;

    for(uint8_t i = 0; i < count; i++) {
        previous_scene = get_previous_scene();
        if(previous_scene == Scene::SceneExit) break;
    }

    if(previous_scene == Scene::SceneExit) {
        return true;
    } else {
        scenes[current_scene]->on_exit(this);
        current_scene = previous_scene;
        scenes[current_scene]->on_enter(this);
        return false;
    }
}

iButtonApp::Scene iButtonApp::get_previous_scene() {
    Scene scene = previous_scenes_list.front();
    previous_scenes_list.pop_front();
    return scene;
}

const GpioPin* iButtonApp::get_ibutton_pin() {
    // TODO open record
    return &ibutton_gpio;
}

KeyWorker* iButtonApp::get_key_worker() {
    return key_worker;
}

iButtonKey* iButtonApp::get_key() {
    return &key;
}

SdCard_Api* iButtonApp::get_sd_ex_api() {
    return sd_ex_api;
}

FS_Api* iButtonApp::get_fs_api() {
    return fs_api;
}

char* iButtonApp::get_file_name() {
    return file_name;
}

uint8_t iButtonApp::get_file_name_size() {
    return file_name_size;
}

void iButtonApp::notify_green_blink() {
    notification_message(notification, &sequence_blink_green_10);
}

void iButtonApp::notify_yellow_blink() {
    notification_message(notification, &sequence_blink_yellow_10);
}

void iButtonApp::notify_red_blink() {
    notification_message(notification, &sequence_blink_red_10);
}

void iButtonApp::notify_error() {
    notification_message(notification, &sequence_error);
}

void iButtonApp::notify_success() {
    notification_message(notification, &sequence_success);
}

void iButtonApp::notify_green_on() {
    notification_message_block(notification, &sequence_set_green_255);
}

void iButtonApp::notify_green_off() {
    notification_message(notification, &sequence_reset_green);
}

void iButtonApp::notify_red_on() {
    notification_message_block(notification, &sequence_set_red_255);
}

void iButtonApp::notify_red_off() {
    notification_message(notification, &sequence_reset_red);
}

void iButtonApp::set_text_store(const char* text...) {
    va_list args;
    va_start(args, text);

    vsnprintf(text_store, text_store_size, text, args);

    va_end(args);
}

char* iButtonApp::get_text_store() {
    return text_store;
}

uint8_t iButtonApp::get_text_store_size() {
    return text_store_size;
}

void iButtonApp::generate_random_name(char* name, uint8_t max_name_size) {
    const uint8_t prefix_size = 9;
    const char* prefix[prefix_size] = {
        "ancient",
        "hollow",
        "strange",
        "disappeared",
        "unknown",
        "unthinkable",
        "unnamable",
        "nameless",
        "my",
    };

    const uint8_t suffix_size = 8;
    const char* suffix[suffix_size] = {
        "door",
        "entrance",
        "doorway",
        "entry",
        "portal",
        "entree",
        "opening",
        "crack",
    };

    sniprintf(
        name, max_name_size, "%s_%s", prefix[rand() % prefix_size], suffix[rand() % suffix_size]);
    // to upper
    name[0] = name[0] - 0x20;
}

// file managment
void iButtonApp::show_file_error_message(const char* error_text) {
    set_text_store(error_text);
    get_sd_ex_api()->show_error(get_sd_ex_api()->context, get_text_store());
}

bool iButtonApp::save_key(const char* key_name) {
    File key_file;
    string_t key_file_name;
    bool result = false;
    FS_Error fs_result;
    uint16_t write_count;

    // Create ibutton directory if necessary
    fs_result = get_fs_api()->common.mkdir(app_folder);
    if(fs_result != FSE_OK && fs_result != FSE_EXIST) {
        show_file_error_message("Cannot create\napplication folder");
        return false;
    };

    // First remove key if it was saved
    string_init_set_str(key_file_name, app_folder);
    string_cat_str(key_file_name, "/");
    string_cat_str(key_file_name, get_key()->get_name());
    string_cat_str(key_file_name, app_extension);
    fs_result = get_fs_api()->common.remove(string_get_cstr(key_file_name));
    if(fs_result != FSE_OK && fs_result != FSE_NOT_EXIST) {
        string_clear(key_file_name);
        show_file_error_message("Cannot remove\nold key file");
        return false;
    };

    // Save the key
    get_key()->set_name(key_name);
    string_set_str(key_file_name, app_folder);
    string_cat_str(key_file_name, "/");
    string_cat_str(key_file_name, get_key()->get_name());
    string_cat_str(key_file_name, app_extension);

    bool res = get_fs_api()->file.open(
        &key_file, string_get_cstr(key_file_name), FSAM_WRITE, FSOM_CREATE_ALWAYS);
    string_clear(key_file_name);

    if(res) {
        // type header
        const char* key_type = "E";

        switch(get_key()->get_key_type()) {
        case iButtonKeyType::KeyCyfral:
            key_type = "C";
            break;
        case iButtonKeyType::KeyDallas:
            key_type = "D";
            break;
        case iButtonKeyType::KeyMetakom:
            key_type = "M";
            break;
        }

        write_count = get_fs_api()->file.write(&key_file, key_type, 1);
        if(key_file.error_id != FSE_OK || write_count != 1) {
            show_file_error_message("Cannot write\nto key file");
            get_fs_api()->file.close(&key_file);
            return false;
        }

        const uint8_t byte_text_size = 4;
        char byte_text[byte_text_size];

        for(uint8_t i = 0; i < get_key()->get_type_data_size(); i++) {
            sniprintf(byte_text, byte_text_size, " %02X", get_key()->get_data()[i]);
            write_count = get_fs_api()->file.write(&key_file, byte_text, 3);
            if(key_file.error_id != FSE_OK || write_count != 3) {
                show_file_error_message("Cannot write\nto key file");
                get_fs_api()->file.close(&key_file);
                return false;
            }
        }
        result = true;
    } else {
        show_file_error_message("Cannot create\nnew key file");
    }

    get_fs_api()->file.close(&key_file);
    get_sd_ex_api()->check_error(get_sd_ex_api()->context);

    return result;
}

bool iButtonApp::load_key() {
    bool result = false;

    // Input events and views are managed by file_select
    bool res = get_sd_ex_api()->file_select(
        get_sd_ex_api()->context,
        app_folder,
        app_extension,
        get_file_name(),
        get_file_name_size(),
        get_key()->get_name());

    if(res) {
        string_t key_str;
        File key_file;
        uint16_t read_count;

        // Get key file path
        string_init_set_str(key_str, app_folder);
        string_cat_str(key_str, "/");
        string_cat_str(key_str, get_file_name());
        string_cat_str(key_str, app_extension);

        // Open key file
        get_fs_api()->file.open(
            &key_file, string_get_cstr(key_str), FSAM_READ, FSOM_OPEN_EXISTING);
        string_clear(key_str);

        if(key_file.error_id != FSE_OK) {
            show_file_error_message("Cannot open\nkey file");
            get_fs_api()->file.close(&key_file);
            return false;
        }

        const uint8_t byte_text_size = 4;
        char byte_text[byte_text_size] = {0, 0, 0, 0};

        // load type header
        read_count = get_fs_api()->file.read(&key_file, byte_text, 1);
        if(key_file.error_id != FSE_OK || read_count != 1) {
            show_file_error_message("Cannot read\nkey file");
            get_fs_api()->file.close(&key_file);
            return false;
        }

        iButtonKeyType key_type = iButtonKeyType::KeyCyfral;
        if(strcmp(byte_text, "C") == 0) {
            key_type = iButtonKeyType::KeyCyfral;
        } else if(strcmp(byte_text, "M") == 0) {
            key_type = iButtonKeyType::KeyMetakom;
        } else if(strcmp(byte_text, "D") == 0) {
            key_type = iButtonKeyType::KeyDallas;
        } else {
            show_file_error_message("Cannot parse\nkey file");
            get_fs_api()->file.close(&key_file);
            return false;
        }

        get_key()->set_type(key_type);

        // load data
        uint8_t key_data[IBUTTON_KEY_DATA_SIZE] = {0, 0, 0, 0, 0, 0, 0, 0};
        for(uint8_t i = 0; i < get_key()->get_type_data_size(); i++) {
            // space
            read_count = get_fs_api()->file.read(&key_file, byte_text, 1);
            if(key_file.error_id != FSE_OK || read_count != 1) {
                show_file_error_message("Cannot read\nkey file");
                get_fs_api()->file.close(&key_file);
                return false;
            }

            // value
            read_count = get_fs_api()->file.read(&key_file, byte_text, 2);
            if(key_file.error_id != FSE_OK || read_count != 2) {
                show_file_error_message("Cannot read\nkey file");
                get_fs_api()->file.close(&key_file);
                return false;
            }

            // convert hex value to byte
            key_data[i] = strtol(byte_text, NULL, 16);
        }

        get_fs_api()->file.close(&key_file);

        get_key()->set_name(get_file_name());
        get_key()->set_type(key_type);
        get_key()->set_data(key_data, IBUTTON_KEY_DATA_SIZE);

        result = true;
    }

    get_sd_ex_api()->check_error(get_sd_ex_api()->context);

    return result;
}

bool iButtonApp::delete_key() {
    iButtonKey* key = get_key();
    string_t key_file_name;
    bool result = false;

    string_init_set_str(key_file_name, app_folder);
    string_cat_str(key_file_name, "/");
    string_cat_str(key_file_name, key->get_name());
    string_cat_str(key_file_name, app_extension);
    result = (get_fs_api()->common.remove(string_get_cstr(key_file_name)) == FSE_OK);
    string_clear(key_file_name);

    return result;
}