d937b0a10c
Summary: Initially, Scudo had a monolithic design where both C and C++ functions were living in the same library. This was not necessarily ideal, and with the work on -fsanitize=scudo, it became more apparent that this needed to change. We are splitting the new/delete interceptor in their own C++ library. This allows more flexibility, notably with regard to std::bad_alloc when the work is done. This also allows us to not link new & delete when using pure C. Additionally, we add the UBSan runtimes with Scudo, in order to be able to have a -fsanitize=scudo,undefined in Clang (see work in D39334). The changes in this patch: - split the cxx specific code in the scudo cmake file into a new library; (remove the spurious foreach loop, that was not necessary) - add the UBSan runtimes (both C and C++); - change the test cmake file to allow for specific C & C++ tests; - make C tests pure C, rename their extension accordingly. Reviewers: alekseyshl Reviewed By: alekseyshl Subscribers: srhines, mgorny, llvm-commits Differential Revision: https://reviews.llvm.org/D39461 llvm-svn: 317097
66 lines
2.0 KiB
C
66 lines
2.0 KiB
C
// RUN: %clang_scudo %s -o %t
|
|
// RUN: %env_scudo_opts="QuarantineSizeKb=0:ThreadLocalQuarantineSizeKb=0" %run %t 5 1000000 2>&1
|
|
// RUN: %env_scudo_opts="QuarantineSizeKb=1024:ThreadLocalQuarantineSizeKb=64" %run %t 5 1000000 2>&1
|
|
|
|
// Tests parallel allocations and deallocations of memory chunks from a number
|
|
// of concurrent threads, with and without quarantine.
|
|
// This test passes if everything executes properly without crashing.
|
|
|
|
#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;
|
|
char go = 0;
|
|
|
|
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 = 1;
|
|
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;
|
|
}
|