310 lines
9.3 KiB
C
310 lines
9.3 KiB
C
|
|
#include "uf2.h"
|
|
#include "flash_nrf5x.h"
|
|
#include <string.h>
|
|
|
|
#include "boards.h"
|
|
#include "tusb.h"
|
|
|
|
#include "bootloader_settings.h"
|
|
#include "bootloader.h"
|
|
|
|
typedef struct {
|
|
uint8_t JumpInstruction[3];
|
|
uint8_t OEMInfo[8];
|
|
uint16_t SectorSize;
|
|
uint8_t SectorsPerCluster;
|
|
uint16_t ReservedSectors;
|
|
uint8_t FATCopies;
|
|
uint16_t RootDirectoryEntries;
|
|
uint16_t TotalSectors16;
|
|
uint8_t MediaDescriptor;
|
|
uint16_t SectorsPerFAT;
|
|
uint16_t SectorsPerTrack;
|
|
uint16_t Heads;
|
|
uint32_t HiddenSectors;
|
|
uint32_t TotalSectors32;
|
|
uint8_t PhysicalDriveNum;
|
|
uint8_t Reserved;
|
|
uint8_t ExtendedBootSig;
|
|
uint32_t VolumeSerialNumber;
|
|
uint8_t VolumeLabel[11];
|
|
uint8_t FilesystemIdentifier[8];
|
|
} __attribute__((packed)) FAT_BootBlock;
|
|
|
|
typedef struct {
|
|
char name[8];
|
|
char ext[3];
|
|
uint8_t attrs;
|
|
uint8_t reserved;
|
|
uint8_t createTimeFine;
|
|
uint16_t createTime;
|
|
uint16_t createDate;
|
|
uint16_t lastAccessDate;
|
|
uint16_t highStartCluster;
|
|
uint16_t updateTime;
|
|
uint16_t updateDate;
|
|
uint16_t startCluster;
|
|
uint32_t size;
|
|
} __attribute__((packed)) DirEntry;
|
|
|
|
STATIC_ASSERT(sizeof(DirEntry) == 32);
|
|
|
|
struct TextFile {
|
|
char const name[11];
|
|
char const *content;
|
|
};
|
|
|
|
#define NUM_FAT_BLOCKS UF2_NUM_BLOCKS
|
|
|
|
#define STR0(x) #x
|
|
#define STR(x) STR0(x)
|
|
const char infoUf2File[] = //
|
|
"UF2 Bootloader " UF2_VERSION "\r\n"
|
|
"Model: " PRODUCT_NAME "\r\n"
|
|
"Board-ID: " BOARD_ID "\r\n"
|
|
"Bootloader: " BOOTLOADER_ID "\r\n"
|
|
"Date: " __DATE__ "\r\n";
|
|
|
|
const char indexFile[] = //
|
|
"<!doctype html>\n"
|
|
"<html>"
|
|
"<body>"
|
|
"<script>\n"
|
|
"location.replace(\"" INDEX_URL "\");\n"
|
|
"</script>"
|
|
"</body>"
|
|
"</html>\n";
|
|
|
|
static struct TextFile const info[] = {
|
|
{.name = "INFO_UF2TXT", .content = infoUf2File},
|
|
{.name = "INDEX HTM", .content = indexFile},
|
|
{.name = "CURRENT UF2"},
|
|
};
|
|
#define NUM_INFO (sizeof(info) / sizeof(info[0]))
|
|
|
|
#define UF2_SIZE (current_flash_size() * 2)
|
|
#define UF2_SECTORS (UF2_SIZE / 512)
|
|
#define UF2_FIRST_SECTOR (NUM_INFO + 1)
|
|
#define UF2_LAST_SECTOR (UF2_FIRST_SECTOR + UF2_SECTORS - 1)
|
|
|
|
#define RESERVED_SECTORS 1
|
|
#define ROOT_DIR_SECTORS 4
|
|
#define SECTORS_PER_FAT ((NUM_FAT_BLOCKS * 2 + 511) / 512)
|
|
|
|
#define START_FAT0 RESERVED_SECTORS
|
|
#define START_FAT1 (START_FAT0 + SECTORS_PER_FAT)
|
|
#define START_ROOTDIR (START_FAT1 + SECTORS_PER_FAT)
|
|
#define START_CLUSTERS (START_ROOTDIR + ROOT_DIR_SECTORS)
|
|
|
|
static FAT_BootBlock const BootBlock = {
|
|
.JumpInstruction = {0xeb, 0x3c, 0x90},
|
|
.OEMInfo = "UF2 UF2 ",
|
|
.SectorSize = 512,
|
|
.SectorsPerCluster = 1,
|
|
.ReservedSectors = RESERVED_SECTORS,
|
|
.FATCopies = 2,
|
|
.RootDirectoryEntries = (ROOT_DIR_SECTORS * 512 / 32),
|
|
.TotalSectors16 = NUM_FAT_BLOCKS - 2,
|
|
.MediaDescriptor = 0xF8,
|
|
.SectorsPerFAT = SECTORS_PER_FAT,
|
|
.SectorsPerTrack = 1,
|
|
.Heads = 1,
|
|
.ExtendedBootSig = 0x29,
|
|
.VolumeSerialNumber = 0x00420042,
|
|
.VolumeLabel = VOLUME_LABEL,
|
|
.FilesystemIdentifier = "FAT16 ",
|
|
};
|
|
|
|
#define NRF_LOG_DEBUG(...)
|
|
#define NRF_LOG_WARNING(...)
|
|
|
|
// get current.uf2 flash size in bytes, round up to 256 bytes
|
|
static uint32_t current_flash_size(void)
|
|
{
|
|
static uint32_t flash_sz = 0;
|
|
|
|
// only need to compute once
|
|
if ( flash_sz == 0 )
|
|
{
|
|
// return 1 block of 256 bytes
|
|
if ( !bootloader_app_is_valid(DFU_BANK_0_REGION_START) )
|
|
{
|
|
flash_sz = 256;
|
|
}else
|
|
{
|
|
bootloader_settings_t const * boot_setting;
|
|
bootloader_util_settings_get(&boot_setting);
|
|
|
|
flash_sz = boot_setting->bank_0_size;
|
|
|
|
// Copy size must be multiple of 256 bytes
|
|
// else we will got an issue copying current.uf2
|
|
if (flash_sz & 0xff)
|
|
{
|
|
flash_sz = (flash_sz & ~0xff) + 256;
|
|
}
|
|
|
|
// if bank0 size is not valid, happens when flashed with jlink
|
|
// use maximum application size
|
|
if ( (flash_sz == 0) || (flash_sz == 0xFFFFFFFFUL) )
|
|
{
|
|
flash_sz = FLASH_SIZE;
|
|
}
|
|
}
|
|
}
|
|
|
|
return flash_sz;
|
|
}
|
|
|
|
void padded_memcpy (char *dst, char const *src, int len)
|
|
{
|
|
for (int i = 0; i < len; ++i) {
|
|
if (*src)
|
|
*dst = *src++;
|
|
else
|
|
*dst = ' ';
|
|
dst++;
|
|
}
|
|
}
|
|
|
|
|
|
/*------------------------------------------------------------------*/
|
|
/* Read
|
|
*------------------------------------------------------------------*/
|
|
void read_block(uint32_t block_no, uint8_t *data) {
|
|
memset(data, 0, 512);
|
|
uint32_t sectionIdx = block_no;
|
|
|
|
if (block_no == 0) {
|
|
memcpy(data, &BootBlock, sizeof(BootBlock));
|
|
data[510] = 0x55;
|
|
data[511] = 0xaa;
|
|
// logval("data[0]", data[0]);
|
|
} else if (block_no < START_ROOTDIR) {
|
|
sectionIdx -= START_FAT0;
|
|
// logval("sidx", sectionIdx);
|
|
if (sectionIdx >= SECTORS_PER_FAT)
|
|
sectionIdx -= SECTORS_PER_FAT;
|
|
if (sectionIdx == 0) {
|
|
data[0] = 0xf0;
|
|
for (int i = 1; i < NUM_INFO * 2 + 4; ++i) {
|
|
data[i] = 0xff;
|
|
}
|
|
}
|
|
for (int i = 0; i < 256; ++i) {
|
|
uint32_t v = sectionIdx * 256 + i;
|
|
if (UF2_FIRST_SECTOR <= v && v <= UF2_LAST_SECTOR)
|
|
((uint16_t *)(void *)data)[i] = v == UF2_LAST_SECTOR ? 0xffff : v + 1;
|
|
}
|
|
} else if (block_no < START_CLUSTERS) {
|
|
sectionIdx -= START_ROOTDIR;
|
|
if (sectionIdx == 0) {
|
|
DirEntry *d = (void *)data;
|
|
padded_memcpy(d->name, (char const *) BootBlock.VolumeLabel, 11);
|
|
d->attrs = 0x28;
|
|
for (int i = 0; i < NUM_INFO; ++i) {
|
|
d++;
|
|
struct TextFile const *inf = &info[i];
|
|
d->size = inf->content ? strlen(inf->content) : UF2_SIZE;
|
|
d->startCluster = i + 2;
|
|
padded_memcpy(d->name, inf->name, 11);
|
|
}
|
|
}
|
|
} else {
|
|
sectionIdx -= START_CLUSTERS;
|
|
if (sectionIdx < NUM_INFO - 1) {
|
|
memcpy(data, info[sectionIdx].content, strlen(info[sectionIdx].content));
|
|
} else {
|
|
sectionIdx -= NUM_INFO - 1;
|
|
uint32_t addr = USER_FLASH_START + sectionIdx * 256;
|
|
if (addr < USER_FLASH_START+FLASH_SIZE) {
|
|
UF2_Block *bl = (void *)data;
|
|
bl->magicStart0 = UF2_MAGIC_START0;
|
|
bl->magicStart1 = UF2_MAGIC_START1;
|
|
bl->magicEnd = UF2_MAGIC_END;
|
|
bl->blockNo = sectionIdx;
|
|
bl->numBlocks = current_flash_size() / 256;
|
|
bl->targetAddr = addr;
|
|
bl->payloadSize = 256;
|
|
bl->flags = UF2_FLAG_FAMILYID;
|
|
bl->familyID = UF2_FAMILY_ID;
|
|
memcpy(bl->data, (void *)addr, bl->payloadSize);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
/*------------------------------------------------------------------*/
|
|
/* Write UF2
|
|
*------------------------------------------------------------------*/
|
|
|
|
/** Write an block
|
|
*
|
|
* @return number of bytes processed, only 3 following values
|
|
* -1 : if not an uf2 block
|
|
* 512 : write is successful
|
|
* 0 : is busy with flashing, tinyusb stack will call write_block again with the same parameters later on
|
|
*/
|
|
int write_block(uint32_t block_no, uint8_t *data, bool quiet, WriteState *state) {
|
|
UF2_Block *bl = (void *)data;
|
|
|
|
NRF_LOG_DEBUG("Write magic: %x", bl->magicStart0);
|
|
|
|
if (!is_uf2_block(bl)) {
|
|
return -1;
|
|
}
|
|
|
|
// only accept block with same family id
|
|
if ( UF2_FAMILY_ID && !((bl->flags & UF2_FLAG_FAMILYID) && (bl->familyID == UF2_FAMILY_ID)) ) {
|
|
return -1;
|
|
}
|
|
|
|
if ((bl->flags & UF2_FLAG_NOFLASH) || bl->payloadSize > 256 || (bl->targetAddr & 0xff) ||
|
|
bl->targetAddr < USER_FLASH_START || bl->targetAddr + bl->payloadSize > USER_FLASH_END) {
|
|
#if USE_DBG_MSC
|
|
if (!quiet)
|
|
logval("invalid target addr", bl->targetAddr);
|
|
#endif
|
|
NRF_LOG_WARNING("Skip block at %x", bl->targetAddr);
|
|
// this happens when we're trying to re-flash CURRENT.UF2 file previously
|
|
// copied from a device; we still want to count these blocks to reset properly
|
|
} else {
|
|
// logval("write block at", bl->targetAddr);
|
|
NRF_LOG_DEBUG("Write block at %x", bl->targetAddr);
|
|
|
|
static bool first_write = true;
|
|
if ( first_write ) {
|
|
first_write = false;
|
|
led_state(STATE_WRITING_STARTED);
|
|
}
|
|
|
|
flash_nrf5x_write(bl->targetAddr, bl->data, bl->payloadSize, true);
|
|
}
|
|
|
|
if (state && bl->numBlocks) {
|
|
if (state->numBlocks != bl->numBlocks) {
|
|
if (bl->numBlocks >= MAX_BLOCKS || state->numBlocks)
|
|
state->numBlocks = 0xffffffff;
|
|
else
|
|
state->numBlocks = bl->numBlocks;
|
|
}
|
|
if (bl->blockNo < MAX_BLOCKS) {
|
|
uint8_t mask = 1 << (bl->blockNo % 8);
|
|
uint32_t pos = bl->blockNo / 8;
|
|
if (!(state->writtenMask[pos] & mask)) {
|
|
// logval("incr", state->numWritten);
|
|
state->writtenMask[pos] |= mask;
|
|
state->numWritten++;
|
|
}
|
|
if (state->numWritten >= state->numBlocks) {
|
|
// flush last blocks
|
|
flash_nrf5x_flush(true);
|
|
}
|
|
}
|
|
NRF_LOG_DEBUG("wr %d=%d (of %d)", state->numWritten, bl->blockNo, bl->numBlocks);
|
|
}
|
|
|
|
return 512;
|
|
}
|