From dd7c04a39bd88da8e381bb88078c41b049a6665c Mon Sep 17 00:00:00 2001
From: Kimplul <kimi.h.kuparinen@gmail.com>
Date: Fri, 21 Oct 2022 21:34:18 +0300
Subject: beautification

---
 arch/riscv64/kernel/cpu.c | 11 +++++++++++
 1 file changed, 11 insertions(+)

(limited to 'arch/riscv64/kernel/cpu.c')

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) :);
 }
 
-- 
cgit v1.3