add blowup test

Blowup tests the described blowup problem of ptmalloc described in the
TCMalloc post.

2 files changed, 115 insertions, 0 deletions

diff --git a/src/benchmarks/blowup/Makefile b/src/benchmarks/blowup/Makefile
new file mode 100644
index 0000000..afb5f2a
--- /dev/null
+++ b/src/benchmarks/blowup/Makefile
@@ -0,0 +1,25 @@
+OBJDIR ?= obj
+
+CC ?= gcc
+
+WARNFLAGS ?= -Wall -Wextra
+COMMONFLAGS ?= -fno-builtin -fPIC -DPIC -pthread
+OPTFLAGS ?= -O3 -DNDEBUG
+
+CFLAGS ?= $(OPTFLAGS) $(WARNFLAGS) $(COMMONFLAGS)
+
+LDFLAGS ?= -pthread -static-libgcc
+
+.PHONY = all clean
+
+all: $(OBJDIR)/blowup
+
+$(OBJDIR)/blowup: blowup.c | $(OBJDIR)
+	@echo compiling $@...
+	$(CC) $(LDFLAGS) $(CFLAGS) -o $@ $<
+
+$(OBJDIR):
+	mkdir -p $@
+
+clean:
+	rm -rf $(OBJDIR)
diff --git a/src/benchmarks/blowup/blowup.c b/src/benchmarks/blowup/blowup.c
new file mode 100644
index 0000000..defb58c
--- /dev/null
+++ b/src/benchmarks/blowup/blowup.c
@@ -0,0 +1,90 @@
+#include <pthread.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+
+#define NUM_THREADS 10
+#define ALLOCATIONS 1024 * 100
+#define SIZE 1024
+
+static void do_work()
+{
+	void** ptrs;
+
+	ptrs = malloc(sizeof(void*) * ALLOCATIONS);
+	if (ptrs == NULL) {
+		perror("malloc");
+		return;
+	}
+
+	for(int i = 0; i < ALLOCATIONS; i++)
+	{
+		ptrs[i] = malloc(SIZE);
+		memset(ptrs[i], 0, SIZE);
+	}
+
+	for(int i = 0; i < ALLOCATIONS; i++)
+	{
+		free(malloc(SIZE));
+	}
+
+	for(int i = 0; i < ALLOCATIONS; i++)
+	{
+		/* memset(ptrs[i], 0, SIZE); */
+		free(ptrs[i]);
+	}
+
+	for(int i = 0; i < ALLOCATIONS; i++)
+	{
+		free(malloc(SIZE));
+	}
+
+	return;
+}
+
+static void* thread_func(void* arg)
+{
+	unsigned int* thread_id = (unsigned int*)arg;
+	pthread_t next_thread;
+
+	printf("thread %d doing work\n", *thread_id);
+	do_work();
+
+	if (*thread_id == NUM_THREADS)
+		return NULL;
+
+	printf("thread %d spawning new thread work\n", *thread_id);
+	*thread_id = (*thread_id) + 1;
+	if (0 != pthread_create(&next_thread, NULL, thread_func, arg)) {
+		perror("pthread_create");
+		return NULL;
+	}
+
+	printf("thread %d joining thread %d work\n", *thread_id-1, *thread_id);
+	if (0 != pthread_join(next_thread, NULL)) {
+		perror("pthread_join");
+		return NULL;
+	}
+
+	return NULL;
+}
+
+int main()
+{
+	unsigned int thread_count = 0;
+	pthread_t first;
+
+	do_work();
+
+	if (0 != pthread_create(&first, NULL, thread_func, &thread_count)) {
+		perror("pthread_create");
+		return 1;
+	}
+
+	if (0 != pthread_join(first, NULL)) {
+		perror("pthread_join");
+		return 1;
+	}
+
+	return 0;
+}