]> git.vpit.fr Git - perl/modules/Variable-Magic.git/blob - Magic.xs
Free the thread local context from an atexit callback
[perl/modules/Variable-Magic.git] / Magic.xs
1 /* This file is part of the Variable::Magic Perl module.
2  * See http://search.cpan.org/dist/Variable-Magic/ */
3
4 #include <stdarg.h> /* <va_list>, va_{start,arg,end}, ... */
5
6 #include <stdio.h>  /* sprintf() */
7
8 #define PERL_NO_GET_CONTEXT
9 #include "EXTERN.h"
10 #include "perl.h"
11 #include "XSUB.h"
12
13 #define __PACKAGE__ "Variable::Magic"
14
15 #ifndef VMG_PERL_PATCHLEVEL
16 # ifdef PERL_PATCHNUM
17 #  define VMG_PERL_PATCHLEVEL PERL_PATCHNUM
18 # else
19 #  define VMG_PERL_PATCHLEVEL 0
20 # endif
21 #endif
22
23 #define VMG_HAS_PERL(R, V, S) (PERL_REVISION > (R) || (PERL_REVISION == (R) && (PERL_VERSION > (V) || (PERL_VERSION == (V) && (PERL_SUBVERSION >= (S))))))
24
25 #define VMG_HAS_PERL_BRANCH(R, V, S) (PERL_REVISION == (R) && PERL_VERSION == (V) && PERL_SUBVERSION >= (S))
26
27 #define VMG_HAS_PERL_MAINT(R, V, S, P) (PERL_REVISION == (R) && PERL_VERSION == (V) && (VMG_PERL_PATCHLEVEL >= (P) || (!VMG_PERL_PATCHLEVEL && PERL_SUBVERSION >= (S))))
28
29 /* --- Threads and multiplicity -------------------------------------------- */
30
31 #ifndef NOOP
32 # define NOOP
33 #endif
34
35 #ifndef dNOOP
36 # define dNOOP
37 #endif
38
39 /* Safe unless stated otherwise in Makefile.PL */
40 #ifndef VMG_FORKSAFE
41 # define VMG_FORKSAFE 1
42 #endif
43
44 #ifndef VMG_MULTIPLICITY
45 # if defined(MULTIPLICITY) || defined(PERL_IMPLICIT_CONTEXT)
46 #  define VMG_MULTIPLICITY 1
47 # else
48 #  define VMG_MULTIPLICITY 0
49 # endif
50 #endif
51 #if VMG_MULTIPLICITY && !defined(tTHX)
52 # define tTHX PerlInterpreter*
53 #endif
54
55 #if VMG_MULTIPLICITY && defined(USE_ITHREADS) && defined(dMY_CXT) && defined(MY_CXT) && defined(START_MY_CXT) && defined(MY_CXT_INIT) && (defined(MY_CXT_CLONE) || defined(dMY_CXT_SV))
56 # define VMG_THREADSAFE 1
57 # ifndef MY_CXT_CLONE
58 #  define MY_CXT_CLONE \
59     dMY_CXT_SV;                                                      \
60     my_cxt_t *my_cxtp = (my_cxt_t*)SvPVX(newSV(sizeof(my_cxt_t)-1)); \
61     Copy(INT2PTR(my_cxt_t*, SvUV(my_cxt_sv)), my_cxtp, 1, my_cxt_t); \
62     sv_setuv(my_cxt_sv, PTR2UV(my_cxtp))
63 # endif
64 #else
65 # define VMG_THREADSAFE 0
66 # undef  dMY_CXT
67 # define dMY_CXT      dNOOP
68 # undef  MY_CXT
69 # define MY_CXT       vmg_globaldata
70 # undef  START_MY_CXT
71 # define START_MY_CXT STATIC my_cxt_t MY_CXT;
72 # undef  MY_CXT_INIT
73 # define MY_CXT_INIT  NOOP
74 # undef  MY_CXT_CLONE
75 # define MY_CXT_CLONE NOOP
76 #endif
77
78 #if VMG_THREADSAFE
79
80 STATIC SV *vmg_clone(pTHX_ SV *sv, tTHX owner) {
81 #define vmg_clone(P, O) vmg_clone(aTHX_ (P), (O))
82  CLONE_PARAMS param;
83
84  param.stashes    = NULL; /* don't need it unless sv is a PVHV */
85  param.flags      = 0;
86  param.proto_perl = owner;
87
88  return SvREFCNT_inc(sv_dup(sv, &param));
89 }
90
91 #endif /* VMG_THREADSAFE */
92
93 /* --- Compatibility ------------------------------------------------------- */
94
95 #ifndef Newx
96 # define Newx(v, n, c) New(0, v, n, c)
97 #endif
98
99 #ifndef SvMAGIC_set
100 # define SvMAGIC_set(sv, val) (SvMAGIC(sv) = (val))
101 #endif
102
103 #ifndef mPUSHu
104 # define mPUSHu(U) PUSHs(sv_2mortal(newSVuv(U)))
105 #endif
106
107 #ifndef SvPV_const
108 # define SvPV_const SvPV
109 #endif
110
111 #ifndef PERL_MAGIC_ext
112 # define PERL_MAGIC_ext '~'
113 #endif
114
115 #ifndef PERL_MAGIC_tied
116 # define PERL_MAGIC_tied 'P'
117 #endif
118
119 #ifndef MGf_COPY
120 # define MGf_COPY 0
121 #endif
122
123 #ifndef MGf_DUP
124 # define MGf_DUP 0
125 #endif
126
127 #ifndef MGf_LOCAL
128 # define MGf_LOCAL 0
129 #endif
130
131 #ifndef IN_PERL_COMPILETIME
132 # define IN_PERL_COMPILETIME (PL_curcop == &PL_compiling)
133 #endif
134
135 #if VMG_HAS_PERL(5, 10, 0) || defined(PL_parser)
136 # ifndef PL_error_count
137 #  define PL_error_count PL_parser->error_count
138 # endif
139 #else
140 # ifndef PL_error_count
141 #  define PL_error_count PL_Ierror_count
142 # endif
143 #endif
144
145 /* uvar magic and Hash::Util::FieldHash were commited with 28419, but only
146  * enable it on 5.10 */
147 #if VMG_HAS_PERL(5, 10, 0)
148 # define VMG_UVAR 1
149 #else
150 # define VMG_UVAR 0
151 #endif
152
153 /* Applied to dev-5.9 as 25854, integrated to maint-5.8 as 28160, partially
154  * reverted to dev-5.11 as 9cdcb38b */
155 #if VMG_HAS_PERL_MAINT(5, 8, 9, 28160) || VMG_HAS_PERL_MAINT(5, 9, 3, 25854) || VMG_HAS_PERL(5, 10, 0)
156 # ifndef VMG_COMPAT_ARRAY_PUSH_NOLEN
157 /* This branch should only apply for perls before the official 5.11.0 release.
158  * Makefile.PL takes care of the higher ones. */
159 #  define VMG_COMPAT_ARRAY_PUSH_NOLEN 1
160 # endif
161 # ifndef VMG_COMPAT_ARRAY_PUSH_NOLEN_VOID
162 #  define VMG_COMPAT_ARRAY_PUSH_NOLEN_VOID 1
163 # endif
164 #else
165 # ifndef VMG_COMPAT_ARRAY_PUSH_NOLEN
166 #  define VMG_COMPAT_ARRAY_PUSH_NOLEN 0
167 # endif
168 # ifndef VMG_COMPAT_ARRAY_PUSH_NOLEN_VOID
169 #  define VMG_COMPAT_ARRAY_PUSH_NOLEN_VOID 0
170 # endif
171 #endif
172
173 /* Applied to dev-5.11 as 34908 */
174 #if VMG_HAS_PERL_MAINT(5, 11, 0, 34908)
175 # define VMG_COMPAT_ARRAY_UNSHIFT_NOLEN_VOID 1
176 #else
177 # define VMG_COMPAT_ARRAY_UNSHIFT_NOLEN_VOID 0
178 #endif
179
180 /* Applied to dev-5.9 as 31473 (see #43357), integrated to maint-5.8 as 32542 */
181 #if VMG_HAS_PERL_MAINT(5, 8, 9, 32542) || VMG_HAS_PERL_MAINT(5, 9, 5, 31473) || VMG_HAS_PERL(5, 10, 0)
182 # define VMG_COMPAT_ARRAY_UNDEF_CLEAR 1
183 #else
184 # define VMG_COMPAT_ARRAY_UNDEF_CLEAR 0
185 #endif
186
187 #if VMG_HAS_PERL_MAINT(5, 11, 0, 32969)
188 # define VMG_COMPAT_SCALAR_LENGTH_NOLEN 1
189 #else
190 # define VMG_COMPAT_SCALAR_LENGTH_NOLEN 0
191 #endif
192
193 #if VMG_UVAR
194
195 /* Bug-free mg_magical - see http://www.xray.mpe.mpg.de/mailing-lists/perl5-porters/2008-01/msg00036.html - but specialized to our needs. */
196 STATIC void vmg_sv_magicuvar(pTHX_ SV *sv, const char *uf, I32 len) {
197 #define vmg_sv_magicuvar(S, U, L) vmg_sv_magicuvar(aTHX_ (S), (U), (L))
198  const MAGIC* mg;
199  sv_magic(sv, NULL, PERL_MAGIC_uvar, uf, len);
200  /* uvar magic has set and get magic, hence this has set SVs_GMG and SVs_SMG. */
201  if ((mg = SvMAGIC(sv))) {
202   SvRMAGICAL_off(sv);
203   do {
204    const MGVTBL* const vtbl = mg->mg_virtual;
205    if (vtbl) {
206     if (vtbl->svt_clear) {
207      SvRMAGICAL_on(sv);
208      break;
209     }
210    }
211   } while ((mg = mg->mg_moremagic));
212  }
213 }
214
215 #endif /* VMG_UVAR */
216
217 /* --- Stolen chunk of B --------------------------------------------------- */
218
219 typedef enum {
220  OPc_NULL   = 0,
221  OPc_BASEOP = 1,
222  OPc_UNOP   = 2,
223  OPc_BINOP  = 3,
224  OPc_LOGOP  = 4,
225  OPc_LISTOP = 5,
226  OPc_PMOP   = 6,
227  OPc_SVOP   = 7,
228  OPc_PADOP  = 8,
229  OPc_PVOP   = 9,
230  OPc_LOOP   = 10,
231  OPc_COP    = 11,
232  OPc_MAX    = 12
233 } opclass;
234
235 STATIC const char *const vmg_opclassnames[] = {
236  "B::NULL",
237  "B::OP",
238  "B::UNOP",
239  "B::BINOP",
240  "B::LOGOP",
241  "B::LISTOP",
242  "B::PMOP",
243  "B::SVOP",
244  "B::PADOP",
245  "B::PVOP",
246  "B::LOOP",
247  "B::COP"
248 };
249
250 STATIC opclass vmg_opclass(const OP *o) {
251 #if 0
252  if (!o)
253   return OPc_NULL;
254 #endif
255
256  if (o->op_type == 0)
257   return (o->op_flags & OPf_KIDS) ? OPc_UNOP : OPc_BASEOP;
258
259  if (o->op_type == OP_SASSIGN)
260   return ((o->op_private & OPpASSIGN_BACKWARDS) ? OPc_UNOP : OPc_BINOP);
261
262  if (o->op_type == OP_AELEMFAST) {
263   if (o->op_flags & OPf_SPECIAL)
264    return OPc_BASEOP;
265   else
266 #ifdef USE_ITHREADS
267    return OPc_PADOP;
268 #else
269    return OPc_SVOP;
270 #endif
271  }
272
273 #ifdef USE_ITHREADS
274  if (o->op_type == OP_GV || o->op_type == OP_GVSV || o->op_type == OP_RCATLINE)
275   return OPc_PADOP;
276 #endif
277
278  switch (PL_opargs[o->op_type] & OA_CLASS_MASK) {
279   case OA_BASEOP:
280    return OPc_BASEOP;
281   case OA_UNOP:
282    return OPc_UNOP;
283   case OA_BINOP:
284    return OPc_BINOP;
285   case OA_LOGOP:
286    return OPc_LOGOP;
287   case OA_LISTOP:
288    return OPc_LISTOP;
289   case OA_PMOP:
290    return OPc_PMOP;
291   case OA_SVOP:
292    return OPc_SVOP;
293   case OA_PADOP:
294    return OPc_PADOP;
295   case OA_PVOP_OR_SVOP:
296    return (o->op_private & (OPpTRANS_TO_UTF|OPpTRANS_FROM_UTF)) ? OPc_SVOP : OPc_PVOP;
297   case OA_LOOP:
298    return OPc_LOOP;
299   case OA_COP:
300    return OPc_COP;
301   case OA_BASEOP_OR_UNOP:
302    return (o->op_flags & OPf_KIDS) ? OPc_UNOP : OPc_BASEOP;
303   case OA_FILESTATOP:
304    return ((o->op_flags & OPf_KIDS) ? OPc_UNOP :
305 #ifdef USE_ITHREADS
306            (o->op_flags & OPf_REF) ? OPc_PADOP : OPc_BASEOP);
307 #else
308            (o->op_flags & OPf_REF) ? OPc_SVOP : OPc_BASEOP);
309 #endif
310   case OA_LOOPEXOP:
311    if (o->op_flags & OPf_STACKED)
312     return OPc_UNOP;
313    else if (o->op_flags & OPf_SPECIAL)
314     return OPc_BASEOP;
315    else
316     return OPc_PVOP;
317  }
318
319  return OPc_BASEOP;
320 }
321
322 /* --- Error messages ------------------------------------------------------ */
323
324 STATIC const char vmg_invalid_wiz[]    = "Invalid wizard object";
325 STATIC const char vmg_wrongargnum[]    = "Wrong number of arguments";
326 STATIC const char vmg_argstorefailed[] = "Error while storing arguments";
327
328 /* --- Signatures ---------------------------------------------------------- */
329
330 #define SIG_WZO ((U16) (0x3891))
331 #define SIG_WIZ ((U16) (0x3892))
332
333 /* --- MGWIZ structure ----------------------------------------------------- */
334
335 typedef struct {
336  MGVTBL *vtbl;
337
338  U8 uvar;
339  U8 opinfo;
340
341  SV *cb_data;
342  SV *cb_get, *cb_set, *cb_len, *cb_clear, *cb_free;
343 #if MGf_COPY
344  SV *cb_copy;
345 #endif /* MGf_COPY */
346 #if MGf_DUP
347  SV *cb_dup;
348 #endif /* MGf_DUP */
349 #if MGf_LOCAL
350  SV *cb_local;
351 #endif /* MGf_LOCAL */
352 #if VMG_UVAR
353  SV *cb_fetch, *cb_store, *cb_exists, *cb_delete;
354 #endif /* VMG_UVAR */
355
356 #if VMG_MULTIPLICITY
357  tTHX owner;
358 #endif /* VMG_MULTIPLICITY */
359 } MGWIZ;
360
361 STATIC void vmg_mgwiz_free(pTHX_ MGWIZ *w) {
362 #define vmg_mgwiz_free(W) vmg_mgwiz_free(aTHX_ (W))
363  if (!w)
364   return;
365
366  if (w->cb_data)   SvREFCNT_dec(w->cb_data);
367  if (w->cb_get)    SvREFCNT_dec(w->cb_get);
368  if (w->cb_set)    SvREFCNT_dec(w->cb_set);
369  if (w->cb_len)    SvREFCNT_dec(w->cb_len);
370  if (w->cb_clear)  SvREFCNT_dec(w->cb_clear);
371  if (w->cb_free)   SvREFCNT_dec(w->cb_free);
372 #if MGf_COPY
373  if (w->cb_copy)   SvREFCNT_dec(w->cb_copy);
374 #endif /* MGf_COPY */
375 #if 0 /* MGf_DUP */
376  if (w->cb_dup)    SvREFCNT_dec(w->cb_dup);
377 #endif /* MGf_DUP */
378 #if MGf_LOCAL
379  if (w->cb_local)  SvREFCNT_dec(w->cb_local);
380 #endif /* MGf_LOCAL */
381 #if VMG_UVAR
382  if (w->cb_fetch)  SvREFCNT_dec(w->cb_fetch);
383  if (w->cb_store)  SvREFCNT_dec(w->cb_store);
384  if (w->cb_exists) SvREFCNT_dec(w->cb_exists);
385  if (w->cb_delete) SvREFCNT_dec(w->cb_delete);
386 #endif /* VMG_UVAR */
387
388  Safefree(w->vtbl);
389  Safefree(w);
390
391  return;
392 }
393
394 #if VMG_THREADSAFE
395
396 #define VMG_CLONE_CB(N) \
397  z->cb_ ## N = (w->cb_ ## N) ? newRV_noinc(vmg_clone(SvRV(w->cb_ ## N), \
398                                            w->owner))                   \
399                              : NULL;
400
401 STATIC MGWIZ *vmg_mgwiz_clone(pTHX_ const MGWIZ *w) {
402 #define vmg_mgwiz_clone(W) vmg_mgwiz_clone(aTHX_ (W))
403  MGVTBL *t;
404  MGWIZ *z;
405
406  if (!w)
407   return NULL;
408
409  Newx(t, 1, MGVTBL);
410  Copy(w->vtbl, t, 1, MGVTBL);
411
412  Newx(z, 1, MGWIZ);
413
414  z->vtbl   = t;
415  z->uvar   = w->uvar;
416  z->opinfo = w->opinfo;
417
418  VMG_CLONE_CB(data);
419  VMG_CLONE_CB(get);
420  VMG_CLONE_CB(set);
421  VMG_CLONE_CB(len);
422  VMG_CLONE_CB(clear);
423  VMG_CLONE_CB(free);
424 #if MGf_COPY
425  VMG_CLONE_CB(copy);
426 #endif /* MGf_COPY */
427 #if MGf_DUP
428  VMG_CLONE_CB(dup);
429 #endif /* MGf_DUP */
430 #if MGf_LOCAL
431  VMG_CLONE_CB(local);
432 #endif /* MGf_LOCAL */
433 #if VMG_UVAR
434  VMG_CLONE_CB(fetch);
435  VMG_CLONE_CB(store);
436  VMG_CLONE_CB(exists);
437  VMG_CLONE_CB(delete);
438 #endif /* VMG_UVAR */
439
440  z->owner = aTHX;
441
442  return z;
443 }
444
445 #endif /* VMG_THREADSAFE */
446
447 /* --- Context-safe global data -------------------------------------------- */
448
449 #if VMG_THREADSAFE
450
451 #define PTABLE_NAME        ptable
452 #define PTABLE_VAL_FREE(V) vmg_mgwiz_free(V)
453
454 #define pPTBL  pTHX
455 #define pPTBL_ pTHX_
456 #define aPTBL  aTHX
457 #define aPTBL_ aTHX_
458
459 #include "ptable.h"
460
461 #define ptable_store(T, K, V) ptable_store(aTHX_ (T), (K), (V))
462 #define ptable_clear(T)       ptable_clear(aTHX_ (T))
463 #define ptable_free(T)        ptable_free(aTHX_ (T))
464
465 #endif /* VMG_THREADSAFE */
466
467 #define MY_CXT_KEY __PACKAGE__ "::_guts" XS_VERSION
468
469 typedef struct {
470 #if VMG_THREADSAFE
471  ptable *wizards;
472  tTHX    owner;
473 #endif
474  HV     *b__op_stashes[OPc_MAX];
475 } my_cxt_t;
476
477 START_MY_CXT
478
479 #if VMG_THREADSAFE
480
481 STATIC void vmg_ptable_clone(pTHX_ ptable_ent *ent, void *ud_) {
482  my_cxt_t *ud = ud_;
483  MGWIZ *w;
484
485  if (ud->owner == aTHX)
486   return;
487
488  w = vmg_mgwiz_clone(ent->val);
489  if (w)
490   ptable_store(ud->wizards, ent->key, w);
491 }
492
493 #endif /* VMG_THREADSAFE */
494
495 /* --- Wizard objects ------------------------------------------------------ */
496
497 STATIC int vmg_wizard_free(pTHX_ SV *sv, MAGIC *mg);
498
499 STATIC MGVTBL vmg_wizard_vtbl = {
500  NULL,            /* get */
501  NULL,            /* set */
502  NULL,            /* len */
503  NULL,            /* clear */
504  vmg_wizard_free, /* free */
505 #if MGf_COPY
506  NULL,            /* copy */
507 #endif /* MGf_COPY */
508 #if MGf_DUP
509  NULL,            /* dup */
510 #endif /* MGf_DUP */
511 #if MGf_LOCAL
512  NULL,            /* local */
513 #endif /* MGf_LOCAL */
514 };
515
516 /* ... Wizard constructor .................................................. */
517
518 STATIC SV *vmg_wizard_new(pTHX_ const MGWIZ *w) {
519 #define vmg_wizard_new(W) vmg_wizard_new(aTHX_ (W))
520  SV *wiz = newSVuv(PTR2IV(w));
521
522  if (w) {
523   MAGIC *mg = sv_magicext(wiz, NULL, PERL_MAGIC_ext, &vmg_wizard_vtbl, NULL, 0);
524   mg->mg_private = SIG_WZO;
525  }
526  SvREADONLY_on(wiz);
527
528  return wiz;
529 }
530
531 STATIC const SV *vmg_wizard_validate(pTHX_ const SV *wiz) {
532 #define vmg_wizard_validate(W) vmg_wizard_validate(aTHX_ (W))
533  if (SvROK(wiz)) {
534   wiz = SvRV(wiz);
535   if (SvIOK(wiz))
536    return wiz;
537  }
538
539  croak(vmg_invalid_wiz);
540 }
541
542 #define vmg_wizard_id(W)         SvIVX((const SV *) (W))
543 #define vmg_wizard_main_mgwiz(W) INT2PTR(const MGWIZ *, vmg_wizard_id(W))
544
545 /* ... Wizard destructor ................................................... */
546
547 STATIC int vmg_wizard_free(pTHX_ SV *sv, MAGIC *mg) {
548  MGWIZ *w;
549
550  if (PL_dirty) /* During global destruction, the context is already freed */
551   return 0;
552
553  w = (MGWIZ *) vmg_wizard_main_mgwiz(sv);
554
555 #if VMG_THREADSAFE
556  {
557   dMY_CXT;
558   ptable_store(MY_CXT.wizards, w, NULL);
559  }
560 #else /* VMG_THREADSAFE */
561  vmg_mgwiz_free(w);
562 #endif /* !VMG_THREADSAFE */
563
564  return 0;
565 }
566
567 #if VMG_THREADSAFE
568
569 STATIC const MGWIZ *vmg_wizard_mgwiz(pTHX_ const SV *wiz) {
570 #define vmg_wizard_mgwiz(W) vmg_wizard_mgwiz(aTHX_ ((const SV *) (W)))
571  const MGWIZ *w;
572
573  w = vmg_wizard_main_mgwiz(wiz);
574  if (w->owner == aTHX)
575   return w;
576
577  {
578   dMY_CXT;
579   return ptable_fetch(MY_CXT.wizards, w);
580  }
581 }
582
583 #else /* VMG_THREADSAFE */
584
585 #define vmg_wizard_mgwiz(W) vmg_wizard_main_mgwiz(W)
586
587 #endif /* !VMG_THREADSAFE */
588
589 /* --- User-level functions implementation --------------------------------- */
590
591 STATIC const MAGIC *vmg_find(const SV *sv, const SV *wiz) {
592  const MAGIC *mg, *moremagic;
593  UV wid;
594
595  if (SvTYPE(sv) < SVt_PVMG)
596   return NULL;
597
598  wid = vmg_wizard_id(wiz);
599  for (mg = SvMAGIC(sv); mg; mg = moremagic) {
600   moremagic = mg->mg_moremagic;
601   if (mg->mg_type == PERL_MAGIC_ext && mg->mg_private == SIG_WIZ) {
602    UV zid = vmg_wizard_id(mg->mg_ptr);
603    if (zid == wid)
604     return mg;
605   }
606  }
607
608  return NULL;
609 }
610
611 /* ... Construct private data .............................................. */
612
613 STATIC SV *vmg_data_new(pTHX_ SV *ctor, SV *sv, AV *args) {
614 #define vmg_data_new(C, S, A) vmg_data_new(aTHX_ (C), (S), (A))
615  SV *nsv;
616  I32 i, alen = (args == NULL) ? 0 : av_len(args);
617
618  dSP;
619
620  ENTER;
621  SAVETMPS;
622
623  PUSHMARK(SP);
624  EXTEND(SP, alen + 1);
625  PUSHs(sv_2mortal(newRV_inc(sv)));
626  for (i = 0; i < alen; ++i)
627   PUSHs(*av_fetch(args, i, 0));
628  PUTBACK;
629
630  call_sv(ctor, G_SCALAR);
631
632  SPAGAIN;
633  nsv = POPs;
634 #if VMG_HAS_PERL(5, 8, 3)
635  SvREFCNT_inc(nsv);    /* Or it will be destroyed in FREETMPS */
636 #else
637  nsv = sv_newref(nsv); /* Workaround some bug in SvREFCNT_inc() */
638 #endif
639  PUTBACK;
640
641  FREETMPS;
642  LEAVE;
643
644  return nsv;
645 }
646
647 STATIC SV *vmg_data_get(pTHX_ SV *sv, const SV *wiz) {
648 #define vmg_data_get(S, W) vmg_data_get(aTHX_ (S), (W))
649  const MAGIC *mg = vmg_find(sv, wiz);
650  return mg ? mg->mg_obj : NULL;
651
652
653 /* ... Magic cast/dispell .................................................. */
654
655 #if VMG_UVAR
656 STATIC I32 vmg_svt_val(pTHX_ IV, SV *);
657
658 STATIC void vmg_uvar_del(SV *sv, MAGIC *prevmagic, MAGIC *mg, MAGIC *moremagic) {
659  if (prevmagic) {
660   prevmagic->mg_moremagic = moremagic;
661  } else {
662   SvMAGIC_set(sv, moremagic);
663  }
664  mg->mg_moremagic = NULL;
665  Safefree(mg->mg_ptr);
666  Safefree(mg);
667 }
668 #endif /* VMG_UVAR */
669
670 STATIC UV vmg_cast(pTHX_ SV *sv, const SV *wiz, AV *args) {
671 #define vmg_cast(S, W, A) vmg_cast(aTHX_ (S), (W), (A))
672  MAGIC       *mg, *moremagic = NULL;
673  SV          *data;
674  const MGWIZ *w;
675  U32          oldgmg;
676
677  if (vmg_find(sv, wiz))
678   return 1;
679
680  w = vmg_wizard_mgwiz(wiz);
681  oldgmg = SvGMAGICAL(sv);
682
683  data = (w->cb_data) ? vmg_data_new(w->cb_data, sv, args) : NULL;
684  mg = sv_magicext(sv, data, PERL_MAGIC_ext, w->vtbl, (const char *) wiz, HEf_SVKEY);
685  mg->mg_private = SIG_WIZ;
686 #if MGf_COPY
687  if (w->cb_copy)
688   mg->mg_flags |= MGf_COPY;
689 #endif /* MGf_COPY */
690 #if 0 /* MGf_DUP */
691  if (w->cb_dup)
692   mg->mg_flags |= MGf_DUP;
693 #endif /* MGf_DUP */
694 #if MGf_LOCAL
695  if (w->cb_local)
696   mg->mg_flags |= MGf_LOCAL;
697 #endif /* MGf_LOCAL */
698
699  if (SvTYPE(sv) < SVt_PVHV)
700   goto done;
701
702  /* The GMAGICAL flag only says that a hash is tied or has uvar magic - get
703   * magic is actually never called for them. If the GMAGICAL flag was off before
704   * calling sv_magicext(), the hash isn't tied and has no uvar magic. If it's
705   * now on, then this wizard has get magic. Hence we can work around the
706   * get/clear shortcoming by turning the GMAGICAL flag off. If the current magic
707   * has uvar callbacks, it will be turned back on later. */
708  if (!oldgmg && SvGMAGICAL(sv))
709   SvGMAGICAL_off(sv);
710
711 #if VMG_UVAR
712  if (w->uvar) {
713   MAGIC *prevmagic;
714   struct ufuncs uf[2];
715
716   uf[0].uf_val   = vmg_svt_val;
717   uf[0].uf_set   = NULL;
718   uf[0].uf_index = 0;
719   uf[1].uf_val   = NULL;
720   uf[1].uf_set   = NULL;
721   uf[1].uf_index = 0;
722
723   /* One uvar magic in the chain is enough. */
724   for (prevmagic = NULL, mg = SvMAGIC(sv); mg; prevmagic = mg, mg = moremagic) {
725    moremagic = mg->mg_moremagic;
726    if (mg->mg_type == PERL_MAGIC_uvar)
727     break;
728   }
729
730   if (mg) { /* Found another uvar magic. */
731    struct ufuncs *olduf = (struct ufuncs *) mg->mg_ptr;
732    if (olduf->uf_val == vmg_svt_val) {
733     /* It's our uvar magic, nothing to do. oldgmg was true. */
734     goto done;
735    } else {
736     /* It's another uvar magic, backup it and replace it by ours. */
737     uf[1] = *olduf;
738     vmg_uvar_del(sv, prevmagic, mg, moremagic);
739    }
740   }
741
742   vmg_sv_magicuvar(sv, (const char *) &uf, sizeof(uf));
743   /* Our hash now carries uvar magic. The uvar/clear shortcoming has to be
744    * handled by our uvar callback. */
745  }
746 #endif /* VMG_UVAR */
747
748 done:
749  return 1;
750 }
751
752 STATIC UV vmg_dispell(pTHX_ SV *sv, const SV *wiz) {
753 #define vmg_dispell(S, W) vmg_dispell(aTHX_ (S), (W))
754 #if VMG_UVAR
755  U32 uvars = 0;
756 #endif /* VMG_UVAR */
757  MAGIC *mg, *prevmagic, *moremagic = NULL;
758  UV wid = vmg_wizard_id(wiz);
759
760  if (SvTYPE(sv) < SVt_PVMG)
761   return 0;
762
763  for (prevmagic = NULL, mg = SvMAGIC(sv); mg; prevmagic = mg, mg = moremagic) {
764   moremagic = mg->mg_moremagic;
765   if (mg->mg_type == PERL_MAGIC_ext && mg->mg_private == SIG_WIZ) {
766    const MGWIZ *z   = vmg_wizard_mgwiz(mg->mg_ptr);
767    UV           zid = vmg_wizard_id(mg->mg_ptr);
768    if (zid == wid) {
769 #if VMG_UVAR
770     /* If the current has no uvar, short-circuit uvar deletion. */
771     uvars = z->uvar ? (uvars + 1) : 0;
772 #endif /* VMG_UVAR */
773     break;
774 #if VMG_UVAR
775    } else if (z->uvar) {
776     ++uvars;
777     /* We can't break here since we need to find the ext magic to delete. */
778 #endif /* VMG_UVAR */
779    }
780   }
781  }
782  if (!mg)
783   return 0;
784
785  if (prevmagic) {
786   prevmagic->mg_moremagic = moremagic;
787  } else {
788   SvMAGIC_set(sv, moremagic);
789  }
790  mg->mg_moremagic = NULL;
791
792  /* Destroy private data */
793  if (mg->mg_obj != sv)
794   SvREFCNT_dec(mg->mg_obj);
795  /* Unreference the wizard */
796  SvREFCNT_dec((SV *) mg->mg_ptr);
797  Safefree(mg);
798
799 #if VMG_UVAR
800  if (uvars == 1 && SvTYPE(sv) >= SVt_PVHV) {
801   /* mg was the first ext magic in the chain that had uvar */
802
803   for (mg = moremagic; mg; mg = mg->mg_moremagic) {
804    if (mg->mg_type == PERL_MAGIC_ext && mg->mg_private == SIG_WIZ) {
805     const MGWIZ *z = vmg_wizard_mgwiz(mg->mg_ptr);
806     if (z->uvar) {
807      ++uvars;
808      break;
809     }
810    }
811   }
812
813   if (uvars == 1) {
814    struct ufuncs *uf;
815    for (prevmagic = NULL, mg = SvMAGIC(sv); mg; prevmagic = mg, mg = moremagic){
816     moremagic = mg->mg_moremagic;
817     if (mg->mg_type == PERL_MAGIC_uvar)
818      break;
819    }
820    /* assert(mg); */
821    uf = (struct ufuncs *) mg->mg_ptr;
822    /* assert(uf->uf_val == vmg_svt_val); */
823    if (uf[1].uf_val || uf[1].uf_set) {
824     /* Revert the original uvar magic. */
825     uf[0] = uf[1];
826     Renew(uf, 1, struct ufuncs);
827     mg->mg_ptr = (char *) uf;
828     mg->mg_len = sizeof(struct ufuncs);
829    } else {
830     /* Remove the uvar magic. */
831     vmg_uvar_del(sv, prevmagic, mg, moremagic);
832    }
833   }
834  }
835 #endif /* VMG_UVAR */
836
837  return 1;
838 }
839
840 /* ... OP info ............................................................. */
841
842 #define VMG_OP_INFO_NAME   1
843 #define VMG_OP_INFO_OBJECT 2
844
845 #if VMG_THREADSAFE
846 STATIC perl_mutex vmg_op_name_init_mutex;
847 #endif
848
849 STATIC U32           vmg_op_name_init      = 0;
850 STATIC unsigned char vmg_op_name_len[MAXO] = { 0 };
851
852 STATIC void vmg_op_info_init(pTHX_ unsigned int opinfo) {
853 #define vmg_op_info_init(W) vmg_op_info_init(aTHX_ (W))
854  switch (opinfo) {
855   case VMG_OP_INFO_NAME:
856 #if VMG_THREADSAFE
857    MUTEX_LOCK(&vmg_op_name_init_mutex);
858 #endif
859    if (!vmg_op_name_init) {
860     OPCODE t;
861     for (t = 0; t < OP_max; ++t)
862      vmg_op_name_len[t] = strlen(PL_op_name[t]);
863     vmg_op_name_init = 1;
864    }
865 #if VMG_THREADSAFE
866    MUTEX_UNLOCK(&vmg_op_name_init_mutex);
867 #endif
868    break;
869   case VMG_OP_INFO_OBJECT: {
870    dMY_CXT;
871    if (!MY_CXT.b__op_stashes[0]) {
872     opclass c;
873     require_pv("B.pm");
874     for (c = 0; c < OPc_MAX; ++c)
875      MY_CXT.b__op_stashes[c] = gv_stashpv(vmg_opclassnames[c], 1);
876    }
877    break;
878   }
879   default:
880    break;
881  }
882 }
883
884 STATIC SV *vmg_op_info(pTHX_ unsigned int opinfo) {
885 #define vmg_op_info(W) vmg_op_info(aTHX_ (W))
886  if (!PL_op)
887   return &PL_sv_undef;
888
889  switch (opinfo) {
890   case VMG_OP_INFO_NAME: {
891    OPCODE t = PL_op->op_type;
892    return sv_2mortal(newSVpvn(PL_op_name[t], vmg_op_name_len[t]));
893   }
894   case VMG_OP_INFO_OBJECT: {
895    dMY_CXT;
896    return sv_bless(sv_2mortal(newRV_noinc(newSViv(PTR2IV(PL_op)))),
897                    MY_CXT.b__op_stashes[vmg_opclass(PL_op)]);
898   }
899   default:
900    break;
901  }
902
903  return &PL_sv_undef;
904 }
905
906 /* ... svt callbacks ....................................................... */
907
908 #define VMG_CB_CALL_SET_RET(D) \
909  {            \
910   SV *svr;    \
911   SPAGAIN;    \
912   svr = POPs; \
913   ret = SvOK(svr) ? SvIV(svr) : (D); \
914   PUTBACK;    \
915  }
916
917 #define VMG_CB_CALL_ARGS_MASK  15
918 #define VMG_CB_CALL_ARGS_SHIFT 4
919 #define VMG_CB_CALL_OPINFO     (VMG_OP_INFO_NAME|VMG_OP_INFO_OBJECT)
920
921 STATIC int vmg_cb_call(pTHX_ SV *cb, unsigned int flags, SV *sv, ...) {
922  va_list ap;
923  int ret;
924  unsigned int i, args, opinfo;
925
926  dSP;
927
928  args    = flags & VMG_CB_CALL_ARGS_MASK;
929  flags >>= VMG_CB_CALL_ARGS_SHIFT;
930  opinfo  = flags & VMG_CB_CALL_OPINFO;
931
932  ENTER;
933  SAVETMPS;
934
935  PUSHMARK(SP);
936  EXTEND(SP, args + 1);
937  PUSHs(sv_2mortal(newRV_inc(sv)));
938  va_start(ap, sv);
939  for (i = 0; i < args; ++i) {
940   SV *sva = va_arg(ap, SV *);
941   PUSHs(sva ? sva : &PL_sv_undef);
942  }
943  va_end(ap);
944  if (opinfo)
945   XPUSHs(vmg_op_info(opinfo));
946  PUTBACK;
947
948  call_sv(cb, G_SCALAR);
949
950  VMG_CB_CALL_SET_RET(0);
951
952  FREETMPS;
953  LEAVE;
954
955  return ret;
956 }
957
958 #define vmg_cb_call1(I, F, S, A1) \
959         vmg_cb_call(aTHX_ (I), (((F) << VMG_CB_CALL_ARGS_SHIFT) | 1), (S), (A1))
960 #define vmg_cb_call2(I, F, S, A1, A2) \
961         vmg_cb_call(aTHX_ (I), (((F) << VMG_CB_CALL_ARGS_SHIFT) | 2), (S), (A1), (A2))
962 #define vmg_cb_call3(I, F, S, A1, A2, A3) \
963         vmg_cb_call(aTHX_ (I), (((F) << VMG_CB_CALL_ARGS_SHIFT) | 3), (S), (A1), (A2), (A3))
964
965 STATIC int vmg_svt_get(pTHX_ SV *sv, MAGIC *mg) {
966  const MGWIZ *w = vmg_wizard_mgwiz(mg->mg_ptr);
967  return vmg_cb_call1(w->cb_get, w->opinfo, sv, mg->mg_obj);
968 }
969
970 STATIC int vmg_svt_set(pTHX_ SV *sv, MAGIC *mg) {
971  const MGWIZ *w = vmg_wizard_mgwiz(mg->mg_ptr);
972  return vmg_cb_call1(w->cb_set, w->opinfo, sv, mg->mg_obj);
973 }
974
975 STATIC U32 vmg_svt_len(pTHX_ SV *sv, MAGIC *mg) {
976  const MGWIZ *w = vmg_wizard_mgwiz(mg->mg_ptr);
977  unsigned int opinfo = w->opinfo;
978  U32 len, ret;
979  svtype t = SvTYPE(sv);
980
981  dSP;
982
983  ENTER;
984  SAVETMPS;
985
986  PUSHMARK(SP);
987  EXTEND(SP, 3);
988  PUSHs(sv_2mortal(newRV_inc(sv)));
989  PUSHs(mg->mg_obj ? mg->mg_obj : &PL_sv_undef);
990  if (t < SVt_PVAV) {
991   STRLEN l;
992   const U8 *s = (const U8 *) SvPV_const(sv, l);
993   if (DO_UTF8(sv))
994    len = utf8_length(s, s + l);
995   else
996    len = l;
997   mPUSHu(len);
998  } else if (t == SVt_PVAV) {
999   len = av_len((AV *) sv) + 1;
1000   mPUSHu(len);
1001  } else {
1002   len = 0;
1003   PUSHs(&PL_sv_undef);
1004  }
1005  if (opinfo)
1006   XPUSHs(vmg_op_info(opinfo));
1007  PUTBACK;
1008
1009  call_sv(w->cb_len, G_SCALAR);
1010
1011  VMG_CB_CALL_SET_RET(len);
1012
1013  FREETMPS;
1014  LEAVE;
1015
1016  return t == SVt_PVAV ? ret - 1 : ret;
1017 }
1018
1019 STATIC int vmg_svt_clear(pTHX_ SV *sv, MAGIC *mg) {
1020  const MGWIZ *w = vmg_wizard_mgwiz(mg->mg_ptr);
1021  return vmg_cb_call1(w->cb_clear, w->opinfo, sv, mg->mg_obj);
1022 }
1023
1024 STATIC int vmg_svt_free(pTHX_ SV *sv, MAGIC *mg) {
1025  const MGWIZ *w;
1026 #if VMG_HAS_PERL(5, 9, 5)
1027  PERL_CONTEXT saved_cx;
1028  I32 cxix;
1029 #endif
1030  unsigned int had_err, has_err, flags = G_SCALAR | G_EVAL;
1031  int ret = 0;
1032
1033  dSP;
1034
1035  /* Don't even bother if we are in global destruction - the wizard is prisoner
1036   * of circular references and we are way beyond user realm */
1037  if (PL_dirty)
1038   return 0;
1039
1040  w = vmg_wizard_mgwiz(mg->mg_ptr);
1041
1042  /* So that it survives the temp cleanup below */
1043  SvREFCNT_inc(sv);
1044
1045 #if !VMG_HAS_PERL_MAINT(5, 11, 0, 32686)
1046  /* The previous magic tokens were freed but the magic chain wasn't updated, so
1047   * if you access the sv from the callback the old deleted magics will trigger
1048   * and cause memory misreads. Change 32686 solved it that way : */
1049  SvMAGIC_set(sv, mg);
1050 #endif
1051
1052  ENTER;
1053  SAVETMPS;
1054
1055  PUSHMARK(SP);
1056  EXTEND(SP, 2);
1057  PUSHs(sv_2mortal(newRV_inc(sv)));
1058  PUSHs(mg->mg_obj ? mg->mg_obj : &PL_sv_undef);
1059  if (w->opinfo)
1060   XPUSHs(vmg_op_info(w->opinfo));
1061  PUTBACK;
1062
1063  had_err = SvTRUE(ERRSV);
1064  if (had_err)
1065   flags |= G_KEEPERR;
1066
1067 #if VMG_HAS_PERL(5, 9, 5)
1068  /* This context should not be used anymore, but since we croak in places the
1069   * core doesn't even dare to, some pointers to it may remain in the upper call
1070   * stack. Make sure call_sv() doesn't clobber it. */
1071  if (cxstack_ix < cxstack_max)
1072   cxix = cxstack_ix + 1;
1073  else
1074   cxix = Perl_cxinc(aTHX);
1075  saved_cx = cxstack[cxix];
1076 #endif
1077
1078  call_sv(w->cb_free, flags);
1079
1080 #if VMG_HAS_PERL(5, 9, 5)
1081  cxstack[cxix] = saved_cx;
1082 #endif
1083
1084  has_err = SvTRUE(ERRSV);
1085  if (IN_PERL_COMPILETIME && !had_err && has_err)
1086   ++PL_error_count;
1087
1088  VMG_CB_CALL_SET_RET(0);
1089
1090  FREETMPS;
1091  LEAVE;
1092
1093  /* Calling SvREFCNT_dec() will trigger destructors in an infinite loop, so
1094   * we have to rely on SvREFCNT() being a lvalue. Heck, even the core does it */
1095  --SvREFCNT(sv);
1096
1097  /* Perl_mg_free will get rid of the magic and decrement mg->mg_obj and
1098   * mg->mg_ptr reference count */
1099  return ret;
1100 }
1101
1102 #if MGf_COPY
1103 STATIC int vmg_svt_copy(pTHX_ SV *sv, MAGIC *mg, SV *nsv, const char *key,
1104 # if VMG_HAS_PERL_MAINT(5, 11, 0, 33256)
1105   I32 keylen
1106 # else
1107   int keylen
1108 # endif
1109  ) {
1110  SV *keysv;
1111  const MGWIZ *w = vmg_wizard_mgwiz(mg->mg_ptr);
1112  int ret;
1113
1114  if (keylen == HEf_SVKEY) {
1115   keysv = (SV *) key;
1116  } else {
1117   keysv = newSVpvn(key, keylen);
1118  }
1119
1120  ret = vmg_cb_call3(w->cb_copy, w->opinfo, sv, mg->mg_obj, keysv, nsv);
1121
1122  if (keylen != HEf_SVKEY) {
1123   SvREFCNT_dec(keysv);
1124  }
1125
1126  return ret;
1127 }
1128 #endif /* MGf_COPY */
1129
1130 #if 0 /*  MGf_DUP */
1131 STATIC int vmg_svt_dup(pTHX_ MAGIC *mg, CLONE_PARAMS *param) {
1132  return 0;
1133 }
1134 #endif /* MGf_DUP */
1135
1136 #if MGf_LOCAL
1137 STATIC int vmg_svt_local(pTHX_ SV *nsv, MAGIC *mg) {
1138  const MGWIZ *w = vmg_wizard_mgwiz(mg->mg_ptr);
1139  return vmg_cb_call1(w->cb_local, w->opinfo, nsv, mg->mg_obj);
1140 }
1141 #endif /* MGf_LOCAL */
1142
1143 #if VMG_UVAR
1144 STATIC OP *vmg_pp_resetuvar(pTHX) {
1145  SvRMAGICAL_on(cSVOP_sv);
1146  return NORMAL;
1147 }
1148
1149 STATIC I32 vmg_svt_val(pTHX_ IV action, SV *sv) {
1150  struct ufuncs *uf;
1151  MAGIC *mg, *umg;
1152  SV *key = NULL, *newkey = NULL;
1153  int tied = 0;
1154
1155  umg = mg_find(sv, PERL_MAGIC_uvar);
1156  /* umg can't be NULL or we wouldn't be there. */
1157  key = umg->mg_obj;
1158  uf  = (struct ufuncs *) umg->mg_ptr;
1159
1160  if (uf[1].uf_val)
1161   uf[1].uf_val(aTHX_ action, sv);
1162  if (uf[1].uf_set)
1163   uf[1].uf_set(aTHX_ action, sv);
1164
1165  action &= HV_FETCH_ISSTORE | HV_FETCH_ISEXISTS | HV_FETCH_LVALUE | HV_DELETE;
1166  for (mg = SvMAGIC(sv); mg; mg = mg->mg_moremagic) {
1167   const MGWIZ *w;
1168   switch (mg->mg_type) {
1169    case PERL_MAGIC_ext:
1170     break;
1171    case PERL_MAGIC_tied:
1172     ++tied;
1173     continue;
1174    default:
1175     continue;
1176   }
1177   if (mg->mg_private != SIG_WIZ) continue;
1178   w = vmg_wizard_mgwiz(mg->mg_ptr);
1179   switch (w->uvar) {
1180    case 0:
1181     continue;
1182    case 2:
1183     if (!newkey)
1184      newkey = key = umg->mg_obj = sv_mortalcopy(umg->mg_obj);
1185   }
1186   switch (action) {
1187    case 0:
1188     if (w->cb_fetch)
1189      vmg_cb_call2(w->cb_fetch, w->opinfo, sv, mg->mg_obj, key);
1190     break;
1191    case HV_FETCH_ISSTORE:
1192    case HV_FETCH_LVALUE:
1193    case (HV_FETCH_ISSTORE|HV_FETCH_LVALUE):
1194     if (w->cb_store)
1195      vmg_cb_call2(w->cb_store, w->opinfo, sv, mg->mg_obj, key);
1196     break;
1197    case HV_FETCH_ISEXISTS:
1198     if (w->cb_exists)
1199      vmg_cb_call2(w->cb_exists, w->opinfo, sv, mg->mg_obj, key);
1200     break;
1201    case HV_DELETE:
1202     if (w->cb_delete)
1203      vmg_cb_call2(w->cb_delete, w->opinfo, sv, mg->mg_obj, key);
1204     break;
1205   }
1206  }
1207
1208  if (SvRMAGICAL(sv) && !tied) {
1209   /* Temporarily hide the RMAGICAL flag of the hash so it isn't wrongly
1210    * mistaken for a tied hash by the rest of hv_common. It will be reset by
1211    * the op_ppaddr of a new fake op injected between the current and the next
1212    * one. */
1213   OP *o = PL_op;
1214   if (!o->op_next || o->op_next->op_ppaddr != vmg_pp_resetuvar) {
1215    SVOP *svop;
1216    NewOp(1101, svop, 1, SVOP);
1217    svop->op_type   = OP_STUB;
1218    svop->op_ppaddr = vmg_pp_resetuvar;
1219    svop->op_next   = o->op_next;
1220    svop->op_flags  = 0;
1221    svop->op_sv     = sv;
1222    o->op_next      = (OP *) svop;
1223   }
1224   SvRMAGICAL_off(sv);
1225  }
1226
1227  return 0;
1228 }
1229 #endif /* VMG_UVAR */
1230
1231 /* --- Macros for the XS section ------------------------------------------- */
1232
1233 #define VMG_SET_CB(S, N)              \
1234  cb = (S);                            \
1235  w->cb_ ## N = (SvOK(cb) && SvROK(cb)) ? newRV_inc(SvRV(cb)) : NULL;
1236
1237 #define VMG_SET_SVT_CB(S, N)          \
1238  cb = (S);                            \
1239  if (SvOK(cb) && SvROK(cb)) {         \
1240   t->svt_ ## N = vmg_svt_ ## N;       \
1241   w->cb_  ## N = newRV_inc(SvRV(cb)); \
1242  } else {                             \
1243   t->svt_ ## N = NULL;                \
1244   w->cb_  ## N = NULL;                \
1245  }
1246
1247 #if VMG_THREADSAFE
1248
1249 STATIC void vmg_cleanup(pTHX_ void *ud) {
1250  dMY_CXT;
1251
1252  ptable_free(MY_CXT.wizards);
1253  MY_CXT.wizards = NULL;
1254 }
1255
1256 #endif /* VMG_THREADSAFE */
1257
1258 /* --- XS ------------------------------------------------------------------ */
1259
1260 MODULE = Variable::Magic            PACKAGE = Variable::Magic
1261
1262 PROTOTYPES: ENABLE
1263
1264 BOOT:
1265 {
1266  HV *stash;
1267
1268  MY_CXT_INIT;
1269 #if VMG_THREADSAFE
1270  MY_CXT.wizards = ptable_new();
1271  MY_CXT.owner   = aTHX;
1272 #endif
1273  MY_CXT.b__op_stashes[0] = NULL;
1274 #if VMG_THREADSAFE
1275  MUTEX_INIT(&vmg_op_name_init_mutex);
1276  call_atexit(vmg_cleanup, NULL);
1277 #endif
1278
1279  stash = gv_stashpv(__PACKAGE__, 1);
1280  newCONSTSUB(stash, "MGf_COPY",  newSVuv(MGf_COPY));
1281  newCONSTSUB(stash, "MGf_DUP",   newSVuv(MGf_DUP));
1282  newCONSTSUB(stash, "MGf_LOCAL", newSVuv(MGf_LOCAL));
1283  newCONSTSUB(stash, "VMG_UVAR",  newSVuv(VMG_UVAR));
1284  newCONSTSUB(stash, "VMG_COMPAT_ARRAY_PUSH_NOLEN",
1285                     newSVuv(VMG_COMPAT_ARRAY_PUSH_NOLEN));
1286  newCONSTSUB(stash, "VMG_COMPAT_ARRAY_PUSH_NOLEN_VOID",
1287                     newSVuv(VMG_COMPAT_ARRAY_PUSH_NOLEN_VOID));
1288  newCONSTSUB(stash, "VMG_COMPAT_ARRAY_UNSHIFT_NOLEN_VOID",
1289                     newSVuv(VMG_COMPAT_ARRAY_UNSHIFT_NOLEN_VOID));
1290  newCONSTSUB(stash, "VMG_COMPAT_ARRAY_UNDEF_CLEAR",
1291                     newSVuv(VMG_COMPAT_ARRAY_UNDEF_CLEAR));
1292  newCONSTSUB(stash, "VMG_COMPAT_SCALAR_LENGTH_NOLEN",
1293                     newSVuv(VMG_COMPAT_SCALAR_LENGTH_NOLEN));
1294  newCONSTSUB(stash, "VMG_PERL_PATCHLEVEL", newSVuv(VMG_PERL_PATCHLEVEL));
1295  newCONSTSUB(stash, "VMG_THREADSAFE",      newSVuv(VMG_THREADSAFE));
1296  newCONSTSUB(stash, "VMG_FORKSAFE",        newSVuv(VMG_FORKSAFE));
1297  newCONSTSUB(stash, "VMG_OP_INFO_NAME",    newSVuv(VMG_OP_INFO_NAME));
1298  newCONSTSUB(stash, "VMG_OP_INFO_OBJECT",  newSVuv(VMG_OP_INFO_OBJECT));
1299 }
1300
1301 #if VMG_THREADSAFE
1302
1303 void
1304 CLONE(...)
1305 PROTOTYPE: DISABLE
1306 PREINIT:
1307  ptable *t;
1308  U32     had_b__op_stash = 0;
1309  opclass c;
1310 PPCODE:
1311  {
1312   my_cxt_t ud;
1313   dMY_CXT;
1314
1315   ud.wizards = t = ptable_new();
1316   ud.owner   = MY_CXT.owner;
1317   ptable_walk(MY_CXT.wizards, vmg_ptable_clone, &ud);
1318
1319   for (c = 0; c < OPc_MAX; ++c) {
1320    if (MY_CXT.b__op_stashes[c])
1321     had_b__op_stash |= (((U32) 1) << c);
1322   }
1323  }
1324  {
1325   MY_CXT_CLONE;
1326   MY_CXT.wizards = t;
1327   MY_CXT.owner   = aTHX;
1328   for (c = 0; c < OPc_MAX; ++c) {
1329    MY_CXT.b__op_stashes[c] = (had_b__op_stash & (((U32) 1) << c))
1330                               ? gv_stashpv(vmg_opclassnames[c], 1) : NULL;
1331   }
1332  }
1333  XSRETURN(0);
1334
1335 #endif /* VMG_THREADSAFE */
1336
1337 SV *_wizard(...)
1338 PROTOTYPE: DISABLE
1339 PREINIT:
1340  I32 i = 0;
1341  MGWIZ *w;
1342  MGVTBL *t;
1343  SV *cb;
1344 CODE:
1345  dMY_CXT;
1346
1347  if (items != 7
1348 #if MGf_COPY
1349               + 1
1350 #endif /* MGf_COPY */
1351 #if MGf_DUP
1352               + 1
1353 #endif /* MGf_DUP */
1354 #if MGf_LOCAL
1355               + 1
1356 #endif /* MGf_LOCAL */
1357 #if VMG_UVAR
1358               + 5
1359 #endif /* VMG_UVAR */
1360               ) { croak(vmg_wrongargnum); }
1361
1362  Newx(t, 1, MGVTBL);
1363  Newx(w, 1, MGWIZ);
1364
1365  VMG_SET_CB(ST(i++), data);
1366  cb = ST(i++);
1367  w->opinfo = SvOK(cb) ? SvUV(cb) : 0;
1368  if (w->opinfo)
1369   vmg_op_info_init(w->opinfo);
1370  VMG_SET_SVT_CB(ST(i++), get);
1371  VMG_SET_SVT_CB(ST(i++), set);
1372  VMG_SET_SVT_CB(ST(i++), len);
1373  VMG_SET_SVT_CB(ST(i++), clear);
1374  VMG_SET_SVT_CB(ST(i++), free);
1375 #if MGf_COPY
1376  VMG_SET_SVT_CB(ST(i++), copy);
1377 #endif /* MGf_COPY */
1378 #if MGf_DUP
1379  /* VMG_SET_SVT_CB(ST(i++), dup); */
1380  i++;
1381  t->svt_dup = NULL;
1382  w->cb_dup  = NULL;
1383 #endif /* MGf_DUP */
1384 #if MGf_LOCAL
1385  VMG_SET_SVT_CB(ST(i++), local);
1386 #endif /* MGf_LOCAL */
1387 #if VMG_UVAR
1388  VMG_SET_CB(ST(i++), fetch);
1389  VMG_SET_CB(ST(i++), store);
1390  VMG_SET_CB(ST(i++), exists);
1391  VMG_SET_CB(ST(i++), delete);
1392  cb = ST(i++);
1393  if (w->cb_fetch || w->cb_store || w->cb_exists || w->cb_delete)
1394   w->uvar = SvTRUE(cb) ? 2 : 1;
1395  else
1396   w->uvar = 0;
1397 #endif /* VMG_UVAR */
1398 #if VMG_MULTIPLICITY
1399  w->owner = aTHX;
1400 #endif /* VMG_MULTIPLICITY */
1401  w->vtbl  = t;
1402 #if VMG_THREADSAFE
1403  ptable_store(MY_CXT.wizards, w, w);
1404 #endif /* VMG_THREADSAFE */
1405
1406  RETVAL = newRV_noinc(vmg_wizard_new(w));
1407 OUTPUT:
1408  RETVAL
1409
1410 SV *cast(SV *sv, SV *wiz, ...)
1411 PROTOTYPE: \[$@%&*]$@
1412 PREINIT:
1413  AV *args = NULL;
1414  SV *ret;
1415 CODE:
1416  if (items > 2) {
1417   I32 i;
1418   args = newAV();
1419   av_fill(args, items - 2);
1420   for (i = 2; i < items; ++i) {
1421    SV *arg = ST(i);
1422    SvREFCNT_inc(arg);
1423    if (av_store(args, i - 2, arg) == NULL) croak(vmg_argstorefailed);
1424   }
1425  }
1426  ret = newSVuv(vmg_cast(SvRV(sv), vmg_wizard_validate(wiz), args));
1427  SvREFCNT_dec(args);
1428  RETVAL = ret;
1429 OUTPUT:
1430  RETVAL
1431
1432 void
1433 getdata(SV *sv, SV *wiz)
1434 PROTOTYPE: \[$@%&*]$
1435 PREINIT:
1436  SV *data;
1437 PPCODE:
1438  data = vmg_data_get(SvRV(sv), vmg_wizard_validate(wiz));
1439  if (!data)
1440   XSRETURN_EMPTY;
1441  ST(0) = data;
1442  XSRETURN(1);
1443
1444 SV *dispell(SV *sv, SV *wiz)
1445 PROTOTYPE: \[$@%&*]$
1446 CODE:
1447  RETVAL = newSVuv(vmg_dispell(SvRV(sv), vmg_wizard_validate(wiz)));
1448 OUTPUT:
1449  RETVAL