aboutsummaryrefslogtreecommitdiff
path: root/src/coro.c
blob: 82ef4f4de0547a7b0ea8ae27085bd1910a078ef7 (plain)
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
#include <u.h>

// -----------------------------------------------------------------------
//  Assembly routines.

extern void    _newcoro(coro *co, uintptr (*func)(coro*, uintptr), void *stk);
extern uintptr _coroyield(coro *co, uintptr arg);

struct coro 
{
    void*   sp;
    void*   bp;
    uintptr size;
    void*   user;
};

coro*
coro·new(uintptr stk, uintptr (*func)(coro*, uintptr))
{
    if (!func) 
        return nil;

    if (stk == 0) stk = 8192;

    byte *block = malloc(stk);
    coro *co    = (coro*)&block[stk - sizeof(coro)];
    co->bp      = block;
    co->size    = stk;

    _newcoro(co, func, co);
    return co;
}

error
coro·free(coro *co)
{
    enum 
    {
        NIL,
        GOOD,
        EMPTY,
        LOST,
    };

    if (!co)           return NIL;
    if (!co->bp)       return LOST;
    if (co->size == 0) return EMPTY;

    free(co->bp);

    return GOOD;
}

uintptr
coro·yield(coro *c, uintptr arg)
{
    return _coroyield(c, arg);
}