Stage 27

clks/kernel/exec.c

@@ -1,13 +1,36 @@
+#include <clks/cpu.h>
 #include <clks/elf64.h>
 #include <clks/exec.h>
 #include <clks/fs.h>
 #include <clks/heap.h>
+#include <clks/interrupts.h>
 #include <clks/log.h>
+#include <clks/string.h>
 #include <clks/types.h>
 
 typedef u64 (*clks_exec_entry_fn)(void);
 
 #define CLKS_EXEC_RUN_STACK_BYTES (64ULL * 1024ULL)
+#define CLKS_EXEC_MAX_PROCS       64U
+#define CLKS_EXEC_MAX_DEPTH       16U
+#define CLKS_EXEC_PATH_MAX        192U
+
+enum clks_exec_proc_state {
+    CLKS_EXEC_PROC_UNUSED = 0,
+    CLKS_EXEC_PROC_RUNNING = 1,
+    CLKS_EXEC_PROC_EXITED = 2,
+};
+
+struct clks_exec_proc_record {
+    clks_bool used;
+    enum clks_exec_proc_state state;
+    u64 pid;
+    u64 ppid;
+    u64 started_tick;
+    u64 exited_tick;
+    u64 exit_status;
+    char path[CLKS_EXEC_PATH_MAX];
+};
 
 #if defined(CLKS_ARCH_X86_64)
 extern u64 clks_exec_call_on_stack_x86_64(void *entry_ptr, void *stack_top);
@@ -17,6 +40,66 @@ static u64 clks_exec_requests = 0ULL;
 static u64 clks_exec_success = 0ULL;
 static u32 clks_exec_running_depth = 0U;
 
+static struct clks_exec_proc_record clks_exec_proc_table[CLKS_EXEC_MAX_PROCS];
+static u64 clks_exec_next_pid = 1ULL;
+static u64 clks_exec_pid_stack[CLKS_EXEC_MAX_DEPTH];
+static clks_bool clks_exec_exit_requested_stack[CLKS_EXEC_MAX_DEPTH];
+static u64 clks_exec_exit_status_stack[CLKS_EXEC_MAX_DEPTH];
+static u32 clks_exec_pid_stack_depth = 0U;
+
+static void clks_exec_copy_path(char *dst, usize dst_size, const char *src) {
+    usize i = 0U;
+
+    if (dst == CLKS_NULL || src == CLKS_NULL || dst_size == 0U) {
+        return;
+    }
+
+    while (src[i] != '\0' && i + 1U < dst_size) {
+        dst[i] = src[i];
+        i++;
+    }
+
+    dst[i] = '\0';
+}
+
+static i32 clks_exec_proc_find_slot_by_pid(u64 pid) {
+    u32 i;
+
+    for (i = 0U; i < CLKS_EXEC_MAX_PROCS; i++) {
+        if (clks_exec_proc_table[i].used == CLKS_TRUE && clks_exec_proc_table[i].pid == pid) {
+            return (i32)i;
+        }
+    }
+
+    return -1;
+}
+
+static i32 clks_exec_proc_alloc_slot(void) {
+    u32 i;
+
+    for (i = 0U; i < CLKS_EXEC_MAX_PROCS; i++) {
+        if (clks_exec_proc_table[i].used == CLKS_FALSE) {
+            return (i32)i;
+        }
+    }
+
+    for (i = 0U; i < CLKS_EXEC_MAX_PROCS; i++) {
+        if (clks_exec_proc_table[i].state == CLKS_EXEC_PROC_EXITED) {
+            return (i32)i;
+        }
+    }
+
+    return -1;
+}
+
+static i32 clks_exec_current_depth_index(void) {
+    if (clks_exec_pid_stack_depth == 0U) {
+        return -1;
+    }
+
+    return (i32)(clks_exec_pid_stack_depth - 1U);
+}
+
 static clks_bool clks_exec_invoke_entry(void *entry_ptr, u64 *out_ret) {
     if (entry_ptr == CLKS_NULL || out_ret == CLKS_NULL) {
         return CLKS_FALSE;
@@ -43,20 +126,20 @@ static clks_bool clks_exec_invoke_entry(void *entry_ptr, u64 *out_ret) {
 #endif
 }
 
-void clks_exec_init(void) {
-    clks_exec_requests = 0ULL;
-    clks_exec_success = 0ULL;
-    clks_exec_running_depth = 0U;
-    clks_log(CLKS_LOG_INFO, "EXEC", "PATH EXEC FRAMEWORK ONLINE");
-}
-
-clks_bool clks_exec_run_path(const char *path, u64 *out_status) {
+static clks_bool clks_exec_run_path_internal(const char *path, u64 *out_status, u64 *out_pid) {
     const void *image;
     u64 image_size = 0ULL;
     struct clks_elf64_info info;
     struct clks_elf64_loaded_image loaded;
+    clks_bool loaded_active = CLKS_FALSE;
     void *entry_ptr;
-    u64 run_ret;
+    u64 run_ret = (u64)-1;
+    u64 pid = (u64)-1;
+    i32 slot;
+    i32 depth_index;
+    struct clks_exec_proc_record *proc;
+
+    clks_memset(&loaded, 0, sizeof(loaded));
 
     clks_exec_requests++;
 
@@ -64,37 +147,78 @@ clks_bool clks_exec_run_path(const char *path, u64 *out_status) {
         *out_status = (u64)-1;
     }
 
+    if (out_pid != CLKS_NULL) {
+        *out_pid = (u64)-1;
+    }
+
     if (path == CLKS_NULL || path[0] != '/') {
         clks_log(CLKS_LOG_WARN, "EXEC", "INVALID EXEC PATH");
         return CLKS_FALSE;
     }
 
+    slot = clks_exec_proc_alloc_slot();
+
+    if (slot < 0) {
+        clks_log(CLKS_LOG_WARN, "EXEC", "PROCESS TABLE FULL");
+        return CLKS_FALSE;
+    }
+
+    if (clks_exec_pid_stack_depth >= CLKS_EXEC_MAX_DEPTH) {
+        clks_log(CLKS_LOG_WARN, "EXEC", "PROCESS STACK DEPTH EXCEEDED");
+        return CLKS_FALSE;
+    }
+
+    pid = clks_exec_next_pid;
+    clks_exec_next_pid++;
+
+    if (clks_exec_next_pid == 0ULL) {
+        clks_exec_next_pid = 1ULL;
+    }
+
+    proc = &clks_exec_proc_table[(u32)slot];
+    clks_memset(proc, 0, sizeof(*proc));
+
+    proc->used = CLKS_TRUE;
+    proc->state = CLKS_EXEC_PROC_RUNNING;
+    proc->pid = pid;
+    proc->ppid = clks_exec_current_pid();
+    proc->started_tick = clks_interrupts_timer_ticks();
+    proc->exit_status = (u64)-1;
+    clks_exec_copy_path(proc->path, sizeof(proc->path), path);
+
+    depth_index = (i32)clks_exec_pid_stack_depth;
+    clks_exec_pid_stack[(u32)depth_index] = pid;
+    clks_exec_exit_requested_stack[(u32)depth_index] = CLKS_FALSE;
+    clks_exec_exit_status_stack[(u32)depth_index] = 0ULL;
+    clks_exec_pid_stack_depth++;
+
     image = clks_fs_read_all(path, &image_size);
 
     if (image == CLKS_NULL || image_size == 0ULL) {
         clks_log(CLKS_LOG_WARN, "EXEC", "EXEC FILE MISSING");
         clks_log(CLKS_LOG_WARN, "EXEC", path);
-        return CLKS_FALSE;
+        goto fail;
     }
 
     if (clks_elf64_inspect(image, image_size, &info) == CLKS_FALSE) {
         clks_log(CLKS_LOG_WARN, "EXEC", "EXEC ELF INVALID");
         clks_log(CLKS_LOG_WARN, "EXEC", path);
-        return CLKS_FALSE;
+        goto fail;
     }
 
     if (clks_elf64_load(image, image_size, &loaded) == CLKS_FALSE) {
         clks_log(CLKS_LOG_WARN, "EXEC", "EXEC ELF LOAD FAILED");
         clks_log(CLKS_LOG_WARN, "EXEC", path);
-        return CLKS_FALSE;
+        goto fail;
     }
 
+    loaded_active = CLKS_TRUE;
+
     entry_ptr = clks_elf64_entry_pointer(&loaded, info.entry);
     if (entry_ptr == CLKS_NULL) {
         clks_log(CLKS_LOG_WARN, "EXEC", "ENTRY POINTER RESOLVE FAILED");
         clks_log(CLKS_LOG_WARN, "EXEC", path);
-        clks_elf64_unload(&loaded);
-        return CLKS_FALSE;
+        goto fail;
     }
 
     clks_log(CLKS_LOG_INFO, "EXEC", "EXEC RUN START");
@@ -110,25 +234,154 @@ clks_bool clks_exec_run_path(const char *path, u64 *out_status) {
 
         clks_log(CLKS_LOG_WARN, "EXEC", "EXEC RUN INVOKE FAILED");
         clks_log(CLKS_LOG_WARN, "EXEC", path);
-        clks_elf64_unload(&loaded);
-        return CLKS_FALSE;
+        goto fail;
     }
+
     if (clks_exec_running_depth > 0U) {
         clks_exec_running_depth--;
     }
 
+    if (clks_exec_exit_requested_stack[(u32)depth_index] == CLKS_TRUE) {
+        run_ret = clks_exec_exit_status_stack[(u32)depth_index];
+    }
+
     clks_log(CLKS_LOG_INFO, "EXEC", "RUN RETURNED");
     clks_log(CLKS_LOG_INFO, "EXEC", path);
     clks_log_hex(CLKS_LOG_INFO, "EXEC", "RET", run_ret);
 
     clks_exec_success++;
 
+    proc->state = CLKS_EXEC_PROC_EXITED;
+    proc->exit_status = run_ret;
+    proc->exited_tick = clks_interrupts_timer_ticks();
+
+    clks_exec_pid_stack_depth--;
+
+    if (loaded_active == CLKS_TRUE) {
+        clks_elf64_unload(&loaded);
+    }
+
     if (out_status != CLKS_NULL) {
         *out_status = run_ret;
     }
 
-    clks_elf64_unload(&loaded);
+    if (out_pid != CLKS_NULL) {
+        *out_pid = pid;
+    }
+
     return CLKS_TRUE;
+
+fail:
+    proc->state = CLKS_EXEC_PROC_EXITED;
+    proc->exit_status = (u64)-1;
+    proc->exited_tick = clks_interrupts_timer_ticks();
+
+    clks_exec_pid_stack_depth--;
+
+    if (loaded_active == CLKS_TRUE) {
+        clks_elf64_unload(&loaded);
+    }
+
+    if (out_status != CLKS_NULL) {
+        *out_status = (u64)-1;
+    }
+
+    return CLKS_FALSE;
+}
+
+void clks_exec_init(void) {
+    clks_exec_requests = 0ULL;
+    clks_exec_success = 0ULL;
+    clks_exec_running_depth = 0U;
+    clks_exec_next_pid = 1ULL;
+    clks_exec_pid_stack_depth = 0U;
+    clks_memset(clks_exec_pid_stack, 0, sizeof(clks_exec_pid_stack));
+    clks_memset(clks_exec_exit_requested_stack, 0, sizeof(clks_exec_exit_requested_stack));
+    clks_memset(clks_exec_exit_status_stack, 0, sizeof(clks_exec_exit_status_stack));
+    clks_memset(clks_exec_proc_table, 0, sizeof(clks_exec_proc_table));
+    clks_log(CLKS_LOG_INFO, "EXEC", "PATH EXEC FRAMEWORK ONLINE");
+}
+
+clks_bool clks_exec_run_path(const char *path, u64 *out_status) {
+    return clks_exec_run_path_internal(path, out_status, CLKS_NULL);
+}
+
+clks_bool clks_exec_spawn_path(const char *path, u64 *out_pid) {
+    u64 status = (u64)-1;
+    return clks_exec_run_path_internal(path, &status, out_pid);
+}
+
+u64 clks_exec_wait_pid(u64 pid, u64 *out_status) {
+    i32 slot;
+    const struct clks_exec_proc_record *proc;
+
+    slot = clks_exec_proc_find_slot_by_pid(pid);
+
+    if (slot < 0) {
+        return (u64)-1;
+    }
+
+    proc = &clks_exec_proc_table[(u32)slot];
+
+    if (proc->state == CLKS_EXEC_PROC_RUNNING) {
+        return 0ULL;
+    }
+
+    if (out_status != CLKS_NULL) {
+        *out_status = proc->exit_status;
+    }
+
+    return 1ULL;
+}
+
+clks_bool clks_exec_request_exit(u64 status) {
+    i32 depth_index = clks_exec_current_depth_index();
+
+    if (depth_index < 0) {
+        return CLKS_FALSE;
+    }
+
+    clks_exec_exit_requested_stack[(u32)depth_index] = CLKS_TRUE;
+    clks_exec_exit_status_stack[(u32)depth_index] = status;
+    return CLKS_TRUE;
+}
+
+u64 clks_exec_current_pid(void) {
+    i32 depth_index = clks_exec_current_depth_index();
+
+    if (depth_index < 0) {
+        return 0ULL;
+    }
+
+    return clks_exec_pid_stack[(u32)depth_index];
+}
+
+u64 clks_exec_sleep_ticks(u64 ticks) {
+    u64 start = clks_interrupts_timer_ticks();
+
+    if (ticks == 0ULL) {
+        return 0ULL;
+    }
+
+    while ((clks_interrupts_timer_ticks() - start) < ticks) {
+#if defined(CLKS_ARCH_X86_64)
+        __asm__ volatile("sti; hlt; cli" : : : "memory");
+#elif defined(CLKS_ARCH_AARCH64)
+        clks_cpu_pause();
+#endif
+    }
+
+    return clks_interrupts_timer_ticks() - start;
+}
+
+u64 clks_exec_yield(void) {
+#if defined(CLKS_ARCH_X86_64)
+    __asm__ volatile("sti; hlt; cli" : : : "memory");
+#elif defined(CLKS_ARCH_AARCH64)
+    clks_cpu_pause();
+#endif
+
+    return clks_interrupts_timer_ticks();
 }
 
 u64 clks_exec_request_count(void) {
@@ -141,5 +394,4 @@ u64 clks_exec_success_count(void) {
 
 clks_bool clks_exec_is_running(void) {
     return (clks_exec_running_depth > 0U) ? CLKS_TRUE : CLKS_FALSE;
-}
-
+}

clks/kernel/syscall.c

@@ -200,6 +200,48 @@ static u64 clks_syscall_exec_path(u64 arg0) {
     return status;
 }
 
+static u64 clks_syscall_getpid(void) {
+    return clks_exec_current_pid();
+}
+
+static u64 clks_syscall_spawn_path(u64 arg0) {
+    char path[CLKS_SYSCALL_PATH_MAX];
+    u64 pid = (u64)-1;
+
+    if (clks_syscall_copy_user_string(arg0, path, sizeof(path)) == CLKS_FALSE) {
+        return (u64)-1;
+    }
+
+    if (clks_exec_spawn_path(path, &pid) == CLKS_FALSE) {
+        return (u64)-1;
+    }
+
+    return pid;
+}
+
+static u64 clks_syscall_waitpid(u64 arg0, u64 arg1) {
+    u64 status = (u64)-1;
+    u64 wait_ret = clks_exec_wait_pid(arg0, &status);
+
+    if (wait_ret == 1ULL && arg1 != 0ULL) {
+        clks_memcpy((void *)arg1, &status, sizeof(status));
+    }
+
+    return wait_ret;
+}
+
+static u64 clks_syscall_exit(u64 arg0) {
+    return (clks_exec_request_exit(arg0) == CLKS_TRUE) ? 1ULL : 0ULL;
+}
+
+static u64 clks_syscall_sleep_ticks(u64 arg0) {
+    return clks_exec_sleep_ticks(arg0);
+}
+
+static u64 clks_syscall_yield(void) {
+    return clks_exec_yield();
+}
+
 static u64 clks_syscall_fs_stat_type(u64 arg0) {
     char path[CLKS_SYSCALL_PATH_MAX];
     struct clks_fs_node_info info;
@@ -436,6 +478,18 @@ u64 clks_syscall_dispatch(void *frame_ptr) {
             return clks_keyboard_drop_count();
         case CLKS_SYSCALL_KBD_HOTKEY_SWITCHES:
             return clks_keyboard_hotkey_switch_count();
+        case CLKS_SYSCALL_GETPID:
+            return clks_syscall_getpid();
+        case CLKS_SYSCALL_SPAWN_PATH:
+            return clks_syscall_spawn_path(frame->rbx);
+        case CLKS_SYSCALL_WAITPID:
+            return clks_syscall_waitpid(frame->rbx, frame->rcx);
+        case CLKS_SYSCALL_EXIT:
+            return clks_syscall_exit(frame->rbx);
+        case CLKS_SYSCALL_SLEEP_TICKS:
+            return clks_syscall_sleep_ticks(frame->rbx);
+        case CLKS_SYSCALL_YIELD:
+            return clks_syscall_yield();
         default:
             return (u64)-1;
     }