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 #ifndef VMG_MULTIPLICITY
  40 # if defined(MULTIPLICITY) || defined(PERL_IMPLICIT_CONTEXT)
  41 #  define VMG_MULTIPLICITY 1
  42 # else
  43 #  define VMG_MULTIPLICITY 0
  44 # endif
  45 #endif
  46 #if VMG_MULTIPLICITY && !defined(tTHX)
  47 # define tTHX PerlInterpreter*
  48 #endif
  49
  50 #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))
  51 # define VMG_THREADSAFE 1
  52 # ifndef MY_CXT_CLONE
  53 #  define MY_CXT_CLONE \
  54     dMY_CXT_SV;                                                      \
  55     my_cxt_t *my_cxtp = (my_cxt_t*)SvPVX(newSV(sizeof(my_cxt_t)-1)); \
  56     Copy(INT2PTR(my_cxt_t*, SvUV(my_cxt_sv)), my_cxtp, 1, my_cxt_t); \
  57     sv_setuv(my_cxt_sv, PTR2UV(my_cxtp))
  58 # endif
  59 #else
  60 # define VMG_THREADSAFE 0
  61 # undef  dMY_CXT
  62 # define dMY_CXT      dNOOP
  63 # undef  MY_CXT
  64 # define MY_CXT       vmg_globaldata
  65 # undef  START_MY_CXT
  66 # define START_MY_CXT STATIC my_cxt_t MY_CXT;
  67 # undef  MY_CXT_INIT
  68 # define MY_CXT_INIT  NOOP
  69 # undef  MY_CXT_CLONE
  70 # define MY_CXT_CLONE NOOP
  71 #endif
  72
  73 #if VMG_THREADSAFE
  74
  75 STATIC SV *vmg_clone(pTHX_ SV *sv, tTHX owner) {
  76 #define vmg_clone(P, O) vmg_clone(aTHX_ (P), (O))
  77  CLONE_PARAMS param;
  78  param.stashes    = NULL; /* don't need it unless sv is a PVHV */
  79  param.flags      = 0;
  80  param.proto_perl = owner;
  81  return sv_dup(sv, &param);
  82 }
  83
  84 #endif /* VMG_THREADSAFE */
  85
  86 /* --- Compatibility ------------------------------------------------------- */
  87
  88 #ifndef Newx
  89 # define Newx(v, n, c) New(0, v, n, c)
  90 #endif
  91
  92 #ifndef SvMAGIC_set
  93 # define SvMAGIC_set(sv, val) (SvMAGIC(sv) = (val))
  94 #endif
  95
  96 #ifndef mPUSHi
  97 # define mPUSHi(I) PUSHs(sv_2mortal(newSViv(I)))
  98 #endif
  99
 100 #ifndef PERL_MAGIC_ext
 101 # define PERL_MAGIC_ext '~'
 102 #endif
 103
 104 #ifndef MGf_COPY
 105 # define MGf_COPY 0
 106 #endif
 107
 108 #ifndef MGf_DUP
 109 # define MGf_DUP 0
 110 #endif
 111
 112 #ifndef MGf_LOCAL
 113 # define MGf_LOCAL 0
 114 #endif
 115
 116 /* uvar magic and Hash::Util::FieldHash were commited with p28419 */
 117 #if VMG_HAS_PERL_MAINT(5, 9, 4, 28419) || VMG_HAS_PERL(5, 10, 0)
 118 # define VMG_UVAR 1
 119 #else
 120 # define VMG_UVAR 0
 121 #endif
 122
 123 #if !defined(VMG_COMPAT_ARRAY_PUSH_NOLEN) && (VMG_HAS_PERL_BRANCH(5, 8, 9) || VMG_HAS_PERL_MAINT(5, 9, 3, 25854) || VMG_HAS_PERL(5, 10, 0))
 124 # define VMG_COMPAT_ARRAY_PUSH_NOLEN 1
 125 #else
 126 # define VMG_COMPAT_ARRAY_PUSH_NOLEN 0
 127 #endif
 128
 129 /* since 5.9.5 - see #43357 */
 130 #if VMG_HAS_PERL_BRANCH(5, 8, 9) || VMG_HAS_PERL_MAINT(5, 9, 5, 31473) || VMG_HAS_PERL(5, 10, 0)
 131 # define VMG_COMPAT_ARRAY_UNDEF_CLEAR 1
 132 #else
 133 # define VMG_COMPAT_ARRAY_UNDEF_CLEAR 0
 134 #endif
 135
 136 #if VMG_HAS_PERL_MAINT(5, 11, 0, 32969)
 137 # define VMG_COMPAT_SCALAR_LENGTH_NOLEN 1
 138 #else
 139 # define VMG_COMPAT_SCALAR_LENGTH_NOLEN 0
 140 #endif
 141
 142 #if VMG_UVAR
 143
 144 /* Bug-free mg_magical - see http://www.xray.mpe.mpg.de/mailing-lists/perl5-porters/2008-01/msg00036.html - but specialized to our needs. */
 145 STATIC void vmg_sv_magicuvar(pTHX_ SV *sv, const char *uf, I32 len) {
 146 #define vmg_sv_magicuvar(S, U, L) vmg_sv_magicuvar(aTHX_ (S), (U), (L))
 147  const MAGIC* mg;
 148  sv_magic(sv, NULL, PERL_MAGIC_uvar, uf, len);
 149  /* uvar magic has set and get magic, hence this has set SVs_GMG and SVs_SMG. */
 150  PERL_UNUSED_CONTEXT;
 151  if ((mg = SvMAGIC(sv))) {
 152   SvRMAGICAL_off(sv);
 153   do {
 154    const MGVTBL* const vtbl = mg->mg_virtual;
 155    if (vtbl) {
 156     if (vtbl->svt_clear) {
 157      SvRMAGICAL_on(sv);
 158      break;
 159     }
 160    }
 161   } while ((mg = mg->mg_moremagic));
 162  }
 163 }
 164
 165 #endif /* VMG_UVAR */
 166
 167 /* --- Context-safe global data -------------------------------------------- */
 168
 169 #define MY_CXT_KEY __PACKAGE__ "::_guts" XS_VERSION
 170
 171 typedef HV * my_cxt_t;
 172
 173 START_MY_CXT
 174
 175 /* --- Signatures ---------------------------------------------------------- */
 176
 177 #define SIG_MIN ((U16) (1u << 8))
 178 #define SIG_MAX ((U16) ((1u << 16) - 1))
 179 #define SIG_NBR (SIG_MAX - SIG_MIN + 1)
 180 #define SIG_WIZ ((U16) ((1u << 8) - 1))
 181
 182 /* ... Generate signatures ................................................. */
 183
 184 STATIC U16 vmg_gensig(pTHX) {
 185 #define vmg_gensig() vmg_gensig(aTHX)
 186  U16 sig;
 187  char buf[8];
 188  dMY_CXT;
 189
 190  do {
 191   sig = SIG_NBR * Drand01() + SIG_MIN;
 192  } while (hv_exists(MY_CXT, buf, sprintf(buf, "%u", sig)));
 193
 194  return sig;
 195 }
 196
 197 /* --- MGWIZ structure ----------------------------------------------------- */
 198
 199 typedef struct {
 200  MGVTBL *vtbl;
 201  U16 sig;
 202  U16 uvar;
 203  SV *cb_data;
 204  SV *cb_get, *cb_set, *cb_len, *cb_clear, *cb_free;
 205 #if MGf_COPY
 206  SV *cb_copy;
 207 #endif /* MGf_COPY */
 208 #if MGf_DUP
 209  SV *cb_dup;
 210 #endif /* MGf_DUP */
 211 #if MGf_LOCAL
 212  SV *cb_local;
 213 #endif /* MGf_LOCAL */
 214 #if VMG_UVAR
 215  SV *cb_fetch, *cb_store, *cb_exists, *cb_delete;
 216 #endif /* VMG_UVAR */
 217 #if VMG_MULTIPLICITY
 218  tTHX owner;
 219 #endif /* VMG_MULTIPLICITY */
 220 } MGWIZ;
 221
 222 #define MGWIZ2SV(W) (newSVuv(PTR2UV(W)))
 223 #define SV2MGWIZ(S) (INT2PTR(MGWIZ*, SvUVX((SV *) (S))))
 224
 225 /* ... Construct private data .............................................. */
 226
 227 STATIC SV *vmg_data_new(pTHX_ SV *ctor, SV *sv, AV *args) {
 228 #define vmg_data_new(C, S, A) vmg_data_new(aTHX_ (C), (S), (A))
 229  SV *nsv;
 230  I32 i, alen = (args == NULL) ? 0 : av_len(args);
 231
 232  dSP;
 233  int count;
 234
 235  ENTER;
 236  SAVETMPS;
 237
 238  PUSHMARK(SP);
 239  EXTEND(SP, alen + 1);
 240  PUSHs(sv_2mortal(newRV_inc(sv)));
 241  for (i = 0; i < alen; ++i)
 242   PUSHs(*av_fetch(args, i, 0));
 243  PUTBACK;
 244
 245  count = call_sv(ctor, G_SCALAR);
 246
 247  SPAGAIN;
 248
 249  if (count != 1) { croak("Callback needs to return 1 scalar\n"); }
 250  nsv = POPs;
 251 #if !VMG_HAS_PERL(5, 8, 2)
 252  nsv = sv_newref(nsv); /* Workaround some bug in SvREFCNT_inc() */
 253 #else
 254  SvREFCNT_inc(nsv);    /* Or it will be destroyed in FREETMPS */
 255 #endif
 256
 257  PUTBACK;
 258
 259  FREETMPS;
 260  LEAVE;
 261
 262  return nsv;
 263 }
 264
 265 STATIC SV *vmg_data_get(SV *sv, U16 sig) {
 266  MAGIC *mg, *moremagic;
 267
 268  if (SvTYPE(sv) >= SVt_PVMG) {
 269   for (mg = SvMAGIC(sv); mg; mg = moremagic) {
 270    moremagic = mg->mg_moremagic;
 271    if ((mg->mg_type == PERL_MAGIC_ext) && (mg->mg_private == sig)) { break; }
 272   }
 273   if (mg) { return mg->mg_obj; }
 274  }
 275
 276  return NULL;
 277 }
 278
 279 /* ... Magic cast/dispell .................................................. */
 280
 281 #if VMG_UVAR
 282 STATIC I32 vmg_svt_val(pTHX_ IV, SV *);
 283
 284 STATIC void vmg_uvar_del(SV *sv, MAGIC *prevmagic, MAGIC *mg, MAGIC *moremagic) {
 285  if (prevmagic) {
 286   prevmagic->mg_moremagic = moremagic;
 287  } else {
 288   SvMAGIC_set(sv, moremagic);
 289  }
 290  mg->mg_moremagic = NULL;
 291  Safefree(mg->mg_ptr);
 292  Safefree(mg);
 293 }
 294 #endif /* VMG_UVAR */
 295
 296 STATIC UV vmg_cast(pTHX_ SV *sv, SV *wiz, AV *args) {
 297 #define vmg_cast(S, W, A) vmg_cast(aTHX_ (S), (W), (A))
 298  MAGIC *mg = NULL, *moremagic = NULL;
 299  MGWIZ *w;
 300  SV *data;
 301
 302  w = SV2MGWIZ(wiz);
 303
 304  if (SvTYPE(sv) >= SVt_PVMG) {
 305   for (mg = SvMAGIC(sv); mg; mg = moremagic) {
 306    moremagic = mg->mg_moremagic;
 307    if ((mg->mg_type == PERL_MAGIC_ext) && (mg->mg_private == w->sig)) { break; }
 308   }
 309   if (mg) { return 1; }
 310  }
 311
 312  data = (w->cb_data) ? vmg_data_new(w->cb_data, sv, args) : NULL;
 313  mg = sv_magicext(sv, data, PERL_MAGIC_ext, w->vtbl, (const char *) wiz, HEf_SVKEY);
 314  mg->mg_private = w->sig;
 315 #if MGf_COPY
 316  if (w->cb_copy)
 317   mg->mg_flags |= MGf_COPY;
 318 #endif /* MGf_COPY */
 319 #if 0 /* MGf_DUP */
 320  if (w->cb_dup)
 321   mg->mg_flags |= MGf_DUP;
 322 #endif /* MGf_DUP */
 323 #if MGf_LOCAL
 324  if (w->cb_local)
 325   mg->mg_flags |= MGf_LOCAL;
 326 #endif /* MGf_LOCAL */
 327
 328 #if VMG_UVAR
 329  if (w->uvar && SvTYPE(sv) >= SVt_PVHV) {
 330   MAGIC *prevmagic;
 331   int add_uvar = 1;
 332   struct ufuncs uf[2];
 333
 334   uf[0].uf_val   = vmg_svt_val;
 335   uf[0].uf_set   = NULL;
 336   uf[0].uf_index = 0;
 337   uf[1].uf_val   = NULL;
 338   uf[1].uf_set   = NULL;
 339   uf[1].uf_index = 0;
 340
 341   /* One uvar magic in the chain is enough. */
 342   for (prevmagic = NULL, mg = SvMAGIC(sv); mg; prevmagic = mg, mg = moremagic) {
 343    moremagic = mg->mg_moremagic;
 344    if (mg->mg_type == PERL_MAGIC_uvar) { break; }
 345   }
 346
 347   if (mg) { /* Found another uvar magic. */
 348    struct ufuncs *olduf = (struct ufuncs *) mg->mg_ptr;
 349    if (olduf->uf_val == vmg_svt_val) {
 350     /* It's our uvar magic, nothing to do. */
 351     add_uvar = 0;
 352    } else {
 353     /* It's another uvar magic, backup it and replace it by ours. */
 354     uf[1] = *olduf;
 355     vmg_uvar_del(sv, prevmagic, mg, moremagic);
 356    }
 357   }
 358
 359   if (add_uvar) {
 360    vmg_sv_magicuvar(sv, (const char *) &uf, sizeof(uf));
 361   }
 362
 363  }
 364 #endif /* VMG_UVAR */
 365
 366  return 1;
 367 }
 368
 369 STATIC UV vmg_dispell(pTHX_ SV *sv, U16 sig) {
 370 #define vmg_dispell(S, Z) vmg_dispell(aTHX_ (S), (Z))
 371 #if VMG_UVAR
 372  U32 uvars = 0;
 373 #endif /* VMG_UVAR */
 374  MAGIC *mg, *prevmagic, *moremagic = NULL;
 375
 376  if (SvTYPE(sv) < SVt_PVMG) { return 0; }
 377
 378  for (prevmagic = NULL, mg = SvMAGIC(sv); mg; prevmagic = mg, mg = moremagic) {
 379   moremagic = mg->mg_moremagic;
 380   if (mg->mg_type == PERL_MAGIC_ext) {
 381    if (mg->mg_private == sig) {
 382 #if VMG_UVAR
 383     /* If the current has no uvar, short-circuit uvar deletion. */
 384     uvars = (SV2MGWIZ(mg->mg_ptr)->uvar) ? (uvars + 1) : 0;
 385 #endif /* VMG_UVAR */
 386     break;
 387 #if VMG_UVAR
 388    } else if ((mg->mg_private >= SIG_MIN) &&
 389               (mg->mg_private <= SIG_MAX) &&
 390                SV2MGWIZ(mg->mg_ptr)->uvar) {
 391     ++uvars;
 392     /* We can't break here since we need to find the ext magic to delete. */
 393 #endif /* VMG_UVAR */
 394    }
 395   }
 396  }
 397  if (!mg) { return 0; }
 398
 399  if (prevmagic) {
 400   prevmagic->mg_moremagic = moremagic;
 401  } else {
 402   SvMAGIC_set(sv, moremagic);
 403  }
 404  mg->mg_moremagic = NULL;
 405
 406  if (mg->mg_obj != sv) { SvREFCNT_dec(mg->mg_obj); } /* Destroy private data */
 407  SvREFCNT_dec((SV *) mg->mg_ptr); /* Unreference the wizard */
 408  Safefree(mg);
 409
 410 #if VMG_UVAR
 411  if (uvars == 1 && SvTYPE(sv) >= SVt_PVHV) {
 412   /* mg was the first ext magic in the chain that had uvar */
 413
 414   for (mg = moremagic; mg; mg = mg->mg_moremagic) {
 415    if ((mg->mg_type == PERL_MAGIC_ext) &&
 416        (mg->mg_private >= SIG_MIN) &&
 417        (mg->mg_private <= SIG_MAX) &&
 418         SV2MGWIZ(mg->mg_ptr)->uvar) {
 419     ++uvars;
 420     break;
 421    }
 422   }
 423
 424   if (uvars == 1) {
 425    struct ufuncs *uf;
 426    for (prevmagic = NULL, mg = SvMAGIC(sv); mg; prevmagic = mg, mg = moremagic){
 427     moremagic = mg->mg_moremagic;
 428     if (mg->mg_type == PERL_MAGIC_uvar) { break; }
 429    }
 430    /* assert(mg); */
 431    uf = (struct ufuncs *) mg->mg_ptr;
 432    /* assert(uf->uf_val == vmg_svt_val); */
 433    if (uf[1].uf_val || uf[1].uf_set) {
 434     /* Revert the original uvar magic. */
 435     uf[0] = uf[1];
 436     Renew(uf, 1, struct ufuncs);
 437     mg->mg_ptr = (char *) uf;
 438     mg->mg_len = sizeof(struct ufuncs);
 439    } else {
 440     /* Remove the uvar magic. */
 441     vmg_uvar_del(sv, prevmagic, mg, moremagic);
 442    }
 443   }
 444  }
 445 #endif /* VMG_UVAR */
 446
 447  return 1;
 448 }
 449
 450 /* ... svt callbacks ....................................................... */
 451
 452 STATIC int vmg_cb_call(pTHX_ SV *cb, SV *sv, SV *data, unsigned int args, ...) {
 453  va_list ap;
 454  SV *svr;
 455  int ret;
 456  unsigned int i;
 457
 458  dSP;
 459  int count;
 460
 461  ENTER;
 462  SAVETMPS;
 463
 464  PUSHMARK(SP);
 465  EXTEND(SP, args + 2);
 466  PUSHs(sv_2mortal(newRV_inc(sv)));
 467  PUSHs(data ? data : &PL_sv_undef);
 468  va_start(ap, args);
 469  for (i = 0; i < args; ++i) {
 470   SV *sva = va_arg(ap, SV *);
 471   PUSHs(sva ? sva : &PL_sv_undef);
 472  }
 473  va_end(ap);
 474  PUTBACK;
 475
 476  count = call_sv(cb, G_SCALAR);
 477
 478  SPAGAIN;
 479
 480  if (count != 1) { croak("Callback needs to return 1 scalar\n"); }
 481  svr = POPs;
 482  ret = SvOK(svr) ? SvIV(svr) : 0;
 483
 484  PUTBACK;
 485
 486  FREETMPS;
 487  LEAVE;
 488
 489  return ret;
 490 }
 491
 492 #define vmg_cb_call1(I, S, D)         vmg_cb_call(aTHX_ (I), (S), (D), 0)
 493 #define vmg_cb_call2(I, S, D, S2)     vmg_cb_call(aTHX_ (I), (S), (D), 1, (S2))
 494 #define vmg_cb_call3(I, S, D, S2, S3) vmg_cb_call(aTHX_ (I), (S), (D), 2, (S2), (S3))
 495
 496 STATIC int vmg_svt_get(pTHX_ SV *sv, MAGIC *mg) {
 497  return vmg_cb_call1(SV2MGWIZ(mg->mg_ptr)->cb_get, sv, mg->mg_obj);
 498 }
 499
 500 STATIC int vmg_svt_set(pTHX_ SV *sv, MAGIC *mg) {
 501  return vmg_cb_call1(SV2MGWIZ(mg->mg_ptr)->cb_set, sv, mg->mg_obj);
 502 }
 503
 504 STATIC U32 vmg_svt_len(pTHX_ SV *sv, MAGIC *mg) {
 505  SV *svr;
 506  I32 len;
 507  U32 ret;
 508
 509  dSP;
 510  int count;
 511
 512  ENTER;
 513  SAVETMPS;
 514
 515  PUSHMARK(SP);
 516  EXTEND(SP, 3);
 517  PUSHs(sv_2mortal(newRV_inc(sv)));
 518  PUSHs(mg->mg_obj ? mg->mg_obj : &PL_sv_undef);
 519  if (SvTYPE(sv) == SVt_PVAV) {
 520   len = av_len((AV *) sv) + 1;
 521   mPUSHi(len);
 522  } else {
 523   len = 1;
 524   PUSHs(&PL_sv_undef);
 525  }
 526  PUTBACK;
 527
 528  count = call_sv(SV2MGWIZ(mg->mg_ptr)->cb_len, G_SCALAR);
 529
 530  SPAGAIN;
 531
 532  if (count != 1) { croak("Callback needs to return 1 scalar\n"); }
 533  svr = POPs;
 534  ret = SvOK(svr) ? SvUV(svr) : len;
 535
 536  PUTBACK;
 537
 538  FREETMPS;
 539  LEAVE;
 540
 541  return ret - 1;
 542 }
 543
 544 STATIC int vmg_svt_clear(pTHX_ SV *sv, MAGIC *mg) {
 545  return vmg_cb_call1(SV2MGWIZ(mg->mg_ptr)->cb_clear, sv, mg->mg_obj);
 546 }
 547
 548 STATIC int vmg_svt_free(pTHX_ SV *sv, MAGIC *mg) {
 549  /* So that it can survive tmp cleanup in vmg_cb_call */
 550  SvREFCNT_inc(sv);
 551 #if !VMG_HAS_PERL_MAINT(5, 11, 0, 32686)
 552  /* The previous magic tokens were freed but the magic chain wasn't updated, so
 553   * if you access the sv from the callback the old deleted magics will trigger
 554   * and cause memory misreads. Change 32686 solved it that way : */
 555  SvMAGIC_set(sv, mg);
 556 #endif
 557  /* Perl_mg_free will get rid of the magic and decrement mg->mg_obj and
 558   * mg->mg_ptr reference count */
 559  return vmg_cb_call1(SV2MGWIZ(mg->mg_ptr)->cb_free, sv, mg->mg_obj);
 560 }
 561
 562 #if MGf_COPY
 563 STATIC int vmg_svt_copy(pTHX_ SV *sv, MAGIC *mg, SV *nsv, const char *key,
 564 # if VMG_HAS_PERL_MAINT(5, 11, 0, 33256)
 565   I32 keylen
 566 # else
 567   int keylen
 568 # endif
 569  ) {
 570  SV *keysv;
 571  int ret;
 572
 573  if (keylen == HEf_SVKEY) {
 574   keysv = (SV *) key;
 575  } else {
 576   keysv = newSVpvn(key, keylen);
 577  }
 578
 579  ret = vmg_cb_call3(SV2MGWIZ(mg->mg_ptr)->cb_copy, sv, mg->mg_obj, keysv, nsv);
 580
 581  if (keylen != HEf_SVKEY) {
 582   SvREFCNT_dec(keysv);
 583  }
 584
 585  return ret;
 586 }
 587 #endif /* MGf_COPY */
 588
 589 #if 0 /*  MGf_DUP */
 590 STATIC int vmg_svt_dup(pTHX_ MAGIC *mg, CLONE_PARAMS *param) {
 591  return 0;
 592 }
 593 #endif /* MGf_DUP */
 594
 595 #if MGf_LOCAL
 596 STATIC int vmg_svt_local(pTHX_ SV *nsv, MAGIC *mg) {
 597  return vmg_cb_call1(SV2MGWIZ(mg->mg_ptr)->cb_local, nsv, mg->mg_obj);
 598 }
 599 #endif /* MGf_LOCAL */
 600
 601 #if VMG_UVAR
 602 STATIC I32 vmg_svt_val(pTHX_ IV action, SV *sv) {
 603  struct ufuncs *uf;
 604  MAGIC *mg;
 605  SV *key = NULL;
 606
 607  mg  = mg_find(sv, PERL_MAGIC_uvar);
 608  /* mg can't be NULL or we wouldn't be there. */
 609  key = mg->mg_obj;
 610  uf  = (struct ufuncs *) mg->mg_ptr;
 611
 612  if (uf[1].uf_val != NULL) { uf[1].uf_val(aTHX_ action, sv); }
 613  if (uf[1].uf_set != NULL) { uf[1].uf_set(aTHX_ action, sv); }
 614
 615  action &= HV_FETCH_ISSTORE | HV_FETCH_ISEXISTS | HV_FETCH_LVALUE | HV_DELETE;
 616  for (mg = SvMAGIC(sv); mg; mg = mg->mg_moremagic) {
 617   MGWIZ *w;
 618   if ((mg->mg_type != PERL_MAGIC_ext)
 619    || (mg->mg_private < SIG_MIN)
 620    || (mg->mg_private > SIG_MAX)) { continue; }
 621   w = SV2MGWIZ(mg->mg_ptr);
 622   if (!w->uvar) { continue; }
 623   switch (action) {
 624    case 0:
 625     if (w->cb_fetch)  { vmg_cb_call2(w->cb_fetch,  sv, mg->mg_obj, key); }
 626     break;
 627    case HV_FETCH_ISSTORE:
 628    case HV_FETCH_LVALUE:
 629    case (HV_FETCH_ISSTORE|HV_FETCH_LVALUE):
 630     if (w->cb_store)  { vmg_cb_call2(w->cb_store,  sv, mg->mg_obj, key); }
 631     break;
 632    case HV_FETCH_ISEXISTS:
 633     if (w->cb_exists) { vmg_cb_call2(w->cb_exists, sv, mg->mg_obj, key); }
 634     break;
 635    case HV_DELETE:
 636     if (w->cb_delete) { vmg_cb_call2(w->cb_delete, sv, mg->mg_obj, key); }
 637     break;
 638   }
 639  }
 640
 641  return 0;
 642 }
 643 #endif /* VMG_UVAR */
 644
 645 /* ... Wizard destructor ................................................... */
 646
 647 STATIC int vmg_wizard_free(pTHX_ SV *wiz, MAGIC *mg) {
 648  char buf[8];
 649  MGWIZ *w;
 650
 651  if (PL_dirty) /* during global destruction, the context is already freed */
 652   return 0;
 653
 654  w = SV2MGWIZ(wiz);
 655 #if VMG_MULTIPLICITY
 656  if (w->owner != aTHX)
 657   return 0;
 658  w->owner = NULL;
 659 #endif /* VMG_MULTIPLICITY */
 660
 661  {
 662   dMY_CXT;
 663   if (hv_delete(MY_CXT, buf, sprintf(buf, "%u", w->sig), 0) != wiz)
 664    return 0;
 665  }
 666  SvFLAGS(wiz) |= SVf_BREAK;
 667  FREETMPS;
 668
 669  if (w->cb_data  != NULL) { SvREFCNT_dec(SvRV(w->cb_data)); }
 670  if (w->cb_get   != NULL) { SvREFCNT_dec(SvRV(w->cb_get)); }
 671  if (w->cb_set   != NULL) { SvREFCNT_dec(SvRV(w->cb_set)); }
 672  if (w->cb_len   != NULL) { SvREFCNT_dec(SvRV(w->cb_len)); }
 673  if (w->cb_clear != NULL) { SvREFCNT_dec(SvRV(w->cb_clear)); }
 674  if (w->cb_free  != NULL) { SvREFCNT_dec(SvRV(w->cb_free)); }
 675 #if MGf_COPY
 676  if (w->cb_copy  != NULL) { SvREFCNT_dec(SvRV(w->cb_copy)); }
 677 #endif /* MGf_COPY */
 678 #if 0 /* MGf_DUP */
 679  if (w->cb_dup   != NULL) { SvREFCNT_dec(SvRV(w->cb_dup)); }
 680 #endif /* MGf_DUP */
 681 #if MGf_LOCAL
 682  if (w->cb_local != NULL) { SvREFCNT_dec(SvRV(w->cb_local)); }
 683 #endif /* MGf_LOCAL */
 684 #if VMG_UVAR
 685  if (w->cb_fetch  != NULL) { SvREFCNT_dec(SvRV(w->cb_fetch)); }
 686  if (w->cb_store  != NULL) { SvREFCNT_dec(SvRV(w->cb_store)); }
 687  if (w->cb_exists != NULL) { SvREFCNT_dec(SvRV(w->cb_exists)); }
 688  if (w->cb_delete != NULL) { SvREFCNT_dec(SvRV(w->cb_delete)); }
 689 #endif /* VMG_UVAR */
 690
 691  Safefree(w->vtbl);
 692  Safefree(w);
 693
 694  return 0;
 695 }
 696
 697 STATIC MGVTBL vmg_wizard_vtbl = {
 698  NULL,            /* get */
 699  NULL,            /* set */
 700  NULL,            /* len */
 701  NULL,            /* clear */
 702  vmg_wizard_free, /* free */
 703 #if MGf_COPY
 704  NULL,            /* copy */
 705 #endif /* MGf_COPY */
 706 #if MGf_DUP
 707  NULL,            /* dup */
 708 #endif /* MGf_DUP */
 709 #if MGf_LOCAL
 710  NULL,            /* local */
 711 #endif /* MGf_LOCAL */
 712 };
 713
 714 STATIC const char vmg_invalid_wiz[]    = "Invalid wizard object";
 715 STATIC const char vmg_invalid_sv[]     = "Invalid variable";
 716 STATIC const char vmg_invalid_sig[]    = "Invalid numeric signature";
 717 STATIC const char vmg_wrongargnum[]    = "Wrong number of arguments";
 718 STATIC const char vmg_toomanysigs[]    = "Too many magic signatures used";
 719 STATIC const char vmg_argstorefailed[] = "Error while storing arguments";
 720
 721 STATIC U16 vmg_sv2sig(pTHX_ SV *sv) {
 722 #define vmg_sv2sig(S) vmg_sv2sig(aTHX_ (S))
 723  U16 sig;
 724
 725  if (SvIOK(sv)) {
 726   sig = SvUVX(sv);
 727  } else if (SvNOK(sv)) {
 728   sig = SvNVX(sv);
 729  } else if ((SvPOK(sv) && grok_number(SvPVX(sv), SvCUR(sv), NULL))) {
 730   sig = SvUV(sv);
 731  } else {
 732   croak(vmg_invalid_sig);
 733  }
 734  if (sig < SIG_MIN) { sig += SIG_MIN; }
 735  if (sig > SIG_MAX) { sig %= SIG_MAX + 1; }
 736
 737  return sig;
 738 }
 739
 740 STATIC U16 vmg_wizard_sig(pTHX_ SV *wiz) {
 741 #define vmg_wizard_sig(W) vmg_wizard_sig(aTHX_ (W))
 742  char buf[8];
 743  U16 sig;
 744
 745  if (SvROK(wiz)) {
 746   sig = SV2MGWIZ(SvRV(wiz))->sig;
 747  } else if (SvOK(wiz)) {
 748   sig = vmg_sv2sig(wiz);
 749  } else {
 750   croak(vmg_invalid_wiz);
 751  }
 752
 753  {
 754   dMY_CXT;
 755   if (!hv_fetch(MY_CXT, buf, sprintf(buf, "%u", sig), 0))
 756    sig = 0;
 757  }
 758  return sig;
 759 }
 760
 761 STATIC SV *vmg_wizard_wiz(pTHX_ SV *wiz) {
 762 #define vmg_wizard_wiz(W) vmg_wizard_wiz(aTHX_ (W))
 763  char buf[8];
 764  SV **old;
 765  U16 sig;
 766
 767  if (SvROK(wiz)) {
 768   wiz = SvRV(wiz);
 769 #if VMG_MULTIPLICITY
 770   if (SV2MGWIZ(wiz)->owner == aTHX)
 771    return wiz;
 772 #endif /* VMG_MULTIPLICITY */
 773   sig = SV2MGWIZ(wiz)->sig;
 774  } else if (SvOK(wiz)) {
 775   sig = vmg_sv2sig(wiz);
 776  } else {
 777   croak(vmg_invalid_wiz);
 778  }
 779
 780  {
 781   dMY_CXT;
 782   return (old = hv_fetch(MY_CXT, buf, sprintf(buf, "%u", sig), 0))
 783           ? *old : NULL;
 784  }
 785 }
 786
 787 #define VMG_SET_CB(S, N)              \
 788  cb = (S);                            \
 789  w->cb_ ## N = (SvOK(cb) && SvROK(cb)) ? newRV_inc(SvRV(cb)) : NULL;
 790
 791 #define VMG_SET_SVT_CB(S, N)          \
 792  cb = (S);                            \
 793  if (SvOK(cb) && SvROK(cb)) {         \
 794   t->svt_ ## N = vmg_svt_ ## N;       \
 795   w->cb_  ## N = newRV_inc(SvRV(cb)); \
 796  } else {                             \
 797   t->svt_ ## N = NULL;                \
 798   w->cb_  ## N = NULL;                \
 799  }
 800
 801 #if VMG_THREADSAFE
 802
 803 #define VMG_CLONE_CB(N) \
 804  z->cb_ ## N = (w->cb_ ## N) ? newRV_noinc(vmg_clone(SvRV(w->cb_ ## N), \
 805                                            w->owner))                   \
 806                              : NULL;
 807
 808 STATIC MGWIZ *vmg_wizard_clone(pTHX_ const MGWIZ *w) {
 809 #define vmg_wizard_clone(W) vmg_wizard_clone(aTHX_ (W))
 810  MGVTBL *t;
 811  MGWIZ *z;
 812
 813  Newx(t, 1, MGVTBL);
 814  Copy(w->vtbl, t, 1, MGVTBL);
 815
 816  Newx(z, 1, MGWIZ);
 817  VMG_CLONE_CB(data);
 818  VMG_CLONE_CB(get);
 819  VMG_CLONE_CB(set);
 820  VMG_CLONE_CB(len);
 821  VMG_CLONE_CB(clear);
 822  VMG_CLONE_CB(free);
 823 #if MGf_COPY
 824  VMG_CLONE_CB(copy);
 825 #endif /* MGf_COPY */
 826 #if MGf_DUP
 827  VMG_CLONE_CB(dup);
 828 #endif /* MGf_DUP */
 829 #if MGf_LOCAL
 830  VMG_CLONE_CB(local);
 831 #endif /* MGf_LOCAL */
 832 #if VMG_UVAR
 833  VMG_CLONE_CB(fetch);
 834  VMG_CLONE_CB(store);
 835  VMG_CLONE_CB(exists);
 836  VMG_CLONE_CB(delete);
 837 #endif /* VMG_UVAR */
 838  z->owner = aTHX;
 839  z->vtbl  = t;
 840  z->sig   = w->sig;
 841  z->uvar  = w->uvar;
 842
 843  return z;
 844 }
 845
 846 #endif /* VMG_THREADSAFE */
 847
 848 /* --- XS ------------------------------------------------------------------ */
 849
 850 MODULE = Variable::Magic            PACKAGE = Variable::Magic
 851
 852 PROTOTYPES: ENABLE
 853
 854 BOOT:
 855 {
 856  HV *stash;
 857  MY_CXT_INIT;
 858  MY_CXT = newHV();
 859  hv_iterinit(MY_CXT); /* Allocate iterator */
 860  stash = gv_stashpv(__PACKAGE__, 1);
 861  newCONSTSUB(stash, "SIG_MIN",   newSVuv(SIG_MIN));
 862  newCONSTSUB(stash, "SIG_MAX",   newSVuv(SIG_MAX));
 863  newCONSTSUB(stash, "SIG_NBR",   newSVuv(SIG_NBR));
 864  newCONSTSUB(stash, "MGf_COPY",  newSVuv(MGf_COPY));
 865  newCONSTSUB(stash, "MGf_DUP",   newSVuv(MGf_DUP));
 866  newCONSTSUB(stash, "MGf_LOCAL", newSVuv(MGf_LOCAL));
 867  newCONSTSUB(stash, "VMG_UVAR",  newSVuv(VMG_UVAR));
 868  newCONSTSUB(stash, "VMG_COMPAT_ARRAY_PUSH_NOLEN",
 869                     newSVuv(VMG_COMPAT_ARRAY_PUSH_NOLEN));
 870  newCONSTSUB(stash, "VMG_COMPAT_ARRAY_UNDEF_CLEAR",
 871                     newSVuv(VMG_COMPAT_ARRAY_UNDEF_CLEAR));
 872  newCONSTSUB(stash, "VMG_COMPAT_SCALAR_LENGTH_NOLEN",
 873                     newSVuv(VMG_COMPAT_SCALAR_LENGTH_NOLEN));
 874  newCONSTSUB(stash, "VMG_PERL_PATCHLEVEL", newSVuv(VMG_PERL_PATCHLEVEL));
 875  newCONSTSUB(stash, "VMG_THREADSAFE",      newSVuv(VMG_THREADSAFE));
 876 }
 877
 878 void
 879 CLONE(...)
 880 PROTOTYPE: DISABLE
 881 PREINIT:
 882  HV *hv;
 883 CODE:
 884 #if VMG_THREADSAFE
 885  {
 886   HE *key;
 887   dMY_CXT;
 888   hv = newHV();
 889   hv_iterinit(hv); /* Allocate iterator */
 890   hv_iterinit(MY_CXT);
 891   while ((key = hv_iternext(MY_CXT))) {
 892    STRLEN len;
 893    char *sig = HePV(key, len);
 894    SV *sv;
 895    MAGIC *mg;
 896    sv = MGWIZ2SV(vmg_wizard_clone(SV2MGWIZ(HeVAL(key))));
 897    mg = sv_magicext(sv, NULL, PERL_MAGIC_ext, &vmg_wizard_vtbl, NULL, 0);
 898    mg->mg_private = SIG_WIZ;
 899    SvREADONLY_on(sv);
 900    hv_store(hv, sig, len, sv, HeHASH(key));
 901   }
 902  }
 903  {
 904   MY_CXT_CLONE;
 905   MY_CXT = hv;
 906  }
 907 #endif /* VMG_THREADSAFE */
 908
 909 SV *_wizard(...)
 910 PROTOTYPE: DISABLE
 911 PREINIT:
 912  I32 i = 0;
 913  U16 sig;
 914  char buf[8];
 915  MGWIZ *w;
 916  MGVTBL *t;
 917  MAGIC *mg;
 918  SV *sv;
 919  SV *svsig;
 920  SV *cb;
 921 CODE:
 922  dMY_CXT;
 923
 924  if (items != 7
 925 #if MGf_COPY
 926               + 1
 927 #endif /* MGf_COPY */
 928 #if MGf_DUP
 929               + 1
 930 #endif /* MGf_DUP */
 931 #if MGf_LOCAL
 932               + 1
 933 #endif /* MGf_LOCAL */
 934 #if VMG_UVAR
 935               + 4
 936 #endif /* VMG_UVAR */
 937               ) { croak(vmg_wrongargnum); }
 938
 939  svsig = ST(i++);
 940  if (SvOK(svsig)) {
 941   SV **old;
 942   sig = vmg_sv2sig(svsig);
 943   if ((old = hv_fetch(MY_CXT, buf, sprintf(buf, "%u", sig), 0))) {
 944    ST(0) = sv_2mortal(newRV_inc(*old));
 945    XSRETURN(1);
 946   }
 947  } else {
 948   if (HvKEYS(MY_CXT) >= SIG_NBR) { croak(vmg_toomanysigs); }
 949   sig = vmg_gensig();
 950  }
 951
 952  Newx(t, 1, MGVTBL);
 953  Newx(w, 1, MGWIZ);
 954
 955  VMG_SET_CB(ST(i++), data);
 956  VMG_SET_SVT_CB(ST(i++), get);
 957  VMG_SET_SVT_CB(ST(i++), set);
 958  VMG_SET_SVT_CB(ST(i++), len);
 959  VMG_SET_SVT_CB(ST(i++), clear);
 960  VMG_SET_SVT_CB(ST(i++), free);
 961 #if MGf_COPY
 962  VMG_SET_SVT_CB(ST(i++), copy);
 963 #endif /* MGf_COPY */
 964 #if MGf_DUP
 965  /* VMG_SET_SVT_CB(ST(i++), dup); */
 966  i++;
 967  t->svt_dup = NULL;
 968  w->cb_dup  = NULL;
 969 #endif /* MGf_DUP */
 970 #if MGf_LOCAL
 971  VMG_SET_SVT_CB(ST(i++), local);
 972 #endif /* MGf_LOCAL */
 973 #if VMG_UVAR
 974  VMG_SET_CB(ST(i++), fetch);
 975  VMG_SET_CB(ST(i++), store);
 976  VMG_SET_CB(ST(i++), exists);
 977  VMG_SET_CB(ST(i++), delete);
 978 #endif /* VMG_UVAR */
 979 #if VMG_MULTIPLICITY
 980  w->owner = aTHX;
 981 #endif /* VMG_MULTIPLICITY */
 982
 983  w->vtbl = t;
 984  w->sig  = sig;
 985 #if VMG_UVAR
 986  w->uvar = (w->cb_fetch || w->cb_store || w->cb_exists || w->cb_delete);
 987 #endif /* VMG_UVAR */
 988
 989  sv = MGWIZ2SV(w);
 990  mg = sv_magicext(sv, NULL, PERL_MAGIC_ext, &vmg_wizard_vtbl, NULL, 0);
 991  mg->mg_private = SIG_WIZ;
 992  SvREADONLY_on(sv);
 993
 994  hv_store(MY_CXT, buf, sprintf(buf, "%u", sig), sv, 0);
 995
 996  RETVAL = newRV_noinc(sv);
 997 OUTPUT:
 998  RETVAL
 999
1000 SV *gensig()
1001 PROTOTYPE:
1002 CODE:
1003  dMY_CXT;
1004  if (HvKEYS(MY_CXT) >= SIG_NBR) { croak(vmg_toomanysigs); }
1005  RETVAL = newSVuv(vmg_gensig());
1006 OUTPUT:
1007  RETVAL
1008
1009 SV *getsig(SV *wiz)
1010 PROTOTYPE: $
1011 CODE:
1012  if (!SvROK(wiz)) { croak(vmg_invalid_wiz); }
1013  RETVAL = newSVuv(SV2MGWIZ(SvRV(wiz))->sig);
1014 OUTPUT:
1015  RETVAL
1016
1017 SV *cast(SV *sv, SV *wiz, ...)
1018 PROTOTYPE: \[$@%&*]$@
1019 PREINIT:
1020  AV *args = NULL;
1021  SV *ret;
1022 CODE:
1023  wiz = vmg_wizard_wiz(wiz);
1024  if (!wiz)
1025   XSRETURN_UNDEF;
1026  if (items > 2) {
1027   I32 i;
1028   args = newAV();
1029   av_fill(args, items - 2);
1030   for (i = 2; i < items; ++i) {
1031    SV *arg = ST(i);
1032    SvREFCNT_inc(arg);
1033    if (av_store(args, i - 2, arg) == NULL) { croak(vmg_argstorefailed); }
1034   }
1035  }
1036  ret = newSVuv(vmg_cast(SvRV(sv), wiz, args));
1037  SvREFCNT_dec(args);
1038  RETVAL = ret;
1039 OUTPUT:
1040  RETVAL
1041
1042 SV *getdata(SV *sv, SV *wiz)
1043 PROTOTYPE: \[$@%&*]$
1044 PREINIT:
1045  SV *data;
1046  U16 sig;
1047 CODE:
1048  sig = vmg_wizard_sig(wiz);
1049  if (!sig)
1050   XSRETURN_UNDEF;
1051  data = vmg_data_get(SvRV(sv), sig);
1052  if (!data) { XSRETURN_UNDEF; }
1053  ST(0) = data;
1054  XSRETURN(1);
1055
1056 SV *dispell(SV *sv, SV *wiz)
1057 PROTOTYPE: \[$@%&*]$
1058 PREINIT:
1059  U16 sig;
1060 CODE:
1061  sig = vmg_wizard_sig(wiz);
1062  if (!sig)
1063   XSRETURN_UNDEF;
1064  RETVAL = newSVuv(vmg_dispell(SvRV(sv), sig));
1065 OUTPUT:
1066  RETVAL