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 /* --- Compatibility wrappers ---------------------------------------------- */
11 #define LT_HAS_PERL(R, V, S) (PERL_REVISION > (R) || (PERL_REVISION == (R) && (PERL_VERSION > (V) || (PERL_VERSION == (V) && (PERL_SUBVERSION >= (S))))))
13 #if LT_HAS_PERL(5, 10, 0) || defined(PL_parser)
14 # ifndef PL_in_my_stash
15 # define PL_in_my_stash PL_parser->in_my_stash
18 # ifndef PL_in_my_stash
19 # define PL_in_my_stash PL_Iin_my_stash
24 # define Newx(v, n, c) New(0, v, n, c)
28 # define HvNAME_get(H) HvNAME(H)
32 # define HvNAMELEN_get(H) strlen(HvNAME_get(H))
35 #define __PACKAGE__ "Lexical::Types"
36 #define __PACKAGE_LEN__ (sizeof(__PACKAGE__)-1)
38 /* --- Helpers ------------------------------------------------------------- */
40 /* ... Hints ............................................................... */
42 STATIC U32 lt_hash = 0;
44 STATIC SV *lt_hint(pTHX) {
45 #define lt_hint() lt_hint(aTHX)
47 #if LT_HAS_PERL(5, 10, 0)
48 id = Perl_refcounted_he_fetch(aTHX_ PL_curcop->cop_hints_hash,
50 __PACKAGE__, __PACKAGE_LEN__,
54 SV **val = hv_fetch(GvHV(PL_hintgv), __PACKAGE__, __PACKAGE_LEN__, lt_hash);
59 return (id && SvOK(id)) ? id : NULL;
62 /* ... op => info map ...................................................... */
64 #define OP2STR_BUF char buf[(CHAR_BIT * sizeof(UV)) / 2]
65 #define OP2STR(O) (sprintf(buf, "%"UVxf, PTR2UV(O)))
67 STATIC HV *lt_op_map = NULL;
73 OP *(*pp_padsv)(pTHX);
76 STATIC void lt_map_store(pTHX_ const OP *o, SV *orig_pkg, SV *type_pkg, SV *type_meth, OP *(*pp_padsv)(pTHX)) {
77 #define lt_map_store(O, P1, P2, M, PP) lt_map_store(aTHX_ (O), (P1), (P2), (M), (PP))
82 Newx(oi, 1, lt_op_info);
83 oi->orig_pkg = orig_pkg;
84 oi->type_pkg = type_pkg;
85 oi->type_meth = type_meth;
86 oi->pp_padsv = pp_padsv;
87 val = newSVuv(PTR2UV(oi));
89 (void)hv_store(lt_op_map, buf, OP2STR(o), val, 0);
92 STATIC const lt_op_info *lt_map_fetch(pTHX_ const OP *o) {
93 #define lt_map_fetch(O) lt_map_fetch(aTHX_ (O))
97 svp = hv_fetch(lt_op_map, buf, OP2STR(o), 0);
99 return svp ? INT2PTR(const lt_op_info *, SvUVX(*svp)) : NULL;
102 /* --- Hooks --------------------------------------------------------------- */
104 /* ... Our pp_padsv ........................................................ */
106 STATIC OP *lt_pp_padsv(pTHX) {
107 const lt_op_info *oi;
109 if ((PL_op->op_private & OPpLVAL_INTRO) && (oi = lt_map_fetch(PL_op))) {
110 PADOFFSET targ = PL_op->op_targ;
111 SV *sv = PAD_SVl(targ);
127 items = call_sv(oi->type_meth, G_ARRAY | G_METHOD);
137 croak("Typed scalar initializer method should return zero or one scalar, but got %d", items);
145 return CALL_FPTR(oi->pp_padsv)(aTHX);
148 return CALL_FPTR(PL_ppaddr[OP_PADSV])(aTHX);
151 STATIC OP *(*lt_pp_padsv_saved)(pTHX) = 0;
153 STATIC void lt_pp_padsv_save(void) {
154 if (lt_pp_padsv_saved)
157 lt_pp_padsv_saved = PL_ppaddr[OP_PADSV];
158 PL_ppaddr[OP_PADSV] = lt_pp_padsv;
161 STATIC void lt_pp_padsv_restore(OP *o) {
162 if (!lt_pp_padsv_saved)
165 if (o->op_ppaddr == lt_pp_padsv)
166 o->op_ppaddr = lt_pp_padsv_saved;
168 PL_ppaddr[OP_PADSV] = lt_pp_padsv_saved;
169 lt_pp_padsv_saved = 0;
172 /* ... Our ck_pad{any,sv} .................................................. */
174 /* Sadly, the PADSV OPs we are interested in don't trigger the padsv check
175 * function, but are instead manually mutated from a PADANY. This is why we set
176 * PL_ppaddr[OP_PADSV] in the padany check function so that PADSV OPs will have
177 * their pp_ppaddr set to our pp_padsv. PL_ppaddr[OP_PADSV] is then reset at the
178 * beginning of every ck_pad{any,sv}. Some unwanted OPs can still call our
179 * pp_padsv, but much less than if we would have set PL_ppaddr[OP_PADSV]
182 STATIC SV *lt_default_meth = NULL;
184 STATIC OP *(*lt_old_ck_padany)(pTHX_ OP *) = 0;
186 STATIC OP *lt_ck_padany(pTHX_ OP *o) {
190 lt_pp_padsv_restore(o);
192 o = CALL_FPTR(lt_old_ck_padany)(aTHX_ o);
194 stash = PL_in_my_stash;
195 if (stash && (hint = lt_hint())) {
196 SV *orig_pkg = newSVpvn(HvNAME_get(stash), HvNAMELEN_get(stash));
197 SV *orig_meth = lt_default_meth;
198 SV *type_pkg = orig_pkg;
199 SV *type_meth = orig_meth;
200 SV *code = INT2PTR(SV *, SvUVX(hint));
202 SvREADONLY_on(orig_pkg);
217 items = call_sv(code, G_ARRAY);
221 croak(__PACKAGE__ " mangler should return zero, one or two scalars, but got %d", items);
223 SvREFCNT_dec(orig_pkg);
230 type_meth = newSVsv(rsv);
231 SvREADONLY_on(type_meth);
236 type_pkg = newSVsv(rsv);
237 SvREADONLY_on(type_pkg);
246 if (type_meth == orig_meth)
247 SvREFCNT_inc(orig_meth);
251 lt_map_store(o, orig_pkg, type_pkg, type_meth, lt_pp_padsv_saved);
258 STATIC OP *(*lt_old_ck_padsv)(pTHX_ OP *) = 0;
260 STATIC OP *lt_ck_padsv(pTHX_ OP *o) {
261 lt_pp_padsv_restore(o);
263 return CALL_FPTR(lt_old_ck_padsv)(aTHX_ o);
266 STATIC U32 lt_initialized = 0;
268 /* --- XS ------------------------------------------------------------------ */
270 MODULE = Lexical::Types PACKAGE = Lexical::Types
276 if (!lt_initialized++) {
277 lt_default_meth = newSVpvn("TYPEDSCALAR", 11);
278 SvREADONLY_on(lt_default_meth);
280 PERL_HASH(lt_hash, __PACKAGE__, __PACKAGE_LEN__);
283 lt_old_ck_padany = PL_check[OP_PADANY];
284 PL_check[OP_PADANY] = MEMBER_TO_FPTR(lt_ck_padany);
285 lt_old_ck_padsv = PL_check[OP_PADSV];
286 PL_check[OP_PADSV] = MEMBER_TO_FPTR(lt_ck_padsv);
294 if (SvOK(ref) && SvROK(ref)) {
297 ret = newSVuv(PTR2UV(sv));