Home | History | Annotate | Download | only in Linux
      1 // RUN: %clangxx_asan -mllvm -asan-coverage=2 -DSHARED %s -shared -o %T/libcoverage_sandboxing_test.so -fPIC
      2 // RUN: %clangxx_asan -mllvm -asan-coverage=1 %s   -o %t -Wl,-R,\$ORIGIN -L%T -lcoverage_sandboxing_test
      3 // RUN: export ASAN_OPTIONS=coverage=1:verbosity=1
      4 // RUN: rm -rf %T/coverage_sandboxing_test
      5 // RUN: mkdir %T/coverage_sandboxing_test && cd %T/coverage_sandboxing_test
      6 // RUN: mkdir vanilla && cd vanilla
      7 // RUN: %run %t 2>&1         | FileCheck %s --check-prefix=CHECK-vanilla
      8 // RUN: mkdir ../sandbox1 && cd ../sandbox1
      9 // RUN: %run %t a 2>&1       | FileCheck %s --check-prefix=CHECK-sandbox
     10 // RUN: %sancov unpack coverage_sandboxing_test.sancov.packed
     11 // RUN: mkdir ../sandbox2 && cd ../sandbox2
     12 // RUN: %run %t a b 2>&1     | FileCheck %s --check-prefix=CHECK-sandbox
     13 // RUN: %sancov unpack coverage_sandboxing_test.sancov.packed
     14 // RUN: cd ..
     15 // RUN: %sancov print vanilla/libcoverage_sandboxing_test.so.*.sancov > vanilla.txt
     16 // RUN: %sancov print sandbox1/libcoverage_sandboxing_test.so.*.sancov > sandbox1.txt
     17 // RUN: %sancov print sandbox2/libcoverage_sandboxing_test.so.*.sancov > sandbox2.txt
     18 // RUN: diff vanilla.txt sandbox1.txt
     19 // RUN: diff vanilla.txt sandbox2.txt
     20 // RUN: cd ../ && rm coverage_sandboxing_test -r
     21 // https://code.google.com/p/address-sanitizer/issues/detail?id=263
     22 // XFAIL: android
     23 
     24 #include <assert.h>
     25 #include <fcntl.h>
     26 #include <stdio.h>
     27 #include <string.h>
     28 #include <unistd.h>
     29 
     30 #include <sanitizer/common_interface_defs.h>
     31 
     32 #define bb0(n)                        \
     33   case n:                             \
     34     fprintf(stderr, "foo: %d\n", n);  \
     35     break;
     36 
     37 #define bb1(n) bb0(n) bb0(n + 1)
     38 #define bb2(n) bb1(n) bb1(n + 2)
     39 #define bb3(n) bb2(n) bb2(n + 4)
     40 #define bb4(n) bb3(n) bb3(n + 8)
     41 #define bb5(n) bb4(n) bb4(n + 16)
     42 #define bb6(n) bb5(n) bb5(n + 32)
     43 #define bb7(n) bb6(n) bb6(n + 64)
     44 #define bb8(n) bb7(n) bb7(n + 128)
     45 
     46 #ifdef SHARED
     47 void foo(int i) {
     48   switch(i) {
     49     // 256 basic blocks
     50     bb8(0)
     51   }
     52 }
     53 #else
     54 extern void foo(int i);
     55 
     56 int main(int argc, char **argv) {
     57   assert(argc <= 3);
     58   for (int i = 0; i < 256; i++) foo(i);
     59   fprintf(stderr, "PID: %d\n", getpid());
     60   if (argc == 1) {
     61     // Vanilla mode, dump to individual files.
     62     return 0;
     63   }
     64   // Dump to packed file.
     65   int fd = creat("coverage_sandboxing_test.sancov.packed", 0660);
     66   __sanitizer_sandbox_arguments args = {0};
     67   args.coverage_sandboxed = 1;
     68   args.coverage_fd = fd;
     69   if (argc == 2)
     70     // Write to packed file, do not split into blocks.
     71     args.coverage_max_block_size = 0;
     72   else if (argc == 3)
     73     // Write to packed file, split into blocks (as if writing to a socket).
     74     args.coverage_max_block_size = 100;
     75   __sanitizer_sandbox_on_notify(&args);
     76   return 0;
     77 }
     78 #endif
     79 
     80 // CHECK-vanilla: PID: [[PID:[0-9]+]]
     81 // CHECK-vanilla: [[PID]].sancov: 1 PCs written
     82 // CHECK-vanilla: .so.[[PID]].sancov: 258 PCs written
     83 
     84 // CHECK-sandbox: PID: [[PID:[0-9]+]]
     85 // CHECK-sandbox: 258 PCs written to packed file
     86