FlipperZeroNRFJammer/jammer.c

#include <furi.h>
#include <gui/gui.h>
#include <dialogs/dialogs.h>
#include <input/input.h>
#include <stdlib.h>
#include <furi_hal.h>
#include <furi_hal_gpio.h>
#include <furi_hal_spi.h>
#include <furi_hal_interrupt.h>
#include <furi_hal_resources.h>
#include <nrf24.h>
#include <notification/notification_messages.h>
#include <dolphin/dolphin.h>
#include "nrf24_jammer_icons.h"
#include "gui/elements.h"
#include <toolbox/stream/file_stream.h>
#include <dialogs/dialogs.h>

#include <stringp.h>

#define TAG "jammer"

#define MARGIN_LEFT 5
#define MARGIN_TOP 5
#define MARGIN_BOT 5
#define MARGIN_RIGHT 5
#define KEY_WIDTH 5
#define KEY_HEIGHT 10
#define KEY_PADDING 2

typedef struct {
    FuriMutex* mutex;
    bool is_thread_running;
    bool is_nrf24_connected;
    bool close_thread_please;
    bool select_file_on_next_render;
    uint8_t jam_type; // 0:narrow, 1:wide, 2:all, 3:custom
    FuriThread* jam_thread;
} PluginState;

typedef enum {
    EventTypeTick,
    EventTypeKey,
} EventType;

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

// various types of hopping I empirically found
uint8_t hopping_channels_2[128];
uint8_t hopping_channels_1[] = {32,34, 46,48, 50, 52, 0, 1, 2, 4, 6, 8, 22, 24, 26, 28, 30, 74, 76, 78, 80, 82, 84,86 };
uint8_t hopping_channels_0[] = {2, 26, 80};
uint8_t hopping_channels_len[] = {3, 24, 42, 0};


int parse_custom_list(char *input, uint8_t **out_arr) {
    // Make a copy of the input string, because strtok modifies it
    char *temp = strdup(input);
    if (!temp) {
        *out_arr = NULL;
        return 0;
    }

    // First pass: count how many integers we have (commas + 1)
    int count = 0;
    for (int i = 0; temp[i] != '\0'; i++) {
        if (temp[i] == ',') {
            count++;
        }
    }
    count++;  // one more than number of commas

    // Allocate the array
    *out_arr = (uint8_t *)malloc(count * sizeof(uint8_t));
    if (!(*out_arr)) {
        free(temp);
        return 0; // allocation failed
    }

    // Second pass: tokenize and parse
    int index = 0;
    char *token = strtok(temp, ",");
    while (token != NULL) {
        (*out_arr)[index++] = (uint8_t) atoi(token);
        token = strtok(NULL, ",");
    }

    free(temp);
    return count;  // number of integers parsed
}

char *jam_types[] = {"narrow", "wide", "full", "custom"};
uint8_t  *hopping_channels;
int hopping_channels_custom_len = 0;
uint8_t *hopping_channels_custom_arr = NULL;

static void render_callback(Canvas* const canvas, void* ctx) {
    furi_assert(ctx);
    PluginState* plugin_state = ctx;
    //furi_mutex_acquire(plugin_state->mutex, FuriWaitForever);

    // border around the edge of the screen
    canvas_draw_frame(canvas, 0, 0, 128, 64);

    canvas_set_font(canvas, FontSecondary);

    char tmp[128];
    snprintf(tmp, 128, "^ type:%s", jam_types[plugin_state->jam_type]);
    canvas_draw_str_aligned(canvas, 10, 3, AlignLeft, AlignTop, tmp);

    if(!plugin_state->is_thread_running && plugin_state->jam_type <= 3) {
        canvas_set_font(canvas, FontSecondary);
        canvas_draw_str_aligned(canvas, 10, 20, AlignLeft, AlignBottom, "Press Ok button to start");
        
        if (plugin_state->jam_type == 3) {
            canvas_draw_str(canvas, 80, 30, "select file >"); 
        }
        
        if(!plugin_state->is_nrf24_connected) {
            canvas_draw_str_aligned(
                canvas, 10, 30, AlignLeft, AlignBottom, "Connect NRF24 to GPIO!");
        }
    } else if(plugin_state->is_thread_running && plugin_state->jam_type <= 3) {
        canvas_set_font(canvas, FontSecondary);
        canvas_draw_str_aligned(canvas, 3, 30, AlignLeft, AlignBottom, "Causing mayhem...");
        canvas_draw_str_aligned(canvas, 3, 40, AlignLeft, AlignBottom, "Please wait!");
        canvas_draw_str_aligned(
            canvas, 3, 50, AlignLeft, AlignBottom, "Press back to exit.");
    } else {
        canvas_draw_str_aligned(canvas, 3, 10, AlignLeft, AlignBottom, "Unknown Error");
        canvas_draw_str_aligned(canvas, 3, 20, AlignLeft, AlignBottom, "press back");
        canvas_draw_str_aligned(canvas, 3, 30, AlignLeft, AlignBottom, "to exit");
    }
    uint8_t limit = hopping_channels_len[plugin_state->jam_type];
    canvas_draw_frame(canvas, 0, 52, 128, 13);
    if (limit > 0) {
        for(int i = 0; i < limit; i++) {
            canvas_draw_line(canvas, hopping_channels[i]+1, 53, hopping_channels[i]+1, 64);
        }
    }
    //furi_mutex_release(plugin_state->mutex);
}

static void input_callback(InputEvent* input_event, void* ctx) {
    furi_assert(ctx);
    FuriMessageQueue* event_queue = ctx;
    PluginEvent event = {.type = EventTypeKey, .input = *input_event};
    furi_message_queue_put(event_queue, &event, 100);
}

static void jammer_state_init(PluginState* const plugin_state) {
    plugin_state->is_thread_running = false;
}

// entrypoint for worker
static int32_t mj_worker_thread(void* ctx) {
    PluginState* plugin_state = ctx;
    plugin_state->is_thread_running = true;
    FURI_LOG_D(TAG, "starting to jam");
    char tmp[128];
    // make sure the NRF24 is powered down so we can do all the initial setup
    nrf24_set_idle(nrf24_HANDLE);
    uint8_t mac[] = { 0xDE, 0xAD}; // DEAD BEEF FEED
    uint8_t ping_packet[] = {0xDE, 0xAD, 0xBE, 0xEF,0xDE, 0xAD, 0xBE, 0xEF,0xDE, 0xAD, 0xBE, 0xEF,0xDE, 0xAD, 0xBE, 0xEF,0xDE, 0xAD, 0xBE, 0xEF,0xDE, 0xAD, 0xBE, 0xEF,0xDE, 0xAD, 0xBE, 0xEF,0xDE, 0xAD, 0xBE, 0xEF}; // 32 bytes, in case we ever need to experiment with bigger packets
       
    uint8_t conf = 0;

    nrf24_configure(nrf24_HANDLE, 2, mac, mac, 2, 1, true, true);
    // set PA level to maximum
    uint8_t setup; 
    nrf24_read_reg(nrf24_HANDLE, REG_RF_SETUP, &setup,1);
    
    setup &= 0xF8;
    setup |= 7;
    
    snprintf(tmp, 128, "NRF24 SETUP REGISTER: %d", setup);
    FURI_LOG_D(TAG, tmp);
    
    nrf24_read_reg(nrf24_HANDLE, REG_CONFIG, &conf,1);
    snprintf(tmp, 128, "NRF24 CONFIG REGISTER: %d", conf);
    FURI_LOG_D(TAG, tmp);
    nrf24_write_reg(nrf24_HANDLE, REG_RF_SETUP, setup);

    #define size 32
    uint8_t status = 0;
    uint8_t tx[size + 1];
    uint8_t rx[size + 1];
    memset(tx, 0, size + 1);
    memset(rx, 0, size + 1);

    tx[0] = W_TX_PAYLOAD_NOACK;

    memcpy(&tx[1], ping_packet, size);

    #define nrf24_TIMEOUT 500
    // push data to the TX register
    nrf24_spi_trx(nrf24_HANDLE, tx, 0, size + 1, nrf24_TIMEOUT);
    // put the module in TX mode
    nrf24_set_tx_mode(nrf24_HANDLE);
    // send one test packet (for debug reasons)
    while(!(status & (TX_DS | MAX_RT))) 
    {
        status = nrf24_status(nrf24_HANDLE);
        snprintf(tmp, 128, "NRF24 STATUS REGISTER: %d", status);
        
        FURI_LOG_D(TAG, tmp);
    }
    NotificationApp* notification = furi_record_open(RECORD_NOTIFICATION);

    
    uint8_t chan = 0;
    uint8_t limit = 0;
    notification_message(notification, &sequence_blink_red_100);

    do {
        limit = hopping_channels_len[plugin_state->jam_type];
        for(int ch = 0;ch < limit; ch++) {
            chan = hopping_channels[ch];
            // change channel
            nrf24_write_reg(nrf24_HANDLE, REG_RF_CH, chan);
            // push new data to the TX register
            nrf24_spi_trx(nrf24_HANDLE, tx, 0, 3, nrf24_TIMEOUT);
        }
    } while(!plugin_state->close_thread_please);
    
    furi_record_close(RECORD_NOTIFICATION);
    
    plugin_state->is_thread_running = false;
    nrf24_set_idle(nrf24_HANDLE);
    return 0;
}

int32_t jammer_app(void* p) {
    UNUSED(p);
    FuriMessageQueue* event_queue = furi_message_queue_alloc(8, sizeof(PluginEvent));
    dolphin_deed(DolphinDeedPluginStart);
    PluginState* plugin_state = malloc(sizeof(PluginState));
    jammer_state_init(plugin_state);
    plugin_state->mutex = furi_mutex_alloc(FuriMutexTypeNormal);
    if(!plugin_state->mutex) {
        FURI_LOG_E("jammer", "cannot create mutex\r\n");
        furi_message_queue_free(event_queue);
        free(plugin_state);
        return 255;
    }

    NotificationApp* notification = furi_record_open(RECORD_NOTIFICATION);

    // Set system callbacks
    ViewPort* view_port = view_port_alloc();
    view_port_draw_callback_set(view_port, render_callback, plugin_state);
    view_port_input_callback_set(view_port, input_callback, event_queue);

    // Open GUI and register view_port
    Gui* gui = furi_record_open(RECORD_GUI);
    gui_add_view_port(gui, view_port, GuiLayerFullscreen);

    plugin_state->jam_thread = furi_thread_alloc();
    furi_thread_set_name(plugin_state->jam_thread, "Jammer Worker");
    furi_thread_set_stack_size(plugin_state->jam_thread, 2048);
    furi_thread_set_context(plugin_state->jam_thread, plugin_state);
    furi_thread_set_callback(plugin_state->jam_thread, mj_worker_thread);
    FURI_LOG_D(TAG, "nrf24 init...");
    nrf24_init();
    FURI_LOG_D(TAG, "nrf24 init done!");
    PluginEvent event;
    for(int i = 0; i < 128; i++) hopping_channels_2[i] = i*2;
    hopping_channels = hopping_channels_0;
    plugin_state->is_nrf24_connected = true;
    if(!nrf24_check_connected(nrf24_HANDLE)) {
        plugin_state->is_nrf24_connected = false;
    }

    for(bool processing = true; processing;) {
        FuriStatus event_status = furi_message_queue_get(event_queue, &event, 100);
        
        if(event_status == FuriStatusOk) {
            // press events
            if(event.type == EventTypeKey) {
                if(event.input.type == InputTypePress) {
                    //furi_mutex_acquire(plugin_state->mutex, FuriWaitForever);

                    switch(event.input.key) {
                    case InputKeyUp:
                        if(!plugin_state -> is_thread_running) {
                            plugin_state->jam_type = (plugin_state->jam_type + 1) % 4;
                        
                            switch(plugin_state->jam_type) {
                                case 0: hopping_channels = hopping_channels_0; break;
                                case 1: hopping_channels = hopping_channels_1; break;
                                case 2: hopping_channels = hopping_channels_2; break;
                                case 3: hopping_channels = hopping_channels_custom_arr;
                                default: break;             
                            }

                            view_port_update(view_port);
                        }
                        break;
                    case InputKeyDown:
                        break;
                    case InputKeyRight:
                        if(plugin_state->jam_type == 3) { 
                            DialogsApp* dialogs = furi_record_open("dialogs");
                            Storage* storage = furi_record_open(RECORD_STORAGE);
                            plugin_state->select_file_on_next_render = true;
                            Stream* file_stream = file_stream_alloc(storage);
                            plugin_state->select_file_on_next_render = false;
                            FuriString* path;
                            path = furi_string_alloc();
                            furi_string_set(path, "/ext/");
                            DialogsFileBrowserOptions browser_options;
                            dialog_file_browser_set_basic_options(&browser_options, ".txt", &I_badusb_10px);
                            browser_options.hide_ext = false;
                            browser_options.base_path = "/ext/";
                            furi_delay_ms(1000);
                            bool ret = dialog_file_browser_show(dialogs, path, path, &browser_options);
                            furi_record_close("dialogs");
                            if(ret) {
                                if(!file_stream_open(file_stream, furi_string_get_cstr(path), FSAM_READ, FSOM_OPEN_EXISTING)) {
                                    FURI_LOG_D(TAG, "Cannot open file \"%s\"", furi_string_get_cstr(path));
                                } else {
                                    size_t file_size = 0;
                                    size_t bytes_read = 0;
                                    uint8_t* file_buf;
                                    FURI_LOG_I(TAG, "opening channel file");
                                    file_size = stream_size(file_stream);
                                    file_buf = malloc(file_size);
                                    memset(file_buf, 0, file_size);
                                    bytes_read = stream_read(file_stream, file_buf, file_size);
                                    if(bytes_read == file_size) {
                                        FURI_LOG_I(TAG, "read file");
                                        hopping_channels_custom_len = parse_custom_list((char *)file_buf, &hopping_channels_custom_arr);

                                        FURI_LOG_D(TAG, "Length: %d\n", hopping_channels_custom_len);           // e.g. 6
                                        for (int i = 0; i < hopping_channels_custom_len; i++) {
                                            FURI_LOG_D(TAG, "%d ", hopping_channels_custom_arr[i]);                // 1 2 3 4 1 2
                                        }
                                        hopping_channels_len[3] = hopping_channels_custom_len;
                                        hopping_channels = hopping_channels_custom_arr;

                                        FURI_LOG_D(TAG, "%s", file_buf);
                                    } else {
                                        FURI_LOG_I(TAG, "load failed. file size: %d", file_size);
                                    }
                                    free(file_buf);
                                }
                            }
                            furi_string_free(path);
                        
                            view_port_update(view_port);
                        }
                        break;
                    case InputKeyLeft:
                        break;
                    case InputKeyOk:
                        if(!plugin_state->is_thread_running) {
                            if(!nrf24_check_connected(nrf24_HANDLE)) {
                                plugin_state->is_nrf24_connected = false;
                                notification_message(notification, &sequence_error);
                            } else {
                                furi_thread_start(plugin_state->jam_thread);
                            }
                            view_port_update(view_port);
                        }
                        
                        break;
                    case InputKeyBack:
                        FURI_LOG_D(TAG, "terminating thread...");
                        if(!plugin_state->is_thread_running) processing = false;

                        plugin_state->close_thread_please = true;
                        
                        if(plugin_state->is_thread_running && plugin_state->jam_thread) {
                            plugin_state->is_thread_running = false;
                            furi_thread_join(plugin_state->jam_thread); // wait until thread is finished
                            view_port_update(view_port);
                        }
                        plugin_state->close_thread_please = false;
                        
                        break;
                    default:
                        break;
                    }
                
                    //view_port_update(view_port);

                }
            
            }
        }

        // furi_mutex_release(plugin_state->mutex);
    }

    furi_thread_free(plugin_state->jam_thread);
    FURI_LOG_D(TAG, "nrf24 deinit...");
    nrf24_deinit();
    view_port_enabled_set(view_port, false);
    gui_remove_view_port(gui, view_port);
    furi_record_close(RECORD_GUI);
    furi_record_close(RECORD_NOTIFICATION);
    view_port_free(view_port);
    furi_message_queue_free(event_queue);
    // furi_mutex_free(plugin_state->mutex);
    free(plugin_state);

    return 0;
}