From 739924cb5cb84e8f29c90b4accd0910daa58a579 Mon Sep 17 00:00:00 2001
From: Ian Moffett <ian@osmora.org>
Date: Sat, 20 Sep 2025 17:13:57 -0400
Subject: kern/amd64: mmu: Add page-level cacheability attrs

This commit adds support for page-level cacheability attributes. We have
added the pmap_set_cache() as well as some MMU_CACHE_* bits

Signed-off-by: Ian Moffett <ian@osmora.org>
---
 src/sys/arch/amd64/cpu/mmu.c | 45 ++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 45 insertions(+)

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

diff --git a/src/sys/arch/amd64/cpu/mmu.c b/src/sys/arch/amd64/cpu/mmu.c
index 2b98d78..dad22dc 100644
--- a/src/sys/arch/amd64/cpu/mmu.c
+++ b/src/sys/arch/amd64/cpu/mmu.c
@@ -404,6 +404,51 @@ mmu_free_vas(struct vm_vas *vas)
     return 0;
 }
 
+/*
+ * Set caching attributes
+ */
+int
+pmap_set_cache(struct vm_vas *vas, vaddr_t va, cacheattr_t attr)
+{
+    uintptr_t *pte, pa;
+    uint32_t flags;
+    int error;
+    size_t idx;
+
+    /* We'll modify page-level attributes */
+    error = mmu_read_level(
+        vas, va, MMU_TBL,
+        &pte, true
+    );
+
+    if (error < 0) {
+        return -EINVAL;
+    }
+
+    /* Uncachable? */
+    if (ISSET(attr, MMU_CACHE_UC)) {
+        flags |= PTE_PCD;
+        flags &= ~PTE_PWT;
+        attr &= ~MMU_CACHE_WT;
+    }
+
+    /* Write through? */
+    if (ISSET(attr, MMU_CACHE_WT)) {
+        flags &= ~PTE_PCD;
+        flags |= PTE_PWT;
+    }
+
+    /* Not global? */
+    if (!ISSET(attr, MMU_CACHE_GL)) {
+        flags &= ~PTE_GLOBAL;
+    }
+
+    /* Update attributes and flush the TLB */
+    pte[idx] = pa | flags;
+    __invlpg((void *)va);
+    return 0;
+}
+
 /*
  * Verify that we are in a known state
  */
-- 
cgit v1.2.3