interception_malloc_test.cc revision 2d1fdb26e458c4ddc04155c1d421bced3ba90cd0 
1// ASan interceptor can be accessed with __interceptor_ prefix.
2
3// RUN: %clangxx_asan -O0 %s -o %t && not %run %t 2>&1 | FileCheck %s
4// RUN: %clangxx_asan -O1 %s -o %t && not %run %t 2>&1 | FileCheck %s
5// RUN: %clangxx_asan -O2 %s -o %t && not %run %t 2>&1 | FileCheck %s
6// RUN: %clangxx_asan -O3 %s -o %t && not %run %t 2>&1 | FileCheck %s
7#include <stdlib.h>
8#include <stdio.h>
9#include <unistd.h>
10
11extern "C" void *__interceptor_malloc(size_t size);
12extern "C" void *malloc(size_t size) {
13  write(2, "malloc call\n", sizeof("malloc call\n") - 1);
14  return __interceptor_malloc(size);
15}
16
17int main() {
18  char *x = (char*)malloc(10 * sizeof(char));
19  free(x);
20  return (int)strtol(x, 0, 10);
21  // CHECK: malloc call
22  // CHECK: heap-use-after-free
23}
24