Add WiFi scan task

nochkin · Jan 11, 2025 · ab0568a · ab0568a
1 parent b1fe71a
commit ab0568a
Show file tree

Hide file tree

Showing 9 changed files with 35 additions and 1 deletion.
diff --git a/include/common_tasks_esp32.h b/include/common_tasks_esp32.h
@@ -1,3 +1,4 @@
 void ws2812_task(void *task);
 void led_task(void *task);
 void info_task(void *task);
+void wifi_task(void *task);
diff --git a/src/common_tasks_esp32.cpp b/src/common_tasks_esp32.cpp
@@ -2,6 +2,7 @@
 
 #include <Arduino.h>
 #include <FastLED.h>
+#include <WiFi.h>
 
 void ws2812_task(void *task) {
     uint8_t led_pin = (uint32_t)task & 0xff;
@@ -69,4 +70,29 @@ void info_task(void *task) {
     vTaskDelete(NULL);
 }
 
+void wifi_task(void *task) {
+    log_d("WiFi task");
+
+    WiFi.mode(WIFI_STA);
+    WiFi.disconnect();
+
+    vTaskDelay(pdMS_TO_TICKS(1000));
+    int16_t count = WiFi.scanNetworks();
+    if (count == 0) {
+        log_v("No WiFI APs found");
+    } else {
+        for (int16_t ii=0; ii<count; ii++) {
+            log_v("%02d: %s (%d dBm) %s [BSSID: %s]",
+                ii+1,
+                WiFi.SSID(ii).c_str(),
+                WiFi.RSSI(ii), 
+                WiFi.encryptionType(ii) == WIFI_AUTH_OPEN ? "Open" : "Protected",
+                WiFi.BSSIDstr(ii).c_str()
+            );
+        }
+    }
+
+    vTaskDelete(NULL);
+}
+
 #endif
diff --git a/src/main_esp32.cpp b/src/main_esp32.cpp
@@ -7,6 +7,7 @@
 
 void setup_custom() {
     xTaskCreate(info_task, "info", CONFIG_ARDUINO_LOOP_STACK_SIZE, NULL, 10, NULL);
+    xTaskCreate(wifi_task, "wifi", CONFIG_ARDUINO_LOOP_STACK_SIZE, NULL, 10, NULL);
 }
 
 void loop_custom() {

diff --git a/src/main_esp32_tdisplay.cpp b/src/main_esp32_tdisplay.cpp
@@ -76,7 +76,7 @@ class LGFX_TDisplay : public lgfx::LGFX_Device
 LGFX_TDisplay tft;
 uint32_t lv_width = LV_HOR_RES;
 uint32_t lv_height = LV_VER_RES;
-uint8_t disp_buf[240 * 50 * sizeof(lv_color_t)];
+uint8_t disp_buf[240 * 15 * sizeof(lv_color_t)];
 
 static void disp_flush(lv_display_t *disp, const lv_area_t *area, unsigned char *color_p) {
     uint32_t w = lv_area_get_width(area);
@@ -137,6 +137,7 @@ void setup_custom() {
     // lv_demo_benchmark();
 
     xTaskCreate(info_task, "info", CONFIG_ARDUINO_LOOP_STACK_SIZE, NULL, 10, NULL);
+    xTaskCreate(wifi_task, "wifi", CONFIG_ARDUINO_LOOP_STACK_SIZE, NULL, 10, NULL);
     xTaskCreate(loop_task, "loop", CONFIG_ARDUINO_LOOP_STACK_SIZE, NULL, 10, NULL);
 }
 

diff --git a/src/main_esp32c3_supermini.cpp b/src/main_esp32c3_supermini.cpp
@@ -7,6 +7,7 @@
 
 void setup_custom() {
     xTaskCreate(info_task, "info", CONFIG_ARDUINO_LOOP_STACK_SIZE, NULL, 10, NULL);
+    xTaskCreate(wifi_task, "wifi", CONFIG_ARDUINO_LOOP_STACK_SIZE, NULL, 10, NULL);
     xTaskCreate(led_task, "led", CONFIG_ARDUINO_LOOP_STACK_SIZE, (void*)MY_LED_PIN, 10, NULL);
 }
 

diff --git a/src/main_esp32c6_supermini.cpp b/src/main_esp32c6_supermini.cpp
@@ -7,6 +7,7 @@
 
 void setup_custom() {
     xTaskCreate(info_task, "info", CONFIG_ARDUINO_LOOP_STACK_SIZE, NULL, 10, NULL);
+    xTaskCreate(wifi_task, "wifi", CONFIG_ARDUINO_LOOP_STACK_SIZE, NULL, 10, NULL);
     xTaskCreate(ws2812_task, "ws2812", CONFIG_ARDUINO_LOOP_STACK_SIZE, (void*)MY_WS2812_PIN, 10, NULL);
 }
 

diff --git a/src/main_esp32s3_n16r8.cpp b/src/main_esp32s3_n16r8.cpp
@@ -9,6 +9,7 @@
 
 void setup_custom() {
     xTaskCreate(info_task, "info", CONFIG_ARDUINO_LOOP_STACK_SIZE, NULL, 10, NULL);
+    xTaskCreate(wifi_task, "wifi", CONFIG_ARDUINO_LOOP_STACK_SIZE, NULL, 10, NULL);
     xTaskCreate(ws2812_task, "ws2812", CONFIG_ARDUINO_LOOP_STACK_SIZE, (void*)MY_WS2812_PIN, 10, NULL);
 }
 

diff --git a/src/main_esp32s3_n8r2.cpp b/src/main_esp32s3_n8r2.cpp
@@ -9,6 +9,7 @@
 
 void setup_custom() {
     xTaskCreate(info_task, "info", CONFIG_ARDUINO_LOOP_STACK_SIZE, NULL, 10, NULL);
+    xTaskCreate(wifi_task, "wifi", CONFIG_ARDUINO_LOOP_STACK_SIZE, NULL, 10, NULL);
     xTaskCreate(ws2812_task, "ws2812", CONFIG_ARDUINO_LOOP_STACK_SIZE, (void*)MY_WS2812_PIN, 10, NULL);
 }
 

diff --git a/src/main_esp32s3_supermini.cpp b/src/main_esp32s3_supermini.cpp
@@ -7,6 +7,7 @@
 
 void setup_custom() {
     xTaskCreate(info_task, "info", CONFIG_ARDUINO_LOOP_STACK_SIZE, NULL, 10, NULL);
+    xTaskCreate(wifi_task, "wifi", CONFIG_ARDUINO_LOOP_STACK_SIZE, NULL, 10, NULL);
     xTaskCreate(ws2812_task, "ws2812", CONFIG_ARDUINO_LOOP_STACK_SIZE, (void*)MY_WS2812_PIN, 10, NULL);
 }