1; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --version 4 2; RUN: opt < %s -passes=instcombine -S | FileCheck %s 3; 4; Test that instcombine folds allocsize function calls properly. 5; Dummy arguments are inserted to verify that allocsize is picking the right 6; args, and to prove that arbitrary unfoldable values don't interfere with 7; allocsize if they're not used by allocsize. 8 9declare ptr @my_malloc(ptr, i32) allocsize(1) 10declare ptr @my_calloc(ptr, ptr, i32, i32) allocsize(2, 3) 11 12define void @test_malloc(ptr %p, ptr %r) { 13; CHECK-LABEL: define void @test_malloc( 14; CHECK-SAME: ptr [[P:%.*]], ptr [[R:%.*]]) { 15; CHECK-NEXT: [[TMP1:%.*]] = call dereferenceable_or_null(100) ptr @my_malloc(ptr null, i32 100) 16; CHECK-NEXT: store ptr [[TMP1]], ptr [[P]], align 8 17; CHECK-NEXT: store i64 100, ptr [[R]], align 8 18; CHECK-NEXT: ret void 19; 20 %1 = call ptr @my_malloc(ptr null, i32 100) 21 store ptr %1, ptr %p, align 8 ; To ensure objectsize isn't killed 22 23 %2 = call i64 @llvm.objectsize.i64.p0(ptr %1, i1 false) 24 store i64 %2, ptr %r, align 8 25 ret void 26} 27 28define void @test_calloc(ptr %p, ptr %r) { 29; CHECK-LABEL: define void @test_calloc( 30; CHECK-SAME: ptr [[P:%.*]], ptr [[R:%.*]]) { 31; CHECK-NEXT: [[TMP1:%.*]] = call dereferenceable_or_null(500) ptr @my_calloc(ptr null, ptr null, i32 100, i32 5) 32; CHECK-NEXT: store ptr [[TMP1]], ptr [[P]], align 8 33; CHECK-NEXT: store i64 500, ptr [[R]], align 8 34; CHECK-NEXT: ret void 35; 36 %1 = call ptr @my_calloc(ptr null, ptr null, i32 100, i32 5) 37 store ptr %1, ptr %p, align 8 ; To ensure objectsize isn't killed 38 39 %2 = call i64 @llvm.objectsize.i64.p0(ptr %1, i1 false) 40 store i64 %2, ptr %r, align 8 41 ret void 42} 43 44; Failure cases with non-constant values... 45define void @test_malloc_fails(ptr %p, ptr %r, i32 %n) { 46; CHECK-LABEL: define void @test_malloc_fails( 47; CHECK-SAME: ptr [[P:%.*]], ptr [[R:%.*]], i32 [[N:%.*]]) { 48; CHECK-NEXT: [[TMP1:%.*]] = call ptr @my_malloc(ptr null, i32 [[N]]) 49; CHECK-NEXT: store ptr [[TMP1]], ptr [[P]], align 8 50; CHECK-NEXT: [[TMP2:%.*]] = call i64 @llvm.objectsize.i64.p0(ptr [[TMP1]], i1 false, i1 false, i1 false) 51; CHECK-NEXT: store i64 [[TMP2]], ptr [[R]], align 8 52; CHECK-NEXT: ret void 53; 54 %1 = call ptr @my_malloc(ptr null, i32 %n) 55 store ptr %1, ptr %p, align 8 ; To ensure objectsize isn't killed 56 57 %2 = call i64 @llvm.objectsize.i64.p0(ptr %1, i1 false) 58 store i64 %2, ptr %r, align 8 59 ret void 60} 61 62define void @test_calloc_fails(ptr %p, ptr %r, i32 %n) { 63; CHECK-LABEL: define void @test_calloc_fails( 64; CHECK-SAME: ptr [[P:%.*]], ptr [[R:%.*]], i32 [[N:%.*]]) { 65; CHECK-NEXT: [[TMP1:%.*]] = call ptr @my_calloc(ptr null, ptr null, i32 [[N]], i32 5) 66; CHECK-NEXT: store ptr [[TMP1]], ptr [[P]], align 8 67; CHECK-NEXT: [[TMP2:%.*]] = call i64 @llvm.objectsize.i64.p0(ptr [[TMP1]], i1 false, i1 false, i1 false) 68; CHECK-NEXT: store i64 [[TMP2]], ptr [[R]], align 8 69; CHECK-NEXT: [[TMP3:%.*]] = call ptr @my_calloc(ptr null, ptr null, i32 100, i32 [[N]]) 70; CHECK-NEXT: store ptr [[TMP3]], ptr [[P]], align 8 71; CHECK-NEXT: [[TMP4:%.*]] = call i64 @llvm.objectsize.i64.p0(ptr [[TMP3]], i1 false, i1 false, i1 false) 72; CHECK-NEXT: store i64 [[TMP4]], ptr [[R]], align 8 73; CHECK-NEXT: ret void 74; 75 %1 = call ptr @my_calloc(ptr null, ptr null, i32 %n, i32 5) 76 store ptr %1, ptr %p, align 8 ; To ensure objectsize isn't killed 77 78 %2 = call i64 @llvm.objectsize.i64.p0(ptr %1, i1 false) 79 store i64 %2, ptr %r, align 8 80 81 82 %3 = call ptr @my_calloc(ptr null, ptr null, i32 100, i32 %n) 83 store ptr %3, ptr %p, align 8 ; To ensure objectsize isn't killed 84 85 %4 = call i64 @llvm.objectsize.i64.p0(ptr %3, i1 false) 86 store i64 %4, ptr %r, align 8 87 ret void 88} 89 90declare ptr @my_malloc_outofline(ptr, i32) #0 91declare ptr @my_calloc_outofline(ptr, ptr, i32, i32) #1 92 93; Verifying that out of line allocsize is parsed correctly 94define void @test_outofline(ptr %p, ptr %r) { 95; CHECK-LABEL: define void @test_outofline( 96; CHECK-SAME: ptr [[P:%.*]], ptr [[R:%.*]]) { 97; CHECK-NEXT: [[TMP1:%.*]] = call dereferenceable_or_null(100) ptr @my_malloc_outofline(ptr null, i32 100) 98; CHECK-NEXT: store ptr [[TMP1]], ptr [[P]], align 8 99; CHECK-NEXT: store i64 100, ptr [[R]], align 8 100; CHECK-NEXT: [[TMP2:%.*]] = call dereferenceable_or_null(500) ptr @my_calloc_outofline(ptr null, ptr null, i32 100, i32 5) 101; CHECK-NEXT: store ptr [[TMP2]], ptr [[P]], align 8 102; CHECK-NEXT: store i64 500, ptr [[R]], align 8 103; CHECK-NEXT: ret void 104; 105 %1 = call ptr @my_malloc_outofline(ptr null, i32 100) 106 store ptr %1, ptr %p, align 8 ; To ensure objectsize isn't killed 107 108 %2 = call i64 @llvm.objectsize.i64.p0(ptr %1, i1 false) 109 store i64 %2, ptr %r, align 8 110 111 112 %3 = call ptr @my_calloc_outofline(ptr null, ptr null, i32 100, i32 5) 113 store ptr %3, ptr %p, align 8 ; To ensure objectsize isn't killed 114 115 %4 = call i64 @llvm.objectsize.i64.p0(ptr %3, i1 false) 116 store i64 %4, ptr %r, align 8 117 ret void 118} 119 120declare ptr @my_malloc_i64(ptr, i64) #0 121declare ptr @my_tiny_calloc(ptr, ptr, i8, i8) #1 122declare ptr @my_varied_calloc(ptr, ptr, i32, i8) #1 123 124define void @test_overflow(ptr %p, ptr %r) { 125 ; (2**31 + 1) * 2 > 2**31. So overflow. Yay. 126; CHECK-LABEL: define void @test_overflow( 127; CHECK-SAME: ptr [[P:%.*]], ptr [[R:%.*]]) { 128; CHECK-NEXT: [[BIG_MALLOC:%.*]] = call dereferenceable_or_null(4294967298) ptr @my_calloc(ptr null, ptr null, i32 -2147483647, i32 2) 129; CHECK-NEXT: store ptr [[BIG_MALLOC]], ptr [[P]], align 8 130; CHECK-NEXT: [[TMP1:%.*]] = call i32 @llvm.objectsize.i32.p0(ptr [[BIG_MALLOC]], i1 false, i1 false, i1 false) 131; CHECK-NEXT: store i32 [[TMP1]], ptr [[R]], align 4 132; CHECK-NEXT: [[BIG_LITTLE_MALLOC:%.*]] = call dereferenceable_or_null(508) ptr @my_tiny_calloc(ptr null, ptr null, i8 127, i8 4) 133; CHECK-NEXT: store ptr [[BIG_LITTLE_MALLOC]], ptr [[P]], align 8 134; CHECK-NEXT: store i32 508, ptr [[R]], align 4 135; CHECK-NEXT: [[BIG_MALLOC_I64:%.*]] = call dereferenceable_or_null(8589934592) ptr @my_malloc_i64(ptr null, i64 8589934592) 136; CHECK-NEXT: store ptr [[BIG_MALLOC_I64]], ptr [[P]], align 8 137; CHECK-NEXT: [[TMP2:%.*]] = call i32 @llvm.objectsize.i32.p0(ptr [[BIG_MALLOC_I64]], i1 false, i1 false, i1 false) 138; CHECK-NEXT: store i32 [[TMP2]], ptr [[R]], align 4 139; CHECK-NEXT: store i64 8589934592, ptr [[R]], align 8 140; CHECK-NEXT: [[VARIED_CALLOC:%.*]] = call dereferenceable_or_null(5000) ptr @my_varied_calloc(ptr null, ptr null, i32 1000, i8 5) 141; CHECK-NEXT: store ptr [[VARIED_CALLOC]], ptr [[P]], align 8 142; CHECK-NEXT: store i32 5000, ptr [[R]], align 4 143; CHECK-NEXT: ret void 144; 145 %big_malloc = call ptr @my_calloc(ptr null, ptr null, i32 2147483649, i32 2) 146 store ptr %big_malloc, ptr %p, align 8 147 148 %1 = call i32 @llvm.objectsize.i32.p0(ptr %big_malloc, i1 false) 149 store i32 %1, ptr %r, align 4 150 151 152 %big_little_malloc = call ptr @my_tiny_calloc(ptr null, ptr null, i8 127, i8 4) 153 store ptr %big_little_malloc, ptr %p, align 8 154 155 %2 = call i32 @llvm.objectsize.i32.p0(ptr %big_little_malloc, i1 false) 156 store i32 %2, ptr %r, align 4 157 158 159 ; malloc(2**33) 160 %big_malloc_i64 = call ptr @my_malloc_i64(ptr null, i64 8589934592) 161 store ptr %big_malloc_i64, ptr %p, align 8 162 163 %3 = call i32 @llvm.objectsize.i32.p0(ptr %big_malloc_i64, i1 false) 164 store i32 %3, ptr %r, align 4 165 166 167 %4 = call i64 @llvm.objectsize.i64.p0(ptr %big_malloc_i64, i1 false) 168 store i64 %4, ptr %r, align 8 169 170 171 ; Just intended to ensure that we properly handle args of different types... 172 %varied_calloc = call ptr @my_varied_calloc(ptr null, ptr null, i32 1000, i8 5) 173 store ptr %varied_calloc, ptr %p, align 8 174 175 %5 = call i32 @llvm.objectsize.i32.p0(ptr %varied_calloc, i1 false) 176 store i32 %5, ptr %r, align 4 177 178 ret void 179} 180 181; We had a bug where `nobuiltin` would cause `allocsize` to be ignored in 182; @llvm.objectsize calculations. 183define void @test_nobuiltin(ptr %p, ptr %r) { 184; CHECK-LABEL: define void @test_nobuiltin( 185; CHECK-SAME: ptr [[P:%.*]], ptr [[R:%.*]]) { 186; CHECK-NEXT: [[TMP1:%.*]] = call dereferenceable_or_null(100) ptr @my_malloc(ptr null, i32 100) #[[ATTR3:[0-9]+]] 187; CHECK-NEXT: store ptr [[TMP1]], ptr [[P]], align 8 188; CHECK-NEXT: store i64 100, ptr [[R]], align 8 189; CHECK-NEXT: ret void 190; 191 %1 = call ptr @my_malloc(ptr null, i32 100) nobuiltin 192 store ptr %1, ptr %p, align 8 193 194 %2 = call i64 @llvm.objectsize.i64.p0(ptr %1, i1 false) 195 store i64 %2, ptr %r, align 8 196 ret void 197} 198 199attributes #0 = { allocsize(1) } 200attributes #1 = { allocsize(2, 3) } 201 202declare i32 @llvm.objectsize.i32.p0(ptr, i1) 203declare i64 @llvm.objectsize.i64.p0(ptr, i1) 204