-
Notifications
You must be signed in to change notification settings - Fork 10
/
Coro.cpp
116 lines (97 loc) · 1.93 KB
/
Coro.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
#include "Coro.h"
#include <memory>
#include <setjmp.h>
#include <assert.h>
// stack size 2M
#define CORO_STACK_SIZE 1024 * 1024 * 2
#define CORO_FUN_INIT 0
#define CORO_FUN_END 1
#define CORO_FUN_YIELD 2
struct Coro;
static struct Coro* current_coro = NULL;
static struct Coro* coro_list = NULL;
static int coro_num = 0;
jmp_buf main_env;
struct Coro
{
struct Coro *prev, *next;
int co_id;
CoroFunType co_fun;
void* co_arg;
void* co_stack;
jmp_buf co_env;
};
static void _start_coro()
{
current_coro->co_fun(current_coro->co_arg);
longjmp(main_env, CORO_FUN_END);
}
void coro_new(CoroFunType f, void* arg)
{
struct Coro* co = (struct Coro*) malloc(sizeof(struct Coro));
assert(co);
if (coro_list)
{
co->prev = coro_list->prev;
co->next = coro_list;
coro_list->prev->next = co;
coro_list->prev = co;
}
else
{
co->prev = co;
co->next = co;
}
coro_list = co;
// init struct Coro
co->co_id = ++coro_num;
co->co_fun = f;
co->co_arg = arg;
co->co_stack = malloc(CORO_STACK_SIZE);
assert(co->co_stack);
// init jmp buf
setjmp(co->co_env);
_JUMP_BUFFER* jmp_buf = (_JUMP_BUFFER*)&(co->co_env);
jmp_buf->Eip = (unsigned long)(_start_coro);
jmp_buf->Esp = (unsigned long)((char*)(co->co_stack) + CORO_STACK_SIZE - 16);
}
void coro_main()
{
if (!coro_list)
return;
struct Coro *co;
switch (setjmp(main_env))
{
case CORO_FUN_INIT:
// scheduling
current_coro = coro_list;
break;
case CORO_FUN_END:
co = current_coro;
if (co->next == co)
{
coro_list = NULL;
free(co->co_stack);
free(co);
return;
}
// scheduling
current_coro = current_coro->next;
co->prev->next = co->next;
co->next->prev = co->prev;
free(co->co_stack);
free(co);
break;
case CORO_FUN_YIELD:
// scheduling
current_coro = current_coro->next;
}
assert(current_coro);
longjmp(current_coro->co_env, 1);
}
void coro_yield()
{
if (setjmp(current_coro->co_env))
return;
longjmp(main_env, CORO_FUN_YIELD);
}