402 lines
13 KiB
C
402 lines
13 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 <esp_log.h>
|
|
#include <esp_err.h>
|
|
#include <esp_wifi.h>
|
|
#include <nvs_flash.h>
|
|
#include <nvs.h>
|
|
|
|
#include <protocomm.h>
|
|
#include <protocomm_httpd.h>
|
|
#include <protocomm_security0.h>
|
|
#include <protocomm_security1.h>
|
|
#include <wifi_provisioning/wifi_config.h>
|
|
#include <custom_provisioning/custom_config.h>
|
|
|
|
#include "app_prov.h"
|
|
|
|
static const char *TAG = "app_prov";
|
|
|
|
/* Handlers for provisioning endpoints */
|
|
extern wifi_prov_config_handlers_t wifi_prov_handlers;
|
|
extern custom_prov_config_handler_t custom_prov_handler;
|
|
|
|
/**
|
|
* @brief Data relevant to provisioning application
|
|
*/
|
|
struct app_prov_data {
|
|
protocomm_t *pc; /*!< Protocomm handler */
|
|
int security; /*!< Type of security to use with protocomm */
|
|
const protocomm_security_pop_t *pop; /*!< Pointer to proof of possession */
|
|
esp_timer_handle_t timer; /*!< Handle to timer */
|
|
|
|
/* State of WiFi Station */
|
|
wifi_prov_sta_state_t wifi_state;
|
|
|
|
/* Code for WiFi station disconnection (if disconnected) */
|
|
wifi_prov_sta_fail_reason_t wifi_disconnect_reason;
|
|
};
|
|
|
|
/* Pointer to provisioning application data */
|
|
static struct app_prov_data *g_prov;
|
|
|
|
static esp_err_t app_prov_start_service(void)
|
|
{
|
|
/* Create new protocomm instance */
|
|
g_prov->pc = protocomm_new();
|
|
if (g_prov->pc == NULL) {
|
|
ESP_LOGE(TAG, "Failed to create new protocomm instance");
|
|
return ESP_FAIL;
|
|
}
|
|
|
|
/* Config for protocomm_httpd_start() */
|
|
protocomm_httpd_config_t pc_config = {
|
|
.data = {
|
|
.config = PROTOCOMM_HTTPD_DEFAULT_CONFIG()
|
|
}
|
|
};
|
|
|
|
/* Start protocomm server on top of HTTP */
|
|
if (protocomm_httpd_start(g_prov->pc, &pc_config) != ESP_OK) {
|
|
ESP_LOGE(TAG, "Failed to start protocomm HTTP server");
|
|
return ESP_FAIL;
|
|
}
|
|
|
|
/* Set protocomm version verification endpoint for protocol */
|
|
protocomm_set_version(g_prov->pc, "proto-ver", "V0.1");
|
|
|
|
/* Set protocomm security type for endpoint */
|
|
if (g_prov->security == 0) {
|
|
protocomm_set_security(g_prov->pc, "prov-session", &protocomm_security0, NULL);
|
|
} else if (g_prov->security == 1) {
|
|
protocomm_set_security(g_prov->pc, "prov-session", &protocomm_security1, g_prov->pop);
|
|
}
|
|
|
|
/* Add endpoint for provisioning to set WiFi STA config */
|
|
if (protocomm_add_endpoint(g_prov->pc, "prov-config",
|
|
wifi_prov_config_data_handler,
|
|
(void *) &wifi_prov_handlers) != ESP_OK) {
|
|
ESP_LOGE(TAG, "Failed to set WiFi provisioning endpoint");
|
|
protocomm_httpd_stop(g_prov->pc);
|
|
return ESP_FAIL;
|
|
}
|
|
|
|
/* Add endpoint for provisioning to set custom config */
|
|
if (protocomm_add_endpoint(g_prov->pc, "custom-config",
|
|
custom_prov_config_data_handler,
|
|
(void *) custom_prov_handler) != ESP_OK) {
|
|
ESP_LOGE(TAG, "Failed to set custom provisioning endpoint");
|
|
protocomm_httpd_stop(g_prov->pc);
|
|
return ESP_FAIL;
|
|
}
|
|
return ESP_OK;
|
|
}
|
|
|
|
static void app_prov_stop_service(void)
|
|
{
|
|
/* Remove provisioning endpoint for custom config */
|
|
protocomm_remove_endpoint(g_prov->pc, "custom-config");
|
|
/* Remove provisioning endpoint for WiFi STA config */
|
|
protocomm_remove_endpoint(g_prov->pc, "prov-config");
|
|
/* Unset provisioning security */
|
|
protocomm_unset_security(g_prov->pc, "prov-session");
|
|
/* Unset provisioning version endpoint */
|
|
protocomm_unset_version(g_prov->pc, "proto-ver");
|
|
/* Stop protocomm server */
|
|
protocomm_httpd_stop(g_prov->pc);
|
|
/* Delete protocomm instance */
|
|
protocomm_delete(g_prov->pc);
|
|
}
|
|
|
|
/* Callback to be invoked by timer */
|
|
static void _stop_softap_cb(void * arg)
|
|
{
|
|
ESP_LOGI(TAG, "Stopping provisioning");
|
|
app_prov_stop_service();
|
|
esp_wifi_set_mode(WIFI_MODE_STA);
|
|
|
|
/* Timer not needed anymore */
|
|
esp_timer_handle_t timer = g_prov->timer;
|
|
esp_timer_delete(timer);
|
|
g_prov->timer = NULL;
|
|
|
|
/* Free provisioning process data */
|
|
free(g_prov);
|
|
g_prov = NULL;
|
|
ESP_LOGI(TAG, "Provisioning stopped");
|
|
}
|
|
|
|
/* Event handler for starting/stopping provisioning.
|
|
* To be called from within the context of the main
|
|
* event handler.
|
|
*/
|
|
esp_err_t app_prov_event_handler(void *ctx, system_event_t *event)
|
|
{
|
|
/* For accessing reason codes in case of disconnection */
|
|
system_event_info_t *info = &event->event_info;
|
|
|
|
/* If pointer to provisioning application data is NULL
|
|
* then provisioning is not running, therefore return without
|
|
* error */
|
|
if (!g_prov) {
|
|
return ESP_OK;
|
|
}
|
|
|
|
switch(event->event_id) {
|
|
case SYSTEM_EVENT_STA_START:
|
|
ESP_LOGI(TAG, "STA Start");
|
|
/* Once configuration is received by protocomm server,
|
|
* device is restarted as both AP and Station.
|
|
* Once station starts, wait for connection to
|
|
* establish with configured host SSID and password */
|
|
g_prov->wifi_state = WIFI_PROV_STA_CONNECTING;
|
|
break;
|
|
|
|
case SYSTEM_EVENT_STA_GOT_IP:
|
|
ESP_LOGI(TAG, "STA Got IP");
|
|
/* Station got IP. That means configuraion is successful.
|
|
* Schedule timer to stop provisioning app after 30 seconds. */
|
|
g_prov->wifi_state = WIFI_PROV_STA_CONNECTED;
|
|
if (g_prov && g_prov->timer) {
|
|
/* Note that, after restarting the WiFi in Station + AP mode, the
|
|
* user gets disconnected from the AP for a while. But at the same
|
|
* time, the user app requests for status update from the device
|
|
* to verify that the provisioning was successful. Therefore, the
|
|
* turning off of the AP must be delayed long enough for the user
|
|
* to reconnect and get STA connection status from the device.
|
|
* Otherwise, the AP will be turned off before the user can
|
|
* reconnect and thus the user app will see connection timed out,
|
|
* signalling a failure in provisioning. */
|
|
esp_timer_start_once(g_prov->timer, 30000*1000U);
|
|
}
|
|
break;
|
|
|
|
case SYSTEM_EVENT_STA_DISCONNECTED:
|
|
ESP_LOGE(TAG, "STA Disconnected");
|
|
/* Station couldn't connect to configured host SSID */
|
|
g_prov->wifi_state = WIFI_PROV_STA_DISCONNECTED;
|
|
ESP_LOGE(TAG, "Disconnect reason : %d", info->disconnected.reason);
|
|
|
|
/* Set code corresponding to the reason for disconnection */
|
|
switch (info->disconnected.reason) {
|
|
case WIFI_REASON_AUTH_EXPIRE:
|
|
case WIFI_REASON_4WAY_HANDSHAKE_TIMEOUT:
|
|
case WIFI_REASON_BEACON_TIMEOUT:
|
|
case WIFI_REASON_AUTH_FAIL:
|
|
case WIFI_REASON_ASSOC_FAIL:
|
|
case WIFI_REASON_HANDSHAKE_TIMEOUT:
|
|
ESP_LOGI(TAG, "STA Auth Error");
|
|
g_prov->wifi_disconnect_reason = WIFI_PROV_STA_AUTH_ERROR;
|
|
break;
|
|
case WIFI_REASON_NO_AP_FOUND:
|
|
ESP_LOGI(TAG, "STA AP Not found");
|
|
g_prov->wifi_disconnect_reason = WIFI_PROV_STA_AP_NOT_FOUND;
|
|
break;
|
|
default:
|
|
/* If none of the expected reasons,
|
|
* retry connecting to host SSID */
|
|
g_prov->wifi_state = WIFI_PROV_STA_CONNECTING;
|
|
esp_wifi_connect();
|
|
}
|
|
break;
|
|
|
|
default:
|
|
break;
|
|
}
|
|
return ESP_OK;
|
|
}
|
|
|
|
esp_err_t app_prov_get_wifi_state(wifi_prov_sta_state_t* state)
|
|
{
|
|
if (g_prov == NULL || state == NULL) {
|
|
return ESP_FAIL;
|
|
}
|
|
|
|
*state = g_prov->wifi_state;
|
|
return ESP_OK;
|
|
}
|
|
|
|
esp_err_t app_prov_get_wifi_disconnect_reason(wifi_prov_sta_fail_reason_t* reason)
|
|
{
|
|
if (g_prov == NULL || reason == NULL) {
|
|
return ESP_FAIL;
|
|
}
|
|
|
|
if (g_prov->wifi_state != WIFI_PROV_STA_DISCONNECTED) {
|
|
return ESP_FAIL;
|
|
}
|
|
|
|
*reason = g_prov->wifi_disconnect_reason;
|
|
return ESP_OK;
|
|
}
|
|
|
|
esp_err_t app_prov_is_provisioned(bool *provisioned)
|
|
{
|
|
#ifdef CONFIG_RESET_PROVISIONED
|
|
nvs_flash_erase();
|
|
#endif
|
|
|
|
if (nvs_flash_init() != ESP_OK) {
|
|
ESP_LOGE(TAG, "Failed to init NVS");
|
|
return ESP_FAIL;
|
|
}
|
|
|
|
wifi_init_config_t cfg = WIFI_INIT_CONFIG_DEFAULT();
|
|
if (esp_wifi_init(&cfg) != ESP_OK) {
|
|
ESP_LOGE(TAG, "Failed to init wifi");
|
|
return ESP_FAIL;
|
|
}
|
|
|
|
/* Get WiFi Station configuration */
|
|
wifi_config_t wifi_cfg;
|
|
if (esp_wifi_get_config(ESP_IF_WIFI_STA, &wifi_cfg) != ESP_OK) {
|
|
*provisioned = false;
|
|
return ESP_FAIL;
|
|
}
|
|
|
|
if (strlen((const char*) wifi_cfg.sta.ssid)) {
|
|
*provisioned = true;
|
|
ESP_LOGI(TAG, "Found ssid %s", (const char*) wifi_cfg.sta.ssid);
|
|
ESP_LOGI(TAG, "Found password %s", (const char*) wifi_cfg.sta.password);
|
|
}
|
|
return ESP_OK;
|
|
}
|
|
|
|
esp_err_t app_prov_configure_sta(wifi_config_t *wifi_cfg)
|
|
{
|
|
/* Configure WiFi as both AP and Station */
|
|
if (esp_wifi_set_mode(WIFI_MODE_APSTA) != ESP_OK) {
|
|
ESP_LOGE(TAG, "Failed to set WiFi mode");
|
|
return ESP_FAIL;
|
|
}
|
|
/* Configure WiFi station with host credentials
|
|
* provided during provisioning */
|
|
if (esp_wifi_set_config(ESP_IF_WIFI_STA, wifi_cfg) != ESP_OK) {
|
|
ESP_LOGE(TAG, "Failed to set WiFi configuration");
|
|
return ESP_FAIL;
|
|
}
|
|
/* Restart WiFi */
|
|
if (esp_wifi_start() != ESP_OK) {
|
|
ESP_LOGE(TAG, "Failed to restart WiFi");
|
|
return ESP_FAIL;
|
|
}
|
|
/* Connect to AP */
|
|
if (esp_wifi_connect() != ESP_OK) {
|
|
ESP_LOGE(TAG, "Failed to connect WiFi");
|
|
return ESP_FAIL;
|
|
}
|
|
|
|
if (g_prov) {
|
|
/* Reset wifi station state for provisioning app */
|
|
g_prov->wifi_state = WIFI_PROV_STA_CONNECTING;
|
|
}
|
|
return ESP_OK;
|
|
}
|
|
|
|
static esp_err_t start_wifi_ap(const char *ssid, const char *pass)
|
|
{
|
|
/* Initialise WiFi with default configuration */
|
|
wifi_init_config_t cfg = WIFI_INIT_CONFIG_DEFAULT();
|
|
esp_err_t err = esp_wifi_init(&cfg);
|
|
if (err != ESP_OK) {
|
|
ESP_LOGE(TAG, "Failed to init WiFi : %d", err);
|
|
return err;
|
|
}
|
|
|
|
/* Build WiFi configuration for AP mode */
|
|
wifi_config_t wifi_config = {
|
|
.ap = {
|
|
.max_connection = 5,
|
|
},
|
|
};
|
|
|
|
strncpy((char *) wifi_config.ap.ssid, ssid, sizeof(wifi_config.ap.ssid));
|
|
wifi_config.ap.ssid_len = strlen(ssid);
|
|
|
|
if (strlen(pass) == 0) {
|
|
memset(wifi_config.ap.password, 0, sizeof(wifi_config.ap.password));
|
|
wifi_config.ap.authmode = WIFI_AUTH_OPEN;
|
|
} else {
|
|
strncpy((char *) wifi_config.ap.password, pass, sizeof(wifi_config.ap.password));
|
|
wifi_config.ap.authmode = WIFI_AUTH_WPA_WPA2_PSK;
|
|
}
|
|
|
|
/* Start WiFi in AP mode with configuration built above */
|
|
err = esp_wifi_set_mode(WIFI_MODE_AP);
|
|
if (err != ESP_OK) {
|
|
ESP_LOGE(TAG, "Failed to set WiFi mode : %d", err);
|
|
return err;
|
|
}
|
|
err = esp_wifi_set_config(ESP_IF_WIFI_AP, &wifi_config);
|
|
if (err != ESP_OK) {
|
|
ESP_LOGE(TAG, "Failed to set WiFi config : %d", err);
|
|
return err;
|
|
}
|
|
err = esp_wifi_start();
|
|
if (err != ESP_OK) {
|
|
ESP_LOGE(TAG, "Failed to start WiFi : %d", err);
|
|
return err;
|
|
}
|
|
|
|
return ESP_OK;
|
|
}
|
|
|
|
esp_err_t app_prov_start_softap_provisioning(const char *ssid, const char *pass,
|
|
int security, const protocomm_security_pop_t *pop)
|
|
{
|
|
/* If provisioning app data present,
|
|
* means provisioning app is already running */
|
|
if (g_prov) {
|
|
ESP_LOGI(TAG, "Invalid provisioning state");
|
|
return ESP_FAIL;
|
|
}
|
|
|
|
/* Allocate memory for provisioning app data */
|
|
g_prov = (struct app_prov_data *) calloc(1, sizeof(struct app_prov_data));
|
|
if (!g_prov) {
|
|
ESP_LOGI(TAG, "Unable to allocate prov data");
|
|
return ESP_ERR_NO_MEM;
|
|
}
|
|
|
|
/* Initialise app data */
|
|
g_prov->pop = pop;
|
|
g_prov->security = security;
|
|
|
|
/* Create timer object as a member of app data */
|
|
esp_timer_create_args_t timer_conf = {
|
|
.callback = _stop_softap_cb,
|
|
.arg = NULL,
|
|
.dispatch_method = ESP_TIMER_TASK,
|
|
.name = "stop_softap_tm"
|
|
};
|
|
esp_err_t err = esp_timer_create(&timer_conf, &g_prov->timer);
|
|
if (err != ESP_OK) {
|
|
ESP_LOGE(TAG, "Failed to create timer");
|
|
return err;
|
|
}
|
|
|
|
/* Start WiFi softAP with specified ssid and password */
|
|
err = start_wifi_ap(ssid, pass);
|
|
if (err != ESP_OK) {
|
|
ESP_LOGE(TAG, "Failed to start WiFi AP");
|
|
return err;
|
|
}
|
|
|
|
/* Start provisioning service through HTTP */
|
|
err = app_prov_start_service();
|
|
if (err != ESP_OK) {
|
|
ESP_LOGE(TAG, "Failed to start provisioning app");
|
|
return err;
|
|
}
|
|
|
|
ESP_LOGI(TAG, "SoftAP Provisioning started with SSID %s, Password %s", ssid, pass);
|
|
return ESP_OK;
|
|
}
|