sysdeps/managarm: Implement sys_tcb_set() on RISC-V

managarm · Nov 27, 2024 · 2ae7477 · 2ae7477
1 parent 3033fb7
commit 2ae7477
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/sysdeps/managarm/rtld-generic/support.cpp b/sysdeps/managarm/rtld-generic/support.cpp
@@ -176,12 +176,17 @@ HelHandleResult *parseHandle(void *&element) {
 namespace mlibc {
 
 int sys_tcb_set(void *pointer) {
-#if defined(__aarch64__)
+#if defined(__x86_64__)
+	HEL_CHECK(helWriteFsBase(pointer));
+#elif defined(__aarch64__)
 	uintptr_t addr = reinterpret_cast<uintptr_t>(pointer);
 	addr += sizeof(Tcb) - 0x10;
 	asm volatile ("msr tpidr_el0, %0" :: "r"(addr));
+#elif defined(__riscv) && __riscv_xlen == 64
+	uintptr_t tp = reinterpret_cast<uintptr_t>(pointer) + sizeof(Tcb);
+	asm volatile ("mv tp, %0" : : "r"(tp) : "memory");
 #else
-	HEL_CHECK(helWriteFsBase(pointer));
+#error Unknown architecture
 #endif
 	return 0;
 }