]> git.vpit.fr Git - perl/modules/autovivification.git/commitdiff
Revamp module setup/teardown
authorVincent Pit <vince@profvince.com>
Thu, 14 May 2015 14:25:45 +0000 (16:25 +0200)
committerVincent Pit <vince@profvince.com>
Thu, 14 May 2015 14:51:27 +0000 (16:51 +0200)
This fixes crashes and bugs when loading the pragma in several threads run
in parallel.

MANIFEST
autovivification.xs
t/09-load-threads.t [new file with mode: 0644]

index 8ba1106d8e525705fc9fe472f020b690d2b9efe1..152f4bc93a57b03961b10deca049f1ae0bf99010 100644 (file)
--- a/MANIFEST
+++ b/MANIFEST
@@ -10,6 +10,7 @@ ptable.h
 samples/bench.pl
 samples/hash2array.pl
 t/00-load.t
 samples/bench.pl
 samples/hash2array.pl
 t/00-load.t
+t/09-load-threads.t
 t/20-hash.t
 t/22-hash-kv.t
 t/23-hash-tied.t
 t/20-hash.t
 t/22-hash-kv.t
 t/23-hash-tied.t
index 9577d87547cbe1170b45919eddd0ae0ea78b8635..23e20729b5db4e3c8b78c08a42c00e219de3b56a 100644 (file)
 # define MY_CXT_CLONE NOOP
 #endif
 
 # define MY_CXT_CLONE NOOP
 #endif
 
+#if A_THREADSAFE
+/* We must use preexistent global mutexes or we will never be able to destroy
+ * them. */
+# if A_HAS_PERL(5, 9, 3)
+#  define A_LOADED_LOCK   MUTEX_LOCK(&PL_my_ctx_mutex)
+#  define A_LOADED_UNLOCK MUTEX_UNLOCK(&PL_my_ctx_mutex)
+# else
+#  define A_LOADED_LOCK   OP_REFCNT_LOCK
+#  define A_LOADED_UNLOCK OP_REFCNT_UNLOCK
+# endif
+#else
+# define A_LOADED_LOCK   NOOP
+# define A_LOADED_UNLOCK NOOP
+#endif
+
 #if defined(OP_CHECK_MUTEX_LOCK) && defined(OP_CHECK_MUTEX_UNLOCK)
 # define A_CHECK_LOCK   OP_CHECK_MUTEX_LOCK
 # define A_CHECK_UNLOCK OP_CHECK_MUTEX_UNLOCK
 #if defined(OP_CHECK_MUTEX_LOCK) && defined(OP_CHECK_MUTEX_UNLOCK)
 # define A_CHECK_LOCK   OP_CHECK_MUTEX_LOCK
 # define A_CHECK_UNLOCK OP_CHECK_MUTEX_UNLOCK
-#else
+#elif A_HAS_PERL(5, 9, 3)
 # define A_CHECK_LOCK   OP_REFCNT_LOCK
 # define A_CHECK_UNLOCK OP_REFCNT_UNLOCK
 # define A_CHECK_LOCK   OP_REFCNT_LOCK
 # define A_CHECK_UNLOCK OP_REFCNT_UNLOCK
+#else
+/* Before perl 5.9.3, indirect_ck_*() calls are already protected by the
+ * A_LOADED mutex, which falls back to the OP_REFCNT mutex. Make sure we don't
+ * lock it twice. */
+# define A_CHECK_LOCK   NOOP
+# define A_CHECK_UNLOCK NOOP
 #endif
 
 typedef OP *(*a_ck_t)(pTHX_ OP *);
 #endif
 
 typedef OP *(*a_ck_t)(pTHX_ OP *);
@@ -132,6 +153,79 @@ static void a_ck_restore(pTHX_ OPCODE type, a_ck_t *old_ck_p) {
 
 /* --- Helpers ------------------------------------------------------------- */
 
 
 /* --- Helpers ------------------------------------------------------------- */
 
+/* ... Check if the module is loaded ....................................... */
+
+static I32 a_loaded = 0;
+
+#if A_THREADSAFE
+
+#define PTABLE_NAME        ptable_loaded
+#define PTABLE_VAL_FREE(V) NOOP
+
+#include "ptable.h"
+
+#define ptable_loaded_store(T, K, V) ptable_loaded_store(aPTBLMS_ (T), (K), (V))
+#define ptable_loaded_delete(T, K)   ptable_loaded_delete(aPTBLMS_ (T), (K))
+#define ptable_loaded_free(T)        ptable_loaded_free(aPTBLMS_ (T))
+
+static ptable *a_loaded_cxts = NULL;
+
+static int a_is_loaded(pTHX_ void *cxt) {
+#define a_is_loaded(C) a_is_loaded(aTHX_ (C))
+ int res = 0;
+
+ A_LOADED_LOCK;
+ if (a_loaded_cxts && ptable_fetch(a_loaded_cxts, cxt))
+  res = 1;
+ A_LOADED_UNLOCK;
+
+ return res;
+}
+
+static int a_set_loaded_locked(pTHX_ void *cxt) {
+#define a_set_loaded_locked(C) a_set_loaded_locked(aTHX_ (C))
+ int global_setup = 0;
+
+ if (a_loaded <= 0) {
+  assert(a_loaded == 0);
+  assert(!a_loaded_cxts);
+  a_loaded_cxts = ptable_new();
+  global_setup  = 1;
+ }
+ ++a_loaded;
+ assert(a_loaded_cxts);
+ ptable_loaded_store(a_loaded_cxts, cxt, cxt);
+
+ return global_setup;
+}
+
+static int a_clear_loaded_locked(pTHX_ void *cxt) {
+#define a_clear_loaded_locked(C) a_clear_loaded_locked(aTHX_ (C))
+ int global_teardown = 0;
+
+ if (a_loaded > 1) {
+  assert(a_loaded_cxts);
+  ptable_loaded_delete(a_loaded_cxts, cxt);
+  --a_loaded;
+ } else if (a_loaded_cxts) {
+  assert(a_loaded == 1);
+  ptable_loaded_free(a_loaded_cxts);
+  a_loaded_cxts   = NULL;
+  a_loaded        = 0;
+  global_teardown = 1;
+ }
+
+ return global_teardown;
+}
+
+#else
+
+#define a_is_loaded(C)           (a_loaded > 0)
+#define a_set_loaded_locked(C)   ((a_loaded++ <= 0) ? 1 : 0)
+#define a_clear_loaded_locked(C) ((--a_loaded <= 0) ? 1 : 0)
+
+#endif
+
 /* ... Thread-safe hints ................................................... */
 
 #if A_WORKAROUND_REQUIRE_PROPAGATION
 /* ... Thread-safe hints ................................................... */
 
 #if A_WORKAROUND_REQUIRE_PROPAGATION
@@ -175,19 +269,20 @@ typedef struct {
 #define MY_CXT_KEY __PACKAGE__ "::_guts" XS_VERSION
 
 typedef struct {
 #define MY_CXT_KEY __PACKAGE__ "::_guts" XS_VERSION
 
 typedef struct {
+ peep_t  old_peep; /* This is actually the rpeep past 5.13.5 */
+ ptable *seen;     /* It really is a ptable_seen */
 #if A_THREADSAFE && A_WORKAROUND_REQUIRE_PROPAGATION
 #if A_THREADSAFE && A_WORKAROUND_REQUIRE_PROPAGATION
- ptable *tbl;   /* It really is a ptable_hints */
+ ptable *tbl;      /* It really is a ptable_hints */
  tTHX    owner;
 #endif /* A_THREADSAFE && A_WORKAROUND_REQUIRE_PROPAGATION */
  tTHX    owner;
 #endif /* A_THREADSAFE && A_WORKAROUND_REQUIRE_PROPAGATION */
- ptable *seen;  /* It really is a ptable_seen */
 } my_cxt_t;
 
 START_MY_CXT
 
 } my_cxt_t;
 
 START_MY_CXT
 
-#if A_THREADSAFE
-
 #if A_WORKAROUND_REQUIRE_PROPAGATION
 
 #if A_WORKAROUND_REQUIRE_PROPAGATION
 
+#if A_THREADSAFE
+
 typedef struct {
  ptable       *tbl;
 #if A_HAS_PERL(5, 13, 2)
 typedef struct {
  ptable       *tbl;
 #if A_HAS_PERL(5, 13, 2)
@@ -225,46 +320,8 @@ static void a_ptable_clone(pTHX_ ptable_ent *ent, void *ud_) {
  ptable_hints_store(ud->tbl, ent->key, h2);
 }
 
  ptable_hints_store(ud->tbl, ent->key, h2);
 }
 
-#endif /* A_WORKAROUND_REQUIRE_PROPAGATION */
-
-static void a_thread_cleanup(pTHX_ void *ud) {
- dMY_CXT;
-
-#if A_WORKAROUND_REQUIRE_PROPAGATION
- ptable_hints_free(MY_CXT.tbl);
- MY_CXT.tbl  = NULL;
-#endif /* A_WORKAROUND_REQUIRE_PROPAGATION */
- ptable_seen_free(MY_CXT.seen);
- MY_CXT.seen = NULL;
-}
-
-static int a_endav_free(pTHX_ SV *sv, MAGIC *mg) {
- SAVEDESTRUCTOR_X(a_thread_cleanup, NULL);
-
- return 0;
-}
-
-static MGVTBL a_endav_vtbl = {
- 0,
- 0,
- 0,
- 0,
- a_endav_free
-#if MGf_COPY
- , 0
-#endif
-#if MGf_DUP
- , 0
-#endif
-#if MGf_LOCAL
- , 0
-#endif
-};
-
 #endif /* A_THREADSAFE */
 
 #endif /* A_THREADSAFE */
 
-#if A_WORKAROUND_REQUIRE_PROPAGATION
-
 static IV a_require_tag(pTHX) {
 #define a_require_tag() a_require_tag(aTHX)
  const CV *cv, *outside;
 static IV a_require_tag(pTHX) {
 #define a_require_tag() a_require_tag(aTHX)
  const CV *cv, *outside;
@@ -390,7 +447,7 @@ static UV a_detag(pTHX_ const SV *hint) {
 #define A_HINT_ROOT   64
 #define A_HINT_DEREF  128
 
 #define A_HINT_ROOT   64
 #define A_HINT_DEREF  128
 
-static U32 a_hash = 0;
+static VOL U32 a_hash = 0;
 
 static UV a_hint(pTHX) {
 #define a_hint() a_hint(aTHX)
 
 static UV a_hint(pTHX) {
 #define a_hint() a_hint(aTHX)
@@ -1335,8 +1392,6 @@ static OP *a_ck_root(pTHX_ OP *o) {
 
 /* ... Our peephole optimizer .............................................. */
 
 
 /* ... Our peephole optimizer .............................................. */
 
-static peep_t a_old_peep = 0; /* This is actually the rpeep past 5.13.5 */
-
 static void a_peep_rec(pTHX_ OP *o, ptable *seen);
 
 static void a_peep_rec(pTHX_ OP *o, ptable *seen) {
 static void a_peep_rec(pTHX_ OP *o, ptable *seen);
 
 static void a_peep_rec(pTHX_ OP *o, ptable *seen) {
@@ -1460,11 +1515,14 @@ static void a_peep_rec(pTHX_ OP *o, ptable *seen) {
 }
 
 static void a_peep(pTHX_ OP *o) {
 }
 
 static void a_peep(pTHX_ OP *o) {
+ ptable *seen;
  dMY_CXT;
  dMY_CXT;
- ptable *seen = MY_CXT.seen;
 
 
- a_old_peep(aTHX_ o);
+ assert(a_is_loaded(&MY_CXT));
+
+ MY_CXT.old_peep(aTHX_ o);
 
 
+ seen = MY_CXT.seen;
  if (seen) {
   ptable_seen_clear(seen);
   a_peep_rec(o);
  if (seen) {
   ptable_seen_clear(seen);
   a_peep_rec(o);
@@ -1472,127 +1530,96 @@ static void a_peep(pTHX_ OP *o) {
  }
 }
 
  }
 }
 
-/* --- Interpreter setup/teardown ------------------------------------------ */
-
-static U32 a_initialized = 0;
+/* --- Module setup/teardown ----------------------------------------------- */
 
 static void a_teardown(pTHX_ void *root) {
 
 static void a_teardown(pTHX_ void *root) {
+ dMY_CXT;
 
 
- if (!a_initialized)
-  return;
+ A_LOADED_LOCK;
 
 
-#if A_MULTIPLICITY
- if (aTHX != root)
-  return;
-#endif
+ if (a_clear_loaded_locked(&MY_CXT)) {
+  a_ck_restore(OP_PADANY, &a_old_ck_padany);
+  a_ck_restore(OP_PADSV,  &a_old_ck_padsv);
 
 
- {
-  dMY_CXT;
-# if A_THREADSAFE && A_WORKAROUND_REQUIRE_PROPAGATION
-  ptable_hints_free(MY_CXT.tbl);
-  MY_CXT.tbl  = NULL;
-# endif /* A_THREADSAFE && A_WORKAROUND_REQUIRE_PROPAGATION */
-  ptable_seen_free(MY_CXT.seen);
-  MY_CXT.seen = NULL;
- }
+  a_ck_restore(OP_AELEM,  &a_old_ck_aelem);
+  a_ck_restore(OP_HELEM,  &a_old_ck_helem);
+  a_ck_restore(OP_RV2SV,  &a_old_ck_rv2sv);
 
 
a_ck_restore(OP_PADANY, &a_old_ck_padany);
a_ck_restore(OP_PADSV,  &a_old_ck_padsv);
 a_ck_restore(OP_RV2AV,  &a_old_ck_rv2av);
 a_ck_restore(OP_RV2HV,  &a_old_ck_rv2hv);
 
 
- a_ck_restore(OP_AELEM,  &a_old_ck_aelem);
- a_ck_restore(OP_HELEM,  &a_old_ck_helem);
- a_ck_restore(OP_RV2SV,  &a_old_ck_rv2sv);
+  a_ck_restore(OP_ASLICE, &a_old_ck_aslice);
+  a_ck_restore(OP_HSLICE, &a_old_ck_hslice);
 
 
- a_ck_restore(OP_RV2AV,  &a_old_ck_rv2av);
- a_ck_restore(OP_RV2HV,  &a_old_ck_rv2hv);
+  a_ck_restore(OP_EXISTS, &a_old_ck_exists);
+  a_ck_restore(OP_DELETE, &a_old_ck_delete);
+  a_ck_restore(OP_KEYS,   &a_old_ck_keys);
+  a_ck_restore(OP_VALUES, &a_old_ck_values);
 
 
- a_ck_restore(OP_ASLICE, &a_old_ck_aslice);
- a_ck_restore(OP_HSLICE, &a_old_ck_hslice);
+  ptable_map_free(a_op_map);
+  a_op_map = NULL;
+ }
 
 
- a_ck_restore(OP_EXISTS, &a_old_ck_exists);
- a_ck_restore(OP_DELETE, &a_old_ck_delete);
- a_ck_restore(OP_KEYS,   &a_old_ck_keys);
- a_ck_restore(OP_VALUES, &a_old_ck_values);
+ A_LOADED_UNLOCK;
 
 
+ if (MY_CXT.old_peep) {
 #if A_HAS_RPEEP
 #if A_HAS_RPEEP
PL_rpeepp  = a_old_peep;
 PL_rpeepp = MY_CXT.old_peep;
 #else
 #else
PL_peepp   = a_old_peep;
 PL_peepp  = MY_CXT.old_peep;
 #endif
 #endif
- a_old_peep = 0;
+  MY_CXT.old_peep = 0;
+ }
+
+ ptable_seen_free(MY_CXT.seen);
+ MY_CXT.seen = NULL;
+
+#if A_THREADSAFE && A_WORKAROUND_REQUIRE_PROPAGATION
+ ptable_hints_free(MY_CXT.tbl);
+ MY_CXT.tbl  = NULL;
+#endif /* A_THREADSAFE && A_WORKAROUND_REQUIRE_PROPAGATION */
 
 
a_initialized = 0;
return;
 }
 
 static void a_setup(pTHX) {
 #define a_setup() a_setup(aTHX)
 }
 
 static void a_setup(pTHX) {
 #define a_setup() a_setup(aTHX)
- if (a_initialized)
-  return;
-
- {
-  MY_CXT_INIT;
-# if A_THREADSAFE && A_WORKAROUND_REQUIRE_PROPAGATION
-  MY_CXT.tbl   = ptable_new();
-  MY_CXT.owner = aTHX;
-# endif /* A_THREADSAFE && A_WORKAROUND_REQUIRE_PROPAGATION */
-  MY_CXT.seen  = ptable_new();
- }
-
- a_ck_replace(OP_PADANY, a_ck_padany, &a_old_ck_padany);
- a_ck_replace(OP_PADSV,  a_ck_padsv,  &a_old_ck_padsv);
+ MY_CXT_INIT; /* Takes/release PL_my_ctx_mutex */
 
 
- a_ck_replace(OP_AELEM,  a_ck_deref,  &a_old_ck_aelem);
- a_ck_replace(OP_HELEM,  a_ck_deref,  &a_old_ck_helem);
- a_ck_replace(OP_RV2SV,  a_ck_deref,  &a_old_ck_rv2sv);
+ A_LOADED_LOCK;
 
 
- a_ck_replace(OP_RV2AV,  a_ck_rv2xv,  &a_old_ck_rv2av);
- a_ck_replace(OP_RV2HV,  a_ck_rv2xv,  &a_old_ck_rv2hv);
-
- a_ck_replace(OP_ASLICE, a_ck_xslice, &a_old_ck_aslice);
- a_ck_replace(OP_HSLICE, a_ck_xslice, &a_old_ck_hslice);
-
- a_ck_replace(OP_EXISTS, a_ck_root,   &a_old_ck_exists);
- a_ck_replace(OP_DELETE, a_ck_root,   &a_old_ck_delete);
- a_ck_replace(OP_KEYS,   a_ck_root,   &a_old_ck_keys);
- a_ck_replace(OP_VALUES, a_ck_root,   &a_old_ck_values);
+ if (a_set_loaded_locked(&MY_CXT)) {
+  PERL_HASH(a_hash, __PACKAGE__, __PACKAGE_LEN__);
 
 
-#if A_HAS_RPEEP
- a_old_peep = PL_rpeepp;
- PL_rpeepp  = a_peep;
-#else
- a_old_peep = PL_peepp;
- PL_peepp   = a_peep;
+  a_op_map = ptable_new();
+#ifdef USE_ITHREADS
+  MUTEX_INIT(&a_op_map_mutex);
 #endif
 
 #endif
 
-#if A_MULTIPLICITY
- call_atexit(a_teardown, aTHX);
-#else
- call_atexit(a_teardown, NULL);
-#endif
+  a_ck_replace(OP_PADANY, a_ck_padany, &a_old_ck_padany);
+  a_ck_replace(OP_PADSV,  a_ck_padsv,  &a_old_ck_padsv);
 
 
- a_initialized = 1;
-}
+  a_ck_replace(OP_AELEM,  a_ck_deref,  &a_old_ck_aelem);
+  a_ck_replace(OP_HELEM,  a_ck_deref,  &a_old_ck_helem);
+  a_ck_replace(OP_RV2SV,  a_ck_deref,  &a_old_ck_rv2sv);
 
 
-static U32 a_booted = 0;
+  a_ck_replace(OP_RV2AV,  a_ck_rv2xv,  &a_old_ck_rv2av);
+  a_ck_replace(OP_RV2HV,  a_ck_rv2xv,  &a_old_ck_rv2hv);
 
 
-/* --- XS ------------------------------------------------------------------ */
+  a_ck_replace(OP_ASLICE, a_ck_xslice, &a_old_ck_aslice);
+  a_ck_replace(OP_HSLICE, a_ck_xslice, &a_old_ck_hslice);
 
 
-MODULE = autovivification      PACKAGE = autovivification
+  a_ck_replace(OP_EXISTS, a_ck_root,   &a_old_ck_exists);
+  a_ck_replace(OP_DELETE, a_ck_root,   &a_old_ck_delete);
+  a_ck_replace(OP_KEYS,   a_ck_root,   &a_old_ck_keys);
+  a_ck_replace(OP_VALUES, a_ck_root,   &a_old_ck_values);
+ }
 
 
-PROTOTYPES: ENABLE
+ A_LOADED_UNLOCK;
 
 
-BOOT:
-{
- if (!a_booted++) {
+ {
   HV *stash;
 
   HV *stash;
 
-  a_op_map = ptable_new();
-#ifdef USE_ITHREADS
-  MUTEX_INIT(&a_op_map_mutex);
-#endif
-
-  PERL_HASH(a_hash, __PACKAGE__, __PACKAGE_LEN__);
-
   stash = gv_stashpvn(__PACKAGE__, __PACKAGE_LEN__, 1);
   newCONSTSUB(stash, "A_HINT_STRICT", newSVuv(A_HINT_STRICT));
   newCONSTSUB(stash, "A_HINT_WARN",   newSVuv(A_HINT_WARN));
   stash = gv_stashpvn(__PACKAGE__, __PACKAGE_LEN__, 1);
   newCONSTSUB(stash, "A_HINT_STRICT", newSVuv(A_HINT_STRICT));
   newCONSTSUB(stash, "A_HINT_WARN",   newSVuv(A_HINT_WARN));
@@ -1605,6 +1632,41 @@ BOOT:
   newCONSTSUB(stash, "A_FORKSAFE",    newSVuv(A_FORKSAFE));
  }
 
   newCONSTSUB(stash, "A_FORKSAFE",    newSVuv(A_FORKSAFE));
  }
 
+#if A_HAS_RPEEP
+ if (PL_rpeepp != a_peep) {
+  MY_CXT.old_peep = PL_rpeepp;
+  PL_rpeepp       = a_peep;
+ }
+#else
+ if (PL_peepp != a_peep) {
+  MY_CXT.old_peep = PL_peepp;
+  PL_peepp        = a_peep;
+ }
+#endif
+ else {
+  MY_CXT.old_peep = 0;
+ }
+
+ MY_CXT.seen = ptable_new();
+
+#if A_THREADSAFE && A_WORKAROUND_REQUIRE_PROPAGATION
+ MY_CXT.tbl   = ptable_new();
+ MY_CXT.owner = aTHX;
+#endif /* A_THREADSAFE && A_WORKAROUND_REQUIRE_PROPAGATION */
+
+ call_atexit(a_teardown, NULL);
+
+ return;
+}
+
+/* --- XS ------------------------------------------------------------------ */
+
+MODULE = autovivification      PACKAGE = autovivification
+
+PROTOTYPES: ENABLE
+
+BOOT:
+{
  a_setup();
 }
 
  a_setup();
 }
 
@@ -1617,50 +1679,34 @@ PREINIT:
 #if A_WORKAROUND_REQUIRE_PROPAGATION
  ptable *t;
 #endif
 #if A_WORKAROUND_REQUIRE_PROPAGATION
  ptable *t;
 #endif
- ptable *s;
- GV     *gv;
 PPCODE:
 PPCODE:
- {
 #if A_WORKAROUND_REQUIRE_PROPAGATION
 #if A_WORKAROUND_REQUIRE_PROPAGATION
+ {
+  a_ptable_clone_ud ud;
   dMY_CXT;
   dMY_CXT;
-  {
-   a_ptable_clone_ud ud;
-
-   t = ptable_new();
-   a_ptable_clone_ud_init(ud, t, MY_CXT.owner);
-   ptable_walk(MY_CXT.tbl, a_ptable_clone, &ud);
-   a_ptable_clone_ud_deinit(ud);
-  }
-#endif
-  s = ptable_new();
+  t = ptable_new();
+  a_ptable_clone_ud_init(ud, t, MY_CXT.owner);
+  ptable_walk(MY_CXT.tbl, a_ptable_clone, &ud);
+  a_ptable_clone_ud_deinit(ud);
  }
  }
+#endif
  {
   MY_CXT_CLONE;
 #if A_WORKAROUND_REQUIRE_PROPAGATION
   MY_CXT.tbl   = t;
   MY_CXT.owner = aTHX;
 #endif
  {
   MY_CXT_CLONE;
 #if A_WORKAROUND_REQUIRE_PROPAGATION
   MY_CXT.tbl   = t;
   MY_CXT.owner = aTHX;
 #endif
-  MY_CXT.seen  = s;
- }
- gv = gv_fetchpv(__PACKAGE__ "::_THREAD_CLEANUP", 0, SVt_PVCV);
- if (gv) {
-  CV *cv = GvCV(gv);
-  if (!PL_endav)
-   PL_endav = newAV();
-  SvREFCNT_inc(cv);
-  if (!av_store(PL_endav, av_len(PL_endav) + 1, (SV *) cv))
-   SvREFCNT_dec(cv);
-  sv_magicext((SV *) PL_endav, NULL, PERL_MAGIC_ext, &a_endav_vtbl, NULL, 0);
+  MY_CXT.seen  = ptable_new();
+  {
+   int global_setup;
+   A_LOADED_LOCK;
+   global_setup = a_set_loaded_locked(&MY_CXT);
+   assert(!global_setup);
+   A_LOADED_UNLOCK;
+  }
  }
  XSRETURN(0);
 
  }
  XSRETURN(0);
 
-void
-_THREAD_CLEANUP(...)
-PROTOTYPE: DISABLE
-PPCODE:
- a_thread_cleanup(aTHX_ NULL);
- XSRETURN(0);
-
 #endif /* A_THREADSAFE */
 
 SV *
 #endif /* A_THREADSAFE */
 
 SV *
diff --git a/t/09-load-threads.t b/t/09-load-threads.t
new file mode 100644 (file)
index 0000000..ddba1b9
--- /dev/null
@@ -0,0 +1,434 @@
+#!perl
+
+use strict;
+use warnings;
+
+my ($module, $thread_safe_var);
+BEGIN {
+ $module          = 'autovivification';
+ $thread_safe_var = 'autovivification::A_THREADSAFE()';
+}
+
+sub load_test {
+ my $x;
+ if (defined &autovivification::unimport) {
+  local $@;
+  eval 'BEGIN { autovivification->unimport } my $y = $x->[0]';
+  $x = $@ if $@;
+ } else {
+  $x = '';
+ }
+ if (not defined $x) {
+  return 1;
+ } elsif (   (not ref $x        and not length $x)
+          or (ref $x eq 'ARRAY' and not @$x      )) {
+  return 0;
+ } else {
+  return "$x";
+ }
+}
+
+# Keep the rest of the file untouched
+
+use lib 't/lib';
+use VPIT::TestHelpers threads => [ $module, $thread_safe_var ];
+
+my $could_not_create_thread = 'Could not create thread';
+
+use Test::Leaner (
+ tests => 1 + (2 + 2 * 2) + 6 + (2 * 4 + 1) * 2 + (4 + 1) + (6 + 1) + 1
+);
+
+sub is_loaded {
+ my ($affirmative, $desc) = @_;
+
+ my $res = load_test();
+
+ my $expected;
+ if ($affirmative) {
+  $expected = 1;
+  $desc     = "$desc: module loaded";
+ } else {
+  $expected = 0;
+  $desc     = "$desc: module not loaded";
+ }
+
+ unless (is $res, $expected, $desc) {
+  $res      = defined $res ? "'$res'" : 'undef';
+  $expected = "'$expected'";
+  diag("Test '$desc' failed: got $res, expected $expected");
+ }
+
+ return;
+}
+
+BEGIN {
+ local $@;
+ my $code = eval "sub { require $module }";
+ die $@ if $@;
+ *do_load = $code;
+}
+
+is_loaded 0, 'main body, beginning';
+
+# Test serial loadings
+
+SKIP: {
+ my $thr = spawn(sub {
+  my $here = "first serial thread";
+  is_loaded 0, "$here, beginning";
+
+  do_load;
+  is_loaded 1, "$here, after loading";
+
+  return;
+ });
+
+ skip "$could_not_create_thread (serial 1)" => 2 unless defined $thr;
+
+ $thr->join;
+ if (my $err = $thr->error) {
+  die $err;
+ }
+}
+
+is_loaded 0, 'main body, in between serial loadings';
+
+SKIP: {
+ my $thr = spawn(sub {
+  my $here = "second serial thread";
+  is_loaded 0, "$here, beginning";
+
+  do_load;
+  is_loaded 1, "$here, after loading";
+
+  return;
+ });
+
+ skip "$could_not_create_thread (serial 2)" => 2 unless defined $thr;
+
+ $thr->join;
+ if (my $err = $thr->error) {
+  die $err;
+ }
+}
+
+is_loaded 0, 'main body, after serial loadings';
+
+# Test nested loadings
+
+SKIP: {
+ my $parent = spawn(sub {
+  my $here = 'parent thread';
+  is_loaded 0, "$here, beginning";
+
+  SKIP: {
+   my $kid = spawn(sub {
+    my $here = 'child thread';
+    is_loaded 0, "$here, beginning";
+
+    do_load;
+    is_loaded 1, "$here, after loading";
+
+    return;
+   });
+
+   skip "$could_not_create_thread (nested child)" => 2 unless defined $kid;
+
+   $kid->join;
+   if (my $err = $kid->error) {
+    die "in child thread: $err\n";
+   }
+  }
+
+  is_loaded 0, "$here, after child terminated";
+
+  do_load;
+  is_loaded 1, "$here, after loading";
+
+  return;
+ });
+
+ skip "$could_not_create_thread (nested parent)" => (3 + 2)
+                                                         unless defined $parent;
+
+ $parent->join;
+ if (my $err = $parent->error) {
+  die $err;
+ }
+}
+
+is_loaded 0, 'main body, after nested loadings';
+
+# Test parallel loadings
+
+use threads;
+use threads::shared;
+
+my $sync_points = 7;
+
+my @locks_down = (1) x $sync_points;
+my @locks_up   = (0) x $sync_points;
+share($_) for @locks_down, @locks_up;
+
+my $default_peers = 2;
+
+sub sync_master {
+ my ($id, $peers) = @_;
+
+ $peers = $default_peers unless defined $peers;
+
+ {
+  lock $locks_down[$id];
+  $locks_down[$id] = 0;
+  cond_broadcast $locks_down[$id];
+ }
+
+ {
+  lock $locks_up[$id];
+  cond_wait $locks_up[$id] until $locks_up[$id] == $peers;
+ }
+}
+
+sub sync_slave {
+ my ($id) = @_;
+
+ {
+  lock $locks_down[$id];
+  cond_wait $locks_down[$id] until $locks_down[$id] == 0;
+ }
+
+ {
+  lock $locks_up[$id];
+  $locks_up[$id]++;
+  cond_signal $locks_up[$id];
+ }
+}
+
+for my $first_thread_ends_first (0, 1) {
+ for my $id (0 .. $sync_points - 1) {
+  {
+   lock $locks_down[$id];
+   $locks_down[$id] = 1;
+  }
+  {
+   lock $locks_up[$id];
+   $locks_up[$id] = 0;
+  }
+ }
+
+ my $thr1_end = 'finishes first';
+ my $thr2_end = 'finishes last';
+
+ ($thr1_end, $thr2_end) = ($thr2_end, $thr1_end)
+                                                unless $first_thread_ends_first;
+
+ SKIP: {
+  my $thr1 = spawn(sub {
+   my $here = "first simultaneous thread ($thr1_end)";
+   sync_slave 0;
+
+   is_loaded 0, "$here, beginning";
+   sync_slave 1;
+
+   do_load;
+   is_loaded 1, "$here, after loading";
+   sync_slave 2;
+   sync_slave 3;
+
+   sync_slave 4;
+   is_loaded 1, "$here, still loaded while also loaded in the other thread";
+   sync_slave 5;
+
+   sync_slave 6 unless $first_thread_ends_first;
+
+   is_loaded 1, "$here, end";
+
+   return;
+  });
+
+  skip "$could_not_create_thread (parallel 1)" => (4 * 2) unless defined $thr1;
+
+  my $thr2 = spawn(sub {
+   my $here = "second simultaneous thread ($thr2_end)";
+   sync_slave 0;
+
+   is_loaded 0, "$here, beginning";
+   sync_slave 1;
+
+   sync_slave 2;
+   sync_slave 3;
+   is_loaded 0, "$here, loaded in other thread but not here";
+
+   do_load;
+   is_loaded 1, "$here, after loading";
+   sync_slave 4;
+   sync_slave 5;
+
+   sync_slave 6 if $first_thread_ends_first;
+
+   is_loaded 1, "$here, end";
+
+   return;
+  });
+
+  sync_master($_) for 0 .. 5;
+
+  if (defined $thr2) {
+   ($thr2, $thr1) = ($thr1, $thr2) unless $first_thread_ends_first;
+
+   $thr1->join;
+   if (my $err = $thr1->error) {
+    die $err;
+   }
+
+   sync_master(6, 1);
+
+   $thr2->join;
+   if (my $err = $thr1->error) {
+    die $err;
+   }
+  } else {
+   sync_master(6, 1) unless $first_thread_ends_first;
+
+   $thr1->join;
+   if (my $err = $thr1->error) {
+    die $err;
+   }
+
+   skip "$could_not_create_thread (parallel 2)" => (4 * 1);
+  }
+ }
+
+ is_loaded 0, 'main body, after simultaneous threads';
+}
+
+# Test simple clone
+
+SKIP: {
+ my $parent = spawn(sub {
+  my $here = 'simple clone, parent thread';
+  is_loaded 0, "$here, beginning";
+
+  do_load;
+  is_loaded 1, "$here, after loading";
+
+  SKIP: {
+   my $kid = spawn(sub {
+    my $here = 'simple clone, child thread';
+
+    is_loaded 1, "$here, beginning";
+
+    return;
+   });
+
+   skip "$could_not_create_thread (simple clone child)" => 1
+                                                            unless defined $kid;
+
+   $kid->join;
+   if (my $err = $kid->error) {
+    die "in child thread: $err\n";
+   }
+  }
+
+  is_loaded 1, "$here, after child terminated";
+
+  return;
+ });
+
+ skip "$could_not_create_thread (simple clone parent)" => (3 + 1)
+                                                         unless defined $parent;
+
+ $parent->join;
+ if (my $err = $parent->error) {
+  die $err;
+ }
+}
+
+is_loaded 0, 'main body, after simple clone';
+
+# Test clone outliving its parent
+
+SKIP: {
+ my $kid_tid;
+ share($kid_tid);
+
+ my $kid_done;
+ share($kid_done);
+
+ my $parent = spawn(sub {
+  my $here = 'outliving clone, parent thread';
+  is_loaded 0, "$here, beginning";
+
+  my $no_kid;
+
+  do_load;
+  is_loaded 1, "$here, after loading";
+
+  SKIP: {
+   my $kid = spawn(sub {
+    my $here = 'outliving clone, child thread';
+
+    is_loaded 1, "$here, beginning";
+
+    {
+     lock $kid_tid;
+     $kid_tid = threads->tid();
+     cond_signal $kid_tid;
+    }
+
+    is_loaded 1, "$here, kid tid was communicated";
+
+    {
+     lock $kid_done;
+     cond_wait $kid_done until $kid_done;
+    }
+
+    is_loaded 1, "$here, end";
+
+    return;
+   });
+
+   unless (defined $kid) {
+    $no_kid = 1;
+    skip "$could_not_create_thread (outliving clone child)" => 3;
+   }
+  }
+
+  is_loaded 1, "$here, end";
+
+  return $no_kid;
+ });
+
+ skip "$could_not_create_thread (outliving clone parent)" => (3 + 3)
+                                                         unless defined $parent;
+
+ my $no_kid = $parent->join;
+ if (my $err = $parent->error) {
+  die $err;
+ }
+
+ unless ($no_kid) {
+  my $tid = do {
+   lock $kid_tid;
+   cond_wait $kid_tid until defined $kid_tid;
+   $kid_tid;
+  };
+
+  my $kid = threads->object($tid);
+  if (defined $kid) {
+   {
+    lock $kid_done;
+    $kid_done = 1;
+    cond_signal $kid_done;
+   }
+
+   $kid->join;
+  }
+ }
+}
+
+is_loaded 0, 'main body, after outliving clone';
+
+do_load;
+is_loaded 1, 'main body, loaded at end';