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/x86_64/crt-src/Scrt1.S |  8 ++++
 lib/mlibc/sysdeps/managarm/x86_64/crt-src/crt0.S  | 10 ++++
 lib/mlibc/sysdeps/managarm/x86_64/crt-src/crti.S  | 15 ++++++
 lib/mlibc/sysdeps/managarm/x86_64/crt-src/crtn.S  | 11 +++++
 lib/mlibc/sysdeps/managarm/x86_64/signals.S       | 10 ++++
 lib/mlibc/sysdeps/managarm/x86_64/thread.cpp      | 56 +++++++++++++++++++++++
 lib/mlibc/sysdeps/managarm/x86_64/thread_entry.S  | 11 +++++
 7 files changed, 121 insertions(+)
 create mode 100644 lib/mlibc/sysdeps/managarm/x86_64/crt-src/Scrt1.S
 create mode 100644 lib/mlibc/sysdeps/managarm/x86_64/crt-src/crt0.S
 create mode 100644 lib/mlibc/sysdeps/managarm/x86_64/crt-src/crti.S
 create mode 100644 lib/mlibc/sysdeps/managarm/x86_64/crt-src/crtn.S
 create mode 100644 lib/mlibc/sysdeps/managarm/x86_64/signals.S
 create mode 100644 lib/mlibc/sysdeps/managarm/x86_64/thread.cpp
 create mode 100644 lib/mlibc/sysdeps/managarm/x86_64/thread_entry.S

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

diff --git a/lib/mlibc/sysdeps/managarm/x86_64/crt-src/Scrt1.S b/lib/mlibc/sysdeps/managarm/x86_64/crt-src/Scrt1.S
new file mode 100644
index 0000000..d0e8213
--- /dev/null
+++ b/lib/mlibc/sysdeps/managarm/x86_64/crt-src/Scrt1.S
@@ -0,0 +1,8 @@
+.section .text
+.global _start
+_start:
+	mov %rsp, %rdi
+	lea main(%rip), %rsi
+	call __mlibc_entry
+.section .note.GNU-stack,"",%progbits
+
diff --git a/lib/mlibc/sysdeps/managarm/x86_64/crt-src/crt0.S b/lib/mlibc/sysdeps/managarm/x86_64/crt-src/crt0.S
new file mode 100644
index 0000000..6afb421
--- /dev/null
+++ b/lib/mlibc/sysdeps/managarm/x86_64/crt-src/crt0.S
@@ -0,0 +1,10 @@
+
+.section .text
+.global _start
+_start:
+	mov %rsp, %rdi
+	mov $main, %rsi
+	call __mlibc_entry
+
+.section .note.GNU-stack,"",%progbits
+
diff --git a/lib/mlibc/sysdeps/managarm/x86_64/crt-src/crti.S b/lib/mlibc/sysdeps/managarm/x86_64/crt-src/crti.S
new file mode 100644
index 0000000..1ca20f2
--- /dev/null
+++ b/lib/mlibc/sysdeps/managarm/x86_64/crt-src/crti.S
@@ -0,0 +1,15 @@
+	.ident "x86_64-managarm-mlibc crti"
+
+	.section .init
+	.globl	_init
+	.type	_init,@function
+_init:
+	push %rax
+
+	.section .fini
+	.globl	_fini
+	.type	_fini,@function
+_fini:
+	push %rax
+
+.section .note.GNU-stack,"",%progbits
diff --git a/lib/mlibc/sysdeps/managarm/x86_64/crt-src/crtn.S b/lib/mlibc/sysdeps/managarm/x86_64/crt-src/crtn.S
new file mode 100644
index 0000000..fdb309e
--- /dev/null
+++ b/lib/mlibc/sysdeps/managarm/x86_64/crt-src/crtn.S
@@ -0,0 +1,11 @@
+.ident "x86_64-managarm-mlibc crtn"
+
+.section .init
+	pop %rax
+	ret
+
+.section .fini
+	pop %rax
+	ret
+
+.section .note.GNU-stack,"",%progbits
diff --git a/lib/mlibc/sysdeps/managarm/x86_64/signals.S b/lib/mlibc/sysdeps/managarm/x86_64/signals.S
new file mode 100644
index 0000000..33ac4a3
--- /dev/null
+++ b/lib/mlibc/sysdeps/managarm/x86_64/signals.S
@@ -0,0 +1,10 @@
+
+.section .text
+.global __mlibc_signal_restore
+__mlibc_signal_restore:
+	mov $0x80000006, %rdi
+	syscall
+	ud2
+
+.section .note.GNU-stack,"",%progbits
+
diff --git a/lib/mlibc/sysdeps/managarm/x86_64/thread.cpp b/lib/mlibc/sysdeps/managarm/x86_64/thread.cpp
new file mode 100644
index 0000000..88f1bf3
--- /dev/null
+++ b/lib/mlibc/sysdeps/managarm/x86_64/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/x86_64/thread_entry.S b/lib/mlibc/sysdeps/managarm/x86_64/thread_entry.S
new file mode 100644
index 0000000..51e703b
--- /dev/null
+++ b/lib/mlibc/sysdeps/managarm/x86_64/thread_entry.S
@@ -0,0 +1,11 @@
+
+.section .text
+.global __mlibc_start_thread
+__mlibc_start_thread:
+	pop %rdi
+	pop %rsi
+	pop %rdx
+	call __mlibc_enter_thread
+
+.section .note.GNU-stack,"",%progbits
+
-- 
cgit v1.2.3