ViewVC Help
View File | Revision Log | Show Annotations | Download File
/cvs/libcoro/coro.h
(Generate patch)

Comparing libcoro/coro.h (file contents):
Revision 1.38 by root, Fri Nov 14 05:27:08 2008 UTC vs.
Revision 1.61 by root, Tue Aug 14 15:46:04 2018 UTC

1/* 1/*
2 * Copyright (c) 2001-2008 Marc Alexander Lehmann <schmorp@schmorp.de> 2 * Copyright (c) 2001-2012,2015 Marc Alexander Lehmann <schmorp@schmorp.de>
3 * 3 *
4 * Redistribution and use in source and binary forms, with or without modifica- 4 * Redistribution and use in source and binary forms, with or without modifica-
5 * tion, are permitted provided that the following conditions are met: 5 * tion, are permitted provided that the following conditions are met:
6 * 6 *
7 * 1. Redistributions of source code must retain the above copyright notice, 7 * 1. Redistributions of source code must retain the above copyright notice,
8 * this list of conditions and the following disclaimer. 8 * this list of conditions and the following disclaimer.
9 * 9 *
10 * 2. Redistributions in binary form must reproduce the above copyright 10 * 2. Redistributions in binary form must reproduce the above copyright
11 * notice, this list of conditions and the following disclaimer in the 11 * notice, this list of conditions and the following disclaimer in the
12 * documentation and/or other materials provided with the distribution. 12 * documentation and/or other materials provided with the distribution.
13 * 13 *
14 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR IMPLIED 14 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR IMPLIED
15 * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MER- 15 * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MER-
16 * CHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO 16 * CHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO
17 * EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPE- 17 * EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPE-
18 * CIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, 18 * CIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
… …
64 * coro_create now allows one to create source coro_contexts. 64 * coro_create now allows one to create source coro_contexts.
65 * do not rely on makecontext passing a void * correctly. 65 * do not rely on makecontext passing a void * correctly.
66 * try harder to get _setjmp/_longjmp. 66 * try harder to get _setjmp/_longjmp.
67 * major code cleanup/restructuring. 67 * major code cleanup/restructuring.
68 * 2008-11-10 the .cfi hacks are no longer needed. 68 * 2008-11-10 the .cfi hacks are no longer needed.
69 * 2008-11-16 work around a freebsd pthread bug.
70 * 2008-11-19 define coro_*jmp symbols for easier porting.
71 * 2009-06-23 tentative win32-backend support for mingw32 (Yasuhiro Matsumoto).
72 * 2010-12-03 tentative support for uclibc (which lacks all sorts of things).
73 * 2011-05-30 set initial callee-saved-registers to zero with CORO_ASM.
74 * use .cfi_undefined rip on linux-amd64 for better backtraces.
75 * 2011-06-08 maybe properly implement weird windows amd64 calling conventions.
76 * 2011-07-03 rely on __GCC_HAVE_DWARF2_CFI_ASM for cfi detection.
77 * 2011-08-08 cygwin trashes stacks, use pthreads with double stack on cygwin.
78 * 2012-12-04 reduce misprediction penalty for x86/amd64 assembly switcher.
79 * 2012-12-05 experimental fiber backend (allocates stack twice).
80 * 2012-12-07 API version 3 - add coro_stack_alloc/coro_stack_free.
81 * 2012-12-21 valgrind stack registering was broken.
82 * 2015-12-05 experimental asm be for arm7, based on a patch by Nick Zavaritsky.
83 * use __name__ for predefined symbols, as in libecb.
84 * enable guard pages on arm, aarch64 and mips.
85 * 2016-08-27 try to disable _FORTIFY_SOURCE with CORO_SJLJ, as it
86 * breaks setjmp/longjmp. Also disable CORO_ASM for asm by default,
87 * as it was reported to crash.
88 * 2016-11-18 disable cfi_undefined again - backtraces might be worse, but
89 * compile compatibility is improved.
90 * 2018-08-14 use a completely different pthread strategy that should allow
91 * sharing of coroutines among different threads. this would
92 * undefined behaviour before as mutexes would be unlocked on
93 * a different thread. overall, this might be slower than
94 * using a pipe for synchronisation, but pipes eat fd's...
69 */ 95 */
70 96
71#ifndef CORO_H 97#ifndef CORO_H
72#define CORO_H 98#define CORO_H
73 99
74#define CORO_VERSION 2 100#if __cplusplus
75 101extern "C" {
76/* 102#endif
77 * Changes since API version 1:
78 * replaced bogus -DCORO_LOOSE with gramatically more correct -DCORO_LOSER
79 */
80 103
81/* 104/*
82 * This library consists of only three files 105 * This library consists of only three files
83 * coro.h, coro.c and LICENSE (and optionally README) 106 * coro.h, coro.c and LICENSE (and optionally README)
84 * 107 *
85 * It implements what is known as coroutines, in a hopefully 108 * It implements what is known as coroutines, in a hopefully
86 * portable way. At the moment you have to define which kind 109 * portable way.
87 * of implementation flavour you want: 110 *
111 * All compiletime symbols must be defined both when including coro.h
112 * (using libcoro) as well as when compiling coro.c (the implementation).
113 *
114 * You can manually specify which flavour you want. If you don't define
115 * any of these, libcoro tries to choose a safe and fast default:
88 * 116 *
89 * -DCORO_UCONTEXT 117 * -DCORO_UCONTEXT
90 * 118 *
91 * This flavour uses SUSv2's get/set/swap/makecontext functions that 119 * This flavour uses SUSv2's get/set/swap/makecontext functions that
92 * unfortunately only newer unices support. 120 * unfortunately only some unices support, and is quite slow.
93 * 121 *
94 * -DCORO_SJLJ 122 * -DCORO_SJLJ
95 * 123 *
96 * This flavour uses SUSv2's setjmp/longjmp and sigaltstack functions to 124 * This flavour uses SUSv2's setjmp/longjmp and sigaltstack functions to
97 * do it's job. Coroutine creation is much slower than UCONTEXT, but 125 * do it's job. Coroutine creation is much slower than UCONTEXT, but
98 * context switching is often a bit cheaper. It should work on almost 126 * context switching is a bit cheaper. It should work on almost all unices.
99 * all unices.
100 * 127 *
101 * -DCORO_LINUX 128 * -DCORO_LINUX
102 * 129 *
130 * CORO_SJLJ variant.
103 * Old GNU/Linux systems (<= glibc-2.1) only work with this implementation 131 * Old GNU/Linux systems (<= glibc-2.1) only work with this implementation
104 * (it is very fast and therefore recommended over other methods, but 132 * (it is very fast and therefore recommended over other methods, but
105 * doesn't work with anything newer). 133 * doesn't work with anything newer).
106 * 134 *
107 * -DCORO_LOSER 135 * -DCORO_LOSER
108 * 136 *
137 * CORO_SJLJ variant.
109 * Microsoft's highly proprietary platform doesn't support sigaltstack, and 138 * Microsoft's highly proprietary platform doesn't support sigaltstack, and
110 * this automatically selects a suitable workaround for this platform. 139 * this selects a suitable workaround for this platform. It might not work
111 * (untested) 140 * with your compiler though - it has only been tested with MSVC 6.
141 *
142 * -DCORO_FIBER
143 *
144 * Slower, but probably more portable variant for the Microsoft operating
145 * system, using fibers. Ignores the passed stack and allocates it internally.
146 * Also, due to bugs in cygwin, this does not work with cygwin.
112 * 147 *
113 * -DCORO_IRIX 148 * -DCORO_IRIX
114 * 149 *
150 * CORO_SJLJ variant.
115 * SGI's version of Microsoft's NT ;) 151 * For SGI's version of Microsoft's NT ;)
116 * 152 *
117 * -DCORO_ASM 153 * -DCORO_ASM
118 * 154 *
119 * Handcoded assembly, known to work only on a few architectures/ABI: 155 * Hand coded assembly, known to work only on a few architectures/ABI:
120 * ELF Linux x86 && amd64 when gcc is used and optimisation is turned on. 156 * GCC + arm7/x86/IA32/amd64/x86_64 + GNU/Linux and a few BSDs. Fastest
157 * choice, if it works.
121 * 158 *
122 * -DCORO_PTHREAD 159 * -DCORO_PTHREAD
123 * 160 *
124 * Use the pthread API. You have to provide <pthread.h> and -lpthread. 161 * Use the pthread API. You have to provide <pthread.h> and -lpthread.
162 * This is likely the slowest backend, and it also does not support fork(),
163 * so avoid it at all costs.
125 * 164 *
126 * If you define neither of these symbols, coro.h will try to autodetect 165 * If you define neither of these symbols, coro.h will try to autodetect
127 * the model. This currently works for CORO_LOSER only. For the other 166 * the best/safest model. To help with the autodetection, you should check
128 * alternatives you should check (e.g. using autoconf) and define the 167 * (e.g. using autoconf) and define the following symbols: HAVE_UCONTEXT_H
129 * following symbols: HAVE_UCONTEXT_H / HAVE_SETJMP_H / HAVE_SIGALTSTACK. 168 * / HAVE_SETJMP_H / HAVE_SIGALTSTACK.
169 */
170
171/*
172 * Changes when the API changes incompatibly.
173 * This is ONLY the API version - there is no ABI compatibility between releases.
130 */ 174 *
175 * Changes in API version 2:
176 * replaced bogus -DCORO_LOOSE with grammatically more correct -DCORO_LOSER
177 * Changes in API version 3:
178 * introduced stack management (CORO_STACKALLOC)
179 */
180#define CORO_VERSION 3
181
182#include <stddef.h>
131 183
132/* 184/*
133 * This is the type for the initialization function of a new coroutine. 185 * This is the type for the initialization function of a new coroutine.
134 */ 186 */
135typedef void (*coro_func)(void *); 187typedef void (*coro_func)(void *);
… …
146 * uninitialised coro_context, it expects a pointer to the entry function 198 * uninitialised coro_context, it expects a pointer to the entry function
147 * and the single pointer value that is given to it as argument. 199 * and the single pointer value that is given to it as argument.
148 * 200 *
149 * Allocating/deallocating the stack is your own responsibility. 201 * Allocating/deallocating the stack is your own responsibility.
150 * 202 *
151 * As a special case, if coro, arg, sptr and ssize are all zero, 203 * As a special case, if coro, arg, sptr and ssze are all zero,
152 * then an "empty" coro_context will be created that is suitable 204 * then an "empty" coro_context will be created that is suitable
153 * as an initial source for coro_transfer. 205 * as an initial source for coro_transfer.
154 * 206 *
155 * This function is not reentrant, but putting a mutex around it 207 * This function is not reentrant, but putting a mutex around it
156 * will work. 208 * will work.
157 */ 209 */
158void coro_create (coro_context *ctx, /* an uninitialised coro_context */ 210void coro_create (coro_context *ctx, /* an uninitialised coro_context */
159 coro_func coro, /* the coroutine code to be executed */ 211 coro_func coro, /* the coroutine code to be executed */
160 void *arg, /* a single pointer passed to the coro */ 212 void *arg, /* a single pointer passed to the coro */
161 void *sptr, /* start of stack area */ 213 void *sptr, /* start of stack area */
162 long ssize); /* size of stack area */ 214 size_t ssze); /* size of stack area in bytes */
163 215
164/* 216/*
165 * The following prototype defines the coroutine switching function. It is 217 * The following prototype defines the coroutine switching function. It is
166 * usually implemented as a macro, so watch out. 218 * sometimes implemented as a macro, so watch out.
167 * 219 *
168 * This function is thread-safe and reentrant. 220 * This function is thread-safe and reentrant.
169 */ 221 */
170#if 0 222#if 0
171void coro_transfer (coro_context *prev, coro_context *next); 223void coro_transfer (coro_context *prev, coro_context *next);
172#endif 224#endif
173 225
174/* 226/*
175 * The following prototype defines the coroutine destroy function. It is 227 * The following prototype defines the coroutine destroy function. It
176 * usually implemented as a macro, so watch out. It also serves 228 * is sometimes implemented as a macro, so watch out. It also serves no
177 * no purpose unless you want to use the CORO_PTHREAD backend, 229 * purpose unless you want to use the CORO_PTHREAD backend, where it is
178 * where it is used to clean up the thread. You are responsible 230 * used to clean up the thread. You are responsible for freeing the stack
179 * for freeing the stack and the context itself. 231 * and the context itself.
180 * 232 *
181 * This function is thread-safe and reentrant. 233 * This function is thread-safe and reentrant.
182 */ 234 */
183#if 0 235#if 0
184void coro_destroy (coro_context *ctx); 236void coro_destroy (coro_context *ctx);
185#endif 237#endif
186 238
187/*
188 * That was it. No other user-visible functions are implemented here.
189 */
190
191/*****************************************************************************/ 239/*****************************************************************************/
240/* optional stack management */
241/*****************************************************************************/
242/*
243 * You can disable all of the stack management functions by
244 * defining CORO_STACKALLOC to 0. Otherwise, they are enabled by default.
245 *
246 * If stack management is enabled, you can influence the implementation via these
247 * symbols:
248 *
249 * -DCORO_USE_VALGRIND
250 *
251 * If defined, then libcoro will include valgrind/valgrind.h and register
252 * and unregister stacks with valgrind.
253 *
254 * -DCORO_GUARDPAGES=n
255 *
256 * libcoro will try to use the specified number of guard pages to protect against
257 * stack overflow. If n is 0, then the feature will be disabled. If it isn't
258 * defined, then libcoro will choose a suitable default. If guardpages are not
259 * supported on the platform, then the feature will be silently disabled.
260 */
261#ifndef CORO_STACKALLOC
262# define CORO_STACKALLOC 1
263#endif
192 264
265#if CORO_STACKALLOC
266
267/*
268 * The only allowed operations on these struct members is to read the
269 * "sptr" and "ssze" members to pass it to coro_create, to read the "sptr"
270 * member to see if it is false, in which case the stack isn't allocated,
271 * and to set the "sptr" member to 0, to indicate to coro_stack_free to
272 * not actually do anything.
273 */
274
275struct coro_stack
276{
277 void *sptr;
278 size_t ssze;
279#if CORO_USE_VALGRIND
280 int valgrind_id;
281#endif
282};
283
284/*
285 * Try to allocate a stack of at least the given size and return true if
286 * successful, or false otherwise.
287 *
288 * The size is *NOT* specified in bytes, but in units of sizeof (void *),
289 * i.e. the stack is typically 4(8) times larger on 32 bit(64 bit) platforms
290 * then the size passed in.
291 *
292 * If size is 0, then a "suitable" stack size is chosen (usually 1-2MB).
293 */
294int coro_stack_alloc (struct coro_stack *stack, unsigned int size);
295
296/*
297 * Free the stack allocated by coro_stack_alloc again. It is safe to
298 * call this function on the coro_stack structure even if coro_stack_alloc
299 * failed.
300 */
301void coro_stack_free (struct coro_stack *stack);
302
303#endif
304
305/*
306 * That was it. No other user-serviceable parts below here.
307 */
308
309/*****************************************************************************/
310
193#if !defined(CORO_LOSER) && !defined(CORO_UCONTEXT) \ 311#if !defined CORO_LOSER && !defined CORO_UCONTEXT \
194 && !defined(CORO_SJLJ) && !defined(CORO_LINUX) \ 312 && !defined CORO_SJLJ && !defined CORO_LINUX \
195 && !defined(CORO_IRIX) && !defined(CORO_ASM) \ 313 && !defined CORO_IRIX && !defined CORO_ASM \
196 && !defined(CORO_PTHREAD) 314 && !defined CORO_PTHREAD && !defined CORO_FIBER
197# if defined(WINDOWS) 315# if defined WINDOWS && (defined __i386__ || (__x86_64__ || defined _M_IX86 || defined _M_AMD64)
316# define CORO_ASM 1
317# elif defined WINDOWS || defined _WIN32
198# define CORO_LOSER 1 /* you don't win with windoze */ 318# define CORO_LOSER 1 /* you don't win with windoze */
199# elif defined(__linux) && (defined(__x86) || defined (__amd64)) 319# elif __linux && (__i386__ || (__x86_64__ && !__ILP32__) /*|| (__arm__ && __ARM_ARCH == 7)), not working */
200# define CORO_ASM 1 320# define CORO_ASM 1
201# elif defined(HAVE_UCONTEXT_H) 321# elif defined HAVE_UCONTEXT_H
202# define CORO_UCONTEXT 1 322# define CORO_UCONTEXT 1
203# elif defined(HAVE_SETJMP_H) && defined(HAVE_SIGALTSTACK) 323# elif defined HAVE_SETJMP_H && defined HAVE_SIGALTSTACK
204# define CORO_SJLJ 1 324# define CORO_SJLJ 1
205# else 325# else
206error unknown or unsupported architecture 326error unknown or unsupported architecture
207# endif 327# endif
208#endif 328#endif
… …
211 331
212#if CORO_UCONTEXT 332#if CORO_UCONTEXT
213 333
214# include <ucontext.h> 334# include <ucontext.h>
215 335
216struct coro_context { 336struct coro_context
337{
217 ucontext_t uc; 338 ucontext_t uc;
218}; 339};
219 340
220# define coro_transfer(p,n) swapcontext (&((p)->uc), &((n)->uc)) 341# define coro_transfer(p,n) swapcontext (&((p)->uc), &((n)->uc))
221# define coro_destroy(ctx) (void *)(ctx) 342# define coro_destroy(ctx) (void *)(ctx)
222 343
223#elif CORO_SJLJ || CORO_LOSER || CORO_LINUX || CORO_IRIX 344#elif CORO_SJLJ || CORO_LOSER || CORO_LINUX || CORO_IRIX
224 345
225# if defined(CORO_LINUX) && !defined(_GNU_SOURCE) 346# if defined(CORO_LINUX) && !defined(_GNU_SOURCE)
226# define _GNU_SOURCE /* for linux libc */ 347# define _GNU_SOURCE /* for glibc */
348# endif
349
350/* try to disable well-meant but buggy checks in some libcs */
351# ifdef _FORTIFY_SOURCE
352# undef _FORTIFY_SOURCE
353# undef __USE_FORTIFY_LEVEL /* helps some more when too much has been included already */
227# endif 354# endif
228 355
229# if !CORO_LOSER 356# if !CORO_LOSER
230# include <unistd.h> 357# include <unistd.h>
231# endif 358# endif
… …
236# define _XOPEN_UNIX 1 363# define _XOPEN_UNIX 1
237# endif 364# endif
238 365
239# include <setjmp.h> 366# include <setjmp.h>
240 367
368# if _XOPEN_UNIX > 0 || defined (_setjmp)
369# define coro_jmp_buf jmp_buf
370# define coro_setjmp(env) _setjmp (env)
371# define coro_longjmp(env) _longjmp ((env), 1)
372# elif CORO_LOSER
373# define coro_jmp_buf jmp_buf
374# define coro_setjmp(env) setjmp (env)
375# define coro_longjmp(env) longjmp ((env), 1)
376# else
377# define coro_jmp_buf sigjmp_buf
378# define coro_setjmp(env) sigsetjmp (env, 0)
379# define coro_longjmp(env) siglongjmp ((env), 1)
380# endif
381
241struct coro_context { 382struct coro_context
242#if _XOPEN_UNIX > 0 || CORO_LOSER 383{
243 jmp_buf env; 384 coro_jmp_buf env;
385};
386
387# define coro_transfer(p,n) do { if (!coro_setjmp ((p)->env)) coro_longjmp ((n)->env); } while (0)
388# define coro_destroy(ctx) (void *)(ctx)
389
390#elif CORO_ASM
391
392struct coro_context
393{
394 void **sp; /* must be at offset 0 */
395};
396
397#if __i386__ || __x86_64__
398void __attribute__ ((__noinline__, __regparm__(2)))
244#else 399#else
245 sigjmp_buf env; 400void __attribute__ ((__noinline__))
246#endif
247};
248
249# if _XOPEN_UNIX > 0
250# define coro_transfer(p,n) do { if (! _setjmp ((p)->env )) _longjmp ((n)->env, 1); } while (0)
251# elif CORO_LOSER
252# define coro_transfer(p,n) do { if (! setjmp ((p)->env )) longjmp ((n)->env, 1); } while (0)
253# else
254# define coro_transfer(p,n) do { if (!sigsetjmp ((p)->env, 0)) siglongjmp ((n)->env, 1); } while (0)
255# endif 401#endif
402coro_transfer (coro_context *prev, coro_context *next);
256 403
257# define coro_destroy(ctx) (void *)(ctx) 404# define coro_destroy(ctx) (void *)(ctx)
258 405
259#elif CORO_ASM
260
261struct coro_context {
262 void **sp; /* must be at offset 0 */
263};
264
265void __attribute__ ((__noinline__, __regparm__(2)))
266coro_transfer (coro_context *prev, coro_context *next);
267
268# define coro_destroy(ctx) (void *)(ctx)
269
270#elif CORO_PTHREAD 406#elif CORO_PTHREAD
271 407
272# include <pthread.h> 408# include <pthread.h>
273 409
274extern pthread_mutex_t coro_mutex; 410extern pthread_mutex_t coro_mutex;
275 411
276struct coro_context { 412struct coro_context
413{
414 int flags;
277 pthread_cond_t cv; 415 pthread_cond_t cv;
278 pthread_t id;
279}; 416};
280 417
281void coro_transfer (coro_context *prev, coro_context *next); 418void coro_transfer (coro_context *prev, coro_context *next);
282void coro_destroy (coro_context *ctx); 419void coro_destroy (coro_context *ctx);
283 420
284#endif 421#elif CORO_FIBER
285 422
286#endif 423struct coro_context
424{
425 void *fiber;
426 /* only used for initialisation */
427 coro_func coro;
428 void *arg;
429};
287 430
431void coro_transfer (coro_context *prev, coro_context *next);
432void coro_destroy (coro_context *ctx);
433
434#endif
435
436#if __cplusplus
437}
438#endif
439
440#endif
441

Diff Legend

– Removed lines
+ Added lines
< Changed lines
> Changed lines