restructure to use rust source tree primarily

1 files changed, 316 insertions, 0 deletions

diff --git a/src-arduino/ui/menu.cpp b/src-arduino/ui/menu.cpp
new file mode 100644
index 0000000..284d214
--- /dev/null
+++ b/src-arduino/ui/menu.cpp
@@ -0,0 +1,316 @@
+#include "menu.h"
+#include "BleMouse.h"
+#include "bluetooth/ble_mouse.h"
+#include "bluetooth/blescanner.h"
+#include "display.h"
+#include "hid/badusb.h"
+#include "nfc/nfc.h"
+#include "rf/cc1101.h"
+#include "rf/nrf24.h"
+#include "utils/buttons.h"
+#include "utils/device_check.h"
+#include "utils/sysinfo.h"
+#include "wifi/wifi_analyzer.h"
+#include "wifi/wifi_scan.h"
+#include <Arduino.h>
+
+// ================= MENU DATA =================
+extern BleMouse bleMouse;
+
+// Root menu
+const char *mainMenuItems[] = {"BadUSB",      "RF Capture",   "NRF Tools",
+                               "BLE Scan",    "Wifi Scan",    "Wifi Analyzer",
+                               "System Info", "Device Check", "Restart",
+                               "Ble Mouse",   "NFC Tools"};
+
+Menu mainMenu = {mainMenuItems,
+                 sizeof(mainMenuItems) / sizeof(mainMenuItems[0])};
+
+// NRF Tools menu
+const char *nrfToolsItems[] = {
+    "BLE Jammer", "Bluetooth Jammer", "Wifi Jammer", "USB Wireless",
+    "Video TX",   "Zigbee",           "RC"
+
+};
+
+Menu nrfToolsMenu = {nrfToolsItems,
+                     sizeof(nrfToolsItems) / sizeof(nrfToolsItems[0])};
+
+const char *badusbItems[] = {"ORION Demo",
+                             "RickRoll",
+                             "Matrix Rain",
+                             "Fake Terminal",
+                             "Wifi Password Extractor",
+                             "Fake Update",
+                             "Fake BSOD",
+                             "Glitch Screen",
+                             "ASCII Spam",
+                             "Hacker Typer",
+                             "PWN Reverse shell",
+                             "Phishing attack",
+                             "Desktop Ghost",
+                             "System Stresser"};
+
+Menu badusbMenu = {badusbItems, sizeof(badusbItems) / sizeof(badusbItems[0])};
+
+// ================= MENU STATE =================
+
+Menu *currentMenu = &mainMenu;
+
+int menuIndex = 0;
+int menuOffset = 0;
+
+#define MENU_VISIBLE_ROWS 4
+
+bool insideFeature = false;
+
+// ================= DRAW =================
+
+void drawMenu() {
+    u8g2.clearBuffer();
+
+    // scroll handling
+    if (menuIndex < menuOffset)
+        menuOffset = menuIndex;
+
+    if (menuIndex >= menuOffset + MENU_VISIBLE_ROWS)
+        menuOffset = menuIndex - MENU_VISIBLE_ROWS + 1;
+
+    for (int i = 0; i < MENU_VISIBLE_ROWS; i++) {
+        int item = menuOffset + i;
+
+        if (item >= currentMenu->size)
+            break;
+
+        if (item == menuIndex)
+            u8g2.drawStr(0, 14 + i * 14, ">");
+
+        u8g2.drawStr(10, 14 + i * 14, currentMenu->items[item]);
+    }
+
+    // scroll indicators
+    if (menuOffset > 0)
+        u8g2.drawStr(118, 10, "^");
+
+    if (menuOffset + MENU_VISIBLE_ROWS < currentMenu->size)
+        u8g2.drawStr(118, 62, "v");
+
+    u8g2.sendBuffer();
+}
+
+// ================= FEATURE EXECUTION =================
+
+void launchFeature() {
+    insideFeature = true;
+
+    if (currentMenu == &mainMenu) {
+        switch (menuIndex) {
+        case 0: // BadUSB → enter submenu
+            currentMenu = &badusbMenu;
+            menuIndex = 0;
+            menuOffset = 0;
+            break;
+
+        case 1:
+            handleMenu();
+            break;
+        case 2:
+            // startNRFJammer();
+            // startBleJammer();
+            // startBluetoothJammer();
+            currentMenu = &nrfToolsMenu;
+            menuIndex = 0;
+            menuOffset = 0;
+            break;
+
+            break;
+        case 3:
+            ble_scan();
+            ble_drawMenu();
+            while (1) {
+                ble_loop();
+                if (btnBack())
+                    break;
+            }
+            break;
+
+        case 4: {
+            // Start scan once
+            wifi_scan_start();
+            wifi_scan_draw();
+
+            while (1) {
+                wifi_scan_loop();
+                // EXIT condition handled ONLY here
+                if (btnBack()) {
+                    delay(150); // debounce
+                    break;
+                }
+            }
+            break;
+        }
+        case 5: {
+
+            wifi_analyzer_start();
+
+            bool prevBack = false;
+            while (1) {
+                wifi_analyzer_loop();
+                bool nowBack = btnBack();
+                if (nowBack && !prevBack) {
+                    delay(150);
+                    break;
+                }
+                prevBack = nowBack;
+            }
+            break;
+        }
+
+        case 6:
+            runSystemInfoFeature();
+            break;
+        case 7:
+            device_check_run();
+            break;
+        case 8: {
+            // wait for button release
+            delay(200);
+
+            while (btnSelect())
+                delay(10);
+            bool confirm = false;
+
+            while (1) {
+                u8g2.clearBuffer();
+                u8g2.setFont(u8g2_font_6x13_tr);
+
+                u8g2.drawStr(18, 18, "Restart Device?");
+
+                if (confirm) {
+                    u8g2.drawBox(10, 35, 45, 15);
+                    u8g2.setDrawColor(0);
+                    u8g2.drawStr(20, 47, "YES");
+                    u8g2.setDrawColor(1);
+
+                    u8g2.drawStr(75, 47, "NO");
+                } else {
+                    u8g2.drawStr(20, 47, "YES");
+
+                    u8g2.drawBox(65, 35, 45, 15);
+                    u8g2.setDrawColor(0);
+                    u8g2.drawStr(78, 47, "NO");
+                    u8g2.setDrawColor(1);
+                }
+
+                u8g2.sendBuffer();
+
+                if (btnLeft() || btnUp()) {
+                    confirm = true;
+                    delay(150);
+                }
+
+                if (btnRight() || btnDown()) {
+                    confirm = false;
+                    delay(150);
+                }
+
+                if (btnSelect()) {
+                    delay(150);
+
+                    if (confirm) {
+                        u8g2.clearBuffer();
+                        u8g2.drawStr(28, 30, "Restarting...");
+                        u8g2.sendBuffer();
+
+                        delay(1000);
+                        ESP.restart();
+                    } else {
+                        break;
+                    }
+                }
+
+                if (btnBack()) {
+                    delay(150);
+                    break;
+                }
+            }
+        } break;
+
+        case 9:
+            // Begin Ble mouse
+            bleMouse.begin();
+            ble_mouse_run();
+            break;
+        case 10:
+            pn532_scan_loop();
+            break;
+        }
+    } else if (currentMenu == &badusbMenu) {
+        badUSBMenu(menuIndex);
+    } else if (currentMenu == &nrfToolsMenu) {
+        NRFToolsMenu(menuIndex);
+    }
+
+    insideFeature = false;
+
+    drawMenu();
+}
+
+// ================= INIT =================
+
+void menuInit() {
+    currentMenu = &mainMenu;
+    menuIndex = 0;
+    menuOffset = 0;
+
+    drawMenu();
+}
+
+// ================= LOOP =================
+
+void menuLoop() {
+    static uint32_t lastPress = 0;
+
+    if (insideFeature)
+        return;
+
+    if (millis() - lastPress < 150)
+        return;
+
+    if (btnUp()) {
+        menuIndex--;
+
+        if (menuIndex < 0)
+            menuIndex = currentMenu->size - 1;
+
+        drawMenu();
+        lastPress = millis();
+    }
+
+    else if (btnDown()) {
+        menuIndex++;
+
+        if (menuIndex >= currentMenu->size)
+            menuIndex = 0;
+
+        drawMenu();
+        lastPress = millis();
+    }
+
+    else if (btnSelect()) {
+        launchFeature();
+        lastPress = millis();
+    }
+
+    else if (btnBack()) {
+        if (currentMenu != &mainMenu) {
+            currentMenu = &mainMenu;
+            menuIndex = 0;
+            menuOffset = 0;
+
+            drawMenu();
+        }
+
+        lastPress = millis();
+    }
+}