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