maintenance mode (improved error handling)

src/boot.cpp

@@ -11,6 +11,7 @@ void start_boot_menu(void) {
 
   uint8_t mac[6];
   char clientId[20];
+  unsigned long timeout = millis();
 
   // hash 6 byte MAC to 4 byte hash
   esp_eth_get_mac(mac);
@@ -21,13 +22,15 @@ void start_boot_menu(void) {
   const char *ssid = WIFI_SSID;
   const char *password = WIFI_PASS;
 
+  // set runmode normal makes watchdog booting to production if triggered
   RTC_runmode = RUNMODE_NORMAL;
 
+  // setup watchdog, based on esp32 timer2 interrupt
   hw_timer_t *timer = NULL;
   timer = timerBegin(2, 80, true);                 // timer 2, div 80, countup
-  timerAttachInterrupt(timer, &esp_restart, true); // callback device reset
-  timerAlarmWrite(timer, BOOTDELAY * 1000000, false); // set time in us
-  timerAlarmEnable(timer);                            // enable interrupt
+  timerAttachInterrupt(timer, &esp_restart, true); // callback for device reset
+  timerAlarmWrite(timer, BOOTTIMEOUT * 1000000, false); // set time in us
+  timerAlarmEnable(timer);                              // enable watchdog
 
   WebServer server(80);
 
@@ -102,53 +105,57 @@ void start_boot_menu(void) {
   // Connect to WiFi network
   // workaround applied here to avoid WIFI_AUTH failure
   // see https://github.com/espressif/arduino-esp32/issues/2501
+
   // 1st try
   WiFi.begin(ssid, password);
   while (WiFi.status() == WL_DISCONNECTED) {
+    if ((long)(millis() - timeout) > (BOOTDELAY * 1000))
+      esp_restart();
+    else
       delay(500);
   }
   // 2nd try
   if (WiFi.status() != WL_CONNECTED) {
     WiFi.begin(ssid, password);
     while (WiFi.status() != WL_CONNECTED) {
+      if ((long)(millis() - timeout) > (BOOTDELAY * 1000))
+        esp_restart();
+      else
         delay(500);
     }
   }
 
   MDNS.begin(host);
+  timerWrite(timer, 0); // reset timer (feed watchdog)
 
   server.on("/", HTTP_GET, [&server, &loginMenu]() {
-    server.sendHeader("Connection", "close");
+    server.sendHeader("Connection", "keep-alive");
     server.send(200, "text/html", loginMenu);
   });
 
   server.on("/serverIndex", HTTP_GET, [&server, &serverIndex, &timer]() {
-    timerAlarmWrite(timer, BOOTTIMEOUT * 1000000, false);
-    server.sendHeader("Connection", "close");
+    timerWrite(timer, 0); // reset timer (feed watchdog)
+    server.sendHeader("Connection", "keep-alive");
     server.send(200, "text/html", serverIndex);
   });
 
-  server.onNotFound([&server, &loginMenu]() {
-    server.sendHeader("Connection", "close");
-    server.send(200, "text/html", loginMenu);
-  });
-
-  // handling uploading firmware file
   server.on(
       "/update", HTTP_POST,
       [&server]() {
         server.sendHeader("Connection", "close");
         server.send(200, "text/plain", (Update.hasError()) ? "FAIL" : "OK");
         WiFi.disconnect(true);
-        if (!Update.hasError())
-          RTC_runmode = RUNMODE_POWERCYCLE;
         esp_restart();
       },
 
+      // handling uploading firmware file
       [&server, &timer]() {
         bool success = false;
         HTTPUpload &upload = server.upload();
 
+        // did we get a file name?
+        if (upload.filename != NULL) {
+
           switch (upload.status) {
 
           case UPLOAD_FILE_START:
@@ -178,10 +185,11 @@ void start_boot_menu(void) {
 
           } // switch
 
+          // don't boot to production if update failed
           if (!success) {
             ESP_LOGE(TAG, "Error: %s", Update.errorString());
-          WiFi.disconnect(true);
-          esp_restart();
+            RTC_runmode = RUNMODE_POWERCYCLE;
+          }
         }
       });
 
@@ -194,6 +202,6 @@ void start_boot_menu(void) {
 
   while (1) {
     server.handleClient();
-    delay(1);
+    delay(2);
   }
 }