#ifndef __LUNAIX_PROCESS_H
#define __LUNAIX_PROCESS_H
-#include <arch/x86/interrupts.h>
#include <lunaix/clock.h>
+#include <lunaix/ds/waitq.h>
+#include <lunaix/fs.h>
+#include <lunaix/iopoll.h>
#include <lunaix/mm/mm.h>
+#include <lunaix/mm/pagetable.h>
+#include <lunaix/mm/region.h>
#include <lunaix/signal.h>
#include <lunaix/timer.h>
#include <lunaix/types.h>
+#include <lunaix/spike.h>
+#include <lunaix/hart_state.h>
#include <stdint.h>
-// 虽然内核不是进程,但为了区分,这里使用Pid=-1来指代内核。这主要是方便物理页所有权检查。
-#define KERNEL_PID -1
-#define PROC_STOPPED 0
-#define PROC_RUNNING 1
-#define PROC_TERMNAT 2
-#define PROC_DESTROY 4
-#define PROC_BLOCKED 8
-#define PROC_CREATED 16
+/*
+ |C|Sp|Bk|De|Tn|Pu|Rn|
+ \----/
+ Dt
-#define PROC_TERMMASK 0x6
+ Group Dt: whether this process is terminated.
-struct proc_mm
-{
- heap_context_t u_heap;
- struct mm_region* regions;
-};
+ Rn: Running
+ Tn: Terminated
+ De: Destoryed
+ Pu: Paused
+ Bk: Blocked
+ Sp: Stopped
+ C : Created
+*/
+
+#define PS_READY 0
+#define PS_RUNNING 1
+#define PS_TERMNAT 2
+#define PS_DESTROY 4
+#define PS_PAUSED 8
+#define PS_BLOCKED 16
+#define PS_STOPPED 32
+#define PS_CREATED 64
+
+#define PS_GrBP (PS_PAUSED | PS_BLOCKED | PS_STOPPED)
+#define PS_GrDT (PS_TERMNAT | PS_DESTROY)
+#define PS_Rn (PS_RUNNING | PS_CREATED)
+
+#define proc_terminated(proc) (((proc)->state) & PS_GrDT)
+#define proc_hanged(proc) (((proc)->state) & PS_BLOCKED)
+#define proc_runnable(proc) (!(proc)->state || !(((proc)->state) & ~PS_Rn))
+
+
+#define TH_DETACHED 0b00000001
+#define TH_STALLED 0b00000010
+
+#define thread_detached(th) ((th)->flags & TH_DETACHED)
+#define detach_thread(th) ((th)->flags |= TH_DETACHED)
+
+#define thread_flags_set(th, flag) ((th)->flags |= (flag))
+#define thread_flags_clear(th, flag) ((th)->flags &= ~(flag))
+#define thread_flags_test(th, flag) ((th)->flags & (flag))
+#define thread_flags_test_all(th, flag) (((th)->flags & (flag)) == (flag))
struct proc_sig
{
- void* signal_handler;
int sig_num;
- isr_param prev_context;
+ void* sigact;
+ void* sighand;
+ struct hart_state* saved_hstate;
+} __attribute__((packed));
+
+
+struct proc_info;
+
+struct haybed {
+ struct llist_header sleepers;
+ time_t wakeup_time;
+ time_t alarm_time;
+};
+
+struct thread_stats
+{
+ // number of times the thread entering kernel space involuntarily
+ unsigned long entry_count_invol;
+ // number of times the thread entering kernel space voluntarily
+ unsigned long entry_count_vol;
+
+ // number of times the thread is preempted in kerenl space
+ unsigned long kpreempt_count;
+
+ // timestamp of last time kernel entry
+ time_t last_entry;
+ // timestamp of last time kernel reentry
+ time_t last_reentry;
+
+ // timestamp of last time kernel leave
+ time_t last_leave;
+ // timestamp of last time the thread is resumed
+ time_t last_resume;
+
+ union {
+ struct {
+ bool at_user;
+ };
+ int flags;
+ };
};
-#define PROC_SIG_SIZE sizeof(struct proc_sig) // size=84
+struct thread
+{
+ /*
+ Any change to *critical section*, including layout, size
+ must be reflected in arch/x86/interrupt.S.inc to avoid
+ disaster!
+ */
+ struct
+ {
+ struct hart_state* hstate;
+ ptr_t ustack_top;
+ }; // *critical section
+
+ struct {
+ tid_t tid;
+ time_t created;
+ int state;
+ int syscall_ret;
+ ptr_t exit_val;
+ int flags;
+ };
+
+ struct {
+ ptr_t kstack; // process local kernel stack
+ struct mm_region* ustack; // process local user stack (NULL for kernel thread)
+ };
+
+ struct thread_stats stats;
+
+ struct haybed sleep;
+
+ struct proc_info* process;
+ struct llist_header proc_sibs; // sibling to process-local threads
+ struct llist_header sched_sibs; // sibling to scheduler (global) threads
+ struct sigctx sigctx;
+ waitq_t waitqueue;
+};
struct proc_info
{
- pid_t pid;
- struct proc_info* parent;
- isr_param intr_ctx; // size=76
- uintptr_t ustack_top;
- void* page_table;
+ // active thread, must be at the very beginning
+ struct thread* th_active;
+
+ struct llist_header threads;
+ int thread_count;
+
+ struct llist_header tasks;
+
struct llist_header siblings;
struct llist_header children;
struct llist_header grp_member;
- struct proc_mm mm;
- time_t created;
- uint8_t state;
- int32_t exit_code;
- int32_t k_status;
- sigset_t sig_pending;
- sigset_t sig_mask;
- void* sig_handler[_SIG_NUM];
- pid_t pgid;
- struct lx_timer* timer;
+
+ struct {
+ struct proc_info* parent;
+ pid_t pid;
+ pid_t pgid;
+ time_t created;
+
+ int state;
+ int exit_code;
+ };
+
+ struct proc_mm* mm;
+ struct sigregistry* sigreg;
+ struct v_fdtable* fdtable;
+ struct v_dnode* cwd;
+ struct {
+ char* cmd;
+ size_t cmd_len;
+ };
+
+ struct iopoll pollctx;
};
extern volatile struct proc_info* __current;
+extern volatile struct thread* current_thread;
-pid_t
-alloc_pid();
+/**
+ * @brief Check if current process belong to kernel itself
+ * (pid=0)
+ */
+#define kernel_process(proc) (!(proc)->pid)
+#define resume_thread(th) (th)->state = PS_READY
+#define pause_thread(th) (th)->state = PS_PAUSED
+#define block_thread(th) (th)->state = PS_BLOCKED
+
+static inline void must_inline
+set_current_executing(struct thread* thread)
+{
+ current_thread = thread;
+ __current = thread->process;
+}
+
+static inline struct proc_mm*
+vmspace(struct proc_info* proc)
+{
+ return proc ? proc->mm : NULL;
+}
+
+static inline ptr_t
+vmroot(struct proc_info* proc)
+{
+ return proc ? proc->mm->vmroot : 0;
+}
+
+static inline vm_regions_t*
+vmregions(struct proc_info* proc)
+{
+ return proc ? &proc->mm->regions : NULL;
+}
+
+
+static inline unsigned int
+procvm_asid(struct proc_mm* mm)
+{
+ return mm->proc->pid;
+}
+
+static inline void
+block_current_thread()
+{
+ block_thread(current_thread);
+}
+
+static inline void
+pause_current_thread()
+{
+ pause_thread(current_thread);
+}
+
+static inline void
+resume_current_thread()
+{
+ resume_thread(current_thread);
+}
+
+static inline int syscall_result(int retval) {
+ return (current_thread->syscall_ret = retval);
+}
+
+/**
+ * @brief Spawn a process with arbitary entry point.
+ * The inherit priviledge level is deduced automatically
+ * from the given entry point
+ *
+ * @param created returned created main thread
+ * @param entry entry point
+ * @param with_ustack whether to pre-allocate a user stack with it
+ * @return int
+ */
+int
+spawn_process(struct thread** created, ptr_t entry, bool with_ustack);
+
+/**
+ * @brief Spawn a process that housing a given executable image as well as
+ * program argument and environment setting
+ *
+ * @param created returned created main thread
+ * @param path file system path to executable
+ * @param argv arguments passed to executable
+ * @param envp environment variables passed to executable
+ * @return int
+ */
+int
+spawn_process_usr(struct thread** created, char* path,
+ const char** argv, const char** envp);
+
+/**
+ * @brief 分配并初始化一个进程控制块
+ *
+ * @return struct proc_info*
+ */
+struct proc_info*
+alloc_process();
+
+/**
+ * @brief 初始化进程用户空间
+ *
+ * @param pcb
+ */
void
-init_proc(struct proc_info* pcb);
+init_proc_user_space(struct proc_info* pcb);
/**
* @brief 向系统发布一个进程,使其可以被调度。
* @param process
*/
void
-push_process(struct proc_info* process);
+commit_process(struct proc_info* process);
pid_t
destroy_process(pid_t pid);
-void
-setup_proc_mem(struct proc_info* proc, uintptr_t kstack_from);
+void
+delete_process(struct proc_info* proc);
/**
* @brief 复制当前进程(LunaixOS的类 fork (unix) 实现)
*
*/
void
-terminate_proc(int exit_code);
+terminate_current(int exit_code);
+
+void
+terminate_proccess(struct proc_info* proc, int exit_code);
int
orphaned_proc(pid_t pid);
struct proc_info*
get_process(pid_t pid);
+/*
+ ========= Thread =========
+*/
+
+void
+commit_thread(struct thread* thread);
+
+struct thread*
+alloc_thread(struct proc_info* process);
+
+void
+destory_thread(struct thread* thread);
+
+void
+terminate_thread(struct thread* thread, ptr_t val);
+
+void
+terminate_current_thread(ptr_t val);
+
+struct thread*
+create_thread(struct proc_info* proc, bool with_ustack);
+
+void
+start_thread(struct thread* th, ptr_t entry);
+
+static inline void
+spawn_kthread(ptr_t entry) {
+ assert(kernel_process(__current));
+
+ struct thread* th = create_thread(__current, false);
+
+ assert(th);
+ start_thread(th, entry);
+}
+
+void
+exit_thread(void* val);
+
+void
+thread_release_mem(struct thread* thread);
+
+/*
+ ========= Signal =========
+*/
+
+#define pending_sigs(thread) ((thread)->sigctx.sig_pending)
+#define raise_signal(thread, sig) sigset_add(pending_sigs(thread), sig)
+#define sigact_of(proc, sig) ((proc)->sigreg->signals[(sig)])
+#define set_sigact(proc, sig, sigact) ((proc)->sigreg->signals[(sig)] = (sigact))
+
+static inline struct sigact*
+active_signal(struct thread* thread) {
+ struct sigctx* sigctx = &thread->sigctx;
+ struct sigregistry* sigreg = thread->process->sigreg;
+ return sigreg->signals[sigctx->sig_active];
+}
+
+static inline void
+sigactive_push(struct thread* thread, int active_sig) {
+ struct sigctx* sigctx = &thread->sigctx;
+ int prev_active = sigctx->sig_active;
+
+ assert(sigact_of(thread->process, active_sig));
+
+ sigctx->sig_order[active_sig] = prev_active;
+ sigctx->sig_active = active_sig;
+}
+
+static inline void
+sigactive_pop(struct thread* thread) {
+ struct sigctx* sigctx = &thread->sigctx;
+ int active_sig = sigctx->sig_active;
+
+ sigctx->sig_active = sigctx->sig_order[active_sig];
+ sigctx->sig_order[active_sig] = active_sig;
+}
+
+void
+proc_setsignal(struct proc_info* proc, signum_t signum);
+
+void
+thread_setsignal(struct thread* thread, signum_t signum);
+
+void
+thread_stats_update(bool inbound, bool voluntary);
+
+static inline void
+thread_stats_update_entering(bool voluntary)
+{
+ thread_stats_update(true, voluntary);
+}
+
+static inline void
+thread_stats_update_leaving()
+{
+ thread_stats_update(false, true);
+}
+
+static inline void
+thread_stats_update_kpreempt()
+{
+ current_thread->stats.kpreempt_count++;
+}
+
+static inline void
+thread_stats_reset_kpreempt()
+{
+ current_thread->stats.kpreempt_count = 0;
+}
+
+static inline ticks_t
+thread_stats_kernel_elapse(struct thread* thread)
+{
+ return clock_systime() - thread->stats.last_reentry;
+}
+
+static inline ticks_t
+thread_stats_user_elapse(struct thread* thread)
+{
+ struct thread_stats* stats;
+ stats = &thread->stats;
+
+ return stats->last_entry - stats->last_leave;
+}
+
+
#endif /* __LUNAIX_PROCESS_H */