From bd5969fc876a10b18613302db7087ef3c40f18e1 Mon Sep 17 00:00:00 2001
From: Ian Moffett <ian@osmora.org>
Date: Thu, 7 Mar 2024 17:28:00 -0500
Subject: lib: Add mlibc

Signed-off-by: Ian Moffett <ian@osmora.org>
---
 lib/mlibc/sysdeps/managarm/aarch64/crt-src/Scrt1.S | 10 ++++
 lib/mlibc/sysdeps/managarm/aarch64/crt-src/crt0.S  | 11 +++++
 lib/mlibc/sysdeps/managarm/aarch64/crt-src/crti.S  | 15 ++++++
 lib/mlibc/sysdeps/managarm/aarch64/crt-src/crtn.S  | 11 +++++
 lib/mlibc/sysdeps/managarm/aarch64/signals.S       | 10 ++++
 lib/mlibc/sysdeps/managarm/aarch64/thread.cpp      | 56 ++++++++++++++++++++++
 lib/mlibc/sysdeps/managarm/aarch64/thread_entry.S  | 11 +++++
 7 files changed, 124 insertions(+)
 create mode 100644 lib/mlibc/sysdeps/managarm/aarch64/crt-src/Scrt1.S
 create mode 100644 lib/mlibc/sysdeps/managarm/aarch64/crt-src/crt0.S
 create mode 100644 lib/mlibc/sysdeps/managarm/aarch64/crt-src/crti.S
 create mode 100644 lib/mlibc/sysdeps/managarm/aarch64/crt-src/crtn.S
 create mode 100644 lib/mlibc/sysdeps/managarm/aarch64/signals.S
 create mode 100644 lib/mlibc/sysdeps/managarm/aarch64/thread.cpp
 create mode 100644 lib/mlibc/sysdeps/managarm/aarch64/thread_entry.S

(limited to 'lib/mlibc/sysdeps/managarm/aarch64')

diff --git a/lib/mlibc/sysdeps/managarm/aarch64/crt-src/Scrt1.S b/lib/mlibc/sysdeps/managarm/aarch64/crt-src/Scrt1.S
new file mode 100644
index 0000000..3e56608
--- /dev/null
+++ b/lib/mlibc/sysdeps/managarm/aarch64/crt-src/Scrt1.S
@@ -0,0 +1,10 @@
+.section .text
+.global _start
+_start:
+	mov x0, sp
+	adr x1, main
+
+	bl __mlibc_entry
+	brk #0
+
+.section .note.GNU-stack,"",%progbits
diff --git a/lib/mlibc/sysdeps/managarm/aarch64/crt-src/crt0.S b/lib/mlibc/sysdeps/managarm/aarch64/crt-src/crt0.S
new file mode 100644
index 0000000..5e59593
--- /dev/null
+++ b/lib/mlibc/sysdeps/managarm/aarch64/crt-src/crt0.S
@@ -0,0 +1,11 @@
+
+.section .text
+.global _start
+_start:
+	mov x0, sp
+	adrp x1, main
+	add x1, x1, :lo12:main
+	bl __mlibc_entry
+
+.section .note.GNU-stack,"",%progbits
+
diff --git a/lib/mlibc/sysdeps/managarm/aarch64/crt-src/crti.S b/lib/mlibc/sysdeps/managarm/aarch64/crt-src/crti.S
new file mode 100644
index 0000000..4307dfb
--- /dev/null
+++ b/lib/mlibc/sysdeps/managarm/aarch64/crt-src/crti.S
@@ -0,0 +1,15 @@
+	.ident "aarch64-managarm-mlibc crti"
+
+	.section .init
+	.global _init
+_init:
+	stp x29, x30, [sp, -16]!
+	mov x29, sp
+
+	.section .fini
+	.global _fini
+_fini:
+	stp x29, x30, [sp, -16]!
+	mov x29, sp
+
+	.section .note.GNU-stack,"",%progbits
diff --git a/lib/mlibc/sysdeps/managarm/aarch64/crt-src/crtn.S b/lib/mlibc/sysdeps/managarm/aarch64/crt-src/crtn.S
new file mode 100644
index 0000000..005d870
--- /dev/null
+++ b/lib/mlibc/sysdeps/managarm/aarch64/crt-src/crtn.S
@@ -0,0 +1,11 @@
+	.ident "aarch64-managarm-mlibc crtn"
+
+	.section .init
+	ldp x29, x30, [sp], #16
+	ret
+
+	.section .fini
+	ldp x29, x30, [sp], #16
+	ret
+
+	.section .note.GNU-stack,"",%progbits
diff --git a/lib/mlibc/sysdeps/managarm/aarch64/signals.S b/lib/mlibc/sysdeps/managarm/aarch64/signals.S
new file mode 100644
index 0000000..044767d
--- /dev/null
+++ b/lib/mlibc/sysdeps/managarm/aarch64/signals.S
@@ -0,0 +1,10 @@
+
+.section .text
+.global __mlibc_signal_restore
+__mlibc_signal_restore:
+	ldr x0, =0x80000006
+	svc 0
+	brk #1
+
+.section .note.GNU-stack,"",%progbits
+
diff --git a/lib/mlibc/sysdeps/managarm/aarch64/thread.cpp b/lib/mlibc/sysdeps/managarm/aarch64/thread.cpp
new file mode 100644
index 0000000..88f1bf3
--- /dev/null
+++ b/lib/mlibc/sysdeps/managarm/aarch64/thread.cpp
@@ -0,0 +1,56 @@
+#include <mlibc/thread-entry.hpp>
+#include <mlibc/all-sysdeps.hpp>
+#include <mlibc/tcb.hpp>
+#include <bits/ensure.h>
+#include <sys/mman.h>
+#include <errno.h>
+#include <stdint.h>
+#include <stddef.h>
+
+extern "C" void __mlibc_enter_thread(void *entry, void *user_arg, Tcb *tcb) {
+	// Wait until our parent sets up the TID.
+	while(!__atomic_load_n(&tcb->tid, __ATOMIC_RELAXED))
+		mlibc::sys_futex_wait(&tcb->tid, 0, nullptr);
+
+	if(mlibc::sys_tcb_set(tcb))
+		__ensure(!"sys_tcb_set() failed");
+
+	tcb->invokeThreadFunc(entry, user_arg);
+
+	auto self = reinterpret_cast<Tcb *>(tcb);
+
+	__atomic_store_n(&self->didExit, 1, __ATOMIC_RELEASE);
+	mlibc::sys_futex_wake(&self->didExit);
+
+	mlibc::sys_thread_exit();
+}
+
+namespace mlibc {
+
+static constexpr size_t default_stacksize = 0x200000;
+
+int sys_prepare_stack(void **stack, void *entry, void *user_arg, void *tcb, size_t *stack_size, size_t *guard_size, void **stack_base) {
+	if (!*stack_size)
+		*stack_size = default_stacksize;
+	*guard_size = 0;
+
+	if (*stack) {
+		*stack_base = *stack;
+	} else {
+		*stack_base = mmap(nullptr, *stack_size, PROT_READ | PROT_WRITE,
+						MAP_PRIVATE | MAP_ANONYMOUS, -1, 0);
+		if(*stack_base == MAP_FAILED) {
+			return errno;
+		}
+	}
+	
+	uintptr_t *sp = reinterpret_cast<uintptr_t *>(reinterpret_cast<uintptr_t>(*stack_base) + *stack_size);
+
+	*--sp = reinterpret_cast<uintptr_t>(tcb);
+	*--sp = reinterpret_cast<uintptr_t>(user_arg);
+	*--sp = reinterpret_cast<uintptr_t>(entry);
+	*stack = reinterpret_cast<void*>(sp);
+	return 0;
+}
+
+} //namespace mlibc
diff --git a/lib/mlibc/sysdeps/managarm/aarch64/thread_entry.S b/lib/mlibc/sysdeps/managarm/aarch64/thread_entry.S
new file mode 100644
index 0000000..4cfcb4c
--- /dev/null
+++ b/lib/mlibc/sysdeps/managarm/aarch64/thread_entry.S
@@ -0,0 +1,11 @@
+
+.section .text
+.global __mlibc_start_thread
+__mlibc_start_thread:
+	ldp x0, x1, [sp]
+	ldr x2, [sp, #16]
+	add sp, sp, #24
+	bl __mlibc_enter_thread
+
+.section .note.GNU-stack,"",%progbits
+
-- 
cgit v1.2.3