/* Tests for the Wi-Fi */ #include "string.h" #include "esp_system.h" #include "unity.h" #include "esp_system.h" #include "esp_event.h" #include "esp_wifi.h" #include "esp_wifi_types.h" #include "esp_log.h" #include "nvs_flash.h" #include "test_utils.h" #include "freertos/task.h" #include "freertos/event_groups.h" static const char* TAG = "test_wifi"; #define DEFAULT_SSID "TEST_SSID" #define DEFAULT_PWD "TEST_PASS" #define GOT_IP_EVENT 0x00000001 #define DISCONNECT_EVENT 0x00000002 #define EVENT_HANDLER_FLAG_DO_NOT_AUTO_RECONNECT 0x00000001 static uint32_t wifi_event_handler_flag; static EventGroupHandle_t wifi_events; static void wifi_event_handler(void* arg, esp_event_base_t event_base, int32_t event_id, void* event_data) { printf("wifi ev_handle_called.\n"); switch(event_id) { case WIFI_EVENT_STA_START: ESP_LOGI(TAG, "WIFI_EVENT_STA_START"); //do not actually connect in test case //; break; case WIFI_EVENT_STA_DISCONNECTED: ESP_LOGI(TAG, "WIFI_EVENT_STA_DISCONNECTED"); if (! (EVENT_HANDLER_FLAG_DO_NOT_AUTO_RECONNECT & wifi_event_handler_flag) ) { TEST_ESP_OK(esp_wifi_connect()); } if (wifi_events) { xEventGroupSetBits(wifi_events, DISCONNECT_EVENT); } break; default: break; } return; } static void ip_event_handler(void* arg, esp_event_base_t event_base, int32_t event_id, void* event_data) { ip_event_got_ip_t *event; printf("ip ev_handle_called.\n"); switch(event_id) { case IP_EVENT_STA_GOT_IP: event = (ip_event_got_ip_t*)event_data; ESP_LOGI(TAG, "IP_EVENT_STA_GOT_IP"); ESP_LOGI(TAG, "got ip:" IPSTR "\n", IP2STR(&event->ip_info.ip)); if (wifi_events) { xEventGroupSetBits(wifi_events, GOT_IP_EVENT); } break; default: break; } return; } static esp_err_t event_init(void) { ESP_ERROR_CHECK(esp_event_loop_create_default()); ESP_ERROR_CHECK(esp_event_handler_register(WIFI_EVENT, ESP_EVENT_ANY_ID, &wifi_event_handler, NULL)); ESP_ERROR_CHECK(esp_event_handler_register(IP_EVENT, ESP_EVENT_ANY_ID, &ip_event_handler, NULL)); esp_netif_create_default_wifi_sta(); esp_netif_create_default_wifi_ap(); return ESP_OK; } #define EMPH_STR(s) "****** "s" ******" static void test_wifi_init_deinit(wifi_init_config_t *cfg, wifi_config_t* wifi_config) { ESP_LOGI(TAG, EMPH_STR("esp_wifi_deinit")); TEST_ESP_ERR(ESP_ERR_WIFI_NOT_INIT, esp_wifi_deinit()); ESP_LOGI(TAG, EMPH_STR("esp_wifi_get_mode")); wifi_mode_t mode_get; TEST_ESP_ERR(ESP_ERR_WIFI_NOT_INIT, esp_wifi_get_mode(&mode_get)); ESP_LOGI(TAG, EMPH_STR("esp_wifi_init")); TEST_ESP_OK(esp_wifi_init(cfg)); ESP_LOGI(TAG, EMPH_STR("esp_wifi_set_mode")); TEST_ESP_OK(esp_wifi_set_mode(WIFI_MODE_STA)); ESP_LOGI(TAG, EMPH_STR("esp_wifi_set_config")); TEST_ESP_OK(esp_wifi_set_config(WIFI_IF_STA, wifi_config)); ESP_LOGI(TAG, EMPH_STR("esp_wifi_deinit...")); TEST_ESP_OK(esp_wifi_deinit()); } static void test_wifi_start_stop(wifi_init_config_t *cfg, wifi_config_t* wifi_config) { ESP_LOGI(TAG, EMPH_STR("esp_wifi_stop")); TEST_ESP_ERR(ESP_ERR_WIFI_NOT_INIT, esp_wifi_stop()); ESP_LOGI(TAG, EMPH_STR("esp_wifi_init")); TEST_ESP_OK(esp_wifi_init(cfg)); ESP_LOGI(TAG, EMPH_STR("esp_wifi_set_mode")); TEST_ESP_OK(esp_wifi_set_mode(WIFI_MODE_STA)); ESP_LOGI(TAG, EMPH_STR("esp_wifi_set_config")); TEST_ESP_OK(esp_wifi_set_config(WIFI_IF_STA, wifi_config)); //now start wifi ESP_LOGI(TAG, EMPH_STR("esp_wifi_start...")); TEST_ESP_OK(esp_wifi_start()); //wifi stop ESP_LOGI(TAG, EMPH_STR("esp_wifi_stop...")); TEST_ESP_OK( esp_wifi_stop() ); ESP_LOGI(TAG, EMPH_STR("esp_wifi_deinit...")); TEST_ESP_OK(esp_wifi_deinit()); } TEST_CASE("wifi stop and deinit","[wifi]") { test_case_uses_tcpip(); wifi_init_config_t cfg = WIFI_INIT_CONFIG_DEFAULT(); wifi_config_t wifi_config = { .sta = { .ssid = DEFAULT_SSID, .password = DEFAULT_PWD }, }; //init nvs ESP_LOGI(TAG, EMPH_STR("nvs_flash_init")); esp_err_t r = nvs_flash_init(); if (r == ESP_ERR_NVS_NO_FREE_PAGES || r == ESP_ERR_NVS_NEW_VERSION_FOUND) { ESP_LOGI(TAG, EMPH_STR("no free pages or nvs version mismatch, erase..")); TEST_ESP_OK(nvs_flash_erase()); r = nvs_flash_init(); } TEST_ESP_OK(r); //init tcpip ESP_LOGI(TAG, EMPH_STR("esp_netif_init")); esp_netif_init(); //init event loop ESP_LOGI(TAG, EMPH_STR("event_init")); event_init(); ESP_LOGI(TAG, "test wifi init & deinit..."); test_wifi_init_deinit(&cfg, &wifi_config); ESP_LOGI(TAG, "wifi init & deinit seem to be OK."); ESP_LOGI(TAG, "test wifi start & stop..."); test_wifi_start_stop(&cfg, &wifi_config); ESP_LOGI(TAG, "wifi start & stop seem to be OK."); ESP_LOGI(TAG, EMPH_STR("nvs_flash_deinit...")); nvs_flash_deinit(); ESP_LOGI(TAG, "test passed..."); TEST_IGNORE_MESSAGE("this test case is ignored due to the critical memory leak of esp_netif and event_loop."); } #if !TEMPORARY_DISABLED_FOR_TARGETS(ESP32S2BETA) static void start_wifi_as_softap(void) { wifi_init_config_t cfg = WIFI_INIT_CONFIG_DEFAULT(); cfg.nvs_enable = false; wifi_config_t w_config = { .ap.ssid = DEFAULT_SSID, .ap.password = DEFAULT_PWD, .ap.ssid_len = 0, .ap.channel = 1, .ap.authmode = WIFI_AUTH_WPA2_PSK, .ap.ssid_hidden = false, .ap.max_connection = 4, .ap.beacon_interval = 100, }; event_init(); // can't deinit event loop, need to reset leak check unity_reset_leak_checks(); if (wifi_events == NULL) { wifi_events = xEventGroupCreate(); } TEST_ESP_OK(esp_wifi_init(&cfg)); TEST_ESP_OK(esp_wifi_set_mode(WIFI_MODE_AP)); TEST_ESP_OK(esp_wifi_set_config(WIFI_IF_AP, &w_config)); TEST_ESP_OK(esp_wifi_start()); } static void start_wifi_as_sta(void) { wifi_init_config_t cfg = WIFI_INIT_CONFIG_DEFAULT(); cfg.nvs_enable = false; // do not auto connect wifi_event_handler_flag |= EVENT_HANDLER_FLAG_DO_NOT_AUTO_RECONNECT; event_init(); // can't deinit event loop, need to reset leak check unity_reset_leak_checks(); if (wifi_events == NULL) { wifi_events = xEventGroupCreate(); } else { xEventGroupClearBits(wifi_events, 0x00ffffff); } TEST_ESP_OK(esp_wifi_init(&cfg)); TEST_ESP_OK(esp_wifi_set_mode(WIFI_MODE_STA)); TEST_ESP_OK(esp_wifi_start()); } static void stop_wifi(void) { printf("stop wifi\n"); TEST_ESP_OK(esp_wifi_stop()); TEST_ESP_OK(esp_wifi_deinit()); if (wifi_events) { vEventGroupDelete(wifi_events); wifi_events = NULL; } vTaskDelay(1000/portTICK_PERIOD_MS); } static void receive_ds2ds_packet(void) { test_case_uses_tcpip(); start_wifi_as_softap(); unity_wait_for_signal("sender ready"); unity_send_signal("receiver ready"); // wait for sender to send packets vTaskDelay(1000/portTICK_PERIOD_MS); stop_wifi(); } static const char ds2ds_pdu[] = { 0x48, 0x03, 0x00, 0x00, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xE8, 0x65, 0xD4, 0xCB, 0x74, 0x19, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0x60, 0x94, 0xE8, 0x65, 0xD4, 0xCB, 0x74, 0x1C, 0x26, 0xB9, 0x0D, 0x02, 0x7D, 0x13, 0x00, 0x00, 0x01, 0xE8, 0x65, 0xD4, 0xCB, 0x74, 0x1C, 0x00, 0x00, 0x26, 0xB9, 0x00, 0x00, 0x00, 0x00 }; static void send_ds2ds_packet(void) { test_case_uses_tcpip(); start_wifi_as_softap(); unity_send_signal("sender ready"); unity_wait_for_signal("receiver ready"); // send packet 20 times to make sure receiver will get this packet for (uint16_t i = 0; i < 20; i++) { esp_wifi_80211_tx(WIFI_IF_AP, ds2ds_pdu, sizeof(ds2ds_pdu), true); vTaskDelay(50 / portTICK_PERIOD_MS); } stop_wifi(); } TEST_CASE_MULTIPLE_DEVICES("receive ds2ds packet without exception", "[wifi][test_env=UT_T2_1]", receive_ds2ds_packet, send_ds2ds_packet); static void wifi_connect_by_bssid(uint8_t *bssid) { EventBits_t bits; wifi_config_t w_config = { .sta.ssid = DEFAULT_SSID, .sta.password = DEFAULT_PWD, .sta.bssid_set = true, }; memcpy(w_config.sta.bssid, bssid, 6); TEST_ESP_OK(esp_wifi_set_config(WIFI_IF_STA, &w_config)); TEST_ESP_OK(esp_wifi_connect()); ESP_LOGI(TAG, "called esp_wifi_connect()"); bits = xEventGroupWaitBits(wifi_events, GOT_IP_EVENT, 1, 0, 7000/portTICK_RATE_MS); TEST_ASSERT(bits == GOT_IP_EVENT); } static void test_wifi_connection_sta(void) { char mac_str[19]; uint8_t mac[6]; EventBits_t bits; test_case_uses_tcpip(); start_wifi_as_sta(); unity_wait_for_signal_param("SoftAP mac", mac_str, 19); TEST_ASSERT_TRUE(unity_util_convert_mac_from_string(mac_str, mac)); wifi_connect_by_bssid(mac); unity_send_signal("STA connected"); bits = xEventGroupWaitBits(wifi_events, DISCONNECT_EVENT, 1, 0, 60000 / portTICK_RATE_MS); // disconnect event not triggered printf("wait finish\n"); TEST_ASSERT(bits == 0); stop_wifi(); } static void test_wifi_connection_softap(void) { char mac_str[19] = {0}; uint8_t mac[6]; test_case_uses_tcpip(); start_wifi_as_softap(); TEST_ESP_OK(esp_wifi_get_mac(WIFI_IF_AP, mac)); sprintf(mac_str, MACSTR, MAC2STR(mac)); unity_send_signal_param("SoftAP mac", mac_str); unity_wait_for_signal("STA connected"); vTaskDelay(60000 / portTICK_PERIOD_MS); stop_wifi(); } TEST_CASE_MULTIPLE_DEVICES("test wifi retain connection for 60s", "[wifi][test_env=UT_T2_1][timeout=90]", test_wifi_connection_sta, test_wifi_connection_softap); #endif