bootloader/bootloader.c

/*
===============================================================================
 Name        : bootloader.c
 Author      : $(author)
 Version     :
 Copyright   : $(copyright)
 Description : main definition
===============================================================================
*/

#ifdef __USE_CMSIS
#include "LPC17xx.h"
#endif

#include <cr_section_macros.h>

#include <stdio.h>
#include <string.h>

#include "uart_commands.h"
#include "timer.h"

error_code status;

enum States
{
    S1,
    S2,
    S3
};


/*
is_even_assembly:
        ldr sp, [r0]
    ret  
*/

enum States state = S1;

char* argvs[3][10];

cmd_t cmd;

uint32_t* size_store = (uint32_t*) 0x3100;
crc_t* crc_store = (crc_t*) 0x3000;

uint32_t offset, total_size, offset_pointer;

crc_t crc_prog, crc_data;


// TODO: insert other include files here

// TODO: insert other definitions and declarations here

char buff[256];

int is_code_valid(){
	crc_t expected_crc = *crc_store;
	size_t application_size = *size_store;
	crc_t real_crc = crc_init();
	if(application_size > MAX_PROG_SIZE) return 0;
	real_crc = crc_update(real_crc, (void*)APP_OFFSET, application_size);
	real_crc = crc_finalize(real_crc);
	error_code is_error = uart_commands_check(real_crc, expected_crc);
	if(is_error == ok){
		return 1;
	}
	else{
		set_colour(RED);//code invalid, red led
		return 0;
	}
}

int is_b_pressed(){
	int tmp = (LPC_GPIO0->FIOPIN)>>19;
	int tmp2 = tmp & 1;
	if(tmp2 == 0) return 1;
	else return 0;
}

void write_app_info(crc_t* checksum, size_t* size){
	iap_prepare_sectors(&status, addr_to_sector(0x3000), addr_to_sector(0x3fff));
	if(status != ok){
		uart_send_err();
	}
	iap_erase_sectors(&status, addr_to_sector(0x3000), addr_to_sector(0x3fff));
	if(status != ok){
		uart_send_err();
	}
	iap_copy_to_flash(&status, (long unsigned int *)crc_store, (long unsigned int*)checksum, 0x100);
	//0x100 a result of trial and error I initially wanted to put 0x4 but noooo it doesn't work
	iap_copy_to_flash(&status, (long unsigned int *)size_store, (long unsigned int*)size, 0x100);
}

void launch_program(){
	SCB->VTOR = APP_OFFSET;
	__asm("ldr r0,=0x4000");
	__asm("ldr sp,[r0]");
	__asm("ldr pc,[r0,#4]");
}

void manage_bootloader(){
	cmd.argv = (char**)argvs;
	int var = 1;
	char digit;
	set_colour(GREEN);
	while(var){
		uart_receive_command(buff);
		uart_parse_command(buff, &cmd);
		set_colour(GREEN);
		if ((state == S1) && (strncmp("GETID", cmd.argv[0], 5)==0)){
			uart_commands_getid(&status);

		}else if((state == S1) && (strncmp("GETSE", cmd.argv[0], 5)==0)){
			uart_commands_getserial(&status);

		}else if((state == S1) && (strncmp("PROG", cmd.argv[0], 4)==0)){
			crc_prog = uart_string_to_int(cmd.argv[3]);
			total_size = uart_string_to_int(cmd.argv[2]);
			offset = uart_string_to_int(cmd.argv[1]);
			write_app_info(&crc_prog, (unsigned int *)&total_size);
			crc_data = uart_commands_prog(&status, total_size);
			if(status == ok){
				state = S2;
			}
			offset_pointer = offset;

		}else if(((state == S2) || (state = S3)) && (strcmp("DATA", cmd.argv[0])==0)){
			int size = uart_string_to_int(cmd.argv[1]);
			crc_t local_checksum = uart_string_to_int(cmd.argv[2]);
			uart_commands_data(&status, size, &crc_data, local_checksum, offset_pointer);
			if(status == ok){
				offset_pointer += size;
				state = S3;
			}else{
				state = S2;
			}

		}else if((state == S3) && (strncmp("CHECK", cmd.argv[0], 5)==0)){
			error_code is_error = uart_commands_check(crc_prog, crc_data);
			if(is_error == ok){
				state = S1;
				uart_send_ok();
				uart_send("bla\r\n", 5);//putting bla here so OK will show up
				//removing bla for the moment will make ok not show up
				var = 0;
				LPC_GPIO2->FIOCLR0 = 0xff;
				launch_program();
			}else{
				state = S2;
				uart_send_err();
				set_colour(RED);
				itoa(is_error, &digit, 10);
				uart_send(&digit, 1);
				//flash LED red, and stays in the bootloader mode
			}
		}else {
			uart_send_cmd_illeg();
		}
	}
}

int main(void) {
    uart_init(115200);
	timer_0();
    if(is_b_pressed() == 0){
		if(is_code_valid()){
			launch_program();
		}else{
			manage_bootloader();
		}
	}else{
		manage_bootloader();
	}
    while(1);
    return 0 ;
}