mirror of
https://github.com/Next-Flip/Momentum-Firmware.git
synced 2026-05-20 04:54:45 -07:00
Merge branch 'dev' of https://github.com/DarkFlippers/unleashed-firmware into xfw-dev
This commit is contained in:
@@ -13,3 +13,4 @@ ADD_SCENE(wifi_marauder, script_stage_edit, ScriptStageEdit)
|
||||
ADD_SCENE(wifi_marauder, script_stage_add, ScriptStageAdd)
|
||||
ADD_SCENE(wifi_marauder, script_stage_edit_list, ScriptStageEditList)
|
||||
ADD_SCENE(wifi_marauder, sniffpmkid_options, SniffPmkidOptions)
|
||||
ADD_SCENE(wifi_marauder, flasher, Flasher)
|
||||
|
||||
@@ -1,5 +1,7 @@
|
||||
#include "../wifi_marauder_app_i.h"
|
||||
|
||||
#include "../wifi_marauder_flasher.h"
|
||||
|
||||
char* _wifi_marauder_get_prefix_from_cmd(const char* command) {
|
||||
int end = strcspn(command, " ");
|
||||
char* prefix = (char*)malloc(sizeof(char) * (end + 1));
|
||||
@@ -101,13 +103,24 @@ void wifi_marauder_scene_console_output_on_enter(void* context) {
|
||||
view_dispatcher_switch_to_view(app->view_dispatcher, WifiMarauderAppViewConsoleOutput);
|
||||
|
||||
// Register callbacks to receive data
|
||||
wifi_marauder_uart_set_handle_rx_data_cb(
|
||||
app->uart,
|
||||
wifi_marauder_console_output_handle_rx_data_cb); // setup callback for general log rx thread
|
||||
// setup callback for general log rx thread
|
||||
if(app->flash_mode) {
|
||||
wifi_marauder_uart_set_handle_rx_data_cb(
|
||||
app->uart,
|
||||
wifi_marauder_flash_handle_rx_data_cb); // setup callback for general log rx thread
|
||||
} else {
|
||||
wifi_marauder_uart_set_handle_rx_data_cb(
|
||||
app->uart,
|
||||
wifi_marauder_console_output_handle_rx_data_cb); // setup callback for general log rx thread
|
||||
}
|
||||
wifi_marauder_uart_set_handle_rx_data_cb(
|
||||
app->lp_uart,
|
||||
wifi_marauder_console_output_handle_rx_packets_cb); // setup callback for packets rx thread
|
||||
|
||||
if(app->flash_mode) {
|
||||
wifi_marauder_flash_start_thread(app);
|
||||
}
|
||||
|
||||
// Get ready to send command
|
||||
if((app->is_command && app->selected_tx_string) || app->script) {
|
||||
const char* prefix =
|
||||
@@ -169,6 +182,11 @@ bool wifi_marauder_scene_console_output_on_event(void* context, SceneManagerEven
|
||||
consumed = true;
|
||||
} else if(event.type == SceneManagerEventTypeTick) {
|
||||
consumed = true;
|
||||
} else {
|
||||
if(app->flash_worker_busy) {
|
||||
// ignore button presses while flashing
|
||||
consumed = true;
|
||||
}
|
||||
}
|
||||
|
||||
return consumed;
|
||||
@@ -183,6 +201,10 @@ void wifi_marauder_scene_console_output_on_exit(void* context) {
|
||||
furi_delay_ms(50);
|
||||
}
|
||||
|
||||
if(app->flash_mode) {
|
||||
wifi_marauder_flash_stop_thread(app);
|
||||
}
|
||||
|
||||
// Unregister rx callback
|
||||
wifi_marauder_uart_set_handle_rx_data_cb(app->uart, NULL);
|
||||
wifi_marauder_uart_set_handle_rx_data_cb(app->lp_uart, NULL);
|
||||
|
||||
266
applications/external/wifi_marauder_companion/scenes/wifi_marauder_scene_flasher.c
vendored
Normal file
266
applications/external/wifi_marauder_companion/scenes/wifi_marauder_scene_flasher.c
vendored
Normal file
@@ -0,0 +1,266 @@
|
||||
#include "../wifi_marauder_app_i.h"
|
||||
#include "../wifi_marauder_flasher.h"
|
||||
|
||||
enum SubmenuIndex {
|
||||
SubmenuIndexS3Mode,
|
||||
SubmenuIndexBoot,
|
||||
SubmenuIndexPart,
|
||||
SubmenuIndexNvs,
|
||||
SubmenuIndexBootApp0,
|
||||
SubmenuIndexApp,
|
||||
SubmenuIndexCustom,
|
||||
SubmenuIndexFlash,
|
||||
};
|
||||
|
||||
static void wifi_marauder_scene_flasher_callback(void* context, uint32_t index) {
|
||||
WifiMarauderApp* app = context;
|
||||
|
||||
scene_manager_set_scene_state(app->scene_manager, WifiMarauderSceneFlasher, index);
|
||||
|
||||
// browse for files
|
||||
FuriString* predefined_filepath = furi_string_alloc_set_str(MARAUDER_APP_FOLDER);
|
||||
FuriString* selected_filepath = furi_string_alloc();
|
||||
DialogsFileBrowserOptions browser_options;
|
||||
dialog_file_browser_set_basic_options(&browser_options, ".bin", &I_Text_10x10);
|
||||
|
||||
// TODO refactor
|
||||
switch(index) {
|
||||
case SubmenuIndexS3Mode:
|
||||
// toggle S3 mode
|
||||
app->selected_flash_options[SelectedFlashS3Mode] =
|
||||
!app->selected_flash_options[SelectedFlashS3Mode];
|
||||
view_dispatcher_send_custom_event(app->view_dispatcher, WifiMarauderEventRefreshSubmenu);
|
||||
break;
|
||||
case SubmenuIndexBoot:
|
||||
app->selected_flash_options[SelectedFlashBoot] =
|
||||
!app->selected_flash_options[SelectedFlashBoot];
|
||||
if(app->selected_flash_options[SelectedFlashBoot]) {
|
||||
if(dialog_file_browser_show(
|
||||
app->dialogs, selected_filepath, predefined_filepath, &browser_options)) {
|
||||
strncpy(
|
||||
app->bin_file_path_boot,
|
||||
furi_string_get_cstr(selected_filepath),
|
||||
sizeof(app->bin_file_path_boot));
|
||||
}
|
||||
}
|
||||
if(app->bin_file_path_boot[0] == '\0') {
|
||||
// if user didn't select a file, leave unselected
|
||||
app->selected_flash_options[SelectedFlashBoot] = false;
|
||||
}
|
||||
view_dispatcher_send_custom_event(app->view_dispatcher, WifiMarauderEventRefreshSubmenu);
|
||||
break;
|
||||
case SubmenuIndexPart:
|
||||
app->selected_flash_options[SelectedFlashPart] =
|
||||
!app->selected_flash_options[SelectedFlashPart];
|
||||
if(dialog_file_browser_show(
|
||||
app->dialogs, selected_filepath, predefined_filepath, &browser_options)) {
|
||||
strncpy(
|
||||
app->bin_file_path_part,
|
||||
furi_string_get_cstr(selected_filepath),
|
||||
sizeof(app->bin_file_path_part));
|
||||
}
|
||||
if(app->bin_file_path_part[0] == '\0') {
|
||||
// if user didn't select a file, leave unselected
|
||||
app->selected_flash_options[SelectedFlashPart] = false;
|
||||
}
|
||||
view_dispatcher_send_custom_event(app->view_dispatcher, WifiMarauderEventRefreshSubmenu);
|
||||
break;
|
||||
case SubmenuIndexNvs:
|
||||
app->selected_flash_options[SelectedFlashNvs] =
|
||||
!app->selected_flash_options[SelectedFlashNvs];
|
||||
if(dialog_file_browser_show(
|
||||
app->dialogs, selected_filepath, predefined_filepath, &browser_options)) {
|
||||
strncpy(
|
||||
app->bin_file_path_nvs,
|
||||
furi_string_get_cstr(selected_filepath),
|
||||
sizeof(app->bin_file_path_nvs));
|
||||
}
|
||||
if(app->bin_file_path_nvs[0] == '\0') {
|
||||
// if user didn't select a file, leave unselected
|
||||
app->selected_flash_options[SelectedFlashNvs] = false;
|
||||
}
|
||||
view_dispatcher_send_custom_event(app->view_dispatcher, WifiMarauderEventRefreshSubmenu);
|
||||
break;
|
||||
case SubmenuIndexBootApp0:
|
||||
app->selected_flash_options[SelectedFlashBootApp0] =
|
||||
!app->selected_flash_options[SelectedFlashBootApp0];
|
||||
if(dialog_file_browser_show(
|
||||
app->dialogs, selected_filepath, predefined_filepath, &browser_options)) {
|
||||
strncpy(
|
||||
app->bin_file_path_boot_app0,
|
||||
furi_string_get_cstr(selected_filepath),
|
||||
sizeof(app->bin_file_path_boot_app0));
|
||||
}
|
||||
if(app->bin_file_path_boot_app0[0] == '\0') {
|
||||
// if user didn't select a file, leave unselected
|
||||
app->selected_flash_options[SelectedFlashBootApp0] = false;
|
||||
}
|
||||
view_dispatcher_send_custom_event(app->view_dispatcher, WifiMarauderEventRefreshSubmenu);
|
||||
break;
|
||||
case SubmenuIndexApp:
|
||||
app->selected_flash_options[SelectedFlashApp] =
|
||||
!app->selected_flash_options[SelectedFlashApp];
|
||||
if(dialog_file_browser_show(
|
||||
app->dialogs, selected_filepath, predefined_filepath, &browser_options)) {
|
||||
strncpy(
|
||||
app->bin_file_path_app,
|
||||
furi_string_get_cstr(selected_filepath),
|
||||
sizeof(app->bin_file_path_app));
|
||||
}
|
||||
if(app->bin_file_path_app[0] == '\0') {
|
||||
// if user didn't select a file, leave unselected
|
||||
app->selected_flash_options[SelectedFlashApp] = false;
|
||||
}
|
||||
view_dispatcher_send_custom_event(app->view_dispatcher, WifiMarauderEventRefreshSubmenu);
|
||||
break;
|
||||
case SubmenuIndexCustom:
|
||||
app->selected_flash_options[SelectedFlashCustom] =
|
||||
!app->selected_flash_options[SelectedFlashCustom];
|
||||
if(dialog_file_browser_show(
|
||||
app->dialogs, selected_filepath, predefined_filepath, &browser_options)) {
|
||||
strncpy(
|
||||
app->bin_file_path_custom,
|
||||
furi_string_get_cstr(selected_filepath),
|
||||
sizeof(app->bin_file_path_custom));
|
||||
}
|
||||
if(app->bin_file_path_custom[0] == '\0') {
|
||||
// if user didn't select a file, leave unselected
|
||||
app->selected_flash_options[SelectedFlashCustom] = false;
|
||||
}
|
||||
view_dispatcher_send_custom_event(app->view_dispatcher, WifiMarauderEventRefreshSubmenu);
|
||||
break;
|
||||
case SubmenuIndexFlash:
|
||||
// count how many options are selected
|
||||
app->num_selected_flash_options = 0;
|
||||
for(bool* option = &app->selected_flash_options[SelectedFlashBoot];
|
||||
option < &app->selected_flash_options[NUM_FLASH_OPTIONS];
|
||||
++option) {
|
||||
if(*option) {
|
||||
++app->num_selected_flash_options;
|
||||
}
|
||||
}
|
||||
if(app->num_selected_flash_options) {
|
||||
// only start next scene if at least one option is selected
|
||||
scene_manager_next_scene(app->scene_manager, WifiMarauderSceneConsoleOutput);
|
||||
}
|
||||
break;
|
||||
}
|
||||
|
||||
furi_string_free(selected_filepath);
|
||||
furi_string_free(predefined_filepath);
|
||||
}
|
||||
|
||||
#define STR_SELECT "[x]"
|
||||
#define STR_UNSELECT "[ ]"
|
||||
#define STR_BOOT "Bootloader (" TOSTRING(ESP_ADDR_BOOT) ")"
|
||||
#define STR_BOOT_S3 "Bootloader (" TOSTRING(ESP_ADDR_BOOT_S3) ")"
|
||||
#define STR_PART "Part Table (" TOSTRING(ESP_ADDR_PART) ")"
|
||||
#define STR_NVS "NVS (" TOSTRING(ESP_ADDR_NVS) ")"
|
||||
#define STR_BOOT_APP0 "boot_app0 (" TOSTRING(ESP_ADDR_BOOT_APP0) ")"
|
||||
#define STR_APP "Firmware (" TOSTRING(ESP_ADDR_APP) ")"
|
||||
#define STR_CUSTOM "Custom"
|
||||
#define STR_FLASH_S3 "[>] FLASH (ESP32-S3)"
|
||||
#define STR_FLASH "[>] FLASH"
|
||||
static void _refresh_submenu(WifiMarauderApp* app) {
|
||||
Submenu* submenu = app->submenu;
|
||||
|
||||
submenu_reset(app->submenu);
|
||||
|
||||
submenu_set_header(submenu, "Browse for files to flash");
|
||||
submenu_add_item(
|
||||
submenu,
|
||||
app->selected_flash_options[SelectedFlashS3Mode] ? "[x] Using ESP32-S3" :
|
||||
"[ ] Check if using S3",
|
||||
SubmenuIndexS3Mode,
|
||||
wifi_marauder_scene_flasher_callback,
|
||||
app);
|
||||
const char* strSelectBootloader = STR_UNSELECT " " STR_BOOT;
|
||||
if(app->selected_flash_options[SelectedFlashS3Mode]) {
|
||||
if(app->selected_flash_options[SelectedFlashBoot]) {
|
||||
strSelectBootloader = STR_SELECT " " STR_BOOT_S3;
|
||||
} else {
|
||||
strSelectBootloader = STR_UNSELECT " " STR_BOOT_S3;
|
||||
}
|
||||
} else {
|
||||
if(app->selected_flash_options[SelectedFlashBoot]) {
|
||||
strSelectBootloader = STR_SELECT " " STR_BOOT;
|
||||
} else {
|
||||
strSelectBootloader = STR_UNSELECT " " STR_BOOT;
|
||||
}
|
||||
}
|
||||
submenu_add_item(
|
||||
submenu, strSelectBootloader, SubmenuIndexBoot, wifi_marauder_scene_flasher_callback, app);
|
||||
submenu_add_item(
|
||||
submenu,
|
||||
app->selected_flash_options[SelectedFlashPart] ? STR_SELECT " " STR_PART :
|
||||
STR_UNSELECT " " STR_PART,
|
||||
SubmenuIndexPart,
|
||||
wifi_marauder_scene_flasher_callback,
|
||||
app);
|
||||
submenu_add_item(
|
||||
submenu,
|
||||
app->selected_flash_options[SelectedFlashNvs] ? STR_SELECT " " STR_NVS :
|
||||
STR_UNSELECT " " STR_NVS,
|
||||
SubmenuIndexNvs,
|
||||
wifi_marauder_scene_flasher_callback,
|
||||
app);
|
||||
submenu_add_item(
|
||||
submenu,
|
||||
app->selected_flash_options[SelectedFlashBootApp0] ? STR_SELECT " " STR_BOOT_APP0 :
|
||||
STR_UNSELECT " " STR_BOOT_APP0,
|
||||
SubmenuIndexBootApp0,
|
||||
wifi_marauder_scene_flasher_callback,
|
||||
app);
|
||||
submenu_add_item(
|
||||
submenu,
|
||||
app->selected_flash_options[SelectedFlashApp] ? STR_SELECT " " STR_APP :
|
||||
STR_UNSELECT " " STR_APP,
|
||||
SubmenuIndexApp,
|
||||
wifi_marauder_scene_flasher_callback,
|
||||
app);
|
||||
// TODO: custom addr
|
||||
//submenu_add_item(
|
||||
// submenu, app->selected_flash_options[SelectedFlashCustom] ? STR_SELECT " " STR_CUSTOM : STR_UNSELECT " " STR_CUSTOM, SubmenuIndexCustom, wifi_marauder_scene_flasher_callback, app);
|
||||
submenu_add_item(
|
||||
submenu,
|
||||
app->selected_flash_options[SelectedFlashS3Mode] ? STR_FLASH_S3 : STR_FLASH,
|
||||
SubmenuIndexFlash,
|
||||
wifi_marauder_scene_flasher_callback,
|
||||
app);
|
||||
|
||||
submenu_set_selected_item(
|
||||
submenu, scene_manager_get_scene_state(app->scene_manager, WifiMarauderSceneFlasher));
|
||||
view_dispatcher_switch_to_view(app->view_dispatcher, WifiMarauderAppViewSubmenu);
|
||||
}
|
||||
|
||||
void wifi_marauder_scene_flasher_on_enter(void* context) {
|
||||
WifiMarauderApp* app = context;
|
||||
|
||||
memset(app->selected_flash_options, 0, sizeof(app->selected_flash_options));
|
||||
app->bin_file_path_boot[0] = '\0';
|
||||
app->bin_file_path_part[0] = '\0';
|
||||
app->bin_file_path_nvs[0] = '\0';
|
||||
app->bin_file_path_boot_app0[0] = '\0';
|
||||
app->bin_file_path_app[0] = '\0';
|
||||
app->bin_file_path_custom[0] = '\0';
|
||||
|
||||
_refresh_submenu(app);
|
||||
}
|
||||
|
||||
bool wifi_marauder_scene_flasher_on_event(void* context, SceneManagerEvent event) {
|
||||
WifiMarauderApp* app = context;
|
||||
bool consumed = false;
|
||||
if(event.type == SceneManagerEventTypeCustom) {
|
||||
if(event.event == WifiMarauderEventRefreshSubmenu) {
|
||||
_refresh_submenu(app);
|
||||
consumed = true;
|
||||
}
|
||||
}
|
||||
|
||||
return consumed;
|
||||
}
|
||||
|
||||
void wifi_marauder_scene_flasher_on_exit(void* context) {
|
||||
WifiMarauderApp* app = context;
|
||||
submenu_reset(app->submenu);
|
||||
}
|
||||
@@ -121,6 +121,7 @@ const WifiMarauderItem items[NUM_MENU_ITEMS] = {
|
||||
{"Update", {"ota", "sd"}, 2, {"update -w", "update -s"}, NO_ARGS, FOCUS_CONSOLE_END, NO_TIP},
|
||||
{"Reboot", {""}, 1, {"reboot"}, NO_ARGS, FOCUS_CONSOLE_END, NO_TIP},
|
||||
{"Help", {""}, 1, {"help"}, NO_ARGS, FOCUS_CONSOLE_START, SHOW_STOPSCAN_TIP},
|
||||
{"Reflash ESP32 (WIP)", {""}, 1, {""}, NO_ARGS, FOCUS_CONSOLE_END, NO_TIP},
|
||||
{"Scripts", {""}, 1, {""}, NO_ARGS, FOCUS_CONSOLE_END, NO_TIP},
|
||||
{"Save to flipper sdcard", // keep as last entry or change logic in callback below
|
||||
{""},
|
||||
@@ -149,6 +150,17 @@ static void wifi_marauder_scene_start_var_list_enter_callback(void* context, uin
|
||||
item->focus_console;
|
||||
app->show_stopscan_tip = item->show_stopscan_tip;
|
||||
|
||||
// TODO cleanup
|
||||
if(index == NUM_MENU_ITEMS - 3) {
|
||||
// flasher
|
||||
app->is_command = false;
|
||||
app->flash_mode = true;
|
||||
view_dispatcher_send_custom_event(app->view_dispatcher, WifiMarauderEventStartFlasher);
|
||||
return;
|
||||
}
|
||||
|
||||
app->flash_mode = false;
|
||||
|
||||
if(!app->is_command && selected_option_index == 0) {
|
||||
// View Log from start
|
||||
view_dispatcher_send_custom_event(app->view_dispatcher, WifiMarauderEventStartLogViewer);
|
||||
@@ -231,7 +243,6 @@ void wifi_marauder_scene_start_on_enter(void* context) {
|
||||
}
|
||||
|
||||
bool wifi_marauder_scene_start_on_event(void* context, SceneManagerEvent event) {
|
||||
UNUSED(context);
|
||||
WifiMarauderApp* app = context;
|
||||
bool consumed = false;
|
||||
|
||||
@@ -260,6 +271,10 @@ bool wifi_marauder_scene_start_on_event(void* context, SceneManagerEvent event)
|
||||
scene_manager_set_scene_state(
|
||||
app->scene_manager, WifiMarauderSceneStart, app->selected_menu_index);
|
||||
scene_manager_next_scene(app->scene_manager, WifiMarauderSceneSniffPmkidOptions);
|
||||
} else if(event.event == WifiMarauderEventStartFlasher) {
|
||||
scene_manager_set_scene_state(
|
||||
app->scene_manager, WifiMarauderSceneStart, app->selected_menu_index);
|
||||
scene_manager_next_scene(app->scene_manager, WifiMarauderSceneFlasher);
|
||||
}
|
||||
consumed = true;
|
||||
} else if(event.type == SceneManagerEventTypeTick) {
|
||||
|
||||
@@ -46,7 +46,9 @@ void wifi_marauder_scene_text_input_on_enter(void* context) {
|
||||
// Setup view
|
||||
WIFI_TextInput* text_input = app->text_input;
|
||||
// Add help message to header
|
||||
if(app->special_case_input_step == 1) {
|
||||
if(app->flash_mode) {
|
||||
wifi_text_input_set_header_text(text_input, "Enter destination address");
|
||||
} else if(app->special_case_input_step == 1) {
|
||||
wifi_text_input_set_header_text(text_input, "Enter source MAC");
|
||||
} else if(0 == strncmp("ssid -a -g", app->selected_tx_string, strlen("ssid -a -g"))) {
|
||||
wifi_text_input_set_header_text(text_input, "Enter # SSIDs to generate");
|
||||
|
||||
Reference in New Issue
Block a user