summaryrefslogtreecommitdiff
path: root/src/kernel/trap/handler.cc
blob: ecb6652ad2be3fa741858da8d19d05b87d270cbb (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
#include "kernel/trap.h"
#include "common/types.h"
#include "drivers/uart.h"
#include "kernel/proc.h"
#include "kernel/timer.h"


void panic(const char *s)
{
    Drivers::uart_puts("\n\n!!! KERNEL PANIC !!!\n");
    Drivers::uart_puts(s);
    Drivers::uart_puts("\nSystem Halted.\n");
    while (1)
        ;
}

namespace Trap
{
    void init()
    {
        uint64 stvec_val = (uint64)kernelvec;
        asm volatile("csrw stvec, %0" : : "r"(stvec_val));

        uint64 sstatus;
        asm volatile("csrr %0, sstatus" : "=r"(sstatus));
        sstatus &= ~(1L << 1);
        asm volatile("csrw sstatus, %0" : : "r"(sstatus));

        Drivers::uart_puts("[Trap] Initialized stvec to: ");
        Drivers::print_hex(stvec_val);
        Drivers::uart_puts("\n");
    }

    extern "C" void kerneltrap(struct Trapframe *tf)
    {
        uint64 scause, stval, sepc;

        asm volatile("csrr %0, scause" : "=r"(scause));
        asm volatile("csrr %0, stval" : "=r"(stval));
        asm volatile("csrr %0, sepc" : "=r"(sepc));

        if (scause & (1L << 63))
        {
            uint64 cause_code = scause & 0x7FFFFFFFFFFFFFFFL;
            switch (cause_code)
            {
            case 1:
                Drivers::uart_puts("[IRQ] Software Interrupt\n");
                asm volatile("csrc sip, %0" : : "r"(1 << 1));
                break;

            case 5:
                Drivers::uart_puts("[IRQ] Timer Interrupt\n");
                Timer::set_next_trigger();
                ProcManager::yield();
                break;
            
            case 9:
                Drivers::uart_puts("[IRQ] External Interrupt\n");
                break;

            default:
                Drivers::uart_puts("[IRQ] Unknown Interrupt: ");
                Drivers::print_hex(cause_code);
                Drivers::uart_puts("\n");
                break;
            }
        }
        else
        {
            switch (scause)
            {
            case 8:
                Drivers::uart_puts("[Syscall] ecall for User Mode\n");
                tf->epc += 4;
                break;
            
            case 12:
            case 13:
            case 15:
                Drivers::uart_puts("\n--- PAGE FAULT ---\n");
                Drivers::uart_puts("Cause Code: ");
                Drivers::print_hex(scause);
                Drivers::uart_puts("\n");
                Drivers::uart_puts("Fault Addr (stval): ");
                Drivers::print_hex(stval);
                Drivers::uart_puts("\n");
                Drivers::uart_puts("Instruction (sepc): ");
                Drivers::print_hex(sepc);
                Drivers::uart_puts("\n");

                if (stval == 0)
                {
                    panic("Null Pointer Dereference!");
                }
                else if (stval < 0x80000000)
                {
                    panic("Invalid Low Memory Access!");
                }
                else
                {
                    panic("Segmentation Fault (Unmapped Area)");
                }
                break;

            case 2:
                Drivers::uart_puts("\n--- ILLEGAL INSTRUCTION ---\n");
                Drivers::uart_puts("Instruction (sepc): ");
                Drivers::print_hex(sepc);
                Drivers::uart_puts("\n");
                panic("CPU encountered an unknown opcode");
                break;

            default:
                Drivers::uart_puts("\n--- UNHANDLED EXCEPTION ---\n");
                Drivers::uart_puts("Cause: ");
                Drivers::print_hex(scause);
                Drivers::uart_puts("\n");
                Drivers::uart_puts("sepc: ");
                Drivers::print_hex(sepc);
                Drivers::uart_puts("\n");
                panic("System Halt");
                break;
            }
        }
    }
}