Momentum-Apps/totp/services/config/migrations/common_migration.c

#include "common_migration.h"
#include "../constants.h"
#include "../../../types/token_info.h"
#include "../../kb_layouts/kb_layout_provider.h"
#include <flipper_format/flipper_format_i.h>

#define TOTP_OLD_CONFIG_KEY_BASE_IV "BaseIV"

bool totp_config_migrate_to_latest(
    FlipperFormat* fff_data_file,
    FlipperFormat* fff_backup_data_file) {
    FuriString* temp_str = furi_string_alloc();
    uint32_t current_version = 0;
    bool result = false;
    do {
        flipper_format_write_header_cstr(
            fff_data_file, CONFIG_FILE_HEADER, CONFIG_FILE_ACTUAL_VERSION);

        if(!flipper_format_read_header(fff_backup_data_file, temp_str, &current_version)) {
            break;
        }

        if(flipper_format_read_string(
               fff_backup_data_file, TOTP_CONFIG_KEY_CRYPTO_VERSION, temp_str)) {
            flipper_format_write_string(fff_data_file, TOTP_CONFIG_KEY_CRYPTO_VERSION, temp_str);
        } else {
            uint32_t old_crypto_version = 1;
            flipper_format_write_uint32(
                fff_data_file, TOTP_CONFIG_KEY_CRYPTO_VERSION, &old_crypto_version, 1);
        }

        flipper_format_rewind(fff_backup_data_file);

        if(flipper_format_read_string(
               fff_backup_data_file, TOTP_CONFIG_KEY_CRYPTO_KEY_SLOT, temp_str)) {
            flipper_format_write_string(fff_data_file, TOTP_CONFIG_KEY_CRYPTO_KEY_SLOT, temp_str);
        } else {
            uint32_t default_old_key_slot = 2;
            flipper_format_write_uint32(
                fff_data_file, TOTP_CONFIG_KEY_CRYPTO_KEY_SLOT, &default_old_key_slot, 1);
        }

        flipper_format_rewind(fff_backup_data_file);

        if(flipper_format_read_string(fff_backup_data_file, TOTP_CONFIG_KEY_SALT, temp_str)) {
            flipper_format_write_string(fff_data_file, TOTP_CONFIG_KEY_SALT, temp_str);
        } else if(
            flipper_format_rewind(fff_backup_data_file) &&
            flipper_format_read_string(
                fff_backup_data_file, TOTP_OLD_CONFIG_KEY_BASE_IV, temp_str)) {
            flipper_format_write_string(fff_data_file, TOTP_CONFIG_KEY_SALT, temp_str);
        }

        flipper_format_rewind(fff_backup_data_file);

        if(flipper_format_read_string(
               fff_backup_data_file, TOTP_CONFIG_KEY_CRYPTO_VERIFY, temp_str)) {
            flipper_format_write_string(fff_data_file, TOTP_CONFIG_KEY_CRYPTO_VERIFY, temp_str);
        }

        flipper_format_rewind(fff_backup_data_file);

        if(flipper_format_read_string(fff_backup_data_file, TOTP_CONFIG_KEY_TIMEZONE, temp_str)) {
            flipper_format_write_string(fff_data_file, TOTP_CONFIG_KEY_TIMEZONE, temp_str);
        }

        flipper_format_rewind(fff_backup_data_file);

        if(flipper_format_read_string(fff_backup_data_file, TOTP_CONFIG_KEY_PINSET, temp_str)) {
            flipper_format_write_string(fff_data_file, TOTP_CONFIG_KEY_PINSET, temp_str);
        }

        flipper_format_rewind(fff_backup_data_file);

        if(flipper_format_read_string(
               fff_backup_data_file, TOTP_CONFIG_KEY_NOTIFICATION_METHOD, temp_str)) {
            flipper_format_write_string(
                fff_data_file, TOTP_CONFIG_KEY_NOTIFICATION_METHOD, temp_str);
        }

        flipper_format_rewind(fff_backup_data_file);

        if(flipper_format_read_string(
               fff_backup_data_file, TOTP_CONFIG_KEY_AUTOMATION_METHOD, temp_str)) {
            flipper_format_write_string(
                fff_data_file, TOTP_CONFIG_KEY_AUTOMATION_METHOD, temp_str);
        }

        flipper_format_rewind(fff_backup_data_file);

        if(flipper_format_read_string(fff_backup_data_file, TOTP_CONFIG_KEY_FONT, temp_str)) {
            flipper_format_write_string(fff_data_file, TOTP_CONFIG_KEY_FONT, temp_str);
        } else {
            uint32_t default_font_index = 0;
            flipper_format_write_uint32(
                fff_data_file, TOTP_CONFIG_KEY_FONT, &default_font_index, 1);
        }

        flipper_format_rewind(fff_backup_data_file);

        uint32_t kb_layout;
        if(!flipper_format_read_uint32(
               fff_backup_data_file, TOTP_CONFIG_KEY_AUTOMATION_KB_LAYOUT, &kb_layout, 1)) {
            kb_layout = TOTP_DEFAULT_KB_LAYOUT;
        }

        flipper_format_write_uint32(
            fff_data_file, TOTP_CONFIG_KEY_AUTOMATION_KB_LAYOUT, &kb_layout, 1);

        flipper_format_rewind(fff_backup_data_file);

        if(flipper_format_read_string(
               fff_backup_data_file, TOTP_CONFIG_KEY_AUTOMATION_INITIAL_DELAY, temp_str)) {
            flipper_format_write_string(
                fff_data_file, TOTP_CONFIG_KEY_AUTOMATION_INITIAL_DELAY, temp_str);
        } else {
            uint32_t default_automation_initial_delay = 500;
            flipper_format_write_uint32(
                fff_data_file,
                TOTP_CONFIG_KEY_AUTOMATION_INITIAL_DELAY,
                &default_automation_initial_delay,
                1);
        }

        flipper_format_rewind(fff_backup_data_file);

        while(true) {
            if(!flipper_format_read_string(
                   fff_backup_data_file, TOTP_CONFIG_KEY_TOKEN_NAME, temp_str)) {
                break;
            }

            flipper_format_write_string(fff_data_file, TOTP_CONFIG_KEY_TOKEN_NAME, temp_str);

            flipper_format_read_string(
                fff_backup_data_file, TOTP_CONFIG_KEY_TOKEN_SECRET, temp_str);
            flipper_format_write_string(fff_data_file, TOTP_CONFIG_KEY_TOKEN_SECRET, temp_str);

            if(current_version > 1) {
                flipper_format_read_string(
                    fff_backup_data_file, TOTP_CONFIG_KEY_TOKEN_ALGO, temp_str);

                if(current_version < 5) {
                    uint32_t algo_as_uint32t = TokenHashAlgoDefault;
                    if(furi_string_cmpi_str(temp_str, TOKEN_HASH_ALGO_SHA256_NAME) == 0) {
                        algo_as_uint32t = TokenHashAlgoSha256;
                    } else if(furi_string_cmpi_str(temp_str, TOKEN_HASH_ALGO_SHA512_NAME) == 0) {
                        algo_as_uint32t = TokenHashAlgoSha512;
                    } else if(furi_string_cmpi_str(temp_str, TOKEN_HASH_ALGO_STEAM_NAME) == 0) {
                        algo_as_uint32t = TokenHashAlgoSteam;
                    }

                    flipper_format_write_uint32(
                        fff_data_file, TOTP_CONFIG_KEY_TOKEN_ALGO, &algo_as_uint32t, 1);
                } else {
                    flipper_format_write_string(
                        fff_data_file, TOTP_CONFIG_KEY_TOKEN_ALGO, temp_str);
                }

                flipper_format_read_string(
                    fff_backup_data_file, TOTP_CONFIG_KEY_TOKEN_DIGITS, temp_str);
                flipper_format_write_string(fff_data_file, TOTP_CONFIG_KEY_TOKEN_DIGITS, temp_str);
            } else {
                const uint32_t default_algo = TokenHashAlgoDefault;
                flipper_format_write_uint32(
                    fff_data_file, TOTP_CONFIG_KEY_TOKEN_ALGO, &default_algo, 1);
                const uint32_t default_digits = TokenDigitsCountSix;
                flipper_format_write_uint32(
                    fff_data_file, TOTP_CONFIG_KEY_TOKEN_DIGITS, &default_digits, 1);
            }

            if(current_version > 2) {
                flipper_format_read_string(
                    fff_backup_data_file, TOTP_CONFIG_KEY_TOKEN_DURATION, temp_str);
                flipper_format_write_string(
                    fff_data_file, TOTP_CONFIG_KEY_TOKEN_DURATION, temp_str);
            } else {
                const uint32_t default_duration = TokenDurationDefault;
                flipper_format_write_uint32(
                    fff_data_file, TOTP_CONFIG_KEY_TOKEN_DURATION, &default_duration, 1);
            }

            if(current_version > 3) {
                flipper_format_read_string(
                    fff_backup_data_file, TOTP_CONFIG_KEY_TOKEN_AUTOMATION_FEATURES, temp_str);
                flipper_format_write_string(
                    fff_data_file, TOTP_CONFIG_KEY_TOKEN_AUTOMATION_FEATURES, temp_str);
            } else {
                const uint32_t default_automation_features = TokenAutomationFeatureNone;
                flipper_format_write_uint32(
                    fff_data_file,
                    TOTP_CONFIG_KEY_TOKEN_AUTOMATION_FEATURES,
                    &default_automation_features,
                    1);
            }

            if(current_version > 9) {
                flipper_format_read_string(
                    fff_backup_data_file, TOTP_CONFIG_KEY_TOKEN_TYPE, temp_str);
                flipper_format_write_string(fff_data_file, TOTP_CONFIG_KEY_TOKEN_TYPE, temp_str);
                flipper_format_read_string(
                    fff_backup_data_file, TOTP_CONFIG_KEY_TOKEN_COUNTER, temp_str);
                flipper_format_write_string(
                    fff_data_file, TOTP_CONFIG_KEY_TOKEN_COUNTER, temp_str);
            } else {
                const uint32_t default_token_type = TokenTypeTOTP;
                flipper_format_write_uint32(
                    fff_data_file, TOTP_CONFIG_KEY_TOKEN_TYPE, &default_token_type, 1);
                const uint64_t default_counter = 0;
                flipper_format_write_hex(
                    fff_data_file,
                    TOTP_CONFIG_KEY_TOKEN_COUNTER,
                    (const uint8_t*)&default_counter,
                    sizeof(default_counter));
            }
        }

        Stream* stream = flipper_format_get_raw_stream(fff_data_file);
        size_t current_pos = stream_tell(stream);
        size_t total_size = stream_size(stream);
        if(current_pos < total_size) {
            stream_delete(stream, total_size - current_pos);
        }

        result = true;
    } while(false);

    furi_string_free(temp_str);
    return result;
}