From 7f2adfc77d17abf19e843ed09ccd698ee541796d Mon Sep 17 00:00:00 2001
From: Marvin Borner
Date: Sat, 13 Mar 2021 15:12:25 +0100
Subject: Some security measures

---
 kernel/drivers/interrupts.c | 4 ++++
 kernel/features/syscall.c   | 2 ++
 2 files changed, 6 insertions(+)

(limited to 'kernel')

diff --git a/kernel/drivers/interrupts.c b/kernel/drivers/interrupts.c
index a0cd106..fe2321a 100644
--- a/kernel/drivers/interrupts.c
+++ b/kernel/drivers/interrupts.c
@@ -232,6 +232,10 @@ static void isr_install(void)
 	idt_set_gate(29, (u32)isr29, 0x08, 0x8E);
 	idt_set_gate(30, (u32)isr30, 0x08, 0x8E);
 	idt_set_gate(31, (u32)isr31, 0x08, 0x8E);
+
+	// Set default routines
+	for (u32 i = 0; i < 256; i++)
+		isr_routines[i] = isr_panic;
 }
 
 /**
diff --git a/kernel/features/syscall.c b/kernel/features/syscall.c
index dc49d67..68bcbaa 100644
--- a/kernel/features/syscall.c
+++ b/kernel/features/syscall.c
@@ -90,6 +90,7 @@ static void syscall_handler(struct regs *r)
 		switch (r->ecx) {
 		case SYS_BOOT_REBOOT:
 			print("Rebooting...\n");
+			outb(0x64, 0xfe);
 			__asm__ volatile("ud2");
 			break;
 		case SYS_BOOT_SHUTDOWN:
@@ -97,6 +98,7 @@ static void syscall_handler(struct regs *r)
 			outw(0xB004, 0x2000);
 			outw(0x604, 0x2000);
 			outw(0x4004, 0x3400);
+			outb(0x64, 0xfe);
 			__asm__ volatile("ud2");
 			break;
 		default:
-- 
cgit v1.2.3