#pragma once
#include "common.h"

#define PROCS_MAX 8             // Maximum ammount of processes
#define PROC_UNUSED 0           // Unused processes control structure
#define PROC_RUNNABLE 1         // Runnable proccess
#define PROC_EXITED   2
#define SATP_SV32 (1u << 31)
#define SSTATUS_SPIE (1 << 5)
#define SCAUSE_ECALL 8
#define PAGE_V (1 << 0) // "Valid" bit (entry is allowed)
#define PAGE_R (1 << 1) // Readable
#define PAGE_W (1 << 2) // Writable
#define PAGE_X (1 << 3) // Executable
#define PAGE_U (1 << 4) // User (accessible in user mode)
#define USER_BASE 0x1000000

struct process {
    int pid;                    // ID of a process
    int state;                  // State of the process: either PROC_UNUSED or PROC_RUNNABLE
    vaddr_t sp;                 // Stack pointer
    uint32_t *page_table;       // points to first level page table
    uint8_t stack[8192];        // Kernel stack
};

struct sbiret {
    long error;
    long value;
};

struct trap_frame {
    uint32_t ra;
    uint32_t gp;
    uint32_t tp;
    uint32_t t0;
    uint32_t t1;
    uint32_t t2;
    uint32_t t3;
    uint32_t t4;
    uint32_t t5;
    uint32_t t6;
    uint32_t a0;
    uint32_t a1;
    uint32_t a2;
    uint32_t a3;
    uint32_t a4;
    uint32_t a5;
    uint32_t a6;
    uint32_t a7;
    uint32_t s0;
    uint32_t s1;
    uint32_t s2;
    uint32_t s3;
    uint32_t s4;
    uint32_t s5;
    uint32_t s6;
    uint32_t s7;
    uint32_t s8;
    uint32_t s9;
    uint32_t s10;
    uint32_t s11;
    uint32_t sp;
} __attribute__ ((packed));

#define READ_CSR(reg)             \
    ({                            \
        unsigned long __tmp;      \
        __asm__ __volatile__("csrr %0, " #reg : "=r"(__tmp)); \
        __tmp;                    \
    })

#define WRITE_CSR(reg, value)     \
    do {                          \
        uint32_t __tmp = (value); \
        __asm__ __volatile__("csrw " #reg ", %0 " ::"r"(__tmp)); \
    } while (0)

#define PANIC(fmt, ...)                                                                             \
    do {                                                                                            \
        printf("YOU GOT A PANIC ERROR: %s:%d: " fmt "\n", __FILE__, __LINE__, ##__VA_ARGS__);       \
        while (1) {}                                                                                \
    } while (0)