[PATCH 06/11] Add a test for the TSD system

[Mike Hommey]

From: Mike Hommey <mh at glandium.org>

---
 Makefile.in  |    3 +-
 test/tsd.c   |  139 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 test/tsd.exp |    2 +
 3 files changed, 143 insertions(+), 1 deletion(-)
 create mode 100644 test/tsd.c
 create mode 100644 test/tsd.exp

diff --git a/Makefile.in b/Makefile.in
index 8a34928..7f5dcf9 100644
--- a/Makefile.in
+++ b/Makefile.in
@@ -89,7 +89,7 @@ DOCS := $(DOCS_HTML) $(DOCS_MAN3)
 CTESTS := $(srcroot)test/aligned_alloc.c $(srcroot)test/allocated.c \
 	$(srcroot)test/bitmap.c $(srcroot)test/mremap.c \
 	$(srcroot)test/posix_memalign.c $(srcroot)test/thread_arena.c \
-	$(srcroot)test/thread_tcache_enabled.c
+	$(srcroot)test/thread_tcache_enabled.c $(srcroot)test/tsd.c
 ifeq ($(enable_experimental), 1)
 CTESTS += $(srcroot)test/allocm.c $(srcroot)test/rallocm.c
 endif
@@ -156,6 +156,7 @@ $(STATIC_LIBS):
 	ar crus $@ $+
 
 $(objroot)test/bitmap$(EXE): $(objroot)src/bitmap.$(O)
+#$(objroot)test/tsd$(EXE): $(objroot)src/tsd.$(O)
 
 $(objroot)test/%$(EXE): $(objroot)test/%.$(O) $(objroot)src/util.$(O) $(DSOS)
 	@mkdir -p $(@D)
diff --git a/test/tsd.c b/test/tsd.c
new file mode 100644
index 0000000..a3fc913
--- /dev/null
+++ b/test/tsd.c
@@ -0,0 +1,139 @@
+#define	JEMALLOC_MANGLE
+#include "jemalloc_test.h"
+
+typedef unsigned int data_t;
+
+void data_cleanup(void *arg);
+malloc_tsd_protos(, data, data_t)
+malloc_tsd_externs(data, data_t)
+#define DATA_INIT 0
+malloc_tsd_data(, data, data_t, DATA_INIT)
+malloc_tsd_funcs(, data, data_t, DATA_INIT, data_cleanup)
+
+static data_t next_cleanup = 0;
+static int ran_cleanups = 0;
+
+static void
+_check_equal(data_t a, data_t b, const char *file, int line)
+{
+	if (a != b)
+		malloc_printf("Got %x. Expected %x. %s:%d\n",
+			a, b, file, line);
+}
+#define check_equal(a, b) \
+	_check_equal(a, b, __FILE__, __LINE__)
+
+void
+data_cleanup(void *arg)
+{
+	data_t *data = (data_t *)arg;
+
+	check_equal(*data, next_cleanup);
+
+	if (ran_cleanups++)
+		return;
+
+	check_equal(*data_tsd_get(), DATA_INIT);
+	*data = 0x3b384d0f;
+	check_equal(*data_tsd_get(), DATA_INIT);
+
+	data_tsd_set(data);
+	next_cleanup = *data;
+	check_equal(*data_tsd_get(), *data);
+}
+
+void *
+thread_start(void *arg)
+{
+	data_t d = (data_t)(uintptr_t) arg;
+
+	if (d == 0)
+		return NULL;
+
+	check_equal(*data_tsd_get(), DATA_INIT);
+
+	if (d == 1)
+		return NULL;
+
+	data_tsd_set(&d);
+	next_cleanup = d;
+	check_equal(*data_tsd_get(), d);
+
+	d = 0;
+	check_equal(*data_tsd_get(), (data_t)(uintptr_t) arg);
+
+	return NULL;
+}
+
+int
+main(void)
+{
+	je_thread_t thread;
+	data_t d = 0xa5f3e329;
+
+	malloc_printf("Test begin\n");
+
+	data_tsd_boot();
+	thread_start((void *)(uintptr_t) d);
+
+	je_thread_create(&thread, thread_start, (void *) 0x72b65c10);
+	je_thread_join(thread, NULL);
+
+	check_equal(*data_tsd_get(), d);
+
+	je_thread_create(&thread, thread_start, NULL);
+	je_thread_join(thread, NULL);
+
+	je_thread_create(&thread, thread_start, (void *) 1);
+	je_thread_join(thread, NULL);
+
+	check_equal(ran_cleanups, 2);
+	malloc_printf("Test end\n");
+	return (0);
+}
+
+/* Stubs to avoid pulling the entire jemalloc internals */
+bool    opt_abort = false;
+
+arena_t *_arenas[1] = { NULL };
+arena_t **arenas = _arenas;
+
+#ifndef JEMALLOC_ENABLE_INLINE
+void *
+arena_malloc(arena_t *arena, size_t size, bool zero, bool try_tcache)
+{
+	return NULL;
+}
+
+void
+idalloc(void *ptr)
+{
+}
+#endif
+
+void *
+malloc_tsd_malloc(size_t size)
+{
+#if 0
+	return NULL;
+#else
+	static unsigned char buf[4096];
+	static unsigned char *next_ptr = buf;
+	unsigned char *ret = next_ptr;
+
+	next_ptr += size;
+	memset(ret, 42, size);
+	return ret;
+#endif
+}
+
+void
+malloc_tsd_dalloc(void *ptr)
+{
+}
+
+#undef malloc_tsd_dalloc
+#define malloc_tsd_dalloc malloc_tsd_dalloc_
+#undef malloc_tsd_malloc
+#define malloc_tsd_malloc malloc_tsd_malloc_
+#include "../src/tsd.c"
diff --git a/test/tsd.exp b/test/tsd.exp
new file mode 100644
index 0000000..369a88d
--- /dev/null
+++ b/test/tsd.exp
@@ -0,0 +1,2 @@
+Test begin
+Test end
-- 
1.7.10