minor enchancement

Author: John Doe

hardware/esp8266com/esp8266/cores/esp8266/Updater.cpp

@@ -106,6 +106,7 @@ bool UpdaterClass::_writeBuffer(){
   interrupts();
   if (rc) {
     _error = UPDATE_ERROR_WRITE;
+    _currentAddress = (_startAddress + _size);
 #ifdef DEBUG_UPDATER
     printError(DEBUG_UPDATER);
 #endif
@@ -157,6 +158,7 @@ size_t UpdaterClass::writeStream(Stream &data){
     toRead = data.readBytes(_buffer + _bufferLen, toRead);
     if(toRead == 0){ //Timeout
       _error = UPDATE_ERROR_STREAM;
+      _currentAddress = (_startAddress + _size);
 #ifdef DEBUG_UPDATER
       printError(DEBUG_UPDATER);
 #endif

hardware/esp8266com/esp8266/cores/esp8266/Updater.h

@@ -57,10 +57,10 @@ class UpdaterClass {
     void clearError(){ _error = UPDATE_ERROR_OK; }
     bool hasError(){ return _error != UPDATE_ERROR_OK; }
     bool isRunning(){ return _size > 0; }
-    bool isFinished(){ return hasError()?true:(_currentAddress == (_startAddress + _size)); }
+    bool isFinished(){ return _currentAddress == (_startAddress + _size); }
     size_t size(){ return _size; }
     size_t progress(){ return _currentAddress - _startAddress; }
-    size_t remaining(){ return hasError()?0:(size() - progress()); }
+    size_t remaining(){ return _size - (_currentAddress - _startAddress); }
 
     /*
       Template to write from objects that expose
@@ -116,4 +116,4 @@ class UpdaterClass {
 
 extern UpdaterClass Update;
 
-#endif
+#endif