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

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