syscall message unmaps upon invokemain

2 files changed, 61 insertions, 0 deletions

diff --git a/arch/x86_64/memory/message.c b/arch/x86_64/memory/message.c
new file mode 100644
index 0000000..1fbdfc8
--- /dev/null
+++ b/arch/x86_64/memory/message.c
@@ -0,0 +1,48 @@
+#include <object.h>
+#include <stddef.h>
+#include <error.h>
+#include <device/processor.h>
+#include "arch/x86_64/page.h"
+#include "arch/x86_64/page-mapping.h"
+#include "print.h"
+
+int
+ko_message_unmap(objdir_entry_t *message, uintptr_t *saveptr)
+{
+    /* message data should point to the virtual address. */
+    /* If it doesn't, fail*/
+    if(!(message->extra & KODE_EX_MESSAGE_MAPPED)) return KE_BADCALL;
+    uintptr_t vptr = message->data;
+
+    processor_t *cproc = processor_current();
+    pmle_t *pml4 = vmem_phys_tovirt(cproc->pdir);
+
+    pmle_t *message_pmle = mem_mapping_vptr_mapping(pml4, 4, vptr);
+    *saveptr = (uintptr_t)vmem_phys_tovirt(message_pmle->paddr << 12);
+
+    message_pmle->p = 0;
+    __asm__ volatile("invlpg (%0)":: "r"(vptr): "memory");
+
+    message->extra &= ~KODE_EX_MESSAGE_MAPPED;
+    return 0;
+}
+
+int
+ko_message_remap(objdir_entry_t *message)
+{
+    if(message->extra & KODE_EX_MESSAGE_MAPPED) return KE_BADCALL;
+    uintptr_t vptr = message->data;
+    
+    processor_t *cproc = processor_current();
+    pmle_t *pml4 = vmem_phys_tovirt(cproc->pdir);
+
+    pmle_t *message_pmle = mem_mapping_vptr_mapping(pml4, 4, vptr);
+
+    message_pmle->p = 1;
+    __asm__ volatile("invlpg (%0)":: "r"(vptr): "memory");
+
+    message->extra |= KODE_EX_MESSAGE_MAPPED;
+    return 0;
+}
+
+void ko_message_move(objdir_entry_t *message, uintptr_t vptr);
diff --git a/arch/x86_64/memory/page-mapping.c b/arch/x86_64/memory/page-mapping.c
index 0de5bfa..d4b4e98 100644
--- a/arch/x86_64/memory/page-mapping.c
+++ b/arch/x86_64/memory/page-mapping.c
@@ -1,6 +1,7 @@
 #include "arch/x86_64/page-mapping.h"
 #include "arch/x86_64/page.h"
 #include <stddef.h>
+#include "print.h"
 
 pmle_t*
 page_mapping_traverse(pmle_t *pml4, uint8_t depth, uint16_t *path)
@@ -13,3 +14,15 @@ page_mapping_traverse(pmle_t *pml4, uint8_t depth, uint16_t *path)
     if(!pmle->p) return NULL;
     return page_mapping_traverse(pmle_table, depth - 1, path + 1);
 }
+
+pmle_t*
+mem_mapping_vptr_mapping(pmle_t *pml4, uint8_t depth, uintptr_t vptr)
+{
+    uint64_t pathval = 0;
+    uint16_t *path = (uint16_t*)&pathval;
+
+    for(uint8_t i = 0; i < depth; i++) {
+        path[i] = PML_I_FOR_LAYER(vptr, 4 - i);
+    }
+    return page_mapping_traverse(pml4, depth - 1, path);
+}