refactor: striped more arch-related code from the kernel code base

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

diff --git a/lunaix-os/kernel/process/sched.c b/lunaix-os/kernel/process/sched.c
index edde8bc79f54f9c677207fce9813f5e79a88a6fa..ec5d6c0fe5edc9f4700cc3566f28b4f94ab7dd24 100644 (file)
--- a/lunaix-os/kernel/process/sched.c
+++ b/lunaix-os/kernel/process/sched.c
@@ -1,5 +1,5 @@
-#include <arch/x86/interrupts.h>
-#include <arch/x86/tss.h>
+#include <sys/abi.h>
+#include <sys/interrupts.h>

 
 #include <hal/apic.h>
 #include <hal/cpu.h>
 
 #include <hal/apic.h>
 #include <hal/cpu.h>


@@ -60,25 +60,13 @@ sched_init_dummy()
     extern void my_dummy();
     static char dummy_stack[DUMMY_STACK_SIZE] __attribute__((aligned(16)));
 
     extern void my_dummy();
     static char dummy_stack[DUMMY_STACK_SIZE] __attribute__((aligned(16)));
 

-    struct exec_param* execp =
-      (void*)dummy_stack + DUMMY_STACK_SIZE - sizeof(struct exec_param);
+    ptr_t stktop = (ptr_t)dummy_stack + DUMMY_STACK_SIZE;

 
 

-    *execp = (struct exec_param){
-        .cs = KCODE_SEG,
-        .eflags = cpu_reflags() | 0x0200,
-        .eip = (void*)my_dummy,
-        .ss = KDATA_SEG,
-    };
-
-    // memset to 0

     dummy_proc = (struct proc_info){};
     dummy_proc = (struct proc_info){};

-    dummy_proc.intr_ctx = (isr_param){ .registers = { .ds = KDATA_SEG,
-                                                      .es = KDATA_SEG,
-                                                      .fs = KDATA_SEG,
-                                                      .gs = KDATA_SEG },
-                                       .execp = execp };

 
 

-    dummy_proc.page_table = cpu_rcr3();
+    proc_init_transfer(&dummy_proc, stktop, (ptr_t)my_dummy, TRANSFER_IE);
+
+    dummy_proc.page_table = cpu_ldvmspace();

     dummy_proc.state = PS_READY;
     dummy_proc.parent = &dummy_proc;
     dummy_proc.pid = KERNEL_PID;
     dummy_proc.state = PS_READY;
     dummy_proc.parent = &dummy_proc;
     dummy_proc.pid = KERNEL_PID;


@@ -99,7 +87,6 @@ run(struct proc_info* proc)
         由于这中间没有进行地址空间的交换，所以第二次跳转使用的是同一个内核栈，而之前默认tss.esp0的值是永远指向最顶部
         这样一来就有可能会覆盖更早的上下文信息（比如嵌套的信号捕获函数）
     */
         由于这中间没有进行地址空间的交换，所以第二次跳转使用的是同一个内核栈，而之前默认tss.esp0的值是永远指向最顶部
         这样一来就有可能会覆盖更早的上下文信息（比如嵌套的信号捕获函数）
     */

-    tss_update_esp(proc->intr_ctx.esp);

 
     apic_done_servicing();
 
 
     apic_done_servicing();
 


@@ -111,14 +98,24 @@ run(struct proc_info* proc)
 int
 can_schedule(struct proc_info* proc)
 {
 int
 can_schedule(struct proc_info* proc)
 {

-    if (__SIGTEST(proc->sig_pending, _SIGCONT)) {
-        __SIGCLEAR(proc->sig_pending, _SIGSTOP);
-    } else if (__SIGTEST(proc->sig_pending, _SIGSTOP)) {
+    if (!proc) {
+        return 0;
+    }
+
+    struct sighail* sh = &proc->sigctx;
+
+    if ((proc->state & PS_PAUSED)) {
+        return !!(sh->sig_pending & ~1);
+    }
+
+    if (sigset_test(sh->sig_pending, _SIGCONT)) {
+        sigset_clear(sh->sig_pending, _SIGSTOP);
+    } else if (sigset_test(sh->sig_pending, _SIGSTOP)) {

         // 如果进程受到SIGSTOP，则该进程不给予调度。
         return 0;
     }
 
         // 如果进程受到SIGSTOP，则该进程不给予调度。
         return 0;
     }
 

-    return 1;
+    return (proc->state == PS_READY);

 }
 
 void
 }
 
 void


@@ -129,7 +126,7 @@ check_sleepers()
     time_t now = clock_systime();
     llist_for_each(pos, n, &leader->sleep.sleepers, sleep.sleepers)
     {
     time_t now = clock_systime();
     llist_for_each(pos, n, &leader->sleep.sleepers, sleep.sleepers)
     {

-        if (PROC_TERMINATED(pos->state)) {
+        if (proc_terminated(pos)) {

             goto del;
         }
 
             goto del;
         }
 


@@ -143,7 +140,7 @@ check_sleepers()
 
         if (atime && now >= atime) {
             pos->sleep.alarm_time = 0;
 
         if (atime && now >= atime) {
             pos->sleep.alarm_time = 0;

-            __SIGSET(pos->sig_pending, _SIGALRM);
+            proc_setsignal(pos, _SIGALRM);

         }
 
         if (!wtime && !atime) {
         }
 
         if (!wtime && !atime) {


@@ -165,6 +162,7 @@ schedule()
     struct proc_info* next;
     int prev_ptr = sched_ctx.procs_index;
     int ptr = prev_ptr;
     struct proc_info* next;
     int prev_ptr = sched_ctx.procs_index;
     int ptr = prev_ptr;

+    int found = 0;

 
     if (!(__current->state & ~PS_RUNNING)) {
         __current->state = PS_READY;
 
     if (!(__current->state & ~PS_RUNNING)) {
         __current->state = PS_READY;


@@ -173,24 +171,19 @@ schedule()
     check_sleepers();
 
     // round-robin scheduler
     check_sleepers();
 
     // round-robin scheduler

-redo:

     do {
         ptr = (ptr + 1) % sched_ctx.ptable_len;
         next = sched_ctx._procs[ptr];
     do {
         ptr = (ptr + 1) % sched_ctx.ptable_len;
         next = sched_ctx._procs[ptr];

-    } while (!next || (next->state != PS_READY && ptr != prev_ptr));
-
-    sched_ctx.procs_index = ptr;

 
 

-    if (next->state != PS_READY) {
-        // schedule the dummy process if we're out of choice
-        next = &dummy_proc;
-        goto done;
-    }
+        if (!(found = can_schedule(next))) {
+            if (ptr == prev_ptr) {
+                next = &dummy_proc;
+                goto done;
+            }
+        }
+    } while (!found);

 
 

-    if (!can_schedule(next)) {
-        // 如果该进程不给予调度，则尝试重新选择
-        goto redo;
-    }
+    sched_ctx.procs_index = ptr;

 
 done:
     run(next);
 
 done:
     run(next);


@@ -200,7 +193,7 @@ void
 sched_yieldk()
 {
     cpu_enable_interrupt();
 sched_yieldk()
 {
     cpu_enable_interrupt();

-    cpu_int(LUNAIX_SCHED);
+    cpu_trap_sched();

 }
 
 __DEFINE_LXSYSCALL1(unsigned int, sleep, unsigned int, seconds)
 }
 
 __DEFINE_LXSYSCALL1(unsigned int, sleep, unsigned int, seconds)


@@ -220,10 +213,12 @@ __DEFINE_LXSYSCALL1(unsigned int, sleep, unsigned int, seconds)
         llist_append(&root_proc->sleep.sleepers, &__current->sleep.sleepers);
     }
 
         llist_append(&root_proc->sleep.sleepers, &__current->sleep.sleepers);
     }
 

-    __current->intr_ctx.registers.eax = seconds;
+    store_retval(seconds);

 
     block_current();
     schedule();
 
     block_current();
     schedule();

+
+    return 0;

 }
 
 __DEFINE_LXSYSCALL1(unsigned int, alarm, unsigned int, seconds)
 }
 
 __DEFINE_LXSYSCALL1(unsigned int, alarm, unsigned int, seconds)


@@ -303,7 +298,6 @@ repeat:
     goto repeat;
 
 done:
     goto repeat;
 
 done:

-    status_flags |= PEXITSIG * (proc->sig_inprogress != 0);

     if (status) {
         *status = proc->exit_code | status_flags;
     }
     if (status) {
         *status = proc->exit_code | status_flags;
     }


@@ -333,8 +327,6 @@ alloc_process()
     proc->created = clock_systime();
     proc->pgid = proc->pid;
     proc->fdtable = vzalloc(sizeof(struct v_fdtable));
     proc->created = clock_systime();
     proc->pgid = proc->pid;
     proc->fdtable = vzalloc(sizeof(struct v_fdtable));

-    proc->fxstate =
-      vzalloc_dma(512); // FXSAVE需要十六位对齐地址，使用DMA块（128位对齐）

 
     llist_init_head(&proc->mm.regions);
     llist_init_head(&proc->tasks);
 
     llist_init_head(&proc->mm.regions);
     llist_init_head(&proc->tasks);


@@ -371,7 +363,7 @@ commit_process(struct proc_info* process)
 
 // from <kernel/process.c>
 extern void
 
 // from <kernel/process.c>
 extern void

-__del_pagetable(pid_t pid, uintptr_t mount_point);
+__del_pagetable(pid_t pid, ptr_t mount_point);

 
 pid_t
 destroy_process(pid_t pid)
 
 pid_t
 destroy_process(pid_t pid)


@@ -379,8 +371,9 @@ destroy_process(pid_t pid)
     int index = pid;
     if (index <= 0 || index > sched_ctx.ptable_len) {
         __current->k_status = EINVAL;
     int index = pid;
     if (index <= 0 || index > sched_ctx.ptable_len) {
         __current->k_status = EINVAL;

-        return;
+        return -1;

     }
     }

+

     struct proc_info* proc = sched_ctx._procs[index];
     sched_ctx._procs[index] = 0;
 
     struct proc_info* proc = sched_ctx._procs[index];
     sched_ctx._procs[index] = 0;
 


@@ -404,7 +397,6 @@ destroy_process(pid_t pid)
     }
 
     vfree(proc->fdtable);
     }
 
     vfree(proc->fdtable);

-    vfree_dma(proc->fxstate);

 
     vmm_mount_pd(VMS_MOUNT_1, proc->page_table);
 
 
     vmm_mount_pd(VMS_MOUNT_1, proc->page_table);
 


@@ -430,7 +422,7 @@ terminate_proc(int exit_code)
     __current->state = PS_TERMNAT;
     __current->exit_code = exit_code;
 
     __current->state = PS_TERMNAT;
     __current->exit_code = exit_code;
 

-    __SIGSET(__current->parent->sig_pending, _SIGCHLD);
+    proc_setsignal(__current->parent, _SIGCHLD);

 }
 
 struct proc_info*
 }
 
 struct proc_info*


@@ -455,5 +447,5 @@ orphaned_proc(pid_t pid)
 
     // 如果其父进程的状态是terminated 或 destroy中的一种
     // 或者其父进程是在该进程之后创建的，那么该进程为孤儿进程
 
     // 如果其父进程的状态是terminated 或 destroy中的一种
     // 或者其父进程是在该进程之后创建的，那么该进程为孤儿进程

-    return PROC_TERMINATED(parent->state) || parent->created > proc->created;
+    return proc_terminated(parent) || parent->created > proc->created;

 }
\ No newline at end of file
 }
\ No newline at end of file