QSPIFormat: add function to restore memory mapped firmware

Author: pennam

libraries/STM32H747_System/examples/QSPIFormat/QSPIFormat.ino

@@ -80,7 +80,8 @@ void setup() {
 
     Serial.println("Do you want to perform a full erase of the QSPI flash before proceeding? Y/[n]");
     Serial.println("Note: Full flash erase can take up to one minute.");
-    if (true == waitResponse()) {
+    bool fullErase = waitResponse();
+    if (fullErase == true) {
       Serial.println("Full erase started, please wait...");
       root->erase(0x0, root->size());
       Serial.println("Full erase completed.");
@@ -109,7 +110,7 @@ void setup() {
     }
 
     bool restore = true;
-    if (reformat) {
+    if (reformat || fullErase) {
       Serial.println("\nDo you want to restore the WiFi firmware and certificates? Y/[n]");
       restore = waitResponse();
     }
@@ -118,6 +119,10 @@ void setup() {
       flashWiFiFirmwareAndCertificates();
     }
 
+    if (fullErase && restore) {
+      flashWiFiFirmwareMapped();
+    }
+
     reformat = true;
     if (!ota_data_fs.mount(&ota_data)) {
       Serial.println("\nPartition 2 already contains a filesystem, do you want to reformat it? Y/[n]");
@@ -161,10 +166,11 @@ void setup() {
   Serial.println("It's now safe to reboot or disconnect your board.");
 }
 
+const uint32_t file_size = 421098;
+extern const unsigned char wifi_firmware_image_data[];
+
 void flashWiFiFirmwareAndCertificates() {
-  extern const unsigned char wifi_firmware_image_data[];
   FILE* fp = fopen("/wlan/4343WA1.BIN", "wb");
-  const uint32_t file_size = 421098;
   uint32_t chunck_size = 1024;
   uint32_t byte_count = 0;
 
@@ -203,6 +209,26 @@ void flashWiFiFirmwareAndCertificates() {
   fclose(fp);
 }
 
+void flashWiFiFirmwareMapped() {
+  uint32_t chunck_size = 1024;
+  uint32_t byte_count = 0;
+  const uint32_t offset = 15 * 1024 * 1024 + 1024 * 512;
+
+  Serial.println("Flashing memory mapped WiFi firmware");
+  printProgress(byte_count, file_size, 10, true);
+  while (byte_count < file_size) {
+    if (byte_count + chunck_size > file_size)
+      chunck_size = file_size - byte_count;
+    int ret = root->program(wifi_firmware_image_data, offset + byte_count, chunck_size);
+    if (ret != 0) {
+      Serial.println("Error writing memory mapped firmware");
+      break;
+    }
+    byte_count += chunck_size;
+    printProgress(byte_count, file_size, 10, false);
+  }
+}
+
 void loop() {
 
 }