From 1ceed7525272b82c1028ebd353e65c3a63bd2714 Mon Sep 17 00:00:00 2001
From: Kimplul <kimi.h.kuparinen@gmail.com>
Date: Sun, 17 Apr 2022 13:40:56 +0300
Subject: expand sbi implementation a bit

---
 arch/riscv64/include/sbi.h  | 36 ++++++++++++++++++++++++++++++++++++
 arch/riscv64/kernel/power.c | 31 +++++++++++++++++++++++++++++++
 arch/riscv64/kernel/sbi.c   | 28 ++++++++++++++++++++++++++++
 3 files changed, 95 insertions(+)
 create mode 100644 arch/riscv64/kernel/power.c
 create mode 100644 arch/riscv64/kernel/sbi.c

(limited to 'arch/riscv64')

diff --git a/arch/riscv64/include/sbi.h b/arch/riscv64/include/sbi.h
index e6ceb92..b750475 100644
--- a/arch/riscv64/include/sbi.h
+++ b/arch/riscv64/include/sbi.h
@@ -1,13 +1,49 @@
 #ifndef APOS_RISCV_SBI_H
+#define APOS_RISCV_SBI_H
+
+#include <apos/types.h>
 
 struct sbiret {
 	long error;
 	long value;
 };
 
+enum {
+	SBI_SUCCESS = 0,
+	SBI_ERR_FAILED = -1,
+	SBI_ERR_NOT_SUPPORTED = -2,
+	SBI_ERR_INVALID_PARAM = -3,
+	SBI_ERR_DENIED = -4,
+	SBI_ERR_INVALID_ADDRESS = -5,
+	SBI_ERR_ALREADY_AVAILABLE = -6,
+	SBI_ERR_ALREADY_STARTED = -7,
+	SBI_ERR_ALREADY_STOPEED = -8,
+};
+
+/* TODO: query which extensions are available */
 struct sbiret sbi_ecall(int ext, int fid, unsigned long arg0,
                         unsigned long arg1, unsigned long arg2,
                         unsigned long arg3, unsigned long arg4,
                         unsigned long arg5);
 
+#define EID_TIME 0x54494D45
+#define FID_SET_TIMER 0
+static inline struct sbiret sbi_set_timer(uint64_t stime_value)
+{
+	/* TODO: read timebase from fdt, seems to be clocks/sec for accurate
+	 * timers */
+#if __riscv_xlen == 32
+	return sbi_ecall(EID_TIME, FID_SET_TIMER, stime_value, stime_value >> 32, 0, 0, 0, 0);
+#else
+	return sbi_ecall(EID_TIME, FID_SET_TIMER, stime_value, 0, 0, 0, 0, 0);
+#endif
+}
+
+#define EID_SRST 0x53525354
+#define FID_RESET 0
+static inline struct sbiret sbi_system_reset(uint32_t reset_type, uint32_t reset_reason)
+{
+	return sbi_ecall(EID_SRST, FID_RESET, reset_type, reset_reason, 0, 0, 0, 0);
+}
+
 #endif /* APOS_RISCV_SBI_H */
diff --git a/arch/riscv64/kernel/power.c b/arch/riscv64/kernel/power.c
new file mode 100644
index 0000000..877c001
--- /dev/null
+++ b/arch/riscv64/kernel/power.c
@@ -0,0 +1,31 @@
+#include <apos/power.h>
+#include <sbi.h>
+
+/* shutdown reason */
+#define SBI_NO_REASON 0
+
+/* shutdown type */
+#define SBI_SHUTDOWN 0
+#define SBI_COLD_REBOOT 1
+#define SBI_WARM_REBOOT 2
+
+stat_t poweroff(enum poweroff_type type)
+{
+	/* TODO: this only shuts down the cpu itself, but may leave the SOC
+	 * active. Should read from fdt poweroff and syscon-poweroff etc */
+	switch (type) {
+		case SHUTDOWN:
+			sbi_system_reset(SBI_SHUTDOWN, SBI_NO_REASON);
+			return ERR_MISC;
+
+		case COLD_REBOOT:
+			sbi_system_reset(SBI_COLD_REBOOT, SBI_NO_REASON);
+			return ERR_MISC;
+
+		case WARM_REBOOT:
+			sbi_system_reset(SBI_WARM_REBOOT, SBI_NO_REASON);
+			return ERR_MISC;
+	};
+
+	return ERR_INVAL;
+}
diff --git a/arch/riscv64/kernel/sbi.c b/arch/riscv64/kernel/sbi.c
new file mode 100644
index 0000000..cd3eb1b
--- /dev/null
+++ b/arch/riscv64/kernel/sbi.c
@@ -0,0 +1,28 @@
+#include <sbi.h>
+
+struct sbiret sbi_ecall(int ext, int fid, unsigned long arg0,
+		unsigned long arg1, unsigned long arg2,
+		unsigned long arg3, unsigned long arg4,
+		unsigned long arg5)
+{
+	struct sbiret ret;
+
+	register uintptr_t a0 __asm__ ("a0") = (uintptr_t)(arg0);
+	register uintptr_t a1 __asm__ ("a1") = (uintptr_t)(arg1);
+	register uintptr_t a2 __asm__ ("a2") = (uintptr_t)(arg2);
+	register uintptr_t a3 __asm__ ("a3") = (uintptr_t)(arg3);
+	register uintptr_t a4 __asm__ ("a4") = (uintptr_t)(arg4);
+	register uintptr_t a5 __asm__ ("a5") = (uintptr_t)(arg5);
+	register uintptr_t a6 __asm__ ("a6") = (uintptr_t)(fid);
+	register uintptr_t a7 __asm__ ("a7") = (uintptr_t)(ext);
+
+	__asm__ volatile ("ecall"
+			: "+r" (a0), "+r" (a1)
+			: "r" (a2), "r" (a3), "r" (a4), "r" (a5), "r" (a6), "r" (a7)
+			: "memory");
+
+	ret.error = a0;
+	ret.value = a1;
+
+	return ret;
+}
-- 
cgit v1.3