#include <stdio.h>
#include <string.h>
#include "types.h"
#include "mem_map.h"
#include "pxi.h"
#include "arm9/console.h"
#include "fatfs/ff.h"
#include "arm9/fsutils.h"
#include "arm9/main.h"
#include "arm9/interrupt.h"

static u32 debugHash = 0;


void debugHashCodeRoData()
{
#ifndef NDEBUG
	extern u32 *__start__;
	extern u32 *__exidx_start;
	
	u32 *start = __start__;
	u32 *end = __exidx_start;
	
	debugHash = 0;
	
	for(u32 *ptr = start; ptr < end; ptr++)
		debugHash += *ptr;
#endif
}

noreturn void panic()
{
	register u32 lr __asm__("lr");

	consoleInit(0, NULL, true);

	printf("\x1b[41m\x1b[0J\x1b[9C****PANIC!!!****\n\nlr = 0x%08" PRIX32 "\n", lr);
	
	fsUnmountAll();
	devs_close();
	
	PXI_sendWord(PXI_CMD_ALLOW_POWER_OFF);

	while(1) waitForInterrupt();
}

noreturn void panicMsg(const char *msg)
{
	register u32 lr __asm__("lr");

	consoleInit(0, NULL, true);

	printf("\x1b[41m\x1b[0J\x1b[9C****PANIC!!!****\n\nlr = 0x%08" PRIX32 "\n", lr);
	printf("\nERROR MESSAGE:\n%s\n", msg);
	
	fsUnmountAll();
	devs_close();
	
	PXI_sendWord(PXI_CMD_ALLOW_POWER_OFF);

	while(1) waitForInterrupt();
}

// Expects the registers in the exception stack to be in the following order:
// r0-r14, pc (unmodified), cpsr
noreturn void guruMeditation(u8 type, const u32 *excStack)
{
	const char *const typeStr[3] = {"Undefined instruction", "Prefetch abort", "Data abort"};
	u32 realPc, instSize = 4;
	bool codeChanged = false;


	// verify text and rodata
	u32 prevHash = debugHash;
	debugHashCodeRoData();
	if(prevHash != debugHash)
		codeChanged = true;

	consoleInit(0, NULL, true);

	if(excStack[16] & 0x20) instSize = 2; // Processor was in Thumb mode?
	if(type == 2) realPc = excStack[15] - (instSize * 2); // Data abort
	else realPc = excStack[15] - instSize; // Other

	printf("\x1b[41m\x1b[0J\x1b[9CGuru Meditation Error!\n\n%s:\n", typeStr[type]);
	printf("CPSR: 0x%08" PRIX32 "\n"
	       "r0 = 0x%08" PRIX32 " r8  = 0x%08" PRIX32 "\n"
	       "r1 = 0x%08" PRIX32 " r9  = 0x%08" PRIX32 "\n"
	       "r2 = 0x%08" PRIX32 " r10 = 0x%08" PRIX32 "\n"
	       "r3 = 0x%08" PRIX32 " r11 = 0x%08" PRIX32 "\n"
	       "r4 = 0x%08" PRIX32 " r12 = 0x%08" PRIX32 "\n"
	       "r5 = 0x%08" PRIX32 " sp  = 0x%08" PRIX32 "\n"
	       "r6 = 0x%08" PRIX32 " lr  = 0x%08" PRIX32 "\n"
	       "r7 = 0x%08" PRIX32 " pc  = 0x%08" PRIX32 "\n\n",
	       excStack[16],
	       excStack[0], excStack[8],
	       excStack[1], excStack[9],
	       excStack[2], excStack[10],
	       excStack[3], excStack[11],
	       excStack[4], excStack[12],
	       excStack[5], excStack[13],
	       excStack[6], excStack[14],
	       excStack[7], realPc);

	puts("Stack dump:");

	u32 sp = excStack[13];
	if(sp >= DTCM_BASE && sp < DTCM_BASE + DTCM_SIZE && !(sp & 3u))
	{
		u32 stackWords = ((DTCM_BASE + DTCM_SIZE - sp) / 4 > 45 ? 45 : (DTCM_BASE + DTCM_SIZE - sp) / 4);

		u32 newlineCounter = 0;
		for(u32 i = 0; i < stackWords; i++)
		{
			if(newlineCounter == 3) {printf("\n"); newlineCounter = 0;}
			printf("0x%08" PRIX32 " ", ((u32*)sp)[i]);
			newlineCounter++;
		}
	}

	if(codeChanged) printf("Attention: RO section data changed!!");

	// avoid fs corruptions
	fsUnmountAll();
	devs_close();

	PXI_sendWord(PXI_CMD_ALLOW_POWER_OFF);

	while(1) waitForInterrupt();
}

void dumpMem(u8 *mem, u32 size, char *filepath)
{
	FIL file;
	UINT bytesWritten;

	if(f_open(&file, filepath, FA_CREATE_ALWAYS | FA_WRITE) != FR_OK)
		return;
		
	f_write(&file, mem, size, &bytesWritten);
	
	f_sync(&file);

	f_close(&file);
}