Home | History | Annotate | Download | only in InstCombine
      1 ; Test lib call simplification of __stpcpy_chk calls with various values
      2 ; for src, dst, and slen.
      3 ;
      4 ; RUN: opt < %s -instcombine -S | FileCheck %s
      5 
      6 target datalayout = "e-p:32:32:32-i1:8:8-i8:8:8-i16:16:16-i32:32:32-i64:32:64-f32:32:32-f64:32:64-v64:64:64-v128:128:128-a0:0:64-f80:128:128"
      7 
      8 @a = common global [60 x i8] zeroinitializer, align 1
      9 @b = common global [60 x i8] zeroinitializer, align 1
     10 @.str = private constant [12 x i8] c"abcdefghijk\00"
     11 
     12 ; Check cases where slen >= strlen (src).
     13 
     14 define i8* @test_simplify1() {
     15 ; CHECK-LABEL: @test_simplify1(
     16   %dst = getelementptr inbounds [60 x i8], [60 x i8]* @a, i32 0, i32 0
     17   %src = getelementptr inbounds [12 x i8], [12 x i8]* @.str, i32 0, i32 0
     18 
     19 ; CHECK-NEXT: call void @llvm.memcpy.p0i8.p0i8.i32(i8* getelementptr inbounds ([60 x i8], [60 x i8]* @a, i32 0, i32 0), i8* getelementptr inbounds ([12 x i8], [12 x i8]* @.str, i32 0, i32 0), i32 12, i32 1, i1 false)
     20 ; CHECK-NEXT: ret i8* getelementptr inbounds ([60 x i8], [60 x i8]* @a, i32 0, i32 11)
     21   %ret = call i8* @__stpcpy_chk(i8* %dst, i8* %src, i32 60)
     22   ret i8* %ret
     23 }
     24 
     25 define i8* @test_simplify2() {
     26 ; CHECK-LABEL: @test_simplify2(
     27   %dst = getelementptr inbounds [60 x i8], [60 x i8]* @a, i32 0, i32 0
     28   %src = getelementptr inbounds [12 x i8], [12 x i8]* @.str, i32 0, i32 0
     29 
     30 ; CHECK-NEXT: call void @llvm.memcpy.p0i8.p0i8.i32(i8* getelementptr inbounds ([60 x i8], [60 x i8]* @a, i32 0, i32 0), i8* getelementptr inbounds ([12 x i8], [12 x i8]* @.str, i32 0, i32 0), i32 12, i32 1, i1 false)
     31 ; CHECK-NEXT: ret i8* getelementptr inbounds ([60 x i8], [60 x i8]* @a, i32 0, i32 11)
     32   %ret = call i8* @__stpcpy_chk(i8* %dst, i8* %src, i32 12)
     33   ret i8* %ret
     34 }
     35 
     36 define i8* @test_simplify3() {
     37 ; CHECK-LABEL: @test_simplify3(
     38   %dst = getelementptr inbounds [60 x i8], [60 x i8]* @a, i32 0, i32 0
     39   %src = getelementptr inbounds [12 x i8], [12 x i8]* @.str, i32 0, i32 0
     40 
     41 ; CHECK-NEXT: call void @llvm.memcpy.p0i8.p0i8.i32(i8* getelementptr inbounds ([60 x i8], [60 x i8]* @a, i32 0, i32 0), i8* getelementptr inbounds ([12 x i8], [12 x i8]* @.str, i32 0, i32 0), i32 12, i32 1, i1 false)
     42 ; CHECK-NEXT: ret i8* getelementptr inbounds ([60 x i8], [60 x i8]* @a, i32 0, i32 11)
     43   %ret = call i8* @__stpcpy_chk(i8* %dst, i8* %src, i32 -1)
     44   ret i8* %ret
     45 }
     46 
     47 ; Check cases where there are no string constants.
     48 
     49 define i8* @test_simplify4() {
     50 ; CHECK-LABEL: @test_simplify4(
     51   %dst = getelementptr inbounds [60 x i8], [60 x i8]* @a, i32 0, i32 0
     52   %src = getelementptr inbounds [60 x i8], [60 x i8]* @b, i32 0, i32 0
     53 
     54 ; CHECK-NEXT: %stpcpy = call i8* @stpcpy(i8* getelementptr inbounds ([60 x i8], [60 x i8]* @a, i32 0, i32 0), i8* getelementptr inbounds ([60 x i8], [60 x i8]* @b, i32 0, i32 0))
     55 ; CHECK-NEXT: ret i8* %stpcpy
     56   %ret = call i8* @__stpcpy_chk(i8* %dst, i8* %src, i32 -1)
     57   ret i8* %ret
     58 }
     59 
     60 ; Check case where the string length is not constant.
     61 
     62 define i8* @test_simplify5() {
     63 ; CHECK-LABEL: @test_simplify5(
     64   %dst = getelementptr inbounds [60 x i8], [60 x i8]* @a, i32 0, i32 0
     65   %src = getelementptr inbounds [12 x i8], [12 x i8]* @.str, i32 0, i32 0
     66 
     67 ; CHECK-NEXT: %len = call i32 @llvm.objectsize.i32.p0i8(i8* getelementptr inbounds ([60 x i8], [60 x i8]* @a, i32 0, i32 0), i1 false)
     68 ; CHECK-NEXT: %1 = call i8* @__memcpy_chk(i8* getelementptr inbounds ([60 x i8], [60 x i8]* @a, i32 0, i32 0), i8* getelementptr inbounds ([12 x i8], [12 x i8]* @.str, i32 0, i32 0), i32 12, i32 %len)
     69 ; CHECK-NEXT: ret i8* getelementptr inbounds ([60 x i8], [60 x i8]* @a, i32 0, i32 11)
     70   %len = call i32 @llvm.objectsize.i32.p0i8(i8* %dst, i1 false)
     71   %ret = call i8* @__stpcpy_chk(i8* %dst, i8* %src, i32 %len)
     72   ret i8* %ret
     73 }
     74 
     75 ; Check case where the source and destination are the same.
     76 
     77 define i8* @test_simplify6() {
     78 ; CHECK-LABEL: @test_simplify6(
     79   %dst = getelementptr inbounds [60 x i8], [60 x i8]* @a, i32 0, i32 0
     80 
     81 ; CHECK-NEXT: %strlen = call i32 @strlen(i8* getelementptr inbounds ([60 x i8], [60 x i8]* @a, i32 0, i32 0))
     82 ; CHECK-NEXT: %1 = getelementptr inbounds [60 x i8], [60 x i8]* @a, i32 0, i32 %strlen
     83 ; CHECK-NEXT: ret i8* %1
     84   %len = call i32 @llvm.objectsize.i32.p0i8(i8* %dst, i1 false)
     85   %ret = call i8* @__stpcpy_chk(i8* %dst, i8* %dst, i32 %len)
     86   ret i8* %ret
     87 }
     88 
     89 ; Check case where slen < strlen (src).
     90 
     91 define i8* @test_no_simplify1() {
     92 ; CHECK-LABEL: @test_no_simplify1(
     93   %dst = getelementptr inbounds [60 x i8], [60 x i8]* @a, i32 0, i32 0
     94   %src = getelementptr inbounds [60 x i8], [60 x i8]* @b, i32 0, i32 0
     95 
     96 ; CHECK-NEXT: %ret = call i8* @__stpcpy_chk(i8* getelementptr inbounds ([60 x i8], [60 x i8]* @a, i32 0, i32 0), i8* getelementptr inbounds ([60 x i8], [60 x i8]* @b, i32 0, i32 0), i32 8)
     97 ; CHECK-NEXT: ret i8* %ret
     98   %ret = call i8* @__stpcpy_chk(i8* %dst, i8* %src, i32 8)
     99   ret i8* %ret
    100 }
    101 
    102 declare i8* @__stpcpy_chk(i8*, i8*, i32) nounwind
    103 declare i32 @llvm.objectsize.i32.p0i8(i8*, i1) nounwind readonly
    104