ViewVC Help
View File | Revision Log | Show Annotations | Download File
/cvs/libcoro/coro.h
Revision: 1.60
Committed: Fri Nov 18 06:22:05 2016 UTC (7 years, 6 months ago) by root
Content type: text/plain
Branch: MAIN
CVS Tags: rel-6_512, rel-6_513, rel-6_514
Changes since 1.59: +2 -1 lines
Log Message:
*** empty log message ***

File Contents

# Content
1 /*
2 * Copyright (c) 2001-2012,2015 Marc Alexander Lehmann <schmorp@schmorp.de>
3 *
4 * Redistribution and use in source and binary forms, with or without modifica-
5 * tion, are permitted provided that the following conditions are met:
6 *
7 * 1. Redistributions of source code must retain the above copyright notice,
8 * this list of conditions and the following disclaimer.
9 *
10 * 2. Redistributions in binary form must reproduce the above copyright
11 * notice, this list of conditions and the following disclaimer in the
12 * documentation and/or other materials provided with the distribution.
13 *
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-
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-
18 * CIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
19 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
20 * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
21 * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTH-
22 * ERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
23 * OF THE POSSIBILITY OF SUCH DAMAGE.
24 *
25 * Alternatively, the contents of this file may be used under the terms of
26 * the GNU General Public License ("GPL") version 2 or any later version,
27 * in which case the provisions of the GPL are applicable instead of
28 * the above. If you wish to allow the use of your version of this file
29 * only under the terms of the GPL and not to allow others to use your
30 * version of this file under the BSD license, indicate your decision
31 * by deleting the provisions above and replace them with the notice
32 * and other provisions required by the GPL. If you do not delete the
33 * provisions above, a recipient may use your version of this file under
34 * either the BSD or the GPL.
35 *
36 * This library is modelled strictly after Ralf S. Engelschalls article at
37 * http://www.gnu.org/software/pth/rse-pmt.ps. So most of the credit must
38 * go to Ralf S. Engelschall <rse@engelschall.com>.
39 *
40 * This coroutine library is very much stripped down. You should either
41 * build your own process abstraction using it or - better - just use GNU
42 * Portable Threads, http://www.gnu.org/software/pth/.
43 *
44 */
45
46 /*
47 * 2006-10-26 Include stddef.h on OS X to work around one of its bugs.
48 * Reported by Michael_G_Schwern.
49 * 2006-11-26 Use _setjmp instead of setjmp on GNU/Linux.
50 * 2007-04-27 Set unwind frame info if gcc 3+ and ELF is detected.
51 * Use _setjmp instead of setjmp on _XOPEN_SOURCE >= 600.
52 * 2007-05-02 Add assembly versions for x86 and amd64 (to avoid reliance
53 * on SIGUSR2 and sigaltstack in Crossfire).
54 * 2008-01-21 Disable CFI usage on anything but GNU/Linux.
55 * 2008-03-02 Switched to 2-clause BSD license with GPL exception.
56 * 2008-04-04 New (but highly unrecommended) pthreads backend.
57 * 2008-04-24 Reinstate CORO_LOSER (had wrong stack adjustments).
58 * 2008-10-30 Support assembly method on x86 with and without frame pointer.
59 * 2008-11-03 Use a global asm statement for CORO_ASM, idea by pippijn.
60 * 2008-11-05 Hopefully fix misaligned stacks with CORO_ASM/SETJMP.
61 * 2008-11-07 rbp wasn't saved in CORO_ASM on x86_64.
62 * introduce coro_destroy, which is a nop except for pthreads.
63 * speed up CORO_PTHREAD. Do no longer leak threads either.
64 * coro_create now allows one to create source coro_contexts.
65 * do not rely on makecontext passing a void * correctly.
66 * try harder to get _setjmp/_longjmp.
67 * major code cleanup/restructuring.
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 */
91
92 #ifndef CORO_H
93 #define CORO_H
94
95 #if __cplusplus
96 extern "C" {
97 #endif
98
99 /*
100 * This library consists of only three files
101 * coro.h, coro.c and LICENSE (and optionally README)
102 *
103 * It implements what is known as coroutines, in a hopefully
104 * portable way.
105 *
106 * All compiletime symbols must be defined both when including coro.h
107 * (using libcoro) as well as when compiling coro.c (the implementation).
108 *
109 * You can manually specify which flavour you want. If you don't define
110 * any of these, libcoro tries to choose a safe and fast default:
111 *
112 * -DCORO_UCONTEXT
113 *
114 * This flavour uses SUSv2's get/set/swap/makecontext functions that
115 * unfortunately only some unices support, and is quite slow.
116 *
117 * -DCORO_SJLJ
118 *
119 * This flavour uses SUSv2's setjmp/longjmp and sigaltstack functions to
120 * do it's job. Coroutine creation is much slower than UCONTEXT, but
121 * context switching is a bit cheaper. It should work on almost all unices.
122 *
123 * -DCORO_LINUX
124 *
125 * CORO_SJLJ variant.
126 * Old GNU/Linux systems (<= glibc-2.1) only work with this implementation
127 * (it is very fast and therefore recommended over other methods, but
128 * doesn't work with anything newer).
129 *
130 * -DCORO_LOSER
131 *
132 * CORO_SJLJ variant.
133 * Microsoft's highly proprietary platform doesn't support sigaltstack, and
134 * this selects a suitable workaround for this platform. It might not work
135 * with your compiler though - it has only been tested with MSVC 6.
136 *
137 * -DCORO_FIBER
138 *
139 * Slower, but probably more portable variant for the Microsoft operating
140 * system, using fibers. Ignores the passed stack and allocates it internally.
141 * Also, due to bugs in cygwin, this does not work with cygwin.
142 *
143 * -DCORO_IRIX
144 *
145 * CORO_SJLJ variant.
146 * For SGI's version of Microsoft's NT ;)
147 *
148 * -DCORO_ASM
149 *
150 * Hand coded assembly, known to work only on a few architectures/ABI:
151 * GCC + arm7/x86/IA32/amd64/x86_64 + GNU/Linux and a few BSDs. Fastest
152 * choice, if it works.
153 *
154 * -DCORO_PTHREAD
155 *
156 * Use the pthread API. You have to provide <pthread.h> and -lpthread.
157 * This is likely the slowest backend, and it also does not support fork(),
158 * so avoid it at all costs.
159 *
160 * If you define neither of these symbols, coro.h will try to autodetect
161 * the best/safest model. To help with the autodetection, you should check
162 * (e.g. using autoconf) and define the following symbols: HAVE_UCONTEXT_H
163 * / HAVE_SETJMP_H / HAVE_SIGALTSTACK.
164 */
165
166 /*
167 * Changes when the API changes incompatibly.
168 * This is ONLY the API version - there is no ABI compatibility between releases.
169 *
170 * Changes in API version 2:
171 * replaced bogus -DCORO_LOOSE with grammatically more correct -DCORO_LOSER
172 * Changes in API version 3:
173 * introduced stack management (CORO_STACKALLOC)
174 */
175 #define CORO_VERSION 3
176
177 #include <stddef.h>
178
179 /*
180 * This is the type for the initialization function of a new coroutine.
181 */
182 typedef void (*coro_func)(void *);
183
184 /*
185 * A coroutine state is saved in the following structure. Treat it as an
186 * opaque type. errno and sigmask might be saved, but don't rely on it,
187 * implement your own switching primitive if you need that.
188 */
189 typedef struct coro_context coro_context;
190
191 /*
192 * This function creates a new coroutine. Apart from a pointer to an
193 * uninitialised coro_context, it expects a pointer to the entry function
194 * and the single pointer value that is given to it as argument.
195 *
196 * Allocating/deallocating the stack is your own responsibility.
197 *
198 * As a special case, if coro, arg, sptr and ssze are all zero,
199 * then an "empty" coro_context will be created that is suitable
200 * as an initial source for coro_transfer.
201 *
202 * This function is not reentrant, but putting a mutex around it
203 * will work.
204 */
205 void coro_create (coro_context *ctx, /* an uninitialised coro_context */
206 coro_func coro, /* the coroutine code to be executed */
207 void *arg, /* a single pointer passed to the coro */
208 void *sptr, /* start of stack area */
209 size_t ssze); /* size of stack area in bytes */
210
211 /*
212 * The following prototype defines the coroutine switching function. It is
213 * sometimes implemented as a macro, so watch out.
214 *
215 * This function is thread-safe and reentrant.
216 */
217 #if 0
218 void coro_transfer (coro_context *prev, coro_context *next);
219 #endif
220
221 /*
222 * The following prototype defines the coroutine destroy function. It
223 * is sometimes implemented as a macro, so watch out. It also serves no
224 * purpose unless you want to use the CORO_PTHREAD backend, where it is
225 * used to clean up the thread. You are responsible for freeing the stack
226 * and the context itself.
227 *
228 * This function is thread-safe and reentrant.
229 */
230 #if 0
231 void coro_destroy (coro_context *ctx);
232 #endif
233
234 /*****************************************************************************/
235 /* optional stack management */
236 /*****************************************************************************/
237 /*
238 * You can disable all of the stack management functions by
239 * defining CORO_STACKALLOC to 0. Otherwise, they are enabled by default.
240 *
241 * If stack management is enabled, you can influence the implementation via these
242 * symbols:
243 *
244 * -DCORO_USE_VALGRIND
245 *
246 * If defined, then libcoro will include valgrind/valgrind.h and register
247 * and unregister stacks with valgrind.
248 *
249 * -DCORO_GUARDPAGES=n
250 *
251 * libcoro will try to use the specified number of guard pages to protect against
252 * stack overflow. If n is 0, then the feature will be disabled. If it isn't
253 * defined, then libcoro will choose a suitable default. If guardpages are not
254 * supported on the platform, then the feature will be silently disabled.
255 */
256 #ifndef CORO_STACKALLOC
257 # define CORO_STACKALLOC 1
258 #endif
259
260 #if CORO_STACKALLOC
261
262 /*
263 * The only allowed operations on these struct members is to read the
264 * "sptr" and "ssze" members to pass it to coro_create, to read the "sptr"
265 * member to see if it is false, in which case the stack isn't allocated,
266 * and to set the "sptr" member to 0, to indicate to coro_stack_free to
267 * not actually do anything.
268 */
269
270 struct coro_stack
271 {
272 void *sptr;
273 size_t ssze;
274 #if CORO_USE_VALGRIND
275 int valgrind_id;
276 #endif
277 };
278
279 /*
280 * Try to allocate a stack of at least the given size and return true if
281 * successful, or false otherwise.
282 *
283 * The size is *NOT* specified in bytes, but in units of sizeof (void *),
284 * i.e. the stack is typically 4(8) times larger on 32 bit(64 bit) platforms
285 * then the size passed in.
286 *
287 * If size is 0, then a "suitable" stack size is chosen (usually 1-2MB).
288 */
289 int coro_stack_alloc (struct coro_stack *stack, unsigned int size);
290
291 /*
292 * Free the stack allocated by coro_stack_alloc again. It is safe to
293 * call this function on the coro_stack structure even if coro_stack_alloc
294 * failed.
295 */
296 void coro_stack_free (struct coro_stack *stack);
297
298 #endif
299
300 /*
301 * That was it. No other user-serviceable parts below here.
302 */
303
304 /*****************************************************************************/
305
306 #if !defined CORO_LOSER && !defined CORO_UCONTEXT \
307 && !defined CORO_SJLJ && !defined CORO_LINUX \
308 && !defined CORO_IRIX && !defined CORO_ASM \
309 && !defined CORO_PTHREAD && !defined CORO_FIBER
310 # if defined WINDOWS && (defined __i386__ || (__x86_64__ || defined _M_IX86 || defined _M_AMD64)
311 # define CORO_ASM 1
312 # elif defined WINDOWS || defined _WIN32
313 # define CORO_LOSER 1 /* you don't win with windoze */
314 # elif __linux && (__i386__ || (__x86_64__ && !__ILP32__) /*|| (__arm__ && __ARM_ARCH == 7)), not working */
315 # define CORO_ASM 1
316 # elif defined HAVE_UCONTEXT_H
317 # define CORO_UCONTEXT 1
318 # elif defined HAVE_SETJMP_H && defined HAVE_SIGALTSTACK
319 # define CORO_SJLJ 1
320 # else
321 error unknown or unsupported architecture
322 # endif
323 #endif
324
325 /*****************************************************************************/
326
327 #if CORO_UCONTEXT
328
329 # include <ucontext.h>
330
331 struct coro_context
332 {
333 ucontext_t uc;
334 };
335
336 # define coro_transfer(p,n) swapcontext (&((p)->uc), &((n)->uc))
337 # define coro_destroy(ctx) (void *)(ctx)
338
339 #elif CORO_SJLJ || CORO_LOSER || CORO_LINUX || CORO_IRIX
340
341 # if defined(CORO_LINUX) && !defined(_GNU_SOURCE)
342 # define _GNU_SOURCE /* for glibc */
343 # endif
344
345 /* try to disable well-meant but buggy checks in some libcs */
346 # ifdef _FORTIFY_SOURCE
347 # undef _FORTIFY_SOURCE
348 # undef __USE_FORTIFY_LEVEL /* helps some more when too much has been included already */
349 # endif
350
351 # if !CORO_LOSER
352 # include <unistd.h>
353 # endif
354
355 /* solaris is hopelessly borked, it expands _XOPEN_UNIX to nothing */
356 # if __sun
357 # undef _XOPEN_UNIX
358 # define _XOPEN_UNIX 1
359 # endif
360
361 # include <setjmp.h>
362
363 # if _XOPEN_UNIX > 0 || defined (_setjmp)
364 # define coro_jmp_buf jmp_buf
365 # define coro_setjmp(env) _setjmp (env)
366 # define coro_longjmp(env) _longjmp ((env), 1)
367 # elif CORO_LOSER
368 # define coro_jmp_buf jmp_buf
369 # define coro_setjmp(env) setjmp (env)
370 # define coro_longjmp(env) longjmp ((env), 1)
371 # else
372 # define coro_jmp_buf sigjmp_buf
373 # define coro_setjmp(env) sigsetjmp (env, 0)
374 # define coro_longjmp(env) siglongjmp ((env), 1)
375 # endif
376
377 struct coro_context
378 {
379 coro_jmp_buf env;
380 };
381
382 # define coro_transfer(p,n) do { if (!coro_setjmp ((p)->env)) coro_longjmp ((n)->env); } while (0)
383 # define coro_destroy(ctx) (void *)(ctx)
384
385 #elif CORO_ASM
386
387 struct coro_context
388 {
389 void **sp; /* must be at offset 0 */
390 };
391
392 #if __i386__ || __x86_64__
393 void __attribute__ ((__noinline__, __regparm__(2)))
394 #else
395 void __attribute__ ((__noinline__))
396 #endif
397 coro_transfer (coro_context *prev, coro_context *next);
398
399 # define coro_destroy(ctx) (void *)(ctx)
400
401 #elif CORO_PTHREAD
402
403 # include <pthread.h>
404
405 extern pthread_mutex_t coro_mutex;
406
407 struct coro_context
408 {
409 pthread_cond_t cv;
410 pthread_t id;
411 };
412
413 void coro_transfer (coro_context *prev, coro_context *next);
414 void coro_destroy (coro_context *ctx);
415
416 #elif CORO_FIBER
417
418 struct coro_context
419 {
420 void *fiber;
421 /* only used for initialisation */
422 coro_func coro;
423 void *arg;
424 };
425
426 void coro_transfer (coro_context *prev, coro_context *next);
427 void coro_destroy (coro_context *ctx);
428
429 #endif
430
431 #if __cplusplus
432 }
433 #endif
434
435 #endif
436