FIXED

Author: PythonPlumber

include/arch/x86/mmu.h

@@ -20,6 +20,7 @@ void mmu_init(void);
 // Map a page
 void mmu_map_page(uintptr_t virt, uintptr_t phys, uint32_t flags);
 void mmu_map_page_dir(uintptr_t* dir, uintptr_t virt, uintptr_t phys, uint32_t flags);
+void mmu_map_page_4mb(uintptr_t virt, uintptr_t phys, uint32_t flags);
 
 // Unmap a page
 void mmu_unmap_page(uintptr_t virt);

include/cpu.h

@@ -42,7 +42,6 @@ static inline void tlb_flush(uintptr_t addr) { mmu_tlb_flush(addr); }
 static inline void tlb_flush_all(void) { mmu_tlb_flush_all(); }
 
 // Paging control
-static inline void paging_switch_directory(uintptr_t dir_phys) { mmu_switch_space(dir_phys); }
 
 // Halt the CPU
 void cpu_halt(void);

include/paging.h

@@ -19,6 +19,7 @@ int paging_clone_cow_range(uint32_t* parent, uint32_t* child, uint32_t start, ui
 static inline void paging_init(void) { mmu_init(); vmm_init(); }
 static inline void map_page(uint32_t virtual_addr, uint32_t physical_addr, uint32_t flags) { mmu_map_page(virtual_addr, physical_addr, flags); }
 static inline void map_page_dir(uint32_t* dir, uint32_t virtual_addr, uint32_t physical_addr, uint32_t flags) { mmu_map_page_dir((uintptr_t*)dir, virtual_addr, physical_addr, flags); }
+static inline void map_page_4mb(uint32_t virtual_addr, uint32_t physical_addr, uint32_t flags) { mmu_map_page_4mb(virtual_addr, physical_addr, flags); }
 static inline void vmm_map_page(uint32_t virtual_addr, uint32_t physical_addr, uint32_t flags) { mmu_map_page(virtual_addr, physical_addr, flags); }
 static inline void vmm_unmap_page(uint32_t virtual_addr) { mmu_unmap_page(virtual_addr); }
 static inline void unmap_page_dir(uint32_t* dir, uint32_t virtual_addr) { mmu_unmap_page_dir((uintptr_t*)dir, virtual_addr); }

src/ai/ai_model.c

@@ -57,7 +57,7 @@ void ai_scheduler_wait(ai_job_t* job) {
     }
 }
 
-static void ai_worker_loop(void* arg) {
+static void __attribute__((unused)) ai_worker_loop(void* arg) {
     (void)arg;
     while (ai_scheduler_running) {
         ai_job_t* job = 0;

src/arch/x86/mmu.c

@@ -139,7 +139,7 @@ void mmu_init(void) {
         }
     }
 
-    paging_switch_directory((uintptr_t)page_directory);
+    mmu_switch_space((uintptr_t)page_directory);
 
     mmu_enable_pse();
     mmu_enable_paging();
@@ -173,6 +173,16 @@ void mmu_map_page(uintptr_t virt, uintptr_t phys, uint32_t flags) {
     mmu_map_page_dir((uintptr_t*)page_directory, virt, phys, flags);
 }
 
+void mmu_map_page_4mb(uintptr_t virt, uintptr_t phys, uint32_t flags) {
+    uint32_t irq_flags = spin_lock_irqsave(&paging_lock);
+    uintptr_t aligned_virt = virt & 0xFFC00000;
+    uintptr_t aligned_phys = phys & 0xFFC00000;
+    uint32_t pd_index = aligned_virt >> 22;
+    page_directory[pd_index] = aligned_phys | flags | PAGE_PRESENT | PAGE_PS;
+    mmu_tlb_flush(aligned_virt);
+    spin_unlock_irqrestore(&paging_lock, irq_flags);
+}
+
 void mmu_unmap_page_dir(uintptr_t* dir, uintptr_t virt) {
     if (!dir) return;
     uint32_t irq_flags = spin_lock_irqsave(&paging_lock);