[Sanitizer] Move test for malloc/free hooks under test/sanitizer_common

llvm-svn: 212474

compiler-rt/test/msan/malloc_hook.cc

@@ -1,36 +0,0 @@
-// RUN: %clangxx_msan -O2 %s -o %t
-// RUN: %run %t 2>&1 | FileCheck %s
-#include <stdlib.h>
-#include <unistd.h>
-#include <sanitizer/allocator_interface.h>
-
-extern "C" {
-
-const volatile void *global_ptr;
-
-// Note: avoid calling functions that allocate memory in malloc/free
-// to avoid infinite recursion.
-void __sanitizer_malloc_hook(const volatile void *ptr, size_t sz) {
-  if (__sanitizer_get_ownership(ptr)) {
-    write(1, "MallocHook\n", sizeof("MallocHook\n"));
-    global_ptr = ptr;
-  }
-}
-void __sanitizer_free_hook(const volatile void *ptr) {
-  if (__sanitizer_get_ownership(ptr) && ptr == global_ptr)
-    write(1, "FreeHook\n", sizeof("FreeHook\n"));
-}
-}  // extern "C"
-
-int main() {
-  volatile int *x = new int;
-  // CHECK: MallocHook
-  // Check that malloc hook was called with correct argument.
-  if (global_ptr != (void*)x) {
-    _exit(1);
-  }
-  *x = 0;
-  delete x;
-  // CHECK: FreeHook
-  return 0;
-}

compiler-rt/test/sanitizer_common/TestCases/malloc_hook.cc

@@ -1,8 +1,7 @@
-// RUN: %clangxx_asan -O2 %s -o %t
-// RUN: %run %t 2>&1 | FileCheck %s
+// RUN: %clangxx -O2 %s -o %t && %run %t 2>&1 | FileCheck %s
 
-// Malloc/free hooks are not supported on Windows.
-// XFAIL: win32
+// Malloc/free hooks are not supported on Windows and doesn't work in LSan.
+// XFAIL: win32, lsan
 
 #include <stdlib.h>
 #include <unistd.h>

compiler-rt/test/tsan/malloc_hook.cc

@@ -1,53 +0,0 @@
-// RUN: %clangxx_tsan -O1 %s -o %t && %deflake %run %t | FileCheck %s
-#include <pthread.h>
-#include <unistd.h>
-#include <stdio.h>
-#include <stdlib.h>
-#include <stddef.h>
-#include <sanitizer/allocator_interface.h>
-
-static int malloc_count;
-static int free_count;
-
-extern "C" {
-void __sanitizer_malloc_hook(const volatile void *ptr, size_t size) {
-  (void)ptr;
-  (void)size;
-  __sync_fetch_and_add(&malloc_count, 1);
-}
-
-void __sanitizer_free_hook(const volatile void *ptr) {
-  (void)ptr;
-  __sync_fetch_and_add(&free_count, 1);
-}
-}
-
-void *Thread1(void *x) {
-  ((int*)x)[0]++;
-  return 0;
-}
-
-void *Thread2(void *x) {
-  sleep(1);
-  ((int*)x)[0]++;
-  return 0;
-}
-
-int main() {
-  int *x = new int;
-  pthread_t t[2];
-  pthread_create(&t[0], 0, Thread1, x);
-  pthread_create(&t[1], 0, Thread2, x);
-  pthread_join(t[0], 0);
-  pthread_join(t[1], 0);
-  delete x;
-  if (malloc_count == 0 || free_count == 0) {
-    fprintf(stderr, "FAILED %d %d\n", malloc_count, free_count);
-    exit(1);
-  }
-  fprintf(stderr, "DONE\n");
-}
-
-// CHECK: WARNING: ThreadSanitizer: data race
-// CHECK-NOT: FAILED
-// CHECK: DONE