test for renaming lexical subs

[p5sagit/Sub-Name.git] / Name.xs

diff --git a/Name.xs b/Name.xs
index 71f516c..9a4d493 100644 (file)
--- a/Name.xs
+++ b/Name.xs
@@ -37,11 +37,12 @@ subname(name, sub)
        CV *cv = NULL;
        GV *gv;
        HV *stash = CopSTASH(PL_curcop);
-       const char *s, *end = NULL;
+       const char *s, *end = NULL, *begin = NULL;
        MAGIC *mg;
        STRLEN namelen;
-       int utf8flag = SvUTF8(name);
        const char* nameptr = SvPV(name, namelen);
+       int utf8flag = SvUTF8(name);
+       int seen_quote = 0, need_subst = 0;
     PPCODE:
        if (!SvROK(sub) && SvGMAGICAL(sub))
                mg_get(sub);
@@ -62,18 +63,47 @@ subname(name, sub)
                croak("Not a subroutine reference");
 
        for (s = nameptr; s <= nameptr + namelen; s++) {
-               if (*s == ':' && s[-1] == ':')
-                       end = ++s;
-               else if (*s && s[-1] == '\'')
-                       end = s;
+               if (s > nameptr && *s == ':' && s[-1] == ':') {
+                       end = s - 1;
+                       begin = ++s;
+                       if (seen_quote)
+                               need_subst++;
+               }
+               else if (s > nameptr && *s != '\0' && s[-1] == '\'') {
+                       end = s - 1;
+                       begin = s;
+                       if (seen_quote++)
+                               need_subst++;
+               }
        }
        s--;
        if (end) {
-               stash = GvHV(gv_fetchpvn_flags(nameptr, end - nameptr, GV_ADD | utf8flag, SVt_PVHV));
-               nameptr = end;
-               namelen -= end - nameptr;
+               SV* tmp;
+               if (need_subst) {
+                       STRLEN length = end - nameptr + seen_quote - (*end == '\'' ? 1 : 0);
+                       char* left;
+                       int i, j;
+                       tmp = newSV(length);
+                       left = SvPVX(tmp);
+                       for (i = 0, j = 0; j < end - nameptr; ++i, ++j) {
+                               if (nameptr[j] == '\'') {
+                                       left[i] = ':';
+                                       left[++i] = ':';
+                               }
+                               else {
+                                       left[i] = nameptr[j];
+                               }
+                       }
+                       stash = gv_stashpvn(left, length, GV_ADD | utf8flag);
+                       SvREFCNT_dec(tmp);
+               }
+               else
+                       stash = gv_stashpvn(nameptr, end - nameptr, GV_ADD | utf8flag);
+               nameptr = begin;
+               namelen -= begin - nameptr;
        }
 
+       #ifdef PERL_VERSION < 10
        /* under debugger, provide information about sub location */
        if (PL_DBsub && CvGV(cv)) {
                HV *hv = GvHV(PL_DBsub);
@@ -107,6 +137,7 @@ subname(name, sub)
                }
                Safefree(full_name);
        }
+       #endif
 
        gv = (GV *) newSV(0);
        gv_init_pvn(gv, stash, nameptr, s - nameptr, GV_ADDMULTI | utf8flag);