[ViewVC] Diff of: cvs/libcoro/coro.c

Comparing libcoro/coro.c (file contents):
Revision 1.12 by root, Tue Feb 22 19:33:17 2005 UTC vs.
Revision 1.38 by root, Fri Nov 7 20:12:26 2008 UTC

…		…
1	/*	1	/*
2	* Copyright (c) 2001 Marc Alexander Lehmann <pcg@goof.com>	2	* Copyright (c) 2001-2008 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	*
14	* 3. The name of the author may not be used to endorse or promote products
15	* derived from this software without specific prior written permission.
16	*	13	*
17	* 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
18	* WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MER-	15	* WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MER-
19	* CHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO	16	* CHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO
20	* 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-
23	* OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,	20	* OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
24	* WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTH-	21	* WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTH-
25	* ERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED	22	* ERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
26	* OF THE POSSIBILITY OF SUCH DAMAGE.	23	* OF THE POSSIBILITY OF SUCH DAMAGE.
27	*	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	*
28	* This library is modelled strictly after Ralf S. Engelschalls article at	36	* This library is modelled strictly after Ralf S. Engelschalls article at
29	* http://www.gnu.org/software/pth/rse-pmt.ps. So most of the credit must	37	* http://www.gnu.org/software/pth/rse-pmt.ps. So most of the credit must
30	* go to Ralf S. Engelschall <rse@engelschall.com>.	38	* go to Ralf S. Engelschall <rse@engelschall.com>.
31	*/	39	*/
32		40
33	#include "coro.h"	41	#include "coro.h"
		42
		43	#include <string.h>
34		44
35	#if !defined(STACK_ADJUST_PTR)	45	#if !defined(STACK_ADJUST_PTR)
36	/* IRIX is decidedly NON-unix */	46	/* IRIX is decidedly NON-unix */
37	# if __sgi	47	# if __sgi
38	# define STACK_ADJUST_PTR(sp,ss) ((char *)(sp) + (ss) - 8)	48	# define STACK_ADJUST_PTR(sp,ss) ((char *)(sp) + (ss) - 8)
39	# define STACK_ADJUST_SIZE(sp,ss) ((ss) - 8)	49	# define STACK_ADJUST_SIZE(sp,ss) ((ss) - 8)
40	# elif __i386__ && CORO_LINUX	50	# elif (__i386__ && CORO_LINUX) \|\| (_M_IX86 && CORO_LOSER)
41	# define STACK_ADJUST_PTR(sp,ss) ((char *)(sp) + (ss))	51	# define STACK_ADJUST_PTR(sp,ss) ((char *)(sp) + (ss))
		52	# define STACK_ADJUST_SIZE(sp,ss) (ss)
		53	# elif (__amd64__ && CORO_LINUX) \|\| ((_M_AMD64 \|\| _M_IA64) && CORO_LOSER)
		54	# define STACK_ADJUST_PTR(sp,ss) ((char *)(sp) + (ss) - 8)
42	# define STACK_ADJUST_SIZE(sp,ss) (ss)	55	# define STACK_ADJUST_SIZE(sp,ss) (ss)
43	# else	56	# else
44	# define STACK_ADJUST_PTR(sp,ss) (sp)	57	# define STACK_ADJUST_PTR(sp,ss) (sp)
45	# define STACK_ADJUST_SIZE(sp,ss) (ss)	58	# define STACK_ADJUST_SIZE(sp,ss) (ss)
46	# endif	59	# endif
47	#endif	60	#endif
48		61
		62	#if CORO_UCONTEXT
		63	# include <stddef.h>
		64	#endif
		65
49	#if CORO_SJLJ \|\| CORO_LOOSE \|\| CORO_LINUX \|\| CORO_IRIX	66	#if CORO_SJLJ \|\| CORO_LOSER \|\| CORO_LINUX \|\| CORO_IRIX \|\| CORO_ASM
50		67
		68	# include <stdlib.h>
		69
		70	# if CORO_SJLJ
		71	# include <stdio.h>
51	#include <signal.h>	72	# include <signal.h>
		73	# include <unistd.h>
		74	# endif
52		75
53	static volatile coro_func coro_init_func;	76	static volatile coro_func coro_init_func;
54	static volatile void *coro_init_arg;	77	static volatile void *coro_init_arg;
55	static volatile coro_context new_coro, create_coro;	78	static volatile coro_context new_coro, create_coro;
56		79
		80	/* what we really want to detect here is wether we use a new-enough version of GAS */
		81	/* instead, check for gcc 3, ELF and GNU/Linux and hope for the best */
		82	# if __GNUC__ >= 3 && __ELF__ && __linux__
		83	# define HAVE_CFI 1
		84	# endif
		85
57	static void	86	static void
58	coro_init (void)	87	coro_init (void)
59	{	88	{
60	volatile coro_func func = coro_init_func;	89	volatile coro_func func = coro_init_func;
61	volatile void *arg = coro_init_arg;	90	volatile void *arg = coro_init_arg;
…		…
72		101
73	static volatile int trampoline_count;	102	static volatile int trampoline_count;
74		103
75	/* trampoline signal handler */	104	/* trampoline signal handler */
76	static void	105	static void
77	trampoline(int sig)	106	trampoline (int sig)
78	{	107	{
79	if (setjmp (((coro_context *)new_coro)->env))	108	if (setjmp (((coro_context *)new_coro)->env))
		109	{
		110	# if HAVE_CFI
		111	asm (".cfi_startproc");
		112	# endif
80	coro_init (); /* start it */	113	coro_init (); /* start it */
		114	# if HAVE_CFI
		115	asm (".cfi_endproc");
		116	# endif
		117	}
81	else	118	else
82	trampoline_count++;	119	trampoline_count++;
83	}	120	}
84		121
85	# endif	122	# endif
86		123
87	#endif	124	#endif
88		125
		126	#if CORO_ASM
		127
		128	asm (
		129	".text\n"
		130	".globl coro_transfer\n"
		131	".type coro_transfer, @function\n"
		132	"coro_transfer:\n"
		133	# if __amd64
		134	# define NUM_SAVED 6
		135	"\tpush %rbp\n"
		136	"\tpush %rbx\n"
		137	"\tpush %r12\n"
		138	"\tpush %r13\n"
		139	"\tpush %r14\n"
		140	"\tpush %r15\n"
		141	"\tmov %rsp, (%rdi)\n"
		142	"\tmov (%rsi), %rsp\n"
		143	"\tpop %r15\n"
		144	"\tpop %r14\n"
		145	"\tpop %r13\n"
		146	"\tpop %r12\n"
		147	"\tpop %rbx\n"
		148	"\tpop %rbp\n"
		149	# elif __i386
		150	# define NUM_SAVED 4
		151	"\tpush %ebp\n"
		152	"\tpush %ebx\n"
		153	"\tpush %esi\n"
		154	"\tpush %edi\n"
		155	"\tmov %esp, (%eax)\n"
		156	"\tmov (%edx), %esp\n"
		157	"\tpop %edi\n"
		158	"\tpop %esi\n"
		159	"\tpop %ebx\n"
		160	"\tpop %ebp\n"
		161	# else
		162	# error unsupported architecture
		163	# endif
		164	"\tret\n"
		165	);
		166
		167	#endif
		168
		169	#if CORO_PTHREAD
		170
		171	/* this mutex will be locked by the running coroutine */
		172	pthread_mutex_t coro_mutex = PTHREAD_MUTEX_INITIALIZER;
		173
		174	struct coro_init_args
		175	{
		176	coro_func func;
		177	void *arg;
		178	coro_context self, main;
		179	};
		180
		181	static pthread_t null_tid;
		182
		183	/* I'd so love to cast pthread_mutex_unlock to void ()(void )... */
		184	static void
		185	mutex_unlock_wrapper (void *arg)
		186	{
		187	pthread_mutex_unlock ((pthread_mutex_t *)arg);
		188	}
		189
		190	static void *
		191	trampoline (void *args_)
		192	{
		193	struct coro_init_args args = (struct coro_init_args )args_;
		194	coro_func func = args->func;
		195	void *arg = args->arg;
		196
		197	pthread_mutex_lock (&coro_mutex);
		198
		199	/* we try to be good citizens and use deferred cancellation and cleanup handlers */
		200	pthread_cleanup_push (mutex_unlock_wrapper, &coro_mutex);
		201	coro_transfer (args->self, args->main);
		202	func (arg);
		203	pthread_cleanup_pop (1);
		204
		205	return 0;
		206	}
		207
		208	void
		209	coro_transfer (coro_context prev, coro_context next)
		210	{
		211	pthread_cond_signal (&next->cv);
		212	pthread_cond_wait (&prev->cv, &coro_mutex);
		213	}
		214
		215	void
		216	coro_destroy (coro_context *ctx)
		217	{
		218	if (!pthread_equal (ctx->id, null_tid))
		219	{
		220	pthread_cancel (ctx->id);
		221	pthread_mutex_unlock (&coro_mutex);
		222	pthread_join (ctx->id, 0);
		223	pthread_mutex_lock (&coro_mutex);
		224	}
		225
		226	pthread_cond_destroy (&ctx->cv);
		227	}
		228
		229	#endif
		230
89	/* initialize a machine state */	231	/* initialize a machine state */
90	void coro_create(coro_context *ctx,	232	void
91	coro_func coro, void *arg,	233	coro_create (coro_context ctx, coro_func coro, void arg, void *sptr, long ssize)
92	void *sptr, long ssize)
93	{	234	{
94	#if CORO_UCONTEXT	235	#if CORO_UCONTEXT
		236
		237	if (!coro)
		238	return;
95		239
96	getcontext (&(ctx->uc));	240	getcontext (&(ctx->uc));
97		241
98	ctx->uc.uc_link = 0;	242	ctx->uc.uc_link = 0;
99	ctx->uc.uc_stack.ss_sp = STACK_ADJUST_PTR (sptr,ssize);	243	ctx->uc.uc_stack.ss_sp = STACK_ADJUST_PTR (sptr,ssize);
100	ctx->uc.uc_stack.ss_size = (size_t) STACK_ADJUST_SIZE (sptr,ssize);	244	ctx->uc.uc_stack.ss_size = (size_t)STACK_ADJUST_SIZE (sptr,ssize);
101	ctx->uc.uc_stack.ss_flags = 0;	245	ctx->uc.uc_stack.ss_flags = 0;
102		246
103	makecontext (&(ctx->uc), (void (*)()) coro, 1, arg);	247	makecontext (&(ctx->uc), (void (*)())coro, 1, arg);
104		248
105	#elif CORO_SJLJ \|\| CORO_LOOSE \|\| CORO_LINUX \|\| CORO_IRIX	249	#elif CORO_SJLJ \|\| CORO_LOSER \|\| CORO_LINUX \|\| CORO_IRIX \|\| CORO_ASM
106		250
107	# if CORO_SJLJ	251	# if CORO_SJLJ
108	stack_t ostk, nstk;	252	stack_t ostk, nstk;
109	struct sigaction osa, nsa;	253	struct sigaction osa, nsa;
110	sigset_t nsig, osig;	254	sigset_t nsig, osig;
111	# endif	255	# endif
112	coro_context nctx;	256	coro_context nctx;
113		257
		258	if (!coro)
		259	return;
		260
114	coro_init_func = coro;	261	coro_init_func = coro;
115	coro_init_arg = arg;	262	coro_init_arg = arg;
116		263
117	new_coro = ctx;	264	new_coro = ctx;
118	create_coro = &nctx;	265	create_coro = &nctx;
…		…
127	nsa.sa_handler = trampoline;	274	nsa.sa_handler = trampoline;
128	sigemptyset (&nsa.sa_mask);	275	sigemptyset (&nsa.sa_mask);
129	nsa.sa_flags = SA_ONSTACK;	276	nsa.sa_flags = SA_ONSTACK;
130		277
131	if (sigaction (SIGUSR2, &nsa, &osa))	278	if (sigaction (SIGUSR2, &nsa, &osa))
		279	{
132	perror ("sigaction");	280	perror ("sigaction");
		281	abort ();
		282	}
133		283
134	/* set the new stack */	284	/* set the new stack */
135	nstk.ss_sp = STACK_ADJUST_PTR (sptr,ssize); /* yes, some platforms (IRIX) get this wrong. */	285	nstk.ss_sp = STACK_ADJUST_PTR (sptr,ssize); /* yes, some platforms (IRIX) get this wrong. */
136	nstk.ss_size = STACK_ADJUST_SIZE (sptr,ssize);	286	nstk.ss_size = STACK_ADJUST_SIZE (sptr,ssize);
137	nstk.ss_flags = 0;	287	nstk.ss_flags = 0;
138		288
139	if (sigaltstack (&nstk, &ostk) < 0)	289	if (sigaltstack (&nstk, &ostk) < 0)
		290	{
140	perror ("sigaltstack");	291	perror ("sigaltstack");
		292	abort ();
		293	}
141		294
142	trampoline_count = 0;	295	trampoline_count = 0;
143	kill (getpid (), SIGUSR2);	296	kill (getpid (), SIGUSR2);
144	sigfillset (&nsig); sigdelset (&nsig, SIGUSR2);	297	sigfillset (&nsig); sigdelset (&nsig, SIGUSR2);
145		298
…		…
156	abort ();	309	abort ();
157		310
158	if (~ostk.ss_flags & SS_DISABLE)	311	if (~ostk.ss_flags & SS_DISABLE)
159	sigaltstack (&ostk, 0);	312	sigaltstack (&ostk, 0);
160		313
161	sigaction (SIGUSR1, &osa, 0);	314	sigaction (SIGUSR2, &osa, 0);
162		315
163	sigprocmask (SIG_SETMASK, &osig, 0);	316	sigprocmask (SIG_SETMASK, &osig, 0);
164		317
165	# elif CORO_LOOSE	318	# elif CORO_LOSER
166		319
167	setjmp (ctx->env);	320	setjmp (ctx->env);
		321	#if __CYGWIN__
168	ctx->env[7] = (long)((char *)sptr + ssize);	322	ctx->env[7] = (long)((char *)sptr + ssize) - sizeof (long);
169	ctx->env[8] = (long)coro_init;	323	ctx->env[8] = (long)coro_init;
		324	#elif defined(_M_IX86)
		325	((_JUMP_BUFFER *)&ctx->env)->Eip = (long)coro_init;
		326	((_JUMP_BUFFER *)&ctx->env)->Esp = (long)STACK_ADJUST_PTR (sptr, ssize) - sizeof (long);
		327	#elif defined(_M_AMD64)
		328	((_JUMP_BUFFER *)&ctx->env)->Rip = (__int64)coro_init;
		329	((_JUMP_BUFFER *)&ctx->env)->Rsp = (__int64)STACK_ADJUST_PTR (sptr, ssize) - sizeof (long);
		330	#elif defined(_M_IA64)
		331	((_JUMP_BUFFER *)&ctx->env)->StIIP = (__int64)coro_init;
		332	((_JUMP_BUFFER *)&ctx->env)->IntSp = (__int64)STACK_ADJUST_PTR (sptr, ssize) - sizeof (long);
		333	#else
		334	# error "microsoft libc or architecture not supported"
		335	#endif
170		336
171	# elif CORO_LINUX	337	# elif CORO_LINUX
172		338
173	setjmp (ctx->env);	339	_setjmp (ctx->env);
174	#if defined(__GLIBC__) && defined(__GLIBC_MINOR__) \
175	&& __GLIBC__ >= 2 && __GLIBC_MINOR__ >= 0 && defined(JB_PC) && defined(JB_SP)	340	#if __GLIBC__ >= 2 && __GLIBC_MINOR__ >= 0 && defined (JB_PC) && defined (JB_SP)
176	ctx->env[0].__jmpbuf[JB_PC] = (long)coro_init;	341	ctx->env[0].__jmpbuf[JB_PC] = (long)coro_init;
177	ctx->env[0].__jmpbuf[JB_SP] = (long)STACK_ADJUST_PTR (sptr,ssize);	342	ctx->env[0].__jmpbuf[JB_SP] = (long)STACK_ADJUST_PTR (sptr, ssize) - sizeof (long);
178	#elif defined(__GLIBC__) && defined(__GLIBC_MINOR__) \
179	&& __GLIBC__ >= 2 && __GLIBC_MINOR__ >= 0 && defined(__mc68000__)	343	#elif __GLIBC__ >= 2 && __GLIBC_MINOR__ >= 0 && defined (__mc68000__)
180	ctx->env[0].__jmpbuf[0].__aregs[0] = (long int)coro_init;	344	ctx->env[0].__jmpbuf[0].__aregs[0] = (long int)coro_init;
181	ctx->env[0].__jmpbuf[0].__sp = (int )((char )sptr + ssize);	345	ctx->env[0].__jmpbuf[0].__sp = (int )((char )sptr + ssize) - sizeof (long);
182	#elif defined(__GNU_LIBRARY__) && defined(__i386__)	346	#elif defined (__GNU_LIBRARY__) && defined (__i386__)
183	ctx->env[0].__jmpbuf[0].__pc = (char *)coro_init;	347	ctx->env[0].__jmpbuf[0].__pc = (char *)coro_init;
184	ctx->env[0].__jmpbuf[0].__sp = (void )((char )sptr + ssize);	348	ctx->env[0].__jmpbuf[0].__sp = (void )((char )sptr + ssize) - sizeof (long);
185	#elif defined(__GNU_LIBRARY__) && defined(__amd64__)	349	#elif defined (__GNU_LIBRARY__) && defined (__amd64__)
186	ctx->env[0].__jmpbuf[JB_PC] = (long)coro_init;	350	ctx->env[0].__jmpbuf[JB_PC] = (long)coro_init;
187	ctx->env[0].__jmpbuf[JB_RSP] = (long)((char *)sptr + ssize);	351	ctx->env[0].__jmpbuf[0].__sp = (void )((char )sptr + ssize) - sizeof (long);
188	#else	352	#else
189	#error "linux libc or architecture not supported"	353	# error "linux libc or architecture not supported"
190	#endif	354	#endif
191		355
192	# elif CORO_IRIX	356	# elif CORO_IRIX
193		357
194	setjmp (ctx->env);	358	setjmp (ctx->env);
195	ctx->env[JB_PC] = (__uint64_t)coro_init;	359	ctx->env[JB_PC] = (__uint64_t)coro_init;
196	ctx->env[JB_SP] = (__uint64_t)STACK_ADJUST_PTR (sptr,ssize);	360	ctx->env[JB_SP] = (__uint64_t)STACK_ADJUST_PTR (sptr, ssize) - sizeof (long);
		361
		362	# elif CORO_ASM
		363
		364	ctx->sp = (volatile void *)(ssize + (char )sptr);
		365	--ctx->sp = (void )abort; /* needed for alignment only */
		366	--ctx->sp = (void )coro_init;
		367	ctx->sp -= NUM_SAVED;
197		368
198	# endif	369	# endif
199		370
200	coro_transfer ((coro_context )create_coro, (coro_context )new_coro);	371	coro_transfer ((coro_context )create_coro, (coro_context )new_coro);
201		372
		373	# elif CORO_PTHREAD
		374
		375	static coro_context nctx;
		376	static int once;
		377
		378	if (!once)
		379	{
		380	once = 1;
		381
		382	pthread_mutex_lock (&coro_mutex);
		383	pthread_cond_init (&nctx.cv, 0);
		384	null_tid = pthread_self ();
		385	}
		386
		387	pthread_cond_init (&ctx->cv, 0);
		388
		389	if (coro)
		390	{
		391	pthread_attr_t attr;
		392	struct coro_init_args args;
		393
		394	args.func = coro;
		395	args.arg = arg;
		396	args.self = ctx;
		397	args.main = &nctx;
		398
		399	pthread_attr_init (&attr);
		400	pthread_attr_setstack (&attr, sptr, (size_t)ssize);
		401	pthread_attr_setscope (&attr, PTHREAD_SCOPE_PROCESS);
		402	pthread_create (&ctx->id, &attr, trampoline, &args);
		403
		404	coro_transfer (args.main, args.self);
		405	}
		406	else
		407	ctx->id = null_tid;
		408
202	#else	409	#else
203	error unsupported architecture	410	# error unsupported backend
204	#endif	411	#endif
205	}	412	}
206		413

Diff Legend

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

Comparing libcoro/coro.c (file contents): Revision 1.12 by root, Tue Feb 22 19:33:17 2005 UTC vs. Revision 1.38 by root, Fri Nov 7 20:12:26 2008 UTC

Diff Legend

Comparing libcoro/coro.c (file contents):
Revision 1.12 by root, Tue Feb 22 19:33:17 2005 UTC vs.
Revision 1.38 by root, Fri Nov 7 20:12:26 2008 UTC