unleashed-firmware/applications/plugins/totp/scenes/authenticate/totp_scene_authenticate.c

#include "totp_scene_authenticate.h"
#include <dialogs/dialogs.h>
#include "../../types/common.h"
#include "../../services/ui/icons.h"
#include "../../services/ui/constants.h"
#include "../../services/config/config.h"
#include "../scene_director.h"
#include "../totp_scenes_enum.h"
#include "../../services/crypto/crypto.h"

#define MAX_CODE_LENGTH TOTP_IV_SIZE

typedef struct {
    uint8_t code_input[MAX_CODE_LENGTH];
    uint8_t code_length;
} SceneState;

void totp_scene_authenticate_init(PluginState* plugin_state) {
    memset(&plugin_state->iv[0], 0, TOTP_IV_SIZE);
}

void totp_scene_authenticate_activate(PluginState* plugin_state) {
    SceneState* scene_state = malloc(sizeof(SceneState));
    scene_state->code_length = 0;
    memset(&scene_state->code_input[0], 0, MAX_CODE_LENGTH);
    plugin_state->current_scene_state = scene_state;
    memset(&plugin_state->iv[0], 0, TOTP_IV_SIZE);
}

void totp_scene_authenticate_render(Canvas* const canvas, PluginState* plugin_state) {
    SceneState* scene_state = (SceneState*)plugin_state->current_scene_state;

    int v_shift = 0;
    if(scene_state->code_length > 0) {
        v_shift = -10;
    }

    if(plugin_state->crypto_verify_data == NULL) {
        canvas_draw_str_aligned(
            canvas,
            SCREEN_WIDTH_CENTER,
            SCREEN_HEIGHT_CENTER - 10 + v_shift,
            AlignCenter,
            AlignCenter,
            "Use arrow keys");
        canvas_draw_str_aligned(
            canvas,
            SCREEN_WIDTH_CENTER,
            SCREEN_HEIGHT_CENTER + 5 + v_shift,
            AlignCenter,
            AlignCenter,
            "to setup new PIN");
    } else {
        canvas_draw_str_aligned(
            canvas,
            SCREEN_WIDTH_CENTER,
            SCREEN_HEIGHT_CENTER + v_shift,
            AlignCenter,
            AlignCenter,
            "Use arrow keys to enter PIN");
    }
    const uint8_t PIN_ASTERISK_RADIUS = 3;
    const uint8_t PIN_ASTERISK_STEP = (PIN_ASTERISK_RADIUS << 1) + 2;
    if(scene_state->code_length > 0) {
        uint8_t left_start_x = (scene_state->code_length - 1) * PIN_ASTERISK_STEP >> 1;
        for(uint8_t i = 0; i < scene_state->code_length; i++) {
            canvas_draw_disc(
                canvas,
                SCREEN_WIDTH_CENTER - left_start_x + i * PIN_ASTERISK_STEP,
                SCREEN_HEIGHT_CENTER + 10,
                PIN_ASTERISK_RADIUS);
        }
    }
}

bool totp_scene_authenticate_handle_event(PluginEvent* const event, PluginState* plugin_state) {
    if(event->type == EventTypeKey) {
        if(event->input.type == InputTypeLong && event->input.key == InputKeyBack) {
            return false;
        } else if(event->input.type == InputTypePress) {
            SceneState* scene_state = (SceneState*)plugin_state->current_scene_state;

            const uint8_t ARROW_UP_CODE = 2;
            const uint8_t ARROW_RIGHT_CODE = 8;
            const uint8_t ARROW_DOWN_CODE = 11;
            const uint8_t ARROW_LEFT_CODE = 5;

            switch(event->input.key) {
            case InputKeyUp:
                if(scene_state->code_length < MAX_CODE_LENGTH) {
                    scene_state->code_input[scene_state->code_length] = ARROW_UP_CODE;
                    scene_state->code_length++;
                }
                break;
            case InputKeyDown:
                if(scene_state->code_length < MAX_CODE_LENGTH) {
                    scene_state->code_input[scene_state->code_length] = ARROW_DOWN_CODE;
                    scene_state->code_length++;
                }
                break;
            case InputKeyRight:
                if(scene_state->code_length < MAX_CODE_LENGTH) {
                    scene_state->code_input[scene_state->code_length] = ARROW_RIGHT_CODE;
                    scene_state->code_length++;
                }
                break;
            case InputKeyLeft:
                if(scene_state->code_length < MAX_CODE_LENGTH) {
                    scene_state->code_input[scene_state->code_length] = ARROW_LEFT_CODE;
                    scene_state->code_length++;
                }
                break;
            case InputKeyOk:
                totp_crypto_seed_iv(
                    plugin_state, &scene_state->code_input[0], scene_state->code_length);

                if(totp_crypto_verify_key(plugin_state)) {
                    FURI_LOG_D(LOGGING_TAG, "PIN is valid");
                    totp_scene_director_activate_scene(plugin_state, TotpSceneGenerateToken, NULL);
                } else {
                    FURI_LOG_D(LOGGING_TAG, "PIN is NOT valid");
                    memset(&scene_state->code_input[0], 0, MAX_CODE_LENGTH);
                    memset(&plugin_state->iv[0], 0, TOTP_IV_SIZE);
                    scene_state->code_length = 0;

                    DialogMessage* message = dialog_message_alloc();
                    dialog_message_set_buttons(message, "Try again", NULL, NULL);
                    dialog_message_set_header(
                        message,
                        "You entered\ninvalid PIN",
                        SCREEN_WIDTH_CENTER - 25,
                        SCREEN_HEIGHT_CENTER - 5,
                        AlignCenter,
                        AlignCenter);
                    dialog_message_set_icon(message, &I_DolphinCommon_56x48, 72, 17);
                    dialog_message_show(plugin_state->dialogs, message);
                    dialog_message_free(message);
                }
                break;
            case InputKeyBack:
                if(scene_state->code_length > 0) {
                    scene_state->code_input[scene_state->code_length - 1] = 0;
                    scene_state->code_length--;
                }
                break;
            default:
                break;
            }
        }
    }

    return true;
}

void totp_scene_authenticate_deactivate(PluginState* plugin_state) {
    if(plugin_state->current_scene_state == NULL) return;
    free(plugin_state->current_scene_state);
    plugin_state->current_scene_state = NULL;
}

void totp_scene_authenticate_free(PluginState* plugin_state) {
    UNUSED(plugin_state);
}