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, (CV*)(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     HV *namespace;
 330     HE *entry;
 331     GV *glob;
 332
 333     namespace = _get_namespace(self);
 334
 335     if (entry = hv_fetch_ent(namespace, varname, 0, 0)) {
 336         glob = (GV*)HeVAL(entry);
 337         if (isGV(glob)) {
 338             croak("_expand_glob called on stash slot with expanded glob");
 339         }
 340         else {
 341             char *varname_pv;
 342             STRLEN varname_len;
 343
 344             varname_pv = SvPV(varname, varname_len);
 345             gv_init(glob, namespace, varname_pv, varname_len, 1);
 346             SvREFCNT_inc(glob);
 347             if (!hv_store_ent(namespace, varname, (SV*)glob, 0)) {
 348                 croak("hv_store failed");
 349             }
 350         }
 351     }
 352     else {
 353         croak("_expand_glob called on nonexistent stash slot");
 354     }
 355 }
 356
 357 static SV *_get_symbol(SV *self, varspec_t *variable, int vivify)
 358 {
 359     HV *namespace;
 360     HE *entry;
 361     GV *glob;
 362
 363     namespace = _get_namespace(self);
 364     entry = hv_fetch_ent(namespace, variable->name, vivify, 0);
 365     if (!entry)
 366         return NULL;
 367
 368     glob = (GV*)(HeVAL(entry));
 369     if (!isGV(glob))
 370         _expand_glob(self, variable->name);
 371
 372     if (vivify) {
 373         switch (variable->type) {
 374         case VAR_SCALAR:
 375             if (!GvSVOK(glob))
 376                 GvSetSV(glob, newSV(0));
 377             break;
 378         case VAR_ARRAY:
 379             if (!GvAVOK(glob))
 380                 GvSetAV(glob, newAV());
 381             break;
 382         case VAR_HASH:
 383             if (!GvHVOK(glob))
 384                 GvSetHV(glob, newHV());
 385             break;
 386         case VAR_CODE:
 387             croak("Don't know how to vivify CODE variables");
 388         case VAR_IO:
 389             if (!GvIOOK(glob))
 390                 GvSetIO(glob, newIO());
 391             break;
 392         default:
 393             croak("Unknown type in vivication");
 394         }
 395     }
 396
 397     switch (variable->type) {
 398     case VAR_SCALAR:
 399         return GvSV(glob);
 400     case VAR_ARRAY:
 401         return (SV*)GvAV(glob);
 402     case VAR_HASH:
 403         return (SV*)GvHV(glob);
 404     case VAR_CODE:
 405         return (SV*)GvCV(glob);
 406     case VAR_IO:
 407         return (SV*)GvIO(glob);
 408     default:
 409         return NULL;
 410     }
 411 }
 412
 413 MODULE = Package::Stash::XS  PACKAGE = Package::Stash::XS
 414
 415 PROTOTYPES: DISABLE
 416
 417 SV*
 418 new(class, package)
 419     SV *class
 420     SV *package
 421   PREINIT:
 422     HV *instance;
 423   CODE:
 424     if (SvPOK(package)) {
 425         if (!_is_valid_module_name(package))
 426             croak("%s is not a module name", SvPV_nolen(package));
 427
 428         instance = newHV();
 429
 430         if (!hv_store(instance, "name", 4, SvREFCNT_inc_simple_NN(package), 0)) {
 431             SvREFCNT_dec(package);
 432             SvREFCNT_dec(instance);
 433             croak("Couldn't initialize the 'name' key, hv_store failed");
 434         }
 435     }
 436     else if (SvROK(package) && SvTYPE(SvRV(package)) == SVt_PVHV) {
 437         instance = newHV();
 438
 439         if (!hv_store(instance, "namespace", 9, SvREFCNT_inc_simple_NN(package), 0)) {
 440             SvREFCNT_dec(package);
 441             SvREFCNT_dec(instance);
 442             croak("Couldn't initialize the 'namespace' key, hv_store failed");
 443         }
 444     }
 445     else {
 446         croak("Package::Stash->new must be passed the name of the package to access");
 447     }
 448
 449     RETVAL = sv_bless(newRV_noinc((SV*)instance), gv_stashsv(class, 0));
 450   OUTPUT:
 451     RETVAL
 452
 453 SV*
 454 name(self)
 455     SV *self
 456   PREINIT:
 457     HE *slot;
 458   CODE:
 459     if (!sv_isobject(self))
 460         croak("Can't call name as a class method");
 461     if (slot = hv_fetch_ent((HV*)SvRV(self), name_key, 0, name_hash)) {
 462         RETVAL = SvREFCNT_inc_simple_NN(HeVAL(slot));
 463     }
 464     else {
 465         croak("Can't get the name of an anonymous package");
 466     }
 467   OUTPUT:
 468     RETVAL
 469
 470 SV*
 471 namespace(self)
 472     SV *self
 473   PREINIT:
 474     HE *slot;
 475     SV *package_name;
 476   CODE:
 477     if (!sv_isobject(self))
 478         croak("Can't call namespace as a class method");
 479 #if PERL_VERSION < 10
 480     package_name = _get_name(self);
 481     RETVAL = newRV_inc((SV*)gv_stashpv(SvPV_nolen(package_name), GV_ADD));
 482 #else
 483     slot = hv_fetch_ent((HV*)SvRV(self), namespace_key, 0, namespace_hash);
 484     if (slot) {
 485         RETVAL = SvREFCNT_inc_simple_NN(HeVAL(slot));
 486     }
 487     else {
 488         HV *namespace;
 489         SV *nsref;
 490
 491         package_name = _get_name(self);
 492         namespace = gv_stashpv(SvPV_nolen(package_name), GV_ADD);
 493         nsref = newRV_inc((SV*)namespace);
 494         sv_rvweaken(nsref);
 495         if (!hv_store((HV*)SvRV(self), "namespace", 9, nsref, 0)) {
 496             SvREFCNT_dec(nsref);
 497             SvREFCNT_dec(self);
 498             croak("Couldn't initialize the 'namespace' key, hv_store failed");
 499         }
 500         RETVAL = SvREFCNT_inc_simple_NN(nsref);
 501     }
 502 #endif
 503   OUTPUT:
 504     RETVAL
 505
 506 void
 507 add_symbol(self, variable, initial=NULL, ...)
 508     SV *self
 509     varspec_t variable
 510     SV *initial
 511   PREINIT:
 512     GV *glob;
 513     HV *namespace;
 514     HE *entry;
 515   CODE:
 516     if (initial && !_valid_for_type(initial, variable.type))
 517         croak("%s is not of type %s",
 518               SvPV_nolen(initial), vartype_to_string(variable.type));
 519
 520     if (items > 2 && (PL_perldb & 0x10) && variable.type == VAR_CODE) {
 521         int i;
 522         char *filename = NULL;
 523         I32 first_line_num = -1, last_line_num = -1;
 524         SV *dbval, *name;
 525         HV *dbsub;
 526
 527         if ((items - 3) % 2)
 528             croak("add_symbol: Odd number of elements in %%opts");
 529
 530         for (i = 3; i < items; i += 2) {
 531             char *key;
 532             key = SvPV_nolen(ST(i));
 533             if (strEQ(key, "filename")) {
 534                 if (!SvPOK(ST(i + 1)))
 535                     croak("add_symbol: filename must be a string");
 536                 filename = SvPV_nolen(ST(i + 1));
 537             }
 538             else if (strEQ(key, "first_line_num")) {
 539                 if (!SvIOK(ST(i + 1)))
 540                     croak("add_symbol: first_line_num must be an integer");
 541                 first_line_num = SvIV(ST(i + 1));
 542             }
 543             else if (strEQ(key, "last_line_num")) {
 544                 if (!SvIOK(ST(i + 1)))
 545                     croak("add_symbol: last_line_num must be an integer");
 546                 last_line_num = SvIV(ST(i + 1));
 547             }
 548         }
 549
 550         if (!filename || first_line_num == -1) {
 551             if (!filename)
 552                 filename = CopFILE(PL_curcop);
 553             if (first_line_num == -1)
 554                 first_line_num = PL_curcop->cop_line;
 555         }
 556
 557         if (last_line_num == -1)
 558             last_line_num = first_line_num;
 559
 560         name = newSVsv(_get_name(self));
 561         sv_catpvs(name, "::");
 562         sv_catsv(name, variable.name);
 563
 564         /* http://perldoc.perl.org/perldebguts.html#Debugger-Internals */
 565         dbsub = get_hv("DB::sub", 1);
 566         dbval = newSVpvf("%s:%d-%d", filename, first_line_num, last_line_num);
 567         if (!hv_store_ent(dbsub, name, dbval, 0)) {
 568             warn("Failed to update $DB::sub for subroutine %s",
 569                  SvPV_nolen(name));
 570             SvREFCNT_dec(dbval);
 571         }
 572
 573         SvREFCNT_dec(name);
 574     }
 575
 576     /* GV_ADDMULTI rather than GV_ADD because otherwise you get 'used only
 577      * once' warnings in some situations... i can't reproduce this, but CMOP
 578      * triggers it */
 579     namespace = _get_namespace(self);
 580     entry = hv_fetch_ent(namespace, variable.name, 0, 0);
 581     if (entry) {
 582         glob = (GV*)HeVAL(entry);
 583     }
 584     else {
 585         char *varname_pv;
 586         STRLEN varname_len;
 587         glob = (GV*)newSV(0);
 588         varname_pv = SvPV(variable.name, varname_len);
 589         gv_init(glob, namespace, varname_pv, varname_len, 1);
 590         if (!hv_store_ent(namespace, variable.name, (SV*)glob, 0)) {
 591             croak("hv_store failed");
 592         }
 593     }
 594
 595     if (initial) {
 596         SV *val;
 597
 598         if (SvROK(initial)) {
 599             val = SvRV(initial);
 600             SvREFCNT_inc_simple_void_NN(val);
 601         }
 602         else {
 603             val = newSVsv(initial);
 604         }
 605
 606         switch (variable.type) {
 607         case VAR_SCALAR:
 608             GvSetSV(glob, val);
 609             break;
 610         case VAR_ARRAY:
 611             GvSetAV(glob, val);
 612             break;
 613         case VAR_HASH:
 614             GvSetHV(glob, val);
 615             break;
 616         case VAR_CODE:
 617             GvSetCV(glob, val);
 618             break;
 619         case VAR_IO:
 620             GvSetIO(glob, val);
 621             break;
 622         }
 623     }
 624
 625 void
 626 remove_glob(self, name)
 627     SV *self
 628     SV *name
 629   CODE:
 630     hv_delete_ent(_get_namespace(self), name, G_DISCARD, 0);
 631
 632 int
 633 has_symbol(self, variable)
 634     SV *self
 635     varspec_t variable
 636   PREINIT:
 637     HV *namespace;
 638     HE *entry;
 639     SV *val;
 640   CODE:
 641     namespace = _get_namespace(self);
 642     entry = hv_fetch_ent(namespace, variable.name, 0, 0);
 643     if (!entry)
 644         XSRETURN_UNDEF;
 645
 646     val = HeVAL(entry);
 647     if (isGV(val)) {
 648         GV *glob = (GV*)val;
 649         switch (variable.type) {
 650         case VAR_SCALAR:
 651             RETVAL = GvSVOK(glob) ? 1 : 0;
 652             break;
 653         case VAR_ARRAY:
 654             RETVAL = GvAVOK(glob) ? 1 : 0;
 655             break;
 656         case VAR_HASH:
 657             RETVAL = GvHVOK(glob) ? 1 : 0;
 658             break;
 659         case VAR_CODE:
 660             RETVAL = GvCVOK(glob) ? 1 : 0;
 661             break;
 662         case VAR_IO:
 663             RETVAL = GvIOOK(glob) ? 1 : 0;
 664             break;
 665         }
 666     }
 667     else {
 668         RETVAL = (variable.type == VAR_CODE);
 669     }
 670   OUTPUT:
 671     RETVAL
 672
 673 SV*
 674 get_symbol(self, variable)
 675     SV *self
 676     varspec_t variable
 677   PREINIT:
 678     SV *val;
 679   CODE:
 680     val = _get_symbol(self, &variable, 0);
 681     if (!val)
 682         XSRETURN_UNDEF;
 683     RETVAL = newRV_inc(val);
 684   OUTPUT:
 685     RETVAL
 686
 687 SV*
 688 get_or_add_symbol(self, variable)
 689     SV *self
 690     varspec_t variable
 691   PREINIT:
 692     SV *val;
 693   CODE:
 694     val = _get_symbol(self, &variable, 1);
 695     if (!val)
 696         XSRETURN_UNDEF;
 697     RETVAL = newRV_inc(val);
 698   OUTPUT:
 699     RETVAL
 700
 701 void
 702 remove_symbol(self, variable)
 703     SV *self
 704     varspec_t variable
 705   PREINIT:
 706     HV *namespace;
 707     HE *entry;
 708     SV *val;
 709   CODE:
 710     namespace = _get_namespace(self);
 711     entry = hv_fetch_ent(namespace, variable.name, 0, 0);
 712     if (!entry)
 713         XSRETURN_EMPTY;
 714
 715     val = HeVAL(entry);
 716     if (isGV(val)) {
 717         GV *glob = (GV*)val;
 718         switch (variable.type) {
 719         case VAR_SCALAR:
 720             GvSetSV(glob, NULL);
 721             break;
 722         case VAR_ARRAY:
 723             GvSetAV(glob, NULL);
 724             break;
 725         case VAR_HASH:
 726             GvSetHV(glob, NULL);
 727             break;
 728         case VAR_CODE:
 729             GvSetCV(glob, NULL);
 730             break;
 731         case VAR_IO:
 732             GvSetIO(glob, NULL);
 733             break;
 734         }
 735     }
 736     else {
 737         if (variable.type == VAR_CODE) {
 738             hv_delete_ent(namespace, variable.name, G_DISCARD, 0);
 739         }
 740     }
 741
 742 void
 743 list_all_symbols(self, vartype=VAR_NONE)
 744     SV *self
 745     vartype_t vartype
 746   PPCODE:
 747     if (vartype == VAR_NONE) {
 748         HV *namespace;
 749         HE *entry;
 750         int keys;
 751
 752         namespace = _get_namespace(self);
 753         keys = hv_iterinit(namespace);
 754         EXTEND(SP, keys);
 755         while ((entry = hv_iternext(namespace))) {
 756             mPUSHs(newSVhe(entry));
 757         }
 758     }
 759     else {
 760         HV *namespace;
 761         SV *val;
 762         char *key;
 763         I32 len;
 764
 765         namespace = _get_namespace(self);
 766         hv_iterinit(namespace);
 767         while ((val = hv_iternextsv(namespace, &key, &len))) {
 768             GV *gv = (GV*)val;
 769             if (isGV(gv)) {
 770                 switch (vartype) {
 771                 case VAR_SCALAR:
 772                     if (GvSVOK(val))
 773                         mXPUSHp(key, len);
 774                     break;
 775                 case VAR_ARRAY:
 776                     if (GvAVOK(val))
 777                         mXPUSHp(key, len);
 778                     break;
 779                 case VAR_HASH:
 780                     if (GvHVOK(val))
 781                         mXPUSHp(key, len);
 782                     break;
 783                 case VAR_CODE:
 784                     if (GvCVOK(val))
 785                         mXPUSHp(key, len);
 786                     break;
 787                 case VAR_IO:
 788                     if (GvIOOK(val))
 789                         mXPUSHp(key, len);
 790                     break;
 791                 }
 792             }
 793             else if (vartype == VAR_CODE) {
 794                 mXPUSHp(key, len);
 795             }
 796         }
 797     }
 798
 799 void
 800 get_all_symbols(self, vartype=VAR_NONE)
 801     SV *self
 802     vartype_t vartype
 803   PREINIT:
 804     HV *namespace, *ret;
 805     SV *val;
 806     char *key;
 807     I32 len;
 808   PPCODE:
 809     namespace = _get_namespace(self);
 810     ret = newHV();
 811
 812     hv_iterinit(namespace);
 813     while ((val = hv_iternextsv(namespace, &key, &len))) {
 814         GV *gv = (GV*)val;
 815
 816         if (!isGV(gv)) {
 817             SV *keysv = newSVpvn(key, len);
 818             _expand_glob(self, keysv);
 819             SvREFCNT_dec(keysv);
 820         }
 821
 822         switch (vartype) {
 823         case VAR_SCALAR:
 824             if (GvSVOK(val))
 825                 hv_store(ret, key, len, newRV_inc(GvSV(gv)), 0);
 826             break;
 827         case VAR_ARRAY:
 828             if (GvAVOK(val))
 829                 hv_store(ret, key, len, newRV_inc((SV*)GvAV(gv)), 0);
 830             break;
 831         case VAR_HASH:
 832             if (GvHVOK(val))
 833                 hv_store(ret, key, len, newRV_inc((SV*)GvHV(gv)), 0);
 834             break;
 835         case VAR_CODE:
 836             if (GvCVOK(val))
 837                 hv_store(ret, key, len, newRV_inc((SV*)GvCV(gv)), 0);
 838             break;
 839         case VAR_IO:
 840             if (GvIOOK(val))
 841                 hv_store(ret, key, len, newRV_inc((SV*)GvIO(gv)), 0);
 842             break;
 843         case VAR_NONE:
 844             hv_store(ret, key, len, SvREFCNT_inc_simple_NN(val), 0);
 845             break;
 846         }
 847     }
 848
 849     mPUSHs(newRV_noinc((SV*)ret));
 850
 851 BOOT:
 852     {
 853         const char *vmre = "\\A[0-9A-Z_a-z]+(?:::[0-9A-Z_a-z]+)*\\z";
 854 #if (PERL_VERSION < 9) || ((PERL_VERSION == 9) && (PERL_SUBVERSION < 5))
 855         PMOP fakepmop;
 856
 857         fakepmop.op_pmflags = 0;
 858         valid_module_regex = pregcomp(vmre, vmre + strlen(vmre), &fakepmop);
 859 #else
 860         SV *re;
 861
 862         re = newSVpv(vmre, 0);
 863         valid_module_regex = pregcomp(re, 0);
 864 #endif
 865
 866         name_key = newSVpvs("name");
 867         PERL_HASH(name_hash, "name", 4);
 868
 869         namespace_key = newSVpvs("namespace");
 870         PERL_HASH(namespace_hash, "namespace", 9);
 871
 872         type_key = newSVpvs("type");
 873         PERL_HASH(type_hash, "type", 4);
 874     }