173 lines
		
	
	
		
			No EOL
		
	
	
		
			6.9 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			173 lines
		
	
	
		
			No EOL
		
	
	
		
			6.9 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
| /* WiFi station Example
 | |
| 
 | |
|    This example code is in the Public Domain (or CC0 licensed, at your option.)
 | |
| 
 | |
|    Unless required by applicable law or agreed to in writing, this
 | |
|    software is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR
 | |
|    CONDITIONS OF ANY KIND, either express or implied.
 | |
| */
 | |
| #include <string.h>
 | |
| #include "freertos/FreeRTOS.h"
 | |
| #include "freertos/task.h"
 | |
| #include "freertos/event_groups.h"
 | |
| #include "esp_system.h"
 | |
| #include "esp_wifi.h"
 | |
| #include "esp_event.h"
 | |
| #include "esp_log.h"
 | |
| #include "nvs_flash.h"
 | |
| 
 | |
| #include "lwip/err.h"
 | |
| #include "lwip/sys.h"
 | |
| 
 | |
| #include "wifi_helper.h"
 | |
| 
 | |
| /* The examples use WiFi configuration that you can set via project configuration menu
 | |
| 
 | |
|    If you'd rather not, just change the below entries to strings with
 | |
|    the config you want - ie #define EXAMPLE_WIFI_SSID "mywifissid"
 | |
| */
 | |
| //#define EXAMPLE_ESP_MAXIMUM_RETRY  10
 | |
| 
 | |
| //#if CONFIG_ESP_STATION_EXAMPLE_WPA3_SAE_PWE_HUNT_AND_PECK
 | |
| //#define ESP_WIFI_SAE_MODE WPA3_SAE_PWE_HUNT_AND_PECK
 | |
| //#define EXAMPLE_H2E_IDENTIFIER ""
 | |
| //#elif CONFIG_ESP_STATION_EXAMPLE_WPA3_SAE_PWE_HASH_TO_ELEMENT
 | |
| //#define ESP_WIFI_SAE_MODE WPA3_SAE_PWE_HASH_TO_ELEMENT
 | |
| //#define EXAMPLE_H2E_IDENTIFIER CONFIG_ESP_WIFI_PW_ID
 | |
| //#elif CONFIG_ESP_STATION_EXAMPLE_WPA3_SAE_PWE_BOTH
 | |
| //#define ESP_WIFI_SAE_MODE WPA3_SAE_PWE_BOTH
 | |
| //#define EXAMPLE_H2E_IDENTIFIER CONFIG_ESP_WIFI_PW_ID
 | |
| //#endif
 | |
| //#if CONFIG_ESP_WIFI_AUTH_OPEN
 | |
| //#define ESP_WIFI_SCAN_AUTH_MODE_THRESHOLD WIFI_AUTH_OPEN
 | |
| //#elif CONFIG_ESP_WIFI_AUTH_WEP
 | |
| //#define ESP_WIFI_SCAN_AUTH_MODE_THRESHOLD WIFI_AUTH_WEP
 | |
| //#elif CONFIG_ESP_WIFI_AUTH_WPA_PSK
 | |
| //#define ESP_WIFI_SCAN_AUTH_MODE_THRESHOLD WIFI_AUTH_WPA_PSK
 | |
| //#elif CONFIG_ESP_WIFI_AUTH_WPA2_PSK
 | |
| #define ESP_WIFI_SCAN_AUTH_MODE_THRESHOLD WIFI_AUTH_WPA2_PSK
 | |
| //#elif CONFIG_ESP_WIFI_AUTH_WPA_WPA2_PSK
 | |
| //#define ESP_WIFI_SCAN_AUTH_MODE_THRESHOLD WIFI_AUTH_WPA_WPA2_PSK
 | |
| //#elif CONFIG_ESP_WIFI_AUTH_WPA3_PSK
 | |
| //#define ESP_WIFI_SCAN_AUTH_MODE_THRESHOLD WIFI_AUTH_WPA3_PSK
 | |
| //#elif CONFIG_ESP_WIFI_AUTH_WPA2_WPA3_PSK
 | |
| //#define ESP_WIFI_SCAN_AUTH_MODE_THRESHOLD WIFI_AUTH_WPA2_WPA3_PSK
 | |
| //#elif CONFIG_ESP_WIFI_AUTH_WAPI_PSK
 | |
| //#define ESP_WIFI_SCAN_AUTH_MODE_THRESHOLD WIFI_AUTH_WAPI_PSK
 | |
| //#endif
 | |
| 
 | |
| /* FreeRTOS event group to signal when we are connected*/
 | |
| static EventGroupHandle_t s_wifi_event_group;
 | |
| 
 | |
| /* The event group allows multiple bits for each event, but we only care about two events:
 | |
|  * - we are connected to the AP with an IP
 | |
|  * - we failed to connect after the maximum amount of retries */
 | |
| #define WIFI_CONNECTED_BIT BIT0
 | |
| #define WIFI_FAIL_BIT      BIT1
 | |
| 
 | |
| static const char *TAG = "wifi station";
 | |
| 
 | |
| //static int s_retry_num = 0;
 | |
| 
 | |
| 
 | |
| static void event_handler(void* arg, esp_event_base_t event_base,
 | |
|                                 int32_t event_id, void* event_data)
 | |
| {
 | |
|     if (event_base == WIFI_EVENT && event_id == WIFI_EVENT_STA_START) {
 | |
|         esp_wifi_connect();
 | |
|     } else if (event_base == WIFI_EVENT && event_id == WIFI_EVENT_STA_DISCONNECTED) {
 | |
|         esp_wifi_connect();
 | |
|         ESP_LOGI(TAG, "retry to connect to the AP");
 | |
|         //if (s_retry_num < EXAMPLE_ESP_MAXIMUM_RETRY) {
 | |
|         //    esp_wifi_connect();
 | |
|         //    s_retry_num++;
 | |
|         //    ESP_LOGI(TAG, "retry to connect to the AP");
 | |
|         //} else {
 | |
|         //    xEventGroupSetBits(s_wifi_event_group, WIFI_FAIL_BIT);
 | |
|         //}
 | |
|         //ESP_LOGI(TAG,"connect to the AP fail");
 | |
|     } 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));
 | |
|         //s_retry_num = 0;
 | |
|         xEventGroupSetBits(s_wifi_event_group, WIFI_CONNECTED_BIT);
 | |
|     }
 | |
| }
 | |
| 
 | |
| void wifi_init_sta(void)
 | |
| {
 | |
|     s_wifi_event_group = xEventGroupCreate();
 | |
| 
 | |
|     ESP_ERROR_CHECK(esp_netif_init());
 | |
| 
 | |
|     ESP_ERROR_CHECK(esp_event_loop_create_default());
 | |
|     esp_netif_create_default_wifi_sta();
 | |
| 
 | |
|     wifi_init_config_t cfg = WIFI_INIT_CONFIG_DEFAULT();
 | |
|     ESP_ERROR_CHECK(esp_wifi_init(&cfg));
 | |
| 
 | |
|     esp_event_handler_instance_t instance_any_id;
 | |
|     esp_event_handler_instance_t instance_got_ip;
 | |
|     ESP_ERROR_CHECK(esp_event_handler_instance_register(WIFI_EVENT,
 | |
|                                                         ESP_EVENT_ANY_ID,
 | |
|                                                         &event_handler,
 | |
|                                                         NULL,
 | |
|                                                         &instance_any_id));
 | |
|     ESP_ERROR_CHECK(esp_event_handler_instance_register(IP_EVENT,
 | |
|                                                         IP_EVENT_STA_GOT_IP,
 | |
|                                                         &event_handler,
 | |
|                                                         NULL,
 | |
|                                                         &instance_got_ip));
 | |
| 
 | |
|     wifi_config_t wifi_config = {
 | |
|         .sta = {
 | |
|             .ssid = CONFIG_ESP_WIFI_SSID,
 | |
|             .password = CONFIG_ESP_WIFI_PASS,
 | |
|             /* Authmode threshold resets to WPA2 as default if password matches WPA2 standards (password len => 8).
 | |
|              * If you want to connect the device to deprecated WEP/WPA networks, Please set the threshold value
 | |
|              * to WIFI_AUTH_WEP/WIFI_AUTH_WPA_PSK and set the password with length and format matching to
 | |
|              * WIFI_AUTH_WEP/WIFI_AUTH_WPA_PSK standards.
 | |
|              */
 | |
|             .threshold.authmode = ESP_WIFI_SCAN_AUTH_MODE_THRESHOLD,
 | |
|             //.sae_pwe_h2e = ESP_WIFI_SAE_MODE,
 | |
|             //.sae_h2e_identifier = EXAMPLE_H2E_IDENTIFIER,
 | |
|         },
 | |
|     };
 | |
|     ESP_ERROR_CHECK(esp_wifi_set_mode(WIFI_MODE_STA) );
 | |
|     ESP_ERROR_CHECK(esp_wifi_set_config(WIFI_IF_STA, &wifi_config) );
 | |
|     ESP_ERROR_CHECK(esp_wifi_start() );
 | |
| 
 | |
|     ESP_LOGI(TAG, "wifi_init_sta finished.");
 | |
| 
 | |
|     /* Waiting until either the connection is established (WIFI_CONNECTED_BIT) or connection failed for the maximum
 | |
|      * number of re-tries (WIFI_FAIL_BIT). The bits are set by event_handler() (see above) */
 | |
|     EventBits_t bits = xEventGroupWaitBits(s_wifi_event_group,
 | |
|             WIFI_CONNECTED_BIT | WIFI_FAIL_BIT,
 | |
|             pdFALSE,
 | |
|             pdFALSE,
 | |
|             portMAX_DELAY);
 | |
| 
 | |
|     /* xEventGroupWaitBits() returns the bits before the call returned, hence we can test which event actually
 | |
|      * happened. */
 | |
|     if (bits & WIFI_CONNECTED_BIT) {
 | |
|         ESP_LOGI(TAG, "connected to ap SSID:%s password:%s",
 | |
|                  CONFIG_ESP_WIFI_SSID, CONFIG_ESP_WIFI_PASS);
 | |
|     } else if (bits & WIFI_FAIL_BIT) {
 | |
|         ESP_LOGI(TAG, "Failed to connect to SSID:%s, password:%s",
 | |
|                  CONFIG_ESP_WIFI_SSID, CONFIG_ESP_WIFI_PASS);
 | |
|     } else {
 | |
|         ESP_LOGE(TAG, "UNEXPECTED EVENT");
 | |
|     }
 | |
| }
 | |
| 
 | |
| void connect_wifi(void)
 | |
| {
 | |
|     if (CONFIG_LOG_MAXIMUM_LEVEL > CONFIG_LOG_DEFAULT_LEVEL) {
 | |
|         /* If you only want to open more logs in the wifi module, you need to make the max level greater than the default level,
 | |
|          * and call esp_log_level_set() before esp_wifi_init() to improve the log level of the wifi module. */
 | |
|         esp_log_level_set("wifi", CONFIG_LOG_MAXIMUM_LEVEL);
 | |
|     }
 | |
| 
 | |
|     ESP_LOGI(TAG, "ESP_WIFI_MODE_STA");
 | |
|     wifi_init_sta();
 | |
| } |