llvm/test/Transforms/InstCombine/stpcpy_chk-1.ll

   1 ; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
   2 ; Test lib call simplification of __stpcpy_chk calls with various values
   3 ; for src, dst, and slen.
   4 ;
   5 ; RUN: opt < %s -instcombine -S | FileCheck %s
   6
   7 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"
   8
   9 @a = common global [60 x i8] zeroinitializer, align 1
  10 @b = common global [60 x i8] zeroinitializer, align 1
  11 @.str = private constant [12 x i8] c"abcdefghijk\00"
  12
  13 ; Check cases where slen >= strlen (src).
  14
  15 define i8* @test_simplify1() {
  16 ; CHECK-LABEL: @test_simplify1(
  17 ; CHECK-NEXT:    call void @llvm.memcpy.p0i8.p0i8.i32(i8* noundef nonnull align 1 dereferenceable(12) getelementptr inbounds ([60 x i8], [60 x i8]* @a, i32 0, i32 0), i8* noundef nonnull align 1 dereferenceable(12) getelementptr inbounds ([12 x i8], [12 x i8]* @.str, i32 0, i32 0), i32 12, i1 false)
  18 ; CHECK-NEXT:    ret i8* getelementptr inbounds ([60 x i8], [60 x i8]* @a, i32 0, i32 11)
  19 ;
  20   %dst = getelementptr inbounds [60 x i8], [60 x i8]* @a, i32 0, i32 0
  21   %src = getelementptr inbounds [12 x i8], [12 x i8]* @.str, i32 0, i32 0
  22
  23   %ret = call i8* @__stpcpy_chk(i8* %dst, i8* %src, i32 60)
  24   ret i8* %ret
  25 }
  26
  27 define i8* @test_simplify2() {
  28 ; CHECK-LABEL: @test_simplify2(
  29 ; CHECK-NEXT:    call void @llvm.memcpy.p0i8.p0i8.i32(i8* noundef nonnull align 1 dereferenceable(12) getelementptr inbounds ([60 x i8], [60 x i8]* @a, i32 0, i32 0), i8* noundef nonnull align 1 dereferenceable(12) getelementptr inbounds ([12 x i8], [12 x i8]* @.str, i32 0, i32 0), i32 12, i1 false)
  30 ; CHECK-NEXT:    ret i8* getelementptr inbounds ([60 x i8], [60 x i8]* @a, i32 0, i32 11)
  31 ;
  32   %dst = getelementptr inbounds [60 x i8], [60 x i8]* @a, i32 0, i32 0
  33   %src = getelementptr inbounds [12 x i8], [12 x i8]* @.str, i32 0, i32 0
  34
  35   %ret = call i8* @__stpcpy_chk(i8* %dst, i8* %src, i32 12)
  36   ret i8* %ret
  37 }
  38
  39 define i8* @test_simplify3() {
  40 ; CHECK-LABEL: @test_simplify3(
  41 ; CHECK-NEXT:    call void @llvm.memcpy.p0i8.p0i8.i32(i8* noundef nonnull align 1 dereferenceable(12) getelementptr inbounds ([60 x i8], [60 x i8]* @a, i32 0, i32 0), i8* noundef nonnull align 1 dereferenceable(12) getelementptr inbounds ([12 x i8], [12 x i8]* @.str, i32 0, i32 0), i32 12, i1 false)
  42 ; CHECK-NEXT:    ret i8* getelementptr inbounds ([60 x i8], [60 x i8]* @a, i32 0, i32 11)
  43 ;
  44   %dst = getelementptr inbounds [60 x i8], [60 x i8]* @a, i32 0, i32 0
  45   %src = getelementptr inbounds [12 x i8], [12 x i8]* @.str, i32 0, i32 0
  46
  47   %ret = call i8* @__stpcpy_chk(i8* %dst, i8* %src, i32 -1)
  48   ret i8* %ret
  49 }
  50
  51 define i8* @test_simplify1_tail() {
  52 ; CHECK-LABEL: @test_simplify1_tail(
  53 ; CHECK-NEXT:    call void @llvm.memcpy.p0i8.p0i8.i32(i8* noundef nonnull align 1 dereferenceable(12) getelementptr inbounds ([60 x i8], [60 x i8]* @a, i32 0, i32 0), i8* noundef nonnull align 1 dereferenceable(12) getelementptr inbounds ([12 x i8], [12 x i8]* @.str, i32 0, i32 0), i32 12, i1 false)
  54 ; CHECK-NEXT:    ret i8* getelementptr inbounds ([60 x i8], [60 x i8]* @a, i32 0, i32 11)
  55 ;
  56   %dst = getelementptr inbounds [60 x i8], [60 x i8]* @a, i32 0, i32 0
  57   %src = getelementptr inbounds [12 x i8], [12 x i8]* @.str, i32 0, i32 0
  58
  59   %ret = tail call i8* @__stpcpy_chk(i8* %dst, i8* %src, i32 60)
  60   ret i8* %ret
  61 }
  62
  63 ; Check cases where there are no string constants.
  64
  65 define i8* @test_simplify4() {
  66 ; CHECK-LABEL: @test_simplify4(
  67 ; 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))
  68 ; CHECK-NEXT:    ret i8* [[STPCPY]]
  69 ;
  70   %dst = getelementptr inbounds [60 x i8], [60 x i8]* @a, i32 0, i32 0
  71   %src = getelementptr inbounds [60 x i8], [60 x i8]* @b, i32 0, i32 0
  72
  73   %ret = call i8* @__stpcpy_chk(i8* %dst, i8* %src, i32 -1)
  74   ret i8* %ret
  75 }
  76
  77 define i8* @test_simplify4_tail() {
  78 ; CHECK-LABEL: @test_simplify4_tail(
  79 ; CHECK-NEXT:    [[STPCPY:%.*]] = tail 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))
  80 ; CHECK-NEXT:    ret i8* [[STPCPY]]
  81 ;
  82   %dst = getelementptr inbounds [60 x i8], [60 x i8]* @a, i32 0, i32 0
  83   %src = getelementptr inbounds [60 x i8], [60 x i8]* @b, i32 0, i32 0
  84
  85   %ret = tail call i8* @__stpcpy_chk(i8* %dst, i8* %src, i32 -1)
  86   ret i8* %ret
  87 }
  88
  89 ; Check case where the string length is not constant.
  90
  91 define i8* @test_simplify5() {
  92 ; CHECK-LABEL: @test_simplify5(
  93 ; 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, i1 false, i1 false)
  94 ; CHECK-NEXT:    [[TMP1:%.*]] = 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]])
  95 ; CHECK-NEXT:    ret i8* getelementptr inbounds ([60 x i8], [60 x i8]* @a, i32 0, i32 11)
  96 ;
  97   %dst = getelementptr inbounds [60 x i8], [60 x i8]* @a, i32 0, i32 0
  98   %src = getelementptr inbounds [12 x i8], [12 x i8]* @.str, i32 0, i32 0
  99
 100   %len = call i32 @llvm.objectsize.i32.p0i8(i8* %dst, i1 false, i1 false, i1 false)
 101   %ret = call i8* @__stpcpy_chk(i8* %dst, i8* %src, i32 %len)
 102   ret i8* %ret
 103 }
 104
 105 ; Check case where the source and destination are the same.
 106
 107 define i8* @test_simplify6() {
 108 ; CHECK-LABEL: @test_simplify6(
 109 ; CHECK-NEXT:    [[STRLEN:%.*]] = call i32 @strlen(i8* noundef nonnull dereferenceable(1) getelementptr inbounds ([60 x i8], [60 x i8]* @a, i32 0, i32 0))
 110 ; CHECK-NEXT:    [[TMP1:%.*]] = getelementptr inbounds [60 x i8], [60 x i8]* @a, i32 0, i32 [[STRLEN]]
 111 ; CHECK-NEXT:    ret i8* [[TMP1]]
 112 ;
 113   %dst = getelementptr inbounds [60 x i8], [60 x i8]* @a, i32 0, i32 0
 114
 115   %len = call i32 @llvm.objectsize.i32.p0i8(i8* %dst, i1 false, i1 false, i1 false)
 116   %ret = call i8* @__stpcpy_chk(i8* %dst, i8* %dst, i32 %len)
 117   ret i8* %ret
 118 }
 119
 120 ; Check cases where there are no string constants, and is a tail call.
 121
 122 define i8* @test_simplify7() {
 123 ; CHECK-LABEL: @test_simplify7(
 124 ; CHECK-NEXT:    [[STPCPY:%.*]] = tail 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))
 125 ; CHECK-NEXT:    ret i8* [[STPCPY]]
 126 ;
 127   %dst = getelementptr inbounds [60 x i8], [60 x i8]* @a, i32 0, i32 0
 128   %src = getelementptr inbounds [60 x i8], [60 x i8]* @b, i32 0, i32 0
 129
 130   %ret = tail call i8* @__stpcpy_chk(i8* %dst, i8* %src, i32 -1)
 131   ret i8* %ret
 132 }
 133
 134 ; Check case where slen < strlen (src).
 135
 136 define i8* @test_no_simplify1() {
 137 ; CHECK-LABEL: @test_no_simplify1(
 138 ; 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)
 139 ; CHECK-NEXT:    ret i8* [[RET]]
 140 ;
 141   %dst = getelementptr inbounds [60 x i8], [60 x i8]* @a, i32 0, i32 0
 142   %src = getelementptr inbounds [60 x i8], [60 x i8]* @b, i32 0, i32 0
 143
 144   %ret = call i8* @__stpcpy_chk(i8* %dst, i8* %src, i32 8)
 145   ret i8* %ret
 146 }
 147
 148 declare i8* @__stpcpy_chk(i8*, i8*, i32) nounwind
 149 declare i32 @llvm.objectsize.i32.p0i8(i8*, i1, i1, i1) nounwind readonly