summaryrefslogtreecommitdiff
path: root/test/asan/TestCases/coverage-pc-buffer.cc
blob: 5895a5c45d1536f152560cbd5282593500c382eb (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
// Test __sanitizer_coverage_pc_buffer().

// RUN: %clangxx_asan -fsanitize-coverage=edge %s -o %t && %run %t

// UNSUPPORTED: android

#include <assert.h>
#include <sanitizer/coverage_interface.h>
#include <stdio.h>

static volatile int sink;
__attribute__((noinline)) void bar() { sink = 2; }
__attribute__((noinline)) void foo() { sink = 1; }

void assertNotZeroPcs(uintptr_t *buf, uintptr_t size) {
  assert(buf);
  for (uintptr_t i = 0; i < size; ++i)
    assert(buf[i]);
}

int main() {
  {
    uintptr_t *buf = NULL;
    uintptr_t sz = __sanitizer_get_coverage_pc_buffer(&buf);
    assertNotZeroPcs(buf, sz);
    assert(sz);
  }

  {
    uintptr_t *buf = NULL;
    uintptr_t sz = __sanitizer_get_coverage_pc_buffer(&buf);
    // call functions for the first time.
    foo();
    bar();
    uintptr_t *buf1 = NULL;
    uintptr_t sz1 = __sanitizer_get_coverage_pc_buffer(&buf1);
    assertNotZeroPcs(buf1, sz1);
    assert(buf1 == buf);
    assert(sz1 > sz);
  }

  {
    uintptr_t *buf = NULL;
    uintptr_t sz = __sanitizer_get_coverage_pc_buffer(&buf);
    // second call shouldn't increase coverage.
    bar();
    uintptr_t *buf1 = NULL;
    uintptr_t sz1 = __sanitizer_get_coverage_pc_buffer(&buf1);
    assertNotZeroPcs(buf1, sz1);
    assert(buf1 == buf);
    assert(sz1 == sz);
  }

  {
    uintptr_t *buf = NULL;
    uintptr_t sz = __sanitizer_get_coverage_pc_buffer(&buf);
    // reset coverage to 0.
    __sanitizer_reset_coverage();
    uintptr_t *buf1 = NULL;
    uintptr_t sz1 = __sanitizer_get_coverage_pc_buffer(&buf1);
    assertNotZeroPcs(buf1, sz1);
    assert(buf1 == buf);
    assert(sz1 < sz);
  }
}