X-Git-Url: https://scm.lunaixsky.com/lunaix-os.git/blobdiff_plain/48b4a227035048fdebcd32532deb7a857c6199ac..9b8e0c494de6b447b44454112748f702dffec90d:/lunaix-os/kernel/mm/cow.c

diff --git a/lunaix-os/kernel/mm/cow.c b/lunaix-os/kernel/mm/cow.c
index 30cead9..2350188 100644
--- a/lunaix-os/kernel/mm/cow.c
+++ b/lunaix-os/kernel/mm/cow.c
@@ -1,16 +1,21 @@
 #include <lunaix/mm/vmm.h>
 
-void* vmm_dup_page(void* va) {    
-    void* new_ppg = pmm_alloc_page(KERNEL_PID, 0);
-    vmm_fmap_page(KERNEL_PID, PG_MOUNT_3, new_ppg, PG_PREM_RW);
+void*
+vmm_dup_page(pid_t pid, void* pa)
+{
+    void* 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);
 
-    asm volatile (
-        "movl %1, %%edi\n"
-        "rep movsl\n"
-        :: "c"(1024), "r"(PG_MOUNT_3), "S"((uintptr_t)va)
-        : "memory", "%edi");
+    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_unset_mapping(PG_MOUNT_3);
+    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