happycodefarm
diff --git a/‎libraries/DNSServer/examples/CaptivePortal/CaptivePortal.ino‎
Lines changed: 44 additions & 37 deletions b/‎libraries/DNSServer/examples/CaptivePortal/CaptivePortal.ino‎
Lines changed: 44 additions & 37 deletions
@@ -1,52 +1,59 @@
+/*
+This example enables catch-all Captive portal for ESP32 Access-Point
+It will allow modern devices/OSes to detect that WiFi connection is
+limited and offer a user to access a banner web-page.
+There is no need to find and open device's IP address/URL, i.e. http://192.168.4.1/
+This works for Android, Ubuntu, FireFox, Windows, maybe others...
+*/
+
+#include <Arduino.h>
 #include <WiFi.h>
 #include <DNSServer.h>
+#include <WebServer.h>
+
 
-const byte DNS_PORT = 53;
-IPAddress apIP(8,8,4,4); // The default android DNS
 DNSServer dnsServer;
-WiFiServer server(80);
+WebServer server(80);
+
+static const char responsePortal[] = R"===(
+<!DOCTYPE html><html><head><title>ESP32 CaptivePortal</title></head><body>
+<h1>Hello World!</h1><p>This is a captive portal example page. All unknown http requests will 
+be redirected here.</p></body></html>
+)===";
+
+// index page handler
+void handleRoot() {
+  server.send(200, "text/plain", "Hello from esp32!");
+}
 
-String responseHTML = ""
-  "<!DOCTYPE html><html><head><title>CaptivePortal</title></head><body>"
-  "<h1>Hello World!</h1><p>This is a captive portal example. All requests will "
-  "be redirected here.</p></body></html>";
+// this will redirect unknown http req's to our captive portal page
+// based on this redirect various systems could detect that WiFi AP has a captive portal page
+void handleNotFound() {
+  server.sendHeader("Location", "/portal");
+  server.send(302, "text/plain", "redirect to captive portal");
+}
 
-void setup() { 
+void setup() {
+  Serial.begin(115200);
   WiFi.mode(WIFI_AP);
   WiFi.softAP("ESP32-DNSServer");
-  WiFi.softAPConfig(apIP, apIP, IPAddress(255, 255, 255, 0));
 
-  // if DNSServer is started with "*" for domain name, it will reply with
-  // provided IP to all DNS request
-  dnsServer.start(DNS_PORT, "*", apIP);
+  // by default DNSServer is started serving any "*" domain name. It will reply
+  // AccessPoint's IP to all DNS request (this is requred for Captive Portal detection)
+  dnsServer.start();
+
+  // serve a simple root page
+  server.on("/", handleRoot);
+
+  // serve portal page
+  server.on("/portal",[](){server.send(200, "text/html", responsePortal);});
 
+  // all unknown pages are redirected to captive portal
+  server.onNotFound(handleNotFound);
   server.begin();
 }
 
 void loop() {
-  dnsServer.processNextRequest();
-  WiFiClient client = server.available();   // listen for incoming clients
-
-  if (client) {
-    String currentLine = "";
-    while (client.connected()) {
-      if (client.available()) {
-        char c = client.read();
-        if (c == '\n') {
-          if (currentLine.length() == 0) {
-            client.println("HTTP/1.1 200 OK");
-            client.println("Content-type:text/html");
-            client.println();
-            client.print(responseHTML);
-            break;
-          } else {
-            currentLine = "";
-          }
-        } else if (c != '\r') {
-          currentLine += c;
-        }
-      }
-    }
-    client.stop();
-  }
+  server.handleClient();
+  delay(5);   // give CPU some idle time
 }