#include <assert.h>

#include <pthread.h>

#include <stdlib.h>

#include <stdio.h>

#include <sanitizer/allocator_interface.h>

int num_threads;

int total_num_alloc;

const int kMaxNumThreads = 500;

pthread_t tid[kMaxNumThreads];

pthread_cond_t cond = PTHREAD_COND_INITIALIZER;

pthread_mutex_t mutex = PTHREAD_MUTEX_INITIALIZER;

bool go = false;

void *thread_fun(void *arg) {

pthread_mutex_lock(&mutex);

while (!go) pthread_cond_wait(&cond, &mutex);

pthread_mutex_unlock(&mutex);

for (int i = 0; i < total_num_alloc / num_threads; i++) {

void *p = malloc(10);

__asm__ __volatile__("" : : "r"(p) : "memory");

free(p);

}

return 0;

}

int main(int argc, char** argv) {

assert(argc == 3);

num_threads = atoi(argv[1]);

assert(num_threads > 0);

assert(num_threads <= kMaxNumThreads);

total_num_alloc = atoi(argv[2]);

assert(total_num_alloc > 0);

printf("%d threads, %d allocations in each\n", num_threads,

total_num_alloc / num_threads);

fprintf(stderr, "Heap size before: %zd\n", __sanitizer_get_heap_size());

fprintf(stderr, "Allocated bytes before: %zd\n",

__sanitizer_get_current_allocated_bytes());

for (int i = 0; i < num_threads; i++)

pthread_create(&tid[i], 0, thread_fun, 0);

pthread_mutex_lock(&mutex);

go = true;

pthread_cond_broadcast(&cond);

pthread_mutex_unlock(&mutex);

for (int i = 0; i < num_threads; i++)

pthread_join(tid[i], 0);

fprintf(stderr, "Heap size after: %zd\n", __sanitizer_get_heap_size());

fprintf(stderr, "Allocated bytes after: %zd\n",

__sanitizer_get_current_allocated_bytes());

return 0;

}