X-Git-Url: http://git.vpit.fr/?p=perl%2Fmodules%2Findirect.git;a=blobdiff_plain;f=indirect.xs;h=e144b02a4febf353155eefc2f432ba7734c5cf4d;hp=1f1ab9a69b3c9642dc37da47da2a615e1a147af1;hb=8fc94cd143b3b33a9642be69b17389375208dbcf;hpb=07de00639dbad8bec6c83de81c3e92336ca85569 diff --git a/indirect.xs b/indirect.xs index 1f1ab9a..e144b02 100644 --- a/indirect.xs +++ b/indirect.xs @@ -18,6 +18,10 @@ # define SvPVX_const SvPVX #endif +#ifndef sv_catpvn_nomg +# define sv_catpvn_nomg sv_catpvn +#endif + /* ... Hints ............................................................... */ STATIC U32 indirect_hash = 0; @@ -62,8 +66,7 @@ STATIC void indirect_map_store(pTHX_ const OP *o, const char *src, SV *sv) { SvUVX(val) = PTR2UV(src); SvIOK_on(val); SvIsUV_on(val); - if (!hv_store(indirect_map, buf, OP2STR(o), val, 0)) - SvREFCNT_dec(val); + if (!hv_store(indirect_map, buf, OP2STR(o), val, 0)) SvREFCNT_dec(val); } STATIC const char *indirect_map_fetch(pTHX_ const OP *o, SV ** const name) { @@ -109,8 +112,13 @@ STATIC const char *indirect_find(pTHX_ SV *sv, const char *s) { STRLEN len; const char *p = NULL, *r = SvPV_const(sv, len); - if (!len) - return s; + if (len >= 1 && *r == '$') { + ++r; + --len; + s = strchr(s, '$'); + if (!s) + return NULL; + } p = strstr(s, r); while (p) { @@ -146,16 +154,59 @@ STATIC OP *(*indirect_old_ck_rv2sv)(pTHX_ OP *) = 0; STATIC OP *indirect_ck_rv2sv(pTHX_ OP *o) { if (indirect_hint()) { OP *op = cUNOPo->op_first; - SV *name = cSVOPx_sv(op); - if (SvPOK(name) && (SvTYPE(name) >= SVt_PV)) { - SV *sv = sv_2mortal(newSVpvn("$", 1)); - sv_catsv(sv, name); - o = CALL_FPTR(indirect_old_ck_rv2sv)(aTHX_ o); - indirect_map_store(o, indirect_find(sv, PL_parser->oldbufptr), sv); - return o; + SV *sv; + const char *name = NULL, *s; + STRLEN len; + + switch (op->op_type) { + case OP_GV: + case OP_GVSV: { + GV *gv = cGVOPx_gv(op); + name = GvNAME(gv); + len = GvNAMELEN(gv); + break; + } + default: { + SV *sv = cSVOPx_sv(op); + if (SvPOK(sv) && (SvTYPE(sv) >= SVt_PV)) + name = SvPV_const(sv, len); + break; + } + } + if (!name) + goto done; + + sv = sv_2mortal(newSVpvn("$", 1)); + sv_catpvn_nomg(sv, name, len); + s = indirect_find(sv, PL_parser->oldbufptr); + if (!s) { /* If it failed, retry without the current stash */ + const char *stash = HvNAME_get(PL_curstash); + STRLEN stashlen = HvNAMELEN_get(PL_curstash); + + if ((len < stashlen + 2) || name != strstr(name, stash)) { + /* Failed again ? Try to remove main */ + stash = "main"; + stashlen = 4; + if ((len < stashlen + 2) || name != strstr(name, stash)) + goto done; + } + if (name[stashlen] != ':' || name[stashlen+1] != ':') + goto done; + + sv_setpvn(sv, "$", 1); + stashlen += 2; + sv_catpvn_nomg(sv, name + stashlen, len - stashlen); + s = indirect_find(sv, PL_parser->oldbufptr); + if (!s) + goto done; } + + o = CALL_FPTR(indirect_old_ck_rv2sv)(aTHX_ o); + indirect_map_store(o, s, sv); + return o; } +done: return CALL_FPTR(indirect_old_ck_rv2sv)(aTHX_ o); } @@ -171,10 +222,13 @@ STATIC OP *indirect_ck_padany(pTHX_ OP *o) { const char *s = PL_parser->oldbufptr, *t = PL_parser->bufptr - 1; while (s < t && isSPACE(*s)) ++s; - while (t > s && isSPACE(*t)) --t; - sv = sv_2mortal(newSVpvn(s, t - s + 1)); - - indirect_map_store(o, s, sv); + if (*s == '$' && ++s <= t) { + while (s < t && isSPACE(*s)) ++s; + while (s < t && isSPACE(*t)) --t; + sv = sv_2mortal(newSVpvn("$", 1)); + sv_catpvn_nomg(sv, s, t - s + 1); + indirect_map_store(o, s, sv); + } } return o;