[FL-3962] BadUSB arbitrary key combinations (#4156)

* badusb: arbitrary modifier key combinations

* badusb: format code

* badusb: add const

* Revert "badusb: add const"

This reverts commit 6ae909fd5d3a5b614712fc92fadda98a6ced2893.

---------

Co-authored-by: hedger <hedger@users.noreply.github.com>

applications/main/bad_usb/helpers/ducky_script.c

@@ -40,11 +40,8 @@ static const uint8_t numpad_keys[10] = {
 };
 
 uint32_t ducky_get_command_len(const char* line) {
-    uint32_t len = strlen(line);
+    char* first_space = strchr(line, ' ');
-    for(uint32_t i = 0; i < len; i++) {
+    return first_space ? (first_space - line) : 0;
-        if(line[i] == ' ') return i;
-    }
-    return 0;
 }
 
 bool ducky_is_line_end(const char chr) {
@@ -180,37 +177,46 @@ static bool ducky_string_next(BadUsbScript* bad_usb) {
 
 static int32_t ducky_parse_line(BadUsbScript* bad_usb, FuriString* line) {
     uint32_t line_len = furi_string_size(line);
-    const char* line_tmp = furi_string_get_cstr(line);
+    const char* line_cstr = furi_string_get_cstr(line);
 
     if(line_len == 0) {
         return SCRIPT_STATE_NEXT_LINE; // Skip empty lines
     }
-    FURI_LOG_D(WORKER_TAG, "line:%s", line_tmp);
+    FURI_LOG_D(WORKER_TAG, "line:%s", line_cstr);
 
     // Ducky Lang Functions
-    int32_t cmd_result = ducky_execute_cmd(bad_usb, line_tmp);
+    int32_t cmd_result = ducky_execute_cmd(bad_usb, line_cstr);
     if(cmd_result != SCRIPT_STATE_CMD_UNKNOWN) {
         return cmd_result;
     }
 
     // Mouse Keys
-    uint16_t key = ducky_get_mouse_keycode_by_name(line_tmp);
+    uint16_t key = ducky_get_mouse_keycode_by_name(line_cstr);
     if(key != HID_MOUSE_INVALID) {
         bad_usb->hid->mouse_press(bad_usb->hid_inst, key);
         bad_usb->hid->mouse_release(bad_usb->hid_inst, key);
         return 0;
     }
 
-    // Special keys + modifiers
+    // Parse chain of modifiers linked by spaces and hyphens
-    key = ducky_get_keycode(bad_usb, line_tmp, false);
+    uint16_t modifiers = 0;
-    if(key == HID_KEYBOARD_NONE) {
+    while(1) {
-        return ducky_error(bad_usb, "No keycode defined for %s", line_tmp);
+        key = ducky_get_next_modifier_keycode_by_name(&line_cstr);
-    }
+        if(key == HID_KEYBOARD_NONE) break;
-    if((key & 0xFF00) != 0) {
+
-        // It's a modifier key
+        modifiers |= key;
-        line_tmp = &line_tmp[ducky_get_command_len(line_tmp) + 1];
+        char next_char = *line_cstr;
-        key |= ducky_get_keycode(bad_usb, line_tmp, true);
+        if(next_char == ' ' || next_char == '-') line_cstr++;
     }
+
+    // Main key
+    char next_char = *line_cstr;
+    uint16_t main_key = ducky_get_keycode_by_name(line_cstr);
+    if(!main_key && next_char) main_key = BADUSB_ASCII_TO_KEY(bad_usb, next_char);
+    key = modifiers | main_key;
+
+    if(key == 0 && next_char) ducky_error(bad_usb, "No keycode defined for %s", line_cstr);
+
     bad_usb->hid->kb_press(bad_usb->hid_inst, key);
     bad_usb->hid->kb_release(bad_usb->hid_inst, key);
     return 0;

applications/main/bad_usb/helpers/ducky_script_i.h

@@ -54,6 +54,8 @@ uint32_t ducky_get_command_len(const char* line);
 
 bool ducky_is_line_end(const char chr);
 
+uint16_t ducky_get_next_modifier_keycode_by_name(const char** param);
+
 uint16_t ducky_get_keycode_by_name(const char* param);
 
 uint16_t ducky_get_media_keycode_by_name(const char* param);

applications/main/bad_usb/helpers/ducky_script_keycodes.c

@@ -6,21 +6,16 @@ typedef struct {
     uint16_t keycode;
 } DuckyKey;
 
-static const DuckyKey ducky_keys[] = {
+static const DuckyKey ducky_modifier_keys[] = {
-    {"CTRL-ALT", KEY_MOD_LEFT_CTRL | KEY_MOD_LEFT_ALT},
-    {"CTRL-SHIFT", KEY_MOD_LEFT_CTRL | KEY_MOD_LEFT_SHIFT},
-    {"ALT-SHIFT", KEY_MOD_LEFT_ALT | KEY_MOD_LEFT_SHIFT},
-    {"ALT-GUI", KEY_MOD_LEFT_ALT | KEY_MOD_LEFT_GUI},
-    {"GUI-SHIFT", KEY_MOD_LEFT_GUI | KEY_MOD_LEFT_SHIFT},
-    {"GUI-CTRL", KEY_MOD_LEFT_GUI | KEY_MOD_LEFT_CTRL},
-
     {"CTRL", KEY_MOD_LEFT_CTRL},
     {"CONTROL", KEY_MOD_LEFT_CTRL},
     {"SHIFT", KEY_MOD_LEFT_SHIFT},
     {"ALT", KEY_MOD_LEFT_ALT},
     {"GUI", KEY_MOD_LEFT_GUI},
     {"WINDOWS", KEY_MOD_LEFT_GUI},
+};
 
+static const DuckyKey ducky_keys[] = {
     {"DOWNARROW", HID_KEYBOARD_DOWN_ARROW},
     {"DOWN", HID_KEYBOARD_DOWN_ARROW},
     {"LEFTARROW", HID_KEYBOARD_LEFT_ARROW},
@@ -119,6 +114,23 @@ static const DuckyKey ducky_mouse_keys[] = {
     {"WHEEL_CLICK", HID_MOUSE_BTN_WHEEL},
 };
 
+uint16_t ducky_get_next_modifier_keycode_by_name(const char** param) {
+    const char* input_str = *param;
+
+    for(size_t i = 0; i < COUNT_OF(ducky_modifier_keys); i++) {
+        size_t key_cmd_len = strlen(ducky_modifier_keys[i].name);
+        if((strncmp(input_str, ducky_modifier_keys[i].name, key_cmd_len) == 0)) {
+            char next_char_after_key = input_str[key_cmd_len];
+            if(ducky_is_line_end(next_char_after_key) || (next_char_after_key == '-')) {
+                *param = &input_str[key_cmd_len];
+                return ducky_modifier_keys[i].keycode;
+            }
+        }
+    }
+
+    return HID_KEYBOARD_NONE;
+}
+
 uint16_t ducky_get_keycode_by_name(const char* param) {
     for(size_t i = 0; i < COUNT_OF(ducky_keys); i++) {
         size_t key_cmd_len = strlen(ducky_keys[i].name);