Add mdnsreset handling

lbussy · Nov 28, 2020 · e342a3b · e342a3b
1 parent 91279e7
commit e342a3b
Show file tree

Hide file tree

Showing 4 changed files with 115 additions and 8 deletions.
diff --git a/src/config.h b/src/config.h
@@ -44,6 +44,82 @@ SOFTWARE. */
 //
 //////////////////////////////////////////////////////////////////////////
 
+//////////////////////////////////////////////////////////////////////////
+//
+// Echo all serial output to telnet
+//
+#ifndef DOTELNET
+#define DOTELNET true
+#endif
+//
+#if DOTELNET == true
+#ifndef TELNETPORT
+#define TELNETPORT 23
+#endif
+#endif
+//
+//////////////////////////////////////////////////////////////////////////
+
+//////////////////////////////////////////////////////////////////////////
+//
+// Use SPIFFS Editor
+//
+#ifndef SPIFFSEDIT
+#define SPIFFSEDIT
+#endif
+//
+//////////////////////////////////////////////////////////////////////////
+
+//////////////////////////////////////////////////////////////////////////
+//
+// SPIFFS Editor Username
+//
+#ifndef SPIFFSEDITUSER
+#define SPIFFSEDITUSER "admin"
+#endif
+//
+//////////////////////////////////////////////////////////////////////////
+
+//////////////////////////////////////////////////////////////////////////
+//
+// SPIFFS Editor Username
+//
+#ifndef SPIFFSEDITPW
+#define SPIFFSEDITPW "p@ssword"
+#endif
+//
+//////////////////////////////////////////////////////////////////////////
+
+//////////////////////////////////////////////////////////////////////////
+//
+// Reset controller every 42 days
+//
+#ifndef ESPREPBOOT
+#define ESPREBOOT (uint32_t)3628800000
+#endif
+//
+//////////////////////////////////////////////////////////////////////////
+
+//////////////////////////////////////////////////////////////////////////
+//
+// Reset NTP every 24 hours
+//
+#ifndef NTPRESET
+#define NTPRESET (uint32_t)86400000
+#endif
+//
+//////////////////////////////////////////////////////////////////////////
+
+//////////////////////////////////////////////////////////////////////////
+//
+// Minumum Free Heap, triggers reset if below this value
+//
+#ifndef MINFREEHEAP
+#define MINFREEHEAP 10000
+#endif
+//
+//////////////////////////////////////////////////////////////////////////
+
 //////////////////////////////////////////////////////////////////////////
 //
 // Define API Key (identifies application to target)
@@ -271,7 +347,7 @@ SOFTWARE. */
 // Poll Brew Bubbles server for available version (time in seconds)
 //
 #ifndef POLLSERVERVERSION
-#define POLLSERVERVERSION 60
+#define POLLSERVERVERSION 3600
 #endif
 //
 //////////////////////////////////////////////////////////////////////////
@@ -310,8 +386,8 @@ SOFTWARE. */
 //
 // Port for local web services
 //
-#ifndef PORT
-#define PORT 80
+#ifndef HTTPPORT
+#define HTTPPORT 80
 #endif
 //
 //////////////////////////////////////////////////////////////////////////

diff --git a/src/mdns.cpp b/src/mdns.cpp
@@ -31,13 +31,43 @@ void mdnssetup()
     else
     {
         Log.notice(F("mDNS responder started for %s.local." CR), config.hostname);
-        if (!MDNS.addService("http", "tcp", PORT))
+        if (!MDNS.addService("http", "tcp", HTTPPORT))
         {
-            Log.error(F("Failed to register mDNS service." CR));
+            Log.error(F("Failed to register Web mDNS service." CR));
         }
         else
         {
-            Log.notice(F("HTTP registered via mDNS on port %i." CR), PORT);
+            Log.notice(F("HTTP registered via mDNS on port %i." CR), HTTPPORT);
         }
+        if (!MDNS.addService(config.hostname, "tcp", HTTPPORT))
+        {
+            Log.error(F("Failed to register %s mDNS service." CR), API_KEY);
+        }
+        else
+        {
+            Log.notice(F("%s registered via mDNS on port %i." CR), API_KEY, HTTPPORT);
+        }
+    }
+}
+
+void mdnsreset()
+{
+    MDNS.end();
+    if (!MDNS.begin(config.hostname))
+    {
+        Log.error(F("Error resetting MDNS responder."));
+    }
+    else
+    {
+#ifdef ESP32
+        Log.notice(F("mDNS responder restarted, hostname: %s.local." CR), WiFi.getHostname());
+#elif ESP8266
+        Log.notice(F("mDNS responder restarted, hostname: %s.local." CR), WiFi.hostname().c_str());
+#endif
+        MDNS.addService("http", "tcp", HTTPPORT);
+        MDNS.addService(config.hostname, "tcp", HTTPPORT);
+#if DOTELNET == true
+        MDNS.addService("telnet", "tcp", TELNETPORT);
+#endif
     }
 }
diff --git a/src/mdns.h b/src/mdns.h
@@ -34,6 +34,7 @@ SOFTWARE. */
 #include <ESP8266WiFi.h>
 
 void mdnssetup();
+void mdnsreset();
 
 extern struct Config config;
 

diff --git a/src/webserver.cpp b/src/webserver.cpp
@@ -22,7 +22,7 @@ SOFTWARE. */
 
 #include "webserver.h"
 
-AsyncWebServer server(PORT);
+AsyncWebServer server(HTTPPORT);
 
 void initWebServer()
 {
@@ -42,7 +42,7 @@ void initWebServer()
 
     server.begin();
 
-    Log.notice(F("Async HTTP server started on port %l." CR), PORT);
+    Log.notice(F("Async HTTP server started on port %l." CR), HTTPPORT);
     Log.verbose(F("Open: http://%s.local to view controller application." CR), WiFi.hostname().c_str());
 }