0 branches 0 tags
4
12
13 19
Thread security added
jonroach
on 9:25 AM Sep 2 2025
11
12
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
#include "generator.h"
#include "coroutine.h"
#include <stdlib.h>
#include <stdio.h>
#include <assert.h>
enum {
Generator_Running,
Generator_Deleting,
Generator_Complete
};
struct Generator {
Coroutine *coroutine;
Coroutine *caller;
void *(*start)(void *);
void *param;
char state;
};
static void on_yield(void *this){
Generator *gen = (Generator *)this;
Coroutine_Continue(gen->caller, Coroutine_GetValue(gen->coroutine), true);
}
static void *do_start(void *this){
Generator *gen = (Generator *)this;
assert(gen->state == Generator_Running || gen->state == Generator_Deleting);
void *value;
if (gen->state == Generator_Running){
value = gen->start(gen->param);
} else {
// we are being deleted
value = NULL;
}
gen->state = Generator_Complete;
Coroutine_Continue(gen->caller, value, true);
return value;
}
Generator *Generator_New(void *(*start)(void *), void *param){
Generator *gen = malloc(sizeof(Generator));
gen->start = start;
gen->param = param;
gen->coroutine = Coroutine_New(gen, on_yield, do_start);
gen->state = Generator_Running;
return gen;
}
void Generator_Delete(Generator *gen){
assert(gen->state != Generator_Deleting);
if (gen->state == Generator_Running){
gen->state = Generator_Deleting;
gen->caller = Coroutine_GetActive();
Coroutine_Continue(gen->coroutine, gen, true);
Coroutine_Yield(NULL);
}
assert(gen->state == Generator_Complete);
Coroutine_Delete(gen->coroutine);
free(gen);
}
bool Generator_Next(Generator *gen, void **value){
assert(gen->state != Generator_Deleting);
if (gen->state == Generator_Complete){
return false;
}
gen->caller = Coroutine_GetActive();
Coroutine_Continue(gen->coroutine, gen, true);
*value = Coroutine_Yield(NULL);
return Coroutine_IsRunning(gen->coroutine);
}
bool Generator_Yield(void *value)
{
Generator *gen = Coroutine_Yield(value);
return gen->state == Generator_Running;
}
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
#include "generator.h"
#include "coroutine.h"
#include <stdlib.h>
#include <stdio.h>
#include <assert.h>
#include "cor_thread_local.h"
enum {
Generator_Running,
Generator_Deleting,
Generator_Complete
};
struct Generator {
Coroutine *coroutine;
Coroutine *caller;
void *(*start)(void *);
void *param;
char state;
};
_Cor_thread_local Generator *g_active_generator = NULL;
static void on_yield_gen(void *this){
Generator *gen = (Generator *)this;
Coroutine_Continue(gen->caller, Coroutine_GetValue(gen->coroutine), true);
}
static void on_yield_gen_caller(void *this){
(void)this;
}
static void *do_start(void *this){
Generator *gen = (Generator *)this;
g_active_generator = this;
assert(gen->state == Generator_Running || gen->state == Generator_Deleting);
void *value;
if (gen->state == Generator_Running){
g_active_generator = gen;
value = gen->start(gen->param);
g_active_generator = NULL;
} else {
// we are being deleted
value = NULL;
}
gen->state = Generator_Complete;
Coroutine_Continue(gen->caller, value, true);
return value;
}
Generator *Generator_New(void *(*start)(void *), void *param){
Generator *gen = malloc(sizeof(Generator));
gen->start = start;
gen->param = param;
gen->coroutine = Coroutine_New(do_start);
gen->state = Generator_Running;
return gen;
}
void Generator_Delete(Generator *gen){
assert(gen->state != Generator_Deleting);
if (gen->state == Generator_Running){
gen->state = Generator_Deleting;
gen->caller = Coroutine_GetActive();
Coroutine_Continue(gen->coroutine, gen, true);
Coroutine_Yield(NULL, on_yield_gen_caller, gen);
}
assert(gen->state == Generator_Complete);
Coroutine_Delete(gen->coroutine);
free(gen);
}
bool Generator_Next(Generator *gen, void **value){
assert(gen->state != Generator_Deleting);
if (gen->state == Generator_Complete){
return false;
}
gen->caller = Coroutine_GetActive();
Coroutine_Continue(gen->coroutine, gen, true);
*value = Coroutine_Yield(NULL, on_yield_gen_caller, gen);
return Coroutine_IsRunning(gen->coroutine);
}
bool Generator_Yield(void *value)
{
assert(g_active_generator);
Generator *gen = g_active_generator;
g_active_generator = NULL;
Coroutine_Yield(value, on_yield_gen, gen);
g_active_generator = gen;
return gen->state == Generator_Running;
}