src/main/fc/firmware_update_common.c

   1
   2 #include <stdbool.h>
   3 #include <stdint.h>
   4
   5 #include "platform.h"
   6
   7 #include "common/log.h"
   8
   9 #include "drivers/flash.h"
  10
  11 #include "fc/firmware_update_common.h"
  12
  13 #include "io/asyncfatfs/asyncfatfs.h"
  14
  15 #ifdef MSP_FIRMWARE_UPDATE
  16
  17 #if !(defined(USE_FLASHFS) || defined(USE_SDCARD))
  18 #error No storage backend available
  19 #endif
  20
  21 #ifdef USE_SDCARD
  22 static afatfsFilePtr_t metaFile = NULL;
  23
  24 static void metaFileOpenCallback(afatfsFilePtr_t file)
  25 {
  26     metaFile = file;
  27 }
  28 #endif
  29
  30 bool firmwareUpdateMetadataRead(firmwareUpdateMetadata_t *updateMetadata)
  31 {
  32 #if defined(USE_SDCARD)
  33     if ((afatfs_getFilesystemState() != AFATFS_FILESYSTEM_STATE_READY)
  34             || !afatfs_fopen(FIRMWARE_UPDATE_META_FILENAME, "r", metaFileOpenCallback)
  35             || (afatfs_freadSync(metaFile, (uint8_t *)updateMetadata, sizeof(*updateMetadata)) != sizeof(*updateMetadata))) {
  36         return false;
  37     }
  38
  39 #elif defined(USE_FLASHFS)
  40     flashPartition_t *flashPartition = flashPartitionFindByType(FLASH_PARTITION_TYPE_FIRMWARE_UPDATE_META);
  41     if (!flashPartition) return false;
  42
  43     const flashGeometry_t *flashGeometry = flashGetGeometry();
  44     uint32_t flashAddress = flashPartition->startSector * flashGeometry->sectorSize;
  45
  46     if (!flashReadBytes(flashAddress, (uint8_t *)updateMetadata, sizeof(*updateMetadata))) {
  47         return false;
  48     }
  49
  50 #endif
  51
  52     return true;
  53 }
  54
  55 bool firmwareUpdateMetadataWrite(firmwareUpdateMetadata_t *updateMetadata)
  56 {
  57     updateMetadata->magic = FIRMWARE_UPDATE_METADATA_MAGIC;
  58
  59 #if defined(USE_SDCARD)
  60     if ((afatfs_getFilesystemState() != AFATFS_FILESYSTEM_STATE_READY)
  61             || !afatfs_fopen(FIRMWARE_UPDATE_META_FILENAME, "w+", metaFileOpenCallback)
  62             || (afatfs_fwriteSync(metaFile, (uint8_t *)updateMetadata, sizeof(*updateMetadata)) != sizeof(*updateMetadata))) {
  63         return false;
  64     }
  65
  66     afatfs_fcloseSync(metaFile);
  67
  68 #elif defined(USE_FLASHFS)
  69     flashPartition_t *flashPartition = flashPartitionFindByType(FLASH_PARTITION_TYPE_FIRMWARE_UPDATE_META);
  70     if (!flashPartition) return false;
  71
  72     const flashGeometry_t *flashGeometry = flashGetGeometry();
  73     const uint32_t flashPartitionSize = (flashPartition->endSector - flashPartition->startSector + 1) * flashGeometry->sectorSize;
  74     uint32_t flashAddress = flashPartition->startSector * flashGeometry->sectorSize;
  75
  76     if (flashPartitionSize < sizeof(*updateMetadata)) return false;
  77
  78     flashEraseSector(flashAddress);
  79     flashWaitForReady(1000);
  80     flashPageProgram(flashAddress, (uint8_t *)updateMetadata, sizeof(*updateMetadata));
  81
  82 #endif
  83
  84     return true;
  85 }
  86
  87 #endif