refactor: Move the kernel post init stage into proc0

[lunaix-os.git] / lunaix-os / kernel / process.c

diff --git a/lunaix-os/kernel/process.c b/lunaix-os/kernel/process.c
index 5112ae19e04a7e78ddf2cc36587e5684799ff5a0..87a7413c963f0db793f22f842c5c59bdaf491e50 100644 (file)
--- a/lunaix-os/kernel/process.c
+++ b/lunaix-os/kernel/process.c
@@ -1,34 +1,34 @@
-#include <lunaix/process.h>
-#include <lunaix/mm/vmm.h>
-#include <lunaix/mm/region.h>
+#include <klibc/string.h>

 #include <lunaix/clock.h>
 #include <lunaix/clock.h>

-#include <lunaix/syslog.h>

 #include <lunaix/common.h>
 #include <lunaix/common.h>

-#include <lunaix/syscall.h>
+#include <lunaix/mm/region.h>
+#include <lunaix/mm/vmm.h>
+#include <lunaix/process.h>

 #include <lunaix/spike.h>
 #include <lunaix/spike.h>

-#include <klibc/string.h>
+#include <lunaix/syscall.h>
+#include <lunaix/syslog.h>

 
 LOG_MODULE("PROC")
 
 
 LOG_MODULE("PROC")
 

-void* __dup_pagetable(pid_t pid, uintptr_t mount_point) {
+void*
+__dup_pagetable(pid_t pid, uintptr_t mount_point)
+{

     void* ptd_pp = pmm_alloc_page(pid, PP_FGPERSIST);
     x86_page_table* ptd = vmm_fmap_page(pid, PG_MOUNT_1, ptd_pp, PG_PREM_RW);
     void* ptd_pp = pmm_alloc_page(pid, PP_FGPERSIST);
     x86_page_table* ptd = vmm_fmap_page(pid, PG_MOUNT_1, ptd_pp, PG_PREM_RW);

-    x86_page_table* pptd = (x86_page_table*) (mount_point | (0x3FF << 12));
+    x86_page_table* pptd = (x86_page_table*)(mount_point | (0x3FF << 12));

 
 

-    for (size_t i = 0; i < PG_MAX_ENTRIES - 1; i++)
-    {
+    for (size_t i = 0; i < PG_MAX_ENTRIES - 1; i++) {

         x86_pte_t ptde = pptd->entry[i];
         if (!ptde || !(ptde & PG_PRESENT)) {
             ptd->entry[i] = ptde;
             continue;
         }
         x86_pte_t ptde = pptd->entry[i];
         if (!ptde || !(ptde & PG_PRESENT)) {
             ptd->entry[i] = ptde;
             continue;
         }

-        
-        x86_page_table* ppt = (x86_page_table*) (mount_point | (i << 12));
+
+        x86_page_table* ppt = (x86_page_table*)(mount_point | (i << 12));

         void* pt_pp = pmm_alloc_page(pid, PP_FGPERSIST);
         x86_page_table* pt = vmm_fmap_page(pid, PG_MOUNT_2, pt_pp, PG_PREM_RW);
 
         void* pt_pp = pmm_alloc_page(pid, PP_FGPERSIST);
         x86_page_table* pt = vmm_fmap_page(pid, PG_MOUNT_2, pt_pp, PG_PREM_RW);
 

-        for (size_t j = 0; j < PG_MAX_ENTRIES; j++)
-        {
+        for (size_t j = 0; j < PG_MAX_ENTRIES; j++) {

             x86_pte_t pte = ppt->entry[j];
             pmm_ref_page(pid, PG_ENTRY_ADDR(pte));
             pt->entry[j] = pte;
             x86_pte_t pte = ppt->entry[j];
             pmm_ref_page(pid, PG_ENTRY_ADDR(pte));
             pt->entry[j] = pte;


@@ -36,30 +36,30 @@ void* __dup_pagetable(pid_t pid, uintptr_t mount_point) {
 
         ptd->entry[i] = (uintptr_t)pt_pp | PG_PREM_RW;
     }
 
         ptd->entry[i] = (uintptr_t)pt_pp | PG_PREM_RW;
     }

-    
+

     ptd->entry[PG_MAX_ENTRIES - 1] = NEW_L1_ENTRY(T_SELF_REF_PERM, ptd_pp);
 
     return ptd_pp;
 }
 
     ptd->entry[PG_MAX_ENTRIES - 1] = NEW_L1_ENTRY(T_SELF_REF_PERM, ptd_pp);
 
     return ptd_pp;
 }
 

-void __del_pagetable(pid_t pid, uintptr_t mount_point) {
-    x86_page_table* pptd = (x86_page_table*) (mount_point | (0x3FF << 12));
+void
+__del_pagetable(pid_t pid, uintptr_t mount_point)
+{
+    x86_page_table* pptd = (x86_page_table*)(mount_point | (0x3FF << 12));

 
 

-    for (size_t i = 0; i < PG_MAX_ENTRIES - 1; i++)
-    {
+    for (size_t i = 0; i < PG_MAX_ENTRIES - 1; i++) {

         x86_pte_t ptde = pptd->entry[i];
         if (!ptde || !(ptde & PG_PRESENT)) {
             continue;
         }
         x86_pte_t ptde = pptd->entry[i];
         if (!ptde || !(ptde & PG_PRESENT)) {
             continue;
         }

-        
-        x86_page_table* ppt = (x86_page_table*) (mount_point | (i << 12));

 
 

-        for (size_t j = 0; j < PG_MAX_ENTRIES; j++)
-        {
+        x86_page_table* ppt = (x86_page_table*)(mount_point | (i << 12));
+
+        for (size_t j = 0; j < PG_MAX_ENTRIES; j++) {

             x86_pte_t pte = ppt->entry[j];
             // free the 4KB data page
             if ((pte & PG_PRESENT)) {
             x86_pte_t pte = ppt->entry[j];
             // free the 4KB data page
             if ((pte & PG_PRESENT)) {

-                pmm_free_page(pid,PG_ENTRY_ADDR(pte));
+                pmm_free_page(pid, PG_ENTRY_ADDR(pte));

             }
         }
         // free the L2 page table
             }
         }
         // free the L2 page table


@@ -69,23 +69,30 @@ void __del_pagetable(pid_t pid, uintptr_t mount_point) {
     pmm_free_page(pid, PG_ENTRY_ADDR(pptd->entry[PG_MAX_ENTRIES - 1]));
 }
 
     pmm_free_page(pid, PG_ENTRY_ADDR(pptd->entry[PG_MAX_ENTRIES - 1]));
 }
 

-void* dup_pagetable(pid_t pid) {
+void*
+dup_pagetable(pid_t pid)
+{

     return __dup_pagetable(pid, PD_REFERENCED);
 }
 
     return __dup_pagetable(pid, PD_REFERENCED);
 }
 

-__DEFINE_LXSYSCALL(pid_t, fork) {
+__DEFINE_LXSYSCALL(pid_t, fork)
+{

     return dup_proc();
 }
 
     return dup_proc();
 }
 

-__DEFINE_LXSYSCALL(pid_t, getpid) {
+__DEFINE_LXSYSCALL(pid_t, getpid)
+{

     return __current->pid;
 }
 
     return __current->pid;
 }
 

-__DEFINE_LXSYSCALL(pid_t, getppid) {
+__DEFINE_LXSYSCALL(pid_t, getppid)
+{

     return __current->parent->pid;
 }
 
     return __current->parent->pid;
 }
 

-void init_proc(struct proc_info *pcb) {
+void
+init_proc(struct proc_info* pcb)
+{

     memset(pcb, 0, sizeof(*pcb));
 
     pcb->pid = alloc_pid();
     memset(pcb, 0, sizeof(*pcb));
 
     pcb->pid = alloc_pid();


@@ -93,7 +100,9 @@ void init_proc(struct proc_info *pcb) {
     pcb->state = PROC_CREATED;
 }
 
     pcb->state = PROC_CREATED;
 }
 

-pid_t dup_proc() {
+pid_t
+dup_proc()
+{

     struct proc_info pcb;
     init_proc(&pcb);
     pcb.mm = __current->mm;
     struct proc_info pcb;
     init_proc(&pcb);
     pcb.mm = __current->mm;


@@ -101,7 +110,7 @@ pid_t dup_proc() {
     pcb.parent = __current;
 
 #ifdef USE_KERNEL_PG
     pcb.parent = __current;
 
 #ifdef USE_KERNEL_PG

-    setup_proc_mem(&pcb, PD_MOUNT_1);    //挂载点#1是当前进程的页表
+    setup_proc_mem(&pcb, PD_MOUNT_1); //挂载点#1是当前进程的页表

 #else
     setup_proc_mem(&pcb, PD_REFERENCED);
 #endif
 #else
     setup_proc_mem(&pcb, PD_REFERENCED);
 #endif


@@ -110,9 +119,11 @@ pid_t dup_proc() {
     if (!__current->mm.regions) {
         goto not_copy;
     }
     if (!__current->mm.regions) {
         goto not_copy;
     }

-    
+
+    llist_init_head(&pcb.mm.regions);

     struct mm_region *pos, *n;
     struct mm_region *pos, *n;

-    llist_for_each(pos, n, &__current->mm.regions->head, head) {
+    llist_for_each(pos, n, &__current->mm.regions->head, head)
+    {

         region_add(&pcb, pos->start, pos->end, pos->attr);
 
         // 如果写共享，则不作处理。
         region_add(&pcb, pos->start, pos->end, pos->attr);
 
         // 如果写共享，则不作处理。


@@ -122,25 +133,23 @@ pid_t dup_proc() {
 
         uintptr_t start_vpn = PG_ALIGN(pos->start) >> 12;
         uintptr_t end_vpn = PG_ALIGN(pos->end) >> 12;
 
         uintptr_t start_vpn = PG_ALIGN(pos->start) >> 12;
         uintptr_t end_vpn = PG_ALIGN(pos->end) >> 12;

-        for (size_t i = start_vpn; i < end_vpn; i++)
-        {
-            x86_pte_t *curproc = &PTE_MOUNTED(PD_MOUNT_1, i);
-            x86_pte_t *newproc = &PTE_MOUNTED(PD_MOUNT_2, i);
-            cpu_invplg(curproc);
+        for (size_t i = start_vpn; i < end_vpn; i++) {
+            x86_pte_t* curproc = &PTE_MOUNTED(PD_MOUNT_1, i);
+            x86_pte_t* newproc = &PTE_MOUNTED(PD_MOUNT_2, i);

             cpu_invplg(newproc);
 
             if (pos->attr == REGION_RSHARED) {
                 // 如果读共享，则将两者的都标注为只读，那么任何写入都将会应用COW策略。
             cpu_invplg(newproc);
 
             if (pos->attr == REGION_RSHARED) {
                 // 如果读共享，则将两者的都标注为只读，那么任何写入都将会应用COW策略。

+                cpu_invplg(curproc);

                 *curproc = *curproc & ~PG_WRITE;
                 *newproc = *newproc & ~PG_WRITE;
                 *curproc = *curproc & ~PG_WRITE;
                 *newproc = *newproc & ~PG_WRITE;

-            }
-            else {
+            } else {

                 // 如果是私有页，则将该页从新进程中移除。
                 *newproc = 0;
             }
         }
     }
                 // 如果是私有页，则将该页从新进程中移除。
                 *newproc = 0;
             }
         }
     }

-    
+

 not_copy:
     vmm_unmount_pd(PD_MOUNT_2);
 
 not_copy:
     vmm_unmount_pd(PD_MOUNT_2);
 


@@ -154,24 +163,26 @@ not_copy:
 
 extern void __kernel_end;
 
 
 extern void __kernel_end;
 

-void setup_proc_mem(struct proc_info* proc, uintptr_t usedMnt) {
+void
+setup_proc_mem(struct proc_info* proc, uintptr_t usedMnt)
+{

     // copy the entire kernel page table
     pid_t pid = proc->pid;
     void* pt_copy = __dup_pagetable(pid, usedMnt);
 
     // copy the entire kernel page table
     pid_t pid = proc->pid;
     void* pt_copy = __dup_pagetable(pid, usedMnt);
 

-    vmm_mount_pd(PD_MOUNT_2, pt_copy);   // 将新进程的页表挂载到挂载点#2
+    vmm_mount_pd(PD_MOUNT_2, pt_copy); // 将新进程的页表挂载到挂载点#2

 
     // copy the kernel stack
 
     // copy the kernel stack

-    for (size_t i = KSTACK_START >> 12; i <= KSTACK_TOP >> 12; i++)
-    {
-        volatile x86_pte_t *ppte = &PTE_MOUNTED(PD_MOUNT_2, i);
+    for (size_t i = KSTACK_START >> 12; i <= KSTACK_TOP >> 12; i++) {
+        volatile x86_pte_t* ppte = &PTE_MOUNTED(PD_MOUNT_2, i);

 
         /*
 
         /*

-            This is a fucking nightmare, the TLB caching keep the rewrite to PTE from updating.
-            Even the Nightmare Moon the Evil is far less nasty than this.
-            It took me hours of debugging to figure this out.
+            This is a fucking nightmare, the TLB caching keep the rewrite to PTE
+           from updating. Even the Nightmare Moon the Evil is far less nasty
+           than this. It took me hours of debugging to figure this out.

 
 

-            In the name of Celestia our glorious goddess, I will fucking HATE the TLB for the rest of my LIFE!
+            In the name of Celestia our glorious goddess, I will fucking HATE
+           the TLB for the rest of my LIFE!

         */
         cpu_invplg(ppte);
 
         */
         cpu_invplg(ppte);
 


@@ -182,8 +193,9 @@ void setup_proc_mem(struct proc_info* proc, uintptr_t usedMnt) {
 
     // 我们不需要分配内核的区域，因为所有的内核代码和数据段只能通过系统调用来访问，任何非法的访问
     // 都会导致eip落在区域外面，从而segmentation fault.
 
     // 我们不需要分配内核的区域，因为所有的内核代码和数据段只能通过系统调用来访问，任何非法的访问
     // 都会导致eip落在区域外面，从而segmentation fault.

-    
-    // 定义用户栈区域，但是不分配实际的物理页。我们会在Page fault handler里面实现动态分配物理页的逻辑。（虚拟内存的好处！）
+
+    // 定义用户栈区域，但是不分配实际的物理页。我们会在Page fault
+    // handler里面实现动态分配物理页的逻辑。（虚拟内存的好处！）

     // FIXME: 这里应该放到spawn_proc里面。
     // region_add(proc, USTACK_END, USTACK_SIZE, REGION_PRIVATE | REGION_RW);
 
     // FIXME: 这里应该放到spawn_proc里面。
     // region_add(proc, USTACK_END, USTACK_SIZE, REGION_PRIVATE | REGION_RW);