From c478ba925b29990df2426ee06c2e629c0a2d7246 Mon Sep 17 00:00:00 2001
From: Ian Moffett <ian@osmora.org>
Date: Wed, 19 Nov 2025 22:57:27 -0500
Subject: kern/amd64: sched: Implement context switching

Signed-off-by: Ian Moffett <ian@osmora.org>
---
 sys/arch/amd64/cpu/idt.S |  5 ++++-
 sys/arch/amd64/cpu/mp.c  | 17 ++++++++++-------
 2 files changed, 14 insertions(+), 8 deletions(-)

(limited to 'sys/arch/amd64/cpu')

diff --git a/sys/arch/amd64/cpu/idt.S b/sys/arch/amd64/cpu/idt.S
index 729d958..cf31fbb 100644
--- a/sys/arch/amd64/cpu/idt.S
+++ b/sys/arch/amd64/cpu/idt.S
@@ -309,9 +309,12 @@ page_fault:
     .globl lapic_tmr_isr
 lapic_tmr_isr:
     KFENCE
+    subq $8, %rsp
     push_frame 0x81
-    nop
+    mov %rsp, %rdi
+    call mu_process_switch
     pop_frame 0x81
+    add $8, %rsp
     KFENCE
     iretq
 
diff --git a/sys/arch/amd64/cpu/mp.c b/sys/arch/amd64/cpu/mp.c
index 6911a49..ea36829 100644
--- a/sys/arch/amd64/cpu/mp.c
+++ b/sys/arch/amd64/cpu/mp.c
@@ -40,6 +40,7 @@
 #include <md/lapic.h>
 #include <md/msr.h>
 #include <md/cpu.h>
+#include <md/gdt.h>
 #include <mu/cpu.h>
 #include <os/process.h>
 #include <os/sched.h>
@@ -116,10 +117,11 @@ static volatile uint32_t aps_up = 0;
 __section(".trampoline") static char ap_code[4096];
 
 static void
-cpu_idle(void)
+cpu_idle(struct mcb *mcb)
 {
+    lapic_oneshot_usec(mcb, SCHED_QUANTUM);
     for (;;) {
-        __asmv("hlt");
+        __asmv("sti; hlt");
     }
 }
 
@@ -308,7 +310,7 @@ cpu_lm_entry(void)
     );
 
     idt_load();
-    cpu_idle();
+    cpu_idle(&ci->mcb);
     __builtin_unreachable();
 }
 
@@ -391,9 +393,10 @@ static void
 cpu_start_idle(void)
 {
     struct process *p;
+    struct cpu_info *core;
     int error;
 
-    for (size_t i = 0; i < ap_count; ++i) {
+    for (size_t i = 0; i < ap_count + 1; ++i) {
         p = kalloc(sizeof(*p));
         if (p == NULL) {
             panic("mp: could not allocate idle thread\n");
@@ -405,7 +408,8 @@ cpu_start_idle(void)
             panic("mp: could not initialize process\n");
         }
 
-        sched_enqueue_proc(p);
+        core = sched_enqueue_proc(p);
+        core->curproc = NULL;
     }
 }
 
@@ -451,6 +455,7 @@ cpu_start_aps(struct cpu_info *ci)
     /* Copy the bring up code to the BUA */
     bua = AP_BUA_VADDR;
     memcpy(bua, ap_code, AP_BUA_LEN);
+    cpu_start_idle();
 
     /* Start up the APs */
     mcb = &self->mcb;
@@ -471,6 +476,4 @@ cpu_start_aps(struct cpu_info *ci)
     } else {
         dtrace("%d processor(s) up\n", aps_up);
     }
-
-    cpu_start_idle();
 }
-- 
cgit v1.2.3