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 47void foo(int i) { 48 switch(i) { 49 // 256 basic blocks 50 bb8(0) 51 } 52} 53#else 54extern void foo(int i); 55 56int 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