12d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines// Check that memset() call from a shared library gets intercepted.
22d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines// Please always keep this file in sync with
32d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines// ../Linux/interception-in-shared-lib-test.cc.
42d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines
52d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines// RUN: %clangxx_asan -O0 %s -DSHARED_LIB \
62d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines// RUN:     -shared -o %T/libinterception-in-shared-lib-test.so \
72d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines// RUN:     -fPIC
82d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines// TODO(glider): figure out how to set rpath in a more portable way and unite
92d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines// this test with ../Linux/interception-in-shared-lib-test.cc.
102d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines// RUN: %clangxx_asan -O0 %s -o %t -Wl,-rpath,@executable-path -L%T -linterception-in-shared-lib-test && \
112d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines// RUN:     not %run %t 2>&1 | FileCheck %s
122d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines
132d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines#include <stdio.h>
142d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines#include <string.h>
152d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines
162d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines#if defined(SHARED_LIB)
172d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hinesextern "C"
182d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hinesvoid my_memset(void *p, size_t sz) {
192d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines  memset(p, 0, sz);
202d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines}
212d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines#else
222d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hinesextern "C" void my_memset(void *p, size_t sz);
232d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines
242d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hinesint main(int argc, char *argv[]) {
252d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines  char buf[10];
262d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines  my_memset(buf, 11);
272d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines  // CHECK: {{.*ERROR: AddressSanitizer: stack-buffer-overflow}}
282d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines  // CHECK: {{WRITE of size 11 at 0x.* thread T0}}
292d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines  // CHECK: {{0x.* in my_memset .*interception-in-shared-lib-test.cc:19}}
302d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines  return 0;
312d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines}
322d1fdb26e458c4ddc04155c1d421bced3ba90cd0Stephen Hines#endif
33