1 /* This file is part of the Lexical-Types Perl module.
2 * See http://search.cpan.org/dist/Lexical-Types/ */
4 #define PERL_NO_GET_CONTEXT
9 #define __PACKAGE__ "Lexical::Types"
10 #define __PACKAGE_LEN__ (sizeof(__PACKAGE__)-1)
12 /* --- Compatibility wrappers ---------------------------------------------- */
14 #define LT_HAS_PERL(R, V, S) (PERL_REVISION > (R) || (PERL_REVISION == (R) && (PERL_VERSION > (V) || (PERL_VERSION == (V) && (PERL_SUBVERSION >= (S))))))
16 #if LT_HAS_PERL(5, 10, 0) || defined(PL_parser)
17 # ifndef PL_in_my_stash
18 # define PL_in_my_stash PL_parser->in_my_stash
21 # ifndef PL_in_my_stash
22 # define PL_in_my_stash PL_Iin_my_stash
27 # define HvNAME_get(H) HvNAME(H)
31 # define HvNAMELEN_get(H) strlen(HvNAME_get(H))
35 # define SvIS_FREED(sv) ((sv)->sv_flags == SVTYPEMASK)
38 /* ... Thread safety and multiplicity ...................................... */
40 #ifndef LT_MULTIPLICITY
41 # if defined(MULTIPLICITY) || defined(PERL_IMPLICIT_CONTEXT)
42 # define LT_MULTIPLICITY 1
44 # define LT_MULTIPLICITY 0
47 #if LT_MULTIPLICITY && !defined(tTHX)
48 # define tTHX PerlInterpreter*
51 #if LT_MULTIPLICITY && defined(USE_ITHREADS) && defined(dMY_CXT) && defined(MY_CXT) && defined(START_MY_CXT) && defined(MY_CXT_INIT) && (defined(MY_CXT_CLONE) || defined(dMY_CXT_SV))
52 # define LT_THREADSAFE 1
54 # define MY_CXT_CLONE \
56 my_cxt_t *my_cxtp = (my_cxt_t*)SvPVX(newSV(sizeof(my_cxt_t)-1)); \
57 Copy(INT2PTR(my_cxt_t*, SvUV(my_cxt_sv)), my_cxtp, 1, my_cxt_t); \
58 sv_setuv(my_cxt_sv, PTR2UV(my_cxtp))
61 # define LT_THREADSAFE 0
64 /* --- Helpers ------------------------------------------------------------- */
66 /* ... Thread-safe hints ................................................... */
70 #define PTABLE_NAME ptable_hints
71 #define PTABLE_VAL_FREE(V) if ((V) && !SvIS_FREED((SV *) (V))) SvREFCNT_dec(V)
80 #define ptable_hints_store(T, K, V) ptable_hints_store(aTHX_ (T), (K), (V))
81 #define ptable_hints_free(T) ptable_hints_free(aTHX_ (T))
83 #define MY_CXT_KEY __PACKAGE__ "::_guts" XS_VERSION
92 STATIC void lt_ptable_hints_clone(pTHX_ ptable_ent *ent, void *ud_) {
96 if (ud->owner != aTHX) {
98 AV *stashes = (SvTYPE(val) == SVt_PVHV && HvNAME_get(val)) ? newAV() : NULL;
99 param.stashes = stashes;
101 param.proto_perl = ud->owner;
102 val = sv_dup(val, ¶m);
105 SvREFCNT_dec(stashes);
109 ptable_hints_store(ud->tbl, ent->key, val);
113 STATIC void lt_thread_cleanup(pTHX_ void *);
115 STATIC void lt_thread_cleanup(pTHX_ void *ud) {
122 SAVEDESTRUCTOR_X(lt_thread_cleanup, level);
126 PerlMemShared_free(level);
127 ptable_hints_free(MY_CXT.tbl);
131 STATIC SV *lt_tag(pTHX_ SV *value) {
132 #define lt_tag(V) lt_tag(aTHX_ (V))
135 value = SvOK(value) && SvROK(value) ? SvRV(value) : NULL;
136 /* We only need for the key to be an unique tag for looking up the value later.
137 * Allocated memory provides convenient unique identifiers, so that's why we
138 * use the value pointer as the key itself. */
139 ptable_hints_store(MY_CXT.tbl, value, value);
142 return newSVuv(PTR2UV(value));
145 STATIC SV *lt_detag(pTHX_ const SV *hint) {
146 #define lt_detag(H) lt_detag(aTHX_ (H))
150 if (!hint || !SvOK(hint) || !SvIOK(hint))
153 tag = INT2PTR(void *, SvIVX(hint));
156 value = ptable_fetch(MY_CXT.tbl, tag);
164 STATIC SV *lt_tag(pTHX_ SV *value) {
165 #define lt_tag(V) lt_tag(aTHX_ (V))
168 if (SvOK(value) && SvROK(value)) {
177 #define lt_detag(H) INT2PTR(SV *, SvUVX(H))
179 #endif /* LT_THREADSAFE */
181 STATIC U32 lt_hash = 0;
183 STATIC SV *lt_hint(pTHX) {
184 #define lt_hint() lt_hint(aTHX)
186 #if LT_HAS_PERL(5, 9, 5)
187 id = Perl_refcounted_he_fetch(aTHX_ PL_curcop->cop_hints_hash,
189 __PACKAGE__, __PACKAGE_LEN__,
193 SV **val = hv_fetch(GvHV(PL_hintgv), __PACKAGE__, __PACKAGE_LEN__, lt_hash);
198 return (id && SvOK(id)) ? id : NULL;
201 /* ... op => info map ...................................................... */
203 #define PTABLE_NAME ptable_map
204 #define PTABLE_VAL_FREE(V) PerlMemShared_free(V)
208 /* PerlMemShared_free() needs the [ap]PTBLMS_? default values */
209 #define ptable_map_store(T, K, V) ptable_map_store(aPTBLMS_ (T), (K), (V))
211 STATIC ptable *lt_op_map = NULL;
214 STATIC perl_mutex lt_op_map_mutex;
221 OP *(*pp_padsv)(pTHX);
224 STATIC void lt_map_store(pPTBLMS_ const OP *o, SV *orig_pkg, SV *type_pkg, SV *type_meth, OP *(*pp_padsv)(pTHX)) {
225 #define lt_map_store(O, OP, TP, TM, PP) lt_map_store(aPTBLMS_ (O), (OP), (TP), (TM), (PP))
229 MUTEX_LOCK(<_op_map_mutex);
232 if (!(oi = ptable_fetch(lt_op_map, o))) {
233 oi = PerlMemShared_malloc(sizeof *oi);
234 ptable_map_store(lt_op_map, o, oi);
237 oi->orig_pkg = orig_pkg;
238 oi->type_pkg = type_pkg;
239 oi->type_meth = type_meth;
240 oi->pp_padsv = pp_padsv;
243 MUTEX_UNLOCK(<_op_map_mutex);
247 STATIC const lt_op_info *lt_map_fetch(const OP *o, lt_op_info *oi) {
248 const lt_op_info *val;
251 MUTEX_LOCK(<_op_map_mutex);
254 val = ptable_fetch(lt_op_map, o);
261 MUTEX_UNLOCK(<_op_map_mutex);
267 /* --- Hooks --------------------------------------------------------------- */
269 /* ... Our pp_padsv ........................................................ */
271 STATIC OP *lt_pp_padsv(pTHX) {
274 if ((PL_op->op_private & OPpLVAL_INTRO) && lt_map_fetch(PL_op, &oi)) {
275 PADOFFSET targ = PL_op->op_targ;
276 SV *sv = PAD_SVl(targ);
292 items = call_sv(oi.type_meth, G_ARRAY | G_METHOD);
302 croak("Typed scalar initializer method should return zero or one scalar, but got %d", items);
310 return CALL_FPTR(oi.pp_padsv)(aTHX);
313 return CALL_FPTR(PL_ppaddr[OP_PADSV])(aTHX);
316 STATIC OP *(*lt_pp_padsv_saved)(pTHX) = 0;
318 STATIC void lt_pp_padsv_save(void) {
319 if (lt_pp_padsv_saved)
322 lt_pp_padsv_saved = PL_ppaddr[OP_PADSV];
323 PL_ppaddr[OP_PADSV] = lt_pp_padsv;
326 STATIC void lt_pp_padsv_restore(OP *o) {
327 if (!lt_pp_padsv_saved)
330 if (o->op_ppaddr == lt_pp_padsv)
331 o->op_ppaddr = lt_pp_padsv_saved;
333 PL_ppaddr[OP_PADSV] = lt_pp_padsv_saved;
334 lt_pp_padsv_saved = 0;
337 /* ... Our ck_pad{any,sv} .................................................. */
339 /* Sadly, the PADSV OPs we are interested in don't trigger the padsv check
340 * function, but are instead manually mutated from a PADANY. This is why we set
341 * PL_ppaddr[OP_PADSV] in the padany check function so that PADSV OPs will have
342 * their pp_ppaddr set to our pp_padsv. PL_ppaddr[OP_PADSV] is then reset at the
343 * beginning of every ck_pad{any,sv}. Some unwanted OPs can still call our
344 * pp_padsv, but much less than if we would have set PL_ppaddr[OP_PADSV]
347 STATIC SV *lt_default_meth = NULL;
349 STATIC OP *(*lt_old_ck_padany)(pTHX_ OP *) = 0;
351 STATIC OP *lt_ck_padany(pTHX_ OP *o) {
355 lt_pp_padsv_restore(o);
357 o = CALL_FPTR(lt_old_ck_padany)(aTHX_ o);
359 stash = PL_in_my_stash;
360 if (stash && (hint = lt_hint())) {
361 SV *orig_pkg = newSVpvn(HvNAME_get(stash), HvNAMELEN_get(stash));
362 SV *orig_meth = lt_default_meth;
364 SV *type_meth = NULL;
365 SV *code = lt_detag(hint);
367 SvREADONLY_on(orig_pkg);
382 items = call_sv(code, G_ARRAY);
386 croak(__PACKAGE__ " mangler should return zero, one or two scalars, but got %d", items);
388 SvREFCNT_dec(orig_pkg);
395 type_meth = newSVsv(rsv);
396 SvREADONLY_on(type_meth);
401 type_pkg = newSVsv(rsv);
402 SvREADONLY_on(type_pkg);
413 SvREFCNT_inc(orig_pkg);
417 type_meth = orig_meth;
418 SvREFCNT_inc(orig_meth);
423 lt_map_store(o, orig_pkg, type_pkg, type_meth, lt_pp_padsv_saved);
430 STATIC OP *(*lt_old_ck_padsv)(pTHX_ OP *) = 0;
432 STATIC OP *lt_ck_padsv(pTHX_ OP *o) {
433 lt_pp_padsv_restore(o);
435 return CALL_FPTR(lt_old_ck_padsv)(aTHX_ o);
438 STATIC U32 lt_initialized = 0;
440 /* --- XS ------------------------------------------------------------------ */
442 MODULE = Lexical::Types PACKAGE = Lexical::Types
448 if (!lt_initialized++) {
452 MY_CXT.tbl = ptable_new();
456 lt_op_map = ptable_new();
458 MUTEX_INIT(<_op_map_mutex);
461 lt_default_meth = newSVpvn("TYPEDSCALAR", 11);
462 SvREADONLY_on(lt_default_meth);
464 PERL_HASH(lt_hash, __PACKAGE__, __PACKAGE_LEN__);
466 lt_old_ck_padany = PL_check[OP_PADANY];
467 PL_check[OP_PADANY] = MEMBER_TO_FPTR(lt_ck_padany);
468 lt_old_ck_padsv = PL_check[OP_PADSV];
469 PL_check[OP_PADSV] = MEMBER_TO_FPTR(lt_ck_padsv);
471 stash = gv_stashpvn(__PACKAGE__, __PACKAGE_LEN__, 1);
472 newCONSTSUB(stash, "LT_THREADSAFE", newSVuv(LT_THREADSAFE));
488 ud.tbl = t = ptable_new();
489 ud.owner = MY_CXT.owner;
490 ptable_walk(MY_CXT.tbl, lt_ptable_hints_clone, &ud);
498 level = PerlMemShared_malloc(sizeof *level);
501 SAVEDESTRUCTOR_X(lt_thread_cleanup, level);
511 RETVAL = lt_tag(value);