X-Git-Url: https://scm.lunaixsky.com/lunaix-os.git/blobdiff_plain/2e21eb2f39dd80aa166216381d5d402be943686e..6f6da1abb22dff69dbb710bd2fd9d95f083f2b43:/lunaix-os/kernel/mm/cow.c

diff --git a/lunaix-os/kernel/mm/cow.c b/lunaix-os/kernel/mm/cow.c
index 85dcd49..a12d6e9 100644
--- a/lunaix-os/kernel/mm/cow.c
+++ b/lunaix-os/kernel/mm/cow.c
@@ -1,19 +1,24 @@
+#include <lunaix/mm/pmm.h>
 #include <lunaix/mm/vmm.h>
 
-void* vmm_dup_page(pid_t pid, void* pa) {    
-    void* new_ppg = pmm_alloc_page(pid, 0);
-    vmm_fmap_page(pid, PG_MOUNT_3, new_ppg, PG_PREM_RW);
-    vmm_fmap_page(pid, PG_MOUNT_4, pa, PG_PREM_RW);
+#include <sys/mm/mempart.h>
 
-    asm volatile (
-        "movl %1, %%edi\n"
-        "movl %2, %%esi\n"
-        "rep movsl\n"
-        :: "c"(1024), "r"(PG_MOUNT_3), "r"(PG_MOUNT_4)
-        : "memory", "%edi", "%esi");
+ptr_t
+vmm_dup_page(pid_t pid, ptr_t pa)
+{
+    ptr_t new_ppg = pmm_alloc_page(pid, 0);
+    vmm_set_mapping(VMS_SELF, PG_MOUNT_3, new_ppg, PG_PREM_RW, VMAP_NULL);
+    vmm_set_mapping(VMS_SELF, PG_MOUNT_4, pa, PG_PREM_RW, VMAP_NULL);
 
-    vmm_unset_mapping(PG_MOUNT_3);
-    vmm_unset_mapping(PG_MOUNT_4);
+    asm volatile("movl %1, %%edi\n"
+                 "movl %2, %%esi\n"
+                 "rep movsl\n" ::"c"(1024),
+                 "r"(PG_MOUNT_3),
+                 "r"(PG_MOUNT_4)
+                 : "memory", "%edi", "%esi");
+
+    vmm_del_mapping(VMS_SELF, PG_MOUNT_3);
+    vmm_del_mapping(VMS_SELF, PG_MOUNT_4);
 
     return new_ppg;
 }
\ No newline at end of file