1 files changed, 28 insertions, 15 deletions

diff --git a/src/benchmarks/loop/loop.c b/src/benchmarks/loop/loop.c
index 6297e49..9593cd5 100644
--- a/src/benchmarks/loop/loop.c
+++ b/src/benchmarks/loop/loop.c
@@ -19,26 +19,29 @@ typedef struct ThreadArgs {
 	double benchmark;
 	int allocations;
 	int max_size;
+#ifdef KEEP_ALLOCS
+	int num_to_keep;
+#endif
 } ThreadArgs;
 
-static void* malloc_then_write(size_t size) {
-	void* ptr = malloc(size);
-	// Write to ptr
-	/* *((char*)ptr) = '!'; */
-	return ptr;
-}
-
-static void read_then_free(void* ptr) {
-	// Read before free
-	/* char s __attribute__((unused)) = *((char*)ptr); */
-	free(ptr);
-}
 static void* test_thread_func(void* arg) {
 	ThreadArgs* args = (ThreadArgs*)arg;
+#ifdef KEEP_ALLOCS
+	void** ptrs = (void**)calloc(args->num_to_keep, sizeof(void*));
+#endif
 
 	for(int i = 0; i < args->allocations; i++) {
-		void* ptr = malloc_then_write((_rand() % args->max_size) + 1);
-		read_then_free(ptr);
+#ifdef KEEP_ALLOCS
+		int pos = i % args->num_to_keep;
+		if (0 == pos && i > 0) {
+			for (int j = 0; j < args->num_to_keep; j++)
+				free(ptrs[j]);
+		}
+		ptrs[pos] = malloc((_rand() % args->max_size) + 1);
+#else
+		void* ptr = malloc((_rand() % args->max_size) + 1);
+		free(ptr);
+#endif
 	}
 	return NULL;
 }
@@ -48,14 +51,24 @@ int main(int argc, char* argv[]) {
 	int num_threads;
 	struct ThreadArgs thread_args;
 
+#ifdef KEEP_ALLOCS
+	if (argc < 5) {
+		fprintf(stderr, "Usage: %s <num threads> <num allocations> <max size> <allocs to keep>\n", argv[0]);
+		return 1;
+	}
+#else
 	if (argc < 4) {
 		fprintf(stderr, "Usage: %s <num threads> <num allocations> <max size>\n", argv[0]);
 		return 1;
 	}
+#endif
 
 	num_threads = atoi(argv[1]);
 	thread_args.allocations = atoi(argv[2]);
 	thread_args.max_size = atoi(argv[3]);
+#ifdef KEEP_ALLOCS
+	thread_args.num_to_keep = atoi(argv[4]);
+#endif
 
 	threads = (pthread_t*)malloc(num_threads * sizeof(pthread_t));
 
@@ -73,7 +86,7 @@ int main(int argc, char* argv[]) {
 		}
 	}
 
-	if (argc == 5)
+	if (argc == 6)
 	{
 		FILE* f = stdout;
 		if (strcmp(argv[4],"stdout") != 0)