compiler-rt/test/asan/TestCases/Posix/free_hook_realloc.cpp

   1 // Check that free hook doesn't conflict with Realloc.
   2 // RUN: %clangxx_asan -O2 %s -o %t
   3 // RUN: %run %t 2>&1 | FileCheck %s
   4
   5 #include <stdlib.h>
   6 #include <unistd.h>
   7 #include <sanitizer/allocator_interface.h>
   8
   9 static void *glob_ptr;
  10
  11 // Required for dyld macOS 12.0+
  12 #if (__APPLE__)
  13 __attribute__((weak))
  14 #endif
  15 extern "C" void
  16 __sanitizer_free_hook(const volatile void *ptr) {
  17   if (ptr == glob_ptr) {
  18     *(int*)ptr = 0;
  19     write(1, "FreeHook\n", sizeof("FreeHook\n"));
  20   }
  21 }
  22
  23 int main() {
  24   int *x = (int*)malloc(100);
  25   x[0] = 42;
  26   glob_ptr = x;
  27   int *y = (int*)realloc(x, 200);
  28   // Verify that free hook was called and didn't spoil the memory.
  29   if (y[0] != 42) {
  30     _exit(1);
  31   }
  32   write(1, "Passed\n", sizeof("Passed\n"));
  33   free(y);
  34   // CHECK: FreeHook
  35   // CHECK: Passed
  36   return 0;
  37 }