Momentum-Apps/swd_probe_app.h

#ifndef __ARHA_FLIPPERAPP_DEMO
#define __ARHA_FLIPPERAPP_DEMO

#include <stdint.h>
#include <stdbool.h>
#include <stdio.h>

#include <furi.h>
#include <furi_hal.h>
#include <furi_hal_speaker.h>
#include <gui/gui.h>
#include <input/input.h>

#include <lib/subghz/receiver.h>
#include <lib/subghz/transmitter.h>
#include <lib/subghz/subghz_file_encoder_worker.h>
#include <notification/notification.h>
#include <notification/notification_messages.h>

#include "usb_uart.h"

#define TAG "SWD"

#define SWD_DELAY_US 0
#define TIMER_HZ 50
#define TIMEOUT 3
#define QUEUE_SIZE 32
#define IDLE_BITS 8
#define CLOCK_DELAY 0

#define MAX_FILE_LENGTH 128
#define SCRIPT_MAX_LINES 1000

typedef enum {
    ModePageScan = 0,
    ModePageFound = 1,
    ModePageDPRegs = 2,
    ModePageDPID = 3,
    ModePageAPID = 4,
    ModePageCount = 5,
    ModePageHexDump = 0x100,
    ModePageScript = 0x101,
} ModePages;

#define CDBGPWRUPREQ (1 << 28)
#define CDBGPWRUPACK (1 << 29)
#define CSYSPWRUPREQ (1 << 30)
#define CSYSPWRUPACK (1 << 31)
#define WDATAERR (1 << 7)
#define STICKYERR (1 << 5)
#define STAT_ERROR_FLAGS (WDATAERR | STICKYERR)

#define REG_IDCODE 0x00
#define REG_CTRLSTAT 0x01
#define REG_CTRLSTAT_BANK 0x00
#define REG_DLCR 0x01
#define REG_DLCR_BANK 0x01
#define REG_TARGETID 0x01
#define REG_TARGETID_BANK 0x02
#define REG_DLPIDR 0x01
#define REG_DLPIDR_BANK 0x03
#define REG_EVENTSTAT 0x01
#define REG_EVENTSTAT_BANK 0x04

#define REG_SELECT 0x02

#define MEMAP_CSW 0x00
#define MEMAP_TAR 0x04
#define MEMAP_DRW 0x0C
#define AP_IDR 0xFC
#define AP_BASE 0xF8

typedef enum { KeyNone, KeyUp, KeyRight, KeyDown, KeyLeft, KeyOK } KeyCode;

typedef enum {
    EventTimerTick,
    EventKeyPress,
} EventType;

typedef struct {
    EventType type;
    InputEvent input;
} AppEvent;

typedef struct {
    uint32_t ctrlstat;
    bool ctrlstat_ok;
    uint32_t dlcr;
    bool dlcr_ok;
    uint32_t dlpidr;
    bool dlpidr_ok;
    uint32_t dpidr;
    bool dpidr_ok;
    uint32_t eventstat;
    bool eventstat_ok;
    uint32_t select;
    bool select_ok;
    uint32_t targetid;
    bool targetid_ok;
} swd_dpreg_t;

typedef struct {
    bool ok;
    bool tested;
    uint8_t revision;
    uint16_t designer;
    uint8_t class;
    uint8_t variant;
    uint8_t type;
    uint32_t base;
} swd_apidr_info_t;

typedef struct {
    uint8_t revision;
    uint8_t partno;
    uint8_t version;
    uint16_t designer;
} swd_dpidr_info_t;

typedef struct {
    uint8_t revision;
    uint16_t partno;
    uint16_t designer;
} swd_targetid_info_t;

typedef struct sScriptContext ScriptContext;

typedef struct {
    Storage* storage;
    Gui* gui;
    DialogsApp* dialogs;
    NotificationApp* notification;

    FuriTimer* timer;
    UsbUart* uart;
    ViewPort* view_port;

    FuriMessageQueue* event_queue;
    FuriMutex* swd_mutex;
    FuriMutex* gui_mutex;

    swd_targetid_info_t targetid_info;
    swd_dpidr_info_t dpidr_info;
    swd_dpreg_t dp_regs;
    swd_apidr_info_t apidr_info[256];

    ScriptContext* script;
    ScriptContext* commandline;

    uint32_t loop_count;
    uint8_t current_mask_id;
    uint32_t current_mask;
    uint8_t io_swc;
    uint8_t io_swd;
    uint8_t io_num_swc;
    uint8_t io_num_swd;
    int32_t detected_timeout;
    uint32_t swd_clock_delay;
    uint32_t swd_idle_bits;
    bool detected;
    bool detected_device;
    bool detected_notified;
    uint32_t mode_page;
    uint8_t ap_pos;
    uint8_t ap_scanned;

    uint32_t hex_addr;
    uint8_t hex_select;
    uint8_t hex_buffer[32];
    uint8_t hex_buffer_valid[8];
    uint8_t hex_read_delay;

    char state_string[64];
    char script_detected[MAX_FILE_LENGTH];
    bool script_detected_executed;
} AppFSM;

struct sScriptContext {
    AppFSM* app;
    ScriptContext* parent;
    char filename[MAX_FILE_LENGTH];

    /* when used with string input */
    char line_data[128];
    uint64_t line_pos;

    /* when used with file input */
    File* script_file;

    uint64_t position;
    uint32_t selected_ap;
    uint32_t max_tries;
    uint32_t block_size;

    bool abort;
    bool restart;
    bool errors_ignore;
    bool status_ignore;
    bool goto_active;
    char goto_label[64];
};

typedef struct {
    const char* prefix;
    bool (*func)(ScriptContext* ctx);
} ScriptFunctionInfo;

const NotificationSequence seq_c_minor = {
    &message_note_c4,
    &message_delay_100,
    &message_sound_off,
    &message_delay_10,

    &message_note_ds4,
    &message_delay_100,
    &message_sound_off,
    &message_delay_10,

    &message_note_g4,
    &message_delay_100,
    &message_sound_off,
    &message_delay_10,

    &message_vibro_on,
    &message_delay_50,
    &message_vibro_off,
    NULL,
};

const NotificationSequence seq_error = {

    &message_vibro_on,
    &message_delay_50,
    &message_vibro_off,

    &message_note_g4,
    &message_delay_100,
    &message_sound_off,
    &message_delay_10,

    &message_note_c4,
    &message_delay_500,
    &message_sound_off,
    &message_delay_10,
    NULL,
};

const NotificationSequence* seq_sounds[] = {&seq_c_minor, &seq_error};

#endif