kernel/irq: use a lookup table for irqs

5 files changed, 22 insertions, 74 deletions

diff --git a/src/kernel/arch/amd64/interrupts/idt.c b/src/kernel/arch/amd64/interrupts/idt.c
index 622ac1a..8f37cb9 100644
--- a/src/kernel/arch/amd64/interrupts/idt.c
+++ b/src/kernel/arch/amd64/interrupts/idt.c
@@ -1,5 +1,5 @@
 #include <kernel/arch/amd64/boot.h>
-#include <kernel/arch/amd64/interrupts/isr.h>
+#include <kernel/arch/amd64/interrupts.h>
 #include <kernel/panic.h>
 #include <stdbool.h>
 #include <stdint.h>
@@ -30,7 +30,6 @@ static struct lidt_arg lidt_arg;
 
 static void idt_prepare(void);
 static void idt_load(void);
-static void idt_test(void);
 
 
 static void idt_prepare(void) {
@@ -55,13 +54,7 @@ static void idt_load(void) {
 	asm("lidt (%0)" : : "r" (&lidt_arg) : "memory");
 }
 
-static void idt_test(void) {
-	asm("int $0x34" : : : "memory");
-	assert(isr_test_interrupt_called);
-}
-
 void idt_init(void) {
 	idt_prepare();
 	idt_load();
-	idt_test();
 }
diff --git a/src/kernel/arch/amd64/interrupts/irq.c b/src/kernel/arch/amd64/interrupts/irq.c
index 818ea8e..e900450 100644
--- a/src/kernel/arch/amd64/interrupts/irq.c
+++ b/src/kernel/arch/amd64/interrupts/irq.c
@@ -1,4 +1,4 @@
-#include <kernel/arch/amd64/interrupts/irq.h>
+#include <kernel/arch/amd64/interrupts.h>
 #include <kernel/arch/amd64/port_io.h>
 #include <stdint.h>
 
diff --git a/src/kernel/arch/amd64/interrupts/irq.h b/src/kernel/arch/amd64/interrupts/irq.h
deleted file mode 100644
index f949824..0000000
--- a/src/kernel/arch/amd64/interrupts/irq.h
+++ /dev/null
@@ -1,14 +0,0 @@
-#pragma once
-#include <stdbool.h>
-#include <stdint.h>
-
-#define IRQ_IBASE 0x20
-#define IRQ_PIT 0
-#define IRQ_PS2KB 1
-#define IRQ_COM1 4
-#define IRQ_PS2MOUSE 12
-
-#define IRQ_RTL8139 11
-
-void irq_init(void);
-void irq_eoi(uint8_t line);
diff --git a/src/kernel/arch/amd64/interrupts/isr.c b/src/kernel/arch/amd64/interrupts/isr.c
index 1059530..ff00f2f 100644
--- a/src/kernel/arch/amd64/interrupts/isr.c
+++ b/src/kernel/arch/amd64/interrupts/isr.c
@@ -1,21 +1,15 @@
-#include <kernel/arch/amd64/driver/ps2.h>
-#include <kernel/arch/amd64/driver/rtl8139.h>
-#include <kernel/arch/amd64/driver/serial.h>
-#include <kernel/arch/amd64/interrupts/irq.h>
-#include <kernel/arch/amd64/interrupts/isr.h>
+#include <kernel/arch/amd64/interrupts.h>
 #include <kernel/arch/amd64/port_io.h>
-#include <kernel/arch/amd64/time.h>
 #include <kernel/arch/generic.h>
-#include <kernel/mem/alloc.h>
 #include <kernel/panic.h>
 #include <kernel/proc.h>
 #include <stdbool.h>
 #include <stdint.h>
 
-bool isr_test_interrupt_called = false;
+void (*irq_fn[16])(void) = {0};
 
 static void log_interrupt(int interrupt, uint64_t *stackframe) {
-	kprintf("interrupt %x, rip = k/%08x, cs 0x%x, code 0x%x\n",
+	kprintf("interrupt %d, rip = k/%08x, cs 0x%x, code 0x%x\n",
 			interrupt, stackframe[0], stackframe[1], stackframe[-1]);
 	if ((stackframe[1] & 0x3) == 0) {
 		uint64_t *stack = (void*)stackframe[3];
@@ -28,42 +22,25 @@ static void log_interrupt(int interrupt, uint64_t *stackframe) {
 	}
 }
 
-void isr_stage3(int interrupt, uint64_t *stackframe) {
-	if (interrupt == 0xe || interrupt == 0xd) stackframe++;
-	switch (interrupt) {
-		case 0x34:
-			isr_test_interrupt_called = true;
-			return;
-
-		case IRQ_IBASE + IRQ_PIT:
-			pit_irq();
-			irq_eoi(IRQ_PIT);
-			return;
-
-		case IRQ_IBASE + IRQ_PS2KB:
-		case IRQ_IBASE + IRQ_PS2MOUSE:
-			ps2_irq();
-			irq_eoi(interrupt - IRQ_IBASE);
-			return;
-
-		case IRQ_IBASE + IRQ_COM1:
-			serial_irq();
-			irq_eoi(IRQ_COM1);
+void isr_stage3(uint8_t interrupt, uint64_t *stackframe) {
+	uint8_t irqn = interrupt - IRQ_IBASE;
+	if (irqn < 16) {
+		if (irq_fn[irqn]) {
+			irq_fn[irqn]();
+			irq_eoi(irqn);
 			return;
+		}
+	}
 
-		case IRQ_IBASE + IRQ_RTL8139:
-			rtl8139_irq();
-			irq_eoi(interrupt - IRQ_IBASE);
-			return;
+	if (interrupt == 0xe || interrupt == 0xd) {
+		stackframe++;
+	}
 
-		default:
-			if ((stackframe[1] & 0x3) == 0) {
-				// mem_debugprint();
-				log_interrupt(interrupt, stackframe);
-				cpu_halt();
-			} else {
-				proc_kill(proc_cur, interrupt);
-				proc_switch_any();
-			}
+	if ((stackframe[1] & 0x3) == 0) { /* in kernel */
+		log_interrupt(interrupt, stackframe);
+		cpu_halt();
+	} else { /* in user */
+		proc_kill(proc_cur, interrupt);
+		proc_switch_any();
 	}
 }
diff --git a/src/kernel/arch/amd64/interrupts/isr.h b/src/kernel/arch/amd64/interrupts/isr.h
deleted file mode 100644
index 01f5236..0000000
--- a/src/kernel/arch/amd64/interrupts/isr.h
+++ /dev/null
@@ -1,8 +0,0 @@
-#pragma once
-#include <stdbool.h>
-#include <stdint.h>
-
-extern bool isr_test_interrupt_called; // used in the self-test in idt.c
-extern const char _isr_stubs;
-
-void isr_stage3(int interrupt, uint64_t *stackframe);