Allocing a TLS in runtime is not too hot.
[p5sagit/p5-mst-13.2.git] / ext / threads / threads.xs
CommitLineData
47ba8780 1#include "threads.h"
2
47ba8780 3/*
b1edfb69 4 * Starts executing the thread. Needs to clean up memory a tad better.
5 */
47ba8780 6
7#ifdef WIN32
e6e315b9 8THREAD_RET_TYPE Perl_thread_run(LPVOID arg) {
47ba8780 9#else
e8f2bb9a 10void* Perl_thread_run(void * arg) {
47ba8780 11#endif
5b414d21 12 ithread* thread = (ithread*) arg;
47ba8780 13 SV* thread_tid_ptr;
14 SV* thread_ptr;
15 dTHXa(thread->interp);
47ba8780 16 PERL_SET_CONTEXT(thread->interp);
17
18#ifdef WIN32
19 thread->thr = GetCurrentThreadId();
20#else
21 thread->thr = pthread_self();
22#endif
23
4f896ddc 24 SHAREDSvLOCK(threads);
47ba8780 25 SHAREDSvEDIT(threads);
7f9a95ef 26 PERL_THREAD_SETSPECIFIC(self_key,INT2PTR(void*,thread->tid));
27 thread_tid_ptr = Perl_newSVuv(PL_sharedsv_space, thread->tid);
b1edfb69 28 thread_ptr = Perl_newSVuv(PL_sharedsv_space, PTR2UV(thread));
47ba8780 29 hv_store_ent((HV*)SHAREDSvGET(threads), thread_tid_ptr, thread_ptr,0);
30 SvREFCNT_dec(thread_tid_ptr);
31 SHAREDSvRELEASE(threads);
4f896ddc 32 SHAREDSvUNLOCK(threads);
47ba8780 33 PL_perl_destruct_level = 2;
4f896ddc 34
47ba8780 35 {
36
37 AV* params;
38 I32 len;
39 int i;
40 dSP;
41 params = (AV*) SvRV(thread->params);
42 len = av_len(params);
43 ENTER;
44 SAVETMPS;
45 PUSHMARK(SP);
46 if(len > -1) {
47 for(i = 0; i < len + 1; i++) {
48 XPUSHs(av_shift(params));
49 }
50 }
51 PUTBACK;
4f896ddc 52 call_sv(thread->init_function, G_DISCARD);
47ba8780 53 FREETMPS;
54 LEAVE;
55
56
57 }
58
47ba8780 59 MUTEX_LOCK(&thread->mutex);
fd58862f 60 PerlIO_flush((PerlIO*)NULL);
47ba8780 61 perl_destruct(thread->interp);
62 perl_free(thread->interp);
63 if(thread->detached == 1) {
64 MUTEX_UNLOCK(&thread->mutex);
e6e315b9 65 Perl_thread_destruct(thread);
47ba8780 66 } else {
67 MUTEX_UNLOCK(&thread->mutex);
68 }
69#ifdef WIN32
70 return (DWORD)0;
e8f2bb9a 71#else
72 return 0;
47ba8780 73#endif
74
75}
76
47ba8780 77/*
b1edfb69 78 * iThread->create();
79 */
47ba8780 80
e6e315b9 81SV* Perl_thread_create(char* class, SV* init_function, SV* params) {
47ba8780 82 ithread* thread = malloc(sizeof(ithread));
83 SV* obj_ref;
84 SV* obj;
85 SV* temp_store;
47ba8780 86 PerlInterpreter *current_perl;
87
88 MUTEX_LOCK(&create_mutex);
89 obj_ref = newSViv(0);
90 obj = newSVrv(obj_ref, class);
b1edfb69 91 sv_setiv(obj, (IV)thread);
92 SvREADONLY_on(obj);
b6a0b930 93 PerlIO_flush((PerlIO*)NULL);
b1edfb69 94 current_perl = PERL_GET_CONTEXT;
47ba8780 95
96 /*
b1edfb69 97 * here we put the values of params and function to call onto
98 * namespace, this is so perl will properly clone them when we
99 * call perl_clone.
100 */
4f896ddc 101
b1edfb69 102 temp_store = Perl_get_sv(current_perl, "threads::paramtempstore",
103 TRUE | GV_ADDMULTI);
485caab9 104 Perl_sv_setsv_flags(current_perl, temp_store,params, SV_GMAGIC);
47ba8780 105 params = NULL;
106 temp_store = NULL;
107
b1edfb69 108 temp_store = Perl_get_sv(current_perl, "threads::calltempstore",
109 TRUE | GV_ADDMULTI);
485caab9 110 Perl_sv_setsv_flags(current_perl,temp_store, init_function, SV_GMAGIC);
4f896ddc 111 init_function = NULL;
112 temp_store = NULL;
47ba8780 113
114#ifdef WIN32
b1edfb69 115 thread->interp = perl_clone(current_perl, 4);
47ba8780 116#else
b1edfb69 117 thread->interp = perl_clone(current_perl, 0);
47ba8780 118#endif
47ba8780 119
b1edfb69 120 thread->init_function = newSVsv(Perl_get_sv(thread->interp,
121 "threads::calltempstore",FALSE));
122 thread->params = newSVsv(Perl_get_sv(thread->interp,
123 "threads::paramtempstore",FALSE));
47ba8780 124
125 /*
b1edfb69 126 * And here we make sure we clean up the data we put in the
127 * namespace of iThread, both in the new and the calling
128 * inteprreter */
47ba8780 129
b1edfb69 130 temp_store = Perl_get_sv(thread->interp, "threads::paramtempstore",FALSE);
485caab9 131 Perl_sv_setsv_flags(thread->interp,temp_store, &PL_sv_undef, SV_GMAGIC);
47ba8780 132
133 temp_store = Perl_get_sv(thread->interp,"threads::calltempstore",FALSE);
485caab9 134 Perl_sv_setsv_flags(thread->interp,temp_store, &PL_sv_undef, SV_GMAGIC);
47ba8780 135
136 PERL_SET_CONTEXT(current_perl);
137
138 temp_store = Perl_get_sv(current_perl,"threads::paramtempstore",FALSE);
485caab9 139 Perl_sv_setsv_flags(current_perl, temp_store, &PL_sv_undef, SV_GMAGIC);
47ba8780 140
141 temp_store = Perl_get_sv(current_perl,"threads::calltempstore",FALSE);
485caab9 142 Perl_sv_setsv_flags(current_perl, temp_store, &PL_sv_undef, SV_GMAGIC);
47ba8780 143
b1edfb69 144 /* let's init the thread */
47ba8780 145
146 MUTEX_INIT(&thread->mutex);
147 thread->tid = tid_counter++;
148 thread->detached = 0;
149 thread->count = 1;
150
151#ifdef WIN32
152
e6e315b9 153 thread->handle = CreateThread(NULL, 0, Perl_thread_run,
47ba8780 154 (LPVOID)thread, 0, &thread->thr);
155
82c40bf6 156
157#else
fa26028c 158 {
159 static pthread_attr_t attr;
160 static int attr_inited = 0;
161 sigset_t fullmask, oldmask;
162 static int attr_joinable = PTHREAD_CREATE_JOINABLE;
163 if (!attr_inited) {
164 attr_inited = 1;
165 pthread_attr_init(&attr);
166 }
167# ifdef PTHREAD_ATTR_SETDETACHSTATE
168 PTHREAD_ATTR_SETDETACHSTATE(&attr, attr_joinable);
169# endif
3eb37d38 170# ifdef THREAD_CREATE_NEEDS_STACK
171 if(pthread_attr_setstacksize(&attr, THREAD_CREATE_NEEDS_STACK))
172 croak("panic: pthread_attr_setstacksize failed");
173# endif
174
3ad0b7d6 175#ifdef OLD_PTHREADS_API
fa26028c 176 pthread_create( &thread->thr, attr, Perl_thread_run, (void *)thread);
47ba8780 177#else
3ad0b7d6 178 pthread_create( &thread->thr, &attr, Perl_thread_run, (void *)thread);
47ba8780 179#endif
3ad0b7d6 180 }
82c40bf6 181#endif
47ba8780 182 MUTEX_UNLOCK(&create_mutex);
183
b1edfb69 184 return obj_ref;
47ba8780 185}
186
187/*
b1edfb69 188 * returns the id of the thread
189 */
e6e315b9 190I32 Perl_thread_tid (SV* obj) {
47ba8780 191 ithread* thread;
192 if(!SvROK(obj)) {
e6e315b9 193 obj = Perl_thread_self(SvPV_nolen(obj));
47ba8780 194 thread = (ithread*)SvIV(SvRV(obj));
195 SvREFCNT_dec(obj);
196 } else {
197 thread = (ithread*)SvIV(SvRV(obj));
198 }
199 return thread->tid;
200}
201
e6e315b9 202SV* Perl_thread_self (char* class) {
47ba8780 203 dTHX;
204 SV* obj_ref;
205 SV* obj;
b1edfb69 206 SV* thread_tid_ptr;
207 SV* thread_ptr;
208 HE* thread_entry;
7f9a95ef 209 void* id;
210 PERL_THREAD_GETSPECIFIC(self_key,id);
4f896ddc 211 SHAREDSvLOCK(threads);
47ba8780 212 SHAREDSvEDIT(threads);
7f9a95ef 213
214 thread_tid_ptr = Perl_newSVuv(PL_sharedsv_space, PTR2UV(id));
82c40bf6 215
b1edfb69 216 thread_entry = Perl_hv_fetch_ent(PL_sharedsv_space,
217 (HV*) SHAREDSvGET(threads),
218 thread_tid_ptr, 0,0);
47ba8780 219 thread_ptr = HeVAL(thread_entry);
220 SvREFCNT_dec(thread_tid_ptr);
47ba8780 221 SHAREDSvRELEASE(threads);
4f896ddc 222 SHAREDSvUNLOCK(threads);
47ba8780 223
224 obj_ref = newSViv(0);
225 obj = newSVrv(obj_ref, class);
4f896ddc 226 sv_setsv(obj, thread_ptr);
47ba8780 227 SvREADONLY_on(obj);
228 return obj_ref;
229}
230
231/*
b1edfb69 232 * joins the thread this code needs to take the returnvalue from the
233 * call_sv and send it back */
47ba8780 234
e6e315b9 235void Perl_thread_join(SV* obj) {
47ba8780 236 ithread* thread = (ithread*)SvIV(SvRV(obj));
237#ifdef WIN32
238 DWORD waitcode;
239 waitcode = WaitForSingleObject(thread->handle, INFINITE);
240#else
241 void *retval;
242 pthread_join(thread->thr,&retval);
243#endif
244}
245
b1edfb69 246/* detaches a thread
247 * needs to better clean up memory */
47ba8780 248
e6e315b9 249void Perl_thread_detach(SV* obj) {
47ba8780 250 ithread* thread = (ithread*)SvIV(SvRV(obj));
251 MUTEX_LOCK(&thread->mutex);
252 thread->detached = 1;
82c40bf6 253 PERL_THREAD_DETACH(thread->thr);
47ba8780 254 MUTEX_UNLOCK(&thread->mutex);
255}
256
e6e315b9 257void Perl_thread_DESTROY (SV* obj) {
47ba8780 258 ithread* thread = (ithread*)SvIV(SvRV(obj));
259
260 MUTEX_LOCK(&thread->mutex);
261 thread->count--;
262 MUTEX_UNLOCK(&thread->mutex);
e6e315b9 263 Perl_thread_destruct(thread);
47ba8780 264}
265
e6e315b9 266void Perl_thread_destruct (ithread* thread) {
47ba8780 267 return;
268 MUTEX_LOCK(&thread->mutex);
269 if(thread->count != 0) {
270 MUTEX_UNLOCK(&thread->mutex);
271 return;
272 }
273 MUTEX_UNLOCK(&thread->mutex);
274 /* it is safe noone is holding a ref to this */
275 /*printf("proper destruction!\n");*/
276}
277
47ba8780 278MODULE = threads PACKAGE = threads
279BOOT:
280 Perl_sharedsv_init(aTHX);
89e0305a 281 PERL_THREAD_ALLOC_SPECIFIC(self_key);
47ba8780 282 PL_perl_destruct_level = 2;
283 threads = Perl_sharedsv_new(aTHX);
284 SHAREDSvEDIT(threads);
e8f2bb9a 285 SHAREDSvGET(threads) = (SV *)newHV();
47ba8780 286 SHAREDSvRELEASE(threads);
287 {
288
289
290 SV* temp = get_sv("threads::sharedsv_space", TRUE | GV_ADDMULTI);
4f896ddc 291 SV* temp2 = newSViv((IV)PL_sharedsv_space );
47ba8780 292 sv_setsv( temp , temp2 );
293 }
294 {
295 ithread* thread = malloc(sizeof(ithread));
296 SV* thread_tid_ptr;
297 SV* thread_ptr;
298 MUTEX_INIT(&thread->mutex);
299 thread->tid = 0;
300#ifdef WIN32
301 thread->thr = GetCurrentThreadId();
302#else
303 thread->thr = pthread_self();
304#endif
d5ecd109 305 SHAREDSvEDIT(threads);
82c40bf6 306 PERL_THREAD_ALLOC_SPECIFIC(self_key);
7f9a95ef 307 PERL_THREAD_SETSPECIFIC(self_key,0);
308 thread_tid_ptr = Perl_newSVuv(PL_sharedsv_space, 0);
b1edfb69 309 thread_ptr = Perl_newSVuv(PL_sharedsv_space, PTR2UV(thread));
47ba8780 310 hv_store_ent((HV*) SHAREDSvGET(threads), thread_tid_ptr, thread_ptr,0);
4f896ddc 311 SvREFCNT_dec(thread_tid_ptr);
d5ecd109 312 SHAREDSvRELEASE(threads);
47ba8780 313 }
314 MUTEX_INIT(&create_mutex);
315
47ba8780 316PROTOTYPES: DISABLE
317
318SV *
319create (class, function_to_call, ...)
320 char * class
321 SV * function_to_call
322 CODE:
323 AV* params = newAV();
324 if(items > 2) {
325 int i;
326 for(i = 2; i < items ; i++) {
327 av_push(params, ST(i));
328 }
329 }
e6e315b9 330 RETVAL = Perl_thread_create(class, function_to_call, newRV_noinc((SV*) params));
47ba8780 331 OUTPUT:
332 RETVAL
333
334SV *
335self (class)
336 char* class
337 CODE:
e6e315b9 338 RETVAL = Perl_thread_self(class);
47ba8780 339 OUTPUT:
340 RETVAL
341
342int
343tid (obj)
344 SV * obj;
345 CODE:
e6e315b9 346 RETVAL = Perl_thread_tid(obj);
47ba8780 347 OUTPUT:
348 RETVAL
349
350void
351join (obj)
352 SV * obj
353 PREINIT:
354 I32* temp;
355 PPCODE:
356 temp = PL_markstack_ptr++;
e6e315b9 357 Perl_thread_join(obj);
47ba8780 358 if (PL_markstack_ptr != temp) {
359 /* truly void, because dXSARGS not invoked */
360 PL_markstack_ptr = temp;
361 XSRETURN_EMPTY; /* return empty stack */
362 }
363 /* must have used dXSARGS; list context implied */
364 return; /* assume stack size is correct */
365
366void
367detach (obj)
368 SV * obj
369 PREINIT:
370 I32* temp;
371 PPCODE:
372 temp = PL_markstack_ptr++;
e6e315b9 373 Perl_thread_detach(obj);
47ba8780 374 if (PL_markstack_ptr != temp) {
375 /* truly void, because dXSARGS not invoked */
376 PL_markstack_ptr = temp;
377 XSRETURN_EMPTY; /* return empty stack */
378 }
379 /* must have used dXSARGS; list context implied */
380 return; /* assume stack size is correct */
381
47ba8780 382void
383DESTROY (obj)
384 SV * obj
385 PREINIT:
386 I32* temp;
387 PPCODE:
388 temp = PL_markstack_ptr++;
e6e315b9 389 Perl_thread_DESTROY(obj);
47ba8780 390 if (PL_markstack_ptr != temp) {
391 /* truly void, because dXSARGS not invoked */
392 PL_markstack_ptr = temp;
393 XSRETURN_EMPTY; /* return empty stack */
394 }
395 /* must have used dXSARGS; list context implied */
396 return; /* assume stack size is correct */
397