… | |
… | |
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 CORO_LOOSE || CORO_SJLJ |
35 | #if !defined(STACK_ADJUST_PTR) |
|
|
36 | /* IRIX is decidedly NON-unix */ |
|
|
37 | # if __sgi |
|
|
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 | # else |
|
|
44 | # define STACK_ADJUST_PTR(sp,ss) (sp) |
|
|
45 | # define STACK_ADJUST_SIZE(sp,ss) (ss) |
|
|
46 | # endif |
|
|
47 | #endif |
|
|
48 | |
|
|
49 | #if CORO_SJLJ || CORO_LOOSE || CORO_LINUX || CORO_IRIX |
36 | |
50 | |
37 | #include <signal.h> |
51 | #include <signal.h> |
38 | |
52 | |
39 | static volatile coro_func coro_init_func; |
53 | static volatile coro_func coro_init_func; |
40 | static volatile void *coro_init_arg; |
54 | static volatile void *coro_init_arg; |
… | |
… | |
44 | coro_init (void) |
58 | coro_init (void) |
45 | { |
59 | { |
46 | volatile coro_func func = coro_init_func; |
60 | volatile coro_func func = coro_init_func; |
47 | volatile void *arg = coro_init_arg; |
61 | volatile void *arg = coro_init_arg; |
48 | |
62 | |
49 | coro_transfer (new_coro, create_coro); |
63 | coro_transfer ((coro_context *)new_coro, (coro_context *)create_coro); |
50 | |
64 | |
51 | func (arg); |
65 | func ((void *)arg); |
52 | |
66 | |
53 | /* the new coro returned. bad. just abort() for now */ |
67 | /* the new coro returned. bad. just abort() for now */ |
54 | abort (); |
68 | abort (); |
55 | } |
69 | } |
56 | |
70 | |
57 | # if CORO_SJLJ |
71 | # if CORO_SJLJ |
58 | # define coro_save(ctx) (void)setjmp((ctx)->env) |
|
|
59 | |
72 | |
60 | static volatile int trampoline_count; |
73 | static volatile int trampoline_count; |
61 | |
74 | |
62 | /* trampoline signal handler */ |
75 | /* trampoline signal handler */ |
63 | static void |
76 | static void |
64 | trampoline(int sig) |
77 | trampoline(int sig) |
65 | { |
78 | { |
66 | if (setjmp (&(new_coro->env))) |
79 | if (setjmp (((coro_context *)new_coro)->env)) |
67 | coro_init (); /* start it */ |
80 | coro_init (); /* start it */ |
68 | else |
81 | else |
69 | trampoline_count++; |
82 | trampoline_count++; |
70 | } |
83 | } |
71 | |
84 | |
72 | # endif |
85 | # endif |
73 | |
86 | |
74 | #elif CORO_UCONTEXT |
|
|
75 | # define coro_save(ctx) getcontext(&((ctx)->uc)) |
|
|
76 | #endif |
87 | #endif |
77 | |
88 | |
78 | /* initialize a machine state */ |
89 | /* initialize a machine state */ |
79 | void coro_create(coro_context *ctx, |
90 | void coro_create(coro_context *ctx, |
80 | coro_func coro, void *arg, |
91 | coro_func coro, void *arg, |
… | |
… | |
82 | { |
93 | { |
83 | #if CORO_UCONTEXT |
94 | #if CORO_UCONTEXT |
84 | |
95 | |
85 | getcontext (&(ctx->uc)); |
96 | getcontext (&(ctx->uc)); |
86 | |
97 | |
87 | ctx->uc.uc_link = 0; |
98 | ctx->uc.uc_link = 0; |
88 | ctx->uc.uc_stack.ss_sp = sptr; |
99 | ctx->uc.uc_stack.ss_sp = STACK_ADJUST_PTR (sptr,ssize); |
89 | ctx->uc.uc_stack.ss_size = (size_t) ssize; |
100 | ctx->uc.uc_stack.ss_size = (size_t) STACK_ADJUST_SIZE (sptr,ssize); |
90 | ctx->uc.uc_stack.ss_flags = 0; |
101 | ctx->uc.uc_stack.ss_flags = 0; |
91 | |
102 | |
92 | makecontext (&(ctx->uc), (void (*)()) coro, 1, arg); |
103 | makecontext (&(ctx->uc), (void (*)()) coro, 1, arg); |
93 | |
104 | |
94 | #elif CORO_SJLJ || CORO_LOOSE |
105 | #elif CORO_SJLJ || CORO_LOOSE || CORO_LINUX || CORO_IRIX |
95 | |
106 | |
96 | # if CORO_SJLJ |
107 | # if CORO_SJLJ |
97 | stack_t ostk, nstk; |
108 | stack_t ostk, nstk; |
98 | struct sigaction osa, nsa; |
109 | struct sigaction osa, nsa; |
99 | sigset_t nsig, osig; |
110 | sigset_t nsig, osig; |
… | |
… | |
119 | |
130 | |
120 | if (sigaction (SIGUSR2, &nsa, &osa)) |
131 | if (sigaction (SIGUSR2, &nsa, &osa)) |
121 | perror ("sigaction"); |
132 | perror ("sigaction"); |
122 | |
133 | |
123 | /* set the new stack */ |
134 | /* set the new stack */ |
124 | nstk.ss_sp = sptr; |
135 | nstk.ss_sp = STACK_ADJUST_PTR (sptr,ssize); /* yes, some platforms (IRIX) get this wrong. */ |
125 | nstk.ss_size = ssize; |
136 | nstk.ss_size = STACK_ADJUST_SIZE (sptr,ssize); |
126 | nstk.ss_flags = 0; |
137 | nstk.ss_flags = 0; |
127 | |
138 | |
128 | if (sigaltstack (&nstk, &ostk) < 0) |
139 | if (sigaltstack (&nstk, &ostk) < 0) |
129 | perror ("sigaltstack"); |
140 | perror ("sigaltstack"); |
130 | |
141 | |
… | |
… | |
149 | |
160 | |
150 | sigaction (SIGUSR1, &osa, 0); |
161 | sigaction (SIGUSR1, &osa, 0); |
151 | |
162 | |
152 | sigprocmask (SIG_SETMASK, &osig, 0); |
163 | sigprocmask (SIG_SETMASK, &osig, 0); |
153 | |
164 | |
154 | # elif MS_LOOSE |
165 | # elif CORO_LOOSE |
155 | |
166 | |
156 | coro_save (ctx); |
167 | setjmp (ctx->env); |
157 | ctx->env[7] = (int)((char *)sptr + ssize); |
168 | ctx->env[7] = (int)((char *)sptr + ssize); |
158 | ctx->env[8] = (int)coro; |
169 | ctx->env[8] = (int)coro_init; |
159 | |
170 | |
|
|
171 | # elif CORO_LINUX |
|
|
172 | |
|
|
173 | setjmp (ctx->env); |
|
|
174 | #if defined(__GLIBC__) && defined(__GLIBC_MINOR__) \ |
|
|
175 | && __GLIBC__ >= 2 && __GLIBC_MINOR__ >= 0 && defined(JB_PC) && defined(JB_SP) |
|
|
176 | ctx->env[0].__jmpbuf[JB_PC] = (long)coro_init; |
|
|
177 | ctx->env[0].__jmpbuf[JB_SP] = (long)STACK_ADJUST_PTR (sptr,ssize); |
|
|
178 | #elif defined(__GLIBC__) && defined(__GLIBC_MINOR__) \ |
|
|
179 | && __GLIBC__ >= 2 && __GLIBC_MINOR__ >= 0 && defined(__mc68000__) |
|
|
180 | ctx->env[0].__jmpbuf[0].__aregs[0] = (long int)coro_init; |
|
|
181 | ctx->env[0].__jmpbuf[0].__sp = (int *)((char *)sptr + ssize); |
|
|
182 | #elif defined(__GNU_LIBRARY__) && defined(__i386__) |
|
|
183 | ctx->env[0].__jmpbuf[0].__pc = (char *)coro_init; |
|
|
184 | ctx->env[0].__jmpbuf[0].__sp = (void *)((char *)sptr + ssize); |
|
|
185 | #else |
|
|
186 | #error "linux libc or architecture not supported" |
160 | # endif |
187 | #endif |
161 | |
188 | |
162 | coro_transfer (create_coro, new_coro); |
189 | # elif CORO_IRIX |
|
|
190 | |
|
|
191 | setjmp (ctx->env); |
|
|
192 | ctx->env[JB_PC] = (__uint64_t)coro_init; |
|
|
193 | ctx->env[JB_SP] = (__uint64_t)STACK_ADJUST_PTR (sptr,ssize); |
|
|
194 | |
|
|
195 | # endif |
|
|
196 | |
|
|
197 | coro_transfer ((coro_context *)create_coro, (coro_context *)new_coro); |
163 | |
198 | |
164 | #else |
199 | #else |
165 | error unsupported architecture |
200 | error unsupported architecture |
166 | #endif |
201 | #endif |
167 | } |
202 | } |