banan-os/kernel/arch/x86_64/IDT.cpp

#include <BAN/Errors.h>
#include <kernel/IDT.h>
#include <kernel/InterruptController.h>
#include <kernel/kmalloc.h>
#include <kernel/Panic.h>

#define REGISTER_ISR_HANDLER(i) register_interrupt_handler(i, isr ## i)
#define REGISTER_IRQ_HANDLER(i) register_interrupt_handler(IRQ_VECTOR_BASE + i, irq ## i)

namespace IDT
{

	struct GateDescriptor
	{
		uint16_t offset1;
		uint16_t selector;
		uint8_t IST;
		uint8_t flags;
		uint16_t offset2;
		uint32_t offset3;
		uint32_t reserved;
	} __attribute__((packed));

	struct IDTR
	{
		uint16_t size;
		uint64_t offset;
	} __attribute__((packed));

	static IDTR				s_idtr;
	static GateDescriptor*	s_idt = nullptr;

	static void (*s_irq_handlers[0x10])() { nullptr };

	static const char* isr_exceptions[] =
	{
		"Division Error",
		"Debug",
		"Non-maskable Interrupt",
		"Breakpoint",
		"Overflow",
		"Bound Range Exception",
		"Invalid Opcode",
		"Device Not Available",
		"Double Fault",
		"Coprocessor Segment Overrun",
		"Invalid TSS",
		"Segment Not Present",
		"Stack-Segment Fault",
		"General Protection Fault",
		"Page Fault",
		"Unknown Exception 0x0F",
		"x87 Floating-Point Exception",
		"Alignment Check",
		"Machine Check",
		"SIMD Floating-Point Exception",
		"Virtualization Exception",
		"Control Protection Exception",
		"Unknown Exception 0x16",
		"Unknown Exception 0x17",
		"Unknown Exception 0x18",
		"Unknown Exception 0x19",
		"Unknown Exception 0x1A",
		"Unknown Exception 0x1B",
		"Hypervisor Injection Exception",
		"VMM Communication Exception",
		"Security Exception",
		"Unkown Exception 0x1F",
	};

	extern "C" void cpp_isr_handler(uint64_t isr, uint64_t error)
	{
		uint64_t rax, rbx, rcx, rdx, rsp, rbp;
		uint64_t cr0, cr2, cr3, cr4;
		asm volatile("":"=a"(rax),"=b"(rbx),"=c"(rcx),"=d"(rdx));
		asm volatile("movq %%rsp, %%rax":"=a"(rsp));
		asm volatile("movq %%rbp, %%rax":"=a"(rbp));
		asm volatile("movq %%cr0, %%rax":"=a"(cr0));
		asm volatile("movq %%cr2, %%rax":"=a"(cr2));
		asm volatile("movq %%cr3, %%rax":"=a"(cr3));
		asm volatile("movq %%cr4, %%rax":"=a"(cr4));

		Kernel::Panic(
			"{} (error code: 0x{16H})\r\n"
			"Register dump\r\n"
			"rax=0x{16H}, rbx=0x{16H}, rcx=0x{16H}, rdx=0x{16H}\r\n"
			"rsp=0x{16H}, rbp=0x{16H}\r\n"
			"CR0=0x{16H}, CR2=0x{16H}, CR3=0x{16H}, CR4=0x{16H}\r\n",
			isr_exceptions[isr], error, rax, rbx, rcx, rdx, rsp, rbp, cr0, cr2, cr3, cr4
		);
	}

	extern "C" void cpp_irq_handler(uint64_t irq)
	{
		if (s_irq_handlers[irq])
			s_irq_handlers[irq]();
		else
			dprintln("no handler for irq 0x{2H}\n", irq);

		InterruptController::Get().EOI(irq);
	}

	static void flush_idt()
	{
		asm volatile("lidt %0"::"m"(s_idtr));
	}

	static void register_interrupt_handler(uint8_t index, void (*f)())
	{
		GateDescriptor& descriptor = s_idt[index];
		descriptor.offset1 = (uint16_t)((uint64_t)f >> 0);
		descriptor.offset2 = (uint16_t)((uint64_t)f >> 16);
		descriptor.offset3 = (uint32_t)((uint64_t)f >> 32);

		descriptor.selector = 0x08;
		descriptor.IST = 0;
		descriptor.flags = 0x8E;
	}

	void register_irq_handler(uint8_t irq, void (*f)())
	{
		s_irq_handlers[irq] = f;
	}

	extern "C" void isr0();
	extern "C" void isr1();
	extern "C" void isr2();
	extern "C" void isr3();
	extern "C" void isr4();
	extern "C" void isr5();
	extern "C" void isr6();
	extern "C" void isr7();
	extern "C" void isr8();
	extern "C" void isr9();
	extern "C" void isr10();
	extern "C" void isr11();
	extern "C" void isr12();
	extern "C" void isr13();
	extern "C" void isr14();
	extern "C" void isr15();
	extern "C" void isr16();
	extern "C" void isr17();
	extern "C" void isr18();
	extern "C" void isr19();
	extern "C" void isr20();
	extern "C" void isr21();
	extern "C" void isr22();
	extern "C" void isr23();
	extern "C" void isr24();
	extern "C" void isr25();
	extern "C" void isr26();
	extern "C" void isr27();
	extern "C" void isr28();
	extern "C" void isr29();
	extern "C" void isr30();
	extern "C" void isr31();

	extern "C" void irq0();
	extern "C" void irq1();
	extern "C" void irq2();
	extern "C" void irq3();
	extern "C" void irq4();
	extern "C" void irq5();
	extern "C" void irq6();
	extern "C" void irq7();
	extern "C" void irq8();
	extern "C" void irq9();
	extern "C" void irq10();
	extern "C" void irq11();
	extern "C" void irq12();
	extern "C" void irq13();
	extern "C" void irq14();
	extern "C" void irq15();

	void initialize()
	{
		s_idt = (GateDescriptor*)kmalloc_eternal(0x100 * sizeof(GateDescriptor));
		memset(s_idt, 0x00, 0x100 * sizeof(GateDescriptor));

		s_idtr.offset = (uint64_t)s_idt;
		s_idtr.size = 0x100 * sizeof(GateDescriptor) - 1;

		REGISTER_ISR_HANDLER(0);
		REGISTER_ISR_HANDLER(1);
		REGISTER_ISR_HANDLER(2);
		REGISTER_ISR_HANDLER(3);
		REGISTER_ISR_HANDLER(4);
		REGISTER_ISR_HANDLER(5);
		REGISTER_ISR_HANDLER(6);
		REGISTER_ISR_HANDLER(7);
		REGISTER_ISR_HANDLER(8);
		REGISTER_ISR_HANDLER(9);
		REGISTER_ISR_HANDLER(10);
		REGISTER_ISR_HANDLER(11);
		REGISTER_ISR_HANDLER(12);
		REGISTER_ISR_HANDLER(13);
		REGISTER_ISR_HANDLER(14);
		REGISTER_ISR_HANDLER(15);
		REGISTER_ISR_HANDLER(16);
		REGISTER_ISR_HANDLER(17);
		REGISTER_ISR_HANDLER(18);
		REGISTER_ISR_HANDLER(19);
		REGISTER_ISR_HANDLER(20);
		REGISTER_ISR_HANDLER(21);
		REGISTER_ISR_HANDLER(22);
		REGISTER_ISR_HANDLER(23);
		REGISTER_ISR_HANDLER(24);
		REGISTER_ISR_HANDLER(25);
		REGISTER_ISR_HANDLER(26);
		REGISTER_ISR_HANDLER(27);
		REGISTER_ISR_HANDLER(28);
		REGISTER_ISR_HANDLER(29);
		REGISTER_ISR_HANDLER(30);
		REGISTER_ISR_HANDLER(31);

		REGISTER_IRQ_HANDLER(0);
		REGISTER_IRQ_HANDLER(1);
		REGISTER_IRQ_HANDLER(2);
		REGISTER_IRQ_HANDLER(3);
		REGISTER_IRQ_HANDLER(4);
		REGISTER_IRQ_HANDLER(5);
		REGISTER_IRQ_HANDLER(6);
		REGISTER_IRQ_HANDLER(7);
		REGISTER_IRQ_HANDLER(8);
		REGISTER_IRQ_HANDLER(9);
		REGISTER_IRQ_HANDLER(10);
		REGISTER_IRQ_HANDLER(11);
		REGISTER_IRQ_HANDLER(12);
		REGISTER_IRQ_HANDLER(13);
		REGISTER_IRQ_HANDLER(14);
		REGISTER_IRQ_HANDLER(15);

		flush_idt();
	}

}