/*****************************************************************
*
*	bootloader.c
*
*	Trivial Microblaze/uClinux bootloader
*
*****************************************************************/

/* Get general hardware memory map */
#include "xparameters.h"
#include "xbasic_types.h"
#include "xstatus.h"
#include "mb_interface.h"

#define FLASH_BASEADDR 0xff000000
#define FLASH_COMMAND_ADDR_555 (FLASH_BASEADDR + 0x01554)
#define FLASH_COMMAND_ADDR_2AA (FLASH_BASEADDR + 0x00aa8)
#define FLASH_COMMAND_AA               0x00AA00AA
#define FLASH_COMMAND_55               0x00550055
#define FLASH_COMMAND_READ_RESET       0x00F000F0
#define FLASH_COMMAND_ERASE            0x00800080
#define FLASH_COMMAND_AUTO_CHIP_ERASE  0x00100010
#define FLASH_COMMAND_AUTO_BLOCK_ERASE 0x00300030
#define FLASH_COMMAND_AUTO_PROGRAM     0x00A000A0
#define FLASH_DQ7                      0x00800080
#define FLASH_DQ5                      0x00200020

unsigned int *flash_start = (unsigned int *)FLASH_BASEADDR;
unsigned int *ddr_start = (unsigned int *)XPAR_DDR_CONTROLLER_BASEADDR;
unsigned int *ddr_end = (unsigned int *)XPAR_DDR_CONTROLLER_HIGHADDR; 
unsigned int *ram_start = (unsigned int *)0xffe00000;
unsigned int *ram_end = (unsigned int *)0xfff00000;

typedef void (*void_fn)(void);
void_fn kernel_start;

void flash_read_reset(void)
{
   *((unsigned int *)FLASH_COMMAND_ADDR_555) = FLASH_COMMAND_AA;
   *((unsigned int *)FLASH_COMMAND_ADDR_2AA) = FLASH_COMMAND_55;
   *((unsigned int *)FLASH_COMMAND_ADDR_555) = FLASH_COMMAND_READ_RESET;
}

void progress_bar(int start, int finish, int current)
{
	int i, j;

	i=j=40*(current-start)/(finish-start);

	while(j--)
		outbyte('.');

	while(i--)
		outbyte('\b');
}

char progress[]={'|','\\','-','/','|','\\','-','/'};

void memory_test(unsigned int *start, unsigned int *end)
{
	unsigned int *dst;

	print("\n\r\n\rTesting memory...");

	for(dst=start; dst<end; dst++)
		*dst=(unsigned int)dst;

	for(dst=start; dst<end; dst++)
		if(*dst!=(unsigned int)dst)
		{
			print("failed at 0x");
			putnum(dst);
			print("\n\rContents:0x");
			putnum(*dst);
			print(" expected 0x");
			putnum((unsigned int)dst);
			print("\n\rHalting...");
			for(;;)
				;
		}
}

void clear_memory(unsigned int *start, unsigned int *end)
{
	unsigned int *dst;

	for(dst=start; dst< end; dst++)
		*dst=0;
}

void copy_image(unsigned int *src, unsigned int *dst, unsigned int len)
{

	print("\n\rCopying Kernel Image from Flash to RAM...");

	flash_read_reset();

	while(len--)
	{
		*dst++=*src++;
	}
	print("...done\n\r");
}

void verify_image(unsigned int *src, unsigned int *dst, unsigned int len)
{
	print("\n\rVerifying Kernel Image...");	

	flash_read_reset();
	while(len--)
	{
		if(*dst++ != *src++)
		{
			print("Error at ");
			putnum(dst-1);
			print("/");
			putnum(src-1);
			print("  Contents ");
			putnum(*(src-1));
			print("/");
			putnum(*(dst-1));
			print("\n\rHalting...");
			for(;;)
				;
		}
	}

	print("...success\n\r");
}

int main(void)
{
	memory_test(ddr_start, ddr_end);
	clear_memory(ddr_start, ddr_end);

	/* Hard code image length, ugh! */
	copy_image(flash_start, ddr_start, 0x100000);
	verify_image(flash_start, ddr_start, 0x100000);

	print("Jumping to kernel startup...\n\r\n\r");
	kernel_start = (void_fn)ddr_start;

gonna_jump:
	kernel_start();
}