187bd39a72b65f1ef5c5a33fc8ba3aebae998903eAlexey Samsonov// Check that free hook doesn't conflict with Realloc. 2fa50ea7718f094f3737dad23e7bc6b6c086e362bAlexey Samsonov// RUN: %clangxx_asan -O2 %s -o %t 32d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines// RUN: %run %t 2>&1 | FileCheck %s 46a211c5814e25d6745a5058cc0e499e5235d3821Stephen Hines 587bd39a72b65f1ef5c5a33fc8ba3aebae998903eAlexey Samsonov#include <stdlib.h> 6fa50ea7718f094f3737dad23e7bc6b6c086e362bAlexey Samsonov#include <unistd.h> 76a211c5814e25d6745a5058cc0e499e5235d3821Stephen Hines#include <sanitizer/allocator_interface.h> 8fa50ea7718f094f3737dad23e7bc6b6c086e362bAlexey Samsonov 9fa50ea7718f094f3737dad23e7bc6b6c086e362bAlexey Samsonovstatic void *glob_ptr; 1087bd39a72b65f1ef5c5a33fc8ba3aebae998903eAlexey Samsonov 1187bd39a72b65f1ef5c5a33fc8ba3aebae998903eAlexey Samsonovextern "C" { 126a211c5814e25d6745a5058cc0e499e5235d3821Stephen Hinesvoid __sanitizer_free_hook(const volatile void *ptr) { 13fa50ea7718f094f3737dad23e7bc6b6c086e362bAlexey Samsonov if (ptr == glob_ptr) { 14fa50ea7718f094f3737dad23e7bc6b6c086e362bAlexey Samsonov *(int*)ptr = 0; 15fa50ea7718f094f3737dad23e7bc6b6c086e362bAlexey Samsonov write(1, "FreeHook\n", sizeof("FreeHook\n")); 16fa50ea7718f094f3737dad23e7bc6b6c086e362bAlexey Samsonov } 1787bd39a72b65f1ef5c5a33fc8ba3aebae998903eAlexey Samsonov} 1887bd39a72b65f1ef5c5a33fc8ba3aebae998903eAlexey Samsonov} 1987bd39a72b65f1ef5c5a33fc8ba3aebae998903eAlexey Samsonov 2087bd39a72b65f1ef5c5a33fc8ba3aebae998903eAlexey Samsonovint main() { 2187bd39a72b65f1ef5c5a33fc8ba3aebae998903eAlexey Samsonov int *x = (int*)malloc(100); 2287bd39a72b65f1ef5c5a33fc8ba3aebae998903eAlexey Samsonov x[0] = 42; 23fa50ea7718f094f3737dad23e7bc6b6c086e362bAlexey Samsonov glob_ptr = x; 2487bd39a72b65f1ef5c5a33fc8ba3aebae998903eAlexey Samsonov int *y = (int*)realloc(x, 200); 25fa50ea7718f094f3737dad23e7bc6b6c086e362bAlexey Samsonov // Verify that free hook was called and didn't spoil the memory. 26fa50ea7718f094f3737dad23e7bc6b6c086e362bAlexey Samsonov if (y[0] != 42) { 27fa50ea7718f094f3737dad23e7bc6b6c086e362bAlexey Samsonov _exit(1); 28fa50ea7718f094f3737dad23e7bc6b6c086e362bAlexey Samsonov } 29fa50ea7718f094f3737dad23e7bc6b6c086e362bAlexey Samsonov write(1, "Passed\n", sizeof("Passed\n")); 304f78b354034f4b6826a59733d774f558ca1eab89Alexey Samsonov free(y); 31fa50ea7718f094f3737dad23e7bc6b6c086e362bAlexey Samsonov // CHECK: FreeHook 32fa50ea7718f094f3737dad23e7bc6b6c086e362bAlexey Samsonov // CHECK: Passed 3387bd39a72b65f1ef5c5a33fc8ba3aebae998903eAlexey Samsonov return 0; 3487bd39a72b65f1ef5c5a33fc8ba3aebae998903eAlexey Samsonov} 35