mirror of
https://github.com/Next-Flip/Momentum-Firmware.git
synced 2026-04-28 03:59:58 -07:00
257 lines
9.3 KiB
C
257 lines
9.3 KiB
C
#include "bad_kb_app_i.h"
|
|
#include "bad_kb_settings_filename.h"
|
|
#include <furi.h>
|
|
#include <furi_hal.h>
|
|
#include <storage/storage.h>
|
|
#include <lib/toolbox/path.h>
|
|
#include <xtreme/settings.h>
|
|
|
|
#include <bt/bt_service/bt_i.h>
|
|
#include <bt/bt_service/bt.h>
|
|
|
|
#define BAD_KB_SETTINGS_PATH BAD_KB_APP_BASE_FOLDER "/" BAD_KB_SETTINGS_FILE_NAME
|
|
|
|
static bool bad_kb_app_custom_event_callback(void* context, uint32_t event) {
|
|
furi_assert(context);
|
|
BadKbApp* app = context;
|
|
return scene_manager_handle_custom_event(app->scene_manager, event);
|
|
}
|
|
|
|
static bool bad_kb_app_back_event_callback(void* context) {
|
|
furi_assert(context);
|
|
BadKbApp* app = context;
|
|
return scene_manager_handle_back_event(app->scene_manager);
|
|
}
|
|
|
|
static void bad_kb_app_tick_event_callback(void* context) {
|
|
furi_assert(context);
|
|
BadKbApp* app = context;
|
|
scene_manager_handle_tick_event(app->scene_manager);
|
|
}
|
|
|
|
static void bad_kb_load_settings(BadKbApp* app) {
|
|
File* settings_file = storage_file_alloc(furi_record_open(RECORD_STORAGE));
|
|
if(storage_file_open(settings_file, BAD_KB_SETTINGS_PATH, FSAM_READ, FSOM_OPEN_EXISTING)) {
|
|
char chr;
|
|
while((storage_file_read(settings_file, &chr, 1) == 1) &&
|
|
!storage_file_eof(settings_file) && !isspace(chr)) {
|
|
furi_string_push_back(app->keyboard_layout, chr);
|
|
}
|
|
} else {
|
|
furi_string_reset(app->keyboard_layout);
|
|
}
|
|
storage_file_close(settings_file);
|
|
storage_file_free(settings_file);
|
|
|
|
if(!furi_string_empty(app->keyboard_layout)) {
|
|
Storage* fs_api = furi_record_open(RECORD_STORAGE);
|
|
FileInfo layout_file_info;
|
|
FS_Error file_check_err = storage_common_stat(
|
|
fs_api, furi_string_get_cstr(app->keyboard_layout), &layout_file_info);
|
|
furi_record_close(RECORD_STORAGE);
|
|
if(file_check_err != FSE_OK) {
|
|
furi_string_reset(app->keyboard_layout);
|
|
return;
|
|
}
|
|
if(layout_file_info.size != 256) {
|
|
furi_string_reset(app->keyboard_layout);
|
|
}
|
|
}
|
|
}
|
|
|
|
static void bad_kb_save_settings(BadKbApp* app) {
|
|
File* settings_file = storage_file_alloc(furi_record_open(RECORD_STORAGE));
|
|
if(storage_file_open(settings_file, BAD_KB_SETTINGS_PATH, FSAM_WRITE, FSOM_OPEN_ALWAYS)) {
|
|
storage_file_write(
|
|
settings_file,
|
|
furi_string_get_cstr(app->keyboard_layout),
|
|
furi_string_size(app->keyboard_layout));
|
|
storage_file_write(settings_file, "\n", 1);
|
|
}
|
|
storage_file_close(settings_file);
|
|
storage_file_free(settings_file);
|
|
}
|
|
|
|
BadKbApp* bad_kb_app_alloc(char* arg) {
|
|
BadKbApp* app = malloc(sizeof(BadKbApp));
|
|
|
|
app->bad_kb_script = NULL;
|
|
|
|
app->file_path = furi_string_alloc();
|
|
app->keyboard_layout = furi_string_alloc();
|
|
if(arg && strlen(arg)) {
|
|
furi_string_set(app->file_path, arg);
|
|
}
|
|
|
|
Storage* storage = furi_record_open(RECORD_STORAGE);
|
|
// Remove old pre-included files to avoid duplicates on migrate
|
|
storage_simply_remove(storage, EXT_PATH("badusb/layouts"));
|
|
storage_simply_remove(storage, EXT_PATH("badusb/.badusb.settings"));
|
|
storage_simply_remove(storage, EXT_PATH("badusb/Kiosk-Evasion-Bruteforce.txt"));
|
|
storage_simply_remove(storage, EXT_PATH("badusb/Wifi-Stealer_ORG.txt"));
|
|
storage_simply_remove(storage, EXT_PATH("badusb/demo_macos.txt"));
|
|
storage_simply_remove(storage, EXT_PATH("badusb/demo_windows.txt"));
|
|
storage_common_migrate(storage, EXT_PATH("badusb"), BAD_KB_APP_BASE_FOLDER);
|
|
storage_simply_mkdir(storage, BAD_KB_APP_BASE_FOLDER);
|
|
furi_record_close(RECORD_STORAGE);
|
|
|
|
bad_kb_load_settings(app);
|
|
|
|
app->gui = furi_record_open(RECORD_GUI);
|
|
app->notifications = furi_record_open(RECORD_NOTIFICATION);
|
|
app->dialogs = furi_record_open(RECORD_DIALOGS);
|
|
|
|
app->view_dispatcher = view_dispatcher_alloc();
|
|
view_dispatcher_enable_queue(app->view_dispatcher);
|
|
|
|
app->scene_manager = scene_manager_alloc(&bad_kb_scene_handlers, app);
|
|
|
|
view_dispatcher_set_event_callback_context(app->view_dispatcher, app);
|
|
view_dispatcher_set_tick_event_callback(
|
|
app->view_dispatcher, bad_kb_app_tick_event_callback, 500);
|
|
view_dispatcher_set_custom_event_callback(
|
|
app->view_dispatcher, bad_kb_app_custom_event_callback);
|
|
view_dispatcher_set_navigation_event_callback(
|
|
app->view_dispatcher, bad_kb_app_back_event_callback);
|
|
|
|
app->connection_init = false;
|
|
|
|
Bt* bt = furi_record_open(RECORD_BT);
|
|
app->bt = bt;
|
|
app->bt->suppress_pin_screen = true;
|
|
app->is_bt = XTREME_SETTINGS()->bad_bt;
|
|
app->bt_remember = XTREME_SETTINGS()->bad_bt_remember;
|
|
const char* adv_name = furi_hal_bt_get_profile_adv_name(FuriHalBtProfileHidKeyboard);
|
|
memcpy(app->name, adv_name, BAD_KB_ADV_NAME_MAX_LEN);
|
|
memcpy(app->bt_old_config.name, adv_name, BAD_KB_ADV_NAME_MAX_LEN);
|
|
|
|
// need to be done before bt init (where mac address get modified if bounding is activated)
|
|
const uint8_t* mac_addr = furi_hal_bt_get_profile_mac_addr(FuriHalBtProfileHidKeyboard);
|
|
memcpy(app->mac, mac_addr, BAD_KB_MAC_ADDRESS_LEN);
|
|
memcpy(app->bt_old_config.mac, mac_addr, BAD_KB_MAC_ADDRESS_LEN);
|
|
|
|
// Custom Widget
|
|
app->widget = widget_alloc();
|
|
view_dispatcher_add_view(
|
|
app->view_dispatcher, BadKbAppViewError, widget_get_view(app->widget));
|
|
|
|
app->var_item_list_bt = variable_item_list_alloc();
|
|
view_dispatcher_add_view(
|
|
app->view_dispatcher,
|
|
BadKbAppViewConfigBt,
|
|
variable_item_list_get_view(app->var_item_list_bt));
|
|
app->var_item_list_usb = variable_item_list_alloc();
|
|
view_dispatcher_add_view(
|
|
app->view_dispatcher,
|
|
BadKbAppViewConfigUsb,
|
|
variable_item_list_get_view(app->var_item_list_usb));
|
|
|
|
app->bad_kb_view = bad_kb_alloc();
|
|
view_dispatcher_add_view(
|
|
app->view_dispatcher, BadKbAppViewWork, bad_kb_get_view(app->bad_kb_view));
|
|
|
|
app->text_input = text_input_alloc();
|
|
view_dispatcher_add_view(
|
|
app->view_dispatcher, BadKbAppViewConfigName, text_input_get_view(app->text_input));
|
|
|
|
app->byte_input = byte_input_alloc();
|
|
view_dispatcher_add_view(
|
|
app->view_dispatcher, BadKbAppViewConfigMac, byte_input_get_view(app->byte_input));
|
|
|
|
view_dispatcher_attach_to_gui(app->view_dispatcher, app->gui, ViewDispatcherTypeFullscreen);
|
|
|
|
if(furi_hal_usb_is_locked()) {
|
|
app->error = BadKbAppErrorCloseRpc;
|
|
app->conn_init_thread = NULL;
|
|
scene_manager_next_scene(app->scene_manager, BadKbSceneError);
|
|
} else {
|
|
app->conn_init_thread = furi_thread_alloc_ex(
|
|
"BadKbConnInit", 512, (FuriThreadCallback)bad_kb_connection_init, app);
|
|
furi_thread_start(app->conn_init_thread);
|
|
if(!furi_string_empty(app->file_path)) {
|
|
app->bad_kb_script = bad_kb_script_open(app->file_path, app->is_bt ? app->bt : NULL);
|
|
bad_kb_script_set_keyboard_layout(app->bad_kb_script, app->keyboard_layout);
|
|
scene_manager_next_scene(app->scene_manager, BadKbSceneWork);
|
|
} else {
|
|
furi_string_set(app->file_path, BAD_KB_APP_BASE_FOLDER);
|
|
scene_manager_next_scene(app->scene_manager, BadKbSceneFileSelect);
|
|
}
|
|
}
|
|
|
|
return app;
|
|
}
|
|
|
|
void bad_kb_app_free(BadKbApp* app) {
|
|
furi_assert(app);
|
|
|
|
if(app->bad_kb_script) {
|
|
bad_kb_script_close(app->bad_kb_script);
|
|
app->bad_kb_script = NULL;
|
|
}
|
|
|
|
// Views
|
|
view_dispatcher_remove_view(app->view_dispatcher, BadKbAppViewWork);
|
|
bad_kb_free(app->bad_kb_view);
|
|
|
|
// Custom Widget
|
|
view_dispatcher_remove_view(app->view_dispatcher, BadKbAppViewError);
|
|
widget_free(app->widget);
|
|
|
|
// Variable item list
|
|
view_dispatcher_remove_view(app->view_dispatcher, BadKbAppViewConfigBt);
|
|
variable_item_list_free(app->var_item_list_bt);
|
|
view_dispatcher_remove_view(app->view_dispatcher, BadKbAppViewConfigUsb);
|
|
variable_item_list_free(app->var_item_list_usb);
|
|
|
|
// Text Input
|
|
view_dispatcher_remove_view(app->view_dispatcher, BadKbAppViewConfigName);
|
|
text_input_free(app->text_input);
|
|
|
|
// Byte Input
|
|
view_dispatcher_remove_view(app->view_dispatcher, BadKbAppViewConfigMac);
|
|
byte_input_free(app->byte_input);
|
|
|
|
// View dispatcher
|
|
view_dispatcher_free(app->view_dispatcher);
|
|
scene_manager_free(app->scene_manager);
|
|
|
|
// Restore bt config
|
|
// BtProfile has already been switched to the previous one
|
|
// So we directly modify the right profile
|
|
if(strcmp(app->bt_old_config.name, app->name) != 0) {
|
|
furi_hal_bt_set_profile_adv_name(FuriHalBtProfileHidKeyboard, app->bt_old_config.name);
|
|
}
|
|
if(memcmp(app->bt_old_config.mac, app->mac, BAD_KB_MAC_ADDRESS_LEN) != 0) {
|
|
furi_hal_bt_set_profile_mac_addr(FuriHalBtProfileHidKeyboard, app->bt_old_config.mac);
|
|
}
|
|
app->bt->suppress_pin_screen = false;
|
|
|
|
// Close records
|
|
furi_record_close(RECORD_GUI);
|
|
furi_record_close(RECORD_NOTIFICATION);
|
|
furi_record_close(RECORD_DIALOGS);
|
|
furi_record_close(RECORD_BT);
|
|
|
|
bad_kb_save_settings(app);
|
|
|
|
furi_string_free(app->file_path);
|
|
furi_string_free(app->keyboard_layout);
|
|
|
|
if(app->conn_init_thread) {
|
|
furi_thread_join(app->conn_init_thread);
|
|
furi_thread_free(app->conn_init_thread);
|
|
}
|
|
bad_kb_connection_deinit(app);
|
|
|
|
free(app);
|
|
}
|
|
|
|
int32_t bad_kb_app(void* p) {
|
|
BadKbApp* bad_kb_app = bad_kb_app_alloc((char*)p);
|
|
|
|
view_dispatcher_run(bad_kb_app->view_dispatcher);
|
|
|
|
bad_kb_app_free(bad_kb_app);
|
|
return 0;
|
|
}
|