Skip to content

Commit

Permalink
Merge pull request #620 from ESurge/420
Browse files Browse the repository at this point in the history
  • Loading branch information
RogueMaster authored May 8, 2023
2 parents 1edacd1 + 6b759c1 commit bd616f4
Show file tree
Hide file tree
Showing 40 changed files with 2,988 additions and 19 deletions.
18 changes: 18 additions & 0 deletions applications/external/bad_bt/application.fam
Original file line number Diff line number Diff line change
@@ -0,0 +1,18 @@
App(
appid="bad_bt",
name="Bad BT",
apptype=FlipperAppType.EXTERNAL,
entry_point="bad_bt_app",
#cdefines=["APP_BAD_BT"],
requires=[
"gui",
"dialogs",
],
stack_size=2 * 1024,
order=70,
fap_libs=["assets"],
fap_category="Bluetooth",
fap_icon="images/bad_bt_10px.png",
fap_icon_assets="images",
fap_icon_assets_symbol="bad_bt",
)
323 changes: 323 additions & 0 deletions applications/external/bad_bt/bad_bt_app.c
Original file line number Diff line number Diff line change
@@ -0,0 +1,323 @@
#include "bad_bt_app.h"
#include "bad_bt_settings_filename.h"
#include <furi.h>
#include <furi_hal.h>
#include <storage/storage.h>
#include <lib/toolbox/path.h>
#include <lib/flipper_format/flipper_format.h>

#include <bt/bt_service/bt_i.h>
#include <bt/bt_service/bt.h>

#define BAD_BT_SETTINGS_PATH BAD_BT_APP_BASE_FOLDER "/" BAD_BT_SETTINGS_FILE_NAME

static bool bad_bt_app_custom_event_callback(void* context, uint32_t event) {
furi_assert(context);
BadBtApp* app = context;
return scene_manager_handle_custom_event(app->scene_manager, event);
}

static bool bad_bt_app_back_event_callback(void* context) {
furi_assert(context);
BadBtApp* app = context;
return scene_manager_handle_back_event(app->scene_manager);
}

static void bad_bt_app_tick_event_callback(void* context) {
furi_assert(context);
BadBtApp* app = context;
scene_manager_handle_tick_event(app->scene_manager);
}

static void bad_bt_load_settings(BadBtApp* app) {
furi_string_reset(app->keyboard_layout);
strcpy(app->config.bt_name, "");
memcpy(
app->config.bt_mac,
furi_hal_bt_get_profile_mac_addr(FuriHalBtProfileHidKeyboard),
BAD_BT_MAC_ADDRESS_LEN);

Storage* storage = furi_record_open(RECORD_STORAGE);
FlipperFormat* file = flipper_format_file_alloc(storage);
if(flipper_format_file_open_existing(file, BAD_BT_SETTINGS_PATH)) {
FuriString* tmp_str = furi_string_alloc();
if(!flipper_format_read_string(file, "Keyboard_Layout", app->keyboard_layout)) {
furi_string_reset(app->keyboard_layout);
}
if(!flipper_format_read_bool(file, "BT_Remember", &(app->bt_remember), 1)) {
app->bt_remember = false;
}
if(flipper_format_read_string(file, "Bt_Name", tmp_str) && !furi_string_empty(tmp_str)) {
strcpy(app->config.bt_name, furi_string_get_cstr(tmp_str));
} else {
strcpy(app->config.bt_name, "");
}
if(!flipper_format_read_hex(
file, "Bt_Mac", (uint8_t*)&app->config.bt_mac, BAD_BT_MAC_ADDRESS_LEN)) {
memcpy(
app->config.bt_mac,
furi_hal_bt_get_profile_mac_addr(FuriHalBtProfileHidKeyboard),
BAD_BT_MAC_ADDRESS_LEN);
}
furi_string_free(tmp_str);
flipper_format_file_close(file);
}
flipper_format_free(file);

if(!furi_string_empty(app->keyboard_layout)) {
FileInfo layout_file_info;
FS_Error file_check_err = storage_common_stat(
storage, furi_string_get_cstr(app->keyboard_layout), &layout_file_info);
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);
}
}

furi_record_close(RECORD_STORAGE);
}

static void bad_bt_save_settings(BadBtApp* app) {
Storage* storage = furi_record_open(RECORD_STORAGE);
FlipperFormat* file = flipper_format_file_alloc(storage);
if(flipper_format_file_open_always(file, BAD_BT_SETTINGS_PATH)) {
flipper_format_write_string(file, "Keyboard_Layout", app->keyboard_layout);
flipper_format_write_bool(file, "BT_Remember", &(app->bt_remember), 1);
flipper_format_write_string_cstr(file, "Bt_Name", app->config.bt_name);
flipper_format_write_hex(
file, "Bt_Mac", (uint8_t*)&app->config.bt_mac, BAD_BT_MAC_ADDRESS_LEN);
flipper_format_file_close(file);
}
flipper_format_free(file);
furi_record_close(RECORD_STORAGE);
}

void bad_bt_reload_worker(BadBtApp* app) {
bad_bt_script_close(app->bad_bt_script);
app->bad_bt_script = bad_bt_script_open(app->file_path, app->bt, app);
bad_bt_script_set_keyboard_layout(app->bad_bt_script, app->keyboard_layout);
}

int32_t bad_bt_config_switch_mode(BadBtApp* app) {
bad_bt_reload_worker(app);
furi_hal_bt_start_advertising();
scene_manager_next_scene(app->scene_manager, BadBtSceneConfig);
scene_manager_previous_scene(app->scene_manager);
return 0;
}

void bad_bt_config_switch_remember_mode(BadBtApp* app) {
if(app->bt_remember) {
furi_hal_bt_set_profile_pairing_method(
FuriHalBtProfileHidKeyboard, GapPairingPinCodeVerifyYesNo);
bt_set_profile_mac_address(app->bt, (uint8_t*)&BAD_BT_BOUND_MAC_ADDRESS);
bt_enable_peer_key_update(app->bt);
} else {
furi_hal_bt_set_profile_pairing_method(FuriHalBtProfileHidKeyboard, GapPairingNone);
bt_set_profile_mac_address(app->bt, app->config.bt_mac);
bt_disable_peer_key_update(app->bt);
}
bad_bt_reload_worker(app);
}

int32_t bad_bt_connection_init(BadBtApp* app) {
strcpy(
app->prev_config.bt_name, furi_hal_bt_get_profile_adv_name(FuriHalBtProfileHidKeyboard));
memcpy(
app->prev_config.bt_mac,
furi_hal_bt_get_profile_mac_addr(FuriHalBtProfileHidKeyboard),
BAD_BT_MAC_ADDRESS_LEN);
app->prev_config.bt_mode = furi_hal_bt_get_profile_pairing_method(FuriHalBtProfileHidKeyboard);

bt_timeout = bt_hid_delays[LevelRssi39_0];
bt_disconnect(app->bt);
bt_keys_storage_set_storage_path(app->bt, BAD_BT_APP_PATH_BOUND_KEYS_FILE);
if(strcmp(app->config.bt_name, "") != 0) {
furi_hal_bt_set_profile_adv_name(FuriHalBtProfileHidKeyboard, app->config.bt_name);
}
if(app->bt_remember) {
furi_hal_bt_set_profile_mac_addr(
FuriHalBtProfileHidKeyboard, (uint8_t*)&BAD_BT_BOUND_MAC_ADDRESS);
furi_hal_bt_set_profile_pairing_method(
FuriHalBtProfileHidKeyboard, GapPairingPinCodeVerifyYesNo);
} else {
if(memcmp(
app->config.bt_mac, (uint8_t*)&BAD_BT_EMPTY_MAC_ADDRESS, BAD_BT_MAC_ADDRESS_LEN) !=
0) {
furi_hal_bt_set_profile_mac_addr(FuriHalBtProfileHidKeyboard, app->config.bt_mac);
}
furi_hal_bt_set_profile_pairing_method(FuriHalBtProfileHidKeyboard, GapPairingNone);
}
bt_set_profile(app->bt, BtProfileHidKeyboard);
if(strcmp(app->config.bt_name, "") == 0) {
strcpy(app->config.bt_name, furi_hal_bt_get_profile_adv_name(FuriHalBtProfileHidKeyboard));
}
if(memcmp(app->config.bt_mac, (uint8_t*)&BAD_BT_EMPTY_MAC_ADDRESS, BAD_BT_MAC_ADDRESS_LEN) ==
0) {
memcpy(
app->config.bt_mac,
furi_hal_bt_get_profile_mac_addr(FuriHalBtProfileHidKeyboard),
BAD_BT_MAC_ADDRESS_LEN);
}

furi_hal_bt_start_advertising();
if(app->bt_remember) {
bt_enable_peer_key_update(app->bt);
} else {
bt_disable_peer_key_update(app->bt);
}

return 0;
}

void bad_bt_connection_deinit(BadBtApp* app) {
bt_disconnect(app->bt);
bt_keys_storage_set_default_path(app->bt);
furi_hal_bt_set_profile_adv_name(FuriHalBtProfileHidKeyboard, app->prev_config.bt_name);
furi_hal_bt_set_profile_mac_addr(FuriHalBtProfileHidKeyboard, app->prev_config.bt_mac);
furi_hal_bt_set_profile_pairing_method(FuriHalBtProfileHidKeyboard, app->prev_config.bt_mode);
bt_set_profile(app->bt, BtProfileSerial);
bt_enable_peer_key_update(app->bt);
}

BadBtApp* bad_bt_app_alloc(char* arg) {
BadBtApp* app = malloc(sizeof(BadBtApp));

app->bad_bt_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);
storage_simply_mkdir(storage, BAD_BT_APP_BASE_FOLDER);
furi_record_close(RECORD_STORAGE);

bad_bt_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_bt_scene_handlers, app);

view_dispatcher_set_event_callback_context(app->view_dispatcher, app);
view_dispatcher_set_tick_event_callback(
app->view_dispatcher, bad_bt_app_tick_event_callback, 500);
view_dispatcher_set_custom_event_callback(
app->view_dispatcher, bad_bt_app_custom_event_callback);
view_dispatcher_set_navigation_event_callback(
app->view_dispatcher, bad_bt_app_back_event_callback);

Bt* bt = furi_record_open(RECORD_BT);
app->bt = bt;
app->bt->suppress_pin_screen = true;

// Custom Widget
app->widget = widget_alloc();
view_dispatcher_add_view(
app->view_dispatcher, BadBtAppViewError, widget_get_view(app->widget));

app->var_item_list = variable_item_list_alloc();
view_dispatcher_add_view(
app->view_dispatcher, BadBtAppViewConfig, variable_item_list_get_view(app->var_item_list));

app->bad_bt_view = bad_bt_alloc();
view_dispatcher_add_view(
app->view_dispatcher, BadBtAppViewWork, bad_bt_get_view(app->bad_bt_view));

app->text_input = text_input_alloc();
view_dispatcher_add_view(
app->view_dispatcher, BadBtAppViewConfigName, text_input_get_view(app->text_input));

app->byte_input = byte_input_alloc();
view_dispatcher_add_view(
app->view_dispatcher, BadBtAppViewConfigMac, byte_input_get_view(app->byte_input));

view_dispatcher_attach_to_gui(app->view_dispatcher, app->gui, ViewDispatcherTypeFullscreen);

app->conn_init_thread = furi_thread_alloc_ex(
"BadBtConnInit", 1024, (FuriThreadCallback)bad_bt_connection_init, app);
furi_thread_start(app->conn_init_thread);
if(!furi_string_empty(app->file_path)) {
app->bad_bt_script = bad_bt_script_open(app->file_path, app->bt, app);
bad_bt_script_set_keyboard_layout(app->bad_bt_script, app->keyboard_layout);
scene_manager_next_scene(app->scene_manager, BadBtSceneWork);
} else {
furi_string_set(app->file_path, BAD_BT_APP_BASE_FOLDER);
scene_manager_next_scene(app->scene_manager, BadBtSceneFileSelect);
}

return app;
}

void bad_bt_app_free(BadBtApp* app) {
furi_assert(app);

if(app->bad_bt_script) {
bad_bt_script_close(app->bad_bt_script);
app->bad_bt_script = NULL;
}

// Views
view_dispatcher_remove_view(app->view_dispatcher, BadBtAppViewWork);
bad_bt_free(app->bad_bt_view);

// Custom Widget
view_dispatcher_remove_view(app->view_dispatcher, BadBtAppViewError);
widget_free(app->widget);

// Variable item list
view_dispatcher_remove_view(app->view_dispatcher, BadBtAppViewConfig);
variable_item_list_free(app->var_item_list);

// Text Input
view_dispatcher_remove_view(app->view_dispatcher, BadBtAppViewConfigName);
text_input_free(app->text_input);

// Byte Input
view_dispatcher_remove_view(app->view_dispatcher, BadBtAppViewConfigMac);
byte_input_free(app->byte_input);

// View dispatcher
view_dispatcher_free(app->view_dispatcher);
scene_manager_free(app->scene_manager);

// Restore bt config
app->bt->suppress_pin_screen = false;
if(app->conn_init_thread) {
furi_thread_join(app->conn_init_thread);
furi_thread_free(app->conn_init_thread);
bad_bt_connection_deinit(app);
}

// Close records
furi_record_close(RECORD_GUI);
furi_record_close(RECORD_NOTIFICATION);
furi_record_close(RECORD_DIALOGS);
furi_record_close(RECORD_BT);

bad_bt_save_settings(app);

furi_string_free(app->file_path);
furi_string_free(app->keyboard_layout);

free(app);
}

int32_t bad_bt_app(void* p) {
BadBtApp* bad_bt_app = bad_bt_app_alloc((char*)p);

view_dispatcher_run(bad_bt_app->view_dispatcher);

bad_bt_app_free(bad_bt_app);
return 0;
}
36 changes: 36 additions & 0 deletions applications/external/bad_bt/bad_bt_app.h
Original file line number Diff line number Diff line change
@@ -0,0 +1,36 @@
#pragma once

#include "scenes/bad_bt_scene.h"
#include "helpers/ducky_script.h"

#include <gui/gui.h>
#include <assets_icons.h>
#include <gui/scene_manager.h>
#include <dialogs/dialogs.h>
#include <notification/notification_messages.h>
#include "bad_bt_icons.h"

#define BAD_BT_APP_BASE_FOLDER EXT_PATH("badbt")
#define BAD_BT_APP_PATH_LAYOUT_FOLDER BAD_BT_APP_BASE_FOLDER "/assets/layouts"
#define BAD_BT_APP_SCRIPT_EXTENSION ".txt"
#define BAD_BT_APP_LAYOUT_EXTENSION ".kl"

typedef enum BadBtCustomEvent {
BadBtAppCustomEventTextEditResult,
BadBtAppCustomEventByteInputDone,
BadBtCustomEventErrorBack
} BadBtCustomEvent;

typedef enum {
BadBtAppViewError,
BadBtAppViewWork,
BadBtAppViewConfig,
BadBtAppViewConfigMac,
BadBtAppViewConfigName
} BadBtAppView;

void bad_bt_config_switch_remember_mode(BadBtApp* app);

int32_t bad_bt_connection_init(BadBtApp* app);

void bad_bt_connection_deinit(BadBtApp* app);
7 changes: 7 additions & 0 deletions applications/external/bad_bt/bad_bt_settings_filename.h
Original file line number Diff line number Diff line change
@@ -0,0 +1,7 @@
#pragma once

#include <storage/storage.h>

#define BAD_BT_SETTINGS_FILE_NAME ".badbt.settings"
#define BAD_BT_APP_PATH_BOUND_KEYS_FOLDER EXT_PATH("badbt/.keys")
#define BAD_BT_APP_PATH_BOUND_KEYS_FILE BAD_BT_APP_PATH_BOUND_KEYS_FOLDER "/.devices.keys"
Loading

0 comments on commit bd616f4

Please sign in to comment.