mirror of
https://github.com/jart/cosmopolitan.git
synced 2025-01-31 19:43:32 +00:00
fad1279c61
If cosmo_dlopen() is linked on AMD64 then the runtime will switch to using %gs for thread-local storage. This eliminates the need for the imported symbol trampoline. It's now safer to pass function pointers back and forth with imported libraries. Your program gets recompiled at runtime to make it happen and the overhead is a few milliseconds.
760 lines
28 KiB
C
760 lines
28 KiB
C
/*-*- mode:c;indent-tabs-mode:nil;c-basic-offset:2;tab-width:8;coding:utf-8 -*-│
|
|
│ vi: set et ft=c ts=2 sts=2 sw=2 fenc=utf-8 :vi │
|
|
╞══════════════════════════════════════════════════════════════════════════════╡
|
|
│ Copyright 2021 Justine Alexandra Roberts Tunney │
|
|
│ │
|
|
│ Permission to use, copy, modify, and/or distribute this software for │
|
|
│ any purpose with or without fee is hereby granted, provided that the │
|
|
│ above copyright notice and this permission notice appear in all copies. │
|
|
│ │
|
|
│ THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL │
|
|
│ WARRANTIES WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED │
|
|
│ WARRANTIES OF MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE │
|
|
│ AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL │
|
|
│ DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR │
|
|
│ PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER │
|
|
│ TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR │
|
|
│ PERFORMANCE OF THIS SOFTWARE. │
|
|
╚─────────────────────────────────────────────────────────────────────────────*/
|
|
#include "libc/sysv/consts/clone.h"
|
|
#include "libc/assert.h"
|
|
#include "libc/atomic.h"
|
|
#include "libc/calls/calls.h"
|
|
#include "libc/calls/state.internal.h"
|
|
#include "libc/calls/struct/sigset.h"
|
|
#include "libc/calls/struct/ucontext-netbsd.internal.h"
|
|
#include "libc/calls/syscall-sysv.internal.h"
|
|
#include "libc/calls/wincrash.internal.h"
|
|
#include "libc/dce.h"
|
|
#include "libc/errno.h"
|
|
#include "libc/intrin/asan.internal.h"
|
|
#include "libc/intrin/atomic.h"
|
|
#include "libc/intrin/describeflags.internal.h"
|
|
#include "libc/intrin/ulock.h"
|
|
#include "libc/intrin/weaken.h"
|
|
#include "libc/limits.h"
|
|
#include "libc/macros.internal.h"
|
|
#include "libc/mem/alloca.h"
|
|
#include "libc/nt/enum/processcreationflags.h"
|
|
#include "libc/nt/runtime.h"
|
|
#include "libc/nt/signals.h"
|
|
#include "libc/nt/synchronization.h"
|
|
#include "libc/nt/thread.h"
|
|
#include "libc/nt/thunk/msabi.h"
|
|
#include "libc/runtime/internal.h"
|
|
#include "libc/runtime/runtime.h"
|
|
#include "libc/runtime/stack.h"
|
|
#include "libc/runtime/syslib.internal.h"
|
|
#include "libc/sock/internal.h"
|
|
#include "libc/stdalign.internal.h"
|
|
#include "libc/str/str.h"
|
|
#include "libc/sysv/consts/arch.h"
|
|
#include "libc/sysv/consts/clone.h"
|
|
#include "libc/sysv/consts/futex.h"
|
|
#include "libc/sysv/consts/nr.h"
|
|
#include "libc/sysv/consts/nrlinux.h"
|
|
#include "libc/sysv/errfuns.h"
|
|
#include "libc/thread/freebsd.internal.h"
|
|
#include "libc/thread/openbsd.internal.h"
|
|
#include "libc/thread/thread.h"
|
|
#include "libc/thread/tls.h"
|
|
#include "libc/thread/tls2.internal.h"
|
|
#include "libc/thread/xnu.internal.h"
|
|
|
|
#define kMaxThreadIds 32768
|
|
#define kMinThreadId 262144
|
|
|
|
#define AMD64_SET_FSBASE 129
|
|
#define AMD64_SET_GSBASE 131
|
|
|
|
#define __NR_thr_new 455
|
|
#define __NR_clone_linux 56
|
|
#define __NR__lwp_create 309
|
|
#define __NR_getcontext_netbsd 307
|
|
#define __NR_bsdthread_create 0x02000168
|
|
#define __NR_thread_fast_set_cthread_self 0x03000003
|
|
#define PTHREAD_START_CUSTOM_XNU 0x01000000
|
|
#define LWP_DETACHED 0x00000040
|
|
#define LWP_SUSPENDED 0x00000080
|
|
|
|
struct CloneArgs {
|
|
_Alignas(16) union {
|
|
struct {
|
|
int tid;
|
|
int this;
|
|
};
|
|
uint32_t utid;
|
|
int64_t tid64;
|
|
};
|
|
int *ptid;
|
|
int *ctid;
|
|
int *ztid;
|
|
char *tls;
|
|
int (*func)(void *, int);
|
|
void *arg;
|
|
};
|
|
|
|
int sys_set_tls();
|
|
int __stack_call(void *, int, long, long, int (*)(void *, int), void *);
|
|
|
|
static struct CloneArgs *AllocateCloneArgs(char *stk, size_t stksz) {
|
|
return (struct CloneArgs *)(((uintptr_t)(stk + stksz) -
|
|
sizeof(struct CloneArgs)) &
|
|
-16);
|
|
}
|
|
|
|
#ifdef __x86_64__
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
// THE NEW TECHNOLOGY
|
|
|
|
__msabi extern typeof(TlsSetValue) *const __imp_TlsSetValue;
|
|
__msabi extern typeof(ExitThread) *const __imp_ExitThread;
|
|
__msabi extern typeof(WakeByAddressAll) *const __imp_WakeByAddressAll;
|
|
|
|
static textwindows dontinstrument wontreturn void //
|
|
WinThreadEntry(int rdi, // rcx
|
|
int rsi, // rdx
|
|
int rdx, // r8
|
|
struct CloneArgs *wt) { // r9
|
|
int rc;
|
|
if (wt->tls) __set_tls_win32(wt->tls);
|
|
*wt->ctid = wt->tid;
|
|
rc = __stack_call(wt->arg, wt->tid, 0, 0, wt->func, wt);
|
|
// we can now clear ctid directly since we're no longer using our own
|
|
// stack memory, which can now be safely free'd by the parent thread.
|
|
*wt->ztid = 0;
|
|
__imp_WakeByAddressAll(wt->ztid);
|
|
// since we didn't indirect this function through NT2SYSV() it's not
|
|
// safe to simply return, and as such, we need ExitThread().
|
|
__imp_ExitThread(rc);
|
|
__builtin_unreachable();
|
|
}
|
|
|
|
static textwindows errno_t CloneWindows(int (*func)(void *, int), char *stk,
|
|
size_t stksz, int flags, void *arg,
|
|
void *tls, int *ptid, int *ctid) {
|
|
int64_t h;
|
|
struct CloneArgs *wt;
|
|
wt = AllocateCloneArgs(stk, stksz);
|
|
wt->ctid = flags & CLONE_CHILD_SETTID ? ctid : &wt->tid;
|
|
wt->ztid = flags & CLONE_CHILD_CLEARTID ? ctid : &wt->tid;
|
|
wt->func = func;
|
|
wt->arg = arg;
|
|
wt->tls = flags & CLONE_SETTLS ? tls : 0;
|
|
if ((h = CreateThread(&kNtIsInheritable, 65536, (void *)WinThreadEntry, wt,
|
|
kNtStackSizeParamIsAReservation, &wt->utid))) {
|
|
if (flags & CLONE_SETTLS) {
|
|
struct CosmoTib *tib = tls;
|
|
tib->tib_syshand = h;
|
|
}
|
|
if (flags & CLONE_PARENT_SETTID) {
|
|
*ptid = wt->tid;
|
|
}
|
|
return 0;
|
|
} else {
|
|
return __dos2errno(GetLastError());
|
|
}
|
|
}
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
// XNU'S NOT UNIX
|
|
|
|
void XnuThreadThunk(void *pthread, // rdi x0
|
|
int machport, // rsi x1
|
|
void *(*func)(void *), // rdx x2
|
|
void *arg, // rcx x3
|
|
intptr_t *stack, // r8 x4
|
|
unsigned xnuflags); // r9 x5
|
|
asm("XnuThreadThunk:\n\t"
|
|
"xor\t%ebp,%ebp\n\t"
|
|
"mov\t%r8,%rsp\n\t"
|
|
"push\t%rax\n\t"
|
|
"jmp\tXnuThreadMain\n\t"
|
|
".size\tXnuThreadThunk,.-XnuThreadThunk");
|
|
__attribute__((__used__))
|
|
|
|
static wontreturn void
|
|
XnuThreadMain(void *pthread, // rdi
|
|
int tid, // rsi
|
|
int (*func)(void *arg, int tid), // rdx
|
|
void *arg, // rcx
|
|
struct CloneArgs *wt, // r8
|
|
unsigned xnuflags) { // r9
|
|
int ax;
|
|
wt->tid = tid;
|
|
*wt->ctid = tid;
|
|
*wt->ptid = tid;
|
|
|
|
if (wt->tls) {
|
|
// XNU uses the same 0x30 offset as the WIN32 TIB x64. They told the
|
|
// Go team at Google that they Apply stands by our ability to use it
|
|
// https://github.com/golang/go/issues/23617#issuecomment-376662373
|
|
asm volatile("syscall"
|
|
: "=a"(ax)
|
|
: "0"(__NR_thread_fast_set_cthread_self), "D"(wt->tls - 0x30)
|
|
: "rcx", "rdx", "r8", "r9", "r10", "r11", "memory", "cc");
|
|
}
|
|
|
|
func(arg, tid);
|
|
|
|
// we no longer use the stack after this point
|
|
// %rax = int bsdthread_terminate(%rdi = void *stackaddr,
|
|
// %rsi = size_t freesize,
|
|
// %rdx = uint32_t port,
|
|
// %r10 = uint32_t sem);
|
|
asm volatile("movl\t$0,(%%rsi)\n\t" // *wt->ztid = 0
|
|
"mov\t$0x101,%%edi\n\t" // wake all
|
|
"xor\t%%edx,%%edx\n\t" // wake_value
|
|
"mov\t$0x02000204,%%eax\n\t" // ulock_wake()
|
|
"syscall\n\t" //
|
|
"xor\t%%edi,%%edi\n\t" // freeaddr
|
|
"xor\t%%esi,%%esi\n\t" // freesize
|
|
"xor\t%%edx,%%edx\n\t" // kport
|
|
"xor\t%%r10d,%%r10d\n\t" // joinsem
|
|
"mov\t$0x02000169,%%eax\n\t" // bsdthread_terminate()
|
|
"syscall"
|
|
: /* no outputs */
|
|
: "S"(wt->ztid)
|
|
: "rax", "rcx", "r10", "r11", "memory");
|
|
__builtin_unreachable();
|
|
}
|
|
|
|
static errno_t CloneXnu(int (*fn)(void *), char *stk, size_t stksz, int flags,
|
|
void *arg, void *tls, int *ptid, int *ctid) {
|
|
static bool once;
|
|
struct CloneArgs *wt;
|
|
if (!once) {
|
|
npassert(sys_bsdthread_register(XnuThreadThunk, 0, 0, 0, 0, 0, 0) != -1);
|
|
once = true;
|
|
}
|
|
wt = AllocateCloneArgs(stk, stksz);
|
|
wt->ptid = flags & CLONE_PARENT_SETTID ? ptid : &wt->tid;
|
|
wt->ctid = flags & CLONE_CHILD_SETTID ? ctid : &wt->tid;
|
|
wt->ztid = flags & CLONE_CHILD_CLEARTID ? ctid : &wt->tid;
|
|
wt->tls = flags & CLONE_SETTLS ? tls : 0;
|
|
return sys_clone_xnu(fn, arg, wt, 0, PTHREAD_START_CUSTOM_XNU);
|
|
}
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
// OPEN BESIYATA DISHMAYA
|
|
|
|
// we can't use address sanitizer because:
|
|
// 1. __asan_handle_no_return wipes stack [todo?]
|
|
static wontreturn void OpenbsdThreadMain(void *p) {
|
|
struct CloneArgs *wt = p;
|
|
*wt->ctid = wt->tid;
|
|
wt->func(wt->arg, wt->tid);
|
|
asm volatile("mov\t%2,%%rsp\n\t" // so syscall can validate stack exists
|
|
"movl\t$0,(%%rdi)\n\t" // *wt->ztid = 0 (old stack now free'd)
|
|
"syscall\n\t" // futex(int*, op, val) will wake wait0
|
|
"xor\t%%edi,%%edi\n\t" // so kernel doesn't write to old stack
|
|
"mov\t$302,%%eax\n\t" // __threxit(int *notdead) doesn't wake
|
|
"syscall"
|
|
: "=m"(*wt->ztid)
|
|
: "a"(83), "m"(__oldstack), "D"(wt->ztid),
|
|
"S"(2 /* FUTEX_WAKE */), "d"(INT_MAX)
|
|
: "rcx", "r11", "memory");
|
|
__builtin_unreachable();
|
|
}
|
|
|
|
static errno_t CloneOpenbsd(int (*func)(void *, int), char *stk, size_t stksz,
|
|
int flags, void *arg, void *tls, int *ptid,
|
|
int *ctid) {
|
|
int rc;
|
|
intptr_t sp;
|
|
struct __tfork *tf;
|
|
struct CloneArgs *wt;
|
|
sp = (intptr_t)stk + stksz;
|
|
sp -= sizeof(struct __tfork);
|
|
sp &= -alignof(struct __tfork);
|
|
tf = (struct __tfork *)sp;
|
|
sp -= sizeof(struct CloneArgs);
|
|
sp &= -alignof(struct CloneArgs);
|
|
wt = (struct CloneArgs *)sp;
|
|
wt->ctid = flags & CLONE_CHILD_SETTID ? ctid : &wt->tid;
|
|
wt->ztid = flags & CLONE_CHILD_CLEARTID ? ctid : &wt->tid;
|
|
wt->arg = arg;
|
|
wt->func = func;
|
|
tf->tf_stack = (char *)wt - 8;
|
|
tf->tf_tcb = flags & CLONE_SETTLS ? tls : 0;
|
|
tf->tf_tid = &wt->tid;
|
|
if ((rc = __tfork_thread(tf, sizeof(*tf), OpenbsdThreadMain, wt)) >= 0) {
|
|
npassert(rc);
|
|
if (flags & CLONE_PARENT_SETTID) {
|
|
*ptid = rc;
|
|
}
|
|
return 0;
|
|
} else {
|
|
return -rc;
|
|
}
|
|
}
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
// NET BESIYATA DISHMAYA
|
|
|
|
static wontreturn void NetbsdThreadMain(void *arg, // rdi
|
|
int (*func)(void *, int), // rsi
|
|
int *tid, // rdx
|
|
int *ctid, // rcx
|
|
int *ztid) { // r9
|
|
int ax, dx;
|
|
// TODO(jart): Why are we seeing flakes where *tid is zero?
|
|
// ax = *tid;
|
|
ax = sys_gettid();
|
|
*ctid = ax;
|
|
func(arg, ax);
|
|
// we no longer use the stack after this point
|
|
// %eax = int __lwp_exit(void);
|
|
asm volatile("movl\t$0,%2\n\t" // *wt->ztid = 0
|
|
"syscall" // __lwp_exit()
|
|
: "=a"(ax), "=d"(dx), "=m"(*ztid)
|
|
: "0"(310)
|
|
: "rcx", "r11", "memory");
|
|
__builtin_unreachable();
|
|
}
|
|
|
|
static int CloneNetbsd(int (*func)(void *, int), char *stk, size_t stksz,
|
|
int flags, void *arg, void *tls, int *ptid, int *ctid) {
|
|
// NetBSD has its own clone() and it works, but it's technically a
|
|
// second-class API, intended to help Linux folks migrate to this.
|
|
bool failed;
|
|
int ax, *tid;
|
|
intptr_t dx, sp;
|
|
static bool once;
|
|
struct ucontext_netbsd *ctx;
|
|
static struct ucontext_netbsd netbsd_clone_template;
|
|
|
|
// memoize arbitrary valid processor state structure
|
|
if (!once) {
|
|
asm volatile(CFLAG_ASM("syscall")
|
|
: CFLAG_CONSTRAINT(failed), "=a"(ax)
|
|
: "1"(__NR_getcontext_netbsd), "D"(&netbsd_clone_template)
|
|
: "rcx", "rdx", "r8", "r9", "r10", "r11", "memory");
|
|
npassert(!failed);
|
|
once = true;
|
|
}
|
|
sp = (intptr_t)(stk + stksz);
|
|
|
|
// allocate memory for tid
|
|
sp -= sizeof(int);
|
|
sp = sp & -alignof(int);
|
|
tid = (int *)sp;
|
|
*tid = 0;
|
|
|
|
// align the stack
|
|
sp = sp & -16;
|
|
|
|
// simulate call to misalign stack and ensure backtrace looks good
|
|
sp -= 8;
|
|
*(intptr_t *)sp = (intptr_t)CloneNetbsd + 1;
|
|
|
|
// place the giant 784 byte ucontext structure in the red zone!
|
|
// it only has to live long enough for the thread to come alive
|
|
ctx = (struct ucontext_netbsd *)((sp - sizeof(struct ucontext_netbsd)) &
|
|
-alignof(struct ucontext_netbsd));
|
|
|
|
// pass parameters in process state
|
|
memcpy(ctx, &netbsd_clone_template, sizeof(*ctx));
|
|
ctx->uc_link = 0;
|
|
ctx->uc_mcontext.rbp = 0;
|
|
ctx->uc_mcontext.rsp = sp;
|
|
ctx->uc_mcontext.rip = (intptr_t)NetbsdThreadMain;
|
|
ctx->uc_mcontext.rdi = (intptr_t)arg;
|
|
ctx->uc_mcontext.rsi = (intptr_t)func;
|
|
ctx->uc_mcontext.rdx = (intptr_t)tid;
|
|
ctx->uc_mcontext.rcx = (intptr_t)(flags & CLONE_CHILD_SETTID ? ctid : tid);
|
|
ctx->uc_mcontext.r8 = (intptr_t)(flags & CLONE_CHILD_CLEARTID ? ctid : tid);
|
|
ctx->uc_flags |= _UC_STACK;
|
|
ctx->uc_stack.ss_sp = stk;
|
|
ctx->uc_stack.ss_size = stksz;
|
|
ctx->uc_stack.ss_flags = 0;
|
|
if (flags & CLONE_SETTLS) {
|
|
ctx->uc_flags |= _UC_TLSBASE;
|
|
ctx->uc_mcontext._mc_tlsbase = (intptr_t)tls;
|
|
}
|
|
|
|
// perform the system call
|
|
asm volatile(CFLAG_ASM("syscall")
|
|
: CFLAG_CONSTRAINT(failed), "=a"(ax), "=d"(dx)
|
|
: "1"(__NR__lwp_create), "D"(ctx), "S"(LWP_DETACHED), "2"(tid)
|
|
: "rcx", "r8", "r9", "r10", "r11", "memory");
|
|
if (!failed) {
|
|
npassert(*tid);
|
|
if (flags & CLONE_PARENT_SETTID) {
|
|
*ptid = *tid;
|
|
}
|
|
return 0;
|
|
} else {
|
|
return ax;
|
|
}
|
|
}
|
|
|
|
#endif /* __x86_64__ */
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
// FREE BESIYATA DISHMAYA
|
|
|
|
static wontreturn void FreebsdThreadMain(void *p) {
|
|
struct CloneArgs *wt = p;
|
|
#ifdef __aarch64__
|
|
asm volatile("mov\tx28,%0" : /* no outputs */ : "r"(wt->tls));
|
|
#elif defined(__x86_64__)
|
|
if (__tls_morphed) {
|
|
sys_set_tls(AMD64_SET_GSBASE, wt->tls);
|
|
}
|
|
#endif
|
|
*wt->ctid = wt->tid;
|
|
wt->func(wt->arg, wt->tid);
|
|
// we no longer use the stack after this point
|
|
// void thr_exit(%rdi = long *state);
|
|
#ifdef __x86_64__
|
|
asm volatile("movl\t$0,%0\n\t" // *wt->ztid = 0
|
|
"syscall\n\t" // _umtx_op(wt->ztid, WAKE, INT_MAX)
|
|
"movl\t$431,%%eax\n\t" // thr_exit(long *nonzeroes_and_wake)
|
|
"xor\t%%edi,%%edi\n\t" // sad we can't use this free futex op
|
|
"syscall\n\t" // thr_exit() fails if thread is orphaned
|
|
"movl\t$1,%%eax\n\t" // _exit()
|
|
"syscall" //
|
|
: "=m"(*wt->ztid)
|
|
: "a"(454), "D"(wt->ztid), "S"(UMTX_OP_WAKE), "d"(INT_MAX)
|
|
: "rcx", "r8", "r9", "r10", "r11", "memory");
|
|
#elif defined(__aarch64__)
|
|
register long x0 asm("x0") = (long)wt->ztid;
|
|
register long x1 asm("x1") = UMTX_OP_WAKE;
|
|
register long x2 asm("x2") = INT_MAX;
|
|
register long x8 asm("x8") = 454; // _umtx_op
|
|
asm volatile("str\twzr,%0\n\t" // *wt->ztid = 0
|
|
"svc\t0\n\t" // _umtx_op(wt->ztid, WAKE, INT_MAX)
|
|
"mov\tx0,#0\n\t" // arg0 = 0
|
|
"mov\tx8,#431\n\t" // thr_exit
|
|
"svc\t0\n\t" // thr_exit(long *nonzeroes_and_wake = 0)
|
|
"mov\tx8,#1\n\t" // _exit
|
|
"svc\t0" // _exit(long *nonzeroes_and_wake = 0)
|
|
: "=m"(*wt->ztid)
|
|
: "r"(x0), "r"(x1), "r"(x2), "r"(x8));
|
|
#else
|
|
#error "unsupported architecture"
|
|
#endif
|
|
__builtin_unreachable();
|
|
}
|
|
|
|
static errno_t CloneFreebsd(int (*func)(void *, int), char *stk, size_t stksz,
|
|
int flags, void *arg, void *tls, int *ptid,
|
|
int *ctid) {
|
|
int64_t tid;
|
|
struct CloneArgs *wt;
|
|
wt = AllocateCloneArgs(stk, stksz);
|
|
wt->ctid = flags & CLONE_CHILD_SETTID ? ctid : &wt->tid;
|
|
wt->ztid = flags & CLONE_CHILD_CLEARTID ? ctid : &wt->tid;
|
|
wt->tls = tls;
|
|
wt->func = func;
|
|
wt->arg = arg;
|
|
struct thr_param params = {
|
|
.start_func = FreebsdThreadMain,
|
|
.arg = wt,
|
|
.stack_base = stk,
|
|
.stack_size = (uintptr_t)wt - (uintptr_t)stk,
|
|
.tls_base = flags & CLONE_SETTLS ? tls : 0,
|
|
.tls_size = 64,
|
|
.child_tid = &wt->tid64,
|
|
.parent_tid = &tid,
|
|
};
|
|
#ifdef __x86_64__
|
|
int ax;
|
|
bool failed;
|
|
asm volatile(CFLAG_ASM("syscall")
|
|
: CFLAG_CONSTRAINT(failed), "=a"(ax)
|
|
: "1"(__NR_thr_new), "D"(¶ms), "S"(sizeof(params))
|
|
: "rcx", "rdx", "r8", "r9", "r10", "r11", "memory");
|
|
if (failed) return ax;
|
|
#elif defined(__aarch64__)
|
|
register long x0 asm("x0") = (long)¶ms;
|
|
register long x1 asm("x1") = sizeof(params);
|
|
register int x8 asm("x8") = 0x1c7; // thr_new
|
|
asm volatile("svc\t0" : "+r"(x0) : "r"(x1), "r"(x8) : "memory");
|
|
if (x0) return x0;
|
|
#else
|
|
#error "unsupported architecture"
|
|
#endif
|
|
if (flags & CLONE_PARENT_SETTID) *ptid = tid;
|
|
return 0;
|
|
}
|
|
|
|
#ifdef __aarch64__
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
// APPLE SILICON
|
|
|
|
static void *SiliconThreadMain(void *arg) {
|
|
struct CloneArgs *wt = arg;
|
|
asm volatile("mov\tx28,%0" : /* no outputs */ : "r"(wt->tls));
|
|
*wt->ctid = wt->this;
|
|
__stack_call(wt->arg, wt->this, 0, 0, wt->func, wt);
|
|
*wt->ztid = 0;
|
|
ulock_wake(UL_COMPARE_AND_WAIT | ULF_WAKE_ALL, wt->ztid, 0);
|
|
return 0;
|
|
}
|
|
|
|
static errno_t CloneSilicon(int (*fn)(void *, int), char *stk, size_t stksz,
|
|
int flags, void *arg, void *tls, int *ptid,
|
|
int *ctid) {
|
|
void *attr;
|
|
errno_t res;
|
|
unsigned tid;
|
|
pthread_t th;
|
|
size_t babystack;
|
|
struct CloneArgs *wt;
|
|
static atomic_uint tids;
|
|
wt = AllocateCloneArgs(stk, stksz);
|
|
tid = atomic_fetch_add_explicit(&tids, 1, memory_order_acq_rel);
|
|
wt->this = tid = (tid & (kMaxThreadIds - 1)) + kMinThreadId;
|
|
wt->ctid = flags & CLONE_CHILD_SETTID ? ctid : &wt->tid;
|
|
wt->ztid = flags & CLONE_CHILD_CLEARTID ? ctid : &wt->tid;
|
|
wt->tls = flags & CLONE_SETTLS ? tls : 0;
|
|
wt->func = fn;
|
|
wt->arg = arg;
|
|
babystack = __syslib->__pthread_stack_min;
|
|
#pragma GCC push_options
|
|
#pragma GCC diagnostic ignored "-Walloca-larger-than="
|
|
attr = alloca(__syslib->__sizeof_pthread_attr_t);
|
|
#pragma GCC pop_options
|
|
unassert(!__syslib->__pthread_attr_init(attr));
|
|
unassert(!__syslib->__pthread_attr_setguardsize(attr, 0));
|
|
unassert(!__syslib->__pthread_attr_setstacksize(attr, babystack));
|
|
if (!(res = __syslib->__pthread_create(&th, attr, SiliconThreadMain, wt)) &&
|
|
(flags & CLONE_PARENT_SETTID)) {
|
|
*ptid = tid;
|
|
if (flags & CLONE_SETTLS) {
|
|
struct CosmoTib *tib = tls;
|
|
tib[-1].tib_syshand = th;
|
|
}
|
|
}
|
|
unassert(!__syslib->__pthread_attr_destroy(attr));
|
|
return res;
|
|
}
|
|
|
|
#endif /* __aarch64__ */
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
// GNU/SYSTEMD
|
|
|
|
struct LinuxCloneArgs {
|
|
int (*func)(void *, int);
|
|
void *arg;
|
|
char *tls;
|
|
int ctid;
|
|
};
|
|
|
|
int sys_clone_linux(int flags, // rdi
|
|
long sp, // rsi
|
|
int *ptid, // rdx
|
|
int *ctid, // rcx
|
|
void *tls, // r8
|
|
void *func, // r9
|
|
void *arg); // 8(rsp)
|
|
|
|
static int LinuxThreadEntry(void *arg, int tid) {
|
|
struct LinuxCloneArgs *wt = arg;
|
|
sys_set_tls(ARCH_SET_GS, wt->tls);
|
|
return wt->func(wt->arg, tid);
|
|
}
|
|
|
|
static int CloneLinux(int (*func)(void *arg, int rc), char *stk, size_t stksz,
|
|
int flags, void *arg, void *tls, int *ptid, int *ctid) {
|
|
int rc;
|
|
long sp;
|
|
struct LinuxCloneArgs *wt;
|
|
sp = (intptr_t)(stk + stksz);
|
|
sp -= sizeof(struct LinuxCloneArgs);
|
|
// align the stack
|
|
#ifdef __aarch64__
|
|
sp = sp & -128; // for kernel 4.6 and earlier
|
|
#else
|
|
sp = sp & -16;
|
|
#endif
|
|
wt = (struct LinuxCloneArgs *)sp;
|
|
#ifdef __x86_64__
|
|
if ((flags & CLONE_SETTLS) && __tls_morphed) {
|
|
flags &= ~CLONE_SETTLS;
|
|
wt->arg = arg;
|
|
wt->tls = tls;
|
|
wt->func = func;
|
|
func = LinuxThreadEntry;
|
|
arg = wt;
|
|
}
|
|
#endif
|
|
if (~flags & CLONE_CHILD_SETTID) {
|
|
flags |= CLONE_CHILD_SETTID;
|
|
ctid = &wt->ctid;
|
|
}
|
|
if ((rc = sys_clone_linux(flags, sp, ptid, ctid, tls, func, arg)) >= 0) {
|
|
// clone() is documented as setting ptid before return
|
|
return 0;
|
|
} else {
|
|
return -rc;
|
|
}
|
|
}
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
// COSMOPOLITAN
|
|
|
|
/**
|
|
* Creates thread without malloc being linked.
|
|
*
|
|
* If you use clone() you're on your own. Example:
|
|
*
|
|
* int worker(void *arg) { return 0; }
|
|
* struct CosmoTib tib = {.tib_self = &tib, .tib_tid = -1};
|
|
* atomic_int tid;
|
|
* char *stk = NewCosmoStack();
|
|
* clone(worker, stk, GetStackSize() - 16,
|
|
* CLONE_VM | CLONE_THREAD | CLONE_FS | CLONE_FILES |
|
|
* CLONE_SYSVSEM | CLONE_SIGHAND | CLONE_PARENT_SETTID |
|
|
* CLONE_CHILD_SETTID | CLONE_CHILD_CLEARTID | CLONE_SETTLS,
|
|
* arg, &tid, &tib, &tib.tib_tid);
|
|
* while (atomic_load(&tid) == 0) sched_yield();
|
|
* // thread is known
|
|
* while (atomic_load(&tib.tib_tid) < 0) sched_yield();
|
|
* // thread is running
|
|
* while (atomic_load(&tib.tib_tid) > 0) sched_yield();
|
|
* // thread has terminated
|
|
* FreeCosmoStack(stk);
|
|
*
|
|
* Threads are created in a detached manner. They currently can't be
|
|
* synchronized using wait() or posix signals. Threads created by this
|
|
* function should be synchronized using shared memory operations.
|
|
*
|
|
* Any memory that's required by this system call wrapper is allocated
|
|
* to the top of your stack. This shouldn't be more than 128 bytes.
|
|
*
|
|
* Your function is called from within the stack you specify. A return
|
|
* address is pushed onto your stack, that causes returning to jump to
|
|
* _Exit1() which terminates the thread. Even though the callback says
|
|
* it supports a return code, that'll only work on Linux and Windows.
|
|
*
|
|
* This function follows the same ABI convention as the Linux userspace
|
|
* libraries, with a few small changes. The varargs has been removed to
|
|
* help prevent broken code, and the stack size and tls size parameters
|
|
* are introduced for compatibility with FreeBSD.
|
|
*
|
|
* To keep this system call lightweight, only the thread creation use
|
|
* case is polyfilled across platforms. For example, if you want fork
|
|
* that works on OpenBSD for example, don't do it with clone(SIGCHLD)
|
|
* and please just call fork(). Even if you do that on Linux, it will
|
|
* effectively work around libc features like atfork(), so that means
|
|
* other calls like getpid() may return incorrect values.
|
|
*
|
|
* @param func is your callback function, which this wrapper requires
|
|
* not be null, otherwise EINVAL is raised. It is passed two args
|
|
* within the child thread: (1) the caller-supplied `arg` and (2)
|
|
* the new tid is always passed in the second arg for convenience
|
|
*
|
|
* @param stk points to the bottom of a caller allocated stack, which
|
|
* must be allocated via mmap() using the MAP_STACK flag, or else
|
|
* you won't get optimal performance and it won't work on OpenBSD
|
|
*
|
|
* @param stksz is the size of that stack in bytes, we recommend that
|
|
* that this be set to GetStackSize() or else memory safety tools
|
|
* like kprintf() can't do as good and quick of a job; this value
|
|
* must be 16-aligned plus it must be at least 4192 bytes in size
|
|
* and it's advised to have the bottom-most page, be a guard page
|
|
*
|
|
* @param flags which SHOULD always have all of these flags:
|
|
*
|
|
* - `CLONE_THREAD`
|
|
* - `CLONE_VM`
|
|
* - `CLONE_FS`
|
|
* - `CLONE_FILES`
|
|
* - `CLONE_SIGHAND`
|
|
* - `CLONE_SYSVSEM`
|
|
*
|
|
* This system call wrapper is intended for threads, and as such, we
|
|
* won't polyfill Linux's ability to simulate unrelated calls (e.g.
|
|
* fork, vfork) via clone() on other platforms. Please just call
|
|
* fork() and vfork() when that's what you want.
|
|
*
|
|
* Your `flags` may also optionally also additionally bitwise-OR any
|
|
* combination of the following additional flags:
|
|
*
|
|
* - `CLONE_CHILD_SETTID` must be specified if you intend to set the
|
|
* `ctid` argument, which will updated with the child tid once the
|
|
* child has started.
|
|
*
|
|
* - `CLONE_PARENT_SETTID` must be specified if you intend to set
|
|
* the `ptid` argument, and it is updated at the most opportune
|
|
* moment. On all platforms except XNU x86, this happens before
|
|
* clone() returns. But since it might not be available yet you
|
|
* need to use pthread_getunique_np() to obtain it.
|
|
*
|
|
* - `CLONE_CHILD_CLEARTID` causes `*ctid = 0` upon child thread
|
|
* termination. This is used to implement join so that the parent
|
|
* may know when it's safe to free the child's stack memory, and
|
|
* as such, is guaranteed to happen AFTER the child thread has
|
|
* either terminated or has finished using its stack memory
|
|
*
|
|
* - `CLONE_SETTLS` is needed if you intend to specify the `tls`
|
|
* argument, which after thread creation may be accessed using
|
|
* __get_tls(). Doing this means that `errno`, gettid(), etc.
|
|
* correctly work. Caveat emptor if you choose not to do this.
|
|
*
|
|
* @param arg is passed as an argument to `func` in the child thread
|
|
* @param tls may be used to set the thread local storage segment;
|
|
* this parameter is ignored if `CLONE_SETTLS` is not set
|
|
* @param ctid lets the child receive its thread id without having to
|
|
* call gettid() and is ignored if `CLONE_CHILD_SETTID` isn't set
|
|
* @return 0 on success, or errno on errno
|
|
*/
|
|
errno_t clone(void *func, void *stk, size_t stksz, int flags, void *arg,
|
|
void *ptid, void *tls, void *ctid) {
|
|
int rc;
|
|
|
|
if (flags & CLONE_THREAD) {
|
|
__enable_threads();
|
|
}
|
|
|
|
if (!func) {
|
|
rc = EINVAL;
|
|
} else if (IsAsan() &&
|
|
(((flags & CLONE_SETTLS) && !__asan_is_valid(tls, 64)) ||
|
|
((flags & CLONE_PARENT_SETTID) &&
|
|
!__asan_is_valid(ptid, sizeof(int))) ||
|
|
((flags & CLONE_CHILD_SETTID) &&
|
|
!__asan_is_valid(ctid, sizeof(int))))) {
|
|
rc = EFAULT;
|
|
} else if (IsLinux()) {
|
|
rc = CloneLinux(func, stk, stksz, flags, arg, tls, ptid, ctid);
|
|
} else if (!IsTiny() &&
|
|
(flags & ~(CLONE_SETTLS | CLONE_PARENT_SETTID |
|
|
CLONE_CHILD_SETTID | CLONE_CHILD_CLEARTID)) !=
|
|
(CLONE_THREAD | CLONE_VM | CLONE_FS | CLONE_FILES |
|
|
CLONE_SIGHAND | CLONE_SYSVSEM)) {
|
|
rc = EINVAL;
|
|
} else if (IsXnu()) {
|
|
#ifdef __x86_64__
|
|
rc = CloneXnu(func, stk, stksz, flags, arg, tls, ptid, ctid);
|
|
#elif defined(__aarch64__)
|
|
rc = CloneSilicon(func, stk, stksz, flags, arg, tls, ptid, ctid);
|
|
#else
|
|
#error "unsupported architecture"
|
|
#endif
|
|
} else if (IsFreebsd()) {
|
|
rc = CloneFreebsd(func, stk, stksz, flags, arg, tls, ptid, ctid);
|
|
#ifdef __x86_64__
|
|
} else if (IsNetbsd()) {
|
|
rc = CloneNetbsd(func, stk, stksz, flags, arg, tls, ptid, ctid);
|
|
} else if (IsOpenbsd()) {
|
|
rc = CloneOpenbsd(func, stk, stksz, flags, arg, tls, ptid, ctid);
|
|
} else if (IsWindows()) {
|
|
rc = CloneWindows(func, stk, stksz, flags, arg, tls, ptid, ctid);
|
|
#endif /* __x86_64__ */
|
|
} else {
|
|
rc = ENOSYS;
|
|
}
|
|
|
|
if (SupportsBsd() && rc == EPROCLIM) {
|
|
rc = EAGAIN;
|
|
}
|
|
|
|
return rc;
|
|
}
|