From ced1752cf7cb45610ccfbecaa3c5fa77aa156f57 Mon Sep 17 00:00:00 2001
From: Ian Moffett <ian@osmora.org>
Date: Thu, 23 May 2024 15:03:02 -0400
Subject: kernel/amd64: mp: Add local timer stat

Signed-off-by: Ian Moffett <ian@osmora.org>
---
 sys/arch/amd64/amd64/cpu_mp.c     | 35 +++++++++++++++++++++++++++++++++++
 sys/arch/amd64/amd64/local_intr.S |  1 +
 2 files changed, 36 insertions(+)

(limited to 'sys/arch')

diff --git a/sys/arch/amd64/amd64/cpu_mp.c b/sys/arch/amd64/amd64/cpu_mp.c
index 0eb6dba..a289b04 100644
--- a/sys/arch/amd64/amd64/cpu_mp.c
+++ b/sys/arch/amd64/amd64/cpu_mp.c
@@ -34,6 +34,7 @@
 #include <sys/syslog.h>
 #include <sys/sched.h>
 #include <sys/cpu.h>
+#include <sys/intr.h>
 #include <vm/dynalloc.h>
 #include <assert.h>
 
@@ -47,17 +48,48 @@ static volatile struct limine_smp_request g_smp_req = {
 };
 
 static bool is_mp_supported = false;
+static struct intr_info *tmr_irqlist[MAXCPUS];
+
+void handle_local_tmr(void);
+
+/*
+ * Add local timer IRQ stat
+ */
+static inline void
+tmr_irqstat_add(struct cpu_info *ci)
+{
+    tmr_irqlist[ci->idx] = intr_info_alloc("LAPIC", "APICTMR");
+    tmr_irqlist[ci->idx]->affinity = ci->idx;
+    intr_register(tmr_irqlist[ci->idx]);
+}
+
+/*
+ * Update local timer IRQ stats from
+ * interrupt.
+ */
+void
+handle_local_tmr(void)
+{
+    struct cpu_info *ci = this_cpu();
+
+    ++tmr_irqlist[ci->idx]->count;
+}
 
 static void
 ap_trampoline(struct limine_smp_info *si)
 {
     static struct spinlock lock = {0};
+    struct cpu_info *ci;
 
     spinlock_acquire(&lock);
 
     pre_init();
     processor_init();
 
+    ci = this_cpu();
+    cpu_attach(ci);
+    tmr_irqstat_add(ci);
+
     spinlock_release(&lock);
     sched_enter();
 
@@ -88,6 +120,9 @@ ap_bootstrap(struct cpu_info *ci)
     cpus = resp->cpus;
     cpu_init_counter = resp->cpu_count - 1;
 
+    cpu_attach(ci);
+    tmr_irqstat_add(ci);
+
     if (resp->cpu_count == 1) {
         KINFO("CPU has 1 core, no APs to bootstrap...\n");
         return;
diff --git a/sys/arch/amd64/amd64/local_intr.S b/sys/arch/amd64/amd64/local_intr.S
index 6d69302..c2d53e9 100644
--- a/sys/arch/amd64/amd64/local_intr.S
+++ b/sys/arch/amd64/amd64/local_intr.S
@@ -41,6 +41,7 @@ lapic_tmr_isr:
     push_trapframe $0
     mov %rsp, %rdi
     call sched_context_switch
+    call handle_local_tmr
     call lapic_send_eoi
     pop_trapframe
     iretq
-- 
cgit v1.2.3