aboutsummaryrefslogtreecommitdiff
path: root/src/coro.c
blob: 4e651d388fc276b6d7a73981314e69a9e168d707 (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
#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
    {
        GOOD,
        NIL,
        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);
}