emscripten/tests/test_fibers.cpp

105 строки
2.5 KiB
C++

// Copyright 2019 The Emscripten Authors. All rights reserved.
// Emscripten is available under two separate licenses, the MIT license and the
// University of Illinois/NCSA Open Source License. Both these licenses can be
// found in the LICENSE file.
// Based on test_coroutines.cpp
#include <stdio.h>
#include <stdlib.h>
#include <emscripten/fiber.h>
struct Fiber {
emscripten_fiber_t context;
char asyncify_stack[1024];
alignas(16) char c_stack[4096];
int result = 0;
void init_with_api(em_arg_callback_func entry, void *arg) {
emscripten_fiber_init(&context, entry, arg, c_stack, sizeof(c_stack), asyncify_stack, sizeof(asyncify_stack));
}
void init_manually(em_arg_callback_func entry, void *arg) {
context.stack_base = c_stack + sizeof(c_stack);
context.stack_limit = c_stack;
context.stack_ptr = context.stack_base;
context.entry = entry;
context.user_data = arg;
context.asyncify_data.stack_ptr = asyncify_stack;
context.asyncify_data.stack_limit = asyncify_stack + sizeof(asyncify_stack);
}
void swap(emscripten_fiber_t *fiber) {
emscripten_fiber_swap(&context, fiber);
}
};
static struct Globals {
emscripten_fiber_t main;
char asyncify_stack[1024];
Fiber fibers[2];
Globals() {
emscripten_fiber_init_from_current_context(&main, asyncify_stack, sizeof(asyncify_stack));
}
} G;
static void leaf(void) {
printf("leaf-");
}
static void fib(void * arg) {
int *p = (int*)arg;
int cur = 1;
int next = 1;
for(int i = 0; i < 9; ++i) {
*p = cur;
G.fibers[0].swap(&G.main);
int next2 = cur + next;
cur = next;
next = next2;
}
}
[[noreturn]]
static void f(void *arg) {
int *p = (int*)arg;
*p = 0;
leaf();
G.fibers[0].swap(&G.main);
fib(arg);
G.fibers[0].result = 1;
G.fibers[0].swap(&G.main);
abort();
}
[[noreturn]]
static void g(void *arg) {
int *p = (int*)arg;
for(int i = 0; i < 10; ++i) {
*p = 100+i;
G.fibers[1].swap(&G.main);
}
G.fibers[1].result = 1;
G.fibers[1].swap(&G.main);
abort();
}
int main(int argc, char **argv) {
int i;
G.fibers[0].init_with_api(f, &i);
G.fibers[1].init_manually(g, &i);
printf("*");
while(emscripten_fiber_swap(&G.main, &G.fibers[0].context), !G.fibers[0].result) {
printf("%d-", i);
emscripten_fiber_swap(&G.main, &G.fibers[1].context);
printf("%d-", i);
}
printf("*\n");
return 0;
}