Skip to content

Aarch64 darwin #16

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 3 commits into
base: main
Choose a base branch
from
Open

Aarch64 darwin #16

Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
3890c66
add assembly macros
ricardo-valero Feb 8, 2025
2201856
add aarch64 darwin support
ricardo-valero Feb 8, 2025
6684f81
it works :)
ricardo-valero Feb 8, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
198 changes: 145 additions & 53 deletions coroutine.c
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -82,75 +82,125 @@ typedef enum {
SM_WRITE,
} Sleep_Mode;

#if __linux__ && __x86_64__
// Linux x86_64 call convention
// %rdi, %rsi, %rdx, %rcx, %r8, and %r9

#define STORE_REGISTERS \
"pushq %rdi\n" \
"pushq %rbp\n" \
"pushq %rbx\n" \
"pushq %r12\n" \
"pushq %r13\n" \
"pushq %r14\n" \
"pushq %r15\n"

#define SLEEP_NONE \
"movq %rsp, %rdi\n" \
"movq $0, %rsi\n" \
"jmp coroutine_switch_context\n"

#define SLEEP_READ \
"movq %rdi, %rdx\n" \
"movq %rsp, %rdi\n" \
"movq $1, %rsi\n" \
"jmp coroutine_switch_context\n"

#define SLEEP_WRITE \
"movq %rdi, %rdx\n" \
"movq %rsp, %rdi\n" \
"movq $2, %rsi\n" \
"jmp coroutine_switch_context\n"

#define RESTORE_REGISTERS \
"movq %rdi, %rsp\n" \
"popq %r15\n" \
"popq %r14\n" \
"popq %r13\n" \
"popq %r12\n" \
"popq %rbx\n" \
"popq %rbp\n" \
"popq %rdi\n" \
"ret\n"

#elif __APPLE__ && __aarch64__

#define STORE_REGISTERS \
"sub sp, sp, #240\n" \
"stp q8, q9, [sp, #0]\n" \
"stp q10, q11, [sp, #32]\n" \
"stp q12, q13, [sp, #64]\n" \
"stp q14, q15, [sp, #96]\n" \
"stp x19, x20, [sp, #128]\n" \
"stp x21, x22, [sp, #144]\n" \
"stp x23, x24, [sp, #160]\n" \
"stp x25, x26, [sp, #176]\n" \
"stp x27, x28, [sp, #192]\n" \
"stp x29, x30, [sp, #208]\n" \
"mov x1, x30\n" \
"str x30, [sp, #224]\n" \
"str x0, [sp, #232]\n"

#define SLEEP_NONE \
"mov x0, sp\n" \
"mov x1, #0\n" \
"b _coroutine_switch_context\n"

#define SLEEP_READ \
"mov x2, x0\n" \
"mov x0, sp\n" \
"mov x1, #1\n" \
"b _coroutine_switch_context\n"

#define SLEEP_WRITE \
"mov x2, x0\n" \
"mov x0, sp\n" \
"mov x1, #2\n" \
"b _coroutine_switch_context\n"

#define RESTORE_REGISTERS \
"mov sp, x0\n" \
"ldp q8, q9, [sp, #0]\n" \
"ldp q10, q11, [sp, #32]\n" \
"ldp q12, q13, [sp, #64]\n" \
"ldp q14, q15, [sp, #96]\n" \
"ldp x19, x20, [sp, #128]\n" \
"ldp x21, x22, [sp, #144]\n" \
"ldp x23, x24, [sp, #160]\n" \
"ldp x25, x26, [sp, #176]\n" \
"ldp x27, x28, [sp, #192]\n" \
"ldp x29, x30, [sp, #208]\n" \
"mov x1, x30\n" \
"ldr x30, [sp, #224]\n" \
"ldr x0, [sp, #232]\n" \
"add sp, sp, #240\n" \
"ret x1\n"

#else
#error Unsupported platform
#endif

void __attribute__((naked)) coroutine_yield(void)
{
// @arch
asm(
" pushq %rdi\n"
" pushq %rbp\n"
" pushq %rbx\n"
" pushq %r12\n"
" pushq %r13\n"
" pushq %r14\n"
" pushq %r15\n"
" movq %rsp, %rdi\n" // rsp
" movq $0, %rsi\n" // sm = SM_NONE
" jmp coroutine_switch_context\n");
asm volatile (STORE_REGISTERS SLEEP_NONE ::: "memory");
}

void __attribute__((naked)) coroutine_sleep_read(int fd)
{
(void) fd;
// @arch
asm(
" pushq %rdi\n"
" pushq %rbp\n"
" pushq %rbx\n"
" pushq %r12\n"
" pushq %r13\n"
" pushq %r14\n"
" pushq %r15\n"
" movq %rdi, %rdx\n" // fd
" movq %rsp, %rdi\n" // rsp
" movq $1, %rsi\n" // sm = SM_READ
" jmp coroutine_switch_context\n");
// (void) fd;
asm volatile(STORE_REGISTERS SLEEP_READ ::: "memory");
}

void __attribute__((naked)) coroutine_sleep_write(int fd)
{
(void) fd;
// @arch
asm(
" pushq %rdi\n"
" pushq %rbp\n"
" pushq %rbx\n"
" pushq %r12\n"
" pushq %r13\n"
" pushq %r14\n"
" pushq %r15\n"
" movq %rdi, %rdx\n" // fd
" movq %rsp, %rdi\n" // rsp
" movq $2, %rsi\n" // sm = SM_WRITE
" jmp coroutine_switch_context\n");
// (void) fd;
asm volatile(STORE_REGISTERS SLEEP_WRITE ::: "memory");
}

void __attribute__((naked)) coroutine_restore_context(void *rsp)
{
// @arch
(void)rsp;
asm(
" movq %rdi, %rsp\n"
" popq %r15\n"
" popq %r14\n"
" popq %r13\n"
" popq %r12\n"
" popq %rbx\n"
" popq %rbp\n"
" popq %rdi\n"
" ret\n");
// (void)rsp;
asm volatile(RESTORE_REGISTERS ::: "memory");
}

void coroutine_switch_context(void *rsp, Sleep_Mode sm, int fd)
Expand Down Expand Up @@ -245,12 +295,20 @@ void coroutine_go(void (*f)(void*), void *arg)
} else {
da_append(&contexts, ((Context){0}));
id = contexts.count-1;
contexts.items[id].stack_base = mmap(NULL, STACK_CAPACITY, PROT_WRITE|PROT_READ, MAP_PRIVATE|MAP_STACK|MAP_ANONYMOUS|MAP_GROWSDOWN, -1, 0);
contexts.items[id].stack_base = mmap(NULL, STACK_CAPACITY, PROT_WRITE | PROT_READ,
// @arch
#ifdef __APPLE__
MAP_PRIVATE | MAP_ANONYMOUS,
#else
MAP_PRIVATE | MAP_STACK | MAP_ANONYMOUS | MAP_GROWSDOWN,
#endif
-1, 0);
assert(contexts.items[id].stack_base != MAP_FAILED);
}

void **rsp = (void**)((char*)contexts.items[id].stack_base + STACK_CAPACITY);
// @arch
#ifdef __x86_64__
*(--rsp) = coroutine__finish_current;
*(--rsp) = f;
*(--rsp) = arg; // push rdi
Expand All @@ -260,6 +318,40 @@ void coroutine_go(void (*f)(void*), void *arg)
*(--rsp) = 0; // push r13
*(--rsp) = 0; // push r14
*(--rsp) = 0; // push r15
#elif __aarch64__
*(--rsp) = arg;
*(--rsp) = coroutine__finish_current;
*(--rsp) = f; // push r0
*(--rsp) = 0; // push r29
*(--rsp) = 0; // push r28
*(--rsp) = 0; // push r27
*(--rsp) = 0; // push r26
*(--rsp) = 0; // push r25
*(--rsp) = 0; // push r24
*(--rsp) = 0; // push r23
*(--rsp) = 0; // push r22
*(--rsp) = 0; // push r21
*(--rsp) = 0; // push r20
*(--rsp) = 0; // push r19
*(--rsp) = 0; // push v15
*(--rsp) = 0;
*(--rsp) = 0; // push v14
*(--rsp) = 0;
*(--rsp) = 0; // push v13
*(--rsp) = 0;
*(--rsp) = 0; // push v12
*(--rsp) = 0;
*(--rsp) = 0; // push v11
*(--rsp) = 0;
*(--rsp) = 0; // push v10
*(--rsp) = 0;
*(--rsp) = 0; // push v09
*(--rsp) = 0;
*(--rsp) = 0; // push v08
*(--rsp) = 0;
#else
#error Unsupported architecture
#endif
contexts.items[id].rsp = rsp;

da_append(&active, id);
Expand Down