#include <string.h>
#include "utils.h"
#include "mmu.h"
#include "memory_map.h"
#include "arm.h"
#include "cpu_context.h"

extern uint8_t __pk2ldr_start__[], __pk2ldr_end__[];

/* start.s */
void __attribute__((noreturn)) __jump_to_lower_el(uint64_t arg, uintptr_t ep, unsigned int el);

void coldboot_main(void) {
    uintptr_t *mmu_l3_table = (uintptr_t *)TZRAM_GET_SEGMENT_ADDRESS(TZRAM_SEGMENT_ID_L3_TRANSLATION_TABLE);
    uintptr_t pk2ldr = TZRAM_GET_SEGMENT_ADDRESS(TZRAM_SEGMENT_ID_PK2LDR);
    uintptr_t ep;
    uint64_t arg;

    /* Clear and unmap pk2ldr (which is reused as exception entry stacks) */
    memset((void *)pk2ldr, 0, __pk2ldr_end__ - __pk2ldr_start__);
    mmu_unmap_range(3, mmu_l3_table, pk2ldr, __pk2ldr_end__ - __pk2ldr_start__);
    tlb_invalidate_all_inner_shareable();

    use_core_entrypoint_and_argument(get_core_id(), &ep, &arg);

    /* Nintendo jumps to EL1, we jump to EL2. Both are supported with all current packages2. */
    /* TODO: Remove this panic() when we're ready to test Kernel handoff. */
    panic(0x7A700001); 
    __jump_to_lower_el(arg, ep, 2);
}