Implement main application logic and UI for Shikigami device
This commit is contained in:
C F committed
1 parent
10e2f921ed
commit
1e9f7ea47f
1 file changed
+770
+770
@@ -0,0 +1,770 @@
|
||||
#include <M5Unified.h>
|
||||
#include <cstdio>
|
||||
#include <esp_sleep.h>
|
||||
#include <esp_wifi.h>
|
||||
#include <esp_netif.h>
|
||||
#include <string.h>
|
||||
#include <SimpleFTPServer.h>
|
||||
#include "wifi_scanner.h"
|
||||
#include "unifont24_vlw.h"
|
||||
|
||||
// ============================================================
|
||||
// Face state enum — drives eye/mouth expressions
|
||||
// ============================================================
|
||||
enum FaceState {
|
||||
FACE_IDLE, // (ʘ_ʘ) default, waiting
|
||||
FACE_SCANNING, // (⌐■_■) active WiFi/BLE sniff
|
||||
FACE_HANDSHAKE, // (★_★) valuable PCAP captured
|
||||
FACE_ERROR, // (>_<) SD failure or system error
|
||||
FACE_HTTP, // (O_O) hosting FTP server + management AP
|
||||
FACE_SLEEP // (x_x) deep-sleep transition
|
||||
};
|
||||
|
||||
// ============================================================
|
||||
// System state
|
||||
// ============================================================
|
||||
bool scanActive = false;
|
||||
bool bleActive = false;
|
||||
bool gpsActive = false;
|
||||
bool autowActive = false;
|
||||
bool ftpActive = false;
|
||||
bool int5Active = false;
|
||||
static const uint8_t INT_INTERVALS[] = {5, 10, 20};
|
||||
uint8_t intIntervalIdx = 0; // index into INT_INTERVALS
|
||||
|
||||
FaceState currentFace = FACE_IDLE;
|
||||
char faceLabel[32] = "IDLE";
|
||||
char faceCaption[32] = "";
|
||||
char discoveredStr[64] = "";
|
||||
uint8_t battPct = 0;
|
||||
|
||||
unsigned long runStart = 0;
|
||||
bool runActive = false;
|
||||
|
||||
// ============================================================
|
||||
// Management AP + FTP server
|
||||
// ============================================================
|
||||
static const char FTP_AP_SSID[] = "Shikigami";
|
||||
static const char FTP_AP_PASS[] = "wardriving"; // WPA2 min-8
|
||||
static const char FTP_USER[] = "admin";
|
||||
static const char FTP_PASS[] = "shiki1337";
|
||||
static const char FTP_AP_IP[] = "192.168.4.1";
|
||||
|
||||
static FtpServer ftpSrv;
|
||||
|
||||
// ============================================================
|
||||
// Auto-wake cycle state — persisted across deep sleep via RTC RAM
|
||||
// ============================================================
|
||||
|
||||
// Maximum BSSIDs tracked across sleep/wake cycles (~2400 bytes RTC RAM)
|
||||
static const int AUTOWAKE_MAX_SEEN = 400;
|
||||
|
||||
RTC_DATA_ATTR static bool rAutoWakeActive = false;
|
||||
RTC_DATA_ATTR static uint8_t rAutoWakeIntervalMins = 5;
|
||||
RTC_DATA_ATTR static int rSeenCount = 0;
|
||||
RTC_DATA_ATTR static uint8_t rSeenBSSIDs[AUTOWAKE_MAX_SEEN][6];
|
||||
|
||||
// Run timer — elapsed ms at time of last sleep, so it survives deep-sleep cycles.
|
||||
RTC_DATA_ATTR static uint32_t rRunElapsedMs = 0;
|
||||
RTC_DATA_ATTR static bool rRunWasActive = false;
|
||||
|
||||
static bool bssidIsSeen(const uint8_t* b) {
|
||||
for (int i = 0; i < rSeenCount; i++) {
|
||||
if (memcmp(rSeenBSSIDs[i], b, 6) == 0) return true;
|
||||
}
|
||||
return false;
|
||||
}
|
||||
|
||||
static void bssidMarkSeen(const uint8_t* b) {
|
||||
if (rSeenCount < AUTOWAKE_MAX_SEEN) {
|
||||
memcpy(rSeenBSSIDs[rSeenCount++], b, 6);
|
||||
}
|
||||
}
|
||||
|
||||
// ============================================================
|
||||
// Layout constants — 540 × 960 display
|
||||
// ============================================================
|
||||
|
||||
// Header bar (full width)
|
||||
static const int HDR_X = 0;
|
||||
static const int HDR_Y = 5;
|
||||
static const int HDR_W = 540;
|
||||
static const int HDR_H = 45;
|
||||
|
||||
// Version string
|
||||
static const int VER_X = 10;
|
||||
static const int VER_Y = 65;
|
||||
|
||||
// Side indicators (GPS left / ACCEL right)
|
||||
static const int IND_Y = 98;
|
||||
|
||||
// Face region bounding box — partial-refresh target
|
||||
static const int FACE_BOX_X = 70;
|
||||
static const int FACE_BOX_Y = 115;
|
||||
static const int FACE_BOX_W = 400;
|
||||
static const int FACE_BOX_H = 285;
|
||||
static const int FACE_CX = 270; // horizontal centre of display
|
||||
static const int FACE_CY = 245; // vertical centre of face bbox
|
||||
|
||||
// Labels above / below face
|
||||
static const int FLAVOR_Y = 123;
|
||||
static const int FLAVOR2_Y = 408;
|
||||
|
||||
// POWER OFF bar (full width)
|
||||
static const int PWROFF_X = 0;
|
||||
static const int PWROFF_Y = 425;
|
||||
static const int PWROFF_W = 540;
|
||||
static const int PWROFF_H = 50;
|
||||
|
||||
// Discovered device text
|
||||
static const int DISC_Y = 502;
|
||||
|
||||
// Button grid
|
||||
static const int BTN_W = 270;
|
||||
static const int BTN_H = 80;
|
||||
static const int COL1_X = 0;
|
||||
static const int COL2_X = 270;
|
||||
static const int ROW1_Y = 640;
|
||||
static const int ROW2_Y = 720;
|
||||
static const int ROW3_Y = 800;
|
||||
static const int ROW4_Y = 880;
|
||||
static const int FULLBTN_W = 540;
|
||||
|
||||
// ============================================================
|
||||
// Forward declarations
|
||||
// ============================================================
|
||||
void drawFullUI();
|
||||
void updateHeader();
|
||||
void drawVersionLine();
|
||||
void drawIndicators();
|
||||
void updateFaceRegion();
|
||||
void drawFaceGraphic();
|
||||
void drawPowerOffBar();
|
||||
void drawDiscoveredText();
|
||||
void updateButtonGrid();
|
||||
void drawButton(int x, int y, int w, int h, const char* label, bool isActive);
|
||||
void handleTouch(int x, int y);
|
||||
|
||||
// ============================================================
|
||||
// drawButton — inverted when active, outlined when inactive
|
||||
// ============================================================
|
||||
void drawButton(int x, int y, int w, int h, const char* label, bool isActive) {
|
||||
if (isActive) {
|
||||
M5.Display.fillRect(x, y, w, h, TFT_BLACK);
|
||||
M5.Display.setTextColor(TFT_WHITE, TFT_BLACK);
|
||||
} else {
|
||||
M5.Display.fillRect(x, y, w, h, TFT_WHITE);
|
||||
// 3-pixel thick border for inactive buttons
|
||||
M5.Display.drawRect(x, y, w, h, TFT_BLACK);
|
||||
M5.Display.drawRect(x+1, y+1, w - 2, h - 2, TFT_BLACK);
|
||||
M5.Display.drawRect(x+2, y+2, w - 4, h - 4, TFT_BLACK);
|
||||
M5.Display.setTextColor(TFT_BLACK, TFT_WHITE);
|
||||
}
|
||||
M5.Display.setTextDatum(middle_center);
|
||||
M5.Display.setTextSize(2);
|
||||
M5.Display.drawString(label, x + w / 2, y + h / 2);
|
||||
}
|
||||
|
||||
// ============================================================
|
||||
// Header bar — STAT / RUN timer / battery
|
||||
// ============================================================
|
||||
void updateHeader() {
|
||||
M5.Display.fillRect(HDR_X, HDR_Y, HDR_W, HDR_H, TFT_WHITE);
|
||||
|
||||
// Battery — right-aligned, nub on LEFT facing the line
|
||||
const int BW = 48, BH = 20, R = 3;
|
||||
const int NUB_W = 5, NUB_H = 10;
|
||||
const int BX = HDR_W - 6 - BW;
|
||||
const int LINE_Y = HDR_Y + HDR_H - BH / 2 - 2; // leave 2px gap below battery
|
||||
const int BY = LINE_Y - BH / 2; // battery centered on the line
|
||||
const int NUB_X = BX - NUB_W;
|
||||
const int NUB_Y = LINE_Y - NUB_H / 2; // centered on the line
|
||||
|
||||
// Nub (solid, left of body)
|
||||
M5.Display.fillRect(NUB_X, NUB_Y, NUB_W, NUB_H, TFT_BLACK);
|
||||
|
||||
// Battery body
|
||||
M5.Display.fillRoundRect(BX, BY, BW, BH, R, TFT_WHITE);
|
||||
M5.Display.drawRoundRect(BX, BY, BW, BH, R, TFT_BLACK);
|
||||
M5.Display.drawRoundRect(BX + 1, BY + 1, BW - 2, BH - 2, R, TFT_BLACK);
|
||||
|
||||
// Fill bar
|
||||
const int fillW = (BW - 6) * battPct / 100;
|
||||
if (fillW > 0)
|
||||
M5.Display.fillRoundRect(BX + 3, BY + 3, fillW, BH - 6, 1, TFT_BLACK);
|
||||
|
||||
// Percent label
|
||||
char batBuf[8];
|
||||
snprintf(batBuf, sizeof(batBuf), "%d%%", battPct);
|
||||
M5.Display.setTextDatum(middle_center);
|
||||
M5.Display.setTextSize(1);
|
||||
M5.Display.setTextColor(fillW > 0 && battPct > 50 ? TFT_WHITE : TFT_BLACK, TFT_WHITE);
|
||||
M5.Display.drawString(batBuf, BX + BW / 2, BY + BH / 2);
|
||||
|
||||
// Bottom line — left edge to 4px before nub
|
||||
M5.Display.drawFastHLine(HDR_X, LINE_Y, NUB_X - 4, TFT_BLACK);
|
||||
M5.Display.drawFastHLine(HDR_X, LINE_Y - 1, NUB_X - 4, TFT_BLACK);
|
||||
M5.Display.drawFastHLine(HDR_X, LINE_Y - 2, NUB_X - 4, TFT_BLACK);
|
||||
|
||||
M5.Display.setTextColor(TFT_BLACK, TFT_WHITE);
|
||||
M5.Display.setTextSize(2);
|
||||
|
||||
// Left — status label reflects current operating state
|
||||
const char* statStr;
|
||||
if (currentFace == FACE_SLEEP) statStr = "STAT: SLEEP";
|
||||
else if (ftpActive) statStr = "STAT: FTP";
|
||||
else if (scanActive) statStr = "STAT: SCAN";
|
||||
else statStr = "STAT: WAIT";
|
||||
M5.Display.setTextDatum(middle_left);
|
||||
M5.Display.drawString(statStr, HDR_X + 8, HDR_Y + HDR_H / 2);
|
||||
|
||||
// Right-aligned — run timer
|
||||
char timeBuf[20];
|
||||
if (runActive) {
|
||||
unsigned long e = (millis() - runStart) / 1000;
|
||||
snprintf(timeBuf, sizeof(timeBuf), "RUN: %02lu:%02lu:%02lu",
|
||||
e / 3600, (e % 3600) / 60, e % 60);
|
||||
} else {
|
||||
snprintf(timeBuf, sizeof(timeBuf), "RUN: 00:00:00");
|
||||
}
|
||||
M5.Display.setTextDatum(middle_right);
|
||||
M5.Display.drawString(timeBuf, NUB_X - 8, HDR_Y + HDR_H / 2);
|
||||
}
|
||||
|
||||
// ============================================================
|
||||
// Version string
|
||||
// ============================================================
|
||||
void drawVersionLine() {
|
||||
M5.Display.fillRect(0, VER_Y - 12, 540, 24, TFT_WHITE);
|
||||
M5.Display.setTextColor(TFT_BLACK, TFT_WHITE);
|
||||
M5.Display.setTextDatum(middle_left);
|
||||
M5.Display.setTextSize(1);
|
||||
M5.Display.drawString("Shikigami v0.1-{git}", VER_X, VER_Y);
|
||||
M5.Display.setTextSize(2);
|
||||
}
|
||||
|
||||
// ============================================================
|
||||
// Side indicators — GPS (left) / ACCEL (right)
|
||||
// ============================================================
|
||||
void drawIndicators() {
|
||||
M5.Display.fillRect(0, IND_Y - 12, 540, 24, TFT_WHITE);
|
||||
M5.Display.setTextColor(TFT_BLACK, TFT_WHITE);
|
||||
M5.Display.setTextSize(1);
|
||||
M5.Display.setTextDatum(middle_left);
|
||||
M5.Display.drawString("GPS", 10, IND_Y);
|
||||
M5.Display.setTextDatum(middle_right);
|
||||
M5.Display.drawString("ACCEL", 530, IND_Y);
|
||||
M5.Display.setTextSize(2);
|
||||
}
|
||||
|
||||
// ============================================================
|
||||
// Face graphic — drawn into FACE_BOX bounding box
|
||||
// Eye geometry shared across all states; only eye/mouth style changes.
|
||||
// ============================================================
|
||||
void drawFaceGraphic() {
|
||||
static const char* const FACE_STR[] = {
|
||||
"(\xCA\x98_\xCA\x98)", // FACE_IDLE (ʘ_ʘ)
|
||||
"(\xE2\x8C\x90\xE2\x96\xA0_\xE2\x96\xA0)", // FACE_SCANNING (⌐■_■)
|
||||
"(\xE2\x98\x85_\xE2\x98\x85)", // FACE_HANDSHAKE (★_★)
|
||||
"(>_<)", // FACE_ERROR
|
||||
"(O_O)", // FACE_HTTP
|
||||
"(x_x)", // FACE_SLEEP
|
||||
};
|
||||
M5.Display.loadFont(unifont24_vlw);
|
||||
M5.Display.setTextSize(4);
|
||||
M5.Display.setTextDatum(middle_center);
|
||||
M5.Display.setTextColor(TFT_BLACK, TFT_WHITE);
|
||||
M5.Display.drawString(FACE_STR[currentFace], FACE_CX, FACE_CY);
|
||||
M5.Display.unloadFont();
|
||||
}
|
||||
|
||||
// Clears the face bounding box and redraws labels + graphic.
|
||||
// When FTP is active, replaces the face with connection info instead.
|
||||
void updateFaceRegion() {
|
||||
M5.Display.fillRect(FACE_BOX_X, FACE_BOX_Y, FACE_BOX_W, FACE_BOX_H, TFT_WHITE);
|
||||
M5.Display.setTextColor(TFT_BLACK, TFT_WHITE);
|
||||
M5.Display.setTextDatum(middle_center);
|
||||
M5.Display.setTextSize(1);
|
||||
M5.Display.drawString(faceLabel, FACE_CX, FLAVOR_Y);
|
||||
|
||||
if (ftpActive) {
|
||||
// Replace face graphic with FTP/AP connection details
|
||||
char line1[48], line2[48], line3[48];
|
||||
snprintf(line1, sizeof(line1), "SSID: %s", FTP_AP_SSID);
|
||||
snprintf(line2, sizeof(line2), "ftp://%s", FTP_AP_IP);
|
||||
snprintf(line3, sizeof(line3), "%s / %s", FTP_USER, FTP_PASS);
|
||||
const int infoSpacing = 45;
|
||||
const int infoTop = FACE_CY - infoSpacing;
|
||||
M5.Display.setTextSize(2);
|
||||
M5.Display.drawString(line1, FACE_CX, infoTop);
|
||||
M5.Display.drawString(line2, FACE_CX, infoTop + infoSpacing);
|
||||
M5.Display.drawString(line3, FACE_CX, infoTop + infoSpacing * 2);
|
||||
} else {
|
||||
drawFaceGraphic();
|
||||
}
|
||||
|
||||
M5.Display.setTextSize(1);
|
||||
M5.Display.setTextColor(TFT_BLACK, TFT_WHITE);
|
||||
M5.Display.setTextDatum(middle_center);
|
||||
M5.Display.drawString(faceCaption, FACE_CX, FLAVOR2_Y);
|
||||
M5.Display.setTextSize(2);
|
||||
}
|
||||
|
||||
// ============================================================
|
||||
// POWER OFF bar — full-width inverted
|
||||
// ============================================================
|
||||
void drawPowerOffBar() {
|
||||
M5.Display.fillRect(PWROFF_X, PWROFF_Y, PWROFF_W, PWROFF_H, TFT_BLACK);
|
||||
M5.Display.setTextColor(TFT_WHITE, TFT_BLACK);
|
||||
M5.Display.setTextDatum(middle_center);
|
||||
M5.Display.setTextSize(2);
|
||||
M5.Display.drawString("POWER OFF", PWROFF_X + PWROFF_W / 2, PWROFF_Y + PWROFF_H / 2);
|
||||
}
|
||||
|
||||
// ============================================================
|
||||
// Discovered device / status text
|
||||
// ============================================================
|
||||
void drawDiscoveredText() {
|
||||
M5.Display.fillRect(0, DISC_Y - 20, 540, 40, TFT_WHITE);
|
||||
M5.Display.setTextColor(TFT_BLACK, TFT_WHITE);
|
||||
M5.Display.setTextDatum(middle_center);
|
||||
M5.Display.setTextSize(1);
|
||||
M5.Display.drawString(discoveredStr, 270, DISC_Y);
|
||||
M5.Display.setTextSize(2);
|
||||
}
|
||||
|
||||
// ============================================================
|
||||
// Button grid — 2×2 split + 2 full-width rows
|
||||
// ============================================================
|
||||
void updateButtonGrid() {
|
||||
drawButton(COL1_X, ROW1_Y, BTN_W, BTN_H, "SCAN", scanActive);
|
||||
drawButton(COL2_X, ROW1_Y, BTN_W, BTN_H, "BLE", bleActive);
|
||||
drawButton(COL1_X, ROW2_Y, BTN_W, BTN_H, "GPS", gpsActive);
|
||||
drawButton(COL2_X, ROW2_Y, BTN_W, BTN_H, "AUTOW", autowActive);
|
||||
drawButton(COL1_X, ROW3_Y, FULLBTN_W, BTN_H, "FTP", ftpActive);
|
||||
char intLabel[16];
|
||||
if (int5Active) {
|
||||
snprintf(intLabel, sizeof(intLabel), "INT %d", INT_INTERVALS[intIntervalIdx]);
|
||||
} else {
|
||||
snprintf(intLabel, sizeof(intLabel), "INT 5");
|
||||
}
|
||||
drawButton(COL1_X, ROW4_Y, FULLBTN_W, BTN_H, intLabel, int5Active);
|
||||
}
|
||||
|
||||
// ============================================================
|
||||
// Full-screen draw — used on boot and scheduled full refresh
|
||||
// ============================================================
|
||||
void drawFullUI() {
|
||||
M5.Display.fillScreen(TFT_WHITE);
|
||||
updateHeader();
|
||||
drawVersionLine();
|
||||
drawIndicators();
|
||||
updateFaceRegion();
|
||||
if (currentFace == FACE_SLEEP) {
|
||||
drawPowerOffBar();
|
||||
} else {
|
||||
M5.Display.fillRect(PWROFF_X, PWROFF_Y, PWROFF_W, PWROFF_H, TFT_WHITE);
|
||||
}
|
||||
drawDiscoveredText();
|
||||
updateButtonGrid();
|
||||
}
|
||||
|
||||
// ============================================================
|
||||
// Deep-sleep entry — draw sleep UI, flush, start timer, sleep
|
||||
// ============================================================
|
||||
static void enterDeepSleep(uint8_t minutes) {
|
||||
wifiScanner_stop();
|
||||
|
||||
// Snapshot run-timer elapsed before millis() disappears in deep sleep
|
||||
rRunWasActive = runActive;
|
||||
rRunElapsedMs = runActive ? static_cast<uint32_t>(millis() - runStart) : rRunElapsedMs;
|
||||
|
||||
// Draw sleep state — switch to quality mode so white-on-black text renders
|
||||
// correctly (epd_fast can't resolve the non-monotone transition in one pass)
|
||||
currentFace = FACE_SLEEP;
|
||||
strncpy(faceLabel, "SLEEPING", sizeof(faceLabel) - 1);
|
||||
snprintf(faceCaption, sizeof(faceCaption), "Wake in %d min", minutes);
|
||||
|
||||
M5.Display.setEpdMode(epd_mode_t::epd_quality);
|
||||
M5.Display.fillScreen(TFT_WHITE);
|
||||
updateHeader();
|
||||
updateFaceRegion();
|
||||
drawPowerOffBar();
|
||||
M5.Display.display(); // blocking full refresh before power-down
|
||||
|
||||
esp_sleep_enable_timer_wakeup((uint64_t)minutes * 60ULL * 1000000ULL);
|
||||
esp_deep_sleep_start();
|
||||
// never returns
|
||||
}
|
||||
|
||||
// ============================================================
|
||||
// Auto-wake scan cycle — runs in setup() on timer wakeup,
|
||||
// never returns (re-enters deep sleep when done).
|
||||
// ============================================================
|
||||
static const unsigned long AUTOWAKE_SCAN_MS = 30000UL;
|
||||
|
||||
static void autoWakeScanCycle() {
|
||||
int rawBatt = M5.Power.getBatteryLevel();
|
||||
battPct = (uint8_t)(rawBatt > 0 ? rawBatt : 0);
|
||||
|
||||
wifiScanner_init();
|
||||
wifiScanner_startTask();
|
||||
wifiScanner_start();
|
||||
|
||||
// Show minimal scanning UI — e-ink retains POWER OFF bar from sleep screen,
|
||||
// only partial-refresh the header and face region
|
||||
currentFace = FACE_SCANNING;
|
||||
strncpy(faceLabel, "AUTO-SCAN", sizeof(faceLabel) - 1);
|
||||
strncpy(faceCaption, "Scanning...", sizeof(faceCaption) - 1);
|
||||
updateHeader();
|
||||
updateFaceRegion();
|
||||
M5.Display.display();
|
||||
|
||||
// Scan for AUTOWAKE_SCAN_MS, draining packet queue to avoid overflow
|
||||
const unsigned long scanEnd = millis() + AUTOWAKE_SCAN_MS;
|
||||
while (millis() < scanEnd) {
|
||||
WifiPacket pkt;
|
||||
while (xQueueReceive(wifiPacketQueue, &pkt, 0) == pdTRUE) {
|
||||
(void)pkt; // TODO: pass to pcapWriter_enqueue()
|
||||
}
|
||||
vTaskDelay(100 / portTICK_PERIOD_MS);
|
||||
}
|
||||
|
||||
wifiScanner_stop();
|
||||
|
||||
// Count new APs not seen in previous cycles
|
||||
const WifiAP* apList = nullptr;
|
||||
int apCount = 0;
|
||||
wifiScanner_getAPList(&apList, &apCount);
|
||||
|
||||
int newCount = 0;
|
||||
if (apList) {
|
||||
for (int i = 0; i < apCount; i++) {
|
||||
if (!bssidIsSeen(apList[i].bssid)) {
|
||||
bssidMarkSeen(apList[i].bssid);
|
||||
newCount++;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// Show results briefly before sleeping again
|
||||
if (newCount > 0) {
|
||||
currentFace = FACE_HANDSHAKE;
|
||||
snprintf(faceCaption, sizeof(faceCaption), "+%d new AP%s", newCount, newCount == 1 ? "" : "s");
|
||||
} else {
|
||||
currentFace = FACE_IDLE;
|
||||
strncpy(faceCaption, "No new APs", sizeof(faceCaption) - 1);
|
||||
}
|
||||
snprintf(faceLabel, sizeof(faceLabel), "Seen: %d total", rSeenCount);
|
||||
// Partial refresh only — POWER OFF bar already on screen from sleep render
|
||||
updateFaceRegion();
|
||||
M5.Display.display();
|
||||
|
||||
vTaskDelay(2500 / portTICK_PERIOD_MS);
|
||||
|
||||
enterDeepSleep(rAutoWakeIntervalMins);
|
||||
// never returns
|
||||
}
|
||||
|
||||
// ============================================================
|
||||
// Management AP helpers
|
||||
// ============================================================
|
||||
|
||||
// AP netif handle — created once, reused on subsequent enable/disable cycles.
|
||||
static esp_netif_t* s_apNetif = nullptr;
|
||||
|
||||
// Start softAP using raw IDF so we stay consistent with the scanner's
|
||||
// esp_wifi_* setup and avoid Arduino WiFi library reinitialising the stack.
|
||||
static void startManagementAP() {
|
||||
if (!s_apNetif) {
|
||||
s_apNetif = esp_netif_create_default_wifi_ap();
|
||||
}
|
||||
|
||||
// Switch to AP-only mode (scanner is stopped before this is called)
|
||||
esp_wifi_set_mode(WIFI_MODE_AP);
|
||||
|
||||
wifi_config_t ap_cfg = {};
|
||||
memcpy(ap_cfg.ap.ssid, FTP_AP_SSID, strlen(FTP_AP_SSID));
|
||||
memcpy(ap_cfg.ap.password, FTP_AP_PASS, strlen(FTP_AP_PASS));
|
||||
ap_cfg.ap.ssid_len = static_cast<uint8_t>(strlen(FTP_AP_SSID));
|
||||
ap_cfg.ap.channel = 6;
|
||||
ap_cfg.ap.max_connection = 2;
|
||||
ap_cfg.ap.authmode = WIFI_AUTH_WPA2_PSK;
|
||||
esp_wifi_set_config(WIFI_IF_AP, &ap_cfg);
|
||||
|
||||
// ftpSrv.begin(FTP_USER, FTP_PASS); // TODO: enable once SD card is mounted
|
||||
}
|
||||
|
||||
static void stopManagementAP() {
|
||||
// Return to STA mode so the scanner can re-enable promiscuous
|
||||
esp_wifi_set_mode(WIFI_MODE_STA);
|
||||
}
|
||||
|
||||
// ============================================================
|
||||
// Run-timer helpers
|
||||
// ============================================================
|
||||
|
||||
// Start the run timer the first time any capture feature is enabled;
|
||||
// stop it when all capture features are disabled.
|
||||
static void updateRunTimer() {
|
||||
bool anyCapture = scanActive || bleActive;
|
||||
if (anyCapture && !runActive) {
|
||||
runActive = true;
|
||||
runStart = millis();
|
||||
} else if (!anyCapture) {
|
||||
runActive = false;
|
||||
}
|
||||
}
|
||||
|
||||
// ============================================================
|
||||
// Touch handling — each button does its own partial redraw
|
||||
// ============================================================
|
||||
void handleTouch(int tx, int ty) {
|
||||
// Row 1 — SCAN / BLE
|
||||
if (ty >= ROW1_Y && ty < ROW1_Y + BTN_H) {
|
||||
if (tx >= COL1_X && tx < COL1_X + BTN_W) {
|
||||
scanActive = !scanActive;
|
||||
if (scanActive) {
|
||||
wifiScanner_start();
|
||||
currentFace = FACE_SCANNING;
|
||||
strncpy(faceLabel, "SCANNING", sizeof(faceLabel) - 1);
|
||||
snprintf(faceCaption, sizeof(faceCaption), "APs: %d", rSeenCount);
|
||||
strncpy(discoveredStr, "Listening...", sizeof(discoveredStr) - 1);
|
||||
} else {
|
||||
wifiScanner_stop();
|
||||
currentFace = FACE_IDLE;
|
||||
strncpy(faceLabel, "IDLE", sizeof(faceLabel) - 1);
|
||||
faceCaption[0] = '\0';
|
||||
discoveredStr[0] = '\0';
|
||||
}
|
||||
updateRunTimer();
|
||||
drawButton(COL1_X, ROW1_Y, BTN_W, BTN_H, "SCAN", scanActive);
|
||||
updateHeader();
|
||||
updateFaceRegion();
|
||||
drawDiscoveredText();
|
||||
} else if (tx >= COL2_X && tx < COL2_X + BTN_W) {
|
||||
bleActive = !bleActive;
|
||||
updateRunTimer();
|
||||
drawButton(COL2_X, ROW1_Y, BTN_W, BTN_H, "BLE", bleActive);
|
||||
updateHeader();
|
||||
}
|
||||
}
|
||||
// Row 2 — GPS / AUTOW
|
||||
else if (ty >= ROW2_Y && ty < ROW2_Y + BTN_H) {
|
||||
if (tx >= COL1_X && tx < COL1_X + BTN_W) {
|
||||
gpsActive = !gpsActive;
|
||||
drawButton(COL1_X, ROW2_Y, BTN_W, BTN_H, "GPS", gpsActive);
|
||||
} else if (tx >= COL2_X && tx < COL2_X + BTN_W) {
|
||||
autowActive = !autowActive;
|
||||
drawButton(COL2_X, ROW2_Y, BTN_W, BTN_H, "AUTOW", autowActive);
|
||||
}
|
||||
}
|
||||
// Row 3 — FTP management AP (full width)
|
||||
// Disables scanner while active; scanner is re-enabled on toggle-off.
|
||||
else if (ty >= ROW3_Y && ty < ROW3_Y + BTN_H) {
|
||||
if (tx >= COL1_X && tx < COL1_X + FULLBTN_W) {
|
||||
ftpActive = !ftpActive;
|
||||
if (ftpActive) {
|
||||
// Stop scanning before switching WiFi mode
|
||||
if (scanActive) {
|
||||
wifiScanner_stop();
|
||||
scanActive = false;
|
||||
updateRunTimer();
|
||||
drawButton(COL1_X, ROW1_Y, BTN_W, BTN_H, "SCAN", false);
|
||||
}
|
||||
startManagementAP();
|
||||
currentFace = FACE_HTTP;
|
||||
strncpy(faceLabel, "FTP SERVER", sizeof(faceLabel) - 1);
|
||||
strncpy(faceCaption, FTP_AP_SSID, sizeof(faceCaption) - 1);
|
||||
} else {
|
||||
stopManagementAP();
|
||||
currentFace = FACE_IDLE;
|
||||
strncpy(faceLabel, "IDLE", sizeof(faceLabel) - 1);
|
||||
faceCaption[0] = '\0';
|
||||
}
|
||||
drawButton(COL1_X, ROW3_Y, FULLBTN_W, BTN_H, "FTP", ftpActive);
|
||||
updateFaceRegion();
|
||||
}
|
||||
}
|
||||
// Row 4 — INT interval: first tap enables auto-wake and sleeps immediately.
|
||||
// While active, subsequent taps cycle the interval and re-sleep.
|
||||
// Final cycle (wraps back to 0) turns auto-wake off.
|
||||
else if (ty >= ROW4_Y && ty < ROW4_Y + BTN_H) {
|
||||
if (tx >= COL1_X && tx < COL1_X + FULLBTN_W) {
|
||||
if (!int5Active) {
|
||||
// Turn on — start at 5 min and enter sleep cycle
|
||||
int5Active = true;
|
||||
intIntervalIdx = 0;
|
||||
rAutoWakeActive = true;
|
||||
rAutoWakeIntervalMins = INT_INTERVALS[intIntervalIdx];
|
||||
char intLabel[16];
|
||||
snprintf(intLabel, sizeof(intLabel), "INT %d", rAutoWakeIntervalMins);
|
||||
drawButton(COL1_X, ROW4_Y, FULLBTN_W, BTN_H, intLabel, true);
|
||||
vTaskDelay(1200 / portTICK_PERIOD_MS);
|
||||
enterDeepSleep(rAutoWakeIntervalMins); // never returns
|
||||
} else {
|
||||
// Cycle interval; wrapping to index 0 turns it off
|
||||
intIntervalIdx = (intIntervalIdx + 1) % 3;
|
||||
if (intIntervalIdx == 0) {
|
||||
int5Active = false;
|
||||
rAutoWakeActive = false;
|
||||
drawButton(COL1_X, ROW4_Y, FULLBTN_W, BTN_H, "INT 5", false);
|
||||
} else {
|
||||
rAutoWakeIntervalMins = INT_INTERVALS[intIntervalIdx];
|
||||
char intLabel[16];
|
||||
snprintf(intLabel, sizeof(intLabel), "INT %d", rAutoWakeIntervalMins);
|
||||
drawButton(COL1_X, ROW4_Y, FULLBTN_W, BTN_H, intLabel, true);
|
||||
vTaskDelay(1200 / portTICK_PERIOD_MS);
|
||||
enterDeepSleep(rAutoWakeIntervalMins); // never returns
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
// POWER OFF bar
|
||||
else if (ty >= PWROFF_Y && ty < PWROFF_Y + PWROFF_H) {
|
||||
// TODO: initiate graceful power-off sequence (flush SD, draw sleep face)
|
||||
}
|
||||
}
|
||||
|
||||
// ============================================================
|
||||
// Setup
|
||||
// ============================================================
|
||||
void setup() {
|
||||
auto cfg = M5.config();
|
||||
M5.begin(cfg);
|
||||
|
||||
// Initial battery reading
|
||||
int rawBatt = M5.Power.getBatteryLevel();
|
||||
battPct = (uint8_t)(rawBatt > 0 ? rawBatt : 0);
|
||||
|
||||
M5.Display.setEpdMode(epd_mode_t::epd_fast);
|
||||
M5.Display.setTextDatum(middle_center);
|
||||
M5.Display.setTextSize(2);
|
||||
|
||||
// ---- Woke from timer: run auto-scan cycle and go back to sleep ----
|
||||
if (esp_sleep_get_wakeup_cause() == ESP_SLEEP_WAKEUP_TIMER && rAutoWakeActive) {
|
||||
autoWakeScanCycle(); // never returns
|
||||
}
|
||||
|
||||
// ---- Normal boot (first power-on or manual reset) ----
|
||||
// Default to INT 5 active: the device is a wardriving appliance —
|
||||
// auto-wake is on from the start.
|
||||
if (esp_sleep_get_wakeup_cause() == ESP_SLEEP_WAKEUP_UNDEFINED) {
|
||||
rAutoWakeActive = true;
|
||||
rAutoWakeIntervalMins = INT_INTERVALS[0]; // 5 min
|
||||
intIntervalIdx = 0;
|
||||
}
|
||||
|
||||
// Restore INT button UI state from RTC vars
|
||||
if (rAutoWakeActive) {
|
||||
int5Active = true;
|
||||
for (uint8_t i = 0; i < 3; i++) {
|
||||
if (INT_INTERVALS[i] == rAutoWakeIntervalMins) { intIntervalIdx = i; break; }
|
||||
}
|
||||
}
|
||||
|
||||
// Restore run timer — offset runStart so elapsed continues from where it paused
|
||||
if (rRunWasActive) {
|
||||
runActive = true;
|
||||
runStart = millis() - rRunElapsedMs;
|
||||
}
|
||||
|
||||
wifiScanner_init();
|
||||
wifiScanner_startTask();
|
||||
|
||||
drawFullUI();
|
||||
}
|
||||
|
||||
// ============================================================
|
||||
// Loop
|
||||
// ============================================================
|
||||
void loop() {
|
||||
M5.update();
|
||||
|
||||
// Drain WiFi packet queue on Core 1 — hand off to PCAP writer (future module).
|
||||
{
|
||||
WifiPacket pkt;
|
||||
while (xQueueReceive(wifiPacketQueue, &pkt, 0) == pdTRUE) {
|
||||
// TODO: pass pkt to pcapWriter_enqueue()
|
||||
(void)pkt;
|
||||
}
|
||||
}
|
||||
|
||||
// Refresh scanner-driven UI fields every 2 s while scanning is active.
|
||||
if (scanActive) {
|
||||
static unsigned long lastScanUIUpdate = 0;
|
||||
const unsigned long now = millis();
|
||||
if (now - lastScanUIUpdate >= 2000UL) {
|
||||
lastScanUIUpdate = now;
|
||||
|
||||
const WifiAP* apList = nullptr;
|
||||
int apCount = 0;
|
||||
wifiScanner_getAPList(&apList, &apCount);
|
||||
|
||||
// Absorb any new BSSIDs into RTC lifetime tally
|
||||
if (apList) {
|
||||
for (int i = 0; i < apCount; i++) {
|
||||
if (!bssidIsSeen(apList[i].bssid)) {
|
||||
bssidMarkSeen(apList[i].bssid);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// faceCaption — lifetime unique AP count across all sessions
|
||||
snprintf(faceCaption, sizeof(faceCaption), "APs: %d", rSeenCount);
|
||||
|
||||
// discoveredStr — most recently seen AP (last entry in the list)
|
||||
if (apList && apCount > 0) {
|
||||
const WifiAP& newest = apList[apCount - 1];
|
||||
const uint8_t* b = newest.bssid;
|
||||
if (newest.ssid[0] != '\0') {
|
||||
snprintf(discoveredStr, sizeof(discoveredStr),
|
||||
"%.32s [%02X:%02X:%02X:%02X:%02X:%02X]",
|
||||
newest.ssid, b[0], b[1], b[2], b[3], b[4], b[5]);
|
||||
} else {
|
||||
snprintf(discoveredStr, sizeof(discoveredStr),
|
||||
"[%02X:%02X:%02X:%02X:%02X:%02X] (hidden)",
|
||||
b[0], b[1], b[2], b[3], b[4], b[5]);
|
||||
}
|
||||
}
|
||||
|
||||
updateFaceRegion();
|
||||
drawDiscoveredText();
|
||||
}
|
||||
}
|
||||
|
||||
auto t = M5.Touch.getDetail();
|
||||
if (t.wasPressed()) {
|
||||
handleTouch(t.x, t.y);
|
||||
}
|
||||
|
||||
// Service FTP server on Core 1 when active.
|
||||
if (ftpActive) {
|
||||
ftpSrv.handleFTP();
|
||||
}
|
||||
|
||||
// Header refresh — tick the run timer every second; poll battery every 30s.
|
||||
{
|
||||
static unsigned long lastHeaderTick = 0;
|
||||
static uint8_t battTickCount = 0;
|
||||
const unsigned long now = millis();
|
||||
if (now - lastHeaderTick >= 1000UL) {
|
||||
lastHeaderTick = now;
|
||||
if (++battTickCount >= 30) {
|
||||
battTickCount = 0;
|
||||
int rawBatt = M5.Power.getBatteryLevel();
|
||||
battPct = (uint8_t)(rawBatt > 0 ? rawBatt : 0);
|
||||
}
|
||||
updateHeader();
|
||||
}
|
||||
}
|
||||
|
||||
// Auto-sleep watchdog — if INT mode is active and we've been awake for the
|
||||
// full interval, go to sleep now (kicks in whether or not the user is scanning).
|
||||
if (int5Active && !ftpActive) {
|
||||
const unsigned long intervalMs = (unsigned long)rAutoWakeIntervalMins * 60UL * 1000UL;
|
||||
if (millis() >= intervalMs) {
|
||||
enterDeepSleep(rAutoWakeIntervalMins); // never returns
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in new issue
Block a user