coroutine.h

Go to the documentation of this file.

#ifndef COROUTINE_H
#define COROUTINE_H

typedef void * co_t;
typedef unsigned int semaphore_t;

enum {
    CO_READY,
    CO_WAIT,
    CO_YIELD,
    CO_END,
    CO_SKIP,
};

#define CORO_CONTEXT( name ) \
    co_t co_##name

#define CORO_CONTEXT_INIT( name ) ({ \
    co_##name = NULL; \
})

#define CORO_LOCAL  static

#define CORO_DEFINE( name ) \
    int coro_##name( co_t *co_p )

#define CORO_BEGIN( initial ) ({ \
    initial; \
    if ( *co_p ) goto **co_p; \
})

#define CORO_END( final ) ({ \
    __label__ L; \
    *co_p = &&L; \
L: \
    final; \
    return CO_END; \
})

#define CORO_YIELD( final ) ({ \
    __label__ L; \
    *co_p = &&L; \
    final; \
    return CO_YIELD; \
L:; \
})

#define CORO_WAIT( cond, ... ) ({ \
    __label__ L; \
    *co_p = &&L; \
L: \
    if (!( cond )) { \
        __VA_ARGS__;  /* final */ \
        return CO_WAIT; \
    } \
})

#define CORO_RESTART( final ) ({ \
    *co_p = NULL; \
    final; \
    return CO_YIELD; \
})

#define CORO_QUIT( final )  CORO_END( final )

#define CORO_CALL( name ) \
    coro_##name( &co_##name )

#define CORO_ALIVE( coro ) (( coro ) < CO_END )

#define CORO_WAIT_CORO( coro, ... ) \
    CORO_WAIT( !CORO_ALIVE( coro ), ## __VA_ARGS__ )

#define SEMAPHORE_INIT( name, val ) ({ \
    name = val; \
})

#define SEMAPHORE_ACQUIRE( name, ... ) ({ \
    CORO_WAIT(( name > 0 ), ## __VA_ARGS__ ); \
    --name; \
})

#define SEMAPHORE_RELEASE( name ) ({ \
    ++name; \
})

#endif /* COROUTINE_H */