Refactor the updaters back into one sketch

Author: sebromero

libraries/Nano33_System/examples/Nano33_updateBootloader/MBR.h renamed to libraries/Nano33_System/examples/Nano33_updateBLandSoftDevice/MBR.h

@@ -0,0 +1,216 @@
+/*
+ *  This sketch allows to support Soft Devices on Nano 33 BLE.
+ *  
+ *  To be able to support Soft Devices, the bootloader first needs to be updated.
+ *  Upload this sketch on the Nano 33 BLE to download the new bootloader in the flash. 
+ *  After flashing the bootloader the sketch asks if you want to upload a SoftDevice.
+ *  This is required for the OpenMV firmware to work.
+ *  
+ *  The new bootloader is fully backwards compatible with standard sketches.
+ *
+ */
+
+#include "FlashIAP.h"
+#include "MBR.h"
+#include "bootloader.h"
+#include "nrf_nvmc.h"
+#include "SoftDevice.h"
+
+#define SOFTDEVICE_ADDR       (0xA0000)
+#define SOFTDEVICE_INFO_ADDR  (0xFF000)
+#define MBR_ADDR              (0x0)
+#define BOOTLOADER_ADDR       (0xE0000)
+#define UICR_BOOT_ADDR        (0x10001014)
+
+#define BOOTLOADER_CRC         0x5E797B91
+#define BOOTLOADER_SIZE        nano33_bootloader_hex_len
+const unsigned int magic = 0x5f27a93d;
+
+mbed::FlashIAP flash;
+
+bool hasLatestBootloader(){
+  //Compute bootloader CRC32
+  uint32_t crc32 = getCurrentBootloaderCrc();
+  return crc32 == BOOTLOADER_CRC;
+}
+
+bool getUserConfirmation(){
+  while (true){
+    if (Serial.available()){
+      char choice = Serial.read();
+      switch (choice){
+        case 'y':
+        case 'Y':        
+          return true;
+        case 'n':
+        case 'N':
+          return false;
+        default:
+          continue;
+      }
+    }
+  }
+}
+
+void applyUpdate(uint32_t address, const unsigned char payload[], long len, uint32_t bin_pointer = 0) {
+  uint32_t flash_pointer = 0;
+  const uint32_t page_size = flash.get_page_size();
+  char *page_buffer = new char[page_size];
+  uint32_t addr = address;
+
+  uint32_t sector_size = flash.get_sector_size(addr);
+  uint32_t next_sector = addr + sector_size;
+  bool sector_erased = false;
+  size_t pages_flashed = 0;
+  uint32_t percent_done = 0;
+
+  while (true) {
+    
+    if (flash_pointer >= len) {
+      break;
+    }
+
+    flash.erase(addr + flash_pointer, sector_size);
+
+    if ((len - flash_pointer) < sector_size) {
+      sector_size = len - flash_pointer;
+    }
+
+    // Program page
+    flash.program(&payload[bin_pointer], addr + flash_pointer, sector_size);
+    Serial.print("Flash Address = ");
+    Serial.println(addr + flash_pointer, HEX);
+    
+    bin_pointer = bin_pointer + sector_size;
+    flash_pointer = flash_pointer + sector_size;
+
+    uint32_t percent_done = flash_pointer * 100 / len;
+    Serial.println("Flashed " + String(percent_done) + "%");
+  }
+  Serial.println();
+
+  delete[] page_buffer;
+}
+
+void updateBootloader(){
+  Serial.println("This sketch modifies the Nano33 bootloader to support Soft Devices.");
+  Serial.println();
+  
+  flash.init();
+  
+  Serial.println("Flasing MBR...");
+  applyUpdate(MBR_ADDR, MBR_bin, MBR_bin_len);
+
+  Serial.println("Flasing bootloader...");
+  applyUpdate(BOOTLOADER_ADDR, nano33_bootloader_hex, nano33_bootloader_hex_len);
+
+  Serial.print("Bootloader 32bit CRC is: ");
+  uint32_t crc32 = getTargetBootloaderCrc();
+  Serial.println(crc32, HEX);
+  
+  Serial.println("Writing in UICR memory the address of the new bootloader...");
+  nrf_nvmc_write_word(UICR_BOOT_ADDR, BOOTLOADER_ADDR);
+  
+  flash.deinit();
+
+  Serial.println();
+  Serial.println("Bootloader update successfully completed!");
+}
+
+void updateSoftDevice(){
+  flash.init();
+
+  Serial.println("Correct booloader found!");
+  Serial.println();
+  
+  Serial.println("Storing SoftDevice length info at 0xFF000...");
+  writeSoftDeviceLen(SOFTDEVICE_INFO_ADDR);
+  
+  Serial.println("Flasing SoftDevice...");
+  applyUpdate(SOFTDEVICE_ADDR, Softdevice_bin, Softdevice_bin_len - 4096, 4096);
+
+  flash.deinit();
+
+  Serial.println();
+  Serial.println("SoftDevice update complete! The board is restarting...");
+  NVIC_SystemReset();
+}
+
+void setup() {  
+  Serial.begin(115200);
+  while (!Serial) {}
+
+  if(!hasLatestBootloader()){
+    Serial.println("Your bootloader version is outdated (update required for Soft Device support).");
+    Serial.println("Would you like to update it? Y/N");
+    
+    if(getUserConfirmation()){
+      updateBootloader();
+    }
+  }
+  
+  if(hasLatestBootloader()){
+    Serial.println("Would you like to install the Soft Device (required for OpenMV)? Y/N");
+    if(getUserConfirmation()){
+      updateSoftDevice();
+    }
+  }
+}
+
+uint32_t getTargetBootloaderCrc() {
+  uint32_t mask = 0;
+  uint32_t crc = 0xFFFFFFFF;
+  uint32_t b = 0;
+  uint8_t bootByte = 0;
+
+  int iterations = BOOTLOADER_SIZE;
+
+  for (int i=0; i<iterations; i=i+4) {
+    b = 0;
+    for (int j=0; j<4; j++) {
+      mask = 0;
+      bootByte = nano33_bootloader_hex[i+j];
+      mask = mask + (uint32_t)bootByte;
+      mask = mask << 8*j;
+      b = b | mask;
+    }
+    crc = crc ^ b;
+  }
+  return crc;
+}
+
+uint32_t getCurrentBootloaderCrc() {
+  uint32_t b = 0;
+  uint32_t crc = 0xFFFFFFFF;
+
+  int iterations = ceil(BOOTLOADER_SIZE/4);
+
+  int addr = BOOTLOADER_ADDR;
+
+  for (int i=0; i<iterations; i++) {
+    //Read 32 bit from flash
+    flash.read(&b, addr, sizeof(b));
+    //Serial.println(b, HEX);
+    //Update crc
+    crc = crc ^ b;
+    //Update pointer 
+    addr = addr + 4;
+  }
+  return crc;
+}
+
+void writeSoftDeviceLen(uint32_t address) {
+  uint32_t sd_addr = SOFTDEVICE_ADDR;
+  flash.erase(address, 16);
+  //Write flag to let Bootloader understand that SoftDevice binary must be moved
+  flash.program(&magic, address, 4);
+  //Write address where the SoftDevice binary has been written
+  flash.program(&sd_addr, address + 4, 4);
+  //Write SoftDevice binary length
+  unsigned int sd_len = Softdevice_bin_len - 4096;
+  flash.program(&sd_len, address + 8, 4);
+}
+
+void loop() {
+  delay(1000);
+}