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

Comparing cvsroot/libcoro/coro.c (file contents):
Revision 1.38 by root, Fri Nov 7 20:12:26 2008 UTC vs.
Revision 1.63 by root, Wed Dec 5 01:09:12 2012 UTC

…		…
1	/*	1	/*
2	* Copyright (c) 2001-2008 Marc Alexander Lehmann <schmorp@schmorp.de>	2	* Copyright (c) 2001-2011 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,
40		40
41	#include "coro.h"	41	#include "coro.h"
42		42
43	#include <string.h>	43	#include <string.h>
44		44
		45	/*****************************************************************************/
		46	/* ucontext/setjmp/asm backends */
		47	/*****************************************************************************/
		48	#if CORO_UCONTEXT \|\| CORO_SJLJ \|\| CORO_LOSER \|\| CORO_LINUX \|\| CORO_IRIX \|\| CORO_ASM
		49
		50	# if CORO_UCONTEXT
		51	# include <stddef.h>
		52	# endif
		53
45	#if !defined(STACK_ADJUST_PTR)	54	# if !defined(STACK_ADJUST_PTR)
		55	# if __sgi
46	/* IRIX is decidedly NON-unix */	56	/* IRIX is decidedly NON-unix */
47	# if __sgi
48	# define STACK_ADJUST_PTR(sp,ss) ((char *)(sp) + (ss) - 8)	57	# define STACK_ADJUST_PTR(sp,ss) ((char *)(sp) + (ss) - 8)
49	# define STACK_ADJUST_SIZE(sp,ss) ((ss) - 8)	58	# define STACK_ADJUST_SIZE(sp,ss) ((ss) - 8)
50	# elif (__i386__ && CORO_LINUX) \|\| (_M_IX86 && CORO_LOSER)	59	# elif (__i386__ && CORO_LINUX) \|\| (_M_IX86 && CORO_LOSER)
51	# define STACK_ADJUST_PTR(sp,ss) ((char *)(sp) + (ss))	60	# define STACK_ADJUST_PTR(sp,ss) ((char *)(sp) + (ss))
52	# define STACK_ADJUST_SIZE(sp,ss) (ss)	61	# define STACK_ADJUST_SIZE(sp,ss) (ss)
53	# elif (__amd64__ && CORO_LINUX) \|\| ((_M_AMD64 \|\| _M_IA64) && CORO_LOSER)	62	# elif (__amd64__ && CORO_LINUX) \|\| ((_M_AMD64 \|\| _M_IA64) && CORO_LOSER)
54	# define STACK_ADJUST_PTR(sp,ss) ((char *)(sp) + (ss) - 8)	63	# define STACK_ADJUST_PTR(sp,ss) ((char *)(sp) + (ss) - 8)
55	# define STACK_ADJUST_SIZE(sp,ss) (ss)	64	# define STACK_ADJUST_SIZE(sp,ss) (ss)
56	# else	65	# else
57	# define STACK_ADJUST_PTR(sp,ss) (sp)	66	# define STACK_ADJUST_PTR(sp,ss) (sp)
58	# define STACK_ADJUST_SIZE(sp,ss) (ss)	67	# define STACK_ADJUST_SIZE(sp,ss) (ss)
59	# endif	68	# endif
60	#endif	69	# endif
61
62	#if CORO_UCONTEXT
63	# include <stddef.h>
64	#endif
65
66	#if CORO_SJLJ \|\| CORO_LOSER \|\| CORO_LINUX \|\| CORO_IRIX \|\| CORO_ASM
67		70
68	# include <stdlib.h>	71	# include <stdlib.h>
69		72
70	# if CORO_SJLJ	73	# if CORO_SJLJ
71	# include <stdio.h>	74	# include <stdio.h>
72	# include <signal.h>	75	# include <signal.h>
73	# include <unistd.h>	76	# include <unistd.h>
74	# endif	77	# endif
75		78
76	static volatile coro_func coro_init_func;	79	static coro_func coro_init_func;
77	static volatile void *coro_init_arg;	80	static void *coro_init_arg;
78	static volatile coro_context new_coro, create_coro;	81	static coro_context new_coro, create_coro;
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		82
86	static void	83	static void
87	coro_init (void)	84	coro_init (void)
88	{	85	{
89	volatile coro_func func = coro_init_func;	86	volatile coro_func func = coro_init_func;
90	volatile void *arg = coro_init_arg;	87	volatile void *arg = coro_init_arg;
91		88
92	coro_transfer ((coro_context )new_coro, (coro_context )create_coro);	89	coro_transfer (new_coro, create_coro);
		90
		91	#if __GCC_HAVE_DWARF2_CFI_ASM && __amd64
		92	asm (".cfi_undefined rip");
		93	#endif
93		94
94	func ((void *)arg);	95	func ((void *)arg);
95		96
96	/* the new coro returned. bad. just abort() for now */	97	/* the new coro returned. bad. just abort() for now */
97	abort ();	98	abort ();
98	}	99	}
99		100
100	# if CORO_SJLJ	101	# if CORO_SJLJ
101		102
102	static volatile int trampoline_count;	103	static volatile int trampoline_done;
103		104
104	/* trampoline signal handler */	105	/* trampoline signal handler */
105	static void	106	static void
106	trampoline (int sig)	107	trampoline (int sig)
107	{	108	{
108	if (setjmp (((coro_context *)new_coro)->env))	109	if (coro_setjmp (new_coro->env))
		110	coro_init (); /* start it */
		111	else
		112	trampoline_done = 1;
		113	}
		114
		115	# endif
		116
		117	# if CORO_ASM
		118
		119	#if _WIN32 \|\| __CYGWIN__
		120	#define CORO_WIN_TIB 1
		121	#endif
		122
		123	asm (
		124	"\t.text\n"
		125	#if _WIN32 \|\| __CYGWIN__
		126	"\t.globl _coro_transfer\n"
		127	"_coro_transfer:\n"
		128	#else
		129	"\t.globl coro_transfer\n"
		130	"coro_transfer:\n"
		131	#endif
		132	/* windows, of course, gives a shit on the amd64 ABI and uses different registers */
		133	/* http://blogs.msdn.com/freik/archive/2005/03/17/398200.aspx */
		134	#if __amd64
		135
		136	#if _WIN32 \|\| __CYGWIN__
		137	#define NUM_SAVED 29
		138	"\tsubq $168, %rsp\t" /* one dummy qword to improve alignment */
		139	"\tmovaps %xmm6, (%rsp)\n"
		140	"\tmovaps %xmm7, 16(%rsp)\n"
		141	"\tmovaps %xmm8, 32(%rsp)\n"
		142	"\tmovaps %xmm9, 48(%rsp)\n"
		143	"\tmovaps %xmm10, 64(%rsp)\n"
		144	"\tmovaps %xmm11, 80(%rsp)\n"
		145	"\tmovaps %xmm12, 96(%rsp)\n"
		146	"\tmovaps %xmm13, 112(%rsp)\n"
		147	"\tmovaps %xmm14, 128(%rsp)\n"
		148	"\tmovaps %xmm15, 144(%rsp)\n"
		149	"\tpushq %rsi\n"
		150	"\tpushq %rdi\n"
		151	"\tpushq %rbp\n"
		152	"\tpushq %rbx\n"
		153	"\tpushq %r12\n"
		154	"\tpushq %r13\n"
		155	"\tpushq %r14\n"
		156	"\tpushq %r15\n"
		157	#if CORO_WIN_TIB
		158	"\tpushq %fs:0x0\n"
		159	"\tpushq %fs:0x8\n"
		160	"\tpushq %fs:0xc\n"
		161	#endif
		162	"\tmovq %rsp, (%rcx)\n"
		163	"\tmovq (%rdx), %rsp\n"
		164	#if CORO_WIN_TIB
		165	"\tpopq %fs:0xc\n"
		166	"\tpopq %fs:0x8\n"
		167	"\tpopq %fs:0x0\n"
		168	#endif
		169	"\tpopq %r15\n"
		170	"\tpopq %r14\n"
		171	"\tpopq %r13\n"
		172	"\tpopq %r12\n"
		173	"\tpopq %rbx\n"
		174	"\tpopq %rbp\n"
		175	"\tpopq %rdi\n"
		176	"\tpopq %rsi\n"
		177	"\tmovaps (%rsp), %xmm6\n"
		178	"\tmovaps 16(%rsp), %xmm7\n"
		179	"\tmovaps 32(%rsp), %xmm8\n"
		180	"\tmovaps 48(%rsp), %xmm9\n"
		181	"\tmovaps 64(%rsp), %xmm10\n"
		182	"\tmovaps 80(%rsp), %xmm11\n"
		183	"\tmovaps 96(%rsp), %xmm12\n"
		184	"\tmovaps 112(%rsp), %xmm13\n"
		185	"\tmovaps 128(%rsp), %xmm14\n"
		186	"\tmovaps 144(%rsp), %xmm15\n"
		187	"\taddq $168, %rsp\n"
		188	#else
		189	#define NUM_SAVED 6
		190	"\tpushq %rbp\n"
		191	"\tpushq %rbx\n"
		192	"\tpushq %r12\n"
		193	"\tpushq %r13\n"
		194	"\tpushq %r14\n"
		195	"\tpushq %r15\n"
		196	"\tmovq %rsp, (%rdi)\n"
		197	"\tmovq (%rsi), %rsp\n"
		198	"\tpopq %r15\n"
		199	"\tpopq %r14\n"
		200	"\tpopq %r13\n"
		201	"\tpopq %r12\n"
		202	"\tpopq %rbx\n"
		203	"\tpopq %rbp\n"
		204	#endif
		205	"\tpopq %rcx\n"
		206	"\tjmpq *%rcx\n"
		207
		208	#elif __i386
		209
		210	#define NUM_SAVED 4
		211	"\tpushl %ebp\n"
		212	"\tpushl %ebx\n"
		213	"\tpushl %esi\n"
		214	"\tpushl %edi\n"
		215	#if CORO_WIN_TIB
		216	#undef NUM_SAVED
		217	#define NUM_SAVED 7
		218	"\tpushl %fs:0\n"
		219	"\tpushl %fs:4\n"
		220	"\tpushl %fs:8\n"
		221	#endif
		222	"\tmovl %esp, (%eax)\n"
		223	"\tmovl (%edx), %esp\n"
		224	#if CORO_WIN_TIB
		225	"\tpopl %fs:8\n"
		226	"\tpopl %fs:4\n"
		227	"\tpopl %fs:0\n"
		228	#endif
		229	"\tpopl %edi\n"
		230	"\tpopl %esi\n"
		231	"\tpopl %ebx\n"
		232	"\tpopl %ebp\n"
		233	"\tpopl %ecx\n"
		234	"\tjmpl *%ecx\n"
		235
		236	#else
		237	#error unsupported architecture
		238	#endif
		239	);
		240
		241	# endif
		242
		243	void
		244	coro_create (coro_context ctx, coro_func coro, void arg, void *sptr, long ssize)
		245	{
		246	coro_context nctx;
		247	# if CORO_SJLJ
		248	stack_t ostk, nstk;
		249	struct sigaction osa, nsa;
		250	sigset_t nsig, osig;
		251	# endif
		252
		253	if (!coro)
		254	return;
		255
		256	coro_init_func = coro;
		257	coro_init_arg = arg;
		258
		259	new_coro = ctx;
		260	create_coro = &nctx;
		261
		262	# if CORO_SJLJ
		263	/* we use SIGUSR2. first block it, then fiddle with it. */
		264
		265	sigemptyset (&nsig);
		266	sigaddset (&nsig, SIGUSR2);
		267	sigprocmask (SIG_BLOCK, &nsig, &osig);
		268
		269	nsa.sa_handler = trampoline;
		270	sigemptyset (&nsa.sa_mask);
		271	nsa.sa_flags = SA_ONSTACK;
		272
		273	if (sigaction (SIGUSR2, &nsa, &osa))
109	{	274	{
110	# if HAVE_CFI	275	perror ("sigaction");
111	asm (".cfi_startproc");	276	abort ();
112	# endif
113	coro_init (); /* start it */
114	# if HAVE_CFI
115	asm (".cfi_endproc");
116	# endif
117	}	277	}
		278
		279	/* set the new stack */
		280	nstk.ss_sp = STACK_ADJUST_PTR (sptr, ssize); /* yes, some platforms (IRIX) get this wrong. */
		281	nstk.ss_size = STACK_ADJUST_SIZE (sptr, ssize);
		282	nstk.ss_flags = 0;
		283
		284	if (sigaltstack (&nstk, &ostk) < 0)
		285	{
		286	perror ("sigaltstack");
		287	abort ();
		288	}
		289
		290	trampoline_done = 0;
		291	kill (getpid (), SIGUSR2);
		292	sigfillset (&nsig); sigdelset (&nsig, SIGUSR2);
		293
		294	while (!trampoline_done)
		295	sigsuspend (&nsig);
		296
		297	sigaltstack (0, &nstk);
		298	nstk.ss_flags = SS_DISABLE;
		299	if (sigaltstack (&nstk, 0) < 0)
		300	perror ("sigaltstack");
		301
		302	sigaltstack (0, &nstk);
		303	if (~nstk.ss_flags & SS_DISABLE)
		304	abort ();
		305
		306	if (~ostk.ss_flags & SS_DISABLE)
		307	sigaltstack (&ostk, 0);
		308
		309	sigaction (SIGUSR2, &osa, 0);
		310	sigprocmask (SIG_SETMASK, &osig, 0);
		311
		312	# elif CORO_LOSER
		313
		314	coro_setjmp (ctx->env);
		315	#if __CYGWIN__ && __i386
		316	ctx->env[8] = (long) coro_init;
		317	ctx->env[7] = (long) ((char *)sptr + ssize) - sizeof (long);
		318	#elif __CYGWIN__ && __x86_64
		319	ctx->env[7] = (long) coro_init;
		320	ctx->env[6] = (long) ((char *)sptr + ssize) - sizeof (long);
		321	#elif defined __MINGW32__
		322	ctx->env[5] = (long) coro_init;
		323	ctx->env[4] = (long) ((char *)sptr + ssize) - sizeof (long);
		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 (__int64);
		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 (__int64);
118	else	333	#else
119	trampoline_count++;	334	#error "microsoft libc or architecture not supported"
120	}
121
122	# endif	335	#endif
123		336
		337	# elif CORO_LINUX
		338
		339	coro_setjmp (ctx->env);
		340	#if __GLIBC__ >= 2 && __GLIBC_MINOR__ >= 0 && defined (JB_PC) && defined (JB_SP)
		341	ctx->env[0].__jmpbuf[JB_PC] = (long) coro_init;
		342	ctx->env[0].__jmpbuf[JB_SP] = (long) STACK_ADJUST_PTR (sptr, ssize) - sizeof (long);
		343	#elif __GLIBC__ >= 2 && __GLIBC_MINOR__ >= 0 && defined (__mc68000__)
		344	ctx->env[0].__jmpbuf[0].__aregs[0] = (long int)coro_init;
		345	ctx->env[0].__jmpbuf[0].__sp = (int ) ((char )sptr + ssize) - sizeof (long);
		346	#elif defined (__GNU_LIBRARY__) && defined (__i386__)
		347	ctx->env[0].__jmpbuf[0].__pc = (char *) coro_init;
		348	ctx->env[0].__jmpbuf[0].__sp = (void ) ((char )sptr + ssize) - sizeof (long);
		349	#elif defined (__GNU_LIBRARY__) && defined (__amd64__)
		350	ctx->env[0].__jmpbuf[JB_PC] = (long) coro_init;
		351	ctx->env[0].__jmpbuf[0].__sp = (void ) ((char )sptr + ssize) - sizeof (long);
		352	#else
		353	#error "linux libc or architecture not supported"
		354	#endif
		355
		356	# elif CORO_IRIX
		357
		358	coro_setjmp (ctx->env, 0);
		359	ctx->env[JB_PC] = (__uint64_t)coro_init;
		360	ctx->env[JB_SP] = (__uint64_t)STACK_ADJUST_PTR (sptr, ssize) - sizeof (long);
		361
		362	# elif CORO_ASM
		363
		364	ctx->sp = (void *)(ssize + (char )sptr);
		365	--ctx->sp = (void )abort; /* needed for alignment only */
		366	--ctx->sp = (void )coro_init;
		367
		368	#if CORO_WIN_TIB
		369	--ctx->sp = 0; / ExceptionList */
		370	--ctx->sp = (char )sptr + ssize; /* StackBase */
		371	--ctx->sp = sptr; / StackLimit */
		372	#endif
		373
		374	ctx->sp -= NUM_SAVED;
		375	memset (ctx->sp, 0, sizeof (ctx->sp) NUM_SAVED);
		376
		377	# elif CORO_UCONTEXT
		378
		379	getcontext (&(ctx->uc));
		380
		381	ctx->uc.uc_link = 0;
		382	ctx->uc.uc_stack.ss_sp = sptr;
		383	ctx->uc.uc_stack.ss_size = (size_t)ssize;
		384	ctx->uc.uc_stack.ss_flags = 0;
		385
		386	makecontext (&(ctx->uc), (void (*)())coro_init, 0);
		387
124	#endif	388	# endif
125		389
126	#if CORO_ASM	390	coro_transfer (create_coro, new_coro);
		391	}
127		392
128	asm (	393	/*****************************************************************************/
129	".text\n"	394	/* pthread backend */
130	".globl coro_transfer\n"	395	/*****************************************************************************/
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	396	#elif CORO_PTHREAD
170		397
171	/* this mutex will be locked by the running coroutine */	398	/* this mutex will be locked by the running coroutine */
172	pthread_mutex_t coro_mutex = PTHREAD_MUTEX_INITIALIZER;	399	pthread_mutex_t coro_mutex = PTHREAD_MUTEX_INITIALIZER;
173		400
174	struct coro_init_args	401	struct coro_init_args
…		…
186	{	413	{
187	pthread_mutex_unlock ((pthread_mutex_t *)arg);	414	pthread_mutex_unlock ((pthread_mutex_t *)arg);
188	}	415	}
189		416
190	static void *	417	static void *
191	trampoline (void *args_)	418	coro_init (void *args_)
192	{	419	{
193	struct coro_init_args args = (struct coro_init_args )args_;	420	struct coro_init_args args = (struct coro_init_args )args_;
194	coro_func func = args->func;	421	coro_func func = args->func;
195	void *arg = args->arg;	422	void *arg = args->arg;
196		423
…		…
208	void	435	void
209	coro_transfer (coro_context prev, coro_context next)	436	coro_transfer (coro_context prev, coro_context next)
210	{	437	{
211	pthread_cond_signal (&next->cv);	438	pthread_cond_signal (&next->cv);
212	pthread_cond_wait (&prev->cv, &coro_mutex);	439	pthread_cond_wait (&prev->cv, &coro_mutex);
		440	#if __FreeBSD__ /* freebsd is of course broken and needs manual testcancel calls... yay... */
		441	pthread_testcancel ();
		442	#endif
		443	}
		444
		445	void
		446	coro_create (coro_context ctx, coro_func coro, void arg, void *sptr, long ssize)
		447	{
		448	static coro_context nctx;
		449	static int once;
		450
		451	if (!once)
		452	{
		453	once = 1;
		454
		455	pthread_mutex_lock (&coro_mutex);
		456	pthread_cond_init (&nctx.cv, 0);
		457	null_tid = pthread_self ();
		458	}
		459
		460	pthread_cond_init (&ctx->cv, 0);
		461
		462	if (coro)
		463	{
		464	pthread_attr_t attr;
		465	struct coro_init_args args;
		466
		467	args.func = coro;
		468	args.arg = arg;
		469	args.self = ctx;
		470	args.main = &nctx;
		471
		472	pthread_attr_init (&attr);
		473	#if __UCLIBC__
		474	/* exists, but is borked */
		475	/pthread_attr_setstacksize (&attr, (size_t)ssize);/
		476	#elif __CYGWIN__
		477	/* POSIX, not here */
		478	pthread_attr_setstacksize (&attr, (size_t)ssize);
		479	#else
		480	pthread_attr_setstack (&attr, sptr, (size_t)ssize);
		481	#endif
		482	pthread_attr_setscope (&attr, PTHREAD_SCOPE_PROCESS);
		483	pthread_create (&ctx->id, &attr, coro_init, &args);
		484
		485	coro_transfer (args.main, args.self);
		486	}
		487	else
		488	ctx->id = null_tid;
213	}	489	}
214		490
215	void	491	void
216	coro_destroy (coro_context *ctx)	492	coro_destroy (coro_context *ctx)
217	{	493	{
…		…
224	}	500	}
225		501
226	pthread_cond_destroy (&ctx->cv);	502	pthread_cond_destroy (&ctx->cv);
227	}	503	}
228		504
229	#endif	505	/*****************************************************************************/
		506	/* fiber backend */
		507	/*****************************************************************************/
		508	#elif CORO_FIBER
230		509
231	/* initialize a machine state */	510	#define WIN32_LEAN_AND_MEAN
		511	#include <windows.h>
		512
		513	#include <stdio.h>//D
		514	VOID CALLBACK
		515	coro_init (PVOID arg)
		516	{
		517	coro_context ctx = (coro_context )arg;
		518
		519	ctx->coro (ctx->arg);
		520	}
		521
		522	void
		523	coro_transfer (coro_context prev, coro_context next)
		524	{
		525	if (!prev->fiber)
		526	{
		527	prev->fiber = GetCurrentFiber ();
		528
		529	if (prev->fiber == 0 \|\| prev->fiber == (void *)0x1e00)
		530	prev->fiber = ConvertThreadToFiber (0);
		531	}
		532
		533	SwitchToFiber (next->fiber);
		534	}
		535
232	void	536	void
233	coro_create (coro_context ctx, coro_func coro, void arg, void *sptr, long ssize)	537	coro_create (coro_context ctx, coro_func coro, void arg, void *sptr, long ssize)
234	{	538	{
235	#if CORO_UCONTEXT	539	ctx->fiber = 0;
		540	ctx->coro = coro;
		541	ctx->arg = arg;
236		542
237	if (!coro)	543	if (!coro)
238	return;	544	return;
239		545
240	getcontext (&(ctx->uc));	546	ctx->fiber = CreateFiber (ssize, coro_init, ctx);
		547	}
241		548
242	ctx->uc.uc_link = 0;	549	void
243	ctx->uc.uc_stack.ss_sp = STACK_ADJUST_PTR (sptr,ssize);	550	coro_destroy (coro_context *ctx)
244	ctx->uc.uc_stack.ss_size = (size_t)STACK_ADJUST_SIZE (sptr,ssize);	551	{
245	ctx->uc.uc_stack.ss_flags = 0;	552	DeleteFiber (ctx->fiber);
246		553	}
247	makecontext (&(ctx->uc), (void (*)())coro, 1, arg);
248
249	#elif CORO_SJLJ \|\| CORO_LOSER \|\| CORO_LINUX \|\| CORO_IRIX \|\| CORO_ASM
250
251	# if CORO_SJLJ
252	stack_t ostk, nstk;
253	struct sigaction osa, nsa;
254	sigset_t nsig, osig;
255	# endif
256	coro_context nctx;
257
258	if (!coro)
259	return;
260
261	coro_init_func = coro;
262	coro_init_arg = arg;
263
264	new_coro = ctx;
265	create_coro = &nctx;
266
267	# if CORO_SJLJ
268	/* we use SIGUSR2. first block it, then fiddle with it. */
269
270	sigemptyset (&nsig);
271	sigaddset (&nsig, SIGUSR2);
272	sigprocmask (SIG_BLOCK, &nsig, &osig);
273
274	nsa.sa_handler = trampoline;
275	sigemptyset (&nsa.sa_mask);
276	nsa.sa_flags = SA_ONSTACK;
277
278	if (sigaction (SIGUSR2, &nsa, &osa))
279	{
280	perror ("sigaction");
281	abort ();
282	}
283
284	/* set the new stack */
285	nstk.ss_sp = STACK_ADJUST_PTR (sptr,ssize); /* yes, some platforms (IRIX) get this wrong. */
286	nstk.ss_size = STACK_ADJUST_SIZE (sptr,ssize);
287	nstk.ss_flags = 0;
288
289	if (sigaltstack (&nstk, &ostk) < 0)
290	{
291	perror ("sigaltstack");
292	abort ();
293	}
294
295	trampoline_count = 0;
296	kill (getpid (), SIGUSR2);
297	sigfillset (&nsig); sigdelset (&nsig, SIGUSR2);
298
299	while (!trampoline_count)
300	sigsuspend (&nsig);
301
302	sigaltstack (0, &nstk);
303	nstk.ss_flags = SS_DISABLE;
304	if (sigaltstack (&nstk, 0) < 0)
305	perror ("sigaltstack");
306
307	sigaltstack (0, &nstk);
308	if (~nstk.ss_flags & SS_DISABLE)
309	abort ();
310
311	if (~ostk.ss_flags & SS_DISABLE)
312	sigaltstack (&ostk, 0);
313
314	sigaction (SIGUSR2, &osa, 0);
315
316	sigprocmask (SIG_SETMASK, &osig, 0);
317
318	# elif CORO_LOSER
319
320	setjmp (ctx->env);
321	#if __CYGWIN__
322	ctx->env[7] = (long)((char *)sptr + ssize) - sizeof (long);
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
336
337	# elif CORO_LINUX
338
339	_setjmp (ctx->env);
340	#if __GLIBC__ >= 2 && __GLIBC_MINOR__ >= 0 && defined (JB_PC) && defined (JB_SP)
341	ctx->env[0].__jmpbuf[JB_PC] = (long)coro_init;
342	ctx->env[0].__jmpbuf[JB_SP] = (long)STACK_ADJUST_PTR (sptr, ssize) - sizeof (long);
343	#elif __GLIBC__ >= 2 && __GLIBC_MINOR__ >= 0 && defined (__mc68000__)
344	ctx->env[0].__jmpbuf[0].__aregs[0] = (long int)coro_init;
345	ctx->env[0].__jmpbuf[0].__sp = (int )((char )sptr + ssize) - sizeof (long);
346	#elif defined (__GNU_LIBRARY__) && defined (__i386__)
347	ctx->env[0].__jmpbuf[0].__pc = (char *)coro_init;
348	ctx->env[0].__jmpbuf[0].__sp = (void )((char )sptr + ssize) - sizeof (long);
349	#elif defined (__GNU_LIBRARY__) && defined (__amd64__)
350	ctx->env[0].__jmpbuf[JB_PC] = (long)coro_init;
351	ctx->env[0].__jmpbuf[0].__sp = (void )((char )sptr + ssize) - sizeof (long);
352	#else
353	# error "linux libc or architecture not supported"
354	#endif
355
356	# elif CORO_IRIX
357
358	setjmp (ctx->env);
359	ctx->env[JB_PC] = (__uint64_t)coro_init;
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;
368
369	# endif
370
371	coro_transfer ((coro_context )create_coro, (coro_context )new_coro);
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		554
409	#else	555	#else
410	# error unsupported backend	556	# error unsupported backend
411	#endif	557	#endif
412	}
413		558

Diff Legend

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

Comparing cvsroot/libcoro/coro.c (file contents): Revision 1.38 by root, Fri Nov 7 20:12:26 2008 UTC vs. Revision 1.63 by root, Wed Dec 5 01:09:12 2012 UTC

Diff Legend

Comparing cvsroot/libcoro/coro.c (file contents):
Revision 1.38 by root, Fri Nov 7 20:12:26 2008 UTC vs.
Revision 1.63 by root, Wed Dec 5 01:09:12 2012 UTC