apply magic properly when initializing globs
[gitmo/Package-Stash-XS.git] / XS.xs
CommitLineData
59017825 1#include "EXTERN.h"
2#include "perl.h"
3#include "XSUB.h"
4
dfabcbae 5#define NEED_newRV_noinc
6#define NEED_sv_2pv_flags
7#include "ppport.h"
8
13f8a7b7 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
3e37e315 20#define newSVhe(he) newSVpv(HePV(he, PL_na), 0)
13f8a7b7 21#endif
22
23#ifndef savesvpv
24#define savesvpv(s) savepv(SvPV_nolen(s))
25#endif
26
22d15de6 27#ifndef GvCV_set
28#define GvCV_set(gv, cv) (GvCV(gv) = (CV*)(cv))
29#endif
30
a829b7a6 31#ifndef SVT_SCALAR
32#define SVT_SCALAR(svt) (svt <= SVt_PVLV)
33#endif
34
35#ifndef SVT_ARRAY
36#define SVT_ARRAY(svt) (svt == SVt_PVAV)
37#endif
38
39#ifndef SVT_HASH
40#define SVT_HASH(svt) (svt == SVt_PVHV)
41#endif
42
43#ifndef SVT_CODE
44#define SVT_CODE(svt) (svt == SVt_PVCV)
45#endif
46
47#ifndef SVT_IO
48#define SVT_IO(svt) (svt == SVt_PVIO)
49#endif
50
51#ifndef SVT_FORMAT
52#define SVT_FORMAT(svt) (svt == SVt_PVFM)
53#endif
54
d551a208 55/* HACK: scalar slots are always populated on perl < 5.10, so treat undef
56 * as nonexistent. this is consistent with the previous behavior of the pure
57 * perl version of this module (since this is the behavior that perl sees
58 * in all versions */
59#if PERL_VERSION < 10
60#define GvSVOK(g) (GvSV(g) && SvTYPE(GvSV(g)) != SVt_NULL)
61#else
62#define GvSVOK(g) GvSV(g)
63#endif
64
65#define GvAVOK(g) GvAV(g)
66#define GvHVOK(g) GvHV(g)
67#define GvCVOK(g) GvCVu(g) /* XXX: should this really be GvCVu? or GvCV? */
68#define GvIOOK(g) GvIO(g)
69
9aa6fe4f 70/* see above - don't let scalar slots become unpopulated, this breaks
71 * assumptions in core */
3720d1a1 72#if PERL_VERSION < 10
73#define GvSetSV(g,v) do { \
74 SV *_v = (SV*)(v); \
75 SvREFCNT_dec(GvSV(g)); \
76 if ((GvSV(g) = _v ? _v : newSV(0))) \
77 GvIMPORTED_SV_on(g); \
78} while (0)
79#else
d551a208 80#define GvSetSV(g,v) do { \
81 SvREFCNT_dec(GvSV(g)); \
82 if ((GvSV(g) = (SV*)(v))) \
83 GvIMPORTED_SV_on(g); \
84} while (0)
3720d1a1 85#endif
86
d551a208 87#define GvSetAV(g,v) do { \
88 SvREFCNT_dec(GvAV(g)); \
89 if ((GvAV(g) = (AV*)(v))) \
90 GvIMPORTED_AV_on(g); \
91} while (0)
92#define GvSetHV(g,v) do { \
93 SvREFCNT_dec(GvHV(g)); \
94 if ((GvHV(g) = (HV*)(v))) \
95 GvIMPORTED_HV_on(g); \
96} while (0)
97#define GvSetCV(g,v) do { \
98 SvREFCNT_dec(GvCV(g)); \
2d533eaf 99 if ((GvCV_set(g, (CV*)(v)))) { \
d551a208 100 GvIMPORTED_CV_on(g); \
101 GvASSUMECV_on(g); \
102 } \
103 GvCVGEN(g) = 0; \
4eb02ba6 104 if (HvENAME_get(GvSTASH(g))) \
105 mro_method_changed_in(GvSTASH(g)); \
d551a208 106} while (0)
107#define GvSetIO(g,v) do { \
108 SvREFCNT_dec(GvIO(g)); \
109 GvIOp(g) = (IO*)(v); \
110} while (0)
111
a382a84b 112typedef enum {
113 VAR_NONE = 0,
114 VAR_SCALAR,
115 VAR_ARRAY,
116 VAR_HASH,
117 VAR_CODE,
118 VAR_IO,
119 VAR_GLOB, /* TODO: unimplemented */
120 VAR_FORMAT /* TODO: unimplemented */
121} vartype_t;
122
123typedef struct {
124 vartype_t type;
4849f4fc 125 SV *name;
a382a84b 126} varspec_t;
127
26f802f7 128static U32 name_hash, namespace_hash, type_hash;
129static SV *name_key, *namespace_key, *type_key;
78daf8ed 130static REGEXP *valid_module_regex;
26f802f7 131
bf460ff3 132static const char *vartype_to_string(vartype_t type)
76f7c306 133{
134 switch (type) {
135 case VAR_SCALAR:
136 return "SCALAR";
137 case VAR_ARRAY:
138 return "ARRAY";
139 case VAR_HASH:
140 return "HASH";
141 case VAR_CODE:
142 return "CODE";
143 case VAR_IO:
144 return "IO";
145 default:
146 return "unknown";
147 }
148}
149
bf460ff3 150static I32 vartype_to_svtype(vartype_t type)
76f7c306 151{
152 switch (type) {
153 case VAR_SCALAR:
154 return SVt_PV; /* or whatever */
155 case VAR_ARRAY:
156 return SVt_PVAV;
157 case VAR_HASH:
158 return SVt_PVHV;
159 case VAR_CODE:
160 return SVt_PVCV;
161 case VAR_IO:
162 return SVt_PVIO;
163 default:
164 return SVt_NULL;
165 }
166}
167
bf460ff3 168static vartype_t string_to_vartype(char *vartype)
a382a84b 169{
170 if (strEQ(vartype, "SCALAR")) {
171 return VAR_SCALAR;
172 }
173 else if (strEQ(vartype, "ARRAY")) {
174 return VAR_ARRAY;
175 }
176 else if (strEQ(vartype, "HASH")) {
177 return VAR_HASH;
178 }
179 else if (strEQ(vartype, "CODE")) {
180 return VAR_CODE;
181 }
182 else if (strEQ(vartype, "IO")) {
183 return VAR_IO;
184 }
185 else {
f88deb66 186 croak("Type must be one of 'SCALAR', 'ARRAY', 'HASH', 'CODE', or 'IO', not '%s'", vartype);
a382a84b 187 }
188}
189
78daf8ed 190static int _is_valid_module_name(SV *package)
191{
192 char *buf;
193 STRLEN len;
194 SV *sv;
195
196 buf = SvPV(package, len);
197
198 /* whee cargo cult */
199 sv = sv_newmortal();
200 sv_upgrade(sv, SVt_PV);
201 SvREADONLY_on(sv);
202 SvLEN(sv) = 0;
203 SvUTF8_on(sv);
204 SvPVX(sv) = buf;
205 SvCUR_set(sv, len);
206 SvPOK_on(sv);
207
208 return pregexec(valid_module_regex, buf, buf + len, buf, 1, sv, 1);
209}
210
bf460ff3 211static void _deconstruct_variable_name(SV *variable, varspec_t *varspec)
a382a84b 212{
4849f4fc 213 char *varpv;
a382a84b 214
4849f4fc 215 if (!SvCUR(variable))
64b79211 216 croak("You must pass a variable name");
a382a84b 217
4849f4fc 218 varspec->name = sv_2mortal(newSVsv(variable));
219
220 varpv = SvPV_nolen(varspec->name);
221 switch (varpv[0]) {
a382a84b 222 case '$':
223 varspec->type = VAR_SCALAR;
4849f4fc 224 sv_chop(varspec->name, &varpv[1]);
a382a84b 225 break;
226 case '@':
227 varspec->type = VAR_ARRAY;
4849f4fc 228 sv_chop(varspec->name, &varpv[1]);
a382a84b 229 break;
230 case '%':
231 varspec->type = VAR_HASH;
4849f4fc 232 sv_chop(varspec->name, &varpv[1]);
a382a84b 233 break;
234 case '&':
235 varspec->type = VAR_CODE;
4849f4fc 236 sv_chop(varspec->name, &varpv[1]);
64b79211 237 break;
238 default:
a382a84b 239 varspec->type = VAR_IO;
64b79211 240 break;
a382a84b 241 }
242}
243
bf460ff3 244static void _deconstruct_variable_hash(HV *variable, varspec_t *varspec)
a382a84b 245{
26f802f7 246 HE *val;
a382a84b 247
26f802f7 248 val = hv_fetch_ent(variable, name_key, 0, name_hash);
a382a84b 249 if (!val)
250 croak("The 'name' key is required in variable specs");
251
061b471e 252 varspec->name = sv_2mortal(newSVsv(HeVAL(val)));
a382a84b 253
26f802f7 254 val = hv_fetch_ent(variable, type_key, 0, type_hash);
a382a84b 255 if (!val)
256 croak("The 'type' key is required in variable specs");
257
061b471e 258 varspec->type = string_to_vartype(SvPV_nolen(HeVAL(val)));
a382a84b 259}
260
1ff84a32 261static void _check_varspec_is_valid(varspec_t *varspec)
262{
263 if (strstr(SvPV_nolen(varspec->name), "::")) {
264 croak("Variable names may not contain ::");
265 }
266}
267
bf460ff3 268static int _valid_for_type(SV *value, vartype_t type)
a382a84b 269{
270 svtype sv_type = SvROK(value) ? SvTYPE(SvRV(value)) : SVt_NULL;
271
272 switch (type) {
273 case VAR_SCALAR:
5a4d63bd 274 /* XXX is a glob a scalar? assigning a glob to the scalar slot seems
275 * to work here, but in pure perl i'm pretty sure it goes to the EGV
276 * slot, which seems more correct to me. just disable it for now
277 * i guess */
278 return SVT_SCALAR(sv_type) && sv_type != SVt_PVGV;
a382a84b 279 case VAR_ARRAY:
a829b7a6 280 return SVT_ARRAY(sv_type);
a382a84b 281 case VAR_HASH:
a829b7a6 282 return SVT_HASH(sv_type);
a382a84b 283 case VAR_CODE:
a829b7a6 284 return SVT_CODE(sv_type);
a382a84b 285 case VAR_IO:
a829b7a6 286 return SVT_IO(sv_type);
a382a84b 287 default:
288 return 0;
289 }
290}
291
bf460ff3 292static HV *_get_namespace(SV *self)
3fd56b4d 293{
294 dSP;
295 SV *ret;
296
297 PUSHMARK(SP);
298 XPUSHs(self);
299 PUTBACK;
300
612dcf3b 301 call_method("namespace", G_SCALAR);
3fd56b4d 302
303 SPAGAIN;
304 ret = POPs;
305 PUTBACK;
306
307 return (HV*)SvRV(ret);
308}
309
bf460ff3 310static SV *_get_name(SV *self)
76f7c306 311{
312 dSP;
313 SV *ret;
314
315 PUSHMARK(SP);
316 XPUSHs(self);
317 PUTBACK;
318
612dcf3b 319 call_method("name", G_SCALAR);
76f7c306 320
321 SPAGAIN;
322 ret = POPs;
323 PUTBACK;
324
325 return ret;
326}
327
e006ab79 328static void _real_gv_init(GV *gv, HV *stash, SV *name)
329{
330 char *name_pv;
331 STRLEN name_len;
332
333 name_pv = SvPV(name, name_len);
334 gv_init(gv, stash, name_pv, name_len, 1);
7ce1a95b 335
336 /* XXX: copied and pasted from gv_fetchpvn_flags and such */
337 /* ignoring the stuff for CORE:: and main:: for now, and also
338 * ignoring the GvMULTI_on bits, since we pass 1 to gv_init above */
339 switch (name_pv[0]) {
340 case 'I':
341 if (strEQ(&name_pv[1], "SA")) {
342 AV *av;
343
344 av = GvAVn(gv);
345 sv_magic(MUTABLE_SV(av), MUTABLE_SV(gv), PERL_MAGIC_isa,
346 NULL, 0);
347 }
348 break;
349 case 'O':
350 if (strEQ(&name_pv[1], "VERLOAD")) {
351 HV *hv;
352
353 hv = GvHVn(gv);
354 hv_magic(hv, NULL, PERL_MAGIC_overload);
355 }
356 break;
357 default:
358 break;
359 }
e006ab79 360}
361
bf460ff3 362static void _expand_glob(SV *self, SV *varname)
60b395a1 363{
5e738664 364 HV *namespace;
365 HE *entry;
366 GV *glob;
367
368 namespace = _get_namespace(self);
369
370 if (entry = hv_fetch_ent(namespace, varname, 0, 0)) {
371 glob = (GV*)HeVAL(entry);
372 if (isGV(glob)) {
373 croak("_expand_glob called on stash slot with expanded glob");
374 }
375 else {
5e738664 376 SvREFCNT_inc(glob);
e006ab79 377 _real_gv_init(glob, namespace, varname);
5e738664 378 if (!hv_store_ent(namespace, varname, (SV*)glob, 0)) {
379 croak("hv_store failed");
380 }
381 }
382 }
383 else {
384 croak("_expand_glob called on nonexistent stash slot");
385 }
60b395a1 386}
387
bf460ff3 388static SV *_get_symbol(SV *self, varspec_t *variable, int vivify)
4871b1a0 389{
390 HV *namespace;
4849f4fc 391 HE *entry;
4871b1a0 392 GV *glob;
393
394 namespace = _get_namespace(self);
4849f4fc 395 entry = hv_fetch_ent(namespace, variable->name, vivify, 0);
4871b1a0 396 if (!entry)
397 return NULL;
398
4849f4fc 399 glob = (GV*)(HeVAL(entry));
60b395a1 400 if (!isGV(glob))
401 _expand_glob(self, variable->name);
4871b1a0 402
403 if (vivify) {
404 switch (variable->type) {
405 case VAR_SCALAR:
d551a208 406 if (!GvSVOK(glob))
407 GvSetSV(glob, newSV(0));
4871b1a0 408 break;
409 case VAR_ARRAY:
d551a208 410 if (!GvAVOK(glob))
411 GvSetAV(glob, newAV());
4871b1a0 412 break;
413 case VAR_HASH:
d551a208 414 if (!GvHVOK(glob))
415 GvSetHV(glob, newHV());
4871b1a0 416 break;
417 case VAR_CODE:
418 croak("Don't know how to vivify CODE variables");
419 case VAR_IO:
d551a208 420 if (!GvIOOK(glob))
421 GvSetIO(glob, newIO());
4871b1a0 422 break;
423 default:
424 croak("Unknown type in vivication");
425 }
426 }
427
428 switch (variable->type) {
429 case VAR_SCALAR:
430 return GvSV(glob);
431 case VAR_ARRAY:
432 return (SV*)GvAV(glob);
433 case VAR_HASH:
434 return (SV*)GvHV(glob);
435 case VAR_CODE:
436 return (SV*)GvCV(glob);
437 case VAR_IO:
438 return (SV*)GvIO(glob);
439 default:
440 return NULL;
441 }
442}
443
c53d2df2 444MODULE = Package::Stash::XS PACKAGE = Package::Stash::XS
59017825 445
be2a7e99 446PROTOTYPES: DISABLE
447
59017825 448SV*
8860963a 449new(class, package)
4849f4fc 450 SV *class
8860963a 451 SV *package
3496f1e8 452 PREINIT:
59017825 453 HV *instance;
59017825 454 CODE:
8860963a 455 if (SvPOK(package)) {
456 if (!_is_valid_module_name(package))
457 croak("%s is not a module name", SvPV_nolen(package));
59017825 458
8860963a 459 instance = newHV();
78daf8ed 460
8860963a 461 if (!hv_store(instance, "name", 4, SvREFCNT_inc_simple_NN(package), 0)) {
462 SvREFCNT_dec(package);
463 SvREFCNT_dec(instance);
464 croak("Couldn't initialize the 'name' key, hv_store failed");
465 }
466 }
467 else if (SvROK(package) && SvTYPE(SvRV(package)) == SVt_PVHV) {
468 instance = newHV();
59017825 469
8860963a 470 if (!hv_store(instance, "namespace", 9, SvREFCNT_inc_simple_NN(package), 0)) {
471 SvREFCNT_dec(package);
472 SvREFCNT_dec(instance);
473 croak("Couldn't initialize the 'namespace' key, hv_store failed");
474 }
475 }
476 else {
477 croak("Package::Stash->new must be passed the name of the package to access");
1d6c978b 478 }
59017825 479
4849f4fc 480 RETVAL = sv_bless(newRV_noinc((SV*)instance), gv_stashsv(class, 0));
59017825 481 OUTPUT:
482 RETVAL
194acf47 483
484SV*
485name(self)
486 SV *self
3496f1e8 487 PREINIT:
26f802f7 488 HE *slot;
194acf47 489 CODE:
490 if (!sv_isobject(self))
491 croak("Can't call name as a class method");
8860963a 492 if (slot = hv_fetch_ent((HV*)SvRV(self), name_key, 0, name_hash)) {
493 RETVAL = SvREFCNT_inc_simple_NN(HeVAL(slot));
494 }
495 else {
496 croak("Can't get the name of an anonymous package");
497 }
194acf47 498 OUTPUT:
499 RETVAL
500
501SV*
502namespace(self)
503 SV *self
3496f1e8 504 PREINIT:
26f802f7 505 HE *slot;
7c3b261c 506 SV *package_name;
194acf47 507 CODE:
508 if (!sv_isobject(self))
509 croak("Can't call namespace as a class method");
7c3b261c 510#if PERL_VERSION < 10
511 package_name = _get_name(self);
512 RETVAL = newRV_inc((SV*)gv_stashpv(SvPV_nolen(package_name), GV_ADD));
513#else
26f802f7 514 slot = hv_fetch_ent((HV*)SvRV(self), namespace_key, 0, namespace_hash);
31263168 515 if (slot) {
516 RETVAL = SvREFCNT_inc_simple_NN(HeVAL(slot));
517 }
518 else {
519 HV *namespace;
7c3b261c 520 SV *nsref;
31263168 521
522 package_name = _get_name(self);
523 namespace = gv_stashpv(SvPV_nolen(package_name), GV_ADD);
524 nsref = newRV_inc((SV*)namespace);
525 sv_rvweaken(nsref);
526 if (!hv_store((HV*)SvRV(self), "namespace", 9, nsref, 0)) {
527 SvREFCNT_dec(nsref);
528 SvREFCNT_dec(self);
529 croak("Couldn't initialize the 'namespace' key, hv_store failed");
530 }
531 RETVAL = SvREFCNT_inc_simple_NN(nsref);
532 }
7c3b261c 533#endif
194acf47 534 OUTPUT:
535 RETVAL
3fd56b4d 536
537void
15c104e2 538add_symbol(self, variable, initial=NULL, ...)
76f7c306 539 SV *self
540 varspec_t variable
541 SV *initial
542 PREINIT:
76f7c306 543 GV *glob;
e290a036 544 HV *namespace;
545 HE *entry;
76f7c306 546 CODE:
547 if (initial && !_valid_for_type(initial, variable.type))
548 croak("%s is not of type %s",
549 SvPV_nolen(initial), vartype_to_string(variable.type));
550
76f7c306 551 if (items > 2 && (PL_perldb & 0x10) && variable.type == VAR_CODE) {
552 int i;
4849f4fc 553 char *filename = NULL;
d1d0e437 554 I32 first_line_num = -1, last_line_num = -1;
e290a036 555 SV *dbval, *name;
d1d0e437 556 HV *dbsub;
76f7c306 557
558 if ((items - 3) % 2)
15c104e2 559 croak("add_symbol: Odd number of elements in %%opts");
76f7c306 560
561 for (i = 3; i < items; i += 2) {
562 char *key;
563 key = SvPV_nolen(ST(i));
564 if (strEQ(key, "filename")) {
565 if (!SvPOK(ST(i + 1)))
15c104e2 566 croak("add_symbol: filename must be a string");
76f7c306 567 filename = SvPV_nolen(ST(i + 1));
568 }
569 else if (strEQ(key, "first_line_num")) {
570 if (!SvIOK(ST(i + 1)))
15c104e2 571 croak("add_symbol: first_line_num must be an integer");
76f7c306 572 first_line_num = SvIV(ST(i + 1));
573 }
574 else if (strEQ(key, "last_line_num")) {
575 if (!SvIOK(ST(i + 1)))
15c104e2 576 croak("add_symbol: last_line_num must be an integer");
76f7c306 577 last_line_num = SvIV(ST(i + 1));
578 }
579 }
580
d1d0e437 581 if (!filename || first_line_num == -1) {
d1d0e437 582 if (!filename)
e0162997 583 filename = CopFILE(PL_curcop);
d1d0e437 584 if (first_line_num == -1)
e0162997 585 first_line_num = PL_curcop->cop_line;
d1d0e437 586 }
587
588 if (last_line_num == -1)
589 last_line_num = first_line_num;
590
e290a036 591 name = newSVsv(_get_name(self));
592 sv_catpvs(name, "::");
593 sv_catsv(name, variable.name);
594
d1d0e437 595 /* http://perldoc.perl.org/perldebguts.html#Debugger-Internals */
596 dbsub = get_hv("DB::sub", 1);
597 dbval = newSVpvf("%s:%d-%d", filename, first_line_num, last_line_num);
4849f4fc 598 if (!hv_store_ent(dbsub, name, dbval, 0)) {
599 warn("Failed to update $DB::sub for subroutine %s",
600 SvPV_nolen(name));
d1d0e437 601 SvREFCNT_dec(dbval);
76f7c306 602 }
e290a036 603
604 SvREFCNT_dec(name);
76f7c306 605 }
76f7c306 606
9aa6fe4f 607 /* GV_ADDMULTI rather than GV_ADD because otherwise you get 'used only
608 * once' warnings in some situations... i can't reproduce this, but CMOP
609 * triggers it */
e290a036 610 namespace = _get_namespace(self);
611 entry = hv_fetch_ent(namespace, variable.name, 0, 0);
612 if (entry) {
613 glob = (GV*)HeVAL(entry);
614 }
615 else {
616 glob = (GV*)newSV(0);
e006ab79 617 _real_gv_init(glob, namespace, variable.name);
e290a036 618 if (!hv_store_ent(namespace, variable.name, (SV*)glob, 0)) {
619 croak("hv_store failed");
620 }
621 }
76f7c306 622
623 if (initial) {
624 SV *val;
625
626 if (SvROK(initial)) {
627 val = SvRV(initial);
a2fec41a 628 SvREFCNT_inc_simple_void_NN(val);
76f7c306 629 }
630 else {
631 val = newSVsv(initial);
632 }
633
634 switch (variable.type) {
635 case VAR_SCALAR:
d551a208 636 GvSetSV(glob, val);
76f7c306 637 break;
638 case VAR_ARRAY:
d551a208 639 GvSetAV(glob, val);
76f7c306 640 break;
641 case VAR_HASH:
d551a208 642 GvSetHV(glob, val);
76f7c306 643 break;
644 case VAR_CODE:
d551a208 645 GvSetCV(glob, val);
76f7c306 646 break;
647 case VAR_IO:
d551a208 648 GvSetIO(glob, val);
76f7c306 649 break;
650 }
651 }
652
653void
4849f4fc 654remove_glob(self, name)
3fd56b4d 655 SV *self
4849f4fc 656 SV *name
3fd56b4d 657 CODE:
4849f4fc 658 hv_delete_ent(_get_namespace(self), name, G_DISCARD, 0);
9b608466 659
34805376 660int
15c104e2 661has_symbol(self, variable)
34805376 662 SV *self
663 varspec_t variable
664 PREINIT:
665 HV *namespace;
4849f4fc 666 HE *entry;
667 SV *val;
34805376 668 CODE:
669 namespace = _get_namespace(self);
4849f4fc 670 entry = hv_fetch_ent(namespace, variable.name, 0, 0);
34805376 671 if (!entry)
672 XSRETURN_UNDEF;
673
4849f4fc 674 val = HeVAL(entry);
675 if (isGV(val)) {
676 GV *glob = (GV*)val;
34805376 677 switch (variable.type) {
678 case VAR_SCALAR:
d551a208 679 RETVAL = GvSVOK(glob) ? 1 : 0;
34805376 680 break;
681 case VAR_ARRAY:
d551a208 682 RETVAL = GvAVOK(glob) ? 1 : 0;
34805376 683 break;
684 case VAR_HASH:
d551a208 685 RETVAL = GvHVOK(glob) ? 1 : 0;
34805376 686 break;
687 case VAR_CODE:
d551a208 688 RETVAL = GvCVOK(glob) ? 1 : 0;
34805376 689 break;
690 case VAR_IO:
d551a208 691 RETVAL = GvIOOK(glob) ? 1 : 0;
34805376 692 break;
693 }
694 }
695 else {
696 RETVAL = (variable.type == VAR_CODE);
697 }
698 OUTPUT:
699 RETVAL
700
e3ad44fd 701SV*
15c104e2 702get_symbol(self, variable)
e3ad44fd 703 SV *self
704 varspec_t variable
705 PREINIT:
fca4ed0c 706 SV *val;
e3ad44fd 707 CODE:
15c104e2 708 val = _get_symbol(self, &variable, 0);
4871b1a0 709 if (!val)
e3ad44fd 710 XSRETURN_UNDEF;
93c36417 711 RETVAL = newRV_inc(val);
4871b1a0 712 OUTPUT:
713 RETVAL
e3ad44fd 714
4871b1a0 715SV*
15c104e2 716get_or_add_symbol(self, variable)
4871b1a0 717 SV *self
718 varspec_t variable
719 PREINIT:
720 SV *val;
721 CODE:
15c104e2 722 val = _get_symbol(self, &variable, 1);
fca4ed0c 723 if (!val)
724 XSRETURN_UNDEF;
93c36417 725 RETVAL = newRV_inc(val);
e3ad44fd 726 OUTPUT:
727 RETVAL
728
9b608466 729void
15c104e2 730remove_symbol(self, variable)
215f49f8 731 SV *self
732 varspec_t variable
733 PREINIT:
734 HV *namespace;
4849f4fc 735 HE *entry;
736 SV *val;
215f49f8 737 CODE:
738 namespace = _get_namespace(self);
4849f4fc 739 entry = hv_fetch_ent(namespace, variable.name, 0, 0);
215f49f8 740 if (!entry)
741 XSRETURN_EMPTY;
742
4849f4fc 743 val = HeVAL(entry);
744 if (isGV(val)) {
745 GV *glob = (GV*)val;
215f49f8 746 switch (variable.type) {
747 case VAR_SCALAR:
d551a208 748 GvSetSV(glob, NULL);
215f49f8 749 break;
750 case VAR_ARRAY:
d551a208 751 GvSetAV(glob, NULL);
215f49f8 752 break;
753 case VAR_HASH:
d551a208 754 GvSetHV(glob, NULL);
215f49f8 755 break;
756 case VAR_CODE:
d551a208 757 GvSetCV(glob, NULL);
215f49f8 758 break;
759 case VAR_IO:
d551a208 760 GvSetIO(glob, NULL);
215f49f8 761 break;
762 }
763 }
764 else {
765 if (variable.type == VAR_CODE) {
4849f4fc 766 hv_delete_ent(namespace, variable.name, G_DISCARD, 0);
215f49f8 767 }
768 }
769
770void
15c104e2 771list_all_symbols(self, vartype=VAR_NONE)
9b608466 772 SV *self
773 vartype_t vartype
774 PPCODE:
775 if (vartype == VAR_NONE) {
776 HV *namespace;
777 HE *entry;
778 int keys;
779
780 namespace = _get_namespace(self);
781 keys = hv_iterinit(namespace);
782 EXTEND(SP, keys);
ebcc8ba8 783 while ((entry = hv_iternext(namespace))) {
13f8a7b7 784 mPUSHs(newSVhe(entry));
9b608466 785 }
786 }
787 else {
788 HV *namespace;
9b608466 789 SV *val;
790 char *key;
64b79211 791 I32 len;
9b608466 792
793 namespace = _get_namespace(self);
794 hv_iterinit(namespace);
ebcc8ba8 795 while ((val = hv_iternextsv(namespace, &key, &len))) {
9b608466 796 GV *gv = (GV*)val;
797 if (isGV(gv)) {
798 switch (vartype) {
799 case VAR_SCALAR:
d551a208 800 if (GvSVOK(val))
9b608466 801 mXPUSHp(key, len);
802 break;
803 case VAR_ARRAY:
d551a208 804 if (GvAVOK(val))
9b608466 805 mXPUSHp(key, len);
806 break;
807 case VAR_HASH:
d551a208 808 if (GvHVOK(val))
9b608466 809 mXPUSHp(key, len);
810 break;
811 case VAR_CODE:
d551a208 812 if (GvCVOK(val))
9b608466 813 mXPUSHp(key, len);
814 break;
815 case VAR_IO:
d551a208 816 if (GvIOOK(val))
9b608466 817 mXPUSHp(key, len);
818 break;
819 }
820 }
821 else if (vartype == VAR_CODE) {
822 mXPUSHp(key, len);
823 }
824 }
825 }
26f802f7 826
d2b55565 827void
828get_all_symbols(self, vartype=VAR_NONE)
829 SV *self
830 vartype_t vartype
831 PREINIT:
832 HV *namespace, *ret;
833 SV *val;
834 char *key;
835 I32 len;
836 PPCODE:
837 namespace = _get_namespace(self);
838 ret = newHV();
839
840 hv_iterinit(namespace);
841 while ((val = hv_iternextsv(namespace, &key, &len))) {
842 GV *gv = (GV*)val;
843
4849f4fc 844 if (!isGV(gv)) {
845 SV *keysv = newSVpvn(key, len);
846 _expand_glob(self, keysv);
847 SvREFCNT_dec(keysv);
848 }
d2b55565 849
850 switch (vartype) {
851 case VAR_SCALAR:
852 if (GvSVOK(val))
853 hv_store(ret, key, len, newRV_inc(GvSV(gv)), 0);
854 break;
855 case VAR_ARRAY:
856 if (GvAVOK(val))
857 hv_store(ret, key, len, newRV_inc((SV*)GvAV(gv)), 0);
858 break;
859 case VAR_HASH:
860 if (GvHVOK(val))
861 hv_store(ret, key, len, newRV_inc((SV*)GvHV(gv)), 0);
862 break;
863 case VAR_CODE:
864 if (GvCVOK(val))
865 hv_store(ret, key, len, newRV_inc((SV*)GvCV(gv)), 0);
866 break;
867 case VAR_IO:
868 if (GvIOOK(val))
869 hv_store(ret, key, len, newRV_inc((SV*)GvIO(gv)), 0);
870 break;
871 case VAR_NONE:
872 hv_store(ret, key, len, SvREFCNT_inc_simple_NN(val), 0);
873 break;
874 }
875 }
876
877 mPUSHs(newRV_noinc((SV*)ret));
878
26f802f7 879BOOT:
880 {
e1b6c960 881 const char *vmre = "\\A[0-9A-Z_a-z]+(?:::[0-9A-Z_a-z]+)*\\z";
882#if (PERL_VERSION < 9) || ((PERL_VERSION == 9) && (PERL_SUBVERSION < 5))
883 PMOP fakepmop;
884
885 fakepmop.op_pmflags = 0;
886 valid_module_regex = pregcomp(vmre, vmre + strlen(vmre), &fakepmop);
887#else
78daf8ed 888 SV *re;
889
e1b6c960 890 re = newSVpv(vmre, 0);
78daf8ed 891 valid_module_regex = pregcomp(re, 0);
e1b6c960 892#endif
78daf8ed 893
26f802f7 894 name_key = newSVpvs("name");
895 PERL_HASH(name_hash, "name", 4);
896
897 namespace_key = newSVpvs("namespace");
898 PERL_HASH(namespace_hash, "namespace", 9);
899
900 type_key = newSVpvs("type");
901 PERL_HASH(type_hash, "type", 4);
902 }