mirror of
https://github.com/DarkFlippers/unleashed-firmware
synced 2024-11-30 16:29:12 +00:00
d075e00ae1
* badusb: vid/pid/strings change * demo script update * removed vid/pid values Co-authored-by: あく <alleteam@gmail.com>
122 lines
3.9 KiB
C
122 lines
3.9 KiB
C
#include <furi.h>
|
|
#include <furi_hal.h>
|
|
|
|
#include <gui/view.h>
|
|
#include <gui/view_dispatcher.h>
|
|
#include <gui/modules/submenu.h>
|
|
#include <gui/gui.h>
|
|
|
|
typedef struct {
|
|
Gui* gui;
|
|
ViewDispatcher* view_dispatcher;
|
|
Submenu* submenu;
|
|
FuriHalUsbHidConfig hid_cfg;
|
|
} UsbTestApp;
|
|
|
|
typedef enum {
|
|
UsbTestSubmenuIndexEnable,
|
|
UsbTestSubmenuIndexDisable,
|
|
UsbTestSubmenuIndexRestart,
|
|
UsbTestSubmenuIndexVcpSingle,
|
|
UsbTestSubmenuIndexVcpDual,
|
|
UsbTestSubmenuIndexHid,
|
|
UsbTestSubmenuIndexHidWithParams,
|
|
UsbTestSubmenuIndexHidU2F,
|
|
} SubmenuIndex;
|
|
|
|
void usb_test_submenu_callback(void* context, uint32_t index) {
|
|
furi_assert(context);
|
|
UsbTestApp* app = context;
|
|
if(index == UsbTestSubmenuIndexEnable) {
|
|
furi_hal_usb_enable();
|
|
} else if(index == UsbTestSubmenuIndexDisable) {
|
|
furi_hal_usb_disable();
|
|
} else if(index == UsbTestSubmenuIndexRestart) {
|
|
furi_hal_usb_reinit();
|
|
} else if(index == UsbTestSubmenuIndexVcpSingle) {
|
|
furi_hal_usb_set_config(&usb_cdc_single, NULL);
|
|
} else if(index == UsbTestSubmenuIndexVcpDual) {
|
|
furi_hal_usb_set_config(&usb_cdc_dual, NULL);
|
|
} else if(index == UsbTestSubmenuIndexHid) {
|
|
furi_hal_usb_set_config(&usb_hid, NULL);
|
|
} else if(index == UsbTestSubmenuIndexHidWithParams) {
|
|
app->hid_cfg.vid = 0x1234;
|
|
app->hid_cfg.pid = 0xabcd;
|
|
strncpy(app->hid_cfg.manuf, "WEN", sizeof(app->hid_cfg.manuf));
|
|
strncpy(app->hid_cfg.product, "FLIP", sizeof(app->hid_cfg.product));
|
|
furi_hal_usb_set_config(&usb_hid, &app->hid_cfg);
|
|
} else if(index == UsbTestSubmenuIndexHidU2F) {
|
|
furi_hal_usb_set_config(&usb_hid_u2f, NULL);
|
|
}
|
|
}
|
|
|
|
uint32_t usb_test_exit(void* context) {
|
|
return VIEW_NONE;
|
|
}
|
|
|
|
UsbTestApp* usb_test_app_alloc() {
|
|
UsbTestApp* app = malloc(sizeof(UsbTestApp));
|
|
|
|
// Gui
|
|
app->gui = furi_record_open("gui");
|
|
|
|
// 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);
|
|
|
|
// Views
|
|
app->submenu = submenu_alloc();
|
|
submenu_add_item(
|
|
app->submenu, "Enable", UsbTestSubmenuIndexEnable, usb_test_submenu_callback, app);
|
|
submenu_add_item(
|
|
app->submenu, "Disable", UsbTestSubmenuIndexDisable, usb_test_submenu_callback, app);
|
|
submenu_add_item(
|
|
app->submenu, "Restart", UsbTestSubmenuIndexRestart, usb_test_submenu_callback, app);
|
|
submenu_add_item(
|
|
app->submenu, "Single VCP", UsbTestSubmenuIndexVcpSingle, usb_test_submenu_callback, app);
|
|
submenu_add_item(
|
|
app->submenu, "Dual VCP", UsbTestSubmenuIndexVcpDual, usb_test_submenu_callback, app);
|
|
submenu_add_item(
|
|
app->submenu, "HID KB+Mouse", UsbTestSubmenuIndexHid, usb_test_submenu_callback, app);
|
|
submenu_add_item(
|
|
app->submenu,
|
|
"HID KB+Mouse custom ID",
|
|
UsbTestSubmenuIndexHidWithParams,
|
|
usb_test_submenu_callback,
|
|
app);
|
|
submenu_add_item(
|
|
app->submenu, "HID U2F", UsbTestSubmenuIndexHidU2F, usb_test_submenu_callback, app);
|
|
view_set_previous_callback(submenu_get_view(app->submenu), usb_test_exit);
|
|
view_dispatcher_add_view(app->view_dispatcher, 0, submenu_get_view(app->submenu));
|
|
|
|
// Switch to menu
|
|
view_dispatcher_switch_to_view(app->view_dispatcher, 0);
|
|
|
|
return app;
|
|
}
|
|
|
|
void usb_test_app_free(UsbTestApp* app) {
|
|
furi_assert(app);
|
|
|
|
// Free views
|
|
view_dispatcher_remove_view(app->view_dispatcher, 0);
|
|
submenu_free(app->submenu);
|
|
view_dispatcher_free(app->view_dispatcher);
|
|
|
|
// Close gui record
|
|
furi_record_close("gui");
|
|
app->gui = NULL;
|
|
|
|
// Free rest
|
|
free(app);
|
|
}
|
|
|
|
int32_t usb_test_app(void* p) {
|
|
UsbTestApp* app = usb_test_app_alloc();
|
|
|
|
view_dispatcher_run(app->view_dispatcher);
|
|
|
|
usb_test_app_free(app);
|
|
return 0;
|
|
}
|