You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
108 lines
2.3 KiB
108 lines
2.3 KiB
9 years ago
|
#include <esp8266.h>
|
||
|
|
#include <httpd.h>
|
||
|
|
#include "wificontrol.h"
|
||
|
|
#include <sbmp.h>
|
||
|
|
#include "datalink.h"
|
||
|
|
#include "reporting.h"
|
||
|
|
|
||
|
|
// --- AP ---
|
||
|
|
|
||
|
|
void FLASH_FN wificontrol_setmode_ap(void)
|
||
|
|
{
|
||
|
|
wifi_station_disconnect(); // disconnect
|
||
|
|
wifi_set_opmode(SOFTAP_MODE); // enter AP mode
|
||
|
|
|
||
|
|
warn("SBMP request to switch to SoftAP mode. Restarting...\n");
|
||
|
|
|
||
|
|
system_restart(); // TODO check if restart is needed
|
||
|
|
}
|
||
|
|
|
||
|
|
// --- WPS ---
|
||
|
|
|
||
|
|
static bool wps_in_progress = false;
|
||
|
|
|
||
|
|
static ETSTimer wps_abort_timer;
|
||
|
|
|
||
|
|
/** Timer CB - abort WPS */
|
||
|
|
static void FLASH_FN abort_wps_cb(void *arg)
|
||
|
|
{
|
||
|
|
(void)arg;
|
||
|
|
error("WPS timeout, stopped.");
|
||
|
|
wifi_wps_disable();
|
||
|
|
wps_in_progress = false;
|
||
|
|
}
|
||
|
|
|
||
|
|
static void FLASH_FN my_wps_cb(int status)
|
||
|
|
{
|
||
|
|
dbg("WPS callback!");
|
||
|
|
wps_in_progress = false;
|
||
|
|
|
||
|
|
switch (status) {
|
||
|
|
case WPS_CB_ST_SUCCESS:
|
||
|
|
info("WPS success!");
|
||
|
|
wifi_wps_disable();
|
||
|
|
wifi_station_connect();
|
||
|
|
break;
|
||
|
|
|
||
|
|
default:
|
||
|
|
error("WPS failed.");
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
|
||
|
|
// TODO send status update to STM32
|
||
|
|
}
|
||
|
|
|
||
|
|
void FLASH_FN wificontrol_start_wps(void)
|
||
|
|
{
|
||
|
|
info("Starting WPS...");
|
||
|
|
|
||
|
|
// Make sure station is enabled
|
||
|
|
if (wifi_get_opmode() == SOFTAP_MODE) {
|
||
|
|
wifi_set_opmode(STATIONAP_MODE);
|
||
|
|
}
|
||
|
|
|
||
|
|
// Disconnect if connected to any station
|
||
|
|
wifi_station_disconnect();
|
||
|
|
|
||
|
|
// enable WPS
|
||
|
|
wifi_set_wps_cb(my_wps_cb);
|
||
|
|
wifi_wps_enable(WPS_TYPE_PBC);
|
||
|
|
|
||
|
|
os_timer_disarm(&wps_abort_timer);
|
||
|
|
os_timer_setfn(&wps_abort_timer, abort_wps_cb, NULL);
|
||
|
|
os_timer_arm(&wps_abort_timer, 15000, false); // 15 seconds
|
||
|
|
|
||
|
|
wps_in_progress = true;
|
||
|
|
}
|
||
|
|
|
||
|
|
// --- Status update ---
|
||
|
|
|
||
|
|
static ETSTimer wifistatus_timer;
|
||
|
|
|
||
|
|
/** Send wifi status update */
|
||
|
|
static void FLASH_FN wifistatus_cb(void *arg)
|
||
|
|
{
|
||
|
|
(void)arg;
|
||
|
|
|
||
|
|
WIFI_MODE opmode = wifi_get_opmode();
|
||
|
|
|
||
|
|
STATION_STATUS status = STATION_IDLE;
|
||
|
|
if (opmode != SOFTAP_MODE) {
|
||
|
|
status = wifi_station_get_connect_status();
|
||
|
|
}
|
||
|
|
|
||
|
|
if (sbmp_ep_start_message(dlnk_ep, DG_WIFI_STATUS, 4, NULL)) {
|
||
|
|
sbmp_ep_send_u8(dlnk_ep, opmode); // 1-Client, 2-SoftAP, 3-STA+AP
|
||
|
|
sbmp_ep_send_u8(dlnk_ep, status == STATION_GOT_IP); // Station connected
|
||
|
|
sbmp_ep_send_u8(dlnk_ep, wps_in_progress); // 1 = WPS is in progress
|
||
|
|
sbmp_ep_send_u8(dlnk_ep, rpt_conf.enabled); // Reporting is enabled
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
void FLASH_FN wificontrol_init(void)
|
||
|
|
{
|
||
|
|
os_timer_disarm(&wifistatus_timer);
|
||
|
|
os_timer_setfn(&wifistatus_timer, wifistatus_cb, NULL);
|
||
|
|
os_timer_arm(&wifistatus_timer, 1000, true);
|
||
|
|
}
|