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

   1 ; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
   2 ; Test that the strchr library call simplifier works correctly.
   3 ; RUN: opt < %s -instcombine -S | FileCheck %s
   4
   5 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-n8:16:32"
   6
   7 @hello = constant [14 x i8] c"hello world\5Cn\00"
   8 @null = constant [1 x i8] zeroinitializer
   9 @newlines = constant [3 x i8] c"\0D\0A\00"
  10 @chp = global i8* zeroinitializer
  11
  12 declare i8* @strchr(i8*, i32)
  13
  14 define void @test_simplify1() {
  15 ; CHECK-LABEL: @test_simplify1(
  16 ; CHECK-NEXT:    store i8* getelementptr inbounds ([14 x i8], [14 x i8]* @hello, i32 0, i32 6), i8** @chp, align 4
  17 ; CHECK-NEXT:    ret void
  18 ;
  19
  20   %str = getelementptr [14 x i8], [14 x i8]* @hello, i32 0, i32 0
  21   %dst = call i8* @strchr(i8* %str, i32 119)
  22   store i8* %dst, i8** @chp
  23   ret void
  24 }
  25
  26 define void @test_simplify2() {
  27 ; CHECK-LABEL: @test_simplify2(
  28 ; CHECK-NEXT:    store i8* null, i8** @chp, align 4
  29 ; CHECK-NEXT:    ret void
  30 ;
  31
  32   %str = getelementptr [1 x i8], [1 x i8]* @null, i32 0, i32 0
  33   %dst = call i8* @strchr(i8* %str, i32 119)
  34   store i8* %dst, i8** @chp
  35   ret void
  36 }
  37
  38 define void @test_simplify3() {
  39 ; CHECK-LABEL: @test_simplify3(
  40 ; CHECK-NEXT:    store i8* getelementptr inbounds ([14 x i8], [14 x i8]* @hello, i32 0, i32 13), i8** @chp, align 4
  41 ; CHECK-NEXT:    ret void
  42 ;
  43
  44   %src = getelementptr [14 x i8], [14 x i8]* @hello, i32 0, i32 0
  45   %dst = call i8* @strchr(i8* %src, i32 0)
  46   store i8* %dst, i8** @chp
  47   ret void
  48 }
  49
  50 define void @test_simplify4(i32 %chr) {
  51 ; CHECK-LABEL: @test_simplify4(
  52 ; CHECK-NEXT:    [[MEMCHR:%.*]] = call i8* @memchr(i8* noundef nonnull dereferenceable(14) getelementptr inbounds ([14 x i8], [14 x i8]* @hello, i32 0, i32 0), i32 [[CHR:%.*]], i32 14)
  53 ; CHECK-NEXT:    store i8* [[MEMCHR]], i8** @chp, align 4
  54 ; CHECK-NEXT:    ret void
  55 ;
  56
  57   %src = getelementptr [14 x i8], [14 x i8]* @hello, i32 0, i32 0
  58   %dst = call i8* @strchr(i8* %src, i32 %chr)
  59   store i8* %dst, i8** @chp
  60   ret void
  61 }
  62
  63 define void @test_simplify5() {
  64 ; CHECK-LABEL: @test_simplify5(
  65 ; CHECK-NEXT:    store i8* getelementptr inbounds ([14 x i8], [14 x i8]* @hello, i32 0, i32 13), i8** @chp, align 4
  66 ; CHECK-NEXT:    ret void
  67 ;
  68
  69   %src = getelementptr [14 x i8], [14 x i8]* @hello, i32 0, i32 0
  70   %dst = call i8* @strchr(i8* %src, i32 65280)
  71   store i8* %dst, i8** @chp
  72   ret void
  73 }
  74
  75 ; Check transformation strchr(p, 0) -> p + strlen(p)
  76 define void @test_simplify6(i8* %str) {
  77 ; CHECK-LABEL: @test_simplify6(
  78 ; CHECK-NEXT:    [[STRLEN:%.*]] = call i32 @strlen(i8* noundef nonnull dereferenceable(1) [[STR:%.*]])
  79 ; CHECK-NEXT:    [[STRCHR:%.*]] = getelementptr i8, i8* [[STR]], i32 [[STRLEN]]
  80 ; CHECK-NEXT:    store i8* [[STRCHR]], i8** @chp, align 4
  81 ; CHECK-NEXT:    ret void
  82 ;
  83
  84   %dst = call i8* @strchr(i8* %str, i32 0)
  85   store i8* %dst, i8** @chp
  86   ret void
  87 }
  88
  89 ; Check transformation strchr("\r\n", C) != nullptr -> (C & 9217) != 0
  90 define i1 @test_simplify7(i32 %C) {
  91 ; CHECK-LABEL: @test_simplify7(
  92 ; CHECK-NEXT:    [[TMP1:%.*]] = trunc i32 [[C:%.*]] to i16
  93 ; CHECK-NEXT:    [[TMP2:%.*]] = and i16 [[TMP1]], 255
  94 ; CHECK-NEXT:    [[MEMCHR_BOUNDS:%.*]] = icmp ult i16 [[TMP2]], 16
  95 ; CHECK-NEXT:    [[TMP3:%.*]] = shl i16 1, [[TMP2]]
  96 ; CHECK-NEXT:    [[TMP4:%.*]] = and i16 [[TMP3]], 9217
  97 ; CHECK-NEXT:    [[MEMCHR_BITS:%.*]] = icmp ne i16 [[TMP4]], 0
  98 ; CHECK-NEXT:    [[MEMCHR1:%.*]] = select i1 [[MEMCHR_BOUNDS]], i1 [[MEMCHR_BITS]], i1 false
  99 ; CHECK-NEXT:    ret i1 [[MEMCHR1]]
 100 ;
 101
 102   %dst = call i8* @strchr(i8* getelementptr inbounds ([3 x i8], [3 x i8]* @newlines, i64 0, i64 0), i32 %C)
 103   %cmp = icmp ne i8* %dst, null
 104   ret i1 %cmp
 105 }
 106
 107 define i8* @test1(i8* %str, i32 %c) {
 108 ; CHECK-LABEL: @test1(
 109 ; CHECK-NEXT:    [[RET:%.*]] = call i8* @strchr(i8* noundef nonnull dereferenceable(1) [[STR:%.*]], i32 [[C:%.*]])
 110 ; CHECK-NEXT:    ret i8* [[RET]]
 111 ;
 112
 113   %ret = call i8* @strchr(i8* %str, i32 %c)
 114   ret i8* %ret
 115 }
 116
 117 define i8* @test2(i8* %str, i32 %c) null_pointer_is_valid {
 118 ; CHECK-LABEL: @test2(
 119 ; CHECK-NEXT:    [[RET:%.*]] = call i8* @strchr(i8* noundef [[STR:%.*]], i32 [[C:%.*]])
 120 ; CHECK-NEXT:    ret i8* [[RET]]
 121 ;
 122
 123   %ret = call i8* @strchr(i8* %str, i32 %c)
 124   ret i8* %ret
 125 }