Add an xs implementation of Class::MOP::is_class_loaded.
[gitmo/Class-MOP.git] / MOP.xs
CommitLineData
e7b69038 1/* There's a lot of cases of doubled parens in here like this:
2
3 while ( (he = ...) ) {
4
5This shuts up warnings from gcc -Wall
6*/
7
e0e4674a 8#include "EXTERN.h"
9#include "perl.h"
10#include "XSUB.h"
15273f3c 11
08f4716f 12#define NEED_newRV_noinc
599791aa 13#define NEED_sv_2pv_flags
15273f3c 14#define NEED_sv_2pv_nolen
b0e94057 15#include "ppport.h"
e0e4674a 16
cc856b56 17SV *key_name;
18U32 hash_name;
19
20SV *key_package;
21U32 hash_package;
22
da88f307 23SV *key_package_name;
24U32 hash_package_name;
25
cc856b56 26SV *key_body;
27U32 hash_body;
28
c686d0ed 29SV *key_package_cache_flag;
30U32 hash_package_cache_flag;
31
32SV *key_methods;
33U32 hash_methods;
34
00c93f7a 35SV *method_metaclass;
36SV *associated_metaclass;
37SV *wrap;
c94afdc4 38
39
40#define check_package_cache_flag(stash) mop_check_package_cache_flag(aTHX_ stash)
1d4cf2a9 41#if PERL_VERSION >= 10
c94afdc4 42
43static UV
44mop_check_package_cache_flag(pTHX_ HV* stash) {
45 assert(SvTYPE(stash) == SVt_PVHV);
46
1d4cf2a9 47 /* here we're trying to implement a c version of mro::get_pkg_gen($stash),
48 * however the perl core doesn't make it easy for us. It doesn't provide an
49 * api that just does what we want.
50 *
51 * However, we know that the information we want is, inside the core,
52 * available using HvMROMETA(stash)->pkg_gen. Unfortunately, although the
53 * HvMROMETA macro is public, it is implemented using Perl_mro_meta_init,
54 * which is not public and only available inside the core, as the mro
55 * interface as well as the structure returned by mro_meta_init isn't
56 * considered to be stable yet.
57 *
58 * Perl_mro_meta_init isn't declared static, so we could just define it
59 * ourselfs if perls headers don't do that for us, except that won't work
60 * on platforms where symbols need to be explicitly exported when linking
61 * shared libraries.
62 *
63 * So our, hopefully temporary, solution is to be even more evil and
64 * basically reimplement HvMROMETA in a very fragile way that'll blow up
65 * when the relevant parts of the mro implementation in core change.
66 *
67 * :-(
68 *
69 */
70
71 return HvAUX(stash)->xhv_mro_meta
72 ? HvAUX(stash)->xhv_mro_meta->pkg_gen
73 : 0;
c94afdc4 74}
75
76#else /* pre 5.10.0 */
77
78static UV
00c93f7a 79mop_check_package_cache_flag(pTHX_ HV *stash) {
c94afdc4 80 PERL_UNUSED_ARG(stash);
81 assert(SvTYPE(stash) == SVt_PVHV);
82
83 return PL_sub_generation;
84}
85#endif
86
87#define call0(s, m) mop_call0(aTHX_ s, m)
00c93f7a 88static SV *
89mop_call0(pTHX_ SV *const self, SV *const method) {
c94afdc4 90 dSP;
00c93f7a 91 SV *ret;
c94afdc4 92
93 PUSHMARK(SP);
94 XPUSHs(self);
95 PUTBACK;
96
97 call_sv(method, G_SCALAR | G_METHOD);
98
99 SPAGAIN;
100 ret = POPs;
101 PUTBACK;
102
103 return ret;
104}
105
ee82a43e 106static int
0e0b5b8d 107get_code_info (SV *coderef, char **pkg, char **name)
108{
109 if (!SvOK(coderef) || !SvROK(coderef) || SvTYPE(SvRV(coderef)) != SVt_PVCV) {
110 return 0;
111 }
6bd19edb 112
0e0b5b8d 113 coderef = SvRV(coderef);
114 /* I think this only gets triggered with a mangled coderef, but if
115 we hit it without the guard, we segfault. The slightly odd return
116 value strikes me as an improvement (mst)
117 */
118#ifdef isGV_with_GP
119 if ( isGV_with_GP(CvGV(coderef)) ) {
cb0ec494 120#endif
0e0b5b8d 121 *pkg = HvNAME( GvSTASH(CvGV(coderef)) );
122 *name = GvNAME( CvGV(coderef) );
123#ifdef isGV_with_GP
124 } else {
125 *pkg = "__UNKNOWN__";
126 *name = "__ANON__";
c94afdc4 127 }
0e0b5b8d 128#endif
129
130 return 1;
c94afdc4 131}
132
b49ee39f 133typedef enum {
134 TYPE_FILTER_NONE,
135 TYPE_FILTER_CODE,
136 TYPE_FILTER_ARRAY,
137 TYPE_FILTER_IO,
138 TYPE_FILTER_HASH,
139 TYPE_FILTER_SCALAR,
140} type_filter_t;
141
142static HV *
143get_all_package_symbols(HV *stash, type_filter_t filter)
144{
145 HE *he;
146 HV *ret = newHV();
147
148 (void)hv_iterinit(stash);
149
150 if (filter == TYPE_FILTER_NONE) {
151 while ( (he = hv_iternext(stash)) ) {
152 STRLEN keylen;
153 char *key = HePV(he, keylen);
7291ecef 154 if (!hv_store(ret, key, keylen, SvREFCNT_inc(HeVAL(he)), 0)) {
155 croak("failed to store glob ref");
156 }
b49ee39f 157 }
158
159 return ret;
160 }
161
162 while ( (he = hv_iternext(stash)) ) {
163 SV *const gv = HeVAL(he);
164 SV *sv = NULL;
165 char *key;
166 STRLEN keylen;
167 char *package;
168 SV *fq;
169
170 switch( SvTYPE(gv) ) {
171#ifndef SVt_RV
172 case SVt_RV:
173#endif
174 case SVt_PV:
175 case SVt_IV:
176 /* expand the gv into a real typeglob if it
177 * contains stub functions and we were asked to
178 * return CODE symbols */
179 if (filter == TYPE_FILTER_CODE) {
180 if (SvROK(gv)) {
181 /* we don't really care about the length,
182 but that's the API */
183 key = HePV(he, keylen);
184 package = HvNAME(stash);
185 fq = newSVpvf("%s::%s", package, key);
186 sv = (SV *)get_cv(SvPV_nolen(fq), 0);
187 break;
188 }
189
190 key = HePV(he, keylen);
191 gv_init((GV *)gv, stash, key, keylen, GV_ADDMULTI);
192 }
193 /* fall through */
194 case SVt_PVGV:
195 switch (filter) {
196 case TYPE_FILTER_CODE: sv = (SV *)GvCVu(gv); break;
197 case TYPE_FILTER_ARRAY: sv = (SV *)GvAV(gv); break;
198 case TYPE_FILTER_IO: sv = (SV *)GvIO(gv); break;
199 case TYPE_FILTER_HASH: sv = (SV *)GvHV(gv); break;
200 case TYPE_FILTER_SCALAR: sv = (SV *)GvSV(gv); break;
201 default:
202 croak("Unknown type");
203 }
204 break;
205 default:
206 continue;
207 }
208
209 if (sv) {
210 char *key = HePV(he, keylen);
7291ecef 211 if (!hv_store(ret, key, keylen, newRV_inc(sv), 0)) {
212 croak("failed to store symbol ref");
213 }
b49ee39f 214 }
215 }
216
217 return ret;
218}
c94afdc4 219
0e0b5b8d 220
221static void
00c93f7a 222mop_update_method_map(pTHX_ SV *const self, SV *const class_name, HV *const stash, HV *const map) {
223 const char *const class_name_pv = HvNAME(stash); /* must be HvNAME(stash), not SvPV_nolen_const(class_name) */
224 SV *method_metaclass_name;
225 char *method_name;
0e0b5b8d 226 I32 method_name_len;
00c93f7a 227 SV *coderef;
228 HV *symbols;
0e0b5b8d 229 dSP;
230
231 symbols = get_all_package_symbols(stash, TYPE_FILTER_CODE);
232
233 (void)hv_iterinit(symbols);
234 while ( (coderef = hv_iternextsv(symbols, &method_name, &method_name_len)) ) {
00c93f7a 235 CV *cv = (CV *)SvRV(coderef);
236 char *cvpkg_name;
237 char *cv_name;
238 SV *method_slot;
239 SV *method_object;
0e0b5b8d 240
241 if (!get_code_info(coderef, &cvpkg_name, &cv_name)) {
242 continue;
243 }
244
245 /* this checks to see that the subroutine is actually from our package */
246 if ( !(strEQ(cvpkg_name, "constant") && strEQ(cv_name, "__ANON__")) ) {
247 if ( strNE(cvpkg_name, class_name_pv) ) {
248 continue;
249 }
250 }
251
252 method_slot = *hv_fetch(map, method_name, method_name_len, TRUE);
253 if ( SvOK(method_slot) ) {
6d915932 254 SV *const body = call0(method_slot, key_body); /* $method_object->body() */
00c93f7a 255 if ( SvROK(body) && ((CV *) SvRV(body)) == cv ) {
0e0b5b8d 256 continue;
257 }
258 }
259
260 method_metaclass_name = call0(self, method_metaclass); /* $self->method_metaclass() */
261
262 /*
263 $method_object = $method_metaclass->wrap(
264 $cv,
265 associated_metaclass => $self,
266 package_name => $class_name,
267 name => $method_name
268 );
269 */
270 ENTER;
271 SAVETMPS;
272
273 PUSHMARK(SP);
274 EXTEND(SP, 8);
275 PUSHs(method_metaclass_name); /* invocant */
00c93f7a 276 mPUSHs(newRV_inc((SV *)cv));
0e0b5b8d 277 PUSHs(associated_metaclass);
278 PUSHs(self);
279 PUSHs(key_package_name);
280 PUSHs(class_name);
281 PUSHs(key_name);
282 mPUSHs(newSVpv(method_name, method_name_len));
283 PUTBACK;
284
285 call_sv(wrap, G_SCALAR | G_METHOD);
286 SPAGAIN;
287 method_object = POPs;
288 PUTBACK;
289 /* $map->{$method_name} = $method_object */
290 sv_setsv(method_slot, method_object);
291
292 FREETMPS;
293 LEAVE;
294 }
295}
296
e0e4674a 297/*
e0e4674a 298get_code_info:
299 Pass in a coderef, returns:
300 [ $pkg_name, $coderef_name ] ie:
301 [ 'Foo::Bar', 'new' ]
302*/
303
304MODULE = Class::MOP PACKAGE = Class::MOP
305
cc856b56 306BOOT:
307 key_name = newSVpvs("name");
308 key_body = newSVpvs("body");
309 key_package = newSVpvs("package");
da88f307 310 key_package_name = newSVpvs("package_name");
c686d0ed 311 key_package_cache_flag = newSVpvs("_package_cache_flag");
312 key_methods = newSVpvs("methods");
cc856b56 313
314 PERL_HASH(hash_name, "name", 4);
315 PERL_HASH(hash_body, "body", 4);
316 PERL_HASH(hash_package, "package", 7);
da88f307 317 PERL_HASH(hash_package_name, "package_name", 12);
c686d0ed 318 PERL_HASH(hash_package_cache_flag, "_package_cache_flag", 19);
319 PERL_HASH(hash_methods, "methods", 7);
cc856b56 320
c94afdc4 321 method_metaclass = newSVpvs("method_metaclass");
322 wrap = newSVpvs("wrap");
323 associated_metaclass = newSVpvs("associated_metaclass");
324
cc856b56 325
d7bf3478 326PROTOTYPES: ENABLE
327
cc856b56 328
e0e4674a 329void
330get_code_info(coderef)
00c93f7a 331 SV *coderef
0e0b5b8d 332 PREINIT:
6d915932 333 char *pkg = NULL;
334 char *name = NULL;
0e0b5b8d 335 PPCODE:
336 if (get_code_info(coderef, &pkg, &name)) {
337 EXTEND(SP, 2);
338 PUSHs(newSVpv(pkg, 0));
339 PUSHs(newSVpv(name, 0));
340 }
e0e4674a 341
863178c8 342PROTOTYPES: DISABLE
343
344void
345is_class_loaded(klass=&PL_sv_undef)
346 SV *klass
347 PREINIT:
348 HV *stash;
349 char *key;
350 I32 keylen;
351 GV *gv;
352 PPCODE:
353 if (!SvPOK(klass) || !SvCUR(klass)) {
354 XSRETURN_NO;
355 }
356
357 stash = gv_stashsv(klass, 0);
358 if (!stash) {
359 XSRETURN_NO;
360 }
361
362 if (hv_exists (stash, "VERSION", 7)) {
363 SV **version = hv_fetch(stash, "VERSION", 7, 0);
364 if (version && *version && GvSV(*version)) {
365 XSRETURN_YES;
366 }
367 }
368
369 if (hv_exists (stash, "ISA", 3)) {
370 SV **isa = hv_fetch(stash, "ISA", 3, 0);
371 if (isa && *isa && GvAV(*isa)) {
372 XSRETURN_YES;
373 }
374 }
375
376 (void)hv_iterinit(stash);
377 while (gv = (GV *)hv_iternextsv(stash, &key, &keylen)) {
378 if (keylen <= 0) {
379 continue;
380 }
381
382 if (key[keylen - 1] == ':' && key[keylen - 2] == ':') {
383 continue;
384 }
385
386 if (!isGV(gv) || GvCV(gv) || GvSV(gv) || GvAV(gv) || GvHV(gv) || GvIO(gv) || GvFORM(gv)) {
387 XSRETURN_YES;
388 }
389 }
390
391 XSRETURN_NO;
15273f3c 392
393MODULE = Class::MOP PACKAGE = Class::MOP::Package
394
863178c8 395PROTOTYPES: ENABLE
396
15273f3c 397void
b49ee39f 398get_all_package_symbols(self, filter=TYPE_FILTER_NONE)
cc856b56 399 SV *self
b49ee39f 400 type_filter_t filter
15273f3c 401 PROTOTYPE: $;$
402 PREINIT:
cc856b56 403 HV *stash = NULL;
b49ee39f 404 HV *symbols = NULL;
75705e60 405 register HE *he;
15273f3c 406 PPCODE:
a7f711e5 407 if ( ! SvROK(self) ) {
988fb42e 408 die("Cannot call get_all_package_symbols as a class method");
409 }
15273f3c 410
b49ee39f 411 if (GIMME_V == G_VOID) {
412 XSRETURN_EMPTY;
413 }
414
15273f3c 415
15273f3c 416 PUTBACK;
417
b49ee39f 418 if ( (he = hv_fetch_ent((HV *)SvRV(self), key_package, 0, hash_package)) ) {
419 stash = gv_stashsv(HeVAL(he), 0);
420 }
15273f3c 421
15273f3c 422
b49ee39f 423 if (!stash) {
6ccb3af5 424 switch (GIMME_V) {
425 case G_SCALAR: XSRETURN_UNDEF; break;
426 case G_ARRAY: XSRETURN_EMPTY; break;
427 }
b49ee39f 428 }
15273f3c 429
b49ee39f 430 symbols = get_all_package_symbols(stash, filter);
15273f3c 431
6ccb3af5 432 switch (GIMME_V) {
433 case G_SCALAR:
434 PUSHs(sv_2mortal(newRV_inc((SV *)symbols)));
435 break;
436 case G_ARRAY:
437 warn("Class::MOP::Package::get_all_package_symbols in list context is deprecated. use scalar context instead.");
438
439 EXTEND(SP, HvKEYS(symbols) * 2);
440
441 while ((he = hv_iternext(symbols))) {
442 PUSHs(hv_iterkeysv(he));
443 PUSHs(sv_2mortal(SvREFCNT_inc(HeVAL(he))));
444 }
445
446 break;
447 default:
448 break;
15273f3c 449 }
450
b49ee39f 451 SvREFCNT_dec((SV *)symbols);
452
e2c189ae 453void
cc856b56 454name(self)
455 SV *self
456 PREINIT:
457 register HE *he;
458 PPCODE:
a7f711e5 459 if ( ! SvROK(self) ) {
988fb42e 460 die("Cannot call name as a class method");
461 }
462
e7b69038 463 if ( (he = hv_fetch_ent((HV *)SvRV(self), key_package, 0, hash_package)) )
cc856b56 464 XPUSHs(HeVAL(he));
465 else
466 ST(0) = &PL_sv_undef;
467
dcbfe027 468MODULE = Class::MOP PACKAGE = Class::MOP::Attribute
cc856b56 469
e2c189ae 470void
cc856b56 471name(self)
472 SV *self
473 PREINIT:
474 register HE *he;
475 PPCODE:
a7f711e5 476 if ( ! SvROK(self) ) {
988fb42e 477 die("Cannot call name as a class method");
478 }
479
e7b69038 480 if ( (he = hv_fetch_ent((HV *)SvRV(self), key_name, 0, hash_name)) )
cc856b56 481 XPUSHs(HeVAL(he));
482 else
483 ST(0) = &PL_sv_undef;
484
dcbfe027 485MODULE = Class::MOP PACKAGE = Class::MOP::Method
cc856b56 486
e2c189ae 487void
da88f307 488name(self)
489 SV *self
490 PREINIT:
491 register HE *he;
492 PPCODE:
a7f711e5 493 if ( ! SvROK(self) ) {
da88f307 494 die("Cannot call name as a class method");
495 }
496
e7b69038 497 if ( (he = hv_fetch_ent((HV *)SvRV(self), key_name, 0, hash_name)) )
da88f307 498 XPUSHs(HeVAL(he));
499 else
500 ST(0) = &PL_sv_undef;
501
e2c189ae 502void
da88f307 503package_name(self)
504 SV *self
505 PREINIT:
506 register HE *he;
507 PPCODE:
a7f711e5 508 if ( ! SvROK(self) ) {
da88f307 509 die("Cannot call package_name as a class method");
510 }
511
e7b69038 512 if ( (he = hv_fetch_ent((HV *)SvRV(self), key_package_name, 0, hash_package_name)) )
da88f307 513 XPUSHs(HeVAL(he));
514 else
515 ST(0) = &PL_sv_undef;
516
e2c189ae 517void
cc856b56 518body(self)
519 SV *self
520 PREINIT:
521 register HE *he;
522 PPCODE:
a7f711e5 523 if ( ! SvROK(self) ) {
da88f307 524 die("Cannot call body as a class method");
525 }
526
e7b69038 527 if ( (he = hv_fetch_ent((HV *)SvRV(self), key_body, 0, hash_body)) )
cc856b56 528 XPUSHs(HeVAL(he));
529 else
530 ST(0) = &PL_sv_undef;
c94afdc4 531
532
533MODULE = Class::MOP PACKAGE = Class::MOP::Class
534
535void
536get_method_map(self)
00c93f7a 537 SV *self
c200a770 538 PREINIT:
c686d0ed 539 HV *const obj = (HV *)SvRV(self);
540 SV *const class_name = HeVAL( hv_fetch_ent(obj, key_package, 0, hash_package) );
541 HV *const stash = gv_stashsv(class_name, 0);
c200a770 542 UV const current = check_package_cache_flag(stash);
c686d0ed 543 SV *const cache_flag = HeVAL( hv_fetch_ent(obj, key_package_cache_flag, TRUE, hash_package_cache_flag));
544 SV *const map_ref = HeVAL( hv_fetch_ent(obj, key_methods, TRUE, hash_methods));
c200a770 545 PPCODE:
c200a770 546 /* in $self->{methods} does not yet exist (or got deleted) */
c686d0ed 547 if ( !SvROK(map_ref) || SvTYPE(SvRV(map_ref)) != SVt_PVHV ) {
00c93f7a 548 SV *new_map_ref = newRV_noinc((SV *)newHV());
c200a770 549 sv_2mortal(new_map_ref);
550 sv_setsv(map_ref, new_map_ref);
551 }
c94afdc4 552
c686d0ed 553 if ( !SvOK(cache_flag) || SvUV(cache_flag) != current ) {
00c93f7a 554 mop_update_method_map(aTHX_ self, class_name, stash, (HV *)SvRV(map_ref));
c200a770 555 sv_setuv(cache_flag, check_package_cache_flag(stash)); /* update_cache_flag() */
c200a770 556 }
557
558 XPUSHs(map_ref);