void flashBootloader(const char * filename)
{
FIL file;
f_open(&file, filename, FA_READ);
uint8_t buffer[1024];
UINT count;
lcd_clear();
displayProgressBar(STR_WRITING);
static uint8_t unlocked = 0;
if (!unlocked) {
unlocked = 1;
unlockFlash();
}
for (int i=0; i<BOOTLOADER_SIZE; i+=1024) {
watchdogSetTimeout(100/*1s*/);
if (f_read(&file, buffer, 1024, &count) != FR_OK || count != 1024) {
POPUP_WARNING(STR_SDCARD_ERROR);
break;
}
if (i==0 && !isBootloaderStart((uint32_t *)buffer)) {
POPUP_WARNING(STR_INCOMPATIBLE);
break;
}
for (int j=0; j<1024; j+=FLASH_PAGESIZE) {
writeFlash(CONVERT_UINT_PTR(FIRMWARE_ADDRESS+i+j), (uint32_t *)(buffer+j));
}
updateProgressBar(i, BOOTLOADER_SIZE);
SIMU_SLEEP(30/*ms*/);
}
if (unlocked) {
lockFlash();
unlocked = 0;
}
f_close(&file);
}
void menuUp1(uint8_t event)
{
FRESULT fr ;
struct fileControl *fc = &FileControl ;
static uint8_t mounted = 0 ;
static uint32_t state ;
static uint32_t firmwareAddress ;
uint32_t i ;
uint32_t width ;
if (UpdateItem == UPDATE_TYPE_BOOTLOADER ) // Bootloader
{
TITLE( "UPDATE BOOT" ) ;
}
else
{
#ifdef PCBX9D
if (UpdateItem == UPDATE_TYPE_SPORT_INT )
{
TITLE( "UPDATE Int. XJT" ) ;
}
else
{
TITLE( "UPDATE Ext. SPort" ) ;
}
#endif
#ifdef PCBSKY
#ifndef REVX
if (UpdateItem == UPDATE_TYPE_COPROCESSOR )
{
TITLE( "UPDATE COPROC" ) ;
}
else
{
TITLE( "UPDATE SPort" ) ;
}
#else
if (UpdateItem == UPDATE_TYPE_SPORT_EXT )
{
TITLE( "UPDATE SPort" ) ;
}
#endif
#endif
}
switch(event)
{
case EVT_ENTRY:
state = UPDATE_NO_FILES ;
if ( mounted == 0 )
{
#if defined(PCBTARANIS)
fr = f_mount(0, &g_FATFS_Obj) ;
#else
fr = f_mount(0, &g_FATFS) ;
#endif
#ifdef PCBX9D
unlockFlash() ;
#endif
}
else
{
fr = FR_OK ;
}
if ( fr == FR_OK)
{
mounted = 1 ;
}
if ( mounted )
{
fr = f_chdir( (TCHAR *)"\\firmware" ) ;
if ( fr == FR_OK )
{
state = UPDATE_NO_FILES ;
fc->index = 0 ;
fr = f_opendir( &Dj, (TCHAR *) "." ) ;
if ( fr == FR_OK )
{
if ( (UpdateItem > 1 ) )
{
fc->ext[0] = 'F' ;
fc->ext[1] = 'R' ;
fc->ext[2] = 'K' ;
}
else
{
fc->ext[0] = 'B' ;
fc->ext[1] = 'I' ;
fc->ext[2] = 'N' ;
}
fc->ext[3] = 0 ;
fc->index = 0 ;
fc->nameCount = fillNames( 0, fc ) ;
fc->hpos = 0 ;
fc->vpos = 0 ;
if ( fc->nameCount )
{
state = UPDATE_FILE_LIST ;
}
}
}
}
break ;
case EVT_KEY_FIRST(KEY_EXIT):
if ( state < UPDATE_ACTION )
{
chainMenu(menuUpdate) ;
killEvents(event) ;
}
break ;
}
switch ( state )
{
case UPDATE_NO_FILES :
lcd_puts_Pleft( 4*FH, "\005No Files" ) ;
lcd_outdez( 21*FW, 4*FH, mounted ) ;
break ;
case UPDATE_FILE_LIST :
SportVerValid = 0 ;
if ( fileList( event, &FileControl ) == 1 )
{
state = UPDATE_CONFIRM ;
}
break ;
case UPDATE_CONFIRM :
if ( (UpdateItem > UPDATE_TYPE_BOOTLOADER ) )
{
#ifdef PCBX9D
if ( (UpdateItem == UPDATE_TYPE_SPORT_INT ) )
{
lcd_puts_Pleft( 2*FH, "Flash Int. XJT from" ) ;
}
else
{
lcd_puts_Pleft( 2*FH, "Flash Ext.SP from" ) ;
}
SportVerValid = 0 ;
#else
#ifndef REVX
if ( (UpdateItem == UPDATE_TYPE_COPROCESSOR ) )
{
lcd_puts_Pleft( 2*FH, "Flash Co-Proc. from" ) ;
}
else
{
lcd_puts_Pleft( 2*FH, "Flash SPort from" ) ;
}
CoProcReady = 0 ;
#else
lcd_puts_Pleft( 2*FH, "Flash SPort from" ) ;
#endif
#endif
}
else
{
lcd_puts_Pleft( 2*FH, "Flash Bootloader from" ) ;
}
cpystr( cpystr( (uint8_t *)FlashFilename, (uint8_t *)"\\firmware\\" ), (uint8_t *)Filenames[fc->vpos] ) ;
#if defined(PCBTARANIS)
lcd_putsnAtt( 0, 4*FH, Filenames[fc->vpos], DISPLAY_CHAR_WIDTH, 0 ) ;
#else
lcd_putsnAtt0( 0, 4*FH, Filenames[fc->vpos], DISPLAY_CHAR_WIDTH, 0 ) ;
#endif
if ( event == EVT_KEY_LONG(KEY_MENU) )
{
state = UPDATE_SELECTED ;
}
if ( event == EVT_KEY_LONG(KEY_EXIT) )
{
state = UPDATE_FILE_LIST ; // Canceled
}
break ;
case UPDATE_SELECTED :
f_open( &FlashFile, FlashFilename, FA_READ ) ;
f_read( &FlashFile, (BYTE *)FileData, 1024, &BlockCount ) ;
i = 1 ;
if (UpdateItem == UPDATE_TYPE_BOOTLOADER ) // Bootloader
{
i = validateFile( (uint32_t *) FileData ) ;
}
if ( i == 0 )
{
state = UPDATE_INVALID ;
}
else
{
if (UpdateItem == UPDATE_TYPE_BOOTLOADER ) // Bootloader
{
#ifdef PCBX9D
firmwareAddress = 0x08000000 ;
#endif
#ifdef PCBSKY
firmwareAddress = 0x00400000 ;
#endif
}
#ifdef PCBSKY
#ifndef REVX
else if (UpdateItem == UPDATE_TYPE_COPROCESSOR ) // Bootloader
{
firmwareAddress = 0x00000080 ;
if ( check_ready() == 0 )
{
CoProcReady = 1 ;
}
}
#endif
#endif
else
{
// SPort update
SportState = SPORT_START ;
FirmwareSize = FileSize[fc->vpos] ;
BlockInUse = 0 ;
f_read( &FlashFile, (BYTE *)ExtraFileData, 1024, &XblockCount ) ;
}
BytesFlashed = 0 ;
BlockOffset = 0 ;
ByteEnd = 1024 ;
state = UPDATE_ACTION ;
}
break ;
case UPDATE_INVALID :
lcd_puts_Pleft( 2*FH, "Invalid File" ) ;
lcd_puts_Pleft( 4*FH, "Press EXIT" ) ;
if ( event == EVT_KEY_FIRST(KEY_EXIT) )
{
state = UPDATE_FILE_LIST ; // Canceled
killEvents(event) ;
}
break ;
case UPDATE_ACTION :
// Do the flashing
lcd_puts_Pleft( 3*FH, "Flashing" ) ;
if (UpdateItem == UPDATE_TYPE_BOOTLOADER ) // Bootloader
{
width = ByteEnd >> 9 ;
if ( BytesFlashed < ByteEnd )
{
program( (uint32_t *)firmwareAddress, &((uint32_t *)FileData)[BlockOffset] ) ; // size is 256 bytes
BlockOffset += 64 ; // 32-bit words (256 bytes)
firmwareAddress += 256 ;
BytesFlashed += 256 ;
}
else
{
if ( ByteEnd >= 32768 )
{
state = UPDATE_COMPLETE ;
}
else
{
f_read( &FlashFile, (BYTE *)FileData, 1024, &BlockCount ) ;
ByteEnd += 1024 ;
BlockOffset = 0 ;
}
}
}
#ifdef PCBSKY
#ifndef REVX
else if (UpdateItem == UPDATE_TYPE_COPROCESSOR ) // CoProcessor
int main()
{
uint8_t index = 0;
uint8_t maxhsize = DISPLAY_CHAR_WIDTH;
FRESULT fr;
uint32_t state = ST_START;
uint32_t nameCount = 0;
uint32_t vpos = 0;
uint32_t hpos = 0;
#if defined(PCBTARANIS)
wdt_reset();
RCC_AHB1PeriphClockCmd(PWR_RCC_AHB1Periph | KEYS_RCC_AHB1Periph | LCD_RCC_AHB1Periph | BACKLIGHT_RCC_AHB1Periph | I2C_RCC_AHB1Periph | SD_RCC_AHB1Periph, ENABLE);
RCC_APB1PeriphClockCmd(LCD_RCC_APB1Periph | BACKLIGHT_RCC_APB1Periph | INTERRUPT_5MS_APB1Periph | I2C_RCC_APB1Periph | SD_RCC_APB1Periph, ENABLE);
RCC_APB2PeriphClockCmd(BACKLIGHT_RCC_APB2Periph, ENABLE);
#endif
pwrInit();
delaysInit(); //needed for lcdInit()
lcdInit();
backlightInit();
lcd_clear();
lcd_putsn(0, 0, (const char *)bootloaderVersion, 0); // trick to avoid bootloaderVersion to be optimized out ...
lcd_putsLeft(0, BOOTLOADER_TITLE);
lcd_invert_line(0);
lcdRefresh();
keysInit();
i2cInit();
__enable_irq();
init10msTimer();
#if defined(PCBTARANIS)
// SD card detect pin
sdInit();
usbInit();
#endif
for (;;) {
wdt_reset();
if (Tenms) {
Tenms = 0;
lcdRefreshWait();
lcd_clear();
lcd_putsLeft(0, BOOTLOADER_TITLE);
lcd_invert_line(0);
uint8_t event = getEvent();
if (state != ST_USB) {
if (usbPlugged()) {
state = ST_USB;
if (!unlocked) {
unlocked = 1;
unlockFlash();
}
usbPluggedIn();
}
}
if (state == ST_START) {
lcd_putsLeft(2*FH, "\010Write Firmware");
lcd_putsLeft(3*FH, "\010Restore EEPROM");
lcd_putsLeft(4*FH, "\010Exit");
lcd_invert_line(2+vpos);
lcd_putsLeft(7*FH, INDENT "Or plug in a USB cable for mass storage");
if (event == EVT_KEY_FIRST(BOOT_KEY_DOWN)) {
vpos == 2 ? vpos = 0 : vpos = vpos+1;
}
else if (event == EVT_KEY_FIRST(BOOT_KEY_UP)) {
vpos == 0 ? vpos = 2 : vpos = vpos-1;
}
else if (event == EVT_KEY_BREAK(BOOT_KEY_MENU)) {
switch (vpos) {
case 0:
state = ST_FLASH_MENU;
break;
case 1:
state = ST_RESTORE_MENU;
break;
default:
state = ST_REBOOT;
}
}
}
if (state == ST_USB) {
lcd_putsLeft(4*FH, "\026USB Connected");
if (usbPlugged() == 0) {
vpos = 0;
if (unlocked) {
lockFlash();
unlocked = 0;
}
state = ST_START;
}
#if defined(PCBSKY9X)
usbMassStorage();
#endif
}
if (state == ST_FLASH_MENU || state == ST_RESTORE_MENU) {
sdInit();
memoryType = (state == ST_RESTORE_MENU ? MEM_EEPROM : MEM_FLASH);
state = ST_DIR_CHECK;
}
else if (state == ST_DIR_CHECK) {
fr = f_chdir(getBinaryPath());
if (fr == FR_OK) {
state = ST_OPEN_DIR;
}
else {
lcd_putsLeft(2*FH, INDENT "Directory is missing!");
if (event == EVT_KEY_BREAK(BOOT_KEY_EXIT) || event == EVT_KEY_BREAK(BOOT_KEY_MENU)) {
vpos = 0;
state = ST_START;
}
}
}
if (state == ST_OPEN_DIR) {
index = 0;
fr = f_opendir(&Dj, ".");
if (fr == FR_OK) {
state = ST_FILE_LIST;
nameCount = fillNames(0);
hpos = 0;
vpos = 0;
}
}
if (state == ST_FILE_LIST) {
uint32_t limit = 6;
if (nameCount < limit) {
limit = nameCount;
}
maxhsize = 0;
for (uint32_t i=0; i<limit; i++) {
uint32_t x;
x = strlen(Filenames[i]);
if (x > maxhsize) {
maxhsize = x;
}
if (x > DISPLAY_CHAR_WIDTH) {
if (hpos + DISPLAY_CHAR_WIDTH > x) {
x = x - DISPLAY_CHAR_WIDTH;
}
else {
x = hpos;
}
}
else {
x = 0;
}
lcd_putsnAtt(INDENT_WIDTH, 16 + FH * i, &Filenames[i][x], DISPLAY_CHAR_WIDTH, 0);
}
if (event == EVT_KEY_REPT(BOOT_KEY_DOWN) || event == EVT_KEY_FIRST(BOOT_KEY_DOWN)) {
if (vpos < limit - 1) {
vpos += 1;
}
else {
if (nameCount > limit) {
index += 1;
nameCount = fillNames(index);
}
}
}
else if (event == EVT_KEY_REPT(BOOT_KEY_UP) || event == EVT_KEY_FIRST(BOOT_KEY_UP)) {
if (vpos > 0) {
vpos -= 1;
}
else {
if (index) {
index -= 1;
nameCount = fillNames(index);
}
}
}
#if !defined(PCBTARANIS)
else if (event == EVT_KEY_REPT(BOOT_KEY_RIGHT) || event == EVT_KEY_FIRST(BOOT_KEY_RIGHT)) {
if (hpos + DISPLAY_CHAR_WIDTH < maxhsize) {
hpos += 1;
}
}
else if (event == EVT_KEY_REPT(BOOT_KEY_LEFT) || event == EVT_KEY_FIRST(BOOT_KEY_LEFT)) {
if (hpos) {
hpos -= 1;
}
}
#endif
else if (event == EVT_KEY_BREAK(BOOT_KEY_MENU)) {
// Select file to flash
state = ST_FLASH_CHECK;
Valid = 0;
}
else if (event == EVT_KEY_FIRST(BOOT_KEY_EXIT)) {
state = ST_START;
vpos = 0;
}
lcd_invert_line(2 + vpos);
}
else if (state == ST_FLASH_CHECK) {
int result = menuFlashFile(vpos, event);
FirmwareSize = FileSize[vpos] - BOOTLOADER_SIZE;
if (result == 0) {
// canceled
state = ST_FILE_LIST;
}
else if (result == 1) {
// confirmed
firmwareAddress = FIRMWARE_ADDRESS + BOOTLOADER_SIZE;
firmwareWritten = 0;
eepromAddress = 0;
eepromWritten = 0;
state = ST_FLASHING;
}
}
else if (state == ST_FLASHING) {
// commit to flashing
lcd_putsLeft(4*FH, "\032Writing...");
if (!unlocked && (memoryType == MEM_FLASH)) {
unlocked = 1;
unlockFlash();
}
int progress;
if (memoryType == MEM_FLASH) {
writeFlashBlock();
firmwareWritten += sizeof(Block_buffer);
progress = (200*firmwareWritten) / FirmwareSize;
}
else {
writeEepromBlock();
eepromWritten += sizeof(Block_buffer);
progress = (200*eepromWritten) / EESIZE;
}
lcd_rect( 3, 6*FH+4, 204, 7);
lcd_hline(5, 6*FH+6, progress, FORCE);
lcd_hline(5, 6*FH+7, progress, FORCE);
lcd_hline(5, 6*FH+8, progress, FORCE);
fr = f_read(&FlashFile, (BYTE *)Block_buffer, sizeof(Block_buffer), &BlockCount);
if (BlockCount == 0) {
state = ST_FLASH_DONE; // EOF
}
if (firmwareWritten >= FLASHSIZE - BOOTLOADER_SIZE) {
state = ST_FLASH_DONE; // Backstop
}
if (eepromWritten >= EESIZE) {
state = ST_FLASH_DONE; // Backstop
}
}
if (state == ST_FLASH_DONE) {
if (unlocked) {
lockFlash();
unlocked = 0;
}
lcd_putsLeft(4*FH, "\024Writing Complete");
if (event == EVT_KEY_FIRST(BOOT_KEY_EXIT) || event == EVT_KEY_BREAK(BOOT_KEY_MENU)) {
state = ST_START;
vpos = 0;
}
}
if (event == EVT_KEY_LONG(BOOT_KEY_EXIT)) {
state = ST_REBOOT;
}
lcdRefresh();
if (PowerUpDelay < 20) { // 200 mS
PowerUpDelay += 1;
}
else {
sdPoll10ms();
}
}
if (state != ST_FLASHING && state != ST_USB) {
#if defined(REV9E)
if (pwrPressed()) {
#else
if (pwrCheck() == e_power_off) {
#endif
lcdOff(); // this drains LCD caps
pwrOff();
for (;;) {
// Wait for power to go off
}
}
}
if (state == ST_REBOOT) {
if (readKeys() == 0) {
lcd_clear();
lcdRefresh();
lcdRefreshWait();
RCC->CSR |= RCC_CSR_RMVF; //clear the reset flags in RCC clock control & status register
NVIC_SystemReset();
}
}
}
return 0;
}
int main()
{
uint8_t index = 0;
uint8_t maxhsize = DISPLAY_CHAR_WIDTH;
FRESULT fr;
uint32_t state = ST_START;
uint32_t nameCount = 0;
uint32_t vpos = 0;
uint32_t hpos = 0;
#if defined(PCBTARANIS)
wdt_reset();
RCC->AHB1ENR |= RCC_AHB1ENR_GPIOAEN; // Enable portA clock
#endif
pwrInit();
#if defined(PCBSKY9X)
MATRIX->CCFG_SYSIO |= 0x000000F0L; // Disable syspins, enable B4,5,6,7
#endif
#if defined(PCBSKY9X)
init_SDcard();
PIOC->PIO_PER = PIO_PC25; // Enable bit C25 (USB-detect)
start_timer0();
#endif
lcdInit();
#if defined(PCBSKY9X)
extern uint8_t OptrexDisplay;
OptrexDisplay = 1;
#endif
lcd_clear();
lcd_putsLeft(0, BOOTLOADER_TITLE);
lcd_invert_line(0);
lcdRefresh();
#if defined(PCBSKY9X)
OptrexDisplay = 0;
lcdRefresh();
#endif
#if defined(PCBTARANIS)
keysInit();
I2C_EE_Init();
init_hw_timer();
#endif
__enable_irq();
init10msTimer();
#if defined(PCBSKY9X)
EblockAddress = -1;
init_spi();
#endif
#if defined(PCBSKY9X)
LockBits = readLockBits();
if (LockBits) {
clearLockBits();
}
#endif
#if defined(PCBTARANIS)
// SD card detect pin
sdInit();
usbInit();
usbStart();
#endif
for (;;) {
wdt_reset();
if (Tenms) {
if (EE_timer) {
if (--EE_timer == 0) {
#if defined(PCBSKY9X)
writeBlock();
#endif
}
}
Tenms = 0;
lcd_clear();
lcd_putsLeft(0, BOOTLOADER_TITLE);
lcd_invert_line(0);
uint8_t event = getEvent();
if (state != ST_USB) {
if (usbPlugged()) {
state = ST_USB;
if (!unlocked) {
unlocked = 1;
unlockFlash();
}
usbPluggedIn();
}
}
if (state == ST_START) {
lcd_putsLeft(2*FH, "\010Write Firmware");
lcd_putsLeft(3*FH, "\010Restore EEPROM");
lcd_putsLeft(4*FH, "\010Exit");
lcd_invert_line(2+vpos);
lcd_putsLeft(7*FH, INDENT "Or plug in a USB cable for mass storage");
if (event == EVT_KEY_FIRST(BOOT_KEY_DOWN)) {
vpos == 2 ? vpos = 0 : vpos = vpos+1;
}
else if (event == EVT_KEY_FIRST(BOOT_KEY_UP)) {
vpos == 0 ? vpos = 2 : vpos = vpos-1;
}
else if (event == EVT_KEY_BREAK(BOOT_KEY_MENU)) {
switch (vpos) {
case 0:
state = ST_FLASH_MENU;
break;
case 1:
state = ST_RESTORE_MENU;
break;
default:
state = ST_REBOOT;
}
}
}
if (state == ST_USB) {
lcd_putsLeft(4*FH, "\026USB Connected");
if (usbPlugged() == 0) {
vpos = 0;
if (unlocked) {
lockFlash();
unlocked = 0;
}
state = ST_START;
}
#if defined(PCBSKY9X)
usbMassStorage();
#endif
}
if (state == ST_FLASH_MENU || state == ST_RESTORE_MENU) {
sdInit();
memoryType = (state == ST_RESTORE_MENU ? MEM_EEPROM : MEM_FLASH);
state = ST_DIR_CHECK;
}
else if (state == ST_DIR_CHECK) {
fr = f_chdir(getBinaryPath());
if (fr == FR_OK) {
state = ST_OPEN_DIR;
}
else {
lcd_putsLeft(2*FH, INDENT "Directory is missing!");
if (event == EVT_KEY_BREAK(BOOT_KEY_EXIT) || event == EVT_KEY_BREAK(BOOT_KEY_MENU)) {
vpos = 0;
state = ST_START;
}
}
}
if (state == ST_OPEN_DIR) {
index = 0;
fr = f_opendir(&Dj, ".");
if (fr == FR_OK) {
state = ST_FILE_LIST;
nameCount = fillNames(0);
hpos = 0;
vpos = 0;
}
}
if (state == ST_FILE_LIST) {
uint32_t limit = 6;
if (nameCount < limit) {
limit = nameCount;
}
maxhsize = 0;
for (uint32_t i=0; i<limit; i++) {
uint32_t x;
x = strlen(Filenames[i]);
if (x > maxhsize) {
maxhsize = x;
}
if (x > DISPLAY_CHAR_WIDTH) {
if (hpos + DISPLAY_CHAR_WIDTH > x) {
x = x - DISPLAY_CHAR_WIDTH;
}
else {
x = hpos;
}
}
else {
x = 0;
}
lcd_putsnAtt(INDENT_WIDTH, 16 + FH * i, &Filenames[i][x], DISPLAY_CHAR_WIDTH, 0);
}
if (event == EVT_KEY_REPT(BOOT_KEY_DOWN) || event == EVT_KEY_FIRST(BOOT_KEY_DOWN)) {
if (vpos < limit - 1) {
vpos += 1;
}
else {
if (nameCount > limit) {
index += 1;
nameCount = fillNames(index);
}
}
}
else if (event == EVT_KEY_REPT(BOOT_KEY_UP) || event == EVT_KEY_FIRST(BOOT_KEY_UP)) {
if (vpos > 0) {
vpos -= 1;
}
else {
if (index) {
index -= 1;
nameCount = fillNames(index);
}
}
}
#if !defined(PCBTARANIS)
else if (event == EVT_KEY_REPT(BOOT_KEY_RIGHT) || event == EVT_KEY_FIRST(BOOT_KEY_RIGHT)) {
if (hpos + DISPLAY_CHAR_WIDTH < maxhsize) {
hpos += 1;
}
}
else if (event == EVT_KEY_REPT(BOOT_KEY_LEFT) || event == EVT_KEY_FIRST(BOOT_KEY_LEFT)) {
if (hpos) {
hpos -= 1;
}
}
#endif
else if (event == EVT_KEY_BREAK(BOOT_KEY_MENU)) {
// Select file to flash
state = ST_FLASH_CHECK;
Valid = 0;
}
else if (event == EVT_KEY_FIRST(BOOT_KEY_EXIT)) {
state = ST_START;
vpos = 0;
}
lcd_invert_line(2 + vpos);
}
else if (state == ST_FLASH_CHECK) {
int result = menuFlashFile(vpos, event);
FirmwareSize = FileSize[vpos] - BOOTLOADER_SIZE;
if (result == 0) {
// canceled
state = ST_FILE_LIST;
}
else if (result == 1) {
// confirmed
firmwareAddress = FIRMWARE_ADDRESS + BOOTLOADER_SIZE;
firmwareWritten = 0;
eepromAddress = 0;
eepromWritten = 0;
state = ST_FLASHING;
}
}
else if (state == ST_FLASHING) {
// commit to flashing
lcd_putsLeft(4*FH, "\032Writing...");
if (!unlocked && (memoryType == MEM_FLASH)) {
unlocked = 1;
unlockFlash();
}
int progress;
if (memoryType == MEM_FLASH) {
writeFlashBlock();
firmwareWritten += sizeof(Block_buffer);
progress = (200*firmwareWritten) / FirmwareSize;
}
else {
writeEepromBlock();
eepromWritten += sizeof(Block_buffer);
progress = (200*eepromWritten) / EESIZE;
}
lcd_rect( 3, 6*FH+4, 204, 7);
lcd_hline(5, 6*FH+6, progress, FORCE);
lcd_hline(5, 6*FH+7, progress, FORCE);
lcd_hline(5, 6*FH+8, progress, FORCE);
fr = f_read(&FlashFile, (BYTE *)Block_buffer, sizeof(Block_buffer), &BlockCount);
if (BlockCount == 0) {
state = ST_FLASH_DONE; // EOF
}
if (firmwareWritten >= FLASHSIZE - BOOTLOADER_SIZE) {
state = ST_FLASH_DONE; // Backstop
}
if (eepromWritten >= EESIZE) {
state = ST_FLASH_DONE; // Backstop
}
}
if (state == ST_FLASH_DONE) {
if (unlocked) {
lockFlash();
unlocked = 0;
}
lcd_putsLeft(4*FH, "\024Writing Complete");
if (event == EVT_KEY_FIRST(BOOT_KEY_EXIT) || event == EVT_KEY_BREAK(BOOT_KEY_MENU)) {
state = ST_START;
vpos = 0;
}
}
if (event == EVT_KEY_LONG(BOOT_KEY_EXIT)) {
state = ST_REBOOT;
}
lcdRefresh();
if (PowerUpDelay < 20) { // 200 mS
PowerUpDelay += 1;
}
else {
sdPoll10ms();
}
}
if (pwrCheck() == e_power_off && state != ST_FLASHING && state != ST_USB) {
pwrOff();
for (;;) {
// Wait for power to go off
}
}
if (state == ST_REBOOT) {
if ((~readKeys() & 0x7E) == 0) {
NVIC_SystemReset();
}
}
}
return 0;
}
void DataTerminal::processByte(uint8_t byte)
{
switch(mState) {
case STATE_WAITING:
if ( byte == START_TRANSFER_CMD ) {
mState = STATE_IN_TRANSFER_HEADER;
mByteCount = 0;
trace_printf("Receiving metadata\n");
GPIO_SetBits(GPIOB, GPIO_Pin_14);
writeCmd(ACK);
}
else {
trace_printf("Bad command: %.2x\n", byte);
fail();
}
break;
case STATE_IN_TRANSFER_HEADER: {
//trace_printf("1 byte\n");
char *p = (char*)&mMetadata;
memcpy(p + mByteCount, &byte, 1);
++mByteCount;
if ( mByteCount == sizeof mMetadata ) {
trace_printf("Magick: %.8x\n", mMetadata.magic);
// Is the metadata sane?
if ( mMetadata.magic == METADATA_MAGIC ) {
mState = STATE_IN_TRANSFER_BLOCK;
trace_printf("Expecting %d bytes\n", mMetadata.size);
printmd5(mMetadata.md5);
GPIO_SetBits(GPIOB, GPIO_Pin_14);
mByteCount = 0;
mWriteAddress = APPLICATION_ADDRESS;
MD5Init(&mMD5);
unlockFlash();
writeCmd(ACK);
}
else {
trace_printf("Bad metadata \n");
fail();
}
}
}
break;
case STATE_IN_TRANSFER_BLOCK:
MD5Update(&mMD5, &byte, 1);
if ( mByteCount == 0 ) {
GPIO_SetBits(GPIOB, GPIO_Pin_15);
}
size_t offset = mByteCount % FLASH_PAGE_SIZE;
assert(offset < sizeof mCurrPage);
memcpy(mCurrPage + offset, &byte, 1);
++mByteCount;
if ( mByteCount % FLASH_PAGE_SIZE == 0 ) {
flushPage();
}
// Acknowledge every 1K
if ( mByteCount % 1024 == 0 ) {
trace_printf("Sending ACK\n");
writeCmd(ACK);
}
if ( mByteCount == mMetadata.size ) {
if ( mMetadata.size % 1024 ) {
// We have a partial page to write
flushPage();
writeCmd(ACK);
}
// Last byte!
trace_printf("Received %d bytes\n", mMetadata.size);
GPIO_ResetBits(GPIOB, GPIO_Pin_14);
mState = STATE_WAITING;
uint8_t d[16];
MD5Final(d, &mMD5);
printmd5(d);
if ( memcmp(d, mMetadata.md5, 16) ) {
// Bad MD5
trace_printf("MD5 mismatch :( \n");
fail();
}
else {
// Good transfer
trace_printf("MD5 match :) \n");
writeCmd(ACK);
flushMetadata();
lockFlash();
NVIC_SystemReset();
}
}
break;
}
}
