diff options
| author | Kimplul <kimi.h.kuparinen@gmail.com> | 2022-10-21 21:34:18 +0300 |
|---|---|---|
| committer | Kimplul <kimi.h.kuparinen@gmail.com> | 2022-10-21 21:34:18 +0300 |
| commit | dd7c04a39bd88da8e381bb88078c41b049a6665c (patch) | |
| tree | f5210700cc1789547ae55926c7f9aaa753fbb791 /arch/riscv64 | |
| parent | ec989bbe7d9bb7a844eeddeceaeb7bb7e4d5dab6 (diff) | |
| download | kmi-dd7c04a39bd88da8e381bb88078c41b049a6665c.tar.gz kmi-dd7c04a39bd88da8e381bb88078c41b049a6665c.zip | |
beautification
Diffstat (limited to 'arch/riscv64')
| -rw-r--r-- | arch/riscv64/init/start.S | 3 | ||||
| -rw-r--r-- | arch/riscv64/kernel/cpu.c | 11 | ||||
| -rw-r--r-- | arch/riscv64/kernel/proc.c | 2 |
3 files changed, 15 insertions, 1 deletions
diff --git a/arch/riscv64/init/start.S b/arch/riscv64/init/start.S index 646a4af..e5b6ced 100644 --- a/arch/riscv64/init/start.S +++ b/arch/riscv64/init/start.S @@ -5,7 +5,10 @@ .global _start /* Entry point to the kernel loader. */ _start: +/* load static stack */ li sp, PM_STACK_TOP +/* make sure there's no garbage in tp, important for id assignment */ +li tp, 0 call init .section .text diff --git a/arch/riscv64/kernel/cpu.c b/arch/riscv64/kernel/cpu.c index 4273d02..a730b87 100644 --- a/arch/riscv64/kernel/cpu.c +++ b/arch/riscv64/kernel/cpu.c @@ -7,10 +7,21 @@ */ #include <apos/tcb.h> +#include <apos/atomic.h> + #include <arch/cpu.h> +static atomic_long cpus = 0; + void cpu_assign(struct tcb *t) { + /* bounce cpu id forward, or if first time here, get a cpu id */ + struct tcb *c = cur_tcb(); + if (likely(c)) + t->cpu_id = c->cpu_id; + else + t->cpu_id = cpus++; + __asm__ volatile ("mv tp, %0\n" : : "r" (t) :); } diff --git a/arch/riscv64/kernel/proc.c b/arch/riscv64/kernel/proc.c index 10f8bfa..b047e2c 100644 --- a/arch/riscv64/kernel/proc.c +++ b/arch/riscv64/kernel/proc.c @@ -40,7 +40,7 @@ void set_args(struct tcb *t, struct sys_ret a) struct sys_ret get_args(struct tcb *t) { struct riscv_regs *r = (struct riscv_regs *)(--t); - return (struct sys_ret){r->a0, r->a1, r->a2, r->a3, r->a4, r->a5}; + return SYS_RET6(r->a0, r->a1, r->a2, r->a3, r->a4, r->a5); } void set_thread(struct tcb *t) |