don't leak old glob values
[gitmo/Package-Stash-XS.git] / Stash.xs
CommitLineData
59017825 1#include "EXTERN.h"
2#include "perl.h"
3#include "XSUB.h"
4
a382a84b 5typedef enum {
6 VAR_NONE = 0,
7 VAR_SCALAR,
8 VAR_ARRAY,
9 VAR_HASH,
10 VAR_CODE,
11 VAR_IO,
12 VAR_GLOB, /* TODO: unimplemented */
13 VAR_FORMAT /* TODO: unimplemented */
14} vartype_t;
15
16typedef struct {
17 vartype_t type;
18 char sigil;
19 char *name;
20} varspec_t;
21
76f7c306 22const char *vartype_to_string(vartype_t type)
23{
24 switch (type) {
25 case VAR_SCALAR:
26 return "SCALAR";
27 case VAR_ARRAY:
28 return "ARRAY";
29 case VAR_HASH:
30 return "HASH";
31 case VAR_CODE:
32 return "CODE";
33 case VAR_IO:
34 return "IO";
35 default:
36 return "unknown";
37 }
38}
39
40I32 vartype_to_svtype(vartype_t type)
41{
42 switch (type) {
43 case VAR_SCALAR:
44 return SVt_PV; /* or whatever */
45 case VAR_ARRAY:
46 return SVt_PVAV;
47 case VAR_HASH:
48 return SVt_PVHV;
49 case VAR_CODE:
50 return SVt_PVCV;
51 case VAR_IO:
52 return SVt_PVIO;
53 default:
54 return SVt_NULL;
55 }
56}
57
a382a84b 58vartype_t string_to_vartype(char *vartype)
59{
60 if (strEQ(vartype, "SCALAR")) {
61 return VAR_SCALAR;
62 }
63 else if (strEQ(vartype, "ARRAY")) {
64 return VAR_ARRAY;
65 }
66 else if (strEQ(vartype, "HASH")) {
67 return VAR_HASH;
68 }
69 else if (strEQ(vartype, "CODE")) {
70 return VAR_CODE;
71 }
72 else if (strEQ(vartype, "IO")) {
73 return VAR_IO;
74 }
75 else {
76 croak("Type must be one of 'SCALAR', 'ARRAY', 'HASH', 'CODE', or 'IO'");
77 }
78}
79
80void _deconstruct_variable_name(char *variable, varspec_t *varspec)
81{
82 if (!variable || !variable[0])
83 croak("You must pass a variable name");
84
85 varspec->type = VAR_NONE;
86
87 switch (variable[0]) {
88 case '$':
89 varspec->type = VAR_SCALAR;
90 break;
91 case '@':
92 varspec->type = VAR_ARRAY;
93 break;
94 case '%':
95 varspec->type = VAR_HASH;
96 break;
97 case '&':
98 varspec->type = VAR_CODE;
99 break;
100 }
101
102 if (varspec->type != VAR_NONE) {
103 varspec->sigil = variable[0];
104 varspec->name = &variable[1];
105 }
106 else {
107 varspec->type = VAR_IO;
108 varspec->sigil = '\0';
109 varspec->name = variable;
110 }
111}
112
113void _deconstruct_variable_hash(HV *variable, varspec_t *varspec)
114{
115 SV **val;
a382a84b 116
117 val = hv_fetch(variable, "name", 4, 0);
118 if (!val)
119 croak("The 'name' key is required in variable specs");
120
121 varspec->name = savesvpv(*val);
301f570b 122 SAVEFREEPV(varspec->name);
a382a84b 123
124 val = hv_fetch(variable, "sigil", 5, 0);
125 if (!val)
126 croak("The 'sigil' key is required in variable specs");
127
128 varspec->sigil = (SvPV_nolen(*val))[0];
129
130 val = hv_fetch(variable, "type", 4, 0);
131 if (!val)
132 croak("The 'type' key is required in variable specs");
133
134 varspec->type = string_to_vartype(SvPV_nolen(*val));
135}
136
137int _valid_for_type(SV *value, vartype_t type)
138{
139 svtype sv_type = SvROK(value) ? SvTYPE(SvRV(value)) : SVt_NULL;
140
141 switch (type) {
142 case VAR_SCALAR:
143 return sv_type == SVt_NULL ||
144 sv_type == SVt_IV ||
145 sv_type == SVt_NV ||
146 sv_type == SVt_PV ||
147 sv_type == SVt_RV;
148 case VAR_ARRAY:
149 return sv_type == SVt_PVAV;
150 case VAR_HASH:
151 return sv_type == SVt_PVHV;
152 case VAR_CODE:
153 return sv_type == SVt_PVCV;
154 case VAR_IO:
76f7c306 155 return sv_type == SVt_PVIO;
a382a84b 156 default:
157 return 0;
158 }
159}
160
3fd56b4d 161HV *_get_namespace(SV *self)
162{
163 dSP;
164 SV *ret;
165
166 PUSHMARK(SP);
167 XPUSHs(self);
168 PUTBACK;
169
612dcf3b 170 call_method("namespace", G_SCALAR);
3fd56b4d 171
172 SPAGAIN;
173 ret = POPs;
174 PUTBACK;
175
176 return (HV*)SvRV(ret);
177}
178
76f7c306 179SV *_get_name(SV *self)
180{
181 dSP;
182 SV *ret;
183
184 PUSHMARK(SP);
185 XPUSHs(self);
186 PUTBACK;
187
612dcf3b 188 call_method("name", G_SCALAR);
76f7c306 189
190 SPAGAIN;
191 ret = POPs;
192 PUTBACK;
193
194 return ret;
195}
196
4871b1a0 197SV *_get_package_symbol(SV *self, varspec_t *variable, int vivify)
198{
199 HV *namespace;
200 SV **entry;
201 GV *glob;
202
203 namespace = _get_namespace(self);
204 entry = hv_fetch(namespace, variable->name, strlen(variable->name), vivify);
205 if (!entry)
206 return NULL;
207
208 glob = (GV*)(*entry);
209 if (!isGV(glob)) {
210 SV *namesv;
4871b1a0 211
212 namesv = newSVsv(_get_name(self));
213 sv_catpvs(namesv, "::");
214 sv_catpv(namesv, variable->name);
215
b55c97c2 216 /* can't use gv_init here, because it screws up @ISA in a way that I
217 * can't reproduce, but that CMOP triggers */
218 gv_fetchsv(namesv, GV_ADD, vartype_to_svtype(variable->type));
4871b1a0 219 }
220
221 if (vivify) {
222 switch (variable->type) {
223 case VAR_SCALAR:
224 if (!GvSV(glob))
225 GvSV(glob) = newSV(0);
226 break;
227 case VAR_ARRAY:
228 if (!GvAV(glob))
229 GvAV(glob) = newAV();
230 break;
231 case VAR_HASH:
232 if (!GvHV(glob))
233 GvHV(glob) = newHV();
234 break;
235 case VAR_CODE:
236 croak("Don't know how to vivify CODE variables");
237 case VAR_IO:
238 if (!GvIO(glob))
239 GvIOp(glob) = newIO();
240 break;
241 default:
242 croak("Unknown type in vivication");
243 }
244 }
245
246 switch (variable->type) {
247 case VAR_SCALAR:
248 return GvSV(glob);
249 case VAR_ARRAY:
250 return (SV*)GvAV(glob);
251 case VAR_HASH:
252 return (SV*)GvHV(glob);
253 case VAR_CODE:
254 return (SV*)GvCV(glob);
255 case VAR_IO:
256 return (SV*)GvIO(glob);
257 default:
258 return NULL;
259 }
260}
261
59017825 262MODULE = Package::Stash PACKAGE = Package::Stash
263
be2a7e99 264PROTOTYPES: DISABLE
265
59017825 266SV*
267new(class, package_name)
268 char *class
269 SV *package_name
3496f1e8 270 PREINIT:
59017825 271 HV *instance;
272 HV *namespace;
273 CODE:
274 if (!SvPOK(package_name))
275 croak("The constructor argument must be the name of a package");
276
6c3e69c6 277 instance = newHV();
59017825 278
ce38493c 279 hv_store(instance, "name", 4, SvREFCNT_inc(package_name), 0);
59017825 280 namespace = gv_stashpv(SvPV_nolen(package_name), GV_ADD);
ce38493c 281 hv_store(instance, "namespace", 9, newRV_inc((SV*)namespace), 0);
59017825 282
6c3e69c6 283 RETVAL = sv_bless(newRV_noinc((SV*)instance), gv_stashpv(class, 0));
59017825 284 OUTPUT:
285 RETVAL
194acf47 286
287SV*
288name(self)
289 SV *self
3496f1e8 290 PREINIT:
194acf47 291 SV **slot;
292 CODE:
293 if (!sv_isobject(self))
294 croak("Can't call name as a class method");
295 slot = hv_fetch((HV*)SvRV(self), "name", 4, 0);
296 RETVAL = slot ? SvREFCNT_inc(*slot) : &PL_sv_undef;
297 OUTPUT:
298 RETVAL
299
300SV*
301namespace(self)
302 SV *self
3496f1e8 303 PREINIT:
194acf47 304 SV **slot;
305 CODE:
306 if (!sv_isobject(self))
307 croak("Can't call namespace as a class method");
308 slot = hv_fetch((HV*)SvRV(self), "namespace", 9, 0);
309 RETVAL = slot ? SvREFCNT_inc(*slot) : &PL_sv_undef;
310 OUTPUT:
311 RETVAL
3fd56b4d 312
313void
76f7c306 314add_package_symbol(self, variable, initial=NULL, ...)
315 SV *self
316 varspec_t variable
317 SV *initial
318 PREINIT:
319 SV *name;
320 GV *glob;
321 CODE:
322 if (initial && !_valid_for_type(initial, variable.type))
323 croak("%s is not of type %s",
324 SvPV_nolen(initial), vartype_to_string(variable.type));
325
326 name = newSVsv(_get_name(self));
327 sv_catpvs(name, "::");
328 sv_catpv(name, variable.name);
329
330 /* XXX: come back to this when i feel like reimplementing caller() */
331/*
332 my $filename = $opts{filename};
333 my $first_line_num = $opts{first_line_num};
334
335 (undef, $filename, $first_line_num) = caller
336 if not defined $filename;
337
338 my $last_line_num = $opts{last_line_num} || ($first_line_num ||= 0);
339
340 # http://perldoc.perl.org/perldebguts.html#Debugger-Internals
341 $DB::sub{$pkg . '::' . $name} = "$filename:$first_line_num-$last_line_num";
342*/
343/*
344 if (items > 2 && (PL_perldb & 0x10) && variable.type == VAR_CODE) {
345 int i;
346 char *filename = NULL, *name;
347 I32 first_line_num, last_line_num;
348
349 if ((items - 3) % 2)
350 croak("add_package_symbol: Odd number of elements in %%opts");
351
352 for (i = 3; i < items; i += 2) {
353 char *key;
354 key = SvPV_nolen(ST(i));
355 if (strEQ(key, "filename")) {
356 if (!SvPOK(ST(i + 1)))
357 croak("add_package_symbol: filename must be a string");
358 filename = SvPV_nolen(ST(i + 1));
359 }
360 else if (strEQ(key, "first_line_num")) {
361 if (!SvIOK(ST(i + 1)))
362 croak("add_package_symbol: first_line_num must be an integer");
363 first_line_num = SvIV(ST(i + 1));
364 }
365 else if (strEQ(key, "last_line_num")) {
366 if (!SvIOK(ST(i + 1)))
367 croak("add_package_symbol: last_line_num must be an integer");
368 last_line_num = SvIV(ST(i + 1));
369 }
370 }
371
372 if (!filename) {
373 }
374 }
375*/
376
fedea7f9 377 glob = gv_fetchsv(name, GV_ADDMULTI, vartype_to_svtype(variable.type));
76f7c306 378
379 if (initial) {
380 SV *val;
381
382 if (SvROK(initial)) {
383 val = SvRV(initial);
384 SvREFCNT_inc(val);
385 }
386 else {
387 val = newSVsv(initial);
388 }
389
390 switch (variable.type) {
391 case VAR_SCALAR:
12134e83 392 if (GvSV(glob))
393 SvREFCNT_dec(GvSV(glob));
76f7c306 394 GvSV(glob) = val;
395 break;
396 case VAR_ARRAY:
12134e83 397 if (GvAV(glob))
398 SvREFCNT_dec((SV*)GvAV(glob));
76f7c306 399 GvAV(glob) = (AV*)val;
400 break;
401 case VAR_HASH:
12134e83 402 if (GvHV(glob))
403 SvREFCNT_dec((SV*)GvHV(glob));
76f7c306 404 GvHV(glob) = (HV*)val;
405 break;
406 case VAR_CODE:
12134e83 407 if (GvCV(glob))
408 SvREFCNT_dec((SV*)GvCV(glob));
76f7c306 409 GvCV(glob) = (CV*)val;
410 break;
411 case VAR_IO:
12134e83 412 if (GvIO(glob))
413 SvREFCNT_dec((SV*)GvIO(glob));
76f7c306 414 GvIOp(glob) = (IO*)val;
415 break;
416 }
417 }
418
5bb2b07b 419 SvREFCNT_dec(name);
420
76f7c306 421void
3fd56b4d 422remove_package_glob(self, name)
423 SV *self
424 char *name
3fd56b4d 425 CODE:
426 hv_delete(_get_namespace(self), name, strlen(name), G_DISCARD);
9b608466 427
34805376 428int
429has_package_symbol(self, variable)
430 SV *self
431 varspec_t variable
432 PREINIT:
433 HV *namespace;
434 SV **entry;
435 CODE:
436 namespace = _get_namespace(self);
437 entry = hv_fetch(namespace, variable.name, strlen(variable.name), 0);
438 if (!entry)
439 XSRETURN_UNDEF;
440
441 if (isGV(*entry)) {
442 GV *glob = (GV*)(*entry);
443 switch (variable.type) {
444 case VAR_SCALAR:
445 RETVAL = GvSV(glob) ? 1 : 0;
446 break;
447 case VAR_ARRAY:
448 RETVAL = GvAV(glob) ? 1 : 0;
449 break;
450 case VAR_HASH:
451 RETVAL = GvHV(glob) ? 1 : 0;
452 break;
453 case VAR_CODE:
454 RETVAL = GvCV(glob) ? 1 : 0;
455 break;
456 case VAR_IO:
457 RETVAL = GvIO(glob) ? 1 : 0;
458 break;
459 }
460 }
461 else {
462 RETVAL = (variable.type == VAR_CODE);
463 }
464 OUTPUT:
465 RETVAL
466
e3ad44fd 467SV*
4871b1a0 468get_package_symbol(self, variable)
e3ad44fd 469 SV *self
470 varspec_t variable
471 PREINIT:
fca4ed0c 472 SV *val;
e3ad44fd 473 CODE:
4871b1a0 474 val = _get_package_symbol(self, &variable, 0);
475 if (!val)
e3ad44fd 476 XSRETURN_UNDEF;
93c36417 477 RETVAL = newRV_inc(val);
4871b1a0 478 OUTPUT:
479 RETVAL
e3ad44fd 480
4871b1a0 481SV*
482get_or_add_package_symbol(self, variable)
483 SV *self
484 varspec_t variable
485 PREINIT:
486 SV *val;
487 CODE:
488 val = _get_package_symbol(self, &variable, 1);
fca4ed0c 489 if (!val)
490 XSRETURN_UNDEF;
93c36417 491 RETVAL = newRV_inc(val);
e3ad44fd 492 OUTPUT:
493 RETVAL
494
9b608466 495void
215f49f8 496remove_package_symbol(self, variable)
497 SV *self
498 varspec_t variable
499 PREINIT:
500 HV *namespace;
501 SV **entry;
502 CODE:
503 namespace = _get_namespace(self);
504 entry = hv_fetch(namespace, variable.name, strlen(variable.name), 0);
505 if (!entry)
506 XSRETURN_EMPTY;
507
508 if (isGV(*entry)) {
509 GV *glob = (GV*)(*entry);
510 switch (variable.type) {
511 case VAR_SCALAR:
4e3b237e 512 GvSV(glob) = (SV *)NULL;
215f49f8 513 break;
514 case VAR_ARRAY:
4e3b237e 515 GvAV(glob) = (AV *)NULL;
215f49f8 516 break;
517 case VAR_HASH:
4e3b237e 518 GvHV(glob) = (HV *)NULL;
215f49f8 519 break;
520 case VAR_CODE:
4e3b237e 521 GvCV(glob) = (CV *)NULL;
215f49f8 522 break;
523 case VAR_IO:
4e3b237e 524 GvIOp(glob) = (IO *)NULL;
215f49f8 525 break;
526 }
527 }
528 else {
529 if (variable.type == VAR_CODE) {
530 hv_delete(namespace, variable.name, strlen(variable.name), G_DISCARD);
531 }
532 }
533
534void
9b608466 535list_all_package_symbols(self, vartype=VAR_NONE)
536 SV *self
537 vartype_t vartype
538 PPCODE:
539 if (vartype == VAR_NONE) {
540 HV *namespace;
541 HE *entry;
542 int keys;
543
544 namespace = _get_namespace(self);
545 keys = hv_iterinit(namespace);
546 EXTEND(SP, keys);
ebcc8ba8 547 while ((entry = hv_iternext(namespace))) {
9b608466 548 mPUSHs(newSVhek(HeKEY_hek(entry)));
549 }
550 }
551 else {
552 HV *namespace;
9b608466 553 SV *val;
554 char *key;
555 int len;
556
557 namespace = _get_namespace(self);
558 hv_iterinit(namespace);
ebcc8ba8 559 while ((val = hv_iternextsv(namespace, &key, &len))) {
9b608466 560 GV *gv = (GV*)val;
561 if (isGV(gv)) {
562 switch (vartype) {
563 case VAR_SCALAR:
564 if (GvSV(val))
565 mXPUSHp(key, len);
566 break;
567 case VAR_ARRAY:
568 if (GvAV(val))
569 mXPUSHp(key, len);
570 break;
571 case VAR_HASH:
572 if (GvHV(val))
573 mXPUSHp(key, len);
574 break;
575 case VAR_CODE:
576 if (GvCVu(val))
577 mXPUSHp(key, len);
578 break;
579 case VAR_IO:
580 if (GvIO(val))
581 mXPUSHp(key, len);
582 break;
583 }
584 }
585 else if (vartype == VAR_CODE) {
586 mXPUSHp(key, len);
587 }
588 }
589 }