16eed7e7eSNico Weber // Test various levels of coverage
26eed7e7eSNico Weber //
36eed7e7eSNico Weber // FIXME: Port the environment variable logic below for the lit shell.
46eed7e7eSNico Weber // REQUIRES: shell
56eed7e7eSNico Weber //
66eed7e7eSNico Weber // RUN: rm -rf %t-dir && mkdir %t-dir
7*a7141480SJinsong Ji // RUN: %clangxx -fsanitize=shift                        -DGOOD_SHIFT=1 -O1 -fsanitize-coverage=func,trace-pc-guard  %s -o %t
86eed7e7eSNico Weber // RUN: %env_ubsan_opts=coverage=1:verbosity=1:coverage_dir='"%t-dir"' %run %t 2>&1 | FileCheck %s --check-prefix=CHECK1 --check-prefix=CHECK_NOWARN
9*a7141480SJinsong Ji // RUN: %clangxx -fsanitize=undefined                    -DGOOD_SHIFT=1 -O1 -fsanitize-coverage=func,trace-pc-guard  %s -o %t
106eed7e7eSNico Weber // RUN: %env_ubsan_opts=coverage=1:verbosity=1:coverage_dir='"%t-dir"' %run %t 2>&1 | FileCheck %s --check-prefix=CHECK1 --check-prefix=CHECK_NOWARN
116eed7e7eSNico Weber 
126eed7e7eSNico Weber // Also works without any sanitizer.
13*a7141480SJinsong Ji // RUN: %clangxx                                         -DGOOD_SHIFT=1 -O1 -fsanitize-coverage=func,trace-pc-guard  %s -o %t
146eed7e7eSNico Weber // RUN: %env_ubsan_opts=coverage=1:verbosity=1:coverage_dir='"%t-dir"' %run %t 2>&1 | FileCheck %s --check-prefix=CHECK1 --check-prefix=CHECK_NOWARN
156eed7e7eSNico Weber 
16*a7141480SJinsong Ji // RUN: %clangxx -fsanitize=shift -O1 -fsanitize-coverage=func,trace-pc-guard  %s -o %t
176eed7e7eSNico Weber // RUN: %env_ubsan_opts=coverage=1:verbosity=1:coverage_dir='"%t-dir"' %run %t 2>&1 | FileCheck %s --check-prefix=CHECK1 --check-prefix=CHECK_WARN
18*a7141480SJinsong Ji // RUN: %clangxx -fsanitize=shift -O1 -fsanitize-coverage=bb,trace-pc-guard  %s -o %t
196eed7e7eSNico Weber // RUN: %env_ubsan_opts=coverage=1:verbosity=1:coverage_dir='"%t-dir"' %run %t 2>&1 | FileCheck %s --check-prefix=CHECK2 --check-prefix=CHECK_WARN
20*a7141480SJinsong Ji // RUN: %clangxx -fsanitize=shift -O1 -fsanitize-coverage=edge,trace-pc-guard  %s -o %t
216eed7e7eSNico Weber // RUN: %env_ubsan_opts=coverage=1:verbosity=1:coverage_dir='"%t-dir"' %run %t 2>&1 | FileCheck %s --check-prefix=CHECK3 --check-prefix=CHECK_WARN
226eed7e7eSNico Weber 
236eed7e7eSNico Weber // Coverage is not yet implemented in TSan.
246eed7e7eSNico Weber // XFAIL: ubsan-tsan
256eed7e7eSNico Weber // UNSUPPORTED: ubsan-standalone-static
266eed7e7eSNico Weber // No coverage support
276eed7e7eSNico Weber // UNSUPPORTED: openbsd
286eed7e7eSNico Weber 
296eed7e7eSNico Weber volatile int sink;
main(int argc,char ** argv)306eed7e7eSNico Weber int main(int argc, char **argv) {
316eed7e7eSNico Weber   int shift = argc * 32;
326eed7e7eSNico Weber #if GOOD_SHIFT
336eed7e7eSNico Weber   shift = 3;
346eed7e7eSNico Weber #endif
356eed7e7eSNico Weber   if ((argc << shift) == 16)  // False.
366eed7e7eSNico Weber     return 1;
376eed7e7eSNico Weber   return 0;
386eed7e7eSNico Weber }
396eed7e7eSNico Weber 
406eed7e7eSNico Weber // CHECK_WARN: shift exponent 32 is too large
416eed7e7eSNico Weber // CHECK_NOWARN-NOT: ERROR
426eed7e7eSNico Weber // FIXME: Currently, coverage instrumentation kicks in after ubsan, so we get
436eed7e7eSNico Weber // more than the minimal number of instrumented blocks.
446eed7e7eSNico Weber // FIXME: Currently, ubsan with -fno-sanitize-recover and w/o asan will fail
456eed7e7eSNico Weber // to dump coverage.
466eed7e7eSNico Weber // CHECK1:  1 PCs written
476eed7e7eSNico Weber // CHECK2:  2 PCs written
486eed7e7eSNico Weber // CHECK3:  2 PCs written
49