117 lines
3.5 KiB
C
117 lines
3.5 KiB
C
/* SoftAP based Custom Provisioning 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 <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 "app_prov.h"
|
|
|
|
#define EXAMPLE_AP_RECONN_ATTEMPTS CONFIG_EXAMPLE_AP_RECONN_ATTEMPTS
|
|
|
|
static const char *TAG = "app";
|
|
|
|
static void event_handler(void* arg, esp_event_base_t event_base,
|
|
int event_id, void* event_data)
|
|
{
|
|
static int s_retry_num = 0;
|
|
|
|
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) {
|
|
if (s_retry_num < EXAMPLE_AP_RECONN_ATTEMPTS) {
|
|
esp_wifi_connect();
|
|
s_retry_num++;
|
|
ESP_LOGI(TAG, "retry to connect to the AP");
|
|
}
|
|
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:%s",
|
|
ip4addr_ntoa(&event->ip_info.ip));
|
|
s_retry_num = 0;
|
|
}
|
|
}
|
|
|
|
static void wifi_init_sta()
|
|
{
|
|
/* Set our event handling */
|
|
ESP_ERROR_CHECK(esp_event_handler_register(WIFI_EVENT, ESP_EVENT_ANY_ID, event_handler, NULL));
|
|
ESP_ERROR_CHECK(esp_event_handler_register(IP_EVENT, IP_EVENT_STA_GOT_IP, event_handler, NULL));
|
|
|
|
/* Start Wi-Fi in station mode with credentials set during provisioning */
|
|
ESP_ERROR_CHECK(esp_wifi_set_mode(WIFI_MODE_STA));
|
|
ESP_ERROR_CHECK(esp_wifi_start());
|
|
}
|
|
|
|
static void start_softap_provisioning()
|
|
{
|
|
/* Security version */
|
|
int security = 0;
|
|
/* Proof of possession */
|
|
const protocomm_security_pop_t *pop = NULL;
|
|
|
|
#ifdef CONFIG_EXAMPLE_USE_SEC_1
|
|
security = 1;
|
|
#endif
|
|
|
|
/* Having proof of possession is optional */
|
|
#ifdef CONFIG_EXAMPLE_USE_POP
|
|
const static protocomm_security_pop_t app_pop = {
|
|
.data = (uint8_t *) CONFIG_EXAMPLE_POP,
|
|
.len = (sizeof(CONFIG_EXAMPLE_POP)-1)
|
|
};
|
|
pop = &app_pop;
|
|
#endif
|
|
|
|
ESP_ERROR_CHECK(app_prov_start_softap_provisioning(
|
|
CONFIG_EXAMPLE_SSID, CONFIG_EXAMPLE_PASS, security, pop));
|
|
}
|
|
|
|
void app_main()
|
|
{
|
|
/* Initialize networking stack */
|
|
tcpip_adapter_init();
|
|
|
|
/* Create default event loop needed by the
|
|
* main app and the provisioning service */
|
|
ESP_ERROR_CHECK(esp_event_loop_create_default());
|
|
|
|
/* Initialize NVS needed by Wi-Fi */
|
|
ESP_ERROR_CHECK(nvs_flash_init());
|
|
|
|
/* Initialize Wi-Fi with default config */
|
|
wifi_init_config_t cfg = WIFI_INIT_CONFIG_DEFAULT();
|
|
ESP_ERROR_CHECK(esp_wifi_init(&cfg));
|
|
|
|
/* Check if device is provisioned */
|
|
bool provisioned;
|
|
if (app_prov_is_provisioned(&provisioned) != ESP_OK) {
|
|
ESP_LOGE(TAG, "Error getting device provisioning state");
|
|
return;
|
|
}
|
|
|
|
if (provisioned == false) {
|
|
/* If not provisioned, start provisioning via soft AP */
|
|
ESP_LOGI(TAG, "Starting WiFi SoftAP provisioning");
|
|
start_softap_provisioning();
|
|
} else {
|
|
/* Start WiFi station with credentials set during provisioning */
|
|
ESP_LOGI(TAG, "Starting WiFi station");
|
|
wifi_init_sta();
|
|
}
|
|
}
|