1; RUN: opt < %s -sancov -sanitizer-coverage-level=4 -sanitizer-coverage-trace-pc  -S -enable-new-pm=0 | FileCheck %s --check-prefix=CHECK_TRACE_PC
2; RUN: opt < %s -sancov -sanitizer-coverage-level=3 -sanitizer-coverage-prune-blocks=1 -S -enable-new-pm=0 | FileCheck %s --check-prefix=CHECKPRUNE
3
4; RUN: opt < %s -passes='module(sancov-module)' -sanitizer-coverage-level=4 -sanitizer-coverage-trace-pc  -S | FileCheck %s --check-prefix=CHECK_TRACE_PC
5; RUN: opt < %s -passes='module(sancov-module)' -sanitizer-coverage-level=3 -sanitizer-coverage-prune-blocks=1 -S | FileCheck %s --check-prefix=CHECKPRUNE
6
7target datalayout = "e-p:64:64:64-i1:8:8-i8:8:8-i16:16:16-i32:32:32-i64:64:64-f32:32:32-f64:64:64-v64:64:64-v128:128:128-a0:0:64-s0:64:64-f80:128:128-n8:16:32:64"
8target triple = "x86_64-unknown-linux-gnu"
9define void @foo(i32* %a) sanitize_address {
10entry:
11  %tobool = icmp eq i32* %a, null
12  br i1 %tobool, label %if.end, label %if.then
13
14  if.then:                                          ; preds = %entry
15  store i32 0, i32* %a, align 4
16  br label %if.end
17
18  if.end:                                           ; preds = %entry, %if.then
19  ret void
20}
21
22; CHECK0-NOT: @llvm.global_ctors = {{.*}}{ i32 2, void ()* @sancov.module_ctor }
23; CHECK1: @llvm.global_ctors = {{.*}}{ i32 2, void ()* @sancov.module_ctor, i8* null }
24; CHECK2: @llvm.global_ctors = {{.*}}{ i32 2, void ()* @sancov.module_ctor, i8* null }
25
26; CHECK0-NOT: call void @__sanitizer_cov(
27; CHECK0-NOT: call void @__sanitizer_cov_module_init(
28
29; CHECK1-LABEL: define void @foo
30; CHECK1: %0 = load atomic i32, i32* {{.*}} monotonic, align 4, !nosanitize
31; CHECK1: %1 = icmp sge i32 0, %0
32; CHECK1: br i1 %1, label %2, label %3
33; CHECK1: call void @__sanitizer_cov(i32*{{.*}})
34; CHECK1: call void asm sideeffect "", ""()
35; CHECK1-NOT: call void @__sanitizer_cov
36; CHECK1: ret void
37
38; CHECK1-LABEL: define internal void @sancov.module_ctor
39; CHECK1-NOT: ret
40; CHECK1: call void @__sanitizer_cov_module_init({{.*}}, i64 2, {{.*}}@__sancov_gen_modname
41; CHECK1: ret
42
43; CHECK_WITH_CHECK-LABEL: define void @foo
44; CHECK_WITH_CHECK: __sanitizer_cov_with_check
45; CHECK_WITH_CHECK: ret void
46; CHECK_WITH_CHECK-LABEL: define internal void @sancov.module_ctor
47; CHECK_WITH_CHECK-NOT: ret
48; CHECK_WITH_CHECK: call void @__sanitizer_cov_module_init({{.*}}, i64 3,
49; CHECK_WITH_CHECK: ret
50
51; CHECK2-LABEL: define void @foo
52; CHECK2: call void @__sanitizer_cov
53; CHECK2: call void asm sideeffect "", ""()
54; CHECK2: call void @__sanitizer_cov
55; CHECK2: call void asm sideeffect "", ""()
56; CHECK2: call void asm sideeffect "", ""()
57; CHECK2-NOT: call void @__sanitizer_cov
58; CHECK2: ret void
59
60; CHECK2-LABEL: define internal void @sancov.module_ctor
61; CHECK2-NOT: ret
62; CHECK2: call void @__sanitizer_cov_module_init({{.*}}, i64 3,
63; CHECK2: ret
64
65; CHECK3-LABEL: define void @foo
66; CHECK3: call void @__sanitizer_cov
67; CHECK3: call void @__sanitizer_cov
68; CHECK3-NOT: ret void
69; CHECK3: call void @__sanitizer_cov
70; CHECK3-NOT: call void @__sanitizer_cov
71; CHECK3: ret void
72
73%struct.StructWithVptr = type { i32 (...)** }
74
75define void @CallViaVptr(%struct.StructWithVptr* %foo) uwtable sanitize_address {
76entry:
77  %0 = bitcast %struct.StructWithVptr* %foo to void (%struct.StructWithVptr*)***
78  %vtable = load void (%struct.StructWithVptr*)**, void (%struct.StructWithVptr*)*** %0, align 8
79  %1 = load void (%struct.StructWithVptr*)*, void (%struct.StructWithVptr*)** %vtable, align 8
80  tail call void %1(%struct.StructWithVptr* %foo)
81  tail call void %1(%struct.StructWithVptr* %foo)
82  tail call void asm sideeffect "", ""()
83  ret void
84}
85
86; CHECK_TRACE_PC-LABEL: define void @foo
87; CHECK_TRACE_PC: call void @__sanitizer_cov_trace_pc
88; CHECK_TRACE_PC: ret void
89
90; CHECK_TRACE_PC-LABEL: define void @CallViaVptr
91; CHECK_TRACE_PC: call void @__sanitizer_cov_trace_pc_indir
92; CHECK_TRACE_PC: call void @__sanitizer_cov_trace_pc_indir
93; CHECK_TRACE_PC: ret void
94
95define void @call_unreachable() uwtable sanitize_address {
96entry:
97  unreachable
98}
99
100; CHECKPRUNE-LABEL: define void @foo
101; CHECKPRUNE: call void @__sanitizer_cov
102; CHECKPRUNE: call void @__sanitizer_cov
103; CHECKPRUNE: call void @__sanitizer_cov
104; CHECKPRUNE-NOT: call void @__sanitizer_cov
105; CHECKPRUNE: ret void
106