test/Transforms/InstCombine/strpbrk-1.ll

   1 ; Test that the strpbrk library call simplifier works correctly.
   2 ;
   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"
   6
   7 @hello = constant [12 x i8] c"hello world\00"
   8 @w = constant [2 x i8] c"w\00"
   9 @null = constant [1 x i8] zeroinitializer
  10
  11 declare i8* @strpbrk(i8*, i8*)
  12
  13 ; Check strpbrk(s, "") -> NULL.
  14
  15 define i8* @test_simplify1(i8* %str) {
  16 ; CHECK-LABEL: @test_simplify1(
  17   %pat = getelementptr [1 x i8], [1 x i8]* @null, i32 0, i32 0
  18
  19   %ret = call i8* @strpbrk(i8* %str, i8* %pat)
  20   ret i8* %ret
  21 ; CHECK-NEXT: ret i8* null
  22 }
  23
  24 ; Check strpbrk("", s) -> NULL.
  25
  26 define i8* @test_simplify2(i8* %pat) {
  27 ; CHECK-LABEL: @test_simplify2(
  28   %str = getelementptr [1 x i8], [1 x i8]* @null, i32 0, i32 0
  29
  30   %ret = call i8* @strpbrk(i8* %str, i8* %pat)
  31   ret i8* %ret
  32 ; CHECK-NEXT: ret i8* null
  33 }
  34
  35 ; Check strpbrk(s1, s2), where s1 and s2 are constants.
  36
  37 define i8* @test_simplify3() {
  38 ; CHECK-LABEL: @test_simplify3(
  39   %str = getelementptr [12 x i8], [12 x i8]* @hello, i32 0, i32 0
  40   %pat = getelementptr [2 x i8], [2 x i8]* @w, i32 0, i32 0
  41
  42   %ret = call i8* @strpbrk(i8* %str, i8* %pat)
  43   ret i8* %ret
  44 ; CHECK-NEXT: ret i8* getelementptr inbounds ([12 x i8], [12 x i8]* @hello, i32 0, i32 6)
  45 }
  46
  47 ; Check strpbrk(s, "a") -> strchr(s, 'a').
  48
  49 define i8* @test_simplify4(i8* %str) {
  50 ; CHECK-LABEL: @test_simplify4(
  51   %pat = getelementptr [2 x i8], [2 x i8]* @w, i32 0, i32 0
  52
  53   %ret = call i8* @strpbrk(i8* %str, i8* %pat)
  54 ; CHECK-NEXT: [[VAR:%[a-z]+]] = call i8* @strchr(i8* %str, i32 119)
  55   ret i8* %ret
  56 ; CHECK-NEXT: ret i8* [[VAR]]
  57 }
  58
  59 ; Check cases that shouldn't be simplified.
  60
  61 define i8* @test_no_simplify1(i8* %str, i8* %pat) {
  62 ; CHECK-LABEL: @test_no_simplify1(
  63
  64   %ret = call i8* @strpbrk(i8* %str, i8* %pat)
  65 ; CHECK-NEXT: %ret = call i8* @strpbrk(i8* %str, i8* %pat)
  66   ret i8* %ret
  67 ; CHECK-NEXT: ret i8* %ret
  68 }