1; NOTE: Assertions have been autogenerated by utils/update_test_checks.py 2; Test that the wcslen library call simplifier works correctly. 3; 4; RUN: opt < %s -instcombine -S | FileCheck %s 5 6target datalayout = "e-m:o-i64:64-f80:128-n8:16:32:64-S128" 7 8; Test behavior for wchar_size==2 9!llvm.module.flags = !{!0} 10!0 = !{i32 1, !"wchar_size", i32 2} 11 12declare i64 @wcslen(i16*) 13 14@hello = constant [6 x i16] [i16 104, i16 101, i16 108, i16 108, i16 111, i16 0] 15@longer = constant [7 x i16] [i16 108, i16 111, i16 110, i16 103, i16 101, i16 114, i16 0] 16@null = constant [1 x i16] zeroinitializer 17@null_hello = constant [7 x i16] [i16 0, i16 104, i16 101, i16 108, i16 108, i16 111, i16 0] 18@nullstring = constant i16 0 19@a = common global [32 x i16] zeroinitializer, align 1 20@null_hello_mid = constant [13 x i16] [i16 104, i16 101, i16 108, i16 108, i16 111, i16 32, i16 119, i16 111, i16 114, i16 0, i16 108, i16 100, i16 0] 21 22define i64 @test_simplify1() { 23; CHECK-LABEL: @test_simplify1( 24; CHECK-NEXT: ret i64 5 25; 26 %hello_p = getelementptr [6 x i16], [6 x i16]* @hello, i64 0, i64 0 27 %hello_l = call i64 @wcslen(i16* %hello_p) 28 ret i64 %hello_l 29} 30 31define i64 @test_simplify2() { 32; CHECK-LABEL: @test_simplify2( 33; CHECK-NEXT: ret i64 0 34; 35 %null_p = getelementptr [1 x i16], [1 x i16]* @null, i64 0, i64 0 36 %null_l = call i64 @wcslen(i16* %null_p) 37 ret i64 %null_l 38} 39 40define i64 @test_simplify3() { 41; CHECK-LABEL: @test_simplify3( 42; CHECK-NEXT: ret i64 0 43; 44 %null_hello_p = getelementptr [7 x i16], [7 x i16]* @null_hello, i64 0, i64 0 45 %null_hello_l = call i64 @wcslen(i16* %null_hello_p) 46 ret i64 %null_hello_l 47} 48 49define i64 @test_simplify4() { 50; CHECK-LABEL: @test_simplify4( 51; CHECK-NEXT: ret i64 0 52; 53 %len = tail call i64 @wcslen(i16* @nullstring) nounwind 54 ret i64 %len 55} 56 57; Check wcslen(x) == 0 --> *x == 0. 58 59define i1 @test_simplify5() { 60; CHECK-LABEL: @test_simplify5( 61; CHECK-NEXT: ret i1 false 62; 63 %hello_p = getelementptr [6 x i16], [6 x i16]* @hello, i64 0, i64 0 64 %hello_l = call i64 @wcslen(i16* %hello_p) 65 %eq_hello = icmp eq i64 %hello_l, 0 66 ret i1 %eq_hello 67} 68 69define i1 @test_simplify6(i16* %str_p) { 70; CHECK-LABEL: @test_simplify6( 71; CHECK-NEXT: [[STRLENFIRST:%.*]] = load i16, i16* [[STR_P:%.*]], align 2 72; CHECK-NEXT: [[EQ_NULL:%.*]] = icmp eq i16 [[STRLENFIRST]], 0 73; CHECK-NEXT: ret i1 [[EQ_NULL]] 74; 75 %str_l = call i64 @wcslen(i16* %str_p) 76 %eq_null = icmp eq i64 %str_l, 0 77 ret i1 %eq_null 78} 79 80; Check wcslen(x) != 0 --> *x != 0. 81 82define i1 @test_simplify7() { 83; CHECK-LABEL: @test_simplify7( 84; CHECK-NEXT: ret i1 true 85; 86 %hello_p = getelementptr [6 x i16], [6 x i16]* @hello, i64 0, i64 0 87 %hello_l = call i64 @wcslen(i16* %hello_p) 88 %ne_hello = icmp ne i64 %hello_l, 0 89 ret i1 %ne_hello 90} 91 92define i1 @test_simplify8(i16* %str_p) { 93; CHECK-LABEL: @test_simplify8( 94; CHECK-NEXT: [[STRLENFIRST:%.*]] = load i16, i16* [[STR_P:%.*]], align 2 95; CHECK-NEXT: [[NE_NULL:%.*]] = icmp ne i16 [[STRLENFIRST]], 0 96; CHECK-NEXT: ret i1 [[NE_NULL]] 97; 98 %str_l = call i64 @wcslen(i16* %str_p) 99 %ne_null = icmp ne i64 %str_l, 0 100 ret i1 %ne_null 101} 102 103define i64 @test_simplify9(i1 %x) { 104; CHECK-LABEL: @test_simplify9( 105; CHECK-NEXT: [[TMP1:%.*]] = select i1 [[X:%.*]], i64 5, i64 6 106; CHECK-NEXT: ret i64 [[TMP1]] 107; 108 %hello = getelementptr [6 x i16], [6 x i16]* @hello, i64 0, i64 0 109 %longer = getelementptr [7 x i16], [7 x i16]* @longer, i64 0, i64 0 110 %s = select i1 %x, i16* %hello, i16* %longer 111 %l = call i64 @wcslen(i16* %s) 112 ret i64 %l 113} 114 115; Check the case that should be simplified to a sub instruction. 116; wcslen(@hello + x) --> 5 - x 117 118define i64 @test_simplify10(i16 %x) { 119; CHECK-LABEL: @test_simplify10( 120; CHECK-NEXT: [[TMP1:%.*]] = sext i16 [[X:%.*]] to i64 121; CHECK-NEXT: [[TMP2:%.*]] = sub nsw i64 5, [[TMP1]] 122; CHECK-NEXT: ret i64 [[TMP2]] 123; 124 %hello_p = getelementptr inbounds [6 x i16], [6 x i16]* @hello, i16 0, i16 %x 125 %hello_l = call i64 @wcslen(i16* %hello_p) 126 ret i64 %hello_l 127} 128 129; wcslen(@null_hello_mid + (x & 7)) --> 9 - (x & 7) 130 131define i64 @test_simplify11(i16 %x) { 132; CHECK-LABEL: @test_simplify11( 133; CHECK-NEXT: [[AND:%.*]] = and i16 [[X:%.*]], 7 134; CHECK-NEXT: [[NARROW:%.*]] = sub nuw nsw i16 9, [[AND]] 135; CHECK-NEXT: [[TMP1:%.*]] = zext i16 [[NARROW]] to i64 136; CHECK-NEXT: ret i64 [[TMP1]] 137; 138 %and = and i16 %x, 7 139 %hello_p = getelementptr inbounds [13 x i16], [13 x i16]* @null_hello_mid, i16 0, i16 %and 140 %hello_l = call i64 @wcslen(i16* %hello_p) 141 ret i64 %hello_l 142} 143 144; Check cases that shouldn't be simplified. 145 146define i64 @test_no_simplify1() { 147; CHECK-LABEL: @test_no_simplify1( 148; CHECK-NEXT: [[A_L:%.*]] = call i64 @wcslen(i16* getelementptr inbounds ([32 x i16], [32 x i16]* @a, i64 0, i64 0)) 149; CHECK-NEXT: ret i64 [[A_L]] 150; 151 %a_p = getelementptr [32 x i16], [32 x i16]* @a, i64 0, i64 0 152 %a_l = call i64 @wcslen(i16* %a_p) 153 ret i64 %a_l 154} 155 156; wcslen(@null_hello + x) should not be simplified to a sub instruction. 157 158define i64 @test_no_simplify2(i16 %x) { 159; CHECK-LABEL: @test_no_simplify2( 160; CHECK-NEXT: [[TMP1:%.*]] = sext i16 [[X:%.*]] to i64 161; CHECK-NEXT: [[HELLO_P:%.*]] = getelementptr inbounds [7 x i16], [7 x i16]* @null_hello, i64 0, i64 [[TMP1]] 162; CHECK-NEXT: [[HELLO_L:%.*]] = call i64 @wcslen(i16* nonnull [[HELLO_P]]) 163; CHECK-NEXT: ret i64 [[HELLO_L]] 164; 165 %hello_p = getelementptr inbounds [7 x i16], [7 x i16]* @null_hello, i16 0, i16 %x 166 %hello_l = call i64 @wcslen(i16* %hello_p) 167 ret i64 %hello_l 168} 169 170; wcslen(@null_hello_mid + (x & 15)) should not be simplified to a sub instruction. 171 172define i64 @test_no_simplify3(i16 %x) { 173; CHECK-LABEL: @test_no_simplify3( 174; CHECK-NEXT: [[AND:%.*]] = and i16 [[X:%.*]], 15 175; CHECK-NEXT: [[TMP1:%.*]] = zext i16 [[AND]] to i64 176; CHECK-NEXT: [[HELLO_P:%.*]] = getelementptr inbounds [13 x i16], [13 x i16]* @null_hello_mid, i64 0, i64 [[TMP1]] 177; CHECK-NEXT: [[HELLO_L:%.*]] = call i64 @wcslen(i16* nonnull [[HELLO_P]]) 178; CHECK-NEXT: ret i64 [[HELLO_L]] 179; 180 %and = and i16 %x, 15 181 %hello_p = getelementptr inbounds [13 x i16], [13 x i16]* @null_hello_mid, i16 0, i16 %and 182 %hello_l = call i64 @wcslen(i16* %hello_p) 183 ret i64 %hello_l 184} 185 186@str32 = constant [1 x i32] [i32 0] 187 188; This could in principle be simplified, but the current implementation bails on 189; type mismatches. 190define i64 @test_no_simplify4() { 191; CHECK-LABEL: @test_no_simplify4( 192; CHECK-NEXT: [[L:%.*]] = call i64 @wcslen(i16* bitcast ([1 x i32]* @str32 to i16*)) 193; CHECK-NEXT: ret i64 [[L]] 194; 195 %l = call i64 @wcslen(i16* bitcast ([1 x i32]* @str32 to i16*)) 196 ret i64 %l 197} 198