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.
583 lines
15 KiB
583 lines
15 KiB
/*
|
|
Cgi/template routines for the /wifi url.
|
|
*/
|
|
|
|
/*
|
|
* ----------------------------------------------------------------------------
|
|
* "THE BEER-WARE LICENSE" (Revision 42):
|
|
* Jeroen Domburg <jeroen@spritesmods.com> wrote this file. As long as you retain
|
|
* this notice you can do whatever you want with this stuff. If we meet some day,
|
|
* and you think this stuff is worth it, you can buy me a beer in return.
|
|
* ----------------------------------------------------------------------------
|
|
*
|
|
* File adapted and improved by Ondřej Hruška <ondra@ondrovo.com>
|
|
*/
|
|
|
|
#include <esp8266.h>
|
|
#include "cgi_wifi.h"
|
|
#include "wifimgr.h"
|
|
#include "persist.h"
|
|
#include "helpers.h"
|
|
|
|
#define SET_REDIR_SUC "/cfg/wifi"
|
|
#define SET_REDIR_ERR SET_REDIR_SUC"?err="
|
|
|
|
/** WiFi access point data */
|
|
typedef struct {
|
|
char ssid[32];
|
|
char bssid[8];
|
|
int channel;
|
|
char rssi;
|
|
char enc;
|
|
} ApData;
|
|
|
|
/** Scan result type */
|
|
typedef struct {
|
|
char scanInProgress; //if 1, don't access the underlying stuff from the webpage.
|
|
ApData **apData;
|
|
int noAps;
|
|
} ScanResultData;
|
|
|
|
/** Static scan status storage. */
|
|
static ScanResultData cgiWifiAps;
|
|
|
|
/** Connection to AP periodic check timer */
|
|
static os_timer_t staCheckTimer;
|
|
|
|
/**
|
|
* Calculate approximate signal strength % from RSSI
|
|
*/
|
|
int ICACHE_FLASH_ATTR rssi2perc(int rssi)
|
|
{
|
|
int r;
|
|
|
|
if (rssi > 200)
|
|
r = 100;
|
|
else if (rssi < 100)
|
|
r = 0;
|
|
else
|
|
r = 100 - 2 * (200 - rssi); // approx.
|
|
|
|
if (r > 100) r = 100;
|
|
if (r < 0) r = 0;
|
|
|
|
return r;
|
|
}
|
|
|
|
/**
|
|
* Convert Auth type to string
|
|
*/
|
|
const ICACHE_FLASH_ATTR char *auth2str(AUTH_MODE auth)
|
|
{
|
|
switch (auth) {
|
|
case AUTH_OPEN:
|
|
return "Open";
|
|
case AUTH_WEP:
|
|
return "WEP";
|
|
case AUTH_WPA_PSK:
|
|
return "WPA";
|
|
case AUTH_WPA2_PSK:
|
|
return "WPA2";
|
|
case AUTH_WPA_WPA2_PSK:
|
|
return "WPA/WPA2";
|
|
default:
|
|
return "Unknown";
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Convert WiFi opmode to string
|
|
*/
|
|
const ICACHE_FLASH_ATTR char *opmode2str(WIFI_MODE opmode)
|
|
{
|
|
switch (opmode) {
|
|
case NULL_MODE:
|
|
return "Disabled";
|
|
case STATION_MODE:
|
|
return "Client";
|
|
case SOFTAP_MODE:
|
|
return "AP only";
|
|
case STATIONAP_MODE:
|
|
return "Client+AP";
|
|
default:
|
|
return "Unknown";
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Callback the code calls when a wlan ap scan is done. Basically stores the result in
|
|
* the static cgiWifiAps struct.
|
|
*
|
|
* @param arg - a pointer to {struct bss_info}, which is a linked list of the found APs
|
|
* @param status - OK if the scan succeeded
|
|
*/
|
|
void ICACHE_FLASH_ATTR wifiScanDoneCb(void *arg, STATUS status)
|
|
{
|
|
int n;
|
|
struct bss_info *bss_link = (struct bss_info *) arg;
|
|
dbg("wifiScanDoneCb %d", status);
|
|
if (status != OK) {
|
|
cgiWifiAps.scanInProgress = 0;
|
|
return;
|
|
}
|
|
|
|
// Clear prev ap data if needed.
|
|
if (cgiWifiAps.apData != NULL) {
|
|
for (n = 0; n < cgiWifiAps.noAps; n++) free(cgiWifiAps.apData[n]);
|
|
free(cgiWifiAps.apData);
|
|
}
|
|
|
|
// Count amount of access points found.
|
|
n = 0;
|
|
while (bss_link != NULL) {
|
|
bss_link = bss_link->next.stqe_next;
|
|
n++;
|
|
}
|
|
// Allocate memory for access point data
|
|
cgiWifiAps.apData = (ApData **) malloc(sizeof(ApData *) * n);
|
|
if (cgiWifiAps.apData == NULL) {
|
|
error("Out of memory allocating apData");
|
|
return;
|
|
}
|
|
cgiWifiAps.noAps = n;
|
|
info("Scan done: found %d APs", n);
|
|
|
|
// Copy access point data to the static struct
|
|
n = 0;
|
|
bss_link = (struct bss_info *) arg;
|
|
while (bss_link != NULL) {
|
|
if (n >= cgiWifiAps.noAps) {
|
|
// This means the bss_link changed under our nose. Shouldn't happen!
|
|
// Break because otherwise we will write in unallocated memory.
|
|
error("Huh? I have more than the allocated %d aps!", cgiWifiAps.noAps);
|
|
break;
|
|
}
|
|
// Save the ap data.
|
|
cgiWifiAps.apData[n] = (ApData *) malloc(sizeof(ApData));
|
|
if (cgiWifiAps.apData[n] == NULL) {
|
|
error("Can't allocate mem for ap buff.");
|
|
cgiWifiAps.scanInProgress = 0;
|
|
return;
|
|
}
|
|
cgiWifiAps.apData[n]->rssi = bss_link->rssi;
|
|
cgiWifiAps.apData[n]->channel = bss_link->channel;
|
|
cgiWifiAps.apData[n]->enc = bss_link->authmode;
|
|
strncpy(cgiWifiAps.apData[n]->ssid, (char *) bss_link->ssid, 32);
|
|
strncpy(cgiWifiAps.apData[n]->bssid, (char *) bss_link->bssid, 6);
|
|
|
|
bss_link = bss_link->next.stqe_next;
|
|
n++;
|
|
}
|
|
// We're done.
|
|
cgiWifiAps.scanInProgress = 0;
|
|
}
|
|
|
|
/**
|
|
* Routine to start a WiFi access point scan.
|
|
*/
|
|
static void ICACHE_FLASH_ATTR wifiStartScan(void)
|
|
{
|
|
if (cgiWifiAps.scanInProgress) return;
|
|
cgiWifiAps.scanInProgress = 1;
|
|
wifi_station_scan(NULL, wifiScanDoneCb);
|
|
}
|
|
|
|
/**
|
|
* This CGI is called from the bit of AJAX-code in wifi.tpl. It will initiate a
|
|
* scan for access points and if available will return the result of an earlier scan.
|
|
* The result is embedded in a bit of JSON parsed by the javascript in wifi.tpl.
|
|
*/
|
|
httpd_cgi_state ICACHE_FLASH_ATTR cgiWiFiScan(HttpdConnData *connData)
|
|
{
|
|
int pos = (int) connData->cgiData;
|
|
int len;
|
|
char buff[256];
|
|
|
|
// 2nd and following runs of the function via MORE:
|
|
if (!cgiWifiAps.scanInProgress && pos != 0) {
|
|
// Fill in json code for an access point
|
|
if (pos - 1 < cgiWifiAps.noAps) {
|
|
int rssi = cgiWifiAps.apData[pos - 1]->rssi;
|
|
|
|
len = sprintf(buff, "{\"essid\": \"%s\", \"bssid\": \""
|
|
MACSTR
|
|
"\", \"rssi\": %d, \"rssi_perc\": %d, \"enc\": %d, \"channel\": %d}%s",
|
|
cgiWifiAps.apData[pos - 1]->ssid,
|
|
MAC2STR(cgiWifiAps.apData[pos - 1]->bssid),
|
|
rssi,
|
|
rssi2perc(rssi),
|
|
cgiWifiAps.apData[pos - 1]->enc,
|
|
cgiWifiAps.apData[pos - 1]->channel,
|
|
(pos - 1 == cgiWifiAps.noAps - 1) ? "\n " : ",\n "); //<-terminator
|
|
|
|
httpdSend(connData, buff, len);
|
|
}
|
|
pos++;
|
|
if ((pos - 1) >= cgiWifiAps.noAps) {
|
|
len = sprintf(buff, " ]\n }\n}"); // terminate the whole object
|
|
httpdSend(connData, buff, len);
|
|
// Also start a new scan.
|
|
wifiStartScan();
|
|
return HTTPD_CGI_DONE;
|
|
}
|
|
else {
|
|
connData->cgiData = (void *) pos;
|
|
return HTTPD_CGI_MORE;
|
|
}
|
|
}
|
|
|
|
// First run of the function
|
|
httpdStartResponse(connData, 200);
|
|
httpdHeader(connData, "Content-Type", "application/json");
|
|
httpdEndHeaders(connData);
|
|
|
|
if (cgiWifiAps.scanInProgress == 1) {
|
|
// We're still scanning. Tell Javascript code that.
|
|
len = sprintf(buff, "{\n \"result\": {\n \"inProgress\": 1\n }\n}");
|
|
httpdSend(connData, buff, len);
|
|
return HTTPD_CGI_DONE;
|
|
}
|
|
else {
|
|
// We have a scan result. Pass it on.
|
|
len = sprintf(buff, "{\n \"result\": {\n \"inProgress\": 0,\n \"APs\": [\n ");
|
|
httpdSend(connData, buff, len);
|
|
if (cgiWifiAps.apData == NULL) cgiWifiAps.noAps = 0;
|
|
connData->cgiData = (void *) 1;
|
|
return HTTPD_CGI_MORE;
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Cgi to get connection status.
|
|
*
|
|
* This endpoint returns JSON with keys:
|
|
* - status = 'idle', 'working' or 'fail',
|
|
* - ip = IP address, after connection succeeds
|
|
*/
|
|
httpd_cgi_state ICACHE_FLASH_ATTR cgiWiFiConnStatus(HttpdConnData *connData)
|
|
{
|
|
char buff[100];
|
|
struct ip_info info;
|
|
|
|
httpdStartResponse(connData, 200);
|
|
httpdHeader(connData, "Content-Type", "application/json");
|
|
httpdEndHeaders(connData);
|
|
|
|
// if bad opmode or no SSID configured, skip any checks
|
|
if (!(wificonf->opmode & STATION_MODE) || wificonf->sta_ssid[0] == 0) {
|
|
httpdSend(connData, "{\"status\": \"disabled\"}", -1);
|
|
return HTTPD_CGI_DONE;
|
|
}
|
|
|
|
STATION_STATUS st = wifi_station_get_connect_status();
|
|
switch(st) {
|
|
case STATION_IDLE:
|
|
sprintf(buff, "{\"status\": \"idle\"}"); // unclear when this is used
|
|
break;
|
|
|
|
case STATION_CONNECTING:
|
|
sprintf(buff, "{\"status\": \"working\"}");
|
|
break;
|
|
|
|
case STATION_WRONG_PASSWORD:
|
|
sprintf(buff, "{\"status\": \"fail\", \"cause\": \"WRONG_PASSWORD\"}");
|
|
break;
|
|
|
|
case STATION_NO_AP_FOUND:
|
|
sprintf(buff, "{\"status\": \"fail\", \"cause\": \"AP_NOT_FOUND\"}");
|
|
break;
|
|
|
|
case STATION_CONNECT_FAIL:
|
|
sprintf(buff, "{\"status\": \"fail\", \"cause\": \"CONNECTION_FAILED\"}");
|
|
break;
|
|
|
|
case STATION_GOT_IP:
|
|
wifi_get_ip_info(STATION_IF, &info);
|
|
sprintf(buff, "{\"status\": \"success\", \"ip\": \""IPSTR"\"}", GOOD_IP2STR(info.ip.addr));
|
|
break;
|
|
}
|
|
|
|
httpdSend(connData, buff, -1);
|
|
return HTTPD_CGI_DONE;
|
|
}
|
|
|
|
/**
|
|
* Callback for async timer
|
|
*/
|
|
static void ICACHE_FLASH_ATTR applyWifiSettingsLaterCb(void *arg)
|
|
{
|
|
(void*)arg;
|
|
wifimgr_apply_settings();
|
|
}
|
|
|
|
/**
|
|
* Universal CGI endpoint to set WiFi params.
|
|
* Note that some may cause a (delayed) restart.
|
|
*/
|
|
httpd_cgi_state ICACHE_FLASH_ATTR cgiWiFiSetParams(HttpdConnData *connData)
|
|
{
|
|
static ETSTimer timer;
|
|
|
|
char buff[50];
|
|
|
|
char redir_url_buf[300];
|
|
char *redir_url = redir_url_buf;
|
|
redir_url += sprintf(redir_url, SET_REDIR_ERR);
|
|
// we'll test if anything was printed by looking for \0 in failed_keys_buf
|
|
|
|
if (connData->conn == NULL) {
|
|
//Connection aborted. Clean up.
|
|
return HTTPD_CGI_DONE;
|
|
}
|
|
|
|
// ---- WiFi opmode ----
|
|
|
|
if (GET_ARG("opmode")) {
|
|
dbg("Setting WiFi opmode to: %s", buff);
|
|
int mode = atoi(buff);
|
|
if (mode > NULL_MODE && mode < MAX_MODE) {
|
|
wificonf->opmode = (WIFI_MODE) mode;
|
|
} else {
|
|
warn("Bad opmode value \"%s\"", buff);
|
|
redir_url += sprintf(redir_url, "opmode,");
|
|
}
|
|
}
|
|
|
|
if (GET_ARG("ap_enable")) {
|
|
dbg("Enable AP: %s", buff);
|
|
int enable = atoi(buff);
|
|
|
|
if (enable) {
|
|
wificonf->opmode |= SOFTAP_MODE;
|
|
} else {
|
|
wificonf->opmode &= ~SOFTAP_MODE;
|
|
}
|
|
}
|
|
|
|
if (GET_ARG("sta_enable")) {
|
|
dbg("Enable STA: %s", buff);
|
|
int enable = atoi(buff);
|
|
|
|
if (enable) {
|
|
wificonf->opmode |= STATION_MODE;
|
|
} else {
|
|
wificonf->opmode &= ~STATION_MODE;
|
|
}
|
|
}
|
|
|
|
// ---- AP transmit power ----
|
|
|
|
if (GET_ARG("tpw")) {
|
|
dbg("Setting AP power to: %s", buff);
|
|
int tpw = atoi(buff);
|
|
if (tpw >= 0 && tpw <= 82) { // 0 actually isn't 0 but quite low. 82 is very strong
|
|
if (wificonf->tpw != tpw) {
|
|
wificonf->tpw = (u8) tpw;
|
|
wifi_change_flags.ap = true;
|
|
}
|
|
} else {
|
|
warn("tpw %s out of allowed range 0-82.", buff);
|
|
redir_url += sprintf(redir_url, "tpw,");
|
|
}
|
|
}
|
|
|
|
// ---- AP channel (applies in AP-only mode) ----
|
|
|
|
if (GET_ARG("ap_channel")) {
|
|
info("ap_channel = %s", buff);
|
|
int channel = atoi(buff);
|
|
if (channel > 0 && channel < 15) {
|
|
if (wificonf->ap_channel != channel) {
|
|
wificonf->ap_channel = (u8) channel;
|
|
wifi_change_flags.ap = true;
|
|
}
|
|
} else {
|
|
warn("Bad channel value \"%s\", allowed 1-14", buff);
|
|
redir_url += sprintf(redir_url, "ap_channel,");
|
|
}
|
|
}
|
|
|
|
// ---- SSID name in AP mode ----
|
|
|
|
if (GET_ARG("ap_ssid")) {
|
|
// Replace all invalid ASCII with underscores
|
|
int i;
|
|
for (i = 0; i < 32; i++) {
|
|
char c = buff[i];
|
|
if (c == 0) break;
|
|
if (c < 32 || c >= 127) buff[i] = '_';
|
|
}
|
|
buff[i] = 0;
|
|
|
|
if (strlen(buff) > 0) {
|
|
if (!streq(wificonf->ap_ssid, buff)) {
|
|
info("Setting SSID to \"%s\"", buff);
|
|
strncpy_safe(wificonf->ap_ssid, buff, SSID_LEN);
|
|
wifi_change_flags.ap = true;
|
|
}
|
|
} else {
|
|
warn("Bad SSID len.");
|
|
redir_url += sprintf(redir_url, "ap_ssid,");
|
|
}
|
|
}
|
|
|
|
// ---- AP password ----
|
|
|
|
if (GET_ARG("ap_password")) {
|
|
// Users are free to use any stupid shit in ther password,
|
|
// but it may lock them out.
|
|
if (strlen(buff) == 0 || (strlen(buff) >= 8 && strlen(buff) < PASSWORD_LEN-1)) {
|
|
if (!streq(wificonf->ap_password, buff)) {
|
|
info("Setting AP password to \"%s\"", buff);
|
|
strncpy_safe(wificonf->ap_password, buff, PASSWORD_LEN);
|
|
wifi_change_flags.ap = true;
|
|
}
|
|
} else {
|
|
warn("Bad password len.");
|
|
redir_url += sprintf(redir_url, "ap_password,");
|
|
}
|
|
}
|
|
|
|
// ---- Hide AP network (do not announce) ----
|
|
|
|
if (GET_ARG("ap_hidden")) {
|
|
dbg("AP hidden = %s", buff);
|
|
int hidden = atoi(buff);
|
|
if (hidden != wificonf->ap_hidden) {
|
|
wificonf->ap_hidden = (hidden != 0);
|
|
wifi_change_flags.ap = true;
|
|
}
|
|
}
|
|
|
|
// ---- Station SSID (to connect to) ----
|
|
|
|
if (GET_ARG("sta_ssid")) {
|
|
if (!streq(wificonf->sta_ssid, buff)) {
|
|
// No verification needed, at worst it fails to connect
|
|
info("Setting station SSID to: \"%s\"", buff);
|
|
strncpy_safe(wificonf->sta_ssid, buff, SSID_LEN);
|
|
wifi_change_flags.sta = true;
|
|
}
|
|
}
|
|
|
|
// ---- Station password (empty for none is allowed) ----
|
|
|
|
if (GET_ARG("sta_password")) {
|
|
if (!streq(wificonf->sta_password, buff)) {
|
|
// No verification needed, at worst it fails to connect
|
|
info("Setting station password to: \"%s\"", buff);
|
|
strncpy_safe(wificonf->sta_password, buff, PASSWORD_LEN);
|
|
wifi_change_flags.sta = true;
|
|
}
|
|
}
|
|
|
|
if (redir_url_buf[strlen(SET_REDIR_ERR)] == 0) {
|
|
// All was OK
|
|
info("Set WiFi params - success, applying in 1000 ms");
|
|
|
|
// Settings are applied only if all was OK
|
|
//
|
|
// This is so that options that consist of multiple keys sent together are not applied
|
|
// only partially if set wrong, which could lead to eg. user losing access and having
|
|
// to reset to defaults.
|
|
persist_store();
|
|
|
|
// Delayed settings apply, so the response page has a chance to load.
|
|
// If user connects via the Station IF, they may not even notice the connection reset.
|
|
os_timer_disarm(&timer);
|
|
os_timer_setfn(&timer, applyWifiSettingsLaterCb, NULL);
|
|
os_timer_arm(&timer, 1000, false);
|
|
|
|
httpdRedirect(connData, SET_REDIR_SUC);
|
|
} else {
|
|
warn("Some WiFi settings did not validate, asking for correction");
|
|
// Some errors, appended to the URL as ?err=
|
|
httpdRedirect(connData, redir_url_buf);
|
|
}
|
|
return HTTPD_CGI_DONE;
|
|
}
|
|
|
|
|
|
//Template code for the WLAN page.
|
|
httpd_cgi_state ICACHE_FLASH_ATTR tplWlan(HttpdConnData *connData, char *token, void **arg)
|
|
{
|
|
char buff[100];
|
|
int x;
|
|
int connectStatus;
|
|
|
|
if (token == NULL) {
|
|
// We're done
|
|
return HTTPD_CGI_DONE;
|
|
}
|
|
|
|
strcpy(buff, ""); // fallback
|
|
|
|
if (streq(token, "opmode_name")) {
|
|
strcpy(buff, opmode2str(wificonf->opmode));
|
|
}
|
|
else if (streq(token, "opmode")) {
|
|
sprintf(buff, "%d", wificonf->opmode);
|
|
}
|
|
else if (streq(token, "sta_enable")) {
|
|
sprintf(buff, "%d", (wificonf->opmode & STATION_MODE) != 0);
|
|
}
|
|
else if (streq(token, "ap_enable")) {
|
|
sprintf(buff, "%d", (wificonf->opmode & SOFTAP_MODE) != 0);
|
|
}
|
|
else if (streq(token, "tpw")) {
|
|
sprintf(buff, "%d", wificonf->tpw);
|
|
}
|
|
else if (streq(token, "ap_channel")) {
|
|
sprintf(buff, "%d", wificonf->ap_channel);
|
|
}
|
|
else if (streq(token, "ap_ssid")) {
|
|
sprintf(buff, "%s", wificonf->ap_ssid);
|
|
}
|
|
else if (streq(token, "ap_password")) {
|
|
sprintf(buff, "%s", wificonf->ap_password);
|
|
}
|
|
else if (streq(token, "ap_hidden")) {
|
|
sprintf(buff, "%d", wificonf->ap_hidden);
|
|
}
|
|
else if (streq(token, "sta_ssid")) {
|
|
sprintf(buff, "%s", wificonf->sta_ssid);
|
|
}
|
|
else if (streq(token, "sta_password")) {
|
|
sprintf(buff, "%s", wificonf->sta_password);
|
|
}
|
|
else if (streq(token, "sta_rssi")) {
|
|
sprintf(buff, "%d", wifi_station_get_rssi());
|
|
}
|
|
else if (streq(token, "sta_active_ssid")) {
|
|
// For display of our current SSID
|
|
connectStatus = wifi_station_get_connect_status();
|
|
x = wifi_get_opmode();
|
|
if (x == SOFTAP_MODE || connectStatus != STATION_GOT_IP) {
|
|
strcpy(buff, "");
|
|
}
|
|
else {
|
|
struct station_config staconf;
|
|
wifi_station_get_config(&staconf);
|
|
strcpy(buff, (char *) staconf.ssid);
|
|
}
|
|
}
|
|
else if (streq(token, "sta_active_ip")) {
|
|
x = wifi_get_opmode();
|
|
connectStatus = wifi_station_get_connect_status();
|
|
|
|
if (x == SOFTAP_MODE || connectStatus != STATION_GOT_IP) {
|
|
strcpy(buff, "");
|
|
}
|
|
else {
|
|
struct ip_info info;
|
|
wifi_get_ip_info(STATION_IF, &info);
|
|
sprintf(buff, "ip: "IPSTR", mask: "IPSTR", gw: "IPSTR,
|
|
GOOD_IP2STR(info.ip.addr),
|
|
GOOD_IP2STR(info.netmask.addr),
|
|
GOOD_IP2STR(info.gw.addr));
|
|
}
|
|
}
|
|
|
|
httpdSend(connData, buff, -1);
|
|
return HTTPD_CGI_DONE;
|
|
}
|
|
|