mirror of
https://github.com/patchzyy/wiicompiled
synced 2026-09-10 17:16:47 -04:00
theofficialgman
91 lines
2.5 KiB
C
91 lines
2.5 KiB
C
/*
|
|
WARNING: the overhead of POSIX ucontext is very high,
|
|
assembly versions of libco or libco_sjlj should be much faster
|
|
|
|
this library only exists for two reasons:
|
|
1: as an initial test for the viability of a ucontext implementation
|
|
2: to demonstrate the power and speed of libco over existing implementations,
|
|
such as pth (which defaults to wrapping ucontext on unix targets)
|
|
|
|
use this library only as a *last resort*
|
|
*/
|
|
|
|
#define _POSIX_C_SOURCE 200112L
|
|
#define LIBCO_C
|
|
#include "libco.h"
|
|
#include "settings.h"
|
|
#include "valgrind.h"
|
|
|
|
#include <stdlib.h>
|
|
#include <ucontext.h>
|
|
|
|
#ifdef __cplusplus
|
|
extern "C" {
|
|
#endif
|
|
|
|
static thread_local ucontext_t co_primary;
|
|
static thread_local ucontext_t* co_running = 0;
|
|
|
|
cothread_t co_active(void) {
|
|
if(!co_running) co_running = &co_primary;
|
|
return (cothread_t)co_running;
|
|
}
|
|
|
|
cothread_t co_derive(void* memory, unsigned int heapsize, void (*coentry)(void)) {
|
|
ucontext_t* thread;
|
|
if(!co_running) co_running = &co_primary;
|
|
thread = (ucontext_t*)memory;
|
|
memory = (unsigned char*)memory + sizeof(ucontext_t);
|
|
heapsize -= sizeof(ucontext_t);
|
|
if(thread) {
|
|
if((!getcontext(thread) && !(thread->uc_stack.ss_sp = 0)) && (thread->uc_stack.ss_sp = memory)) {
|
|
thread->uc_link = co_running;
|
|
thread->uc_stack.ss_size = heapsize;
|
|
makecontext(thread, coentry, 0);
|
|
VALGRIND_STACK_REGISTER(thread->uc_stack.ss_sp, thread->uc_stack.ss_sp + heapsize);
|
|
} else {
|
|
thread = 0;
|
|
}
|
|
}
|
|
return (cothread_t)thread;
|
|
}
|
|
|
|
cothread_t co_create(unsigned int heapsize, void (*coentry)(void)) {
|
|
ucontext_t* thread;
|
|
if(!co_running) co_running = &co_primary;
|
|
thread = (ucontext_t*)malloc(sizeof(ucontext_t));
|
|
if(thread) {
|
|
if((!getcontext(thread) && !(thread->uc_stack.ss_sp = 0)) && (thread->uc_stack.ss_sp = malloc(heapsize))) {
|
|
thread->uc_link = co_running;
|
|
thread->uc_stack.ss_size = heapsize;
|
|
makecontext(thread, coentry, 0);
|
|
VALGRIND_STACK_REGISTER(thread->uc_stack.ss_sp, thread->uc_stack.ss_sp + heapsize);
|
|
} else {
|
|
co_delete((cothread_t)thread);
|
|
thread = 0;
|
|
}
|
|
}
|
|
return (cothread_t)thread;
|
|
}
|
|
|
|
void co_delete(cothread_t cothread) {
|
|
if(cothread) {
|
|
if(((ucontext_t*)cothread)->uc_stack.ss_sp) { free(((ucontext_t*)cothread)->uc_stack.ss_sp); }
|
|
free(cothread);
|
|
}
|
|
}
|
|
|
|
void co_switch(cothread_t cothread) {
|
|
ucontext_t* old_thread = co_running;
|
|
co_running = (ucontext_t*)cothread;
|
|
swapcontext(old_thread, co_running);
|
|
}
|
|
|
|
int co_serializable(void) {
|
|
return 0;
|
|
}
|
|
|
|
#ifdef __cplusplus
|
|
}
|
|
#endif
|