#include "portal.h" #include "config.h" #include "led.h" #include "api.h" #include "store.h" #include #include #include namespace { WebServer server(80); DNSServer dns; String networksHtml; String lastError; const char PAGE_HEAD[] PROGMEM = R"( SMASHR setup)"; String esc(const String& s) { String o; for (char c : s) { if (c == '<') o += "<"; else if (c == '>') o += ">"; else if (c == '&') o += "&"; else if (c == '"') o += """; else o += c; } return o; } void scan() { int n = WiFi.scanNetworks(); networksHtml = ""; for (int i = 0; i < n && i < 20; i++) { String ssid = WiFi.SSID(i); if (!ssid.length() || networksHtml.indexOf(">" + esc(ssid) + "<") >= 0) continue; networksHtml += ""; } } void handleRoot() { String h = FPSTR(PAGE_HEAD); h += "

Set up your button

Pick your home Wi-Fi (2.4 GHz) and type the 6 character code from the SMASHR dashboard.

"; if (lastError.length()) h += "

" + esc(lastError) + "

"; h += ""; server.send(200, "text/html", h); } void handleSave() { String ssid = server.arg("ssid2").length() ? server.arg("ssid2") : server.arg("ssid"); String pass = server.arg("pass"); String code = server.arg("code"); code.trim(); code.toUpperCase(); led::set(LED_CONNECTING); WiFi.begin(ssid.c_str(), pass.c_str()); uint32_t t0 = millis(); while (WiFi.status() != WL_CONNECTED && millis() - t0 < WIFI_CONNECT_MS) { delay(100); dns.processNextRequest(); } if (WiFi.status() != WL_CONNECTED) { lastError = "Could not join " + ssid + ". Check the password, and that it is a 2.4 GHz network."; led::set(LED_PORTAL); server.sendHeader("Location", "/"); server.send(303); return; } configTime(0, 0, "pool.ntp.org", "time.google.com"); String id, secret, err; if (!api::pair(code, id, secret, err)) { lastError = err; led::set(LED_PORTAL); server.sendHeader("Location", "/"); server.send(303); return; } store::saveWifi(ssid, pass); store::saveIdentity(id, secret); String h = FPSTR(PAGE_HEAD); h += "

Paired.

This is " + esc(id) + ". The ring turns green, then red. Go press it.

"; server.send(200, "text/html", h); led::settle(LED_OK, 2400, LED_IDLE); delay(2500); ESP.restart(); } void redirect() { server.sendHeader("Location", "http://192.168.4.1/"); server.send(302); } } namespace portal { void run() { String mac = WiFi.macAddress(); mac.replace(":", ""); String apName = String(AP_PREFIX) + mac.substring(8); WiFi.mode(WIFI_AP_STA); WiFi.softAP(apName.c_str()); delay(200); scan(); dns.start(53, "*", WiFi.softAPIP()); server.on("/", HTTP_GET, handleRoot); server.on("/save", HTTP_POST, handleSave); // OS captive-portal probes server.on("/generate_204", redirect); server.on("/hotspot-detect.html", redirect); server.on("/connecttest.txt", redirect); server.onNotFound(redirect); server.begin(); led::set(LED_PORTAL); Serial.printf("[portal] join %s\n", apName.c_str()); uint32_t t0 = millis(); while (millis() - t0 < PORTAL_TIMEOUT_MS) { dns.processNextRequest(); server.handleClient(); delay(2); } ESP.restart(); } }