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

Comparing libcoro/coro.c (file contents):
Revision 1.15 by root, Mon Mar 21 14:35:22 2005 UTC vs.
Revision 1.26 by root, Fri Apr 27 19:35:28 2007 UTC

1/* 1/*
2 * Copyright (c) 2001-2005 Marc Alexander Lehmann <schmorp@schmorp.de> 2 * Copyright (c) 2001-2006 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,
38# define STACK_ADJUST_PTR(sp,ss) ((char *)(sp) + (ss) - 8) 38# define STACK_ADJUST_PTR(sp,ss) ((char *)(sp) + (ss) - 8)
39# define STACK_ADJUST_SIZE(sp,ss) ((ss) - 8) 39# define STACK_ADJUST_SIZE(sp,ss) ((ss) - 8)
40# elif __i386__ && CORO_LINUX 40# elif __i386__ && CORO_LINUX
41# define STACK_ADJUST_PTR(sp,ss) ((char *)(sp) + (ss)) 41# define STACK_ADJUST_PTR(sp,ss) ((char *)(sp) + (ss))
42# define STACK_ADJUST_SIZE(sp,ss) (ss) 42# define STACK_ADJUST_SIZE(sp,ss) (ss)
43# elif __amd64__ && CORO_LINUX
44# define STACK_ADJUST_PTR(sp,ss) ((char *)(sp) + (ss) - 8)
45# define STACK_ADJUST_SIZE(sp,ss) (ss)
43# else 46# else
44# define STACK_ADJUST_PTR(sp,ss) (sp) 47# define STACK_ADJUST_PTR(sp,ss) (sp)
45# define STACK_ADJUST_SIZE(sp,ss) (ss) 48# define STACK_ADJUST_SIZE(sp,ss) (ss)
46# endif 49# endif
47#endif 50#endif
48 51
52#if CORO_UCONTEXT
53# include <stddef.h>
54#endif
55
49#if CORO_SJLJ || CORO_LOOSE || CORO_LINUX || CORO_IRIX 56#if CORO_SJLJ || CORO_LOSER || CORO_LINUX || CORO_IRIX
57
58#include <stdlib.h>
50 59
51#if CORO_SJLJ 60#if CORO_SJLJ
52# include <stdio.h> 61# include <stdio.h>
53# include <signal.h> 62# include <signal.h>
63# include <unistd.h>
54#endif 64#endif
55 65
56static volatile coro_func coro_init_func; 66static volatile coro_func coro_init_func;
57static volatile void *coro_init_arg; 67static volatile void *coro_init_arg;
58static volatile coro_context *new_coro, *create_coro; 68static volatile coro_context *new_coro, *create_coro;
69
70/* what we really want to detect here is wether we use a new-enough version of GAS */
71/* instead, check for gcc 3 and ELF and hope for the best */
72#if __GNUC__ >= 3 && __ELF__
73# define HAVE_CFI 1
74#endif
59 75
60static void 76static void
61coro_init (void) 77coro_init (void)
62{ 78{
63 volatile coro_func func = coro_init_func; 79 volatile coro_func func = coro_init_func;
78/* trampoline signal handler */ 94/* trampoline signal handler */
79static void 95static void
80trampoline (int sig) 96trampoline (int sig)
81{ 97{
82 if (setjmp (((coro_context *)new_coro)->env)) 98 if (setjmp (((coro_context *)new_coro)->env))
99 {
100#if HAVE_CFI
101 asm (".cfi_startproc");
102#endif
83 coro_init (); /* start it */ 103 coro_init (); /* start it */
104#if HAVE_CFI
105 asm (".cfi_endproc");
106#endif
107 }
84 else 108 else
85 trampoline_count++; 109 trampoline_count++;
86} 110}
87 111
88# endif 112# endif
89 113
90#endif 114#endif
91 115
92/* initialize a machine state */ 116/* initialize a machine state */
93void coro_create(coro_context *ctx, 117void coro_create (coro_context *ctx,
94 coro_func coro, void *arg, 118 coro_func coro, void *arg,
95 void *sptr, long ssize) 119 void *sptr, long ssize)
96{ 120{
97#if CORO_UCONTEXT 121#if CORO_UCONTEXT
98 122
99 getcontext (&(ctx->uc)); 123 getcontext (&(ctx->uc));
100 124
103 ctx->uc.uc_stack.ss_size = (size_t) STACK_ADJUST_SIZE (sptr,ssize); 127 ctx->uc.uc_stack.ss_size = (size_t) STACK_ADJUST_SIZE (sptr,ssize);
104 ctx->uc.uc_stack.ss_flags = 0; 128 ctx->uc.uc_stack.ss_flags = 0;
105 129
106 makecontext (&(ctx->uc), (void (*)()) coro, 1, arg); 130 makecontext (&(ctx->uc), (void (*)()) coro, 1, arg);
107 131
108#elif CORO_SJLJ || CORO_LOOSE || CORO_LINUX || CORO_IRIX 132#elif CORO_SJLJ || CORO_LOSER || CORO_LINUX || CORO_IRIX
109 133
110# if CORO_SJLJ 134# if CORO_SJLJ
111 stack_t ostk, nstk; 135 stack_t ostk, nstk;
112 struct sigaction osa, nsa; 136 struct sigaction osa, nsa;
113 sigset_t nsig, osig; 137 sigset_t nsig, osig;
130 nsa.sa_handler = trampoline; 154 nsa.sa_handler = trampoline;
131 sigemptyset (&nsa.sa_mask); 155 sigemptyset (&nsa.sa_mask);
132 nsa.sa_flags = SA_ONSTACK; 156 nsa.sa_flags = SA_ONSTACK;
133 157
134 if (sigaction (SIGUSR2, &nsa, &osa)) 158 if (sigaction (SIGUSR2, &nsa, &osa))
159 {
135 perror ("sigaction"); 160 perror ("sigaction");
161 abort ();
162 }
136 163
137 /* set the new stack */ 164 /* set the new stack */
138 nstk.ss_sp = STACK_ADJUST_PTR (sptr,ssize); /* yes, some platforms (IRIX) get this wrong. */ 165 nstk.ss_sp = STACK_ADJUST_PTR (sptr,ssize); /* yes, some platforms (IRIX) get this wrong. */
139 nstk.ss_size = STACK_ADJUST_SIZE (sptr,ssize); 166 nstk.ss_size = STACK_ADJUST_SIZE (sptr,ssize);
140 nstk.ss_flags = 0; 167 nstk.ss_flags = 0;
141 168
142 if (sigaltstack (&nstk, &ostk) < 0) 169 if (sigaltstack (&nstk, &ostk) < 0)
170 {
143 perror ("sigaltstack"); 171 perror ("sigaltstack");
172 abort ();
173 }
144 174
145 trampoline_count = 0; 175 trampoline_count = 0;
146 kill (getpid (), SIGUSR2); 176 kill (getpid (), SIGUSR2);
147 sigfillset (&nsig); sigdelset (&nsig, SIGUSR2); 177 sigfillset (&nsig); sigdelset (&nsig, SIGUSR2);
148 178
159 abort (); 189 abort ();
160 190
161 if (~ostk.ss_flags & SS_DISABLE) 191 if (~ostk.ss_flags & SS_DISABLE)
162 sigaltstack (&ostk, 0); 192 sigaltstack (&ostk, 0);
163 193
164 sigaction (SIGUSR1, &osa, 0); 194 sigaction (SIGUSR2, &osa, 0);
165 195
166 sigprocmask (SIG_SETMASK, &osig, 0); 196 sigprocmask (SIG_SETMASK, &osig, 0);
167 197
168# elif CORO_LOOSE 198# elif CORO_LOSER
169 199
170 setjmp (ctx->env); 200 setjmp (ctx->env);
201#if __CYGWIN__
171 ctx->env[7] = (long)((char *)sptr + ssize); 202 ctx->env[7] = (long)((char *)sptr + ssize);
172 ctx->env[8] = (long)coro_init; 203 ctx->env[8] = (long)coro_init;
204#elif defined(_M_IX86)
205 ((_JUMP_BUFFER *)&ctx->env)->Eip = (long)coro_init;
206 ((_JUMP_BUFFER *)&ctx->env)->Esp = (long)STACK_ADJUST_PTR (sptr,ssize);
207#elif defined(_M_AMD64)
208 ((_JUMP_BUFFER *)&ctx->env)->Rip = (__int64)coro_init;
209 ((_JUMP_BUFFER *)&ctx->env)->Rsp = (__int64)STACK_ADJUST_PTR (sptr,ssize);
210#elif defined(_M_IA64)
211 ((_JUMP_BUFFER *)&ctx->env)->StIIP = (__int64)coro_init;
212 ((_JUMP_BUFFER *)&ctx->env)->IntSp = (__int64)STACK_ADJUST_PTR (sptr,ssize);
213#else
214#error "microsoft libc or architecture not supported"
215#endif
173 216
174# elif CORO_LINUX 217# elif CORO_LINUX
175 218
176 setjmp (ctx->env); 219 _setjmp (ctx->env);
177#if defined(__GLIBC__) && defined(__GLIBC_MINOR__) \
178 && __GLIBC__ >= 2 && __GLIBC_MINOR__ >= 0 && defined(JB_PC) && defined(JB_SP) 220#if __GLIBC__ >= 2 && __GLIBC_MINOR__ >= 0 && defined (JB_PC) && defined (JB_SP)
179 ctx->env[0].__jmpbuf[JB_PC] = (long)coro_init; 221 ctx->env[0].__jmpbuf[JB_PC] = (long)coro_init;
180 ctx->env[0].__jmpbuf[JB_SP] = (long)STACK_ADJUST_PTR (sptr,ssize); 222 ctx->env[0].__jmpbuf[JB_SP] = (long)STACK_ADJUST_PTR (sptr, ssize);
181#elif defined(__GLIBC__) && defined(__GLIBC_MINOR__) \
182 && __GLIBC__ >= 2 && __GLIBC_MINOR__ >= 0 && defined(__mc68000__) 223#elif __GLIBC__ >= 2 && __GLIBC_MINOR__ >= 0 && defined (__mc68000__)
183 ctx->env[0].__jmpbuf[0].__aregs[0] = (long int)coro_init; 224 ctx->env[0].__jmpbuf[0].__aregs[0] = (long int)coro_init;
184 ctx->env[0].__jmpbuf[0].__sp = (int *)((char *)sptr + ssize); 225 ctx->env[0].__jmpbuf[0].__sp = (int *)((char *)sptr + ssize);
185#elif defined(__GNU_LIBRARY__) && defined(__i386__) 226#elif defined (__GNU_LIBRARY__) && defined (__i386__)
186 ctx->env[0].__jmpbuf[0].__pc = (char *)coro_init; 227 ctx->env[0].__jmpbuf[0].__pc = (char *)coro_init;
187 ctx->env[0].__jmpbuf[0].__sp = (void *)((char *)sptr + ssize); 228 ctx->env[0].__jmpbuf[0].__sp = (void *)((char *)sptr + ssize);
188#elif defined(__GNU_LIBRARY__) && defined(__amd64__) 229#elif defined (__GNU_LIBRARY__) && defined (__amd64__)
189 ctx->env[0].__jmpbuf[JB_PC] = (long)coro_init; 230 ctx->env[0].__jmpbuf[JB_PC] = (long)coro_init;
190 ctx->env[0].__jmpbuf[JB_RSP] = (long)((char *)sptr + ssize); 231 ctx->env[0].__jmpbuf[JB_RSP] = (long)STACK_ADJUST_PTR (sptr, ssize);
191#else 232#else
192#error "linux libc or architecture not supported" 233#error "linux libc or architecture not supported"
193#endif 234#endif
194 235
195# elif CORO_IRIX 236# elif CORO_IRIX
196 237
197 setjmp (ctx->env); 238 setjmp (ctx->env);
198 ctx->env[JB_PC] = (__uint64_t)coro_init; 239 ctx->env[JB_PC] = (__uint64_t)coro_init;
199 ctx->env[JB_SP] = (__uint64_t)STACK_ADJUST_PTR (sptr,ssize); 240 ctx->env[JB_SP] = (__uint64_t)STACK_ADJUST_PTR (sptr, ssize);
200 241
201# endif 242# endif
202 243
203 coro_transfer ((coro_context *)create_coro, (coro_context *)new_coro); 244 coro_transfer ((coro_context *)create_coro, (coro_context *)new_coro);
204 245

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines