• Home
  • History
  • Annotate
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 // Test __sanitizer_reset_coverage().
2 
3 // RUN: %clangxx_asan -fsanitize-coverage=1 %s -o %t
4 // RUN: ASAN_OPTIONS=coverage=1 %run %t
5 
6 #include <sanitizer/coverage_interface.h>
7 #include <stdio.h>
8 #include <assert.h>
9 static volatile int sink;
bar()10 __attribute__((noinline)) void bar() { sink = 2; }
foo()11 __attribute__((noinline)) void foo() { sink = 1; }
12 
13 #define GET_AND_PRINT_COVERAGE()                                       \
14   bitset = 0;                                                  \
15   for (size_t i = 0; i < n_guards; i++)                        \
16     if (guards[i]) bitset |= 1U << i;                          \
17   printf("line %d: bitset %zd total: %zd\n", __LINE__, bitset, \
18          __sanitizer_get_total_unique_coverage());
19 
20 #define IS_POWER_OF_TWO(a) ((a & ((a) - 1)) == 0)
21 
main()22 int main() {
23   size_t *guards = 0;
24   size_t bitset;
25   size_t n_guards = __sanitizer_get_coverage_guards(&guards);
26 
27   GET_AND_PRINT_COVERAGE();
28   size_t main_bit = bitset;
29   assert(IS_POWER_OF_TWO(main_bit));
30 
31   foo();
32   GET_AND_PRINT_COVERAGE();
33   size_t foo_bit = bitset & ~main_bit;
34   assert(IS_POWER_OF_TWO(foo_bit));
35 
36   bar();
37   GET_AND_PRINT_COVERAGE();
38   size_t bar_bit = bitset & ~(main_bit | foo_bit);
39   assert(IS_POWER_OF_TWO(bar_bit));
40 
41   __sanitizer_reset_coverage();
42   GET_AND_PRINT_COVERAGE();
43   assert(bitset == 0);
44 
45   foo();
46   GET_AND_PRINT_COVERAGE();
47   assert(bitset == foo_bit);
48 
49   bar();
50   GET_AND_PRINT_COVERAGE();
51   assert(bitset == (foo_bit | bar_bit));
52 }
53