gitweb.ps.run Git - matrix_esp_thesis/blob - esp32/esp_project/main/wifi.c

   1 /*\r
   2  * SPDX-FileCopyrightText: 2010-2022 Espressif Systems (Shanghai) CO LTD\r
   3  *\r
   4  * SPDX-License-Identifier: CC0-1.0\r
   5  */\r
   6 \r
   7 \r
   8 #include <string.h>\r
   9 #include "esp_event.h"\r
  10 #include "esp_log.h"\r
  11 #include "esp_system.h"\r
  12 #include "esp_wifi.h"\r
  13 #include "freertos/FreeRTOS.h"\r
  14 #include "freertos/event_groups.h"\r
  15 #include "freertos/task.h"\r
  16 #include "nvs_flash.h"\r
  17 \r
  18 #include "lwip/err.h"\r
  19 #include "lwip/sys.h"\r
  20 \r
  21 static EventGroupHandle_t s_wifi_event_group;\r
  22 \r
  23 /* The event group allows multiple bits for each event, but we only care about\r
  24  * two events:\r
  25  * - we are connected to the AP with an IP\r
  26  * - we failed to connect after the maximum amount of retries */\r
  27 #define WIFI_CONNECTED_BIT BIT0\r
  28 #define WIFI_FAIL_BIT BIT1\r
  29 \r
  30 static int s_retry_num = 0;\r
  31 \r
  32 static void event_handler(void *arg, esp_event_base_t event_base,\r
  33                           int32_t event_id, void *event_data) {\r
  34   if (event_base == WIFI_EVENT && event_id == WIFI_EVENT_STA_START) {\r
  35     esp_wifi_connect();\r
  36   } else if (event_base == WIFI_EVENT &&\r
  37              event_id == WIFI_EVENT_STA_DISCONNECTED) {\r
  38     if (s_retry_num < 3) {\r
  39       esp_wifi_connect();\r
  40       s_retry_num++;\r
  41       printf("retry to connect to the AP\n");\r
  42     } else {\r
  43       xEventGroupSetBits(s_wifi_event_group, WIFI_FAIL_BIT);\r
  44     }\r
  45     printf("connect to the AP fail\n");\r
  46   } else if (event_base == IP_EVENT && event_id == IP_EVENT_STA_GOT_IP) {\r
  47     ip_event_got_ip_t *event = (ip_event_got_ip_t *) event_data;\r
  48     printf("IP ADDRESS:" IPSTR "\n", IP2STR(&event->ip_info.ip));\r
  49     s_retry_num = 0;\r
  50     xEventGroupSetBits(s_wifi_event_group, WIFI_CONNECTED_BIT);\r
  51   }\r
  52 }\r
  53 \r
  54 void wifi_init(const char *ssid, const char *pass) {\r
  55   esp_err_t ret = nvs_flash_init();\r
  56   if (ret == ESP_ERR_NVS_NO_FREE_PAGES ||\r
  57       ret == ESP_ERR_NVS_NEW_VERSION_FOUND) {\r
  58     ESP_ERROR_CHECK(nvs_flash_erase());\r
  59     ret = nvs_flash_init();\r
  60   }\r
  61   ESP_ERROR_CHECK(ret);\r
  62 \r
  63   s_wifi_event_group = xEventGroupCreate();\r
  64 \r
  65   ESP_ERROR_CHECK(esp_netif_init());\r
  66 \r
  67   ESP_ERROR_CHECK(esp_event_loop_create_default());\r
  68   esp_netif_create_default_wifi_sta();\r
  69 \r
  70   wifi_init_config_t cfg = WIFI_INIT_CONFIG_DEFAULT();\r
  71   ESP_ERROR_CHECK(esp_wifi_init(&cfg));\r
  72 \r
  73   esp_event_handler_instance_t instance_any_id;\r
  74   esp_event_handler_instance_t instance_got_ip;\r
  75   ESP_ERROR_CHECK(esp_event_handler_instance_register(\r
  76       WIFI_EVENT, ESP_EVENT_ANY_ID, &event_handler, NULL, &instance_any_id));\r
  77   ESP_ERROR_CHECK(esp_event_handler_instance_register(\r
  78       IP_EVENT, IP_EVENT_STA_GOT_IP, &event_handler, NULL, &instance_got_ip));\r
  79 \r
  80   wifi_config_t c = {.sta = {.threshold = {.authmode = WIFI_AUTH_WPA2_PSK},\r
  81                              .pmf_cfg = {.capable = true, .required = false}}};\r
  82   snprintf((char *) c.sta.ssid, sizeof(c.sta.ssid), "%s", ssid);\r
  83   snprintf((char *) c.sta.password, sizeof(c.sta.password), "%s", pass);\r
  84   ESP_ERROR_CHECK(esp_wifi_set_mode(WIFI_MODE_STA));\r
  85   ESP_ERROR_CHECK(esp_wifi_set_config(ESP_IF_WIFI_STA, &c));\r
  86   ESP_ERROR_CHECK(esp_wifi_start());\r
  87   printf("wifi_init_sta finished.\n");\r
  88 \r
  89   EventBits_t bits = xEventGroupWaitBits(s_wifi_event_group,\r
  90                                          WIFI_CONNECTED_BIT | WIFI_FAIL_BIT,\r
  91                                          pdFALSE, pdFALSE, portMAX_DELAY);\r
  92 \r
  93   if (bits & WIFI_CONNECTED_BIT) {\r
  94     printf("connected to ap SSID:%s password:%s\n", ssid, pass);\r
  95   } else if (bits & WIFI_FAIL_BIT) {\r
  96     printf("Failed to connect to SSID:%s, password:%s\n", ssid, pass);\r
  97   } else {\r
  98     printf("UNEXPECTED EVENT\n");\r
  99   }\r
 100 \r
 101   /* The event will not be processed after unregister */\r
 102   ESP_ERROR_CHECK(esp_event_handler_instance_unregister(\r
 103       IP_EVENT, IP_EVENT_STA_GOT_IP, instance_got_ip));\r
 104   ESP_ERROR_CHECK(esp_event_handler_instance_unregister(\r
 105       WIFI_EVENT, ESP_EVENT_ANY_ID, instance_any_id));\r
 106   vEventGroupDelete(s_wifi_event_group);\r
 107 }