maiana/latest/Firmware/WiFiAdapter/core/wifi.c

#include "freertos/FreeRTOS.h"
#include "freertos/task.h"
#include "esp_mac.h"
#include "esp_wifi.h"
#include "esp_log.h"
#include "wifi.h"
#include "configuration.h"
#include "../bsp/bsp.h"

static const char *TAG = "wifi";

#define MAX_WIFI_RETRIES      5


static void wifi_event_handler(void* arg, esp_event_base_t event_base, int32_t event_id, void* event_data)
{
  static int retries = 0;
  switch(event_id)
  {
    case WIFI_EVENT_AP_STACONNECTED:
    {
      wifi_event_ap_staconnected_t* event = (wifi_event_ap_staconnected_t*) event_data;
      ESP_LOGI(TAG, "station "MACSTR" join, AID=%d",
                MAC2STR(event->mac), event->aid);
      break;
    }
    case WIFI_EVENT_AP_STADISCONNECTED:
    {
      wifi_event_ap_stadisconnected_t* event = (wifi_event_ap_stadisconnected_t*) event_data;
      ESP_LOGI(TAG, "station "MACSTR" leave, AID=%d",
                MAC2STR(event->mac), event->aid);
      break;
    }
    case WIFI_EVENT_STA_START:
    {
      esp_wifi_connect();
      break;
    }
    case WIFI_EVENT_STA_CONNECTED:
    {
      if ( !config_has_wifi_success() )      
        config_set_wifi_success();
      ESP_LOGI(TAG, "Successfully attached to network");
      break;
    }
    case WIFI_EVENT_STA_DISCONNECTED:
    {
      if ( !config_has_wifi_success() )
        {
          ++retries;
          if ( retries > MAX_WIFI_RETRIES )
          {
            config_reset_wifi();
            bsp_reboot();
          }
        }

      esp_wifi_connect();
      break;
    }
    default:
      break;
  }

}

static void ip_event_handler(void* arg, esp_event_base_t event_base, int32_t event_id, void* event_data)
{
  char __ip_str[32];
  ip_event_got_ip_t *event = (ip_event_got_ip_t *)event_data;
  sprintf(__ip_str, IPSTR, IP2STR(&event->ip_info.ip));
  ESP_LOGI(TAG, "IP: %s", __ip_str);
}

void wifi_start()
{
    esp_netif_init();
    wifi_operation_mode_t mode = config_get_wifi_operation_mode();

    if ( mode != WIFI_STATION )
      esp_netif_create_default_wifi_ap();
    else
      esp_netif_create_default_wifi_sta();

    wifi_init_config_t cfg = WIFI_INIT_CONFIG_DEFAULT();
    ESP_ERROR_CHECK(esp_wifi_init(&cfg));


    wifi_config_t wifi_config = {0};
    const char *ssid = config_get_ssid();
    const char *password = config_get_password();

    esp_event_handler_instance_register(WIFI_EVENT, ESP_EVENT_ANY_ID, wifi_event_handler, NULL, NULL);
    esp_event_handler_instance_register(IP_EVENT, IP_EVENT_STA_GOT_IP, ip_event_handler, NULL, NULL);


    switch(mode)
    {
      case WIFI_OPEN_AP:
      {
        ESP_LOGI(TAG, "Starting open access point with SSID %s", ssid);
        memcpy(wifi_config.ap.ssid, ssid, strlen(ssid));
        wifi_config.ap.ssid_len = strlen(ssid);
        wifi_config.ap.authmode = WIFI_AUTH_OPEN;
        wifi_config.ap.max_connection = 8;
        esp_wifi_set_mode(WIFI_MODE_AP);
        esp_wifi_set_config(WIFI_IF_AP, &wifi_config);
      }
      break;
      case WIFI_SECURE_AP:
      {
        ESP_LOGI(TAG, "Starting secure access point with SSID %s and password %s", ssid, password);
        memcpy(wifi_config.ap.ssid, ssid, strlen(ssid));
        wifi_config.ap.ssid_len = strlen(ssid);
        strncpy((char *)wifi_config.ap.password, password, sizeof wifi_config.ap.password);
      
        wifi_config.ap.max_connection = 8;
        wifi_config.ap.authmode = WIFI_AUTH_WPA_WPA2_PSK;
        esp_wifi_set_mode(WIFI_MODE_AP);
        esp_wifi_set_config(WIFI_IF_AP, &wifi_config);
      } 
      break;
      case WIFI_STATION:
      {
        ESP_LOGI(TAG, "Starting station with SSID %s", ssid);
        memcpy(wifi_config.sta.ssid, ssid, strlen(ssid));
        strncpy((char *)wifi_config.sta.password, password, sizeof wifi_config.ap.password);
        esp_wifi_set_mode(WIFI_MODE_STA);
        esp_wifi_set_config(WIFI_IF_STA, &wifi_config);        
      }
      break;         
    }

    esp_wifi_start();
}