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

diff --git a/lunaix-os/kernel/mm/cow.c b/lunaix-os/kernel/mm/cow.c
index 30cead9..85dcd49 100644
--- a/lunaix-os/kernel/mm/cow.c
+++ b/lunaix-os/kernel/mm/cow.c
@@ -1,16 +1,19 @@
 #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_fmap_page(pid, PG_MOUNT_3, new_ppg, PG_PREM_RW);
+    vmm_fmap_page(pid, PG_MOUNT_4, pa, PG_PREM_RW);
 
     asm volatile (
         "movl %1, %%edi\n"
+        "movl %2, %%esi\n"
         "rep movsl\n"
-        :: "c"(1024), "r"(PG_MOUNT_3), "S"((uintptr_t)va)
-        : "memory", "%edi");
+        :: "c"(1024), "r"(PG_MOUNT_3), "r"(PG_MOUNT_4)
+        : "memory", "%edi", "%esi");
 
     vmm_unset_mapping(PG_MOUNT_3);
+    vmm_unset_mapping(PG_MOUNT_4);
 
     return new_ppg;
 }
\ No newline at end of file