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