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 #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
  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
  70 /* see above - don't let scalar slots become unpopulated, this breaks
  71  * assumptions in core */
  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
  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)
  85 #endif
  86
  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));              \
  99     if ((GvCV_set(g, v))) {             \
 100         GvIMPORTED_CV_on(g);            \
 101         GvASSUMECV_on(g);               \
 102     }                                   \
 103     GvCVGEN(g) = 0;                     \
 104     mro_method_changed_in(GvSTASH(g));  \
 105 } while (0)
 106 #define GvSetIO(g,v) do {               \
 107     SvREFCNT_dec(GvIO(g));              \
 108     GvIOp(g) = (IO*)(v);                \
 109 } while (0)
 110
 111 typedef enum {
 112     VAR_NONE = 0,
 113     VAR_SCALAR,
 114     VAR_ARRAY,
 115     VAR_HASH,
 116     VAR_CODE,
 117     VAR_IO,
 118     VAR_GLOB,  /* TODO: unimplemented */
 119     VAR_FORMAT /* TODO: unimplemented */
 120 } vartype_t;
 121
 122 typedef struct {
 123     vartype_t type;
 124     SV *name;
 125 } varspec_t;
 126
 127 static U32 name_hash, namespace_hash, type_hash;
 128 static SV *name_key, *namespace_key, *type_key;
 129 static REGEXP *valid_module_regex;
 130
 131 static const char *vartype_to_string(vartype_t type)
 132 {
 133     switch (type) {
 134     case VAR_SCALAR:
 135         return "SCALAR";
 136     case VAR_ARRAY:
 137         return "ARRAY";
 138     case VAR_HASH:
 139         return "HASH";
 140     case VAR_CODE:
 141         return "CODE";
 142     case VAR_IO:
 143         return "IO";
 144     default:
 145         return "unknown";
 146     }
 147 }
 148
 149 static I32 vartype_to_svtype(vartype_t type)
 150 {
 151     switch (type) {
 152     case VAR_SCALAR:
 153         return SVt_PV; /* or whatever */
 154     case VAR_ARRAY:
 155         return SVt_PVAV;
 156     case VAR_HASH:
 157         return SVt_PVHV;
 158     case VAR_CODE:
 159         return SVt_PVCV;
 160     case VAR_IO:
 161         return SVt_PVIO;
 162     default:
 163         return SVt_NULL;
 164     }
 165 }
 166
 167 static vartype_t string_to_vartype(char *vartype)
 168 {
 169     if (strEQ(vartype, "SCALAR")) {
 170         return VAR_SCALAR;
 171     }
 172     else if (strEQ(vartype, "ARRAY")) {
 173         return VAR_ARRAY;
 174     }
 175     else if (strEQ(vartype, "HASH")) {
 176         return VAR_HASH;
 177     }
 178     else if (strEQ(vartype, "CODE")) {
 179         return VAR_CODE;
 180     }
 181     else if (strEQ(vartype, "IO")) {
 182         return VAR_IO;
 183     }
 184     else {
 185         croak("Type must be one of 'SCALAR', 'ARRAY', 'HASH', 'CODE', or 'IO', not '%s'", vartype);
 186     }
 187 }
 188
 189 static int _is_valid_module_name(SV *package)
 190 {
 191     char *buf;
 192     STRLEN len;
 193     SV *sv;
 194
 195     buf = SvPV(package, len);
 196
 197     /* whee cargo cult */
 198     sv = sv_newmortal();
 199     sv_upgrade(sv, SVt_PV);
 200     SvREADONLY_on(sv);
 201     SvLEN(sv) = 0;
 202     SvUTF8_on(sv);
 203     SvPVX(sv) = buf;
 204     SvCUR_set(sv, len);
 205     SvPOK_on(sv);
 206
 207     return pregexec(valid_module_regex, buf, buf + len, buf, 1, sv, 1);
 208 }
 209
 210 static void _deconstruct_variable_name(SV *variable, varspec_t *varspec)
 211 {
 212     char *varpv;
 213
 214     if (!SvCUR(variable))
 215         croak("You must pass a variable name");
 216
 217     varspec->name = sv_2mortal(newSVsv(variable));
 218
 219     varpv = SvPV_nolen(varspec->name);
 220     switch (varpv[0]) {
 221     case '$':
 222         varspec->type = VAR_SCALAR;
 223         sv_chop(varspec->name, &varpv[1]);
 224         break;
 225     case '@':
 226         varspec->type = VAR_ARRAY;
 227         sv_chop(varspec->name, &varpv[1]);
 228         break;
 229     case '%':
 230         varspec->type = VAR_HASH;
 231         sv_chop(varspec->name, &varpv[1]);
 232         break;
 233     case '&':
 234         varspec->type = VAR_CODE;
 235         sv_chop(varspec->name, &varpv[1]);
 236         break;
 237     default:
 238         varspec->type = VAR_IO;
 239         break;
 240     }
 241 }
 242
 243 static void _deconstruct_variable_hash(HV *variable, varspec_t *varspec)
 244 {
 245     HE *val;
 246
 247     val = hv_fetch_ent(variable, name_key, 0, name_hash);
 248     if (!val)
 249         croak("The 'name' key is required in variable specs");
 250
 251     varspec->name = sv_2mortal(newSVsv(HeVAL(val)));
 252
 253     val = hv_fetch_ent(variable, type_key, 0, type_hash);
 254     if (!val)
 255         croak("The 'type' key is required in variable specs");
 256
 257     varspec->type = string_to_vartype(SvPV_nolen(HeVAL(val)));
 258 }
 259
 260 static void _check_varspec_is_valid(varspec_t *varspec)
 261 {
 262     if (strstr(SvPV_nolen(varspec->name), "::")) {
 263         croak("Variable names may not contain ::");
 264     }
 265 }
 266
 267 static int _valid_for_type(SV *value, vartype_t type)
 268 {
 269     svtype sv_type = SvROK(value) ? SvTYPE(SvRV(value)) : SVt_NULL;
 270
 271     switch (type) {
 272     case VAR_SCALAR:
 273         /* XXX is a glob a scalar? assigning a glob to the scalar slot seems
 274          * to work here, but in pure perl i'm pretty sure it goes to the EGV
 275          * slot, which seems more correct to me. just disable it for now
 276          * i guess */
 277         return SVT_SCALAR(sv_type) && sv_type != SVt_PVGV;
 278     case VAR_ARRAY:
 279         return SVT_ARRAY(sv_type);
 280     case VAR_HASH:
 281         return SVT_HASH(sv_type);
 282     case VAR_CODE:
 283         return SVT_CODE(sv_type);
 284     case VAR_IO:
 285         return SVT_IO(sv_type);
 286     default:
 287         return 0;
 288     }
 289 }
 290
 291 static HV *_get_namespace(SV *self)
 292 {
 293     dSP;
 294     SV *ret;
 295
 296     PUSHMARK(SP);
 297     XPUSHs(self);
 298     PUTBACK;
 299
 300     call_method("namespace", G_SCALAR);
 301
 302     SPAGAIN;
 303     ret = POPs;
 304     PUTBACK;
 305
 306     return (HV*)SvRV(ret);
 307 }
 308
 309 static SV *_get_name(SV *self)
 310 {
 311     dSP;
 312     SV *ret;
 313
 314     PUSHMARK(SP);
 315     XPUSHs(self);
 316     PUTBACK;
 317
 318     call_method("name", G_SCALAR);
 319
 320     SPAGAIN;
 321     ret = POPs;
 322     PUTBACK;
 323
 324     return ret;
 325 }
 326
 327 static void _expand_glob(SV *self, SV *varname)
 328 {
 329     SV *name;
 330
 331     name = newSVsv(_get_name(self));
 332     sv_catpvs(name, "::");
 333     sv_catsv(name, varname);
 334
 335     /* can't use gv_init here, because it screws up @ISA in a way that I
 336      * can't reproduce, but that CMOP triggers */
 337     gv_fetchsv(name, GV_ADD, SVt_NULL);
 338     SvREFCNT_dec(name);
 339 }
 340
 341 static SV *_get_symbol(SV *self, varspec_t *variable, int vivify)
 342 {
 343     HV *namespace;
 344     HE *entry;
 345     GV *glob;
 346
 347     namespace = _get_namespace(self);
 348     entry = hv_fetch_ent(namespace, variable->name, vivify, 0);
 349     if (!entry)
 350         return NULL;
 351
 352     glob = (GV*)(HeVAL(entry));
 353     if (!isGV(glob))
 354         _expand_glob(self, variable->name);
 355
 356     if (vivify) {
 357         switch (variable->type) {
 358         case VAR_SCALAR:
 359             if (!GvSVOK(glob))
 360                 GvSetSV(glob, newSV(0));
 361             break;
 362         case VAR_ARRAY:
 363             if (!GvAVOK(glob))
 364                 GvSetAV(glob, newAV());
 365             break;
 366         case VAR_HASH:
 367             if (!GvHVOK(glob))
 368                 GvSetHV(glob, newHV());
 369             break;
 370         case VAR_CODE:
 371             croak("Don't know how to vivify CODE variables");
 372         case VAR_IO:
 373             if (!GvIOOK(glob))
 374                 GvSetIO(glob, newIO());
 375             break;
 376         default:
 377             croak("Unknown type in vivication");
 378         }
 379     }
 380
 381     switch (variable->type) {
 382     case VAR_SCALAR:
 383         return GvSV(glob);
 384     case VAR_ARRAY:
 385         return (SV*)GvAV(glob);
 386     case VAR_HASH:
 387         return (SV*)GvHV(glob);
 388     case VAR_CODE:
 389         return (SV*)GvCV(glob);
 390     case VAR_IO:
 391         return (SV*)GvIO(glob);
 392     default:
 393         return NULL;
 394     }
 395 }
 396
 397 MODULE = Package::Stash::XS  PACKAGE = Package::Stash::XS
 398
 399 PROTOTYPES: DISABLE
 400
 401 SV*
 402 new(class, package_name)
 403     SV *class
 404     SV *package_name
 405   PREINIT:
 406     HV *instance;
 407   CODE:
 408     if (!SvPOK(package_name))
 409         croak("Package::Stash->new must be passed the name of the package to access");
 410
 411     if (!_is_valid_module_name(package_name))
 412         croak("%s is not a module name", SvPV_nolen(package_name));
 413
 414     instance = newHV();
 415
 416     if (!hv_store(instance, "name", 4, SvREFCNT_inc_simple_NN(package_name), 0)) {
 417         SvREFCNT_dec(package_name);
 418         SvREFCNT_dec(instance);
 419         croak("Couldn't initialize the 'name' key, hv_store failed");
 420     }
 421
 422     RETVAL = sv_bless(newRV_noinc((SV*)instance), gv_stashsv(class, 0));
 423   OUTPUT:
 424     RETVAL
 425
 426 SV*
 427 name(self)
 428     SV *self
 429   PREINIT:
 430     HE *slot;
 431   CODE:
 432     if (!sv_isobject(self))
 433         croak("Can't call name as a class method");
 434     slot = hv_fetch_ent((HV*)SvRV(self), name_key, 0, name_hash);
 435     RETVAL = slot ? SvREFCNT_inc_simple_NN(HeVAL(slot)) : &PL_sv_undef;
 436   OUTPUT:
 437     RETVAL
 438
 439 SV*
 440 namespace(self)
 441     SV *self
 442   PREINIT:
 443     HE *slot;
 444     SV *package_name;
 445   CODE:
 446     if (!sv_isobject(self))
 447         croak("Can't call namespace as a class method");
 448 #if PERL_VERSION < 10
 449     package_name = _get_name(self);
 450     RETVAL = newRV_inc((SV*)gv_stashpv(SvPV_nolen(package_name), GV_ADD));
 451 #else
 452     slot = hv_fetch_ent((HV*)SvRV(self), namespace_key, 0, namespace_hash);
 453     if (slot) {
 454         RETVAL = SvREFCNT_inc_simple_NN(HeVAL(slot));
 455     }
 456     else {
 457         HV *namespace;
 458         SV *nsref;
 459
 460         package_name = _get_name(self);
 461         namespace = gv_stashpv(SvPV_nolen(package_name), GV_ADD);
 462         nsref = newRV_inc((SV*)namespace);
 463         sv_rvweaken(nsref);
 464         if (!hv_store((HV*)SvRV(self), "namespace", 9, nsref, 0)) {
 465             SvREFCNT_dec(nsref);
 466             SvREFCNT_dec(self);
 467             croak("Couldn't initialize the 'namespace' key, hv_store failed");
 468         }
 469         RETVAL = SvREFCNT_inc_simple_NN(nsref);
 470     }
 471 #endif
 472   OUTPUT:
 473     RETVAL
 474
 475 void
 476 add_symbol(self, variable, initial=NULL, ...)
 477     SV *self
 478     varspec_t variable
 479     SV *initial
 480   PREINIT:
 481     SV *name;
 482     GV *glob;
 483   CODE:
 484     if (initial && !_valid_for_type(initial, variable.type))
 485         croak("%s is not of type %s",
 486               SvPV_nolen(initial), vartype_to_string(variable.type));
 487
 488     name = newSVsv(_get_name(self));
 489     sv_catpvs(name, "::");
 490     sv_catsv(name, variable.name);
 491
 492     if (items > 2 && (PL_perldb & 0x10) && variable.type == VAR_CODE) {
 493         int i;
 494         char *filename = NULL;
 495         I32 first_line_num = -1, last_line_num = -1;
 496         SV *dbval;
 497         HV *dbsub;
 498
 499         if ((items - 3) % 2)
 500             croak("add_symbol: Odd number of elements in %%opts");
 501
 502         for (i = 3; i < items; i += 2) {
 503             char *key;
 504             key = SvPV_nolen(ST(i));
 505             if (strEQ(key, "filename")) {
 506                 if (!SvPOK(ST(i + 1)))
 507                     croak("add_symbol: filename must be a string");
 508                 filename = SvPV_nolen(ST(i + 1));
 509             }
 510             else if (strEQ(key, "first_line_num")) {
 511                 if (!SvIOK(ST(i + 1)))
 512                     croak("add_symbol: first_line_num must be an integer");
 513                 first_line_num = SvIV(ST(i + 1));
 514             }
 515             else if (strEQ(key, "last_line_num")) {
 516                 if (!SvIOK(ST(i + 1)))
 517                     croak("add_symbol: last_line_num must be an integer");
 518                 last_line_num = SvIV(ST(i + 1));
 519             }
 520         }
 521
 522         if (!filename || first_line_num == -1) {
 523             if (!filename)
 524                 filename = CopFILE(PL_curcop);
 525             if (first_line_num == -1)
 526                 first_line_num = PL_curcop->cop_line;
 527         }
 528
 529         if (last_line_num == -1)
 530             last_line_num = first_line_num;
 531
 532         /* http://perldoc.perl.org/perldebguts.html#Debugger-Internals */
 533         dbsub = get_hv("DB::sub", 1);
 534         dbval = newSVpvf("%s:%d-%d", filename, first_line_num, last_line_num);
 535         if (!hv_store_ent(dbsub, name, dbval, 0)) {
 536             warn("Failed to update $DB::sub for subroutine %s",
 537                  SvPV_nolen(name));
 538             SvREFCNT_dec(dbval);
 539         }
 540     }
 541
 542     /* GV_ADDMULTI rather than GV_ADD because otherwise you get 'used only
 543      * once' warnings in some situations... i can't reproduce this, but CMOP
 544      * triggers it */
 545     glob = gv_fetchsv(name, GV_ADDMULTI, vartype_to_svtype(variable.type));
 546
 547     if (initial) {
 548         SV *val;
 549
 550         if (SvROK(initial)) {
 551             val = SvRV(initial);
 552             SvREFCNT_inc_simple_void_NN(val);
 553         }
 554         else {
 555             val = newSVsv(initial);
 556         }
 557
 558         switch (variable.type) {
 559         case VAR_SCALAR:
 560             GvSetSV(glob, val);
 561             break;
 562         case VAR_ARRAY:
 563             GvSetAV(glob, val);
 564             break;
 565         case VAR_HASH:
 566             GvSetHV(glob, val);
 567             break;
 568         case VAR_CODE:
 569             GvSetCV(glob, val);
 570             break;
 571         case VAR_IO:
 572             GvSetIO(glob, val);
 573             break;
 574         }
 575     }
 576
 577     SvREFCNT_dec(name);
 578
 579 void
 580 remove_glob(self, name)
 581     SV *self
 582     SV *name
 583   CODE:
 584     hv_delete_ent(_get_namespace(self), name, G_DISCARD, 0);
 585
 586 int
 587 has_symbol(self, variable)
 588     SV *self
 589     varspec_t variable
 590   PREINIT:
 591     HV *namespace;
 592     HE *entry;
 593     SV *val;
 594   CODE:
 595     namespace = _get_namespace(self);
 596     entry = hv_fetch_ent(namespace, variable.name, 0, 0);
 597     if (!entry)
 598         XSRETURN_UNDEF;
 599
 600     val = HeVAL(entry);
 601     if (isGV(val)) {
 602         GV *glob = (GV*)val;
 603         switch (variable.type) {
 604         case VAR_SCALAR:
 605             RETVAL = GvSVOK(glob) ? 1 : 0;
 606             break;
 607         case VAR_ARRAY:
 608             RETVAL = GvAVOK(glob) ? 1 : 0;
 609             break;
 610         case VAR_HASH:
 611             RETVAL = GvHVOK(glob) ? 1 : 0;
 612             break;
 613         case VAR_CODE:
 614             RETVAL = GvCVOK(glob) ? 1 : 0;
 615             break;
 616         case VAR_IO:
 617             RETVAL = GvIOOK(glob) ? 1 : 0;
 618             break;
 619         }
 620     }
 621     else {
 622         RETVAL = (variable.type == VAR_CODE);
 623     }
 624   OUTPUT:
 625     RETVAL
 626
 627 SV*
 628 get_symbol(self, variable)
 629     SV *self
 630     varspec_t variable
 631   PREINIT:
 632     SV *val;
 633   CODE:
 634     val = _get_symbol(self, &variable, 0);
 635     if (!val)
 636         XSRETURN_UNDEF;
 637     RETVAL = newRV_inc(val);
 638   OUTPUT:
 639     RETVAL
 640
 641 SV*
 642 get_or_add_symbol(self, variable)
 643     SV *self
 644     varspec_t variable
 645   PREINIT:
 646     SV *val;
 647   CODE:
 648     val = _get_symbol(self, &variable, 1);
 649     if (!val)
 650         XSRETURN_UNDEF;
 651     RETVAL = newRV_inc(val);
 652   OUTPUT:
 653     RETVAL
 654
 655 void
 656 remove_symbol(self, variable)
 657     SV *self
 658     varspec_t variable
 659   PREINIT:
 660     HV *namespace;
 661     HE *entry;
 662     SV *val;
 663   CODE:
 664     namespace = _get_namespace(self);
 665     entry = hv_fetch_ent(namespace, variable.name, 0, 0);
 666     if (!entry)
 667         XSRETURN_EMPTY;
 668
 669     val = HeVAL(entry);
 670     if (isGV(val)) {
 671         GV *glob = (GV*)val;
 672         switch (variable.type) {
 673         case VAR_SCALAR:
 674             GvSetSV(glob, NULL);
 675             break;
 676         case VAR_ARRAY:
 677             GvSetAV(glob, NULL);
 678             break;
 679         case VAR_HASH:
 680             GvSetHV(glob, NULL);
 681             break;
 682         case VAR_CODE:
 683             GvSetCV(glob, NULL);
 684             break;
 685         case VAR_IO:
 686             GvSetIO(glob, NULL);
 687             break;
 688         }
 689     }
 690     else {
 691         if (variable.type == VAR_CODE) {
 692             hv_delete_ent(namespace, variable.name, G_DISCARD, 0);
 693         }
 694     }
 695
 696 void
 697 list_all_symbols(self, vartype=VAR_NONE)
 698     SV *self
 699     vartype_t vartype
 700   PPCODE:
 701     if (vartype == VAR_NONE) {
 702         HV *namespace;
 703         HE *entry;
 704         int keys;
 705
 706         namespace = _get_namespace(self);
 707         keys = hv_iterinit(namespace);
 708         EXTEND(SP, keys);
 709         while ((entry = hv_iternext(namespace))) {
 710             mPUSHs(newSVhe(entry));
 711         }
 712     }
 713     else {
 714         HV *namespace;
 715         SV *val;
 716         char *key;
 717         I32 len;
 718
 719         namespace = _get_namespace(self);
 720         hv_iterinit(namespace);
 721         while ((val = hv_iternextsv(namespace, &key, &len))) {
 722             GV *gv = (GV*)val;
 723             if (isGV(gv)) {
 724                 switch (vartype) {
 725                 case VAR_SCALAR:
 726                     if (GvSVOK(val))
 727                         mXPUSHp(key, len);
 728                     break;
 729                 case VAR_ARRAY:
 730                     if (GvAVOK(val))
 731                         mXPUSHp(key, len);
 732                     break;
 733                 case VAR_HASH:
 734                     if (GvHVOK(val))
 735                         mXPUSHp(key, len);
 736                     break;
 737                 case VAR_CODE:
 738                     if (GvCVOK(val))
 739                         mXPUSHp(key, len);
 740                     break;
 741                 case VAR_IO:
 742                     if (GvIOOK(val))
 743                         mXPUSHp(key, len);
 744                     break;
 745                 }
 746             }
 747             else if (vartype == VAR_CODE) {
 748                 mXPUSHp(key, len);
 749             }
 750         }
 751     }
 752
 753 void
 754 get_all_symbols(self, vartype=VAR_NONE)
 755     SV *self
 756     vartype_t vartype
 757   PREINIT:
 758     HV *namespace, *ret;
 759     SV *val;
 760     char *key;
 761     I32 len;
 762   PPCODE:
 763     namespace = _get_namespace(self);
 764     ret = newHV();
 765
 766     hv_iterinit(namespace);
 767     while ((val = hv_iternextsv(namespace, &key, &len))) {
 768         GV *gv = (GV*)val;
 769
 770         if (!isGV(gv)) {
 771             SV *keysv = newSVpvn(key, len);
 772             _expand_glob(self, keysv);
 773             SvREFCNT_dec(keysv);
 774         }
 775
 776         switch (vartype) {
 777         case VAR_SCALAR:
 778             if (GvSVOK(val))
 779                 hv_store(ret, key, len, newRV_inc(GvSV(gv)), 0);
 780             break;
 781         case VAR_ARRAY:
 782             if (GvAVOK(val))
 783                 hv_store(ret, key, len, newRV_inc((SV*)GvAV(gv)), 0);
 784             break;
 785         case VAR_HASH:
 786             if (GvHVOK(val))
 787                 hv_store(ret, key, len, newRV_inc((SV*)GvHV(gv)), 0);
 788             break;
 789         case VAR_CODE:
 790             if (GvCVOK(val))
 791                 hv_store(ret, key, len, newRV_inc((SV*)GvCV(gv)), 0);
 792             break;
 793         case VAR_IO:
 794             if (GvIOOK(val))
 795                 hv_store(ret, key, len, newRV_inc((SV*)GvIO(gv)), 0);
 796             break;
 797         case VAR_NONE:
 798             hv_store(ret, key, len, SvREFCNT_inc_simple_NN(val), 0);
 799             break;
 800         }
 801     }
 802
 803     mPUSHs(newRV_noinc((SV*)ret));
 804
 805 BOOT:
 806     {
 807         const char *vmre = "\\A[0-9A-Z_a-z]+(?:::[0-9A-Z_a-z]+)*\\z";
 808 #if (PERL_VERSION < 9) || ((PERL_VERSION == 9) && (PERL_SUBVERSION < 5))
 809         PMOP fakepmop;
 810
 811         fakepmop.op_pmflags = 0;
 812         valid_module_regex = pregcomp(vmre, vmre + strlen(vmre), &fakepmop);
 813 #else
 814         SV *re;
 815
 816         re = newSVpv(vmre, 0);
 817         valid_module_regex = pregcomp(re, 0);
 818 #endif
 819
 820         name_key = newSVpvs("name");
 821         PERL_HASH(name_hash, "name", 4);
 822
 823         namespace_key = newSVpvs("namespace");
 824         PERL_HASH(namespace_hash, "namespace", 9);
 825
 826         type_key = newSVpvs("type");
 827         PERL_HASH(type_hash, "type", 4);
 828     }