From 22743edeb7cee283229254f1a9076d914a1a1978 Mon Sep 17 00:00:00 2001
From: Max Wash <mbswash@gmail.com>
Date: Thu, 9 Mar 2023 19:50:22 +0000
Subject: [PATCH] pmap: add get_kernel_pmap()

---
 arch/x86_64/pmap.c   | 5 +++++
 include/socks/pmap.h | 1 +
 2 files changed, 6 insertions(+)

diff --git a/arch/x86_64/pmap.c b/arch/x86_64/pmap.c
index f1636ea..7d5ce20 100644
--- a/arch/x86_64/pmap.c
+++ b/arch/x86_64/pmap.c
@@ -222,6 +222,11 @@ static kern_status_t do_pmap_add(pmap_t pmap, void *p, pfn_t pfn, vm_prot_t prot
 	return KERN_OK;
 }
 
+pmap_t get_kernel_pmap(void)
+{
+	return kernel_pmap;
+}
+
 void pmap_bootstrap(void)
 {
 	can_use_gbpages = gigabyte_pages();
diff --git a/include/socks/pmap.h b/include/socks/pmap.h
index 6216ed6..b225023 100644
--- a/include/socks/pmap.h
+++ b/include/socks/pmap.h
@@ -18,6 +18,7 @@ typedef enum pmap_flags {
 } pmap_flags_t;
 
 extern void pmap_bootstrap(void);
+extern pmap_t get_kernel_pmap(void);
 
 extern pmap_t pmap_create(void);
 extern void pmap_destroy(pmap_t pmap);