mirror of
https://github.com/DarkFlippers/unleashed-firmware.git
synced 2024-12-16 09:12:39 +03:00
e6642b332c
* [FL-1930] USB HID keyboard test * [FL-1930] HID mouse demo app * [FL-1930] BadUSB: RubberDucky script parser. BadUSB test app Co-authored-by: Aleksandr Kutuzov <alleteam@gmail.com>
104 lines
3.1 KiB
C
104 lines
3.1 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>
|
|
#include <cmsis_os.h>
|
|
|
|
typedef struct {
|
|
Gui* gui;
|
|
ViewDispatcher* view_dispatcher;
|
|
Submenu* submenu;
|
|
} UsbTestApp;
|
|
|
|
typedef enum {
|
|
UsbTestSubmenuIndexEnable,
|
|
UsbTestSubmenuIndexDisable,
|
|
UsbTestSubmenuIndexVcpSingle,
|
|
UsbTestSubmenuIndexVcpDual,
|
|
UsbTestSubmenuIndexHid,
|
|
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 == UsbTestSubmenuIndexVcpSingle) {
|
|
furi_hal_usb_set_config(UsbModeVcpSingle);
|
|
} else if(index == UsbTestSubmenuIndexVcpDual) {
|
|
furi_hal_usb_set_config(UsbModeVcpDual);
|
|
} else if(index == UsbTestSubmenuIndexHid) {
|
|
furi_hal_usb_set_config(UsbModeHid);
|
|
} else if(index == UsbTestSubmenuIndexHidU2F) {
|
|
//furi_hal_usb_set_config(UsbModeU2F);
|
|
}
|
|
}
|
|
|
|
uint32_t usb_test_exit(void* context) {
|
|
return VIEW_NONE;
|
|
}
|
|
|
|
UsbTestApp* usb_test_app_alloc() {
|
|
UsbTestApp* app = furi_alloc(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, "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, "TODO: 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;
|
|
}
|