74 lines
1.6 KiB
C++
74 lines
1.6 KiB
C++
/* Copyright (c) Kuba Szczodrzyński 2022-06-27. */
|
|
|
|
#include "WiFiPriv.h"
|
|
|
|
static void scanHandler(void *ctx, uint8_t param) {
|
|
LT_HEAP_I();
|
|
WiFiClass *cls = (WiFiClass *)ctx;
|
|
if (!cls) {
|
|
LT_W("Called without ctx");
|
|
return;
|
|
}
|
|
WiFiScanData *scan = cls->scan;
|
|
if (!scan) {
|
|
LT_W("Called without cls->scan");
|
|
return;
|
|
}
|
|
|
|
ScanResult_adv result;
|
|
if (wlan_sta_scan_result(&result)) {
|
|
LT_E("Failed to get scan result");
|
|
goto end;
|
|
}
|
|
LT_D_WG("Found %d APs", result.ApNum);
|
|
|
|
cls->scanAlloc(result.ApNum);
|
|
if (!scan->ap) {
|
|
LT_W("scan->ap alloc failed");
|
|
goto end;
|
|
}
|
|
|
|
for (uint8_t i = 0; i < result.ApNum; i++) {
|
|
scan->ap[i].ssid = strdup(result.ApList[i].ssid);
|
|
scan->ap[i].auth = securityTypeToAuthMode(result.ApList[i].security);
|
|
scan->ap[i].rssi = result.ApList[i].ApPower;
|
|
scan->ap[i].channel = result.ApList[i].channel;
|
|
memcpy(scan->ap[i].bssid.addr, result.ApList[i].bssid, 6);
|
|
}
|
|
|
|
cls->data.scannedAt = millis();
|
|
|
|
end:
|
|
scan->running = false;
|
|
xSemaphoreGive(cls->data.scanSem);
|
|
LT_HEAP_I();
|
|
return;
|
|
}
|
|
|
|
int16_t WiFiClass::scanNetworks(bool async, bool showHidden, bool passive, uint32_t maxMsPerChannel, uint8_t channel) {
|
|
if (scan && scan->running)
|
|
return WIFI_SCAN_RUNNING;
|
|
enableSTA(true);
|
|
scanDelete();
|
|
scanInit();
|
|
|
|
LT_I("Starting WiFi scan");
|
|
|
|
__wrap_bk_printf_disable();
|
|
mhdr_scanu_reg_cb(scanHandler, this);
|
|
bk_wlan_start_scan();
|
|
__wrap_bk_printf_enable();
|
|
|
|
LT_HEAP_I();
|
|
|
|
scan->running = true;
|
|
|
|
if (!async) {
|
|
LT_I("Waiting for results");
|
|
xSemaphoreTake(data.scanSem, 1); // reset the semaphore quickly
|
|
xSemaphoreTake(data.scanSem, pdMS_TO_TICKS(maxMsPerChannel * 20));
|
|
return scan->count;
|
|
}
|
|
return WIFI_SCAN_RUNNING;
|
|
}
|