1 /* This file is part of the autovivification Perl module.
2 * See http://search.cpan.org/dist/autovivification/ */
4 #define PERL_NO_GET_CONTEXT
9 /* --- XS helpers ---------------------------------------------------------- */
11 #define XSH_PACKAGE "autovivification"
18 /* ... Lexical hints ....................................................... */
20 /* Used both for hints and op flags */
21 #define A_HINT_STRICT 1
23 #define A_HINT_FETCH 4
24 #define A_HINT_STORE 8
25 #define A_HINT_KEYS 16
26 #define A_HINT_VALUES 32
27 #define A_HINT_EXISTS 64
28 #define A_HINT_DELETE 128
29 #define A_HINT_NOTIFY (A_HINT_STRICT|A_HINT_WARN)
30 #define A_HINT_DO (A_HINT_FETCH|A_HINT_STORE|A_HINT_KEYS|A_HINT_VALUES|A_HINT_EXISTS|A_HINT_DELETE)
31 #define A_HINT_MASK (A_HINT_NOTIFY|A_HINT_DO)
33 /* Only used in op flags */
34 #define A_HINT_ROOT 256
35 #define A_HINT_SECOND 512
36 #define A_HINT_DEREF 1024
38 #define XSH_HINTS_TYPE_UV 1
40 #include "xsh/hints.h"
42 #define a_hint() xsh_hints_detag(xsh_hints_fetch())
44 /* ... Thread-local storage ................................................ */
46 #define XSH_THREADS_COMPILE_TIME_PROTECTION 1
47 #define XSH_THREADS_USER_CONTEXT 0
49 #include "xsh/threads.h"
51 /* --- Compatibility wrappers ---------------------------------------------- */
54 # define HvNAME_get(H) HvNAME(H)
58 # define HvNAMELEN_get(H) strlen(HvNAME_get(H))
61 #ifndef A_HAS_MULTIDEREF
62 # define A_HAS_MULTIDEREF XSH_HAS_PERL(5, 21, 7)
65 #ifndef A_HAS_SCALARKEYS_OPT
66 # define A_HAS_SCALARKEYS_OPT XSH_HAS_PERL(5, 27, 3)
69 /* ... Our vivify_ref() .................................................... */
71 /* Perl_vivify_ref() is not exported, so we have to reimplement it. */
75 static SV *a_vivify_ref(pTHX_ SV *sv, int to_hash) {
76 #define a_vivify_ref(S, TH) a_vivify_ref(aTHX_ (S), (TH))
83 Perl_croak_no_modify();
85 /* Inlined prepare_SV_for_RV() */
86 if (SvTYPE(sv) < SVt_PV && SvTYPE(sv) != SVt_IV) {
87 sv_upgrade(sv, SVt_IV);
88 } else if (SvTYPE(sv) >= SVt_PV) {
94 val = to_hash ? MUTABLE_SV(newHV()) : MUTABLE_SV(newAV());
101 if (SvGMAGICAL(sv)) {
102 SV *msv = sv_newmortal();
103 sv_setsv_nomg(msv, sv);
110 #endif /* A_HAS_MULTIDEREF */
112 /* --- op => info map ------------------------------------------------------ */
120 #define PTABLE_NAME ptable_map
121 #define PTABLE_VAL_FREE(V) PerlMemShared_free(V)
122 #define PTABLE_VAL_NEED_CONTEXT 0
123 #define PTABLE_NEED_DELETE 1
124 #define PTABLE_NEED_WALK 0
126 #include "xsh/ptable.h"
128 #define ptable_map_store(T, K, V) ptable_map_store(aPMS_ (T), (K), (V))
129 #define ptable_map_delete(T, K) ptable_map_delete(aPMS_ (T), (K))
130 #define ptable_map_free(T) ptable_map_free(aPMS_ (T))
132 static ptable *a_op_map = NULL;
136 #define dA_MAP_THX a_op_info a_op_map_tmp_oi
138 static perl_mutex a_op_map_mutex;
140 static const a_op_info *a_map_fetch(const OP *o, a_op_info *oi) {
141 const a_op_info *val;
143 XSH_LOCK(&a_op_map_mutex);
145 val = ptable_fetch(a_op_map, o);
151 XSH_UNLOCK(&a_op_map_mutex);
156 #define a_map_fetch(O) a_map_fetch((O), &a_op_map_tmp_oi)
158 #else /* USE_ITHREADS */
160 #define dA_MAP_THX dNOOP
162 #define a_map_fetch(O) ptable_fetch(a_op_map, (O))
164 #endif /* !USE_ITHREADS */
166 static const a_op_info *a_map_store_locked(pPMS_ const OP *o, OP *(*old_pp)(pTHX), void *next, UV flags) {
167 #define a_map_store_locked(O, PP, N, F) a_map_store_locked(aPMS_ (O), (PP), (N), (F))
170 if (!(oi = ptable_fetch(a_op_map, o))) {
171 oi = PerlMemShared_malloc(sizeof *oi);
172 ptable_map_store(a_op_map, o, oi);
182 static void a_map_store(pTHX_ const OP *o, OP *(*old_pp)(pTHX), void *next, UV flags) {
183 #define a_map_store(O, PP, N, F) a_map_store(aTHX_ (O), (PP), (N), (F))
184 XSH_LOCK(&a_op_map_mutex);
186 a_map_store_locked(o, old_pp, next, flags);
188 XSH_UNLOCK(&a_op_map_mutex);
191 static void a_map_delete(pTHX_ const OP *o) {
192 #define a_map_delete(O) a_map_delete(aTHX_ (O))
193 XSH_LOCK(&a_op_map_mutex);
195 ptable_map_delete(a_op_map, o);
197 XSH_UNLOCK(&a_op_map_mutex);
200 static const OP *a_map_descend(const OP *o) {
201 switch (PL_opargs[o->op_type] & OA_CLASS_MASK) {
205 case OA_BASEOP_OR_UNOP:
206 return cUNOPo->op_first;
209 return cLISTOPo->op_last;
215 static void a_map_store_root(pTHX_ const OP *root, OP *(*old_pp)(pTHX), UV flags) {
216 #define a_map_store_root(R, PP, F) a_map_store_root(aTHX_ (R), (PP), (F))
217 const a_op_info *roi;
221 XSH_LOCK(&a_op_map_mutex);
223 roi = a_map_store_locked(o, old_pp, (OP *) root, flags | A_HINT_ROOT);
225 while (o->op_flags & OPf_KIDS) {
226 o = a_map_descend(o);
229 if ((oi = ptable_fetch(a_op_map, o))) {
230 oi->flags &= ~A_HINT_ROOT;
231 oi->next = (a_op_info *) roi;
236 XSH_UNLOCK(&a_op_map_mutex);
241 static void a_map_update_flags_topdown(const OP *root, UV mask, UV flags) {
245 XSH_LOCK(&a_op_map_mutex);
251 if ((oi = ptable_fetch(a_op_map, o)))
252 oi->flags = (oi->flags & mask) | flags;
253 if (!(o->op_flags & OPf_KIDS))
255 o = a_map_descend(o);
258 XSH_UNLOCK(&a_op_map_mutex);
263 static void a_map_update_flags_bottomup(const OP *o, UV flags, UV rflags) {
266 XSH_LOCK(&a_op_map_mutex);
268 flags &= ~A_HINT_ROOT;
269 rflags |= A_HINT_ROOT;
271 oi = ptable_fetch(a_op_map, o);
272 while (!(oi->flags & A_HINT_ROOT)) {
278 XSH_UNLOCK(&a_op_map_mutex);
283 /* ... Decide whether this expression should be autovivified or not ........ */
285 static UV a_map_resolve(const OP *o, const a_op_info *oi) {
286 UV flags = 0, rflags;
288 const a_op_info *roi = oi;
290 while (!(roi->flags & A_HINT_ROOT))
295 rflags = roi->flags & ~A_HINT_ROOT;
300 if (root->op_flags & OPf_MOD) {
301 if (rflags & A_HINT_STORE)
302 flags = (A_HINT_STORE|A_HINT_DEREF);
304 if (rflags & (A_HINT_FETCH|A_HINT_KEYS|A_HINT_VALUES))
305 flags = (rflags|A_HINT_DEREF);
310 a_map_update_flags_bottomup(o, 0, 0);
314 flags |= (rflags & A_HINT_NOTIFY);
315 a_map_update_flags_bottomup(o, flags, 0);
317 return oi->flags & A_HINT_ROOT ? 0 : flags;
320 /* ... Inspired from pp_defined() .......................................... */
322 static int a_undef(pTHX_ SV *sv) {
323 #define a_undef(S) a_undef(aTHX_ (S))
324 switch (SvTYPE(sv)) {
328 if (AvMAX(sv) >= 0 || SvGMAGICAL(sv)
329 || (SvRMAGICAL(sv) && mg_find(sv, PERL_MAGIC_tied)))
333 if (HvARRAY(sv) || SvGMAGICAL(sv)
334 || (SvRMAGICAL(sv) && mg_find(sv, PERL_MAGIC_tied)))
346 /* --- PP functions -------------------------------------------------------- */
348 /* Be aware that we restore PL_op->op_ppaddr from the pointer table old_pp
349 * value, another extension might have saved our pp replacement as the ppaddr
350 * for this op, so this doesn't ensure that our function will never be called
351 * again. That's why we don't remove the op info from our map, so that it can
352 * still run correctly if required. */
354 /* ... pp_rv2av ............................................................ */
356 static OP *a_pp_rv2av(pTHX) {
361 oi = a_map_fetch(PL_op);
363 if (oi->flags & A_HINT_DEREF) {
365 /* We always need to push an empty array to fool the pp_aelem() that comes
369 av = sv_2mortal((SV *) newAV());
375 return oi->old_pp(aTHX);
378 /* ... pp_rv2hv ............................................................ */
380 static OP *a_pp_rv2hv_simple(pTHX) {
385 oi = a_map_fetch(PL_op);
387 if (oi->flags & A_HINT_DEREF) {
392 return oi->old_pp(aTHX);
395 static OP *a_pp_rv2hv(pTHX) {
400 oi = a_map_fetch(PL_op);
402 if (oi->flags & A_HINT_DEREF) {
406 hv = sv_2mortal((SV *) newHV());
412 return oi->old_pp(aTHX);
415 #if A_HAS_SCALARKEYS_OPT
417 static OP *a_pp_rv2hv_dokeys(pTHX) {
422 oi = a_map_fetch(PL_op);
424 if (oi->flags & A_HINT_KEYS) {
433 return oi->old_pp(aTHX);
438 /* ... pp_deref (aelem,helem,rv2sv,padsv) .................................. */
440 static void a_cannot_vivify(pTHX_ UV flags) {
441 #define a_cannot_vivify(F) a_cannot_vivify(aTHX_ (F))
442 if (flags & A_HINT_STRICT)
443 croak("Reference vivification forbidden");
444 else if (flags & A_HINT_WARN)
445 warn("Reference was vivified");
446 else /* A_HINT_STORE */
447 croak("Can't vivify reference");
450 static OP *a_pp_deref(pTHX) {
456 oi = a_map_fetch(PL_op);
459 if (flags & A_HINT_DEREF) {
462 o = oi->old_pp(aTHX);
464 if (flags & (A_HINT_NOTIFY|A_HINT_STORE)) {
467 a_cannot_vivify(flags);
473 return oi->old_pp(aTHX);
476 /* ... pp_root (exists,delete,keys,values) ................................. */
478 static OP *a_pp_root_unop(pTHX) {
483 /* Can only be reached by keys or values */
484 if (GIMME_V == G_SCALAR) {
493 const a_op_info *oi = a_map_fetch(PL_op);
494 return oi->old_pp(aTHX);
498 static OP *a_pp_root_binop(pTHX) {
501 if (a_undef(TOPm1s)) {
504 if (PL_op->op_type == OP_EXISTS)
512 const a_op_info *oi = a_map_fetch(PL_op);
513 return oi->old_pp(aTHX);
519 /* ... pp_multideref ....................................................... */
521 /* This pp replacement is actually only called for topmost exists/delete ops,
522 * because we hijack the [ah]elem check functions and this disables the
523 * optimization for lvalue and rvalue dereferencing. In particular, the
524 * OPf_MOD branches should never be covered. In the future, the multideref
525 * optimization might also be disabled for custom exists/delete check functions,
526 * which will make this section unnecessary. However, the code tries to be as
527 * general as possible in case I think of a way to reenable the multideref
528 * optimization even when this module is in use. */
530 static UV a_do_multideref(const OP *o, UV flags) {
531 UV isexdel, other_flags;
533 XSH_ASSERT(o->op_type == OP_MULTIDEREF);
535 other_flags = flags & ~A_HINT_DO;
537 isexdel = o->op_private & (OPpMULTIDEREF_EXISTS|OPpMULTIDEREF_DELETE);
539 if (isexdel & OPpMULTIDEREF_EXISTS) {
540 flags &= A_HINT_EXISTS;
542 flags &= A_HINT_DELETE;
545 if (o->op_flags & OPf_MOD) {
546 flags &= A_HINT_STORE;
548 flags &= A_HINT_FETCH;
552 return flags ? (flags | other_flags) : 0;
555 static SV *a_do_fake_pp(pTHX_ OP *op) {
556 #define a_do_fake_pp(O) a_do_fake_pp(aTHX_ (O))
562 PL_op->op_ppaddr(aTHX);
576 static void a_do_fake_pp_unop_init(pTHX_ UNOP *unop, U32 type, U32 flags) {
577 #define a_do_fake_pp_unop_init(O, T, F) a_do_fake_pp_unop_init(aTHX_ (O), (T), (F))
578 unop->op_type = type;
579 unop->op_flags = OPf_WANT_SCALAR | (~OPf_WANT & flags);
580 unop->op_private = 0;
581 unop->op_first = NULL;
582 unop->op_ppaddr = PL_ppaddr[type];
585 static SV *a_do_fake_pp_unop_arg1(pTHX_ U32 type, U32 flags, SV *arg) {
586 #define a_do_fake_pp_unop_arg1(T, F, A) a_do_fake_pp_unop_arg1(aTHX_ (T), (F), (A))
590 a_do_fake_pp_unop_init(&unop, type, flags);
596 return a_do_fake_pp((OP *) &unop);
599 static SV *a_do_fake_pp_unop_arg2(pTHX_ U32 type, U32 flags, SV *arg1, SV *arg2) {
600 #define a_do_fake_pp_unop_arg2(T, F, A1, A2) a_do_fake_pp_unop_arg2(aTHX_ (T), (F), (A1), (A2))
604 a_do_fake_pp_unop_init(&unop, type, flags);
611 return a_do_fake_pp((OP *) &unop);
614 #define a_do_pp_rv2av(R) a_do_fake_pp_unop_arg1(OP_RV2AV, OPf_REF, (R))
615 #define a_do_pp_afetch(A, I) a_do_fake_pp_unop_arg2(OP_AELEM, 0, (A), (I))
616 #define a_do_pp_afetch_lv(A, I) a_do_fake_pp_unop_arg2(OP_AELEM, OPf_MOD, (A), (I))
617 #define a_do_pp_aexists(A, I) a_do_fake_pp_unop_arg2(OP_EXISTS, OPf_SPECIAL, (A), (I))
618 #define a_do_pp_adelete(A, I) a_do_fake_pp_unop_arg2(OP_DELETE, OPf_SPECIAL, (A), (I))
620 #define a_do_pp_rv2hv(R) a_do_fake_pp_unop_arg1(OP_RV2HV, OPf_REF, (R))
621 #define a_do_pp_hfetch(H, K) a_do_fake_pp_unop_arg2(OP_HELEM, 0, (H), (K))
622 #define a_do_pp_hfetch_lv(H, K) a_do_fake_pp_unop_arg2(OP_HELEM, OPf_MOD, (H), (K))
623 #define a_do_pp_hexists(H, K) a_do_fake_pp_unop_arg2(OP_EXISTS, 0, (H), (K))
624 #define a_do_pp_hdelete(H, K) a_do_fake_pp_unop_arg2(OP_DELETE, 0, (H), (K))
626 static OP *a_pp_multideref(pTHX) {
627 UNOP_AUX_item *items;
635 const a_op_info *oi = a_map_fetch(PL_op);
637 flags = a_do_multideref(PL_op, oi->flags);
639 return oi->old_pp(aTHX);
642 items = cUNOP_AUXx(PL_op)->op_aux;
645 PL_multideref_pc = items;
648 switch (actions & MDEREF_ACTION_MASK) {
650 actions = (++items)->uv;
652 case MDEREF_AV_padav_aelem: /* $lex[...] */
653 sv = PAD_SVl((++items)->pad_offset);
657 case MDEREF_AV_gvav_aelem: /* $pkg[...] */
658 sv = UNOP_AUX_item_sv(++items);
659 XSH_ASSERT(isGV_with_GP(sv));
660 sv = (SV *) GvAVn((GV *) sv);
664 case MDEREF_AV_pop_rv2av_aelem: /* expr->[...] */
668 goto do_AV_rv2av_aelem;
669 case MDEREF_AV_gvsv_vivify_rv2av_aelem: /* $pkg->[...] */
670 sv = UNOP_AUX_item_sv(++items);
671 XSH_ASSERT(isGV_with_GP(sv));
672 sv = GvSVn((GV *) sv);
675 goto do_AV_vivify_rv2av_aelem;
676 case MDEREF_AV_padsv_vivify_rv2av_aelem: /* $lex->[...] */
677 sv = PAD_SVl((++items)->pad_offset);
679 case MDEREF_AV_vivify_rv2av_aelem: /* vivify, ->[...] */
682 do_AV_vivify_rv2av_aelem:
683 sv = a_vivify_ref(sv, 0);
685 sv = a_do_pp_rv2av(sv);
689 XSH_ASSERT(SvTYPE(sv) == SVt_PVAV);
690 switch (actions & MDEREF_INDEX_MASK) {
691 case MDEREF_INDEX_none:
693 case MDEREF_INDEX_const:
694 esv = sv_2mortal(newSViv((++items)->iv));
696 case MDEREF_INDEX_padsv:
697 esv = PAD_SVl((++items)->pad_offset);
699 case MDEREF_INDEX_gvsv:
700 esv = UNOP_AUX_item_sv(++items);
701 XSH_ASSERT(isGV_with_GP(esv));
702 esv = GvSVn((GV *) esv);
704 if (UNLIKELY(SvROK(esv) && !SvGAMAGIC(esv) && ckWARN(WARN_MISC)))
705 Perl_warner(aTHX_ packWARN(WARN_MISC),
706 "Use of reference \"%"SVf"\" as array index",
710 PL_multideref_pc = items;
711 if (actions & MDEREF_FLAG_last) {
712 switch (flags & A_HINT_DO) {
714 sv = a_do_pp_afetch(sv, esv);
717 sv = a_do_pp_afetch_lv(sv, esv);
720 sv = a_do_pp_aexists(sv, esv);
723 sv = a_do_pp_adelete(sv, esv);
728 sv = a_do_pp_afetch(sv, esv);
731 case MDEREF_HV_padhv_helem: /* $lex{...} */
732 sv = PAD_SVl((++items)->pad_offset);
736 case MDEREF_HV_gvhv_helem: /* $pkg{...} */
737 sv = UNOP_AUX_item_sv(++items);
738 XSH_ASSERT(isGV_with_GP(sv));
739 sv = (SV *) GvHVn((GV *) sv);
743 case MDEREF_HV_pop_rv2hv_helem: /* expr->{...} */
747 goto do_HV_rv2hv_helem;
748 case MDEREF_HV_gvsv_vivify_rv2hv_helem: /* $pkg->{...} */
749 sv = UNOP_AUX_item_sv(++items);
750 XSH_ASSERT(isGV_with_GP(sv));
751 sv = GvSVn((GV *) sv);
754 goto do_HV_vivify_rv2hv_helem;
755 case MDEREF_HV_padsv_vivify_rv2hv_helem: /* $lex->{...} */
756 sv = PAD_SVl((++items)->pad_offset);
758 case MDEREF_HV_vivify_rv2hv_helem: /* vivify, ->{...} */
761 do_HV_vivify_rv2hv_helem:
762 sv = a_vivify_ref(sv, 1);
764 sv = a_do_pp_rv2hv(sv);
768 XSH_ASSERT(SvTYPE(sv) == SVt_PVHV);
769 switch (actions & MDEREF_INDEX_MASK) {
770 case MDEREF_INDEX_none:
772 case MDEREF_INDEX_const:
773 key = UNOP_AUX_item_sv(++items);
775 case MDEREF_INDEX_padsv:
776 key = PAD_SVl((++items)->pad_offset);
778 case MDEREF_INDEX_gvsv:
779 key = UNOP_AUX_item_sv(++items);
780 XSH_ASSERT(isGV_with_GP(key));
781 key = GvSVn((GV *) key);
784 PL_multideref_pc = items;
785 if (actions & MDEREF_FLAG_last) {
786 switch (flags & A_HINT_DO) {
788 sv = a_do_pp_hfetch(sv, key);
791 sv = a_do_pp_hfetch_lv(sv, key);
794 sv = a_do_pp_hexists(sv, key);
797 sv = a_do_pp_hdelete(sv, key);
804 sv = a_do_pp_hfetch(sv, key);
809 actions >>= MDEREF_SHIFT;
813 if (flags & (A_HINT_NOTIFY|A_HINT_STORE))
814 a_cannot_vivify(flags);
815 if (flags & A_HINT_EXISTS)
824 #endif /* A_HAS_MULTIDEREF */
826 /* --- Check functions ----------------------------------------------------- */
828 static void a_recheck_rv2xv(pTHX_ OP *o, OPCODE type, OP *(*new_pp)(pTHX)) {
829 #define a_recheck_rv2xv(O, T, PP) a_recheck_rv2xv(aTHX_ (O), (T), (PP))
831 if (o->op_type == type && o->op_ppaddr != new_pp
832 && cUNOPo->op_first->op_type != OP_GV) {
834 const a_op_info *oi = a_map_fetch(o);
836 a_map_store(o, o->op_ppaddr, oi->next, oi->flags);
837 o->op_ppaddr = new_pp;
844 /* ... ck_pad{any,sv} ...................................................... */
846 /* Sadly, the padsv OPs we are interested in don't trigger the padsv check
847 * function, but are instead manually mutated from a padany. So we store
848 * the op entry in the op map in the padany check function, and we set their
849 * op_ppaddr member in our peephole optimizer replacement below. */
851 static OP *(*a_old_ck_padany)(pTHX_ OP *) = 0;
853 static OP *a_ck_padany(pTHX_ OP *o) {
856 o = a_old_ck_padany(aTHX_ o);
859 if (hint & A_HINT_DO)
860 a_map_store_root(o, o->op_ppaddr, hint);
867 static OP *(*a_old_ck_padsv)(pTHX_ OP *) = 0;
869 static OP *a_ck_padsv(pTHX_ OP *o) {
872 o = a_old_ck_padsv(aTHX_ o);
875 if (hint & A_HINT_DO) {
876 a_map_store_root(o, o->op_ppaddr, hint);
877 o->op_ppaddr = a_pp_deref;
884 /* ... ck_deref (aelem,helem,rv2sv) ........................................ */
886 /* Those ops appear both at the root and inside an expression but there's no
887 * way to distinguish both situations. Worse, we can't even know if we are in a
888 * modifying context, so the expression can't be resolved yet. It will be at the
889 * first invocation of a_pp_deref() for this expression. */
891 static OP *(*a_old_ck_aelem)(pTHX_ OP *) = 0;
892 static OP *(*a_old_ck_helem)(pTHX_ OP *) = 0;
893 static OP *(*a_old_ck_rv2sv)(pTHX_ OP *) = 0;
895 static OP *a_ck_deref(pTHX_ OP *o) {
896 OP * (*old_ck)(pTHX_ OP *o) = 0;
899 switch (o->op_type) {
901 old_ck = a_old_ck_aelem;
902 if ((hint & A_HINT_DO) && !(hint & A_HINT_STRICT))
903 a_recheck_rv2xv(cUNOPo->op_first, OP_RV2AV, a_pp_rv2av);
906 old_ck = a_old_ck_helem;
907 if ((hint & A_HINT_DO) && !(hint & A_HINT_STRICT))
908 a_recheck_rv2xv(cUNOPo->op_first, OP_RV2HV, a_pp_rv2hv_simple);
911 old_ck = a_old_ck_rv2sv;
917 if (old_ck == a_old_ck_rv2sv && o->op_flags & OPf_KIDS) {
918 OP *kid = cUNOPo->op_first;
919 if (kid && kid->op_type == OP_GV) {
920 if (hint & A_HINT_DO)
921 a_map_store(kid, kid->op_ppaddr, NULL, hint);
928 if (hint & A_HINT_DO) {
929 a_map_store_root(o, o->op_ppaddr, hint);
930 o->op_ppaddr = a_pp_deref;
937 /* ... ck_rv2xv (rv2av,rv2hv) .............................................. */
939 /* Those ops also appear both inisde and at the root, hence the caveats for
940 * a_ck_deref() still apply here. Since a padsv/rv2sv must appear before a
941 * rv2[ah]v, resolution is handled by the first call to a_pp_deref() in the
944 static OP *(*a_old_ck_rv2av)(pTHX_ OP *) = 0;
945 static OP *(*a_old_ck_rv2hv)(pTHX_ OP *) = 0;
947 static OP *a_ck_rv2xv(pTHX_ OP *o) {
948 OP * (*old_ck)(pTHX_ OP *o) = 0;
949 OP * (*new_pp)(pTHX) = 0;
952 switch (o->op_type) {
953 case OP_RV2AV: old_ck = a_old_ck_rv2av; new_pp = a_pp_rv2av; break;
954 case OP_RV2HV: old_ck = a_old_ck_rv2hv; new_pp = a_pp_rv2hv_simple; break;
958 if (cUNOPo->op_first->op_type == OP_GV)
962 if (hint & A_HINT_DO && !(hint & A_HINT_STRICT)) {
963 a_map_store_root(o, o->op_ppaddr, hint);
964 o->op_ppaddr = new_pp;
971 /* ... ck_xslice (aslice,hslice) ........................................... */
973 /* I think those are only found at the root, but there's nothing that really
974 * prevent them to be inside the expression too. We only need to update the
975 * root so that the rest of the expression will see the right context when
976 * resolving. That's why we don't replace the ppaddr. */
978 static OP *(*a_old_ck_aslice)(pTHX_ OP *) = 0;
979 static OP *(*a_old_ck_hslice)(pTHX_ OP *) = 0;
981 static OP *a_ck_xslice(pTHX_ OP *o) {
982 OP * (*old_ck)(pTHX_ OP *o) = 0;
985 switch (o->op_type) {
987 old_ck = a_old_ck_aslice;
990 old_ck = a_old_ck_hslice;
991 if (hint & A_HINT_DO)
992 a_recheck_rv2xv(OpSIBLING(cUNOPo->op_first), OP_RV2HV, a_pp_rv2hv);
997 if (hint & A_HINT_DO) {
998 a_map_store_root(o, 0, hint);
1005 /* ... ck_root (exists,delete,keys,values) ................................. */
1007 /* Those ops are only found at the root of a dereferencing expression. We can
1008 * then resolve at compile time if vivification must take place or not. */
1010 static OP *(*a_old_ck_exists)(pTHX_ OP *) = 0;
1011 static OP *(*a_old_ck_delete)(pTHX_ OP *) = 0;
1012 static OP *(*a_old_ck_keys) (pTHX_ OP *) = 0;
1013 static OP *(*a_old_ck_values)(pTHX_ OP *) = 0;
1015 static OP *a_ck_root(pTHX_ OP *o) {
1016 OP * (*old_ck)(pTHX_ OP *o) = 0;
1017 OP * (*new_pp)(pTHX) = 0;
1021 switch (o->op_type) {
1023 old_ck = a_old_ck_exists;
1024 new_pp = a_pp_root_binop;
1025 enabled = hint & A_HINT_EXISTS;
1028 old_ck = a_old_ck_delete;
1029 new_pp = a_pp_root_binop;
1030 enabled = hint & A_HINT_DELETE;
1033 old_ck = a_old_ck_keys;
1034 new_pp = a_pp_root_unop;
1035 enabled = hint & A_HINT_KEYS;
1038 old_ck = a_old_ck_values;
1039 new_pp = a_pp_root_unop;
1040 enabled = hint & A_HINT_VALUES;
1043 o = old_ck(aTHX_ o);
1045 if (hint & A_HINT_DO) {
1047 #if A_HAS_SCALARKEYS_OPT
1048 if ((enabled == A_HINT_KEYS) && (o->op_flags & OPf_KIDS)) {
1049 OP *kid = cUNOPo->op_first;
1050 if (kid->op_type == OP_RV2HV) {
1052 const a_op_info *koi = a_map_fetch(kid);
1053 a_map_store(kid, koi ? koi->old_pp : kid->op_ppaddr, NULL,
1054 hint | A_HINT_SECOND);
1056 kid->op_ppaddr = a_pp_rv2hv;
1060 a_map_update_flags_topdown(o, A_HINT_SECOND, hint | A_HINT_DEREF);
1061 a_map_store_root(o, o->op_ppaddr, hint);
1062 o->op_ppaddr = new_pp;
1064 a_map_update_flags_topdown(o, 0, 0);
1072 /* --- Our peephole optimizer ---------------------------------------------- */
1074 static void xsh_peep_rec(pTHX_ OP *o, ptable *seen) {
1075 for (; o; o = o->op_next) {
1077 const a_op_info *oi = NULL;
1080 if (xsh_peep_seen(o, seen))
1083 switch (o->op_type) {
1085 if (o->op_ppaddr != a_pp_deref) {
1086 oi = a_map_fetch(o);
1087 if (oi && (oi->flags & A_HINT_DO)) {
1088 a_map_store(o, o->op_ppaddr, oi->next, oi->flags);
1089 o->op_ppaddr = a_pp_deref;
1097 if (o->op_ppaddr != a_pp_deref)
1099 oi = a_map_fetch(o);
1103 if (!(flags & A_HINT_DEREF)
1104 && (flags & A_HINT_DO)
1105 && (o->op_private & OPpDEREF || flags & A_HINT_ROOT)) {
1106 /* Decide if the expression must autovivify or not. */
1107 flags = a_map_resolve(o, oi);
1109 if (flags & A_HINT_DEREF)
1110 o->op_private = ((o->op_private & ~OPpDEREF) | OPpLVAL_DEFER);
1112 o->op_ppaddr = oi->old_pp;
1115 if (o->op_ppaddr != a_pp_rv2av)
1117 oi = a_map_fetch(o);
1120 if (!(oi->flags & A_HINT_DEREF))
1121 o->op_ppaddr = oi->old_pp;
1124 if (o->op_ppaddr != a_pp_rv2hv && o->op_ppaddr != a_pp_rv2hv_simple)
1126 oi = a_map_fetch(o);
1129 if (!(oi->flags & A_HINT_DEREF)) {
1130 o->op_ppaddr = oi->old_pp;
1133 #if A_HAS_SCALARKEYS_OPT
1135 if ((flags & A_HINT_KEYS) && (flags & A_HINT_SECOND)) {
1136 U8 want = o->op_flags & OPf_WANT;
1137 if (want == OPf_WANT_VOID || want == OPf_WANT_SCALAR)
1138 o->op_ppaddr = a_pp_rv2hv_dokeys;
1139 else if (oi->old_pp == a_pp_rv2hv || oi->old_pp == a_pp_rv2hv_simple)
1140 o->op_ppaddr = oi->old_pp;
1144 #if A_HAS_MULTIDEREF
1146 if (o->op_ppaddr != a_pp_multideref) {
1147 oi = a_map_fetch(cUNOPo->op_first);
1151 if (a_do_multideref(o, flags)) {
1152 a_map_store_root(o, o->op_ppaddr, flags & ~A_HINT_DEREF);
1153 o->op_ppaddr = a_pp_multideref;
1159 xsh_peep_maybe_recurse(o, seen);
1165 /* --- Module setup/teardown ----------------------------------------------- */
1167 static void xsh_user_global_setup(pTHX) {
1168 a_op_map = ptable_new(32);
1171 MUTEX_INIT(&a_op_map_mutex);
1174 xsh_ck_replace(OP_PADANY, a_ck_padany, &a_old_ck_padany);
1175 xsh_ck_replace(OP_PADSV, a_ck_padsv, &a_old_ck_padsv);
1177 xsh_ck_replace(OP_AELEM, a_ck_deref, &a_old_ck_aelem);
1178 xsh_ck_replace(OP_HELEM, a_ck_deref, &a_old_ck_helem);
1179 xsh_ck_replace(OP_RV2SV, a_ck_deref, &a_old_ck_rv2sv);
1181 xsh_ck_replace(OP_RV2AV, a_ck_rv2xv, &a_old_ck_rv2av);
1182 xsh_ck_replace(OP_RV2HV, a_ck_rv2xv, &a_old_ck_rv2hv);
1184 xsh_ck_replace(OP_ASLICE, a_ck_xslice, &a_old_ck_aslice);
1185 xsh_ck_replace(OP_HSLICE, a_ck_xslice, &a_old_ck_hslice);
1187 xsh_ck_replace(OP_EXISTS, a_ck_root, &a_old_ck_exists);
1188 xsh_ck_replace(OP_DELETE, a_ck_root, &a_old_ck_delete);
1189 xsh_ck_replace(OP_KEYS, a_ck_root, &a_old_ck_keys);
1190 xsh_ck_replace(OP_VALUES, a_ck_root, &a_old_ck_values);
1195 static void xsh_user_local_setup(pTHX) {
1198 stash = gv_stashpvn(XSH_PACKAGE, XSH_PACKAGE_LEN, 1);
1199 newCONSTSUB(stash, "A_HINT_STRICT", newSVuv(A_HINT_STRICT));
1200 newCONSTSUB(stash, "A_HINT_WARN", newSVuv(A_HINT_WARN));
1201 newCONSTSUB(stash, "A_HINT_FETCH", newSVuv(A_HINT_FETCH));
1202 newCONSTSUB(stash, "A_HINT_STORE", newSVuv(A_HINT_STORE));
1203 newCONSTSUB(stash, "A_HINT_KEYS", newSVuv(A_HINT_KEYS));
1204 newCONSTSUB(stash, "A_HINT_VALUES", newSVuv(A_HINT_VALUES));
1205 newCONSTSUB(stash, "A_HINT_EXISTS", newSVuv(A_HINT_EXISTS));
1206 newCONSTSUB(stash, "A_HINT_DELETE", newSVuv(A_HINT_DELETE));
1207 newCONSTSUB(stash, "A_HINT_MASK", newSVuv(A_HINT_MASK));
1208 newCONSTSUB(stash, "A_THREADSAFE", newSVuv(XSH_THREADSAFE));
1209 newCONSTSUB(stash, "A_FORKSAFE", newSVuv(XSH_FORKSAFE));
1214 static void xsh_user_local_teardown(pTHX) {
1218 static void xsh_user_global_teardown(pTHX) {
1219 xsh_ck_restore(OP_PADANY, &a_old_ck_padany);
1220 xsh_ck_restore(OP_PADSV, &a_old_ck_padsv);
1222 xsh_ck_restore(OP_AELEM, &a_old_ck_aelem);
1223 xsh_ck_restore(OP_HELEM, &a_old_ck_helem);
1224 xsh_ck_restore(OP_RV2SV, &a_old_ck_rv2sv);
1226 xsh_ck_restore(OP_RV2AV, &a_old_ck_rv2av);
1227 xsh_ck_restore(OP_RV2HV, &a_old_ck_rv2hv);
1229 xsh_ck_restore(OP_ASLICE, &a_old_ck_aslice);
1230 xsh_ck_restore(OP_HSLICE, &a_old_ck_hslice);
1232 xsh_ck_restore(OP_EXISTS, &a_old_ck_exists);
1233 xsh_ck_restore(OP_DELETE, &a_old_ck_delete);
1234 xsh_ck_restore(OP_KEYS, &a_old_ck_keys);
1235 xsh_ck_restore(OP_VALUES, &a_old_ck_values);
1237 ptable_map_free(a_op_map);
1241 MUTEX_DESTROY(&a_op_map_mutex);
1247 /* --- XS ------------------------------------------------------------------ */
1249 MODULE = autovivification PACKAGE = autovivification
1267 #endif /* XSH_THREADSAFE */
1273 RETVAL = xsh_hints_tag(SvOK(hint) ? SvUV(hint) : 0);
1283 RETVAL = newSVuv(xsh_hints_detag(tag));