amt630h-sdk-beta/STEPLDR/Src/Entry.c

/*
**********************************************************************
Copyright (c)2021 Arkmicro Technologies Inc.  All Rights Reserved 
Filename: Entry.c
Version : 1.00
Date    : 2021.08.20
Author  : Sim
***********************************************************************
*/
#include "typedef.h"
#include "amt630h.h"
#include "uart.h"
#include "timer.h"
#include "aic.h"
#include "mmu.h"
#include "clock.h"
#include "lcd.h"
#include "spi.h"
#include "sysinfo.h"
#include "sysctl.h"
#include "board.h"
#include "fs/ff.h"
#include "fs/diskio.h"
#include "crc32.h"

#include <intrinsics.h>

extern void bootFromSPI(void);
extern void updateFromSD(int chipid);
extern void updateFromUSB(void);
int wdt_init(void);
void wdt_stop(void);
void wdt_start(void);

static SysInfo *pSysInfo = NULL;

void UpdateFromMedia(int drv)
{
	FRESULT  fret;
	FIL fp = {0};
	UINT32 size;
	unsigned int checksum, calc_checksum;
	int timeout = 0;
	int update_ok = 0;
	SysInfo *sysinfo = GetSysInfo();
	char loaderfile[32];
	char stepldrfile[32];
	char appfile[32];

	if (drv == SDMMC) {
		strcpy(loaderfile, LOADER_FILE_NAME);
		strcpy(stepldrfile, STEPLDR_FILE_NAME);
		strcpy(appfile, APP_FILE_NAME);
	} else if (drv == USB) {
		strcpy(loaderfile, "1:/");
		strcat(loaderfile, LOADER_FILE_NAME);
		strcpy(stepldrfile, "1:/");
		strcat(stepldrfile, STEPLDR_FILE_NAME);
		strcpy(appfile, "1:/");
		strcat(appfile, APP_FILE_NAME);
	} else {
		SendUartString("Unknown disk drv\r\n");
		return;
	}

	fret = f_mount(drv, &g_fs);
	if(fret == FR_OK)
		SendUartString("Mount file ok\r\n");
	else
	{
		SendUartString("Mount file fail\r\n");
		return;
	}

	SendUartString("burn loader start... \r\n");
readloader:
	fret = f_open(&fp, loaderfile, FA_OPEN_EXISTING | FA_READ);
	if(fret != FR_OK) {
		SendUartString("Open file fail, don't update.\r\n");
	} else {
		fret = f_read(&fp, (void *)IMAGE_ENTRY, LOADER_MAX_SIZE, &size);
		f_close(&fp);
		if(fret != FR_OK) {
			if (timeout++ < 3) {
				SendUartString("Read file fail, read again.\r\n");
				goto readloader;
			}
		} else {
			checksum = *(unsigned int*)(IMAGE_ENTRY + APPLDR_CHECKSUM_OFFSET);
			*(unsigned int*)(IMAGE_ENTRY + APPLDR_CHECKSUM_OFFSET) = 0;
			calc_checksum = xcrc32((unsigned char*)IMAGE_ENTRY, size, 0xffffffff);
			*(unsigned int*)(IMAGE_ENTRY + APPLDR_CHECKSUM_OFFSET) = checksum;
			if (calc_checksum != checksum) {
				if (timeout++ < 3) {
					SendUartString("read loader checksum fail, read again.\n");
					goto readloader;
				}
				else {
					SendUartString("read loader fail, don't update.\n");
				}
			} else {
				if (SpiNorBurn((void*)IMAGE_ENTRY, LOADER_OFFSET, size, 0)) {
					SendUartString("burn loader fail.\n");
					goto end;
				}
			}
		}
	}
	SendUartString("burn loader end. \r\n");

	timeout = 0;
	SendUartString("burn stepldr start... \r\n");
readstepldr:
	fret = f_open(&fp, stepldrfile, FA_OPEN_EXISTING | FA_READ);
	if(fret != FR_OK) {
		SendUartString("Open file fail, don't update.\r\n");
	} else {
		fret = f_read(&fp, (void *)IMAGE_ENTRY, STEPLDR_MAX_SIZE, &size);
		f_close(&fp);
		if(fret != FR_OK) {
			if (timeout++ < 3) {
				SendUartString("Read file fail, read again.\r\n");
				goto readstepldr;
			} else {
				SendUartString("read stepldr fail, don't update.\n");
			}
		} else {
			checksum = *(unsigned int*)(IMAGE_ENTRY + APPLDR_CHECKSUM_OFFSET);
			*(unsigned int*)(IMAGE_ENTRY + APPLDR_CHECKSUM_OFFSET) = 0;
			calc_checksum = xcrc32((unsigned char*)IMAGE_ENTRY, size, 0xffffffff);
			*(unsigned int*)(IMAGE_ENTRY + APPLDR_CHECKSUM_OFFSET) = checksum;
			if (calc_checksum != checksum) {
				if (timeout++ < 3) {
					SendUartString("read stepldr checksum fail, read again.\n");
					goto readstepldr;
				}
				else {
					SendUartString("read stepldr fail, don't update.\n");
				}
			} else {
				if (sysinfo->stepldr_offset == STEPLDRA_OFFSET)
					sysinfo->stepldr_offset = STEPLDRB_OFFSET;
				else
					sysinfo->stepldr_offset = STEPLDRA_OFFSET;
				sysinfo->stepldr_size = size;
				if (SpiNorBurn((void*)IMAGE_ENTRY, sysinfo->stepldr_offset, size, 0)) {
					SendUartString("burn stepldr fail.\n");
					goto end;
				}
			}
		}
	}
	SendUartString("burn stepldr end. \r\n");

	timeout = 0;
	SendUartString("burn app start... \r\n");
	update_logo_init();
readapp:
	fret = f_open(&fp, appfile, FA_OPEN_EXISTING | FA_READ);
	if(fret != FR_OK) {
		SendUartString("Open file fail, don't update.\r\n");
	} else {
		fret = f_read(&fp, (void *)IMAGE_ENTRY, IMAGE_MAX_SIZE, &size);
		f_close(&fp);
		if(fret != FR_OK) {
			SendUartString("Read file fail\r\n");
			if (timeout++ < 3) {
				SendUartString("Read file fail, read again.\r\n");
				goto readapp;
			} else {
				SendUartString("read update file fail, don't update.\n");
			}
		} else {
			UpFileHeader *header = (UpFileHeader *)IMAGE_ENTRY;
			checksum = header->checksum;
			header->checksum = 0;
			calc_checksum = xcrc32((unsigned char*)IMAGE_ENTRY, size, 0xffffffff);
			if (calc_checksum != checksum) {
				if (timeout++ < 3) {
					SendUartString("read update file checksum fail, read again.\n");
					goto readapp;
				}
				else {
					SendUartString("read update file fail, don't update.\n");
				}
			} else {
				sysinfo->app_checksum = header->checksum = checksum;
				if (!SpiNorBurn((void*)IMAGE_ENTRY, IMAGE_OFFSET, size, 1))
					update_ok = 1;
			}
		}
	}
	SendUartString("burn app end.\r\n");

	if (update_ok) {
		SendUartString("update ok, save sysinfo.\r\n");
		sysinfo->update_status = UPDATE_STATUS_END;
		SaveSysInfo(sysinfo);
	}
end:
	SendUartString("Update is finished.\r\n");
}

void main(void)
{
	wdt_init();
	timer_init();
	AIC_Initialize();
	InitUart(115200);
	SendUartString("ARK AMT630H STEPLDR V 1.0\r\n");

	vClkInit();
	lcd_init();
	SpiSelectPad();
	SpiInit();

#ifdef MMU_ENABLE
	MMU_Init();
#endif

	if (ReadSysInfo()) {
		SendUartString("read sysinfo fail, use default.\r\n");
		SetDefaultSysInfo();
	}

	pSysInfo = GetSysInfo();
	//pSysInfo->update_status = UPDATE_STATUS_START;
	//pSysInfo->update_media_type = UPDATE_MEDIA_UART;
	if (pSysInfo->update_status == UPDATE_STATUS_START) {
		wdt_stop();
		switch (pSysInfo->update_media_type) {
		case UPDATE_MEDIA_SD:
			updateFromSD(0);
			break;
		case UPDATE_MEDIA_USB:
			updateFromUSB();
			break;
		case UPDATE_MEDIA_UART:
			__enable_irq();
			/* uart1 pad select */
			vSysctlConfigure(SYS_PAD_CTRL02, 16, 0xf, 0x5);
			updateFromUart(UART_MCU_PORT); 
			break;
		}
		wdt_start();
	}

	bootFromSPI();
}