]> git.vpit.fr Git - perl/modules/Lexical-Types.git/blobdiff - Types.xs
This is 0.16
[perl/modules/Lexical-Types.git] / Types.xs
index 7d50b19fdbbf1a393773dbee7a5ad6e840ac3484..7de168535aa48a10e28178e6dc5c869309203259 100644 (file)
--- a/Types.xs
+++ b/Types.xs
 #include "perl.h"
 #include "XSUB.h"
 
-/* --- Compatibility wrappers ---------------------------------------------- */
+/* --- XS helpers ---------------------------------------------------------- */
 
-#define LT_HAS_PERL(R, V, S) (PERL_REVISION > (R) || (PERL_REVISION == (R) && (PERL_VERSION > (V) || (PERL_VERSION == (V) && (PERL_SUBVERSION >= (S))))))
+#define XSH_PACKAGE "Lexical::Types"
 
-#if LT_HAS_PERL(5, 10, 0) || defined(PL_parser)
-# ifndef PL_in_my_stash
-#  define PL_in_my_stash PL_parser->in_my_stash
-# endif
-#else
-# ifndef PL_in_my_stash
-#  define PL_in_my_stash PL_Iin_my_stash
-# endif
-#endif
+#include "xsh/caps.h"
+#include "xsh/util.h"
+#include "xsh/mem.h"
+#include "xsh/ops.h"
+#include "xsh/peep.h"
 
-#ifndef Newx
-# define Newx(v, n, c) New(0, v, n, c)
-#endif
+/* ... Lexical hints ....................................................... */
 
-#ifndef HvNAME_get
-# define HvNAME_get(H) HvNAME(H)
-#endif
+#define XSH_HINTS_TYPE_SV 1
 
-#ifndef HvNAMELEN_get
-# define HvNAMELEN_get(H) strlen(HvNAME_get(H))
-#endif
+#include "xsh/hints.h"
 
-#define __PACKAGE__     "Lexical::Types"
-#define __PACKAGE_LEN__ (sizeof(__PACKAGE__)-1)
+#define lt_hint() xsh_hints_detag(xsh_hints_fetch())
 
-/* --- Helpers ------------------------------------------------------------- */
+/* ... Thread-local storage ................................................ */
 
-/* ... Hints ............................................................... */
+typedef struct {
+ SV *default_meth;
+} xsh_user_cxt_t;
 
-STATIC U32 lt_hash = 0;
+#define XSH_THREADS_COMPILE_TIME_PROTECTION 1
+#define XSH_THREADS_USER_CLONE_NEEDS_DUP    1
 
-STATIC SV *lt_hint(pTHX) {
-#define lt_hint() lt_hint(aTHX)
- SV *id;
-#if LT_HAS_PERL(5, 10, 0)
- id = Perl_refcounted_he_fetch(aTHX_ PL_curcop->cop_hints_hash,
-                                     NULL,
-                                     __PACKAGE__, __PACKAGE_LEN__,
-                                     0,
-                                     lt_hash);
-#else
- SV **val = hv_fetch(GvHV(PL_hintgv), __PACKAGE__, __PACKAGE_LEN__, lt_hash);
- if (!val)
-  return 0;
- id = *val;
-#endif
- return (id && SvOK(id)) ? id : NULL;
+#if XSH_THREADSAFE
+
+static void xsh_user_clone(pTHX_ const xsh_user_cxt_t *old_cxt, xsh_user_cxt_t *new_cxt, CLONE_PARAMS *params) {
+ new_cxt->default_meth = xsh_dup_inc(old_cxt->default_meth, params);
+
+ return;
 }
 
+#endif /* XSH_THREADSAFE */
+
+#include "xsh/threads.h"
+
 /* ... op => info map ...................................................... */
 
-#define OP2STR_BUF char buf[(CHAR_BIT * sizeof(UV)) / 2]
-#define OP2STR(O)  (sprintf(buf, "%"UVxf, PTR2UV(O)))
+#define PTABLE_NAME             ptable_map
+#define PTABLE_VAL_FREE(V)      XSH_SHARED_FREE((V), 0, char)
+#define PTABLE_VAL_NEED_CONTEXT 0
+#define PTABLE_NEED_DELETE      1
+#define PTABLE_NEED_WALK        0
+
+#include "xsh/ptable.h"
+
+#define ptable_map_store(T, K, V) ptable_map_store(aPMS_ (T), (K), (V))
+#define ptable_map_delete(T, K)   ptable_map_delete(aPMS_ (T), (K))
+#define ptable_map_free(T)        ptable_map_free(aPMS_ (T))
+
+#ifdef USE_ITHREADS
+
+static perl_mutex lt_op_map_mutex;
 
-STATIC HV *lt_op_map = NULL;
+#endif /* USE_ITHREADS */
+
+static ptable *lt_op_padxv_map = NULL;
 
 typedef struct {
+ OP *(*old_pp)(pTHX);
+#ifdef MULTIPLICITY
+ STRLEN buf_size, orig_pkg_len, type_pkg_len, type_meth_len;
+ char *buf;
+#else /* MULTIPLICITY */
  SV *orig_pkg;
  SV *type_pkg;
  SV *type_meth;
- OP *(*pp_padsv)(pTHX);
-} lt_op_info;
+#endif /* !MULTIPLICITY */
+} lt_op_padxv_info;
+
+static void lt_op_padxv_info_call(pTHX_ const lt_op_padxv_info *oi, SV *sv) {
+#define lt_op_padxv_info_call(O, S) lt_op_padxv_info_call(aTHX_ (O), (S))
+ SV *orig_pkg, *type_pkg, *type_meth;
+ int items;
+ dSP;
+
+ ENTER;
+ SAVETMPS;
+
+#ifdef MULTIPLICITY
+ {
+  STRLEN op_len = oi->orig_pkg_len, tp_len = oi->type_pkg_len;
+  char *buf = oi->buf;
+  orig_pkg  = sv_2mortal(newSVpvn(buf, op_len));
+  SvREADONLY_on(orig_pkg);
+  buf      += op_len;
+  type_pkg  = sv_2mortal(newSVpvn(buf, tp_len));
+  SvREADONLY_on(type_pkg);
+  buf      += tp_len;
+  type_meth = sv_2mortal(newSVpvn(buf, oi->type_meth_len));
+  SvREADONLY_on(type_meth);
+ }
+#else /* MULTIPLICITY */
+ orig_pkg  = oi->orig_pkg;
+ type_pkg  = oi->type_pkg;
+ type_meth = oi->type_meth;
+#endif /* !MULTIPLICITY */
+
+ PUSHMARK(SP);
+ EXTEND(SP, 3);
+ PUSHs(type_pkg);
+ PUSHs(sv);
+ PUSHs(orig_pkg);
+ PUTBACK;
+
+ items = call_sv(type_meth, G_ARRAY | G_METHOD);
+
+ SPAGAIN;
+ switch (items) {
+  case 0:
+   break;
+  case 1:
+   sv_setsv(sv, POPs);
+   break;
+  default:
+   croak("Typed scalar initializer method should return zero or one scalar, but got %d", items);
+ }
+ PUTBACK;
 
-STATIC void lt_map_store(pTHX_ const OP *o, SV *orig_pkg, SV *type_pkg, SV *type_meth, OP *(*pp_padsv)(pTHX)) {
-#define lt_map_store(O, P1, P2, M, PP) lt_map_store(aTHX_ (O), (P1), (P2), (M), (PP))
- OP2STR_BUF;
- SV *val;
- lt_op_info *oi;
+ FREETMPS;
+ LEAVE;
 
- Newx(oi, 1, lt_op_info);
+ return;
+}
+
+static void lt_padxv_map_store(pTHX_ const OP *o, SV *orig_pkg, SV *type_pkg, SV *type_meth, OP *(*old_pp)(pTHX)) {
+#define lt_padxv_map_store(O, OP, TP, TM, PP) lt_padxv_map_store(aTHX_ (O), (OP), (TP), (TM), (PP))
+ lt_op_padxv_info *oi;
+
+ XSH_LOCK(&lt_op_map_mutex);
+
+ if (!(oi = ptable_fetch(lt_op_padxv_map, o))) {
+  XSH_SHARED_ALLOC(oi, 1, lt_op_padxv_info);
+  ptable_map_store(lt_op_padxv_map, o, oi);
+#ifdef MULTIPLICITY
+  oi->buf      = NULL;
+  oi->buf_size = 0;
+#else /* MULTIPLICITY */
+ } else {
+  SvREFCNT_dec(oi->orig_pkg);
+  SvREFCNT_dec(oi->type_pkg);
+  SvREFCNT_dec(oi->type_meth);
+#endif /* !MULTIPLICITY */
+ }
+
+#ifdef MULTIPLICITY
+ {
+  STRLEN op_len       = SvCUR(orig_pkg);
+  STRLEN tp_len       = SvCUR(type_pkg);
+  STRLEN tm_len       = SvCUR(type_meth);
+  STRLEN new_buf_size = op_len + tp_len + tm_len;
+  char *buf;
+  if (new_buf_size > oi->buf_size) {
+   XSH_SHARED_REALLOC(oi->buf, oi->buf_size, new_buf_size, char);
+   oi->buf_size = new_buf_size;
+  }
+  buf  = oi->buf;
+  Copy(SvPVX(orig_pkg),  buf, op_len, char);
+  buf += op_len;
+  Copy(SvPVX(type_pkg),  buf, tp_len, char);
+  buf += tp_len;
+  Copy(SvPVX(type_meth), buf, tm_len, char);
+  oi->orig_pkg_len  = op_len;
+  oi->type_pkg_len  = tp_len;
+  oi->type_meth_len = tm_len;
+  SvREFCNT_dec(orig_pkg);
+  SvREFCNT_dec(type_pkg);
+  SvREFCNT_dec(type_meth);
+ }
+#else /* MULTIPLICITY */
  oi->orig_pkg  = orig_pkg;
  oi->type_pkg  = type_pkg;
  oi->type_meth = type_meth;
- oi->pp_padsv  = pp_padsv;
- val = newSVuv(PTR2UV(oi));
+#endif /* !MULTIPLICITY */
 
- (void)hv_store(lt_op_map, buf, OP2STR(o), val, 0);
+ oi->old_pp = old_pp;
+
+ XSH_UNLOCK(&lt_op_map_mutex);
 }
 
-STATIC const lt_op_info *lt_map_fetch(pTHX_ const OP *o) {
-#define lt_map_fetch(O) lt_map_fetch(aTHX_ (O))
- OP2STR_BUF;
- SV **svp;
+static const lt_op_padxv_info *lt_padxv_map_fetch(const OP *o, lt_op_padxv_info *oi) {
+ const lt_op_padxv_info *val;
 
svp = hv_fetch(lt_op_map, buf, OP2STR(o), 0);
XSH_LOCK(&lt_op_map_mutex);
 
- return svp ? INT2PTR(const lt_op_info *, SvUVX(*svp)) : NULL;
+ val = ptable_fetch(lt_op_padxv_map, o);
+ if (val) {
+  *oi = *val;
+  val = oi;
+ }
+
+ XSH_UNLOCK(&lt_op_map_mutex);
+
+ return val;
 }
 
-/* --- Hooks --------------------------------------------------------------- */
+#if XSH_HAS_PERL(5, 17, 6)
 
-/* ... Our pp_padsv ........................................................ */
+static ptable *lt_op_padrange_map = NULL;
 
-STATIC OP *lt_pp_padsv(pTHX) {
- const lt_op_info *oi;
+typedef struct {
+ OP *(*old_pp)(pTHX);
+ const OP *padxv_start;
+} lt_op_padrange_info;
 
- if ((PL_op->op_private & OPpLVAL_INTRO) && (oi = lt_map_fetch(PL_op))) {
-  PADOFFSET targ = PL_op->op_targ;
 SV *sv         = PAD_SVl(targ);
+static void lt_padrange_map_store(pTHX_ const OP *o, const OP *s, OP *(*old_pp)(pTHX)) {
+#define lt_padrange_map_store(O, S, PP) lt_padrange_map_store(aTHX_ (O), (S), (PP))
lt_op_padrange_info *oi;
 
-  if (sv) {
-   int items;
-   dSP;
+ XSH_LOCK(&lt_op_map_mutex);
 
-   ENTER;
-   SAVETMPS;
+ if (!(oi = ptable_fetch(lt_op_padrange_map, o))) {
+  XSH_SHARED_ALLOC(oi, 1, lt_op_padrange_info);
+  ptable_map_store(lt_op_padrange_map, o, oi);
+ }
 
-   PUSHMARK(SP);
-   EXTEND(SP, 3);
-   PUSHs(oi->type_pkg);
-   PUSHs(sv);
-   PUSHs(oi->orig_pkg);
-   PUTBACK;
+ oi->old_pp      = old_pp;
+ oi->padxv_start = s;
 
-   items = call_sv(oi->type_meth, G_ARRAY | G_METHOD);
+ XSH_UNLOCK(&lt_op_map_mutex);
+}
 
-   SPAGAIN;
-   switch (items) {
-    case 0:
-     break;
-    case 1:
-     sv_setsv(sv, POPs);
-     break;
-    default:
-     croak("Typed scalar initializer method should return zero or one scalar, but got %d", items);
-   }
-   PUTBACK;
+static const lt_op_padrange_info *lt_padrange_map_fetch(const OP *o, lt_op_padrange_info *oi) {
+ const lt_op_padrange_info *val;
 
-   FREETMPS;
-   LEAVE;
-  }
+ XSH_LOCK(&lt_op_map_mutex);
 
-  return CALL_FPTR(oi->pp_padsv)(aTHX);
+ val = ptable_fetch(lt_op_padrange_map, o);
+ if (val) {
+  *oi = *val;
+  val = oi;
  }
 
- return CALL_FPTR(PL_ppaddr[OP_PADSV])(aTHX);
+ XSH_UNLOCK(&lt_op_map_mutex);
+
+ return val;
 }
 
-STATIC OP *(*lt_pp_padsv_saved)(pTHX) = 0;
+#endif
 
-STATIC void lt_pp_padsv_save(void) {
- if (lt_pp_padsv_saved)
 return;
+static void lt_map_delete(pTHX_ const OP *o) {
+#define lt_map_delete(O) lt_map_delete(aTHX_ (O))
XSH_LOCK(&lt_op_map_mutex);
 
- lt_pp_padsv_saved   = PL_ppaddr[OP_PADSV];
- PL_ppaddr[OP_PADSV] = lt_pp_padsv;
+ ptable_map_delete(lt_op_padxv_map,    o);
+#if XSH_HAS_PERL(5, 17, 6)
+ ptable_map_delete(lt_op_padrange_map, o);
+#endif
+
+ XSH_UNLOCK(&lt_op_map_mutex);
 }
 
-STATIC void lt_pp_padsv_restore(OP *o) {
- if (!lt_pp_padsv_saved)
-  return;
+/* --- Compatibility wrappers ---------------------------------------------- */
 
- if (o->op_ppaddr == lt_pp_padsv)
-  o->op_ppaddr = lt_pp_padsv_saved;
+#if XSH_HAS_PERL(5, 10, 0) || defined(PL_parser)
+# ifndef PL_in_my_stash
+#  define PL_in_my_stash PL_parser->in_my_stash
+# endif
+#else
+# ifndef PL_in_my_stash
+#  define PL_in_my_stash PL_Iin_my_stash
+# endif
+#endif
+
+#ifndef HvNAME_get
+# define HvNAME_get(H) HvNAME(H)
+#endif
+
+#ifndef HvNAMELEN_get
+# define HvNAMELEN_get(H) strlen(HvNAME_get(H))
+#endif
+
+#ifndef SvREFCNT_inc_simple_void_NN
+# define SvREFCNT_inc_simple_void_NN(S) ((void) SvREFCNT_inc(S))
+#endif
+
+/* --- PP functions -------------------------------------------------------- */
+
+/* ... pp_padsv ............................................................ */
+
+static OP *lt_pp_padsv(pTHX) {
+ lt_op_padxv_info oi;
+
+ if (lt_padxv_map_fetch(PL_op, &oi)) {
+  dTARGET;
+  lt_op_padxv_info_call(&oi, TARG);
+  return oi.old_pp(aTHX);
+ }
 
- PL_ppaddr[OP_PADSV] = lt_pp_padsv_saved;
- lt_pp_padsv_saved   = 0;
+ return PL_op->op_ppaddr(aTHX);
 }
 
-/* ... Our ck_pad{any,sv} .................................................. */
+/* ... pp_padrange (on perl 5.17.6 and above) .............................. */
 
-/* Sadly, the PADSV OPs we are interested in don't trigger the padsv check
- * function, but are instead manually mutated from a PADANY. This is why we set
- * PL_ppaddr[OP_PADSV] in the padany check function so that PADSV OPs will have
- * their pp_ppaddr set to our pp_padsv. PL_ppaddr[OP_PADSV] is then reset at the
- * beginning of every ck_pad{any,sv}. Some unwanted OPs can still call our
- * pp_padsv, but much less than if we would have set PL_ppaddr[OP_PADSV]
- * globally. */
+#if XSH_HAS_PERL(5, 17, 6)
 
-STATIC SV *lt_default_meth = NULL;
+static OP *lt_pp_padrange(pTHX) {
+ lt_op_padrange_info roi;
 
-STATIC OP *(*lt_old_ck_padany)(pTHX_ OP *) = 0;
+ if (lt_padrange_map_fetch(PL_op, &roi)) {
+  PADOFFSET i, base, count;
+  const OP *p;
 
-STATIC OP *lt_ck_padany(pTHX_ OP *o) {
- HV *stash;
- SV *hint;
+  base  = PL_op->op_targ;
+  count = PL_op->op_private & OPpPADRANGE_COUNTMASK;
 
- lt_pp_padsv_restore(o);
+  for (i = 0, p = roi.padxv_start; i < count && p; ++i, p = p->op_next) {
+   lt_op_padxv_info oi;
+   while (p->op_type == OP_NULL)
+    p = p->op_next;
+   if (p->op_type == OP_PADSV && lt_padxv_map_fetch(p, &oi))
+    lt_op_padxv_info_call(&oi, PAD_SV(base + i));
+  }
 
- o = CALL_FPTR(lt_old_ck_padany)(aTHX_ o);
+  return roi.old_pp(aTHX);
+ }
+
+ return PL_op->op_ppaddr(aTHX);
+}
+
+#endif
+
+/* --- Check functions ----------------------------------------------------- */
+
+/* ... ck_pad{any,sv} ...................................................... */
+
+/* Sadly, the padsv OPs we are interested in don't trigger the padsv check
+ * function, but are instead manually mutated from a padany. So we store
+ * the op entry in the op map in the padany check function, and we set their
+ * op_ppaddr member in our peephole optimizer replacement below. */
+
+static OP *(*lt_old_ck_padany)(pTHX_ OP *) = 0;
+
+static OP *lt_ck_padany(pTHX_ OP *o) {
+ HV *stash;
+ SV *code;
+
+ o = lt_old_ck_padany(aTHX_ o);
 
  stash = PL_in_my_stash;
- if (stash && (hint = lt_hint())) {
+ if (stash && (code = lt_hint())) {
+  dXSH_CXT;
   SV *orig_pkg  = newSVpvn(HvNAME_get(stash), HvNAMELEN_get(stash));
-  SV *orig_meth = lt_default_meth;
-  SV *type_pkg  = orig_pkg;
-  SV *type_meth = orig_meth;
-  SV *code      = INT2PTR(SV *, SvUVX(hint));
+  SV *orig_meth = XSH_CXT.default_meth; /* Guarded by lt_hint() */
+  SV *type_pkg  = NULL;
+  SV *type_meth = NULL;
+  int items;
 
-  SvREADONLY_on(orig_pkg);
+  dSP;
 
-  if (code) {
-   int items;
-   dSP;
+  SvREADONLY_on(orig_pkg);
 
-   ENTER;
-   SAVETMPS;
+  ENTER;
+  SAVETMPS;
 
-   PUSHMARK(SP);
-   EXTEND(SP, 2);
-   PUSHs(orig_pkg);
-   PUSHs(orig_meth);
-   PUTBACK;
+  PUSHMARK(SP);
+  EXTEND(SP, 2);
+  PUSHs(orig_pkg);
+  PUSHs(orig_meth);
+  PUTBACK;
 
-   items = call_sv(code, G_ARRAY);
+  items = call_sv(code, G_ARRAY);
 
-   SPAGAIN;
-   if (items > 2)
-    croak(__PACKAGE__ " mangler should return zero, one or two scalars, but got %d", items);
-   if (items == 0) {
-    SvREFCNT_dec(orig_pkg);
-    goto skip;
-   } else {
-    SV *rsv;
-    if (items > 1) {
-     rsv = POPs;
-     if (SvOK(rsv)) {
-      type_meth = newSVsv(rsv);
-      SvREADONLY_on(type_meth);
-     }
-    }
+  SPAGAIN;
+  if (items > 2)
+   croak(XSH_PACKAGE " mangler should return zero, one or two scalars, but got %d", items);
+  if (items == 0) {
+   SvREFCNT_dec(orig_pkg);
+   FREETMPS;
+   LEAVE;
+   goto skip;
+  } else {
+   SV *rsv;
+   if (items > 1) {
     rsv = POPs;
     if (SvOK(rsv)) {
-     type_pkg = newSVsv(rsv);
-     SvREADONLY_on(type_pkg);
+     type_meth = newSVsv(rsv);
+     SvREADONLY_on(type_meth);
     }
    }
-   PUTBACK;
-
-   FREETMPS;
-   LEAVE;
+   rsv = POPs;
+   if (SvOK(rsv)) {
+    type_pkg = newSVsv(rsv);
+    SvREADONLY_on(type_pkg);
+   }
   }
+  PUTBACK;
 
-  if (type_meth == orig_meth)
-   SvREFCNT_inc(orig_meth);
+  FREETMPS;
+  LEAVE;
 
-  lt_pp_padsv_save();
+  if (!type_pkg) {
+   type_pkg = orig_pkg;
+   SvREFCNT_inc_simple_void_NN(orig_pkg);
+  }
 
-  lt_map_store(o, orig_pkg, type_pkg, type_meth, lt_pp_padsv_saved);
- }
+  if (!type_meth) {
+   type_meth = orig_meth;
+   SvREFCNT_inc_simple_void_NN(orig_meth);
+  }
 
+  lt_padxv_map_store(o, orig_pkg, type_pkg, type_meth, o->op_ppaddr);
+ } else {
 skip:
+  lt_map_delete(o);
+ }
+
  return o;
 }
 
-STATIC OP *(*lt_old_ck_padsv)(pTHX_ OP *) = 0;
+static OP *(*lt_old_ck_padsv)(pTHX_ OP *) = 0;
 
-STATIC OP *lt_ck_padsv(pTHX_ OP *o) {
- lt_pp_padsv_restore(o);
+static OP *lt_ck_padsv(pTHX_ OP *o) {
+ lt_map_delete(o);
 
- return CALL_FPTR(lt_old_ck_padsv)(aTHX_ o);
+ return lt_old_ck_padsv(aTHX_ o);
 }
 
-STATIC U32 lt_initialized = 0;
+/* --- Our peephole optimizer ---------------------------------------------- */
 
-/* --- XS ------------------------------------------------------------------ */
+#if XSH_HAS_PERL(5, 17, 6)
 
-MODULE = Lexical::Types      PACKAGE = Lexical::Types
+static int lt_maybe_padrange_setup(pTHX_ OP *o, const OP *start) {
+#define lt_maybe_padrange_setup(O, S) lt_maybe_padrange_setup(aTHX_ (O), (S))
+ PADOFFSET i, count;
+ const OP *p;
 
-PROTOTYPES: DISABLE
+ count = o->op_private & OPpPADRANGE_COUNTMASK;
+
+ for (i = 0, p = start; i < count && p; ++i, p = p->op_next) {
+  if (p->op_type == OP_PADSV) {
+   /* In a padrange sequence, either all lexicals are typed, or none are.
+    * Thus we can stop at the first padsv op. However, note that these
+    * lexicals can need to call different methods in different packages. */
+   XSH_LOCK(&lt_op_map_mutex);
+   if (ptable_fetch(lt_op_padxv_map, p)) {
+    XSH_UNLOCK(&lt_op_map_mutex);
+    lt_padrange_map_store(o, start, o->op_ppaddr);
+    o->op_ppaddr = lt_pp_padrange;
+   } else {
+    XSH_UNLOCK(&lt_op_map_mutex);
+   }
+   return 1;
+  }
+ }
 
-BOOT: 
-{                                    
- if (!lt_initialized++) {
-  lt_default_meth = newSVpvn("TYPEDSCALAR", 11);
-  SvREADONLY_on(lt_default_meth);
+ return 0;
+}
 
-  PERL_HASH(lt_hash, __PACKAGE__, __PACKAGE_LEN__);
-  lt_op_map = newHV();
+#endif
 
-  lt_old_ck_padany    = PL_check[OP_PADANY];
-  PL_check[OP_PADANY] = MEMBER_TO_FPTR(lt_ck_padany);
-  lt_old_ck_padsv     = PL_check[OP_PADSV];
-  PL_check[OP_PADSV]  = MEMBER_TO_FPTR(lt_ck_padsv);
+static void xsh_peep_rec(pTHX_ OP *o, ptable *seen) {
+ for (; o; o = o->op_next) {
+  if (xsh_peep_seen(o, seen))
+   break;
+
+  switch (o->op_type) {
+   case OP_PADSV:
+    if (o->op_ppaddr != lt_pp_padsv && o->op_private & OPpLVAL_INTRO) {
+     lt_op_padxv_info *oi;
+     XSH_LOCK(&lt_op_map_mutex);
+     oi = ptable_fetch(lt_op_padxv_map, o);
+     if (oi) {
+      oi->old_pp   = o->op_ppaddr;
+      o->op_ppaddr = lt_pp_padsv;
+     }
+     XSH_UNLOCK(&lt_op_map_mutex);
+    }
+    break;
+#if XSH_HAS_PERL(5, 17, 6)
+   case OP_PADRANGE:
+    /* We deal with special padrange ops later, in the aassign op they belong
+     * to. */
+    if (o->op_ppaddr != lt_pp_padrange && o->op_private & OPpLVAL_INTRO
+                                       && !(o->op_flags & OPf_SPECIAL)) {
+     /* A padrange op is guaranteed to have previously been a pushmark.
+      * Moreover, for non-special padrange ops (i.e. that aren't for
+      * my (...) = @_), the first original padxv is its sibling or nephew.
+      */
+     OP *kid = OpSIBLING(o);
+     if (kid->op_type == OP_NULL && kid->op_flags & OPf_KIDS) {
+      kid = kUNOP->op_first;
+      if (kid->op_type == OP_NULL)
+       kid = OpSIBLING(kid);
+     }
+     lt_maybe_padrange_setup(o, kid);
+    }
+    break;
+   case OP_AASSIGN: {
+    OP *op;
+    if (cBINOPo->op_first && cBINOPo->op_first->op_flags & OPf_KIDS
+                          && (op = cUNOPx(cBINOPo->op_first)->op_first)
+                          && op->op_type == OP_PADRANGE
+                          && op->op_ppaddr != lt_pp_padrange
+                          && op->op_private & OPpLVAL_INTRO
+                          && op->op_flags & OPf_SPECIAL) {
+     const OP *start = cUNOPx(cBINOPo->op_last)->op_first;
+     if (start->op_type == OP_PUSHMARK)
+      start = OpSIBLING(start);
+     lt_maybe_padrange_setup(op, start);
+    }
+    break;
+   }
+#endif
+   default:
+    xsh_peep_maybe_recurse(o, seen);
+    break;
+  }
  }
 }
 
-SV *_tag(SV *ref)
-PREINIT:
- SV *ret;
+/* --- Module setup/teardown ----------------------------------------------- */
+
+static void xsh_user_global_setup(pTHX) {
+ lt_op_padxv_map    = ptable_new(32);
+#if XSH_HAS_PERL(5, 17, 6)
+ lt_op_padrange_map = ptable_new(32);
+#endif
+
+#ifdef USE_ITHREADS
+ MUTEX_INIT(&lt_op_map_mutex);
+#endif
+
+ xsh_ck_replace(OP_PADANY, lt_ck_padany, &lt_old_ck_padany);
+ xsh_ck_replace(OP_PADSV,  lt_ck_padsv,  &lt_old_ck_padsv);
+
+ return;
+}
+
+static void xsh_user_local_setup(pTHX_ xsh_user_cxt_t *cxt) {
+ HV *stash;
+
+ stash = gv_stashpvn(XSH_PACKAGE, XSH_PACKAGE_LEN, 1);
+ newCONSTSUB(stash, "LT_THREADSAFE", newSVuv(XSH_THREADSAFE));
+ newCONSTSUB(stash, "LT_FORKSAFE",   newSVuv(XSH_FORKSAFE));
+
+ cxt->default_meth = newSVpvn("TYPEDSCALAR", 11);
+ SvREADONLY_on(cxt->default_meth);
+
+ return;
+}
+
+static void xsh_user_local_teardown(pTHX_ xsh_user_cxt_t *cxt) {
+ SvREFCNT_dec(cxt->default_meth);
+ cxt->default_meth = NULL;
+
+ return;
+}
+
+static void xsh_user_global_teardown(pTHX) {
+ xsh_ck_restore(OP_PADANY, &lt_old_ck_padany);
+ xsh_ck_restore(OP_PADSV,  &lt_old_ck_padsv);
+
+ ptable_map_free(lt_op_padxv_map);
+ lt_op_padxv_map    = NULL;
+
+#if XSH_HAS_PERL(5, 17, 6)
+ ptable_map_free(lt_op_padrange_map);
+ lt_op_padrange_map = NULL;
+#endif
+
+#ifdef USE_ITHREADS
+ MUTEX_DESTROY(&lt_op_map_mutex);
+#endif
+
+ return;
+}
+
+/* --- XS ------------------------------------------------------------------ */
+
+MODULE = Lexical::Types      PACKAGE = Lexical::Types
+
+PROTOTYPES: ENABLE
+
+BOOT:
+{
+ xsh_setup();
+}
+
+#if XSH_THREADSAFE
+
+void
+CLONE(...)
+PROTOTYPE: DISABLE
+PPCODE:
+ xsh_clone();
+ XSRETURN(0);
+
+#endif
+
+SV *
+_tag(SV *code)
+PROTOTYPE: $
 CODE:
- if (SvOK(ref) && SvROK(ref)) {
-  SV *sv = SvRV(ref);
-  SvREFCNT_inc(sv);
-  ret = newSVuv(PTR2UV(sv));
- } else {
-  ret = newSVuv(0);
- }
- RETVAL = ret;
+ if (!SvOK(code))
+  code = NULL;
+ else if (SvROK(code))
+  code = SvRV(code);
+ RETVAL = xsh_hints_tag(code);
 OUTPUT:
  RETVAL