disable caching of the namespace on 5.8
[gitmo/Package-Stash-XS.git] / XS.xs
1 #include "EXTERN.h"
2 #include "perl.h"
3 #include "XSUB.h"
4
5 #define NEED_newRV_noinc
6 #define NEED_sv_2pv_flags
7 #include "ppport.h"
8
9 #ifndef gv_fetchsv
10 #define gv_fetchsv(n,f,t) gv_fetchpv(SvPV_nolen(n), f, t)
11 #endif
12
13 #ifndef mro_method_changed_in
14 #define mro_method_changed_in(x) PL_sub_generation++
15 #endif
16
17 #ifdef newSVhek
18 #define newSVhe(he) newSVhek(HeKEY_hek(he))
19 #else
20 #define newSVhe(he) newSVpv(HePV(he, PL_na), 0)
21 #endif
22
23 #ifndef savesvpv
24 #define savesvpv(s) savepv(SvPV_nolen(s))
25 #endif
26
27 #ifndef GvCV_set
28 #define GvCV_set(gv, cv) (GvCV(gv) = (CV*)(cv))
29 #endif
30
31 /* HACK: scalar slots are always populated on perl < 5.10, so treat undef
32  * as nonexistent. this is consistent with the previous behavior of the pure
33  * perl version of this module (since this is the behavior that perl sees
34  * in all versions */
35 #if PERL_VERSION < 10
36 #define GvSVOK(g) (GvSV(g) && SvTYPE(GvSV(g)) != SVt_NULL)
37 #else
38 #define GvSVOK(g) GvSV(g)
39 #endif
40
41 #define GvAVOK(g) GvAV(g)
42 #define GvHVOK(g) GvHV(g)
43 #define GvCVOK(g) GvCVu(g) /* XXX: should this really be GvCVu? or GvCV? */
44 #define GvIOOK(g) GvIO(g)
45
46 /* see above - don't let scalar slots become unpopulated, this breaks
47  * assumptions in core */
48 #if PERL_VERSION < 10
49 #define GvSetSV(g,v) do {               \
50     SV *_v = (SV*)(v);                  \
51     SvREFCNT_dec(GvSV(g));              \
52     if ((GvSV(g) = _v ? _v : newSV(0))) \
53         GvIMPORTED_SV_on(g);            \
54 } while (0)
55 #else
56 #define GvSetSV(g,v) do {               \
57     SvREFCNT_dec(GvSV(g));              \
58     if ((GvSV(g) = (SV*)(v)))           \
59         GvIMPORTED_SV_on(g);            \
60 } while (0)
61 #endif
62
63 #define GvSetAV(g,v) do {               \
64     SvREFCNT_dec(GvAV(g));              \
65     if ((GvAV(g) = (AV*)(v)))           \
66         GvIMPORTED_AV_on(g);            \
67 } while (0)
68 #define GvSetHV(g,v) do {               \
69     SvREFCNT_dec(GvHV(g));              \
70     if ((GvHV(g) = (HV*)(v)))           \
71         GvIMPORTED_HV_on(g);            \
72 } while (0)
73 #define GvSetCV(g,v) do {               \
74     SvREFCNT_dec(GvCV(g));              \
75     if ((GvCV_set(g, v))) {             \
76         GvIMPORTED_CV_on(g);            \
77         GvASSUMECV_on(g);               \
78     }                                   \
79     GvCVGEN(g) = 0;                     \
80     mro_method_changed_in(GvSTASH(g));  \
81 } while (0)
82 #define GvSetIO(g,v) do {               \
83     SvREFCNT_dec(GvIO(g));              \
84     GvIOp(g) = (IO*)(v);                \
85 } while (0)
86
87 typedef enum {
88     VAR_NONE = 0,
89     VAR_SCALAR,
90     VAR_ARRAY,
91     VAR_HASH,
92     VAR_CODE,
93     VAR_IO,
94     VAR_GLOB,  /* TODO: unimplemented */
95     VAR_FORMAT /* TODO: unimplemented */
96 } vartype_t;
97
98 typedef struct {
99     vartype_t type;
100     SV *name;
101 } varspec_t;
102
103 static U32 name_hash, namespace_hash, type_hash;
104 static SV *name_key, *namespace_key, *type_key;
105
106 const char *vartype_to_string(vartype_t type)
107 {
108     switch (type) {
109     case VAR_SCALAR:
110         return "SCALAR";
111     case VAR_ARRAY:
112         return "ARRAY";
113     case VAR_HASH:
114         return "HASH";
115     case VAR_CODE:
116         return "CODE";
117     case VAR_IO:
118         return "IO";
119     default:
120         return "unknown";
121     }
122 }
123
124 I32 vartype_to_svtype(vartype_t type)
125 {
126     switch (type) {
127     case VAR_SCALAR:
128         return SVt_PV; /* or whatever */
129     case VAR_ARRAY:
130         return SVt_PVAV;
131     case VAR_HASH:
132         return SVt_PVHV;
133     case VAR_CODE:
134         return SVt_PVCV;
135     case VAR_IO:
136         return SVt_PVIO;
137     default:
138         return SVt_NULL;
139     }
140 }
141
142 vartype_t string_to_vartype(char *vartype)
143 {
144     if (strEQ(vartype, "SCALAR")) {
145         return VAR_SCALAR;
146     }
147     else if (strEQ(vartype, "ARRAY")) {
148         return VAR_ARRAY;
149     }
150     else if (strEQ(vartype, "HASH")) {
151         return VAR_HASH;
152     }
153     else if (strEQ(vartype, "CODE")) {
154         return VAR_CODE;
155     }
156     else if (strEQ(vartype, "IO")) {
157         return VAR_IO;
158     }
159     else {
160         croak("Type must be one of 'SCALAR', 'ARRAY', 'HASH', 'CODE', or 'IO', not '%s'", vartype);
161     }
162 }
163
164 void _deconstruct_variable_name(SV *variable, varspec_t *varspec)
165 {
166     char *varpv;
167
168     if (!SvCUR(variable))
169         croak("You must pass a variable name");
170
171     varspec->name = sv_2mortal(newSVsv(variable));
172
173     varpv = SvPV_nolen(varspec->name);
174     switch (varpv[0]) {
175     case '$':
176         varspec->type = VAR_SCALAR;
177         sv_chop(varspec->name, &varpv[1]);
178         break;
179     case '@':
180         varspec->type = VAR_ARRAY;
181         sv_chop(varspec->name, &varpv[1]);
182         break;
183     case '%':
184         varspec->type = VAR_HASH;
185         sv_chop(varspec->name, &varpv[1]);
186         break;
187     case '&':
188         varspec->type = VAR_CODE;
189         sv_chop(varspec->name, &varpv[1]);
190         break;
191     default:
192         varspec->type = VAR_IO;
193         break;
194     }
195 }
196
197 void _deconstruct_variable_hash(HV *variable, varspec_t *varspec)
198 {
199     HE *val;
200
201     val = hv_fetch_ent(variable, name_key, 0, name_hash);
202     if (!val)
203         croak("The 'name' key is required in variable specs");
204
205     varspec->name = sv_2mortal(newSVsv(HeVAL(val)));
206
207     val = hv_fetch_ent(variable, type_key, 0, type_hash);
208     if (!val)
209         croak("The 'type' key is required in variable specs");
210
211     varspec->type = string_to_vartype(SvPV_nolen(HeVAL(val)));
212 }
213
214 int _valid_for_type(SV *value, vartype_t type)
215 {
216     svtype sv_type = SvROK(value) ? SvTYPE(SvRV(value)) : SVt_NULL;
217
218     switch (type) {
219     case VAR_SCALAR:
220         /* XXX: something weird is going on here - apparently values can
221          * be SVt_NULL but also be SvROK (and also, SVt_NULL isn't SvOK) */
222         if (sv_type == SVt_NULL)
223             return 1;
224         return SvROK(value) ? SvOK(SvRV(value)) : SvOK(value);
225     case VAR_ARRAY:
226         return sv_type == SVt_PVAV;
227     case VAR_HASH:
228         return sv_type == SVt_PVHV;
229     case VAR_CODE:
230         return sv_type == SVt_PVCV;
231     case VAR_IO:
232         return sv_type == SVt_PVIO;
233     default:
234         return 0;
235     }
236 }
237
238 HV *_get_namespace(SV *self)
239 {
240     dSP;
241     SV *ret;
242
243     PUSHMARK(SP);
244     XPUSHs(self);
245     PUTBACK;
246
247     call_method("namespace", G_SCALAR);
248
249     SPAGAIN;
250     ret = POPs;
251     PUTBACK;
252
253     return (HV*)SvRV(ret);
254 }
255
256 SV *_get_name(SV *self)
257 {
258     dSP;
259     SV *ret;
260
261     PUSHMARK(SP);
262     XPUSHs(self);
263     PUTBACK;
264
265     call_method("name", G_SCALAR);
266
267     SPAGAIN;
268     ret = POPs;
269     PUTBACK;
270
271     return ret;
272 }
273
274 void _expand_glob(SV *self, SV *varname)
275 {
276     SV *name;
277
278     name = newSVsv(_get_name(self));
279     sv_catpvs(name, "::");
280     sv_catsv(name, varname);
281
282     /* can't use gv_init here, because it screws up @ISA in a way that I
283      * can't reproduce, but that CMOP triggers */
284     gv_fetchsv(name, GV_ADD, SVt_NULL);
285     SvREFCNT_dec(name);
286 }
287
288 SV *_get_symbol(SV *self, varspec_t *variable, int vivify)
289 {
290     HV *namespace;
291     HE *entry;
292     GV *glob;
293
294     namespace = _get_namespace(self);
295     entry = hv_fetch_ent(namespace, variable->name, vivify, 0);
296     if (!entry)
297         return NULL;
298
299     glob = (GV*)(HeVAL(entry));
300     if (!isGV(glob))
301         _expand_glob(self, variable->name);
302
303     if (vivify) {
304         switch (variable->type) {
305         case VAR_SCALAR:
306             if (!GvSVOK(glob))
307                 GvSetSV(glob, newSV(0));
308             break;
309         case VAR_ARRAY:
310             if (!GvAVOK(glob))
311                 GvSetAV(glob, newAV());
312             break;
313         case VAR_HASH:
314             if (!GvHVOK(glob))
315                 GvSetHV(glob, newHV());
316             break;
317         case VAR_CODE:
318             croak("Don't know how to vivify CODE variables");
319         case VAR_IO:
320             if (!GvIOOK(glob))
321                 GvSetIO(glob, newIO());
322             break;
323         default:
324             croak("Unknown type in vivication");
325         }
326     }
327
328     switch (variable->type) {
329     case VAR_SCALAR:
330         return GvSV(glob);
331     case VAR_ARRAY:
332         return (SV*)GvAV(glob);
333     case VAR_HASH:
334         return (SV*)GvHV(glob);
335     case VAR_CODE:
336         return (SV*)GvCV(glob);
337     case VAR_IO:
338         return (SV*)GvIO(glob);
339     default:
340         return NULL;
341     }
342 }
343
344 MODULE = Package::Stash::XS  PACKAGE = Package::Stash::XS
345
346 PROTOTYPES: DISABLE
347
348 SV*
349 new(class, package_name)
350     SV *class
351     SV *package_name
352   PREINIT:
353     HV *instance;
354   CODE:
355     if (!SvPOK(package_name))
356         croak("The constructor argument must be the name of a package");
357
358     instance = newHV();
359
360     if (!hv_store(instance, "name", 4, SvREFCNT_inc_simple_NN(package_name), 0)) {
361         SvREFCNT_dec(package_name);
362         SvREFCNT_dec(instance);
363         croak("Couldn't initialize the 'name' key, hv_store failed");
364     }
365
366     RETVAL = sv_bless(newRV_noinc((SV*)instance), gv_stashsv(class, 0));
367   OUTPUT:
368     RETVAL
369
370 SV*
371 name(self)
372     SV *self
373   PREINIT:
374     HE *slot;
375   CODE:
376     if (!sv_isobject(self))
377         croak("Can't call name as a class method");
378     slot = hv_fetch_ent((HV*)SvRV(self), name_key, 0, name_hash);
379     RETVAL = slot ? SvREFCNT_inc_simple_NN(HeVAL(slot)) : &PL_sv_undef;
380   OUTPUT:
381     RETVAL
382
383 SV*
384 namespace(self)
385     SV *self
386   PREINIT:
387     HE *slot;
388     SV *package_name;
389   CODE:
390     if (!sv_isobject(self))
391         croak("Can't call namespace as a class method");
392 #if PERL_VERSION < 10
393     package_name = _get_name(self);
394     RETVAL = newRV_inc((SV*)gv_stashpv(SvPV_nolen(package_name), GV_ADD));
395 #else
396     slot = hv_fetch_ent((HV*)SvRV(self), namespace_key, 0, namespace_hash);
397     if (slot) {
398         RETVAL = SvREFCNT_inc_simple_NN(HeVAL(slot));
399     }
400     else {
401         HV *namespace;
402         SV *nsref;
403
404         package_name = _get_name(self);
405         namespace = gv_stashpv(SvPV_nolen(package_name), GV_ADD);
406         nsref = newRV_inc((SV*)namespace);
407         sv_rvweaken(nsref);
408         if (!hv_store((HV*)SvRV(self), "namespace", 9, nsref, 0)) {
409             SvREFCNT_dec(nsref);
410             SvREFCNT_dec(self);
411             croak("Couldn't initialize the 'namespace' key, hv_store failed");
412         }
413         RETVAL = SvREFCNT_inc_simple_NN(nsref);
414     }
415 #endif
416   OUTPUT:
417     RETVAL
418
419 void
420 add_symbol(self, variable, initial=NULL, ...)
421     SV *self
422     varspec_t variable
423     SV *initial
424   PREINIT:
425     SV *name;
426     GV *glob;
427   CODE:
428     if (initial && !_valid_for_type(initial, variable.type))
429         croak("%s is not of type %s",
430               SvPV_nolen(initial), vartype_to_string(variable.type));
431
432     name = newSVsv(_get_name(self));
433     sv_catpvs(name, "::");
434     sv_catsv(name, variable.name);
435
436     if (items > 2 && (PL_perldb & 0x10) && variable.type == VAR_CODE) {
437         int i;
438         char *filename = NULL;
439         I32 first_line_num = -1, last_line_num = -1;
440         STRLEN namelen;
441         SV *dbval;
442         HV *dbsub;
443
444         if ((items - 3) % 2)
445             croak("add_symbol: Odd number of elements in %%opts");
446
447         for (i = 3; i < items; i += 2) {
448             char *key;
449             key = SvPV_nolen(ST(i));
450             if (strEQ(key, "filename")) {
451                 if (!SvPOK(ST(i + 1)))
452                     croak("add_symbol: filename must be a string");
453                 filename = SvPV_nolen(ST(i + 1));
454             }
455             else if (strEQ(key, "first_line_num")) {
456                 if (!SvIOK(ST(i + 1)))
457                     croak("add_symbol: first_line_num must be an integer");
458                 first_line_num = SvIV(ST(i + 1));
459             }
460             else if (strEQ(key, "last_line_num")) {
461                 if (!SvIOK(ST(i + 1)))
462                     croak("add_symbol: last_line_num must be an integer");
463                 last_line_num = SvIV(ST(i + 1));
464             }
465         }
466
467         if (!filename || first_line_num == -1) {
468             if (!filename)
469                 filename = CopFILE(PL_curcop);
470             if (first_line_num == -1)
471                 first_line_num = PL_curcop->cop_line;
472         }
473
474         if (last_line_num == -1)
475             last_line_num = first_line_num;
476
477         /* http://perldoc.perl.org/perldebguts.html#Debugger-Internals */
478         dbsub = get_hv("DB::sub", 1);
479         dbval = newSVpvf("%s:%d-%d", filename, first_line_num, last_line_num);
480         if (!hv_store_ent(dbsub, name, dbval, 0)) {
481             warn("Failed to update $DB::sub for subroutine %s",
482                  SvPV_nolen(name));
483             SvREFCNT_dec(dbval);
484         }
485     }
486
487     /* GV_ADDMULTI rather than GV_ADD because otherwise you get 'used only
488      * once' warnings in some situations... i can't reproduce this, but CMOP
489      * triggers it */
490     glob = gv_fetchsv(name, GV_ADDMULTI, vartype_to_svtype(variable.type));
491
492     if (initial) {
493         SV *val;
494
495         if (SvROK(initial)) {
496             val = SvRV(initial);
497             SvREFCNT_inc_simple_void_NN(val);
498         }
499         else {
500             val = newSVsv(initial);
501         }
502
503         switch (variable.type) {
504         case VAR_SCALAR:
505             GvSetSV(glob, val);
506             break;
507         case VAR_ARRAY:
508             GvSetAV(glob, val);
509             break;
510         case VAR_HASH:
511             GvSetHV(glob, val);
512             break;
513         case VAR_CODE:
514             GvSetCV(glob, val);
515             break;
516         case VAR_IO:
517             GvSetIO(glob, val);
518             break;
519         }
520     }
521
522     SvREFCNT_dec(name);
523
524 void
525 remove_glob(self, name)
526     SV *self
527     SV *name
528   CODE:
529     hv_delete_ent(_get_namespace(self), name, G_DISCARD, 0);
530
531 int
532 has_symbol(self, variable)
533     SV *self
534     varspec_t variable
535   PREINIT:
536     HV *namespace;
537     HE *entry;
538     SV *val;
539   CODE:
540     namespace = _get_namespace(self);
541     entry = hv_fetch_ent(namespace, variable.name, 0, 0);
542     if (!entry)
543         XSRETURN_UNDEF;
544
545     val = HeVAL(entry);
546     if (isGV(val)) {
547         GV *glob = (GV*)val;
548         switch (variable.type) {
549         case VAR_SCALAR:
550             RETVAL = GvSVOK(glob) ? 1 : 0;
551             break;
552         case VAR_ARRAY:
553             RETVAL = GvAVOK(glob) ? 1 : 0;
554             break;
555         case VAR_HASH:
556             RETVAL = GvHVOK(glob) ? 1 : 0;
557             break;
558         case VAR_CODE:
559             RETVAL = GvCVOK(glob) ? 1 : 0;
560             break;
561         case VAR_IO:
562             RETVAL = GvIOOK(glob) ? 1 : 0;
563             break;
564         }
565     }
566     else {
567         RETVAL = (variable.type == VAR_CODE);
568     }
569   OUTPUT:
570     RETVAL
571
572 SV*
573 get_symbol(self, variable)
574     SV *self
575     varspec_t variable
576   PREINIT:
577     SV *val;
578   CODE:
579     val = _get_symbol(self, &variable, 0);
580     if (!val)
581         XSRETURN_UNDEF;
582     RETVAL = newRV_inc(val);
583   OUTPUT:
584     RETVAL
585
586 SV*
587 get_or_add_symbol(self, variable)
588     SV *self
589     varspec_t variable
590   PREINIT:
591     SV *val;
592   CODE:
593     val = _get_symbol(self, &variable, 1);
594     if (!val)
595         XSRETURN_UNDEF;
596     RETVAL = newRV_inc(val);
597   OUTPUT:
598     RETVAL
599
600 void
601 remove_symbol(self, variable)
602     SV *self
603     varspec_t variable
604   PREINIT:
605     HV *namespace;
606     HE *entry;
607     SV *val;
608   CODE:
609     namespace = _get_namespace(self);
610     entry = hv_fetch_ent(namespace, variable.name, 0, 0);
611     if (!entry)
612         XSRETURN_EMPTY;
613
614     val = HeVAL(entry);
615     if (isGV(val)) {
616         GV *glob = (GV*)val;
617         switch (variable.type) {
618         case VAR_SCALAR:
619             GvSetSV(glob, NULL);
620             break;
621         case VAR_ARRAY:
622             GvSetAV(glob, NULL);
623             break;
624         case VAR_HASH:
625             GvSetHV(glob, NULL);
626             break;
627         case VAR_CODE:
628             GvSetCV(glob, NULL);
629             break;
630         case VAR_IO:
631             GvSetIO(glob, NULL);
632             break;
633         }
634     }
635     else {
636         if (variable.type == VAR_CODE) {
637             hv_delete_ent(namespace, variable.name, G_DISCARD, 0);
638         }
639     }
640
641 void
642 list_all_symbols(self, vartype=VAR_NONE)
643     SV *self
644     vartype_t vartype
645   PPCODE:
646     if (vartype == VAR_NONE) {
647         HV *namespace;
648         HE *entry;
649         int keys;
650
651         namespace = _get_namespace(self);
652         keys = hv_iterinit(namespace);
653         EXTEND(SP, keys);
654         while ((entry = hv_iternext(namespace))) {
655             mPUSHs(newSVhe(entry));
656         }
657     }
658     else {
659         HV *namespace;
660         SV *val;
661         char *key;
662         I32 len;
663
664         namespace = _get_namespace(self);
665         hv_iterinit(namespace);
666         while ((val = hv_iternextsv(namespace, &key, &len))) {
667             GV *gv = (GV*)val;
668             if (isGV(gv)) {
669                 switch (vartype) {
670                 case VAR_SCALAR:
671                     if (GvSVOK(val))
672                         mXPUSHp(key, len);
673                     break;
674                 case VAR_ARRAY:
675                     if (GvAVOK(val))
676                         mXPUSHp(key, len);
677                     break;
678                 case VAR_HASH:
679                     if (GvHVOK(val))
680                         mXPUSHp(key, len);
681                     break;
682                 case VAR_CODE:
683                     if (GvCVOK(val))
684                         mXPUSHp(key, len);
685                     break;
686                 case VAR_IO:
687                     if (GvIOOK(val))
688                         mXPUSHp(key, len);
689                     break;
690                 }
691             }
692             else if (vartype == VAR_CODE) {
693                 mXPUSHp(key, len);
694             }
695         }
696     }
697
698 void
699 get_all_symbols(self, vartype=VAR_NONE)
700     SV *self
701     vartype_t vartype
702   PREINIT:
703     HV *namespace, *ret;
704     SV *val;
705     char *key;
706     I32 len;
707   PPCODE:
708     namespace = _get_namespace(self);
709     ret = newHV();
710
711     hv_iterinit(namespace);
712     while ((val = hv_iternextsv(namespace, &key, &len))) {
713         GV *gv = (GV*)val;
714
715         if (!isGV(gv)) {
716             SV *keysv = newSVpvn(key, len);
717             _expand_glob(self, keysv);
718             SvREFCNT_dec(keysv);
719         }
720
721         switch (vartype) {
722         case VAR_SCALAR:
723             if (GvSVOK(val))
724                 hv_store(ret, key, len, newRV_inc(GvSV(gv)), 0);
725             break;
726         case VAR_ARRAY:
727             if (GvAVOK(val))
728                 hv_store(ret, key, len, newRV_inc((SV*)GvAV(gv)), 0);
729             break;
730         case VAR_HASH:
731             if (GvHVOK(val))
732                 hv_store(ret, key, len, newRV_inc((SV*)GvHV(gv)), 0);
733             break;
734         case VAR_CODE:
735             if (GvCVOK(val))
736                 hv_store(ret, key, len, newRV_inc((SV*)GvCV(gv)), 0);
737             break;
738         case VAR_IO:
739             if (GvIOOK(val))
740                 hv_store(ret, key, len, newRV_inc((SV*)GvIO(gv)), 0);
741             break;
742         case VAR_NONE:
743             hv_store(ret, key, len, SvREFCNT_inc_simple_NN(val), 0);
744             break;
745         }
746     }
747
748     mPUSHs(newRV_noinc((SV*)ret));
749
750 BOOT:
751     {
752         name_key = newSVpvs("name");
753         PERL_HASH(name_hash, "name", 4);
754
755         namespace_key = newSVpvs("namespace");
756         PERL_HASH(namespace_hash, "namespace", 9);
757
758         type_key = newSVpvs("type");
759         PERL_HASH(type_hash, "type", 4);
760     }