RockOS/kernel/interrupts.c

#include <stdint.h>

#include <lib/print.h>

#include "syscall.h"

#include <drivers/pic/pic.h>
#include <drivers/ps2/ps2.h>

#define IDT_MAX_DESCRIPTORS 256
#define ATTR_KERN_32 0x8E // active(1), ring 0 (00), interrupt gate(0), 32 bit (1110)
#define ATTR_USER_32 0xEE // active, ring 3, 32bit
#define SEGMENT_KERN 0x08

typedef struct __attribute__((packed)) {
    uint16_t isr_low; 
    uint16_t selector; 
    uint8_t  reserved;  
    uint8_t  attributes;
    uint16_t isr_high; 
} idt_entry;

typedef struct __attribute__((packed)) {
    uint16_t limit;
    uint32_t base;
} idt_ptr;

typedef struct __attribute__((packed)) {
    uint32_t edi, esi, ebp, esp_dummy, ebx, edx, ecx, eax;
    
    uint32_t int_no;    
    uint32_t error_code;
    
    uint32_t eip, cs, eflags, useresp, ss;
} registers;

// (interrupts.S)
extern void load_idt(void);
extern void isr0();
extern void isr1();
extern void isr8();
extern void isr13();
extern void isr14();
extern void isr32();
extern void isr33();

static void* isr_stub_table[256] = {
    [0] = isr0,
    [1] = isr1,
    [8] = isr8,
    [13] = isr13,
    [14] = isr14,
    [32] = isr32,
    [33] = isr33,
};

static const char* err_messages[32] = {
    [0] = "Divide by 0",
    [1] = "",
    [8] = "",
    [13] = "General Protection Fault",
    [14] = "Page Fault",
};

idt_entry idt[IDT_MAX_DESCRIPTORS];
idt_ptr idtr;

static void set_idtr(uint8_t vector, void* isr, uint16_t sel, uint8_t attributes) {
    idt_entry* desc = &idt[vector];

    desc->isr_low    = (uint32_t)isr & 0xFFFF;
    desc->selector  = SEGMENT_KERN;
    desc->reserved   = 0;
    desc->attributes = attributes;
    desc->isr_high   = ((uint32_t)isr >> 16) & 0xFFFF;
}

void init_idt() {
    idtr.limit = (sizeof(idt_entry) * IDT_MAX_DESCRIPTORS) - 1;
    idtr.base  = (uint32_t)&idt;

    for (int i = 0; i < IDT_MAX_DESCRIPTORS; i++) {
        if (isr_stub_table[i] != 0) {
            set_idtr(i, isr_stub_table[i], SEGMENT_KERN, ATTR_KERN_32);
        } else {
            set_idtr(i, 0, SEGMENT_KERN, 0);
        }
    }

    set_idtr(0x80, syscall_handler, SEGMENT_KERN, ATTR_USER_32);

    load_idt();
}

void common_interrupt_handler(registers *args) {
    switch (args->int_no) {
        case 0:
        case 1:
        case 2:
        case 3:
        case 4:
        case 5:
        case 6:
        case 7:
        case 8:
        case 9:
        case 10:
        case 11:
        case 12:
        case 13:
        case 15:
        case 16:
            kprintf("CPU Error: %s", err_messages[args->int_no]);
            break;
        case 14:{
            uint32_t faulting_address;
            asm volatile("mov %%cr2, %0" : "=r"(faulting_address));
            kprintf("\n--- PAGE FAULT ---\n");
            kprintf("Faulted Virtual Address: 0x%x\n", faulting_address);
            kprintf("Error Code: 0x%x\n", args->error_code);
            kprintf("Instruction Pointer: 0x%x\n", args->eip);
            kprintf("Caused by: %s, %s, running in %s mode\n",
                (args->error_code & 0x01) ? "Protection Violation" : "Page Not Present",
                (args->error_code & 0x02) ? "Write" : "Read",
                (args->error_code & 0x04) ? "User" : "Supervisor");
            while(1);
            break;
        }
        case 33: {
            uint8_t data = ps2_read_data();
            send_eoi_master();
            break;
        }
        case 46: { // IDE PRIMARY
            send_eoi_slave();
            break;
        }
        case 47: {  // IDE SLAVE
            send_eoi_slave();
            break;
        }
        default:
            kprintf("Received unhandled interrupt: %d\n", args->int_no);
            return;
    }
}