arduino
diff --git a/‎libraries/SBU/extras/SBUBoot/SBUBoot.ino
Copy file name to clipboardExpand all lines: libraries/SBU/extras/SBUBoot/SBUBoot.ino
+18-6Lines changed: 18 additions & 6 deletions b/‎libraries/SBU/extras/SBUBoot/SBUBoot.ino
Copy file name to clipboardExpand all lines: libraries/SBU/extras/SBUBoot/SBUBoot.ino
+18-6Lines changed: 18 additions & 6 deletions
@@ -29,7 +29,7 @@ static constexpr char CHECK_FILE_NAME[] = "UPDATE.OK";
 
 FlashClass mcu_flash;
 
-NBFileUtils  fileUtils;
+NBFileUtils  fileUtils(true);
 
 extern "C" void __libc_init_array(void);
 
@@ -51,30 +51,42 @@ int main()
   // Try to update only if update file
   // has been download successfully.
 
-  if (fileUtils.listFile(CHECK_FILE_NAME)) {
+  if (fileUtils.existFile(CHECK_FILE_NAME)) {
     Serial1.println("Update file exists");
     uint32_t updateSize = fileUtils.listFile(UPDATE_FILE_NAME);
-    size_t cycles = (updateSize / blockSize) + 1;
+    uint32_t tot_bytes = 0;
+    uint32_t read_bytes = 0;
 
     if (updateSize > SBU_SIZE) {
       updateSize = updateSize - SBU_SIZE - SBU_START;
+      size_t cycles = (updateSize / blockSize);
+      size_t spare_bytes = (updateSize % blockSize);
       /* Erase the MCU flash */
       uint32_t flash_address = (uint32_t)SKETCH_START;
       mcu_flash.erase((void*)flash_address, updateSize);
 
       for (auto i = 0; i < cycles; i++) {
         uint8_t block[blockSize] { 0 };
         digitalWrite(LED_BUILTIN, LOW);
-        uint32_t read_bytes = fileUtils.readBlock(UPDATE_FILE_NAME, (i * blockSize) + SBU_SIZE + SBU_START, blockSize, block);
+        read_bytes = fileUtils.readBlock(UPDATE_FILE_NAME, (i * blockSize) + SBU_SIZE + SBU_START, blockSize, block);
+        digitalWrite(LED_BUILTIN, HIGH);
+        mcu_flash.write((void*)flash_address, block, read_bytes);
+        flash_address += read_bytes;
+        tot_bytes += read_bytes;
+      }
+
+      if (spare_bytes){
+        uint8_t block[spare_bytes] { 0 };
+        digitalWrite(LED_BUILTIN, LOW);
+        read_bytes = fileUtils.readBlock(UPDATE_FILE_NAME, tot_bytes + SBU_SIZE + SBU_START, spare_bytes, block);
         digitalWrite(LED_BUILTIN, HIGH);
         mcu_flash.write((void*)flash_address, block, read_bytes);
         flash_address += read_bytes;
       }
       update_success = true;
     }
     if (update_success) {
-      fileUtils.deleteFile(UPDATE_FILE_NAME);
-      fileUtils.deleteFile(CHECK_FILE_NAME);
+      fileUtils.deleteFiles();
     }
   }
   else {