Skip to content

Commit f2f0963

Browse files
committed
[MemorySanitizer] Generate check lines for some vararg tests (NFC)
Use UTC_ARGS: --disable to skip the tests with many arguments.
1 parent 3c0f7b1 commit f2f0963

File tree

5 files changed

+484
-117
lines changed

5 files changed

+484
-117
lines changed

llvm/test/Instrumentation/MemorySanitizer/LoongArch/vararg-loongarch64.ll

Lines changed: 64 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
1+
; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --version 6
12
; RUN: opt < %s -S -passes=msan 2>&1 | FileCheck %s
23

34
target datalayout = "e-m:e-p:64:64-i64:64-i128:128-n64-S128"
@@ -9,12 +10,36 @@ declare void @llvm.va_start(ptr) #2
910
declare void @llvm.va_end(ptr) #2
1011
declare void @llvm.lifetime.end.p0(ptr nocapture) #1
1112
define i32 @foo(i32 %guard, ...) {
12-
; CHECK-LABEL: @foo
13-
; CHECK: [[TMP1:%.*]] = load {{.*}} @__msan_va_arg_overflow_size_tls
14-
; CHECK: [[TMP3:%.*]] = alloca {{.*}} [[TMP1]]
15-
; CHECK: call void @llvm.memset.p0.i64(ptr align 8 [[TMP3]], i8 0, i64 [[TMP1]], i1 false)
16-
; CHECK: [[TMP4:%.*]] = call i64 @llvm.umin.i64(i64 [[TMP1]], i64 800)
17-
; CHECK: call void @llvm.memcpy.p0.p0.i64(ptr align 8 [[TMP3]], ptr align 8 @__msan_va_arg_tls, i64 [[TMP4]], i1 false)
13+
; CHECK-LABEL: define i32 @foo(
14+
; CHECK-SAME: i32 [[GUARD:%.*]], ...) {
15+
; CHECK-NEXT: [[TMP1:%.*]] = load i64, ptr @__msan_va_arg_overflow_size_tls, align 8
16+
; CHECK-NEXT: [[TMP2:%.*]] = alloca i8, i64 [[TMP1]], align 8
17+
; CHECK-NEXT: call void @llvm.memset.p0.i64(ptr align 8 [[TMP2]], i8 0, i64 [[TMP1]], i1 false)
18+
; CHECK-NEXT: [[TMP3:%.*]] = call i64 @llvm.umin.i64(i64 [[TMP1]], i64 800)
19+
; CHECK-NEXT: call void @llvm.memcpy.p0.p0.i64(ptr align 8 [[TMP2]], ptr align 8 @__msan_va_arg_tls, i64 [[TMP3]], i1 false)
20+
; CHECK-NEXT: call void @llvm.donothing()
21+
; CHECK-NEXT: [[VL:%.*]] = alloca ptr, align 8
22+
; CHECK-NEXT: [[TMP4:%.*]] = ptrtoint ptr [[VL]] to i64
23+
; CHECK-NEXT: [[TMP5:%.*]] = xor i64 [[TMP4]], 87960930222080
24+
; CHECK-NEXT: [[TMP6:%.*]] = inttoptr i64 [[TMP5]] to ptr
25+
; CHECK-NEXT: call void @llvm.memset.p0.i64(ptr align 8 [[TMP6]], i8 0, i64 8, i1 false)
26+
; CHECK-NEXT: call void @llvm.lifetime.start.p0(ptr [[VL]])
27+
; CHECK-NEXT: [[TMP7:%.*]] = ptrtoint ptr [[VL]] to i64
28+
; CHECK-NEXT: [[TMP8:%.*]] = xor i64 [[TMP7]], 87960930222080
29+
; CHECK-NEXT: [[TMP9:%.*]] = inttoptr i64 [[TMP8]] to ptr
30+
; CHECK-NEXT: call void @llvm.memset.p0.i64(ptr align 8 [[TMP9]], i8 0, i64 8, i1 false)
31+
; CHECK-NEXT: call void @llvm.va_start.p0(ptr [[VL]])
32+
; CHECK-NEXT: [[TMP10:%.*]] = ptrtoint ptr [[VL]] to i64
33+
; CHECK-NEXT: [[TMP11:%.*]] = inttoptr i64 [[TMP10]] to ptr
34+
; CHECK-NEXT: [[TMP12:%.*]] = load ptr, ptr [[TMP11]], align 8
35+
; CHECK-NEXT: [[TMP13:%.*]] = ptrtoint ptr [[TMP12]] to i64
36+
; CHECK-NEXT: [[TMP14:%.*]] = xor i64 [[TMP13]], 87960930222080
37+
; CHECK-NEXT: [[TMP15:%.*]] = inttoptr i64 [[TMP14]] to ptr
38+
; CHECK-NEXT: call void @llvm.memcpy.p0.p0.i64(ptr align 8 [[TMP15]], ptr align 8 [[TMP2]], i64 [[TMP1]], i1 false)
39+
; CHECK-NEXT: call void @llvm.va_end.p0(ptr [[VL]])
40+
; CHECK-NEXT: call void @llvm.lifetime.end.p0(ptr [[VL]])
41+
; CHECK-NEXT: store i32 0, ptr @__msan_retval_tls, align 8
42+
; CHECK-NEXT: ret i32 0
1843
;
1944
%vl = alloca ptr, align 8
2045
call void @llvm.lifetime.start.p0(ptr %vl)
@@ -27,11 +52,22 @@ define i32 @foo(i32 %guard, ...) {
2752
;; Save the incoming shadow value from the arguments in the __msan_va_arg_tls
2853
;; array.
2954
define i32 @bar() {
30-
; CHECK-LABEL: @bar
31-
; CHECK: store i32 0, ptr @__msan_va_arg_tls, align 8
32-
; CHECK: store i64 0, ptr inttoptr (i64 add (i64 ptrtoint (ptr @__msan_va_arg_tls to i64), i64 8) to ptr), align 8
33-
; CHECK: store i64 0, ptr inttoptr (i64 add (i64 ptrtoint (ptr @__msan_va_arg_tls to i64), i64 16) to ptr), align 8
34-
; CHECK: store {{.*}} 24, {{.*}} @__msan_va_arg_overflow_size_tls
55+
; CHECK-LABEL: define i32 @bar() {
56+
; CHECK-NEXT: [[TMP1:%.*]] = load i64, ptr @__msan_va_arg_overflow_size_tls, align 8
57+
; CHECK-NEXT: call void @llvm.donothing()
58+
; CHECK-NEXT: store i32 0, ptr @__msan_param_tls, align 8
59+
; CHECK-NEXT: store i32 0, ptr inttoptr (i64 add (i64 ptrtoint (ptr @__msan_param_tls to i64), i64 8) to ptr), align 8
60+
; CHECK-NEXT: store i64 0, ptr inttoptr (i64 add (i64 ptrtoint (ptr @__msan_param_tls to i64), i64 16) to ptr), align 8
61+
; CHECK-NEXT: store i64 0, ptr inttoptr (i64 add (i64 ptrtoint (ptr @__msan_param_tls to i64), i64 24) to ptr), align 8
62+
; CHECK-NEXT: store i32 0, ptr @__msan_va_arg_tls, align 8
63+
; CHECK-NEXT: store i64 0, ptr inttoptr (i64 add (i64 ptrtoint (ptr @__msan_va_arg_tls to i64), i64 8) to ptr), align 8
64+
; CHECK-NEXT: store i64 0, ptr inttoptr (i64 add (i64 ptrtoint (ptr @__msan_va_arg_tls to i64), i64 16) to ptr), align 8
65+
; CHECK-NEXT: store i64 24, ptr @__msan_va_arg_overflow_size_tls, align 8
66+
; CHECK-NEXT: store i32 0, ptr @__msan_retval_tls, align 8
67+
; CHECK-NEXT: [[TMP2:%.*]] = call i32 (i32, ...) @foo(i32 0, i32 1, i64 2, double 3.000000e+00)
68+
; CHECK-NEXT: [[_MSRET:%.*]] = load i32, ptr @__msan_retval_tls, align 8
69+
; CHECK-NEXT: store i32 0, ptr @__msan_retval_tls, align 8
70+
; CHECK-NEXT: ret i32 [[TMP2]]
3571
;
3672
%1 = call i32 (i32, ...) @foo(i32 0, i32 1, i64 2, double 3.000000e+00)
3773
ret i32 %1
@@ -40,15 +76,28 @@ define i32 @bar() {
4076
;; Check multiple fixed arguments.
4177
declare i32 @foo2(i32 %g1, i32 %g2, ...)
4278
define i32 @bar2() {
43-
; CHECK-LABEL: @bar2
44-
; CHECK: store i64 0, ptr @__msan_va_arg_tls, align 8
45-
; CHECK: store i64 0, ptr inttoptr (i64 add (i64 ptrtoint (ptr @__msan_va_arg_tls to i64), i64 8) to ptr), align 8
46-
; CHECK: store {{.*}} 16, {{.*}} @__msan_va_arg_overflow_size_tls
79+
; CHECK-LABEL: define i32 @bar2() {
80+
; CHECK-NEXT: [[TMP1:%.*]] = load i64, ptr @__msan_va_arg_overflow_size_tls, align 8
81+
; CHECK-NEXT: call void @llvm.donothing()
82+
; CHECK-NEXT: store i32 0, ptr @__msan_param_tls, align 8
83+
; CHECK-NEXT: store i32 0, ptr inttoptr (i64 add (i64 ptrtoint (ptr @__msan_param_tls to i64), i64 8) to ptr), align 8
84+
; CHECK-NEXT: store i64 0, ptr inttoptr (i64 add (i64 ptrtoint (ptr @__msan_param_tls to i64), i64 16) to ptr), align 8
85+
; CHECK-NEXT: store i64 0, ptr inttoptr (i64 add (i64 ptrtoint (ptr @__msan_param_tls to i64), i64 24) to ptr), align 8
86+
; CHECK-NEXT: store i64 0, ptr @__msan_va_arg_tls, align 8
87+
; CHECK-NEXT: store i64 0, ptr inttoptr (i64 add (i64 ptrtoint (ptr @__msan_va_arg_tls to i64), i64 8) to ptr), align 8
88+
; CHECK-NEXT: store i64 16, ptr @__msan_va_arg_overflow_size_tls, align 8
89+
; CHECK-NEXT: store i32 0, ptr @__msan_retval_tls, align 8
90+
; CHECK-NEXT: [[TMP2:%.*]] = call i32 (i32, i32, ...) @foo2(i32 0, i32 1, i64 2, double 3.000000e+00)
91+
; CHECK-NEXT: [[_MSRET:%.*]] = load i32, ptr @__msan_retval_tls, align 8
92+
; CHECK-NEXT: store i32 0, ptr @__msan_retval_tls, align 8
93+
; CHECK-NEXT: ret i32 [[TMP2]]
4794
;
4895
%1 = call i32 (i32, i32, ...) @foo2(i32 0, i32 1, i64 2, double 3.000000e+00)
4996
ret i32 %1
5097
}
5198

99+
; UTC_ARGS: --disable
100+
52101
;; Test that MSan doesn't generate code overflowing __msan_va_arg_tls when too many arguments are
53102
;; passed to a variadic function.
54103
declare i64 @sum(i64 %n, ...)

llvm/test/Instrumentation/MemorySanitizer/Mips/vararg-mips64.ll

Lines changed: 67 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,41 @@
1+
; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --version 6
12
; RUN: opt < %s -S -passes=msan 2>&1 | FileCheck %s
23

34
target datalayout = "E-m:m-i8:8:32-i16:16:32-i64:64-n32:64-S128"
45
target triple = "mips64--linux"
56

67
define i32 @foo(i32 %guard, ...) {
8+
; CHECK-LABEL: define i32 @foo(
9+
; CHECK-SAME: i32 [[GUARD:%.*]], ...) {
10+
; CHECK-NEXT: [[TMP1:%.*]] = load i64, ptr @__msan_va_arg_overflow_size_tls, align 8
11+
; CHECK-NEXT: [[TMP2:%.*]] = alloca i8, i64 [[TMP1]], align 8
12+
; CHECK-NEXT: call void @llvm.memset.p0.i64(ptr align 8 [[TMP2]], i8 0, i64 [[TMP1]], i1 false)
13+
; CHECK-NEXT: [[TMP3:%.*]] = call i64 @llvm.umin.i64(i64 [[TMP1]], i64 800)
14+
; CHECK-NEXT: call void @llvm.memcpy.p0.p0.i64(ptr align 8 [[TMP2]], ptr align 8 @__msan_va_arg_tls, i64 [[TMP3]], i1 false)
15+
; CHECK-NEXT: call void @llvm.donothing()
16+
; CHECK-NEXT: [[VL:%.*]] = alloca ptr, align 8
17+
; CHECK-NEXT: [[TMP4:%.*]] = ptrtoint ptr [[VL]] to i64
18+
; CHECK-NEXT: [[TMP5:%.*]] = xor i64 [[TMP4]], 549755813888
19+
; CHECK-NEXT: [[TMP6:%.*]] = inttoptr i64 [[TMP5]] to ptr
20+
; CHECK-NEXT: call void @llvm.memset.p0.i64(ptr align 8 [[TMP6]], i8 0, i64 8, i1 false)
21+
; CHECK-NEXT: call void @llvm.lifetime.start.p0(ptr [[VL]])
22+
; CHECK-NEXT: [[TMP7:%.*]] = ptrtoint ptr [[VL]] to i64
23+
; CHECK-NEXT: [[TMP8:%.*]] = xor i64 [[TMP7]], 549755813888
24+
; CHECK-NEXT: [[TMP9:%.*]] = inttoptr i64 [[TMP8]] to ptr
25+
; CHECK-NEXT: call void @llvm.memset.p0.i64(ptr align 8 [[TMP9]], i8 0, i64 8, i1 false)
26+
; CHECK-NEXT: call void @llvm.va_start.p0(ptr [[VL]])
27+
; CHECK-NEXT: [[TMP10:%.*]] = ptrtoint ptr [[VL]] to i64
28+
; CHECK-NEXT: [[TMP11:%.*]] = inttoptr i64 [[TMP10]] to ptr
29+
; CHECK-NEXT: [[TMP12:%.*]] = load ptr, ptr [[TMP11]], align 8
30+
; CHECK-NEXT: [[TMP13:%.*]] = ptrtoint ptr [[TMP12]] to i64
31+
; CHECK-NEXT: [[TMP14:%.*]] = xor i64 [[TMP13]], 549755813888
32+
; CHECK-NEXT: [[TMP15:%.*]] = inttoptr i64 [[TMP14]] to ptr
33+
; CHECK-NEXT: call void @llvm.memcpy.p0.p0.i64(ptr align 8 [[TMP15]], ptr align 8 [[TMP2]], i64 [[TMP1]], i1 false)
34+
; CHECK-NEXT: call void @llvm.va_end.p0(ptr [[VL]])
35+
; CHECK-NEXT: call void @llvm.lifetime.end.p0(ptr [[VL]])
36+
; CHECK-NEXT: store i32 0, ptr @__msan_retval_tls, align 8
37+
; CHECK-NEXT: ret i32 0
38+
;
739
%vl = alloca ptr, align 8
840
call void @llvm.lifetime.start.p0(ptr %vl)
941
call void @llvm.va_start(ptr %vl)
@@ -12,47 +44,62 @@ define i32 @foo(i32 %guard, ...) {
1244
ret i32 0
1345
}
1446

15-
; First, check allocation of the save area.
16-
17-
; CHECK-LABEL: @foo
18-
; CHECK: [[A:%.*]] = load {{.*}} @__msan_va_arg_overflow_size_tls
19-
; CHECK: [[C:%.*]] = alloca {{.*}} [[A]]
20-
21-
; CHECK: call void @llvm.memset.p0.i64(ptr align 8 [[C]], i8 0, i64 [[A]], i1 false)
22-
23-
; CHECK: [[D:%.*]] = call i64 @llvm.umin.i64(i64 [[A]], i64 800)
24-
; CHECK: call void @llvm.memcpy.p0.p0.i64(ptr align 8 [[C]], ptr align 8 @__msan_va_arg_tls, i64 [[D]], i1 false)
25-
2647
declare void @llvm.lifetime.start.p0(ptr nocapture) #1
2748
declare void @llvm.va_start(ptr) #2
2849
declare void @llvm.va_end(ptr) #2
2950
declare void @llvm.lifetime.end.p0(ptr nocapture) #1
3051

3152
define i32 @bar() {
53+
; CHECK-LABEL: define i32 @bar() {
54+
; CHECK-NEXT: [[TMP1:%.*]] = load i64, ptr @__msan_va_arg_overflow_size_tls, align 8
55+
; CHECK-NEXT: call void @llvm.donothing()
56+
; CHECK-NEXT: store i32 0, ptr @__msan_param_tls, align 8
57+
; CHECK-NEXT: store i32 0, ptr inttoptr (i64 add (i64 ptrtoint (ptr @__msan_param_tls to i64), i64 8) to ptr), align 8
58+
; CHECK-NEXT: store i64 0, ptr inttoptr (i64 add (i64 ptrtoint (ptr @__msan_param_tls to i64), i64 16) to ptr), align 8
59+
; CHECK-NEXT: store i64 0, ptr inttoptr (i64 add (i64 ptrtoint (ptr @__msan_param_tls to i64), i64 24) to ptr), align 8
60+
; CHECK-NEXT: store i32 0, ptr inttoptr (i64 add (i64 ptrtoint (ptr @__msan_va_arg_tls to i64), i64 4) to ptr), align 8
61+
; CHECK-NEXT: store i64 0, ptr inttoptr (i64 add (i64 ptrtoint (ptr @__msan_va_arg_tls to i64), i64 8) to ptr), align 8
62+
; CHECK-NEXT: store i64 0, ptr inttoptr (i64 add (i64 ptrtoint (ptr @__msan_va_arg_tls to i64), i64 16) to ptr), align 8
63+
; CHECK-NEXT: store i64 24, ptr @__msan_va_arg_overflow_size_tls, align 8
64+
; CHECK-NEXT: store i32 0, ptr @__msan_retval_tls, align 8
65+
; CHECK-NEXT: [[TMP2:%.*]] = call i32 (i32, ...) @foo(i32 0, i32 1, i64 2, double 3.000000e+00)
66+
; CHECK-NEXT: [[_MSRET:%.*]] = load i32, ptr @__msan_retval_tls, align 8
67+
; CHECK-NEXT: store i32 0, ptr @__msan_retval_tls, align 8
68+
; CHECK-NEXT: ret i32 [[TMP2]]
69+
;
3270
%1 = call i32 (i32, ...) @foo(i32 0, i32 1, i64 2, double 3.000000e+00)
3371
ret i32 %1
3472
}
3573

3674
; Save the incoming shadow value from the arguments in the __msan_va_arg_tls
3775
; array. The first argument is stored at position 4, since it's right
3876
; justified.
39-
; CHECK-LABEL: @bar
40-
; CHECK: store i32 0, ptr inttoptr (i64 add (i64 ptrtoint (ptr @__msan_va_arg_tls to i64), i64 4) to ptr), align 8
41-
; CHECK: store i64 0, ptr inttoptr (i64 add (i64 ptrtoint (ptr @__msan_va_arg_tls to i64), i64 8) to ptr), align 8
42-
; CHECK: store i64 0, ptr inttoptr (i64 add (i64 ptrtoint (ptr @__msan_va_arg_tls to i64), i64 16) to ptr), align 8
43-
; CHECK: store {{.*}} 24, {{.*}} @__msan_va_arg_overflow_size_tls
4477

4578
; Check multiple fixed arguments.
4679
declare i32 @foo2(i32 %g1, i32 %g2, ...)
4780
define i32 @bar2() {
81+
; CHECK-LABEL: define i32 @bar2() {
82+
; CHECK-NEXT: [[TMP1:%.*]] = load i64, ptr @__msan_va_arg_overflow_size_tls, align 8
83+
; CHECK-NEXT: call void @llvm.donothing()
84+
; CHECK-NEXT: store i32 0, ptr @__msan_param_tls, align 8
85+
; CHECK-NEXT: store i32 0, ptr inttoptr (i64 add (i64 ptrtoint (ptr @__msan_param_tls to i64), i64 8) to ptr), align 8
86+
; CHECK-NEXT: store i64 0, ptr inttoptr (i64 add (i64 ptrtoint (ptr @__msan_param_tls to i64), i64 16) to ptr), align 8
87+
; CHECK-NEXT: store i64 0, ptr inttoptr (i64 add (i64 ptrtoint (ptr @__msan_param_tls to i64), i64 24) to ptr), align 8
88+
; CHECK-NEXT: store i64 0, ptr @__msan_va_arg_tls, align 8
89+
; CHECK-NEXT: store i64 0, ptr inttoptr (i64 add (i64 ptrtoint (ptr @__msan_va_arg_tls to i64), i64 8) to ptr), align 8
90+
; CHECK-NEXT: store i64 16, ptr @__msan_va_arg_overflow_size_tls, align 8
91+
; CHECK-NEXT: store i32 0, ptr @__msan_retval_tls, align 8
92+
; CHECK-NEXT: [[TMP2:%.*]] = call i32 (i32, i32, ...) @foo2(i32 0, i32 1, i64 2, double 3.000000e+00)
93+
; CHECK-NEXT: [[_MSRET:%.*]] = load i32, ptr @__msan_retval_tls, align 8
94+
; CHECK-NEXT: store i32 0, ptr @__msan_retval_tls, align 8
95+
; CHECK-NEXT: ret i32 [[TMP2]]
96+
;
4897
%1 = call i32 (i32, i32, ...) @foo2(i32 0, i32 1, i64 2, double 3.000000e+00)
4998
ret i32 %1
5099
}
51100

52-
; CHECK-LABEL: @bar2
53-
; CHECK: store i64 0, ptr @__msan_va_arg_tls, align 8
54-
; CHECK: store i64 0, ptr inttoptr (i64 add (i64 ptrtoint (ptr @__msan_va_arg_tls to i64), i64 8) to ptr), align 8
55-
; CHECK: store {{.*}} 16, {{.*}} @__msan_va_arg_overflow_size_tls
101+
102+
; UTC_ARGS: --disable
56103

57104
; Test that MSan doesn't generate code overflowing __msan_va_arg_tls when too many arguments are
58105
; passed to a variadic function.

0 commit comments

Comments
 (0)