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_EXISTS 16
26 #define A_HINT_DELETE 32
27 #define A_HINT_NOTIFY (A_HINT_STRICT|A_HINT_WARN)
28 #define A_HINT_DO (A_HINT_FETCH|A_HINT_STORE|A_HINT_EXISTS|A_HINT_DELETE)
29 #define A_HINT_MASK (A_HINT_NOTIFY|A_HINT_DO)
31 /* Only used in op flags */
32 #define A_HINT_ROOT 64
33 #define A_HINT_DEREF 128
35 #define XSH_HINTS_TYPE_UV 1
37 #include "xsh/hints.h"
39 #define a_hint() xsh_hints_detag(xsh_hints_fetch())
41 /* ... Thread-local storage ................................................ */
43 #define XSH_THREADS_COMPILE_TIME_PROTECTION 1
44 #define XSH_THREADS_USER_CONTEXT 0
46 #include "xsh/threads.h"
48 /* --- Compatibility wrappers ---------------------------------------------- */
51 # define HvNAME_get(H) HvNAME(H)
55 # define HvNAMELEN_get(H) strlen(HvNAME_get(H))
58 #ifndef A_HAS_MULTIDEREF
59 # define A_HAS_MULTIDEREF XSH_HAS_PERL(5, 21, 7)
62 /* ... Our vivify_ref() .................................................... */
64 /* Perl_vivify_ref() is not exported, so we have to reimplement it. */
68 static SV *a_vivify_ref(pTHX_ SV *sv, int to_hash) {
69 #define a_vivify_ref(S, TH) a_vivify_ref(aTHX_ (S), (TH))
76 Perl_croak_no_modify();
78 /* Inlined prepare_SV_for_RV() */
79 if (SvTYPE(sv) < SVt_PV && SvTYPE(sv) != SVt_IV) {
80 sv_upgrade(sv, SVt_IV);
81 } else if (SvTYPE(sv) >= SVt_PV) {
87 val = to_hash ? MUTABLE_SV(newHV()) : MUTABLE_SV(newAV());
95 SV *msv = sv_newmortal();
96 sv_setsv_nomg(msv, sv);
103 #endif /* A_HAS_MULTIDEREF */
105 /* --- op => info map ------------------------------------------------------ */
113 #define PTABLE_NAME ptable_map
114 #define PTABLE_VAL_FREE(V) PerlMemShared_free(V)
115 #define PTABLE_VAL_NEED_CONTEXT 0
116 #define PTABLE_NEED_DELETE 1
117 #define PTABLE_NEED_WALK 0
119 #include "xsh/ptable.h"
121 #define ptable_map_store(T, K, V) ptable_map_store(aPMS_ (T), (K), (V))
122 #define ptable_map_delete(T, K) ptable_map_delete(aPMS_ (T), (K))
123 #define ptable_map_free(T) ptable_map_free(aPMS_ (T))
125 static ptable *a_op_map = NULL;
129 #define dA_MAP_THX a_op_info a_op_map_tmp_oi
131 static perl_mutex a_op_map_mutex;
133 static const a_op_info *a_map_fetch(const OP *o, a_op_info *oi) {
134 const a_op_info *val;
136 XSH_LOCK(&a_op_map_mutex);
138 val = ptable_fetch(a_op_map, o);
144 XSH_UNLOCK(&a_op_map_mutex);
149 #define a_map_fetch(O) a_map_fetch((O), &a_op_map_tmp_oi)
151 #else /* USE_ITHREADS */
153 #define dA_MAP_THX dNOOP
155 #define a_map_fetch(O) ptable_fetch(a_op_map, (O))
157 #endif /* !USE_ITHREADS */
159 static const a_op_info *a_map_store_locked(pPMS_ const OP *o, OP *(*old_pp)(pTHX), void *next, UV flags) {
160 #define a_map_store_locked(O, PP, N, F) a_map_store_locked(aPMS_ (O), (PP), (N), (F))
163 if (!(oi = ptable_fetch(a_op_map, o))) {
164 oi = PerlMemShared_malloc(sizeof *oi);
165 ptable_map_store(a_op_map, o, oi);
175 static void a_map_store(pTHX_ const OP *o, OP *(*old_pp)(pTHX), void *next, UV flags) {
176 #define a_map_store(O, PP, N, F) a_map_store(aTHX_ (O), (PP), (N), (F))
177 XSH_LOCK(&a_op_map_mutex);
179 a_map_store_locked(o, old_pp, next, flags);
181 XSH_UNLOCK(&a_op_map_mutex);
184 static void a_map_delete(pTHX_ const OP *o) {
185 #define a_map_delete(O) a_map_delete(aTHX_ (O))
186 XSH_LOCK(&a_op_map_mutex);
188 ptable_map_delete(a_op_map, o);
190 XSH_UNLOCK(&a_op_map_mutex);
193 static const OP *a_map_descend(const OP *o) {
194 switch (PL_opargs[o->op_type] & OA_CLASS_MASK) {
198 case OA_BASEOP_OR_UNOP:
199 return cUNOPo->op_first;
202 return cLISTOPo->op_last;
208 static void a_map_store_root(pTHX_ const OP *root, OP *(*old_pp)(pTHX), UV flags) {
209 #define a_map_store_root(R, PP, F) a_map_store_root(aTHX_ (R), (PP), (F))
210 const a_op_info *roi;
214 XSH_LOCK(&a_op_map_mutex);
216 roi = a_map_store_locked(o, old_pp, (OP *) root, flags | A_HINT_ROOT);
218 while (o->op_flags & OPf_KIDS) {
219 o = a_map_descend(o);
222 if ((oi = ptable_fetch(a_op_map, o))) {
223 oi->flags &= ~A_HINT_ROOT;
224 oi->next = (a_op_info *) roi;
229 XSH_UNLOCK(&a_op_map_mutex);
234 static void a_map_update_flags_topdown(const OP *root, UV flags) {
238 XSH_LOCK(&a_op_map_mutex);
240 flags &= ~A_HINT_ROOT;
243 if ((oi = ptable_fetch(a_op_map, o)))
244 oi->flags = (oi->flags & A_HINT_ROOT) | flags;
245 if (!(o->op_flags & OPf_KIDS))
247 o = a_map_descend(o);
250 XSH_UNLOCK(&a_op_map_mutex);
255 #define a_map_cancel(R) a_map_update_flags_topdown((R), 0)
257 static void a_map_update_flags_bottomup(const OP *o, UV flags, UV rflags) {
260 XSH_LOCK(&a_op_map_mutex);
262 flags &= ~A_HINT_ROOT;
263 rflags |= A_HINT_ROOT;
265 oi = ptable_fetch(a_op_map, o);
266 while (!(oi->flags & A_HINT_ROOT)) {
272 XSH_UNLOCK(&a_op_map_mutex);
277 /* ... Decide whether this expression should be autovivified or not ........ */
279 static UV a_map_resolve(const OP *o, const a_op_info *oi) {
280 UV flags = 0, rflags;
282 const a_op_info *roi = oi;
284 while (!(roi->flags & A_HINT_ROOT))
289 rflags = roi->flags & ~A_HINT_ROOT;
294 if (root->op_flags & OPf_MOD) {
295 if (rflags & A_HINT_STORE)
296 flags = (A_HINT_STORE|A_HINT_DEREF);
297 } else if (rflags & A_HINT_FETCH)
298 flags = (A_HINT_FETCH|A_HINT_DEREF);
302 a_map_update_flags_bottomup(o, 0, 0);
306 flags |= (rflags & A_HINT_NOTIFY);
307 a_map_update_flags_bottomup(o, flags, 0);
309 return oi->flags & A_HINT_ROOT ? 0 : flags;
312 /* ... Inspired from pp_defined() .......................................... */
314 static int a_undef(pTHX_ SV *sv) {
315 #define a_undef(S) a_undef(aTHX_ (S))
316 switch (SvTYPE(sv)) {
320 if (AvMAX(sv) >= 0 || SvGMAGICAL(sv)
321 || (SvRMAGICAL(sv) && mg_find(sv, PERL_MAGIC_tied)))
325 if (HvARRAY(sv) || SvGMAGICAL(sv)
326 || (SvRMAGICAL(sv) && mg_find(sv, PERL_MAGIC_tied)))
338 /* --- PP functions -------------------------------------------------------- */
340 /* Be aware that we restore PL_op->op_ppaddr from the pointer table old_pp
341 * value, another extension might have saved our pp replacement as the ppaddr
342 * for this op, so this doesn't ensure that our function will never be called
343 * again. That's why we don't remove the op info from our map, so that it can
344 * still run correctly if required. */
348 static UV a_cop_hint(pTHX_ COP *cop) {
349 #define a_cop_hint(C) a_cop_hint(aTHX_ (C))
355 hint = cop_hints_fetch_pvn(cop, XSH_HINTS_KEY, XSH_HINTS_KEY_LEN, xsh_hints_key_hash, 0);
357 return hint ? xsh_hints_detag(hint) : 0;
362 /* ... pp_rv2av ............................................................ */
364 static OP *a_pp_rv2av(pTHX) {
369 oi = a_map_fetch(PL_op);
371 XSH_ASSERT(a_cop_hint(PL_curcop));
372 XSH_ASSERT(oi->flags & A_HINT_DEREF);
373 XSH_ASSERT(oi->flags & A_HINT_DO);
376 /* We always need to push an empty array to fool the pp_aelem() that comes
380 av = sv_2mortal((SV *) newAV());
385 return oi->old_pp(aTHX);
388 /* ... pp_rv2hv ............................................................ */
390 static OP *a_pp_rv2hv_simple(pTHX) {
395 oi = a_map_fetch(PL_op);
397 XSH_ASSERT(a_cop_hint(PL_curcop));
398 XSH_ASSERT(oi->flags & A_HINT_DEREF);
399 XSH_ASSERT(oi->flags & A_HINT_DO);
404 return oi->old_pp(aTHX);
407 static OP *a_pp_rv2hv(pTHX) {
412 oi = a_map_fetch(PL_op);
414 XSH_ASSERT(a_cop_hint(PL_curcop));
415 XSH_ASSERT(oi->flags & A_HINT_DEREF);
416 XSH_ASSERT(oi->flags & (A_HINT_FETCH|A_HINT_STORE));
421 hv = sv_2mortal((SV *) newHV());
426 return oi->old_pp(aTHX);
429 /* ... pp_deref (aelem,helem,rv2sv,padsv) .................................. */
431 static void a_cannot_vivify(pTHX_ UV flags) {
432 #define a_cannot_vivify(F) a_cannot_vivify(aTHX_ (F))
433 if (flags & A_HINT_STRICT)
434 croak("Reference vivification forbidden");
435 else if (flags & A_HINT_WARN)
436 warn("Reference was vivified");
437 else /* A_HINT_STORE */
438 croak("Can't vivify reference");
441 static OP *a_pp_deref(pTHX) {
448 oi = a_map_fetch(PL_op);
450 XSH_ASSERT(a_cop_hint(PL_curcop));
452 XSH_ASSERT(oi->flags & A_HINT_DEREF);
453 XSH_ASSERT(oi->flags & A_HINT_DO);
455 o = oi->old_pp(aTHX);
457 if (flags & (A_HINT_NOTIFY|A_HINT_STORE)) {
460 a_cannot_vivify(flags);
466 /* ... pp_root (exists,delete,keys,values) ................................. */
468 static OP *a_pp_root_unop(pTHX) {
473 /* Can only be reached by keys or values */
474 if (GIMME_V == G_SCALAR) {
483 const a_op_info *oi = a_map_fetch(PL_op);
484 return oi->old_pp(aTHX);
488 static OP *a_pp_root_binop(pTHX) {
491 if (a_undef(TOPm1s)) {
494 if (PL_op->op_type == OP_EXISTS)
502 const a_op_info *oi = a_map_fetch(PL_op);
503 return oi->old_pp(aTHX);
509 /* ... pp_multideref ....................................................... */
511 /* This pp replacement is actually only called for topmost exists/delete ops,
512 * because we hijack the [ah]elem check functions and this disables the
513 * optimization for lvalue and rvalue dereferencing. In particular, the
514 * OPf_MOD branches should never be covered. In the future, the multideref
515 * optimization might also be disabled for custom exists/delete check functions,
516 * which will make this section unnecessary. However, the code tries to be as
517 * general as possible in case I think of a way to reenable the multideref
518 * optimization even when this module is in use. */
520 static UV a_do_multideref(const OP *o, UV flags) {
521 UV isexdel, other_flags;
523 XSH_ASSERT(o->op_type == OP_MULTIDEREF);
525 other_flags = flags & ~A_HINT_DO;
527 isexdel = o->op_private & (OPpMULTIDEREF_EXISTS|OPpMULTIDEREF_DELETE);
529 if (isexdel & OPpMULTIDEREF_EXISTS) {
530 flags &= A_HINT_EXISTS;
532 flags &= A_HINT_DELETE;
535 if (o->op_flags & OPf_MOD) {
536 flags &= A_HINT_STORE;
538 flags &= A_HINT_FETCH;
542 return flags ? (flags | other_flags) : 0;
545 static SV *a_do_fake_pp(pTHX_ OP *op) {
546 #define a_do_fake_pp(O) a_do_fake_pp(aTHX_ (O))
552 PL_op->op_ppaddr(aTHX);
566 static void a_do_fake_pp_unop_init(pTHX_ UNOP *unop, U32 type, U32 flags) {
567 #define a_do_fake_pp_unop_init(O, T, F) a_do_fake_pp_unop_init(aTHX_ (O), (T), (F))
568 unop->op_type = type;
569 unop->op_flags = OPf_WANT_SCALAR | (~OPf_WANT & flags);
570 unop->op_private = 0;
571 unop->op_first = NULL;
572 unop->op_ppaddr = PL_ppaddr[type];
575 static SV *a_do_fake_pp_unop_arg1(pTHX_ U32 type, U32 flags, SV *arg) {
576 #define a_do_fake_pp_unop_arg1(T, F, A) a_do_fake_pp_unop_arg1(aTHX_ (T), (F), (A))
580 a_do_fake_pp_unop_init(&unop, type, flags);
586 return a_do_fake_pp((OP *) &unop);
589 static SV *a_do_fake_pp_unop_arg2(pTHX_ U32 type, U32 flags, SV *arg1, SV *arg2) {
590 #define a_do_fake_pp_unop_arg2(T, F, A1, A2) a_do_fake_pp_unop_arg2(aTHX_ (T), (F), (A1), (A2))
594 a_do_fake_pp_unop_init(&unop, type, flags);
601 return a_do_fake_pp((OP *) &unop);
604 #define a_do_pp_rv2av(R) a_do_fake_pp_unop_arg1(OP_RV2AV, OPf_REF, (R))
605 #define a_do_pp_afetch(A, I) a_do_fake_pp_unop_arg2(OP_AELEM, 0, (A), (I))
606 #define a_do_pp_afetch_lv(A, I) a_do_fake_pp_unop_arg2(OP_AELEM, OPf_MOD, (A), (I))
607 #define a_do_pp_aexists(A, I) a_do_fake_pp_unop_arg2(OP_EXISTS, OPf_SPECIAL, (A), (I))
608 #define a_do_pp_adelete(A, I) a_do_fake_pp_unop_arg2(OP_DELETE, OPf_SPECIAL, (A), (I))
610 #define a_do_pp_rv2hv(R) a_do_fake_pp_unop_arg1(OP_RV2HV, OPf_REF, (R))
611 #define a_do_pp_hfetch(H, K) a_do_fake_pp_unop_arg2(OP_HELEM, 0, (H), (K))
612 #define a_do_pp_hfetch_lv(H, K) a_do_fake_pp_unop_arg2(OP_HELEM, OPf_MOD, (H), (K))
613 #define a_do_pp_hexists(H, K) a_do_fake_pp_unop_arg2(OP_EXISTS, 0, (H), (K))
614 #define a_do_pp_hdelete(H, K) a_do_fake_pp_unop_arg2(OP_DELETE, 0, (H), (K))
616 static OP *a_pp_multideref(pTHX) {
617 UNOP_AUX_item *items;
623 XSH_ASSERT(a_cop_hint(PL_curcop));
627 const a_op_info *oi = a_map_fetch(PL_op);
629 flags = a_do_multideref(PL_op, oi->flags);
630 XSH_ASSERT(flags & A_HINT_DO);
633 items = cUNOP_AUXx(PL_op)->op_aux;
636 PL_multideref_pc = items;
639 switch (actions & MDEREF_ACTION_MASK) {
641 actions = (++items)->uv;
643 case MDEREF_AV_padav_aelem: /* $lex[...] */
644 sv = PAD_SVl((++items)->pad_offset);
648 case MDEREF_AV_gvav_aelem: /* $pkg[...] */
649 sv = UNOP_AUX_item_sv(++items);
650 XSH_ASSERT(isGV_with_GP(sv));
651 sv = (SV *) GvAVn((GV *) sv);
655 case MDEREF_AV_pop_rv2av_aelem: /* expr->[...] */
659 goto do_AV_rv2av_aelem;
660 case MDEREF_AV_gvsv_vivify_rv2av_aelem: /* $pkg->[...] */
661 sv = UNOP_AUX_item_sv(++items);
662 XSH_ASSERT(isGV_with_GP(sv));
663 sv = GvSVn((GV *) sv);
666 goto do_AV_vivify_rv2av_aelem;
667 case MDEREF_AV_padsv_vivify_rv2av_aelem: /* $lex->[...] */
668 sv = PAD_SVl((++items)->pad_offset);
670 case MDEREF_AV_vivify_rv2av_aelem: /* vivify, ->[...] */
673 do_AV_vivify_rv2av_aelem:
674 sv = a_vivify_ref(sv, 0);
676 sv = a_do_pp_rv2av(sv);
680 XSH_ASSERT(SvTYPE(sv) == SVt_PVAV);
681 switch (actions & MDEREF_INDEX_MASK) {
682 case MDEREF_INDEX_none:
684 case MDEREF_INDEX_const:
685 esv = sv_2mortal(newSViv((++items)->iv));
687 case MDEREF_INDEX_padsv:
688 esv = PAD_SVl((++items)->pad_offset);
690 case MDEREF_INDEX_gvsv:
691 esv = UNOP_AUX_item_sv(++items);
692 XSH_ASSERT(isGV_with_GP(esv));
693 esv = GvSVn((GV *) esv);
695 if (UNLIKELY(SvROK(esv) && !SvGAMAGIC(esv) && ckWARN(WARN_MISC)))
696 Perl_warner(aTHX_ packWARN(WARN_MISC),
697 "Use of reference \"%"SVf"\" as array index",
701 PL_multideref_pc = items;
702 if (actions & MDEREF_FLAG_last) {
703 switch (flags & A_HINT_DO) {
705 sv = a_do_pp_afetch(sv, esv);
708 sv = a_do_pp_afetch_lv(sv, esv);
711 sv = a_do_pp_aexists(sv, esv);
714 sv = a_do_pp_adelete(sv, esv);
719 sv = a_do_pp_afetch(sv, esv);
722 case MDEREF_HV_padhv_helem: /* $lex{...} */
723 sv = PAD_SVl((++items)->pad_offset);
727 case MDEREF_HV_gvhv_helem: /* $pkg{...} */
728 sv = UNOP_AUX_item_sv(++items);
729 XSH_ASSERT(isGV_with_GP(sv));
730 sv = (SV *) GvHVn((GV *) sv);
734 case MDEREF_HV_pop_rv2hv_helem: /* expr->{...} */
738 goto do_HV_rv2hv_helem;
739 case MDEREF_HV_gvsv_vivify_rv2hv_helem: /* $pkg->{...} */
740 sv = UNOP_AUX_item_sv(++items);
741 XSH_ASSERT(isGV_with_GP(sv));
742 sv = GvSVn((GV *) sv);
745 goto do_HV_vivify_rv2hv_helem;
746 case MDEREF_HV_padsv_vivify_rv2hv_helem: /* $lex->{...} */
747 sv = PAD_SVl((++items)->pad_offset);
749 case MDEREF_HV_vivify_rv2hv_helem: /* vivify, ->{...} */
752 do_HV_vivify_rv2hv_helem:
753 sv = a_vivify_ref(sv, 1);
755 sv = a_do_pp_rv2hv(sv);
759 XSH_ASSERT(SvTYPE(sv) == SVt_PVHV);
760 switch (actions & MDEREF_INDEX_MASK) {
761 case MDEREF_INDEX_none:
763 case MDEREF_INDEX_const:
764 key = UNOP_AUX_item_sv(++items);
766 case MDEREF_INDEX_padsv:
767 key = PAD_SVl((++items)->pad_offset);
769 case MDEREF_INDEX_gvsv:
770 key = UNOP_AUX_item_sv(++items);
771 XSH_ASSERT(isGV_with_GP(key));
772 key = GvSVn((GV *) key);
775 PL_multideref_pc = items;
776 if (actions & MDEREF_FLAG_last) {
777 switch (flags & A_HINT_DO) {
779 sv = a_do_pp_hfetch(sv, key);
782 sv = a_do_pp_hfetch_lv(sv, key);
785 sv = a_do_pp_hexists(sv, key);
788 sv = a_do_pp_hdelete(sv, key);
795 sv = a_do_pp_hfetch(sv, key);
800 actions >>= MDEREF_SHIFT;
804 if (flags & (A_HINT_NOTIFY|A_HINT_STORE))
805 a_cannot_vivify(flags);
806 if (flags & A_HINT_EXISTS)
815 #endif /* A_HAS_MULTIDEREF */
817 /* --- Check functions ----------------------------------------------------- */
819 static void a_recheck_rv2xv(pTHX_ OP *o, OPCODE type, OP *(*new_pp)(pTHX)) {
820 #define a_recheck_rv2xv(O, T, PP) a_recheck_rv2xv(aTHX_ (O), (T), (PP))
822 if (o->op_type == type && o->op_ppaddr != new_pp
823 && cUNOPo->op_first->op_type != OP_GV) {
825 const a_op_info *oi = a_map_fetch(o);
827 a_map_store(o, o->op_ppaddr, oi->next, oi->flags);
828 o->op_ppaddr = new_pp;
835 /* ... ck_pad{any,sv} ...................................................... */
837 /* Sadly, the padsv OPs we are interested in don't trigger the padsv check
838 * function, but are instead manually mutated from a padany. So we store
839 * the op entry in the op map in the padany check function, and we set their
840 * op_ppaddr member in our peephole optimizer replacement below. */
842 static OP *(*a_old_ck_padany)(pTHX_ OP *) = 0;
844 static OP *a_ck_padany(pTHX_ OP *o) {
847 o = a_old_ck_padany(aTHX_ o);
850 if (hint & A_HINT_DO)
851 a_map_store_root(o, o->op_ppaddr, hint);
858 static OP *(*a_old_ck_padsv)(pTHX_ OP *) = 0;
860 static OP *a_ck_padsv(pTHX_ OP *o) {
863 o = a_old_ck_padsv(aTHX_ o);
866 if (hint & A_HINT_DO) {
867 a_map_store_root(o, o->op_ppaddr, hint);
868 o->op_ppaddr = a_pp_deref;
875 /* ... ck_deref (aelem,helem,rv2sv) ........................................ */
877 /* Those ops appear both at the root and inside an expression but there's no
878 * way to distinguish both situations. Worse, we can't even know if we are in a
879 * modifying context, so the expression can't be resolved yet. It will be at the
880 * first invocation of a_pp_deref() for this expression. */
882 static OP *(*a_old_ck_aelem)(pTHX_ OP *) = 0;
883 static OP *(*a_old_ck_helem)(pTHX_ OP *) = 0;
884 static OP *(*a_old_ck_rv2sv)(pTHX_ OP *) = 0;
886 static OP *a_ck_deref(pTHX_ OP *o) {
887 OP * (*old_ck)(pTHX_ OP *o) = 0;
890 switch (o->op_type) {
892 old_ck = a_old_ck_aelem;
893 if ((hint & A_HINT_DO) && !(hint & A_HINT_STRICT))
894 a_recheck_rv2xv(cUNOPo->op_first, OP_RV2AV, a_pp_rv2av);
897 old_ck = a_old_ck_helem;
898 if ((hint & A_HINT_DO) && !(hint & A_HINT_STRICT))
899 a_recheck_rv2xv(cUNOPo->op_first, OP_RV2HV, a_pp_rv2hv_simple);
902 old_ck = a_old_ck_rv2sv;
908 if (old_ck == a_old_ck_rv2sv && o->op_flags & OPf_KIDS) {
909 OP *kid = cUNOPo->op_first;
910 if (kid && kid->op_type == OP_GV) {
911 if (hint & A_HINT_DO)
912 a_map_store(kid, kid->op_ppaddr, NULL, hint);
919 if (hint & A_HINT_DO) {
920 a_map_store_root(o, o->op_ppaddr, hint);
921 o->op_ppaddr = a_pp_deref;
928 /* ... ck_rv2xv (rv2av,rv2hv) .............................................. */
930 /* Those ops also appear both inisde and at the root, hence the caveats for
931 * a_ck_deref() still apply here. Since a padsv/rv2sv must appear before a
932 * rv2[ah]v, resolution is handled by the first call to a_pp_deref() in the
935 static OP *(*a_old_ck_rv2av)(pTHX_ OP *) = 0;
936 static OP *(*a_old_ck_rv2hv)(pTHX_ OP *) = 0;
938 static OP *a_ck_rv2xv(pTHX_ OP *o) {
939 OP * (*old_ck)(pTHX_ OP *o) = 0;
940 OP * (*new_pp)(pTHX) = 0;
943 switch (o->op_type) {
944 case OP_RV2AV: old_ck = a_old_ck_rv2av; new_pp = a_pp_rv2av; break;
945 case OP_RV2HV: old_ck = a_old_ck_rv2hv; new_pp = a_pp_rv2hv_simple; break;
949 if (cUNOPo->op_first->op_type == OP_GV)
953 if (hint & A_HINT_DO && !(hint & A_HINT_STRICT)) {
954 a_map_store_root(o, o->op_ppaddr, hint);
955 o->op_ppaddr = new_pp;
962 /* ... ck_xslice (aslice,hslice) ........................................... */
964 /* I think those are only found at the root, but there's nothing that really
965 * prevent them to be inside the expression too. We only need to update the
966 * root so that the rest of the expression will see the right context when
967 * resolving. That's why we don't replace the ppaddr. */
969 static OP *(*a_old_ck_aslice)(pTHX_ OP *) = 0;
970 static OP *(*a_old_ck_hslice)(pTHX_ OP *) = 0;
972 static OP *a_ck_xslice(pTHX_ OP *o) {
973 OP * (*old_ck)(pTHX_ OP *o) = 0;
976 switch (o->op_type) {
978 old_ck = a_old_ck_aslice;
981 old_ck = a_old_ck_hslice;
982 if (hint & A_HINT_DO)
983 a_recheck_rv2xv(OpSIBLING(cUNOPo->op_first), OP_RV2HV, a_pp_rv2hv);
988 if (hint & A_HINT_DO) {
989 a_map_store_root(o, 0, hint);
996 /* ... ck_root (exists,delete,keys,values) ................................. */
998 /* Those ops are only found at the root of a dereferencing expression. We can
999 * then resolve at compile time if vivification must take place or not. */
1001 static OP *(*a_old_ck_exists)(pTHX_ OP *) = 0;
1002 static OP *(*a_old_ck_delete)(pTHX_ OP *) = 0;
1003 static OP *(*a_old_ck_keys) (pTHX_ OP *) = 0;
1004 static OP *(*a_old_ck_values)(pTHX_ OP *) = 0;
1006 static OP *a_ck_root(pTHX_ OP *o) {
1007 OP * (*old_ck)(pTHX_ OP *o) = 0;
1008 OP * (*new_pp)(pTHX) = 0;
1009 bool enabled = FALSE;
1012 switch (o->op_type) {
1014 old_ck = a_old_ck_exists;
1015 new_pp = a_pp_root_binop;
1016 enabled = hint & A_HINT_EXISTS;
1019 old_ck = a_old_ck_delete;
1020 new_pp = a_pp_root_binop;
1021 enabled = hint & A_HINT_DELETE;
1024 old_ck = a_old_ck_keys;
1025 new_pp = a_pp_root_unop;
1026 enabled = hint & A_HINT_FETCH;
1029 old_ck = a_old_ck_values;
1030 new_pp = a_pp_root_unop;
1031 enabled = hint & A_HINT_FETCH;
1034 o = old_ck(aTHX_ o);
1036 if (hint & A_HINT_DO) {
1038 a_map_update_flags_topdown(o, hint | A_HINT_DEREF);
1039 a_map_store_root(o, o->op_ppaddr, hint);
1040 o->op_ppaddr = new_pp;
1050 /* --- Our peephole optimizer ---------------------------------------------- */
1052 static void xsh_peep_rec(pTHX_ OP *o, ptable *seen) {
1054 COP *last_cop = NULL;
1057 for (; o; o = o->op_next) {
1059 const a_op_info *oi = NULL;
1062 if (xsh_peep_seen(o, seen))
1065 switch (o->op_type) {
1073 if (o->op_ppaddr != a_pp_deref) {
1074 oi = a_map_fetch(o);
1075 if (oi && (oi->flags & A_HINT_DO)) {
1076 a_map_store(o, o->op_ppaddr, oi->next, oi->flags);
1078 XSH_ASSERT(!last_cop || a_cop_hint(last_cop));
1080 o->op_ppaddr = a_pp_deref;
1088 if (o->op_ppaddr != a_pp_deref)
1090 oi = a_map_fetch(o);
1094 if (!(flags & A_HINT_DEREF)
1095 && (flags & A_HINT_DO)
1096 && (o->op_private & OPpDEREF || flags & A_HINT_ROOT)) {
1097 /* Decide if the expression must autovivify or not. */
1098 flags = a_map_resolve(o, oi);
1100 if (flags & A_HINT_DEREF)
1101 o->op_private = ((o->op_private & ~OPpDEREF) | OPpLVAL_DEFER);
1103 o->op_ppaddr = oi->old_pp;
1107 if ( o->op_ppaddr != a_pp_rv2av
1108 && o->op_ppaddr != a_pp_rv2hv
1109 && o->op_ppaddr != a_pp_rv2hv_simple)
1111 oi = a_map_fetch(o);
1114 if (!(oi->flags & A_HINT_DEREF))
1115 o->op_ppaddr = oi->old_pp;
1117 #if A_HAS_MULTIDEREF
1119 if (o->op_ppaddr != a_pp_multideref) {
1120 oi = a_map_fetch(cUNOPo->op_first);
1124 if (a_do_multideref(o, flags)) {
1125 a_map_store_root(o, o->op_ppaddr, flags & ~A_HINT_DEREF);
1126 o->op_ppaddr = a_pp_multideref;
1132 xsh_peep_maybe_recurse(o, seen);
1138 /* --- Module setup/teardown ----------------------------------------------- */
1140 static void xsh_user_global_setup(pTHX) {
1141 a_op_map = ptable_new(32);
1144 MUTEX_INIT(&a_op_map_mutex);
1147 xsh_ck_replace(OP_PADANY, a_ck_padany, &a_old_ck_padany);
1148 xsh_ck_replace(OP_PADSV, a_ck_padsv, &a_old_ck_padsv);
1150 xsh_ck_replace(OP_AELEM, a_ck_deref, &a_old_ck_aelem);
1151 xsh_ck_replace(OP_HELEM, a_ck_deref, &a_old_ck_helem);
1152 xsh_ck_replace(OP_RV2SV, a_ck_deref, &a_old_ck_rv2sv);
1154 xsh_ck_replace(OP_RV2AV, a_ck_rv2xv, &a_old_ck_rv2av);
1155 xsh_ck_replace(OP_RV2HV, a_ck_rv2xv, &a_old_ck_rv2hv);
1157 xsh_ck_replace(OP_ASLICE, a_ck_xslice, &a_old_ck_aslice);
1158 xsh_ck_replace(OP_HSLICE, a_ck_xslice, &a_old_ck_hslice);
1160 xsh_ck_replace(OP_EXISTS, a_ck_root, &a_old_ck_exists);
1161 xsh_ck_replace(OP_DELETE, a_ck_root, &a_old_ck_delete);
1162 xsh_ck_replace(OP_KEYS, a_ck_root, &a_old_ck_keys);
1163 xsh_ck_replace(OP_VALUES, a_ck_root, &a_old_ck_values);
1168 static void xsh_user_local_setup(pTHX) {
1171 stash = gv_stashpvn(XSH_PACKAGE, XSH_PACKAGE_LEN, 1);
1172 newCONSTSUB(stash, "A_HINT_STRICT", newSVuv(A_HINT_STRICT));
1173 newCONSTSUB(stash, "A_HINT_WARN", newSVuv(A_HINT_WARN));
1174 newCONSTSUB(stash, "A_HINT_FETCH", newSVuv(A_HINT_FETCH));
1175 newCONSTSUB(stash, "A_HINT_STORE", newSVuv(A_HINT_STORE));
1176 newCONSTSUB(stash, "A_HINT_EXISTS", newSVuv(A_HINT_EXISTS));
1177 newCONSTSUB(stash, "A_HINT_DELETE", newSVuv(A_HINT_DELETE));
1178 newCONSTSUB(stash, "A_HINT_MASK", newSVuv(A_HINT_MASK));
1179 newCONSTSUB(stash, "A_THREADSAFE", newSVuv(XSH_THREADSAFE));
1180 newCONSTSUB(stash, "A_FORKSAFE", newSVuv(XSH_FORKSAFE));
1185 static void xsh_user_local_teardown(pTHX) {
1189 static void xsh_user_global_teardown(pTHX) {
1190 xsh_ck_restore(OP_PADANY, &a_old_ck_padany);
1191 xsh_ck_restore(OP_PADSV, &a_old_ck_padsv);
1193 xsh_ck_restore(OP_AELEM, &a_old_ck_aelem);
1194 xsh_ck_restore(OP_HELEM, &a_old_ck_helem);
1195 xsh_ck_restore(OP_RV2SV, &a_old_ck_rv2sv);
1197 xsh_ck_restore(OP_RV2AV, &a_old_ck_rv2av);
1198 xsh_ck_restore(OP_RV2HV, &a_old_ck_rv2hv);
1200 xsh_ck_restore(OP_ASLICE, &a_old_ck_aslice);
1201 xsh_ck_restore(OP_HSLICE, &a_old_ck_hslice);
1203 xsh_ck_restore(OP_EXISTS, &a_old_ck_exists);
1204 xsh_ck_restore(OP_DELETE, &a_old_ck_delete);
1205 xsh_ck_restore(OP_KEYS, &a_old_ck_keys);
1206 xsh_ck_restore(OP_VALUES, &a_old_ck_values);
1208 ptable_map_free(a_op_map);
1212 MUTEX_DESTROY(&a_op_map_mutex);
1218 /* --- XS ------------------------------------------------------------------ */
1220 MODULE = autovivification PACKAGE = autovivification
1238 #endif /* XSH_THREADSAFE */
1244 RETVAL = xsh_hints_tag(SvOK(hint) ? SvUV(hint) : 0);
1254 RETVAL = newSVuv(xsh_hints_detag(tag));