ESP32掃描環境中的所有WiFi並且通過串列埠選擇需要連線的WiFi
阿新 • • 發佈:2021-10-17
#include <stdio.h> #include "freertos/FreeRTOS.h" #include "freertos/event_groups.h" #include "esp_wifi.h" #include "esp_log.h" #include "esp_event.h" #include "nvs_flash.h" #include "esp_system.h" #include "driver/uart.h" #include "string.h" #include "driver/gpio.h" //#include "uart0.h" // #include "esp_spi_flash.h" /* Set the SSID and Password via project configuration, or can set directly here */ #define DEFAULT_SSID CONFIG_EXAMPLE_WIFI_SSID #define DEFAULT_PWD CONFIG_EXAMPLE_WIFI_PASSWORD #if CONFIG_EXAMPLE_WIFI_ALL_CHANNEL_SCAN #define DEFAULT_SCAN_METHOD WIFI_ALL_CHANNEL_SCAN #elif CONFIG_EXAMPLE_WIFI_FAST_SCAN #define DEFAULT_SCAN_METHOD WIFI_FAST_SCAN #else #define DEFAULT_SCAN_METHOD WIFI_FAST_SCAN #endif /*CONFIG_EXAMPLE_SCAN_METHOD*/ #if CONFIG_EXAMPLE_WIFI_CONNECT_AP_BY_SIGNAL #define DEFAULT_SORT_METHOD WIFI_CONNECT_AP_BY_SIGNAL #elif CONFIG_EXAMPLE_WIFI_CONNECT_AP_BY_SECURITY #define DEFAULT_SORT_METHOD WIFI_CONNECT_AP_BY_SECURITY #else #define DEFAULT_SORT_METHOD WIFI_CONNECT_AP_BY_SIGNAL #endif /*CONFIG_EXAMPLE_SORT_METHOD*/ #if CONFIG_EXAMPLE_FAST_SCAN_THRESHOLD #define DEFAULT_RSSI CONFIG_EXAMPLE_FAST_SCAN_MINIMUM_SIGNAL #if CONFIG_EXAMPLE_FAST_SCAN_WEAKEST_AUTHMODE_OPEN #define DEFAULT_AUTHMODE WIFI_AUTH_OPEN #elif CONFIG_EXAMPLE_FAST_SCAN_WEAKEST_AUTHMODE_WEP #define DEFAULT_AUTHMODE WIFI_AUTH_WEP #elif CONFIG_EXAMPLE_FAST_SCAN_WEAKEST_AUTHMODE_WPA #define DEFAULT_AUTHMODE WIFI_AUTH_WPA_PSK #elif CONFIG_EXAMPLE_FAST_SCAN_WEAKEST_AUTHMODE_WPA2 #define DEFAULT_AUTHMODE WIFI_AUTH_WPA2_PSK #else #define DEFAULT_AUTHMODE WIFI_AUTH_OPEN #endif #else #define DEFAULT_RSSI -127 #define DEFAULT_AUTHMODE WIFI_AUTH_OPEN #endif /*CONFIG_EXAMPLE_FAST_SCAN_THRESHOLD*/ const int SCAN_DONE_BIT = BIT0; static const char *TAG = "scan"; static void event_handler(void* arg, esp_event_base_t event_base, int32_t event_id, void* event_data) { static uint8_t i=0; if (event_base == WIFI_EVENT && event_id == WIFI_EVENT_STA_START) { //if(i++==2) esp_wifi_connect(); } else if (event_base == WIFI_EVENT && event_id == WIFI_EVENT_STA_DISCONNECTED) { printf("wifi disconnected\n****************\n * \n !!! \n***************\n"); // esp_wifi_connect(); } else if (event_base == WIFI_EVENT && event_id == WIFI_EVENT_SCAN_DONE) { printf("scan done!\n****************\n * \n***************\n"); printf("i value is %d\n",i); if(i++==2) esp_wifi_connect(); } else if (event_base == IP_EVENT && event_id == IP_EVENT_STA_GOT_IP) { ip_event_got_ip_t* event = (ip_event_got_ip_t*) event_data; ESP_LOGI(TAG, "got ip:" IPSTR, IP2STR(&event->ip_info.ip)); } } /* Initialize Wi-Fi as sta and set scan method */ void uart_config(void) { //串列埠配置結構體 uart_config_t uart0_config, uart1_config,uart2_config; //串列埠引數配置->uart0 uart0_config.baud_rate = 115200; //波特率 uart0_config.data_bits = UART_DATA_8_BITS; //資料位 uart0_config.parity = UART_PARITY_DISABLE; //校驗位 uart0_config.stop_bits = UART_STOP_BITS_1; //停止位 uart0_config.flow_ctrl = UART_HW_FLOWCTRL_DISABLE; //硬體流控 uart_param_config(UART_NUM_0, &uart0_config); //設定串列埠 //IO對映-> T:IO1 R:IO3 // uart_set_pin(UART_NUM_0, TXD0_PIN, RXD0_PIN, UART_PIN_NO_CHANGE, UART_PIN_NO_CHANGE); // 註冊串列埠服務即使能+設定快取區大小 uart_driver_install(UART_NUM_0, 1024 * 2, 1024 * 2, 0, NULL, 0); } /****************/ static void fast_scan(void) { uint16_t ap_number=0; //uint16_t number = 20; // 預設掃描列表大小 // wifi_ap_record_t ap_records[20]; // AP資訊結構體大小 uint8_t* get_cmd; uint8_t recv_length=0; uint8_t i=0; uint8_t ssid[32]; wifi_ap_record_t *ap_records; ESP_ERROR_CHECK(esp_netif_init()); // 建立LwIP核心任務並初始化與LwIP相關的工作 ESP_ERROR_CHECK(esp_event_loop_create_default()); wifi_init_config_t cfg = WIFI_INIT_CONFIG_DEFAULT(); ESP_ERROR_CHECK(esp_wifi_init(&cfg)); ESP_ERROR_CHECK(esp_event_handler_instance_register(WIFI_EVENT, ESP_EVENT_ANY_ID, &event_handler, NULL, NULL)); // esp_wifi/include/esp_wifi_types.h ESP_ERROR_CHECK(esp_event_handler_instance_register(IP_EVENT, IP_EVENT_STA_GOT_IP, &event_handler, NULL, NULL)); // esp_netif/include/esp_netif_types.h // Initialize default station as network interface instance (esp-netif) esp_netif_t *sta_netif = esp_netif_create_default_wifi_sta(); // 分配內網ip和掩碼 assert(sta_netif); // Initialize and start WiFi ESP_ERROR_CHECK(esp_wifi_set_mode(WIFI_MODE_STA)); // 設定接入模式為 sta ESP_ERROR_CHECK(esp_wifi_start()); ESP_ERROR_CHECK(esp_wifi_scan_start(NULL,true)); // 如果要掃描到所有的wifi就必須配置為null ESP_ERROR_CHECK(esp_wifi_scan_get_ap_num(&ap_number)); ap_records=(wifi_ap_record_t *)malloc(ap_number*sizeof(wifi_ap_record_t)); ESP_ERROR_CHECK(esp_wifi_scan_get_ap_records(&ap_number,ap_records)); for(;i<ap_number;i++) { printf("ssid=%s bssid=%s primary=%d authmode=%d \n",ap_records[i].ssid,ap_records[i].bssid,ap_records[i].primary,ap_records[i].authmode); } get_cmd=(uint8_t*)malloc(20*sizeof(uint8_t)); printf("which ssid choose\n"); while(recv_length==0) { recv_length = uart_read_bytes(UART_NUM_0, get_cmd,20, 1000 / portTICK_RATE_MS); } i=get_cmd[0]-48; // 字元轉數字 printf("receive value is %d,strlen is %d\n",get_cmd[i],strlen((const char *)get_cmd)); printf("choose ssid is %s\n",ap_records[i].ssid); recv_length = 0; wifi_config_t wifi_config = { .sta = { // .ssid = ssid, // .password = DEFAULT_PWD, .scan_method = DEFAULT_SCAN_METHOD, .sort_method = DEFAULT_SORT_METHOD, .threshold.rssi = DEFAULT_RSSI, .threshold.authmode = DEFAULT_AUTHMODE, //ap_records[i].authmode, }, }; memcpy(wifi_config.sta.ssid,ap_records[i].ssid,strlen((const char *)ap_records[i].ssid)); printf("wifi_config.sta.ssid %s\n",wifi_config.sta.ssid); printf("\nprint password\n"); while(recv_length==0) { recv_length = uart_read_bytes(UART_NUM_0, get_cmd, 20, 1000 / portTICK_RATE_MS); } printf("receive string is %s,strlen is %d\n",get_cmd,strlen((const char *)get_cmd)); memcpy(wifi_config.sta.password,get_cmd,strlen((const char *)get_cmd)); printf("wifi_config.sta.password %s\n",wifi_config.sta.password); recv_length = 0; ESP_ERROR_CHECK(esp_wifi_set_config(WIFI_IF_STA, &wifi_config)); ESP_ERROR_CHECK(esp_wifi_start()); ESP_ERROR_CHECK(esp_wifi_connect()); printf("free memary\n"); free(ap_records); free(get_cmd); } void app_main(void) { // Initialize NVS esp_err_t ret = nvs_flash_init(); if (ret == ESP_ERR_NVS_NO_FREE_PAGES || ret == ESP_ERR_NVS_NEW_VERSION_FOUND) { ESP_ERROR_CHECK(nvs_flash_erase()); ret = nvs_flash_init(); } ESP_ERROR_CHECK( ret ); uart_config(); fast_scan(); }