mirror of
https://github.com/DarkFlippers/unleashed-firmware
synced 2024-11-23 21:13:16 +00:00
4cee550cc6
* bt: disconnect first on profile change * bt keys: rework bt keys * saved struct: add payload size getter to API * bt: rework bt with new key storage API * bt: add keys storage operation to bt API * hid: save bt keys on sd card * bt: add unit tests for key storage * bt: working profile switch * bt: cleanup * bt hid: change keys storage path Co-authored-by: あく <alleteam@gmail.com>
404 lines
13 KiB
C
404 lines
13 KiB
C
#include "hid.h"
|
|
#include "views.h"
|
|
#include <notification/notification_messages.h>
|
|
#include <dolphin/dolphin.h>
|
|
|
|
#define TAG "HidApp"
|
|
|
|
enum HidDebugSubmenuIndex {
|
|
HidSubmenuIndexKeynote,
|
|
HidSubmenuIndexKeyboard,
|
|
HidSubmenuIndexMedia,
|
|
HidSubmenuIndexTikTok,
|
|
HidSubmenuIndexMouse,
|
|
HidSubmenuIndexMouseJiggler,
|
|
};
|
|
|
|
static void hid_submenu_callback(void* context, uint32_t index) {
|
|
furi_assert(context);
|
|
Hid* app = context;
|
|
if(index == HidSubmenuIndexKeynote) {
|
|
app->view_id = HidViewKeynote;
|
|
view_dispatcher_switch_to_view(app->view_dispatcher, HidViewKeynote);
|
|
} else if(index == HidSubmenuIndexKeyboard) {
|
|
app->view_id = HidViewKeyboard;
|
|
view_dispatcher_switch_to_view(app->view_dispatcher, HidViewKeyboard);
|
|
} else if(index == HidSubmenuIndexMedia) {
|
|
app->view_id = HidViewMedia;
|
|
view_dispatcher_switch_to_view(app->view_dispatcher, HidViewMedia);
|
|
} else if(index == HidSubmenuIndexMouse) {
|
|
app->view_id = HidViewMouse;
|
|
view_dispatcher_switch_to_view(app->view_dispatcher, HidViewMouse);
|
|
} else if(index == HidSubmenuIndexTikTok) {
|
|
app->view_id = BtHidViewTikTok;
|
|
view_dispatcher_switch_to_view(app->view_dispatcher, BtHidViewTikTok);
|
|
} else if(index == HidSubmenuIndexMouseJiggler) {
|
|
app->view_id = HidViewMouseJiggler;
|
|
view_dispatcher_switch_to_view(app->view_dispatcher, HidViewMouseJiggler);
|
|
}
|
|
}
|
|
|
|
static void bt_hid_connection_status_changed_callback(BtStatus status, void* context) {
|
|
furi_assert(context);
|
|
Hid* hid = context;
|
|
bool connected = (status == BtStatusConnected);
|
|
if(connected) {
|
|
notification_internal_message(hid->notifications, &sequence_set_blue_255);
|
|
} else {
|
|
notification_internal_message(hid->notifications, &sequence_reset_blue);
|
|
}
|
|
hid_keynote_set_connected_status(hid->hid_keynote, connected);
|
|
hid_keyboard_set_connected_status(hid->hid_keyboard, connected);
|
|
hid_media_set_connected_status(hid->hid_media, connected);
|
|
hid_mouse_set_connected_status(hid->hid_mouse, connected);
|
|
hid_mouse_jiggler_set_connected_status(hid->hid_mouse_jiggler, connected);
|
|
hid_tiktok_set_connected_status(hid->hid_tiktok, connected);
|
|
}
|
|
|
|
static void hid_dialog_callback(DialogExResult result, void* context) {
|
|
furi_assert(context);
|
|
Hid* app = context;
|
|
if(result == DialogExResultLeft) {
|
|
view_dispatcher_stop(app->view_dispatcher);
|
|
} else if(result == DialogExResultRight) {
|
|
view_dispatcher_switch_to_view(app->view_dispatcher, app->view_id); // Show last view
|
|
} else if(result == DialogExResultCenter) {
|
|
view_dispatcher_switch_to_view(app->view_dispatcher, HidViewSubmenu);
|
|
}
|
|
}
|
|
|
|
static uint32_t hid_exit_confirm_view(void* context) {
|
|
UNUSED(context);
|
|
return HidViewExitConfirm;
|
|
}
|
|
|
|
static uint32_t hid_exit(void* context) {
|
|
UNUSED(context);
|
|
return VIEW_NONE;
|
|
}
|
|
|
|
Hid* hid_alloc(HidTransport transport) {
|
|
Hid* app = malloc(sizeof(Hid));
|
|
app->transport = transport;
|
|
|
|
// Gui
|
|
app->gui = furi_record_open(RECORD_GUI);
|
|
|
|
// Bt
|
|
app->bt = furi_record_open(RECORD_BT);
|
|
|
|
// Notifications
|
|
app->notifications = furi_record_open(RECORD_NOTIFICATION);
|
|
|
|
// View dispatcher
|
|
app->view_dispatcher = view_dispatcher_alloc();
|
|
view_dispatcher_enable_queue(app->view_dispatcher);
|
|
view_dispatcher_attach_to_gui(app->view_dispatcher, app->gui, ViewDispatcherTypeFullscreen);
|
|
// Device Type Submenu view
|
|
app->device_type_submenu = submenu_alloc();
|
|
submenu_add_item(
|
|
app->device_type_submenu, "Keynote", HidSubmenuIndexKeynote, hid_submenu_callback, app);
|
|
submenu_add_item(
|
|
app->device_type_submenu, "Keyboard", HidSubmenuIndexKeyboard, hid_submenu_callback, app);
|
|
submenu_add_item(
|
|
app->device_type_submenu, "Media", HidSubmenuIndexMedia, hid_submenu_callback, app);
|
|
submenu_add_item(
|
|
app->device_type_submenu, "Mouse", HidSubmenuIndexMouse, hid_submenu_callback, app);
|
|
if(app->transport == HidTransportBle) {
|
|
submenu_add_item(
|
|
app->device_type_submenu,
|
|
"TikTok Controller",
|
|
HidSubmenuIndexTikTok,
|
|
hid_submenu_callback,
|
|
app);
|
|
}
|
|
submenu_add_item(
|
|
app->device_type_submenu,
|
|
"Mouse Jiggler",
|
|
HidSubmenuIndexMouseJiggler,
|
|
hid_submenu_callback,
|
|
app);
|
|
view_set_previous_callback(submenu_get_view(app->device_type_submenu), hid_exit);
|
|
view_dispatcher_add_view(
|
|
app->view_dispatcher, HidViewSubmenu, submenu_get_view(app->device_type_submenu));
|
|
app->view_id = HidViewSubmenu;
|
|
view_dispatcher_switch_to_view(app->view_dispatcher, app->view_id);
|
|
return app;
|
|
}
|
|
|
|
Hid* hid_app_alloc_view(void* context) {
|
|
furi_assert(context);
|
|
Hid* app = context;
|
|
// Dialog view
|
|
app->dialog = dialog_ex_alloc();
|
|
dialog_ex_set_result_callback(app->dialog, hid_dialog_callback);
|
|
dialog_ex_set_context(app->dialog, app);
|
|
dialog_ex_set_left_button_text(app->dialog, "Exit");
|
|
dialog_ex_set_right_button_text(app->dialog, "Stay");
|
|
dialog_ex_set_center_button_text(app->dialog, "Menu");
|
|
dialog_ex_set_header(app->dialog, "Close Current App?", 16, 12, AlignLeft, AlignTop);
|
|
view_dispatcher_add_view(
|
|
app->view_dispatcher, HidViewExitConfirm, dialog_ex_get_view(app->dialog));
|
|
|
|
// Keynote view
|
|
app->hid_keynote = hid_keynote_alloc(app);
|
|
view_set_previous_callback(hid_keynote_get_view(app->hid_keynote), hid_exit_confirm_view);
|
|
view_dispatcher_add_view(
|
|
app->view_dispatcher, HidViewKeynote, hid_keynote_get_view(app->hid_keynote));
|
|
|
|
// Keyboard view
|
|
app->hid_keyboard = hid_keyboard_alloc(app);
|
|
view_set_previous_callback(hid_keyboard_get_view(app->hid_keyboard), hid_exit_confirm_view);
|
|
view_dispatcher_add_view(
|
|
app->view_dispatcher, HidViewKeyboard, hid_keyboard_get_view(app->hid_keyboard));
|
|
|
|
// Media view
|
|
app->hid_media = hid_media_alloc(app);
|
|
view_set_previous_callback(hid_media_get_view(app->hid_media), hid_exit_confirm_view);
|
|
view_dispatcher_add_view(
|
|
app->view_dispatcher, HidViewMedia, hid_media_get_view(app->hid_media));
|
|
|
|
// TikTok view
|
|
app->hid_tiktok = hid_tiktok_alloc(app);
|
|
view_set_previous_callback(hid_tiktok_get_view(app->hid_tiktok), hid_exit_confirm_view);
|
|
view_dispatcher_add_view(
|
|
app->view_dispatcher, BtHidViewTikTok, hid_tiktok_get_view(app->hid_tiktok));
|
|
|
|
// Mouse view
|
|
app->hid_mouse = hid_mouse_alloc(app);
|
|
view_set_previous_callback(hid_mouse_get_view(app->hid_mouse), hid_exit_confirm_view);
|
|
view_dispatcher_add_view(
|
|
app->view_dispatcher, HidViewMouse, hid_mouse_get_view(app->hid_mouse));
|
|
|
|
// Mouse jiggler view
|
|
app->hid_mouse_jiggler = hid_mouse_jiggler_alloc(app);
|
|
view_set_previous_callback(
|
|
hid_mouse_jiggler_get_view(app->hid_mouse_jiggler), hid_exit_confirm_view);
|
|
view_dispatcher_add_view(
|
|
app->view_dispatcher,
|
|
HidViewMouseJiggler,
|
|
hid_mouse_jiggler_get_view(app->hid_mouse_jiggler));
|
|
|
|
return app;
|
|
}
|
|
|
|
void hid_free(Hid* app) {
|
|
furi_assert(app);
|
|
|
|
// Reset notification
|
|
notification_internal_message(app->notifications, &sequence_reset_blue);
|
|
|
|
// Free views
|
|
view_dispatcher_remove_view(app->view_dispatcher, HidViewSubmenu);
|
|
submenu_free(app->device_type_submenu);
|
|
view_dispatcher_remove_view(app->view_dispatcher, HidViewExitConfirm);
|
|
dialog_ex_free(app->dialog);
|
|
view_dispatcher_remove_view(app->view_dispatcher, HidViewKeynote);
|
|
hid_keynote_free(app->hid_keynote);
|
|
view_dispatcher_remove_view(app->view_dispatcher, HidViewKeyboard);
|
|
hid_keyboard_free(app->hid_keyboard);
|
|
view_dispatcher_remove_view(app->view_dispatcher, HidViewMedia);
|
|
hid_media_free(app->hid_media);
|
|
view_dispatcher_remove_view(app->view_dispatcher, HidViewMouse);
|
|
hid_mouse_free(app->hid_mouse);
|
|
view_dispatcher_remove_view(app->view_dispatcher, HidViewMouseJiggler);
|
|
hid_mouse_jiggler_free(app->hid_mouse_jiggler);
|
|
view_dispatcher_remove_view(app->view_dispatcher, BtHidViewTikTok);
|
|
hid_tiktok_free(app->hid_tiktok);
|
|
view_dispatcher_free(app->view_dispatcher);
|
|
|
|
// Close records
|
|
furi_record_close(RECORD_GUI);
|
|
app->gui = NULL;
|
|
furi_record_close(RECORD_NOTIFICATION);
|
|
app->notifications = NULL;
|
|
furi_record_close(RECORD_BT);
|
|
app->bt = NULL;
|
|
|
|
// Free rest
|
|
free(app);
|
|
}
|
|
|
|
void hid_hal_keyboard_press(Hid* instance, uint16_t event) {
|
|
furi_assert(instance);
|
|
if(instance->transport == HidTransportBle) {
|
|
furi_hal_bt_hid_kb_press(event);
|
|
} else if(instance->transport == HidTransportUsb) {
|
|
furi_hal_hid_kb_press(event);
|
|
} else {
|
|
furi_crash(NULL);
|
|
}
|
|
}
|
|
|
|
void hid_hal_keyboard_release(Hid* instance, uint16_t event) {
|
|
furi_assert(instance);
|
|
if(instance->transport == HidTransportBle) {
|
|
furi_hal_bt_hid_kb_release(event);
|
|
} else if(instance->transport == HidTransportUsb) {
|
|
furi_hal_hid_kb_release(event);
|
|
} else {
|
|
furi_crash(NULL);
|
|
}
|
|
}
|
|
|
|
void hid_hal_keyboard_release_all(Hid* instance) {
|
|
furi_assert(instance);
|
|
if(instance->transport == HidTransportBle) {
|
|
furi_hal_bt_hid_kb_release_all();
|
|
} else if(instance->transport == HidTransportUsb) {
|
|
furi_hal_hid_kb_release_all();
|
|
} else {
|
|
furi_crash(NULL);
|
|
}
|
|
}
|
|
|
|
void hid_hal_consumer_key_press(Hid* instance, uint16_t event) {
|
|
furi_assert(instance);
|
|
if(instance->transport == HidTransportBle) {
|
|
furi_hal_bt_hid_consumer_key_press(event);
|
|
} else if(instance->transport == HidTransportUsb) {
|
|
furi_hal_hid_consumer_key_press(event);
|
|
} else {
|
|
furi_crash(NULL);
|
|
}
|
|
}
|
|
|
|
void hid_hal_consumer_key_release(Hid* instance, uint16_t event) {
|
|
furi_assert(instance);
|
|
if(instance->transport == HidTransportBle) {
|
|
furi_hal_bt_hid_consumer_key_release(event);
|
|
} else if(instance->transport == HidTransportUsb) {
|
|
furi_hal_hid_consumer_key_release(event);
|
|
} else {
|
|
furi_crash(NULL);
|
|
}
|
|
}
|
|
|
|
void hid_hal_consumer_key_release_all(Hid* instance) {
|
|
furi_assert(instance);
|
|
if(instance->transport == HidTransportBle) {
|
|
furi_hal_bt_hid_consumer_key_release_all();
|
|
} else if(instance->transport == HidTransportUsb) {
|
|
furi_hal_hid_kb_release_all();
|
|
} else {
|
|
furi_crash(NULL);
|
|
}
|
|
}
|
|
|
|
void hid_hal_mouse_move(Hid* instance, int8_t dx, int8_t dy) {
|
|
furi_assert(instance);
|
|
if(instance->transport == HidTransportBle) {
|
|
furi_hal_bt_hid_mouse_move(dx, dy);
|
|
} else if(instance->transport == HidTransportUsb) {
|
|
furi_hal_hid_mouse_move(dx, dy);
|
|
} else {
|
|
furi_crash(NULL);
|
|
}
|
|
}
|
|
|
|
void hid_hal_mouse_scroll(Hid* instance, int8_t delta) {
|
|
furi_assert(instance);
|
|
if(instance->transport == HidTransportBle) {
|
|
furi_hal_bt_hid_mouse_scroll(delta);
|
|
} else if(instance->transport == HidTransportUsb) {
|
|
furi_hal_hid_mouse_scroll(delta);
|
|
} else {
|
|
furi_crash(NULL);
|
|
}
|
|
}
|
|
|
|
void hid_hal_mouse_press(Hid* instance, uint16_t event) {
|
|
furi_assert(instance);
|
|
if(instance->transport == HidTransportBle) {
|
|
furi_hal_bt_hid_mouse_press(event);
|
|
} else if(instance->transport == HidTransportUsb) {
|
|
furi_hal_hid_mouse_press(event);
|
|
} else {
|
|
furi_crash(NULL);
|
|
}
|
|
}
|
|
|
|
void hid_hal_mouse_release(Hid* instance, uint16_t event) {
|
|
furi_assert(instance);
|
|
if(instance->transport == HidTransportBle) {
|
|
furi_hal_bt_hid_mouse_release(event);
|
|
} else if(instance->transport == HidTransportUsb) {
|
|
furi_hal_hid_mouse_release(event);
|
|
} else {
|
|
furi_crash(NULL);
|
|
}
|
|
}
|
|
|
|
void hid_hal_mouse_release_all(Hid* instance) {
|
|
furi_assert(instance);
|
|
if(instance->transport == HidTransportBle) {
|
|
furi_hal_bt_hid_mouse_release_all();
|
|
} else if(instance->transport == HidTransportUsb) {
|
|
furi_hal_hid_mouse_release(HID_MOUSE_BTN_LEFT);
|
|
furi_hal_hid_mouse_release(HID_MOUSE_BTN_RIGHT);
|
|
} else {
|
|
furi_crash(NULL);
|
|
}
|
|
}
|
|
|
|
int32_t hid_usb_app(void* p) {
|
|
UNUSED(p);
|
|
Hid* app = hid_alloc(HidTransportUsb);
|
|
app = hid_app_alloc_view(app);
|
|
FuriHalUsbInterface* usb_mode_prev = furi_hal_usb_get_config();
|
|
furi_hal_usb_unlock();
|
|
furi_check(furi_hal_usb_set_config(&usb_hid, NULL) == true);
|
|
|
|
bt_hid_connection_status_changed_callback(BtStatusConnected, app);
|
|
|
|
DOLPHIN_DEED(DolphinDeedPluginStart);
|
|
|
|
view_dispatcher_run(app->view_dispatcher);
|
|
|
|
furi_hal_usb_set_config(usb_mode_prev, NULL);
|
|
|
|
hid_free(app);
|
|
|
|
return 0;
|
|
}
|
|
|
|
int32_t hid_ble_app(void* p) {
|
|
UNUSED(p);
|
|
Hid* app = hid_alloc(HidTransportBle);
|
|
app = hid_app_alloc_view(app);
|
|
|
|
bt_disconnect(app->bt);
|
|
|
|
// Wait 2nd core to update nvm storage
|
|
furi_delay_ms(200);
|
|
|
|
bt_keys_storage_set_storage_path(app->bt, HID_BT_KEYS_STORAGE_PATH);
|
|
|
|
if(!bt_set_profile(app->bt, BtProfileHidKeyboard)) {
|
|
FURI_LOG_E(TAG, "Failed to switch to HID profile");
|
|
}
|
|
|
|
furi_hal_bt_start_advertising();
|
|
bt_set_status_changed_callback(app->bt, bt_hid_connection_status_changed_callback, app);
|
|
|
|
DOLPHIN_DEED(DolphinDeedPluginStart);
|
|
|
|
view_dispatcher_run(app->view_dispatcher);
|
|
|
|
bt_set_status_changed_callback(app->bt, NULL, NULL);
|
|
|
|
bt_disconnect(app->bt);
|
|
|
|
// Wait 2nd core to update nvm storage
|
|
furi_delay_ms(200);
|
|
|
|
bt_keys_storage_set_default_path(app->bt);
|
|
|
|
if(!bt_set_profile(app->bt, BtProfileSerial)) {
|
|
FURI_LOG_E(TAG, "Failed to switch to Serial profile");
|
|
}
|
|
|
|
hid_free(app);
|
|
|
|
return 0;
|
|
}
|