From 9db1d5a1a4a341ae6c9c264a97c318eaa1f6c659 Mon Sep 17 00:00:00 2001
From: David Refoua <David@Refoua.me>
Date: Sun, 17 Mar 2024 18:23:26 +0330
Subject: [PATCH 1/3] add missing update error codes

---
 cores/esp8266/Updater.h | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/cores/esp8266/Updater.h b/cores/esp8266/Updater.h
index ad652d3806..dd2539aa65 100644
--- a/cores/esp8266/Updater.h
+++ b/cores/esp8266/Updater.h
@@ -21,6 +21,8 @@
 #define UPDATE_ERROR_SIGN               (12)
 #define UPDATE_ERROR_NO_DATA            (13)
 #define UPDATE_ERROR_OOM                (14)
+#define UPDATE_ERROR_RUNNING_ALREADY    (15)
+#define UPDATE_ERROR_UNKNOWN_COMMAND    (16)
 
 #define U_FLASH   0
 #define U_FS      100

From ef5c85633ef252692e5fd071b48570ef55eae851 Mon Sep 17 00:00:00 2001
From: David Refoua <David@Refoua.me>
Date: Sun, 17 Mar 2024 18:26:31 +0330
Subject: [PATCH 2/3] add missing error codes for Update.begin()

---
 cores/esp8266/Updater.cpp | 14 ++++++++------
 1 file changed, 8 insertions(+), 6 deletions(-)

diff --git a/cores/esp8266/Updater.cpp b/cores/esp8266/Updater.cpp
index 91f2a9e6ae..7d721609bf 100644
--- a/cores/esp8266/Updater.cpp
+++ b/cores/esp8266/Updater.cpp
@@ -70,6 +70,7 @@ bool UpdaterClass::begin(size_t size, int command, int ledPin, uint8_t ledOn) {
 #ifdef DEBUG_UPDATER
     DEBUG_UPDATER.println(F("[begin] already running"));
 #endif
+    _setError(UPDATE_ERROR_RUNNING_ALREADY);
     return false;
   }
 
@@ -86,7 +87,7 @@ bool UpdaterClass::begin(size_t size, int command, int ledPin, uint8_t ledOn) {
     _setError(UPDATE_ERROR_BOOTSTRAP);
     return false;
   }
-  
+
 #ifdef DEBUG_UPDATER
   if (command == U_FS) {
     DEBUG_UPDATER.println(F("[begin] Update Filesystem."));
@@ -133,7 +134,7 @@ bool UpdaterClass::begin(size_t size, int command, int ledPin, uint8_t ledOn) {
 
     //make sure that the size of both sketches is less than the total space (updateEndAddress)
     if(updateStartAddress < currentSketchSize) {
-      _setError(UPDATE_ERROR_SPACE);    
+      _setError(UPDATE_ERROR_SPACE);
       return false;
     }
   }
@@ -162,6 +163,7 @@ bool UpdaterClass::begin(size_t size, int command, int ledPin, uint8_t ledOn) {
 #ifdef DEBUG_UPDATER
     DEBUG_UPDATER.println(F("[begin] Unknown update command."));
 #endif
+    _setError(UPDATE_ERROR_UNKNOWN_COMMAND);
     return false;
   }
 
@@ -404,7 +406,7 @@ bool UpdaterClass::_writeBuffer(){
       modifyFlashMode = true;
     }
   }
-  
+
   if (eraseResult) {
     if(!_async) yield();
     writeResult = ESP.flashWrite(_currentAddress, _buffer, _bufferLen);
@@ -488,7 +490,7 @@ bool UpdaterClass::_verifyEnd() {
         uint8_t buf[4] __attribute__((aligned(4)));
         if(!ESP.flashRead(_startAddress, (uint32_t *) &buf[0], 4)) {
             _currentAddress = (_startAddress);
-            _setError(UPDATE_ERROR_READ);            
+            _setError(UPDATE_ERROR_READ);
             return false;
         }
 
@@ -500,7 +502,7 @@ bool UpdaterClass::_verifyEnd() {
             return true;
         } else if (buf[0] != 0xE9) {
             _currentAddress = (_startAddress);
-            _setError(UPDATE_ERROR_MAGIC_BYTE);            
+            _setError(UPDATE_ERROR_MAGIC_BYTE);
             return false;
         }
 
@@ -512,7 +514,7 @@ bool UpdaterClass::_verifyEnd() {
         // check if new bin fits to SPI flash
         if(bin_flash_size > ESP.getFlashChipRealSize()) {
             _currentAddress = (_startAddress);
-            _setError(UPDATE_ERROR_NEW_FLASH_CONFIG);            
+            _setError(UPDATE_ERROR_NEW_FLASH_CONFIG);
             return false;
         }
 #endif

From 0116f9c729a03dc52ad743cbf9e12c497154fd09 Mon Sep 17 00:00:00 2001
From: David Refoua <David@Refoua.me>
Date: Sun, 17 Mar 2024 18:52:40 +0000
Subject: [PATCH 3/3] add the missing getErrorString() cases

---
 cores/esp8266/Updater.cpp | 6 ++++++
 cores/esp8266/Updater.h   | 6 +++---
 2 files changed, 9 insertions(+), 3 deletions(-)

diff --git a/cores/esp8266/Updater.cpp b/cores/esp8266/Updater.cpp
index 7d721609bf..b4961b616d 100644
--- a/cores/esp8266/Updater.cpp
+++ b/cores/esp8266/Updater.cpp
@@ -651,6 +651,12 @@ String UpdaterClass::getErrorString() const {
   case UPDATE_ERROR_OOM:
     out = F("Out of memory");
     break;
+  case UPDATE_ERROR_RUNNING_ALREADY:
+    out = F("Update already running");
+    break;
+  case UPDATE_ERROR_UNKNOWN_COMMAND:
+    out = F("Unknown update command");
+    break;
   default:
     out = F("UNKNOWN");
     break;
diff --git a/cores/esp8266/Updater.h b/cores/esp8266/Updater.h
index dd2539aa65..7ee1d28311 100644
--- a/cores/esp8266/Updater.h
+++ b/cores/esp8266/Updater.h
@@ -57,7 +57,7 @@ class UpdaterClass {
     using THandlerFunction_Progress = std::function<void(size_t, size_t)>;
     using THandlerFunction_Error = std::function<void(uint8_t)>;
     using THandlerFunction = std::function<void()>;
-  
+
     UpdaterClass();
     ~UpdaterClass();
 
@@ -71,7 +71,7 @@ class UpdaterClass {
     bool begin(size_t size, int command = U_FLASH, int ledPin = -1, uint8_t ledOn = LOW);
 
     /*
-      Run Updater from asynchronous callbacs
+      Run Updater from asynchronous callbacks
     */
     void runAsync(bool async){ _async = async; }
 
@@ -218,7 +218,7 @@ class UpdaterClass {
     bool _verifyHeader(uint8_t data);
     bool _verifyEnd();
 
-    void _setError(int error);    
+    void _setError(int error);
 
     bool _async = false;
     uint8_t _error = 0;