Fix leak when setting a fully-qualified name
[p5sagit/Sub-Name.git] / Name.xs
1 /* Copyright (C) 2004, 2008  Matthijs van Duin.  All rights reserved.
2  * Copyright (C) 2014, cPanel Inc.  All rights reserved.
3  * This program is free software; you can redistribute it and/or modify
4  * it under the same terms as Perl itself.
5  */
6
7 #include "EXTERN.h"
8 #include "perl.h"
9 #include "XSUB.h"
10
11 static MGVTBL subname_vtbl;
12
13 #ifndef PERL_MAGIC_ext
14 # define PERL_MAGIC_ext '~'
15 #endif
16
17 #ifndef SvMAGIC_set
18 #define SvMAGIC_set(sv, val) (SvMAGIC(sv) = (val))
19 #endif
20
21
22 MODULE = Sub::Name  PACKAGE = Sub::Name
23
24 PROTOTYPES: DISABLE
25
26 void
27 subname(name, sub)
28         char *name
29         SV *sub
30     PREINIT:
31         CV *cv = NULL;
32         GV *gv;
33         HV *stash = CopSTASH(PL_curcop);
34         char *s, *end = NULL;
35         MAGIC *mg;
36     PPCODE:
37         if (!SvROK(sub) && SvGMAGICAL(sub))
38                 mg_get(sub);
39         if (SvROK(sub))
40                 cv = (CV *) SvRV(sub);
41         else if (SvTYPE(sub) == SVt_PVGV)
42                 cv = GvCVu(sub);
43         else if (!SvOK(sub))
44                 croak(PL_no_usym, "a subroutine");
45         else if (PL_op->op_private & HINT_STRICT_REFS)
46                 croak("Can't use string (\"%.32s\") as %s ref while \"strict refs\" in use",
47                       SvPV_nolen(sub), "a subroutine");
48         else if ((gv = gv_fetchpv(SvPV_nolen(sub), FALSE, SVt_PVCV)))
49                 cv = GvCVu(gv);
50         if (!cv)
51                 croak("Undefined subroutine %s", SvPV_nolen(sub));
52         if (SvTYPE(cv) != SVt_PVCV && SvTYPE(cv) != SVt_PVFM)
53                 croak("Not a subroutine reference");
54         for (s = name; *s++; ) {
55                 if (*s == ':' && s[-1] == ':')
56                         end = ++s;
57                 else if (*s && s[-1] == '\'')
58                         end = s;
59         }
60         s--;
61         if (end) {
62                 char *namepv = savepvn(name, end - name);
63                 stash = GvHV(gv_fetchpv(namepv, TRUE, SVt_PVHV));
64                 Safefree(namepv);
65                 name = end;
66         }
67         gv = (GV *) newSV(0);
68         gv_init(gv, stash, name, s - name, TRUE);
69
70         mg = SvMAGIC(cv);
71         while (mg && mg->mg_virtual != &subname_vtbl)
72                 mg = mg->mg_moremagic;
73         if (!mg) {
74                 Newz(702, mg, 1, MAGIC);
75                 mg->mg_moremagic = SvMAGIC(cv);
76                 mg->mg_type = PERL_MAGIC_ext;
77                 mg->mg_virtual = &subname_vtbl;
78                 SvMAGIC_set(cv, mg);
79         }
80         if (mg->mg_flags & MGf_REFCOUNTED)
81                 SvREFCNT_dec(mg->mg_obj);
82         mg->mg_flags |= MGf_REFCOUNTED;
83         mg->mg_obj = (SV *) gv;
84         SvRMAGICAL_on(cv);
85         CvANON_off(cv);
86 #ifndef CvGV_set
87         CvGV(cv) = gv;
88 #else
89         CvGV_set(cv, gv);
90 #endif
91         PUSHs(sub);