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

Comparing libcoro/coro.h (file contents):
Revision 1.32 by root, Fri Nov 7 20:12:26 2008 UTC vs.
Revision 1.59 by root, Fri Nov 18 05:45:00 2016 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,
… …
60 * 2008-11-05 Hopefully fix misaligned stacks with CORO_ASM/SETJMP. 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. 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. 62 * introduce coro_destroy, which is a nop except for pthreads.
63 * speed up CORO_PTHREAD. Do no longer leak threads either. 63 * speed up CORO_PTHREAD. Do no longer leak threads either.
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.
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 experimental fix for gcc 6 on amd64 (missing cfi_startproc).
65 */ 89 */
66 90
67#ifndef CORO_H 91#ifndef CORO_H
68#define CORO_H 92#define CORO_H
69 93
70#define CORO_VERSION 2 94#if __cplusplus
71 95extern "C" {
72/* 96#endif
73 * Changes since API version 1:
74 * replaced bogus -DCORO_LOOSE with gramatically more correct -DCORO_LOSER
75 */
76 97
77/* 98/*
78 * This library consists of only three files 99 * This library consists of only three files
79 * coro.h, coro.c and LICENSE (and optionally README) 100 * coro.h, coro.c and LICENSE (and optionally README)
80 * 101 *
81 * It implements what is known as coroutines, in a hopefully 102 * It implements what is known as coroutines, in a hopefully
82 * portable way. At the moment you have to define which kind 103 * portable way.
83 * of implementation flavour you want: 104 *
105 * All compiletime symbols must be defined both when including coro.h
106 * (using libcoro) as well as when compiling coro.c (the implementation).
107 *
108 * You can manually specify which flavour you want. If you don't define
109 * any of these, libcoro tries to choose a safe and fast default:
84 * 110 *
85 * -DCORO_UCONTEXT 111 * -DCORO_UCONTEXT
86 * 112 *
87 * This flavour uses SUSv2's get/set/swap/makecontext functions that 113 * This flavour uses SUSv2's get/set/swap/makecontext functions that
88 * unfortunately only newer unices support. 114 * unfortunately only some unices support, and is quite slow.
89 * 115 *
90 * -DCORO_SJLJ 116 * -DCORO_SJLJ
91 * 117 *
92 * This flavour uses SUSv2's setjmp/longjmp and sigaltstack functions to 118 * This flavour uses SUSv2's setjmp/longjmp and sigaltstack functions to
93 * do it's job. Coroutine creation is much slower than UCONTEXT, but 119 * do it's job. Coroutine creation is much slower than UCONTEXT, but
94 * context switching is often a bit cheaper. It should work on almost 120 * context switching is a bit cheaper. It should work on almost all unices.
95 * all unices.
96 * 121 *
97 * -DCORO_LINUX 122 * -DCORO_LINUX
98 * 123 *
124 * CORO_SJLJ variant.
99 * Old GNU/Linux systems (<= glibc-2.1) only work with this implementation 125 * Old GNU/Linux systems (<= glibc-2.1) only work with this implementation
100 * (it is very fast and therefore recommended over other methods, but 126 * (it is very fast and therefore recommended over other methods, but
101 * doesn't work with anything newer). 127 * doesn't work with anything newer).
102 * 128 *
103 * -DCORO_LOSER 129 * -DCORO_LOSER
104 * 130 *
131 * CORO_SJLJ variant.
105 * Microsoft's highly proprietary platform doesn't support sigaltstack, and 132 * Microsoft's highly proprietary platform doesn't support sigaltstack, and
106 * this automatically selects a suitable workaround for this platform. 133 * this selects a suitable workaround for this platform. It might not work
107 * (untested) 134 * with your compiler though - it has only been tested with MSVC 6.
135 *
136 * -DCORO_FIBER
137 *
138 * Slower, but probably more portable variant for the Microsoft operating
139 * system, using fibers. Ignores the passed stack and allocates it internally.
140 * Also, due to bugs in cygwin, this does not work with cygwin.
108 * 141 *
109 * -DCORO_IRIX 142 * -DCORO_IRIX
110 * 143 *
144 * CORO_SJLJ variant.
111 * SGI's version of Microsoft's NT ;) 145 * For SGI's version of Microsoft's NT ;)
112 * 146 *
113 * -DCORO_ASM 147 * -DCORO_ASM
114 * 148 *
115 * Handcoded assembly, known to work only on a few architectures/ABI: 149 * Hand coded assembly, known to work only on a few architectures/ABI:
116 * ELF Linux x86 && amd64 when gcc is used and optimisation is turned on. 150 * GCC + arm7/x86/IA32/amd64/x86_64 + GNU/Linux and a few BSDs. Fastest
151 * choice, if it works.
117 * 152 *
118 * -DCORO_PTHREAD 153 * -DCORO_PTHREAD
119 * 154 *
120 * Use the pthread API. You have to provide <pthread.h> and -lpthread. 155 * Use the pthread API. You have to provide <pthread.h> and -lpthread.
156 * This is likely the slowest backend, and it also does not support fork(),
157 * so avoid it at all costs.
121 * 158 *
122 * If you define neither of these symbols, coro.h will try to autodetect 159 * If you define neither of these symbols, coro.h will try to autodetect
123 * the model. This currently works for CORO_LOSER only. For the other 160 * the best/safest model. To help with the autodetection, you should check
124 * alternatives you should check (e.g. using autoconf) and define the 161 * (e.g. using autoconf) and define the following symbols: HAVE_UCONTEXT_H
125 * following symbols: HAVE_UCONTEXT_H / HAVE_SETJMP_H / HAVE_SIGALTSTACK. 162 * / HAVE_SETJMP_H / HAVE_SIGALTSTACK.
163 */
164
165/*
166 * Changes when the API changes incompatibly.
167 * This is ONLY the API version - there is no ABI compatibility between releases.
126 */ 168 *
169 * Changes in API version 2:
170 * replaced bogus -DCORO_LOOSE with grammatically more correct -DCORO_LOSER
171 * Changes in API version 3:
172 * introduced stack management (CORO_STACKALLOC)
173 */
174#define CORO_VERSION 3
175
176#include <stddef.h>
127 177
128/* 178/*
129 * This is the type for the initialization function of a new coroutine. 179 * This is the type for the initialization function of a new coroutine.
130 */ 180 */
131typedef void (*coro_func)(void *); 181typedef void (*coro_func)(void *);
… …
142 * uninitialised coro_context, it expects a pointer to the entry function 192 * uninitialised coro_context, it expects a pointer to the entry function
143 * and the single pointer value that is given to it as argument. 193 * and the single pointer value that is given to it as argument.
144 * 194 *
145 * Allocating/deallocating the stack is your own responsibility. 195 * Allocating/deallocating the stack is your own responsibility.
146 * 196 *
147 * As a special case, if coro, arg, sptr and ssize are all zero, 197 * As a special case, if coro, arg, sptr and ssze are all zero,
148 * then an "empty" coro_contetx will be created that is suitable 198 * then an "empty" coro_context will be created that is suitable
149 * as an initial source for coro_transfer. 199 * as an initial source for coro_transfer.
150 * 200 *
151 * This function is not reentrant, but putting a mutex around it 201 * This function is not reentrant, but putting a mutex around it
152 * will work. 202 * will work.
153 */ 203 */
154void coro_create (coro_context *ctx, /* an uninitialised coro_context */ 204void coro_create (coro_context *ctx, /* an uninitialised coro_context */
155 coro_func coro, /* the coroutine code to be executed */ 205 coro_func coro, /* the coroutine code to be executed */
156 void *arg, /* a single pointer passed to the coro */ 206 void *arg, /* a single pointer passed to the coro */
157 void *sptr, /* start of stack area */ 207 void *sptr, /* start of stack area */
158 long ssize); /* size of stack area */ 208 size_t ssze); /* size of stack area in bytes */
159 209
160/* 210/*
161 * The following prototype defines the coroutine switching function. It is 211 * The following prototype defines the coroutine switching function. It is
162 * usually implemented as a macro, so watch out. 212 * sometimes implemented as a macro, so watch out.
163 * 213 *
164 * This function is thread-safe and reentrant. 214 * This function is thread-safe and reentrant.
165 */ 215 */
166#if 0 216#if 0
167void coro_transfer (coro_context *prev, coro_context *next); 217void coro_transfer (coro_context *prev, coro_context *next);
168#endif 218#endif
169 219
170/* 220/*
171 * The following prototype defines the coroutine destroy function. It is 221 * The following prototype defines the coroutine destroy function. It
172 * usually implemented as a macro, so watch out. It also serves 222 * is sometimes implemented as a macro, so watch out. It also serves no
173 * no purpose unless you want to use the CORO_PTHREAD backend. 223 * purpose unless you want to use the CORO_PTHREAD backend, where it is
224 * used to clean up the thread. You are responsible for freeing the stack
225 * and the context itself.
174 * 226 *
175 * This function is thread-safe and reentrant. 227 * This function is thread-safe and reentrant.
176 */ 228 */
177#if 0 229#if 0
178void coro_destroy (coro_context *ctx); 230void coro_destroy (coro_context *ctx);
179#endif 231#endif
180 232
181/*
182 * That was it. No other user-visible functions are implemented here.
183 */
184
185/*****************************************************************************/ 233/*****************************************************************************/
234/* optional stack management */
235/*****************************************************************************/
236/*
237 * You can disable all of the stack management functions by
238 * defining CORO_STACKALLOC to 0. Otherwise, they are enabled by default.
239 *
240 * If stack management is enabled, you can influence the implementation via these
241 * symbols:
242 *
243 * -DCORO_USE_VALGRIND
244 *
245 * If defined, then libcoro will include valgrind/valgrind.h and register
246 * and unregister stacks with valgrind.
247 *
248 * -DCORO_GUARDPAGES=n
249 *
250 * libcoro will try to use the specified number of guard pages to protect against
251 * stack overflow. If n is 0, then the feature will be disabled. If it isn't
252 * defined, then libcoro will choose a suitable default. If guardpages are not
253 * supported on the platform, then the feature will be silently disabled.
254 */
255#ifndef CORO_STACKALLOC
256# define CORO_STACKALLOC 1
257#endif
186 258
259#if CORO_STACKALLOC
260
261/*
262 * The only allowed operations on these struct members is to read the
263 * "sptr" and "ssze" members to pass it to coro_create, to read the "sptr"
264 * member to see if it is false, in which case the stack isn't allocated,
265 * and to set the "sptr" member to 0, to indicate to coro_stack_free to
266 * not actually do anything.
267 */
268
269struct coro_stack
270{
271 void *sptr;
272 size_t ssze;
273#if CORO_USE_VALGRIND
274 int valgrind_id;
275#endif
276};
277
278/*
279 * Try to allocate a stack of at least the given size and return true if
280 * successful, or false otherwise.
281 *
282 * The size is *NOT* specified in bytes, but in units of sizeof (void *),
283 * i.e. the stack is typically 4(8) times larger on 32 bit(64 bit) platforms
284 * then the size passed in.
285 *
286 * If size is 0, then a "suitable" stack size is chosen (usually 1-2MB).
287 */
288int coro_stack_alloc (struct coro_stack *stack, unsigned int size);
289
290/*
291 * Free the stack allocated by coro_stack_alloc again. It is safe to
292 * call this function on the coro_stack structure even if coro_stack_alloc
293 * failed.
294 */
295void coro_stack_free (struct coro_stack *stack);
296
297#endif
298
299/*
300 * That was it. No other user-serviceable parts below here.
301 */
302
303/*****************************************************************************/
304
187#if !defined(CORO_LOSER) && !defined(CORO_UCONTEXT) \ 305#if !defined CORO_LOSER && !defined CORO_UCONTEXT \
188 && !defined(CORO_SJLJ) && !defined(CORO_LINUX) \ 306 && !defined CORO_SJLJ && !defined CORO_LINUX \
189 && !defined(CORO_IRIX) && !defined(CORO_ASM) \ 307 && !defined CORO_IRIX && !defined CORO_ASM \
190 && !defined(CORO_PTHREAD) 308 && !defined CORO_PTHREAD && !defined CORO_FIBER
191# if defined(WINDOWS) 309# if defined WINDOWS && (defined __i386__ || (__x86_64__ || defined _M_IX86 || defined _M_AMD64)
310# define CORO_ASM 1
311# elif defined WINDOWS || defined _WIN32
192# define CORO_LOSER 1 /* you don't win with windoze */ 312# define CORO_LOSER 1 /* you don't win with windoze */
193# elif defined(__linux) && (defined(__x86) || defined (__amd64)) 313# elif __linux && (__i386__ || (__x86_64__ && !__ILP32__) /*|| (__arm__ && __ARM_ARCH == 7)), not working */
194# define CORO_ASM 1 314# define CORO_ASM 1
195# elif defined(HAVE_UCONTEXT_H) 315# elif defined HAVE_UCONTEXT_H
196# define CORO_UCONTEXT 1 316# define CORO_UCONTEXT 1
197# elif defined(HAVE_SETJMP_H) && defined(HAVE_SIGALTSTACK) 317# elif defined HAVE_SETJMP_H && defined HAVE_SIGALTSTACK
198# define CORO_SJLJ 1 318# define CORO_SJLJ 1
199# else 319# else
200error unknown or unsupported architecture 320error unknown or unsupported architecture
201# endif 321# endif
202#endif 322#endif
… …
205 325
206#if CORO_UCONTEXT 326#if CORO_UCONTEXT
207 327
208# include <ucontext.h> 328# include <ucontext.h>
209 329
210struct coro_context { 330struct coro_context
331{
211 ucontext_t uc; 332 ucontext_t uc;
212}; 333};
213 334
214# define coro_transfer(p,n) swapcontext (&((p)->uc), &((n)->uc)) 335# define coro_transfer(p,n) swapcontext (&((p)->uc), &((n)->uc))
215# define coro_destroy(ctx) (void *)(ctx) 336# define coro_destroy(ctx) (void *)(ctx)
216 337
217#elif CORO_SJLJ || CORO_LOSER || CORO_LINUX || CORO_IRIX 338#elif CORO_SJLJ || CORO_LOSER || CORO_LINUX || CORO_IRIX
218 339
219# if defined(CORO_LINUX) && !defined(_GNU_SOURCE) 340# if defined(CORO_LINUX) && !defined(_GNU_SOURCE)
220# define _GNU_SOURCE /* for linux libc */ 341# define _GNU_SOURCE /* for glibc */
342# endif
343
344/* try to disable well-meant but buggy checks in some libcs */
345# ifdef _FORTIFY_SOURCE
346# undef _FORTIFY_SOURCE
347# undef __USE_FORTIFY_LEVEL /* helps some more when too much has been included already */
348# endif
349
350# if !CORO_LOSER
351# include <unistd.h>
352# endif
353
354/* solaris is hopelessly borked, it expands _XOPEN_UNIX to nothing */
355# if __sun
356# undef _XOPEN_UNIX
357# define _XOPEN_UNIX 1
221# endif 358# endif
222 359
223# include <setjmp.h> 360# include <setjmp.h>
224 361
225struct coro_context { 362# if _XOPEN_UNIX > 0 || defined (_setjmp)
226 jmp_buf env; 363# define coro_jmp_buf jmp_buf
227}; 364# define coro_setjmp(env) _setjmp (env)
228 365# define coro_longjmp(env) _longjmp ((env), 1)
229# if CORO_LINUX || (_XOPEN_SOURCE >= 600) 366# elif CORO_LOSER
230# define coro_transfer(p,n) do { if (!_setjmp ((p)->env)) _longjmp ((n)->env, 1); } while (0) 367# define coro_jmp_buf jmp_buf
368# define coro_setjmp(env) setjmp (env)
369# define coro_longjmp(env) longjmp ((env), 1)
231# else 370# else
371# define coro_jmp_buf sigjmp_buf
372# define coro_setjmp(env) sigsetjmp (env, 0)
373# define coro_longjmp(env) siglongjmp ((env), 1)
374# endif
375
376struct coro_context
377{
378 coro_jmp_buf env;
379};
380
232# define coro_transfer(p,n) do { if (!setjmp ((p)->env)) longjmp ((n)->env, 1); } while (0) 381# define coro_transfer(p,n) do { if (!coro_setjmp ((p)->env)) coro_longjmp ((n)->env); } while (0)
233# endif
234
235# define coro_destroy(ctx) (void *)(ctx) 382# define coro_destroy(ctx) (void *)(ctx)
236 383
237#elif CORO_ASM 384#elif CORO_ASM
238 385
239struct coro_context { 386struct coro_context
387{
240 volatile void **sp; /* must be at offset 0 */ 388 void **sp; /* must be at offset 0 */
241}; 389};
242 390
391#if __i386__ || __x86_64__
243void __attribute__ ((__noinline__, __regparm__(2))) 392void __attribute__ ((__noinline__, __regparm__(2)))
393#else
394void __attribute__ ((__noinline__))
395#endif
244coro_transfer (coro_context *prev, coro_context *next); 396coro_transfer (coro_context *prev, coro_context *next);
245 397
246# define coro_destroy(ctx) (void *)(ctx) 398# define coro_destroy(ctx) (void *)(ctx)
247 399
248#elif CORO_PTHREAD 400#elif CORO_PTHREAD
249 401
250# include <pthread.h> 402# include <pthread.h>
251 403
252extern pthread_mutex_t coro_mutex; 404extern pthread_mutex_t coro_mutex;
253 405
254struct coro_context { 406struct coro_context
407{
255 pthread_cond_t cv; 408 pthread_cond_t cv;
256 pthread_t id; 409 pthread_t id;
257}; 410};
258 411
259void coro_transfer (coro_context *prev, coro_context *next); 412void coro_transfer (coro_context *prev, coro_context *next);
260void coro_destroy (coro_context *ctx); 413void coro_destroy (coro_context *ctx);
261 414
262#endif 415#elif CORO_FIBER
263 416
264#endif 417struct coro_context
418{
419 void *fiber;
420 /* only used for initialisation */
421 coro_func coro;
422 void *arg;
423};
265 424
425void coro_transfer (coro_context *prev, coro_context *next);
426void coro_destroy (coro_context *ctx);
427
428#endif
429
430#if __cplusplus
431}
432#endif
433
434#endif
435

Diff Legend

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