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, |
… | |
… | |
47 | # define STACK_ADJUST_PTR(sp,ss) (sp) |
47 | # define STACK_ADJUST_PTR(sp,ss) (sp) |
48 | # define STACK_ADJUST_SIZE(sp,ss) (ss) |
48 | # define STACK_ADJUST_SIZE(sp,ss) (ss) |
49 | # endif |
49 | # endif |
50 | #endif |
50 | #endif |
51 | |
51 | |
|
|
52 | #if CORO_UCONTEXT |
|
|
53 | # include <stddef.h> |
|
|
54 | #endif |
|
|
55 | |
52 | #if CORO_SJLJ || CORO_LOSER || CORO_LINUX || CORO_IRIX |
56 | #if CORO_SJLJ || CORO_LOSER || CORO_LINUX || CORO_IRIX |
53 | |
57 | |
54 | #include <stdlib.h> |
58 | #include <stdlib.h> |
55 | |
59 | |
56 | #if CORO_SJLJ |
60 | #if CORO_SJLJ |
… | |
… | |
60 | #endif |
64 | #endif |
61 | |
65 | |
62 | static volatile coro_func coro_init_func; |
66 | static volatile coro_func coro_init_func; |
63 | static volatile void *coro_init_arg; |
67 | static volatile void *coro_init_arg; |
64 | static volatile coro_context *new_coro, *create_coro; |
68 | static 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 |
65 | |
75 | |
66 | static void |
76 | static void |
67 | coro_init (void) |
77 | coro_init (void) |
68 | { |
78 | { |
69 | volatile coro_func func = coro_init_func; |
79 | volatile coro_func func = coro_init_func; |
70 | volatile void *arg = coro_init_arg; |
80 | volatile void *arg = coro_init_arg; |
71 | |
81 | |
72 | coro_transfer ((coro_context *)new_coro, (coro_context *)create_coro); |
82 | coro_transfer ((coro_context *)new_coro, (coro_context *)create_coro); |
73 | |
83 | |
|
|
84 | #if HAVE_CFI |
|
|
85 | asm (".cfi_startproc"); |
|
|
86 | #endif |
74 | func ((void *)arg); |
87 | func ((void *)arg); |
|
|
88 | #if HAVE_CFI |
|
|
89 | asm (".cfi_endproc"); |
|
|
90 | #endif |
75 | |
91 | |
76 | /* the new coro returned. bad. just abort() for now */ |
92 | /* the new coro returned. bad. just abort() for now */ |
77 | abort (); |
93 | abort (); |
78 | } |
94 | } |
79 | |
95 | |
… | |
… | |
94 | # endif |
110 | # endif |
95 | |
111 | |
96 | #endif |
112 | #endif |
97 | |
113 | |
98 | /* initialize a machine state */ |
114 | /* initialize a machine state */ |
99 | void coro_create(coro_context *ctx, |
115 | void coro_create (coro_context *ctx, |
100 | coro_func coro, void *arg, |
116 | coro_func coro, void *arg, |
101 | void *sptr, long ssize) |
117 | void *sptr, long ssize) |
102 | { |
118 | { |
103 | #if CORO_UCONTEXT |
119 | #if CORO_UCONTEXT |
104 | |
120 | |
105 | getcontext (&(ctx->uc)); |
121 | getcontext (&(ctx->uc)); |
106 | |
122 | |
… | |
… | |
136 | nsa.sa_handler = trampoline; |
152 | nsa.sa_handler = trampoline; |
137 | sigemptyset (&nsa.sa_mask); |
153 | sigemptyset (&nsa.sa_mask); |
138 | nsa.sa_flags = SA_ONSTACK; |
154 | nsa.sa_flags = SA_ONSTACK; |
139 | |
155 | |
140 | if (sigaction (SIGUSR2, &nsa, &osa)) |
156 | if (sigaction (SIGUSR2, &nsa, &osa)) |
|
|
157 | { |
141 | perror ("sigaction"); |
158 | perror ("sigaction"); |
|
|
159 | abort (); |
|
|
160 | } |
142 | |
161 | |
143 | /* set the new stack */ |
162 | /* set the new stack */ |
144 | nstk.ss_sp = STACK_ADJUST_PTR (sptr,ssize); /* yes, some platforms (IRIX) get this wrong. */ |
163 | nstk.ss_sp = STACK_ADJUST_PTR (sptr,ssize); /* yes, some platforms (IRIX) get this wrong. */ |
145 | nstk.ss_size = STACK_ADJUST_SIZE (sptr,ssize); |
164 | nstk.ss_size = STACK_ADJUST_SIZE (sptr,ssize); |
146 | nstk.ss_flags = 0; |
165 | nstk.ss_flags = 0; |
147 | |
166 | |
148 | if (sigaltstack (&nstk, &ostk) < 0) |
167 | if (sigaltstack (&nstk, &ostk) < 0) |
|
|
168 | { |
149 | perror ("sigaltstack"); |
169 | perror ("sigaltstack"); |
|
|
170 | abort (); |
|
|
171 | } |
150 | |
172 | |
151 | trampoline_count = 0; |
173 | trampoline_count = 0; |
152 | kill (getpid (), SIGUSR2); |
174 | kill (getpid (), SIGUSR2); |
153 | sigfillset (&nsig); sigdelset (&nsig, SIGUSR2); |
175 | sigfillset (&nsig); sigdelset (&nsig, SIGUSR2); |
154 | |
176 | |
… | |
… | |
165 | abort (); |
187 | abort (); |
166 | |
188 | |
167 | if (~ostk.ss_flags & SS_DISABLE) |
189 | if (~ostk.ss_flags & SS_DISABLE) |
168 | sigaltstack (&ostk, 0); |
190 | sigaltstack (&ostk, 0); |
169 | |
191 | |
170 | sigaction (SIGUSR1, &osa, 0); |
192 | sigaction (SIGUSR2, &osa, 0); |
171 | |
193 | |
172 | sigprocmask (SIG_SETMASK, &osig, 0); |
194 | sigprocmask (SIG_SETMASK, &osig, 0); |
173 | |
195 | |
174 | # elif CORO_LOSER |
196 | # elif CORO_LOSER |
175 | |
197 | |
176 | setjmp (ctx->env); |
198 | setjmp (ctx->env); |
|
|
199 | #if __CYGWIN__ |
177 | ctx->env[7] = (long)((char *)sptr + ssize); |
200 | ctx->env[7] = (long)((char *)sptr + ssize); |
178 | ctx->env[8] = (long)coro_init; |
201 | ctx->env[8] = (long)coro_init; |
|
|
202 | #elif defined(_M_IX86) |
|
|
203 | ((_JUMP_BUFFER *)&ctx->env)->Eip = (long)coro_init; |
|
|
204 | ((_JUMP_BUFFER *)&ctx->env)->Esp = (long)STACK_ADJUST_PTR (sptr,ssize); |
|
|
205 | #elif defined(_M_AMD64) |
|
|
206 | ((_JUMP_BUFFER *)&ctx->env)->Rip = (__int64)coro_init; |
|
|
207 | ((_JUMP_BUFFER *)&ctx->env)->Rsp = (__int64)STACK_ADJUST_PTR (sptr,ssize); |
|
|
208 | #elif defined(_M_IA64) |
|
|
209 | ((_JUMP_BUFFER *)&ctx->env)->StIIP = (__int64)coro_init; |
|
|
210 | ((_JUMP_BUFFER *)&ctx->env)->IntSp = (__int64)STACK_ADJUST_PTR (sptr,ssize); |
|
|
211 | #else |
|
|
212 | #error "microsoft libc or architecture not supported" |
|
|
213 | #endif |
179 | |
214 | |
180 | # elif CORO_LINUX |
215 | # elif CORO_LINUX |
181 | |
216 | |
182 | setjmp (ctx->env); |
217 | _setjmp (ctx->env); |
183 | #if defined(__GLIBC__) && defined(__GLIBC_MINOR__) \ |
|
|
184 | && __GLIBC__ >= 2 && __GLIBC_MINOR__ >= 0 && defined(JB_PC) && defined(JB_SP) |
218 | #if __GLIBC__ >= 2 && __GLIBC_MINOR__ >= 0 && defined (JB_PC) && defined (JB_SP) |
185 | ctx->env[0].__jmpbuf[JB_PC] = (long)coro_init; |
219 | ctx->env[0].__jmpbuf[JB_PC] = (long)coro_init; |
186 | ctx->env[0].__jmpbuf[JB_SP] = (long)STACK_ADJUST_PTR (sptr,ssize); |
220 | ctx->env[0].__jmpbuf[JB_SP] = (long)STACK_ADJUST_PTR (sptr, ssize); |
187 | #elif defined(__GLIBC__) && defined(__GLIBC_MINOR__) \ |
|
|
188 | && __GLIBC__ >= 2 && __GLIBC_MINOR__ >= 0 && defined(__mc68000__) |
221 | #elif __GLIBC__ >= 2 && __GLIBC_MINOR__ >= 0 && defined (__mc68000__) |
189 | ctx->env[0].__jmpbuf[0].__aregs[0] = (long int)coro_init; |
222 | ctx->env[0].__jmpbuf[0].__aregs[0] = (long int)coro_init; |
190 | ctx->env[0].__jmpbuf[0].__sp = (int *)((char *)sptr + ssize); |
223 | ctx->env[0].__jmpbuf[0].__sp = (int *)((char *)sptr + ssize); |
191 | #elif defined(__GNU_LIBRARY__) && defined(__i386__) |
224 | #elif defined (__GNU_LIBRARY__) && defined (__i386__) |
192 | ctx->env[0].__jmpbuf[0].__pc = (char *)coro_init; |
225 | ctx->env[0].__jmpbuf[0].__pc = (char *)coro_init; |
193 | ctx->env[0].__jmpbuf[0].__sp = (void *)((char *)sptr + ssize); |
226 | ctx->env[0].__jmpbuf[0].__sp = (void *)((char *)sptr + ssize); |
194 | #elif defined(__GNU_LIBRARY__) && defined(__amd64__) |
227 | #elif defined (__GNU_LIBRARY__) && defined (__amd64__) |
195 | ctx->env[0].__jmpbuf[JB_PC] = (long)coro_init; |
228 | ctx->env[0].__jmpbuf[JB_PC] = (long)coro_init; |
196 | ctx->env[0].__jmpbuf[JB_RSP] = (long)STACK_ADJUST_PTR (sptr,ssize); |
229 | ctx->env[0].__jmpbuf[JB_RSP] = (long)STACK_ADJUST_PTR (sptr, ssize); |
197 | #else |
230 | #else |
198 | #error "linux libc or architecture not supported" |
231 | #error "linux libc or architecture not supported" |
199 | #endif |
232 | #endif |
200 | |
233 | |
201 | # elif CORO_IRIX |
234 | # elif CORO_IRIX |
202 | |
235 | |
203 | setjmp (ctx->env); |
236 | setjmp (ctx->env); |
204 | ctx->env[JB_PC] = (__uint64_t)coro_init; |
237 | ctx->env[JB_PC] = (__uint64_t)coro_init; |
205 | ctx->env[JB_SP] = (__uint64_t)STACK_ADJUST_PTR (sptr,ssize); |
238 | ctx->env[JB_SP] = (__uint64_t)STACK_ADJUST_PTR (sptr, ssize); |
206 | |
239 | |
207 | # endif |
240 | # endif |
208 | |
241 | |
209 | coro_transfer ((coro_context *)create_coro, (coro_context *)new_coro); |
242 | coro_transfer ((coro_context *)create_coro, (coro_context *)new_coro); |
210 | |
243 | |