1 //===- llvm/unittest/Support/RegexTest.cpp - Regex tests --===// 2 // 3 // The LLVM Compiler Infrastructure 4 // 5 // This file is distributed under the University of Illinois Open Source 6 // License. See LICENSE.TXT for details. 7 // 8 //===----------------------------------------------------------------------===// 9 10 #include "llvm/Support/Regex.h" 11 #include "llvm/ADT/SmallVector.h" 12 #include "gtest/gtest.h" 13 #include <cstring> 14 15 using namespace llvm; 16 namespace { 17 18 class RegexTest : public ::testing::Test { 19 }; 20 21 TEST_F(RegexTest, Basics) { 22 Regex r1("^[0-9]+$"); 23 EXPECT_TRUE(r1.match("916")); 24 EXPECT_TRUE(r1.match("9")); 25 EXPECT_FALSE(r1.match("9a")); 26 27 SmallVector<StringRef, 1> Matches; 28 Regex r2("[0-9]+"); 29 EXPECT_TRUE(r2.match("aa216b", &Matches)); 30 EXPECT_EQ(1u, Matches.size()); 31 EXPECT_EQ("216", Matches[0].str()); 32 33 Regex r3("[0-9]+([a-f])?:([0-9]+)"); 34 EXPECT_TRUE(r3.match("9a:513b", &Matches)); 35 EXPECT_EQ(3u, Matches.size()); 36 EXPECT_EQ("9a:513", Matches[0].str()); 37 EXPECT_EQ("a", Matches[1].str()); 38 EXPECT_EQ("513", Matches[2].str()); 39 40 EXPECT_TRUE(r3.match("9:513b", &Matches)); 41 EXPECT_EQ(3u, Matches.size()); 42 EXPECT_EQ("9:513", Matches[0].str()); 43 EXPECT_EQ("", Matches[1].str()); 44 EXPECT_EQ("513", Matches[2].str()); 45 46 Regex r4("a[^b]+b"); 47 std::string String="axxb"; 48 String[2] = '\0'; 49 EXPECT_FALSE(r4.match("abb")); 50 EXPECT_TRUE(r4.match(String, &Matches)); 51 EXPECT_EQ(1u, Matches.size()); 52 EXPECT_EQ(String, Matches[0].str()); 53 54 std::string NulPattern="X[0-9]+X([a-f])?:([0-9]+)"; 55 String="YX99a:513b"; 56 NulPattern[7] = '\0'; 57 Regex r5(NulPattern); 58 EXPECT_FALSE(r5.match(String)); 59 EXPECT_FALSE(r5.match("X9")); 60 String[3]='\0'; 61 EXPECT_TRUE(r5.match(String)); 62 } 63 64 TEST_F(RegexTest, Backreferences) { 65 Regex r1("([a-z]+)_\\1"); 66 SmallVector<StringRef, 4> Matches; 67 EXPECT_TRUE(r1.match("abc_abc", &Matches)); 68 EXPECT_EQ(2u, Matches.size()); 69 EXPECT_FALSE(r1.match("abc_ab", &Matches)); 70 71 Regex r2("a([0-9])b\\1c\\1"); 72 EXPECT_TRUE(r2.match("a4b4c4", &Matches)); 73 EXPECT_EQ(2u, Matches.size()); 74 EXPECT_EQ("4", Matches[1].str()); 75 EXPECT_FALSE(r2.match("a2b2c3")); 76 77 Regex r3("a([0-9])([a-z])b\\1\\2"); 78 EXPECT_TRUE(r3.match("a6zb6z", &Matches)); 79 EXPECT_EQ(3u, Matches.size()); 80 EXPECT_EQ("6", Matches[1].str()); 81 EXPECT_EQ("z", Matches[2].str()); 82 EXPECT_FALSE(r3.match("a6zb6y")); 83 EXPECT_FALSE(r3.match("a6zb7z")); 84 } 85 86 TEST_F(RegexTest, Substitution) { 87 std::string Error; 88 89 EXPECT_EQ("aNUMber", Regex("[0-9]+").sub("NUM", "a1234ber")); 90 91 // Standard Escapes 92 EXPECT_EQ("a\\ber", Regex("[0-9]+").sub("\\\\", "a1234ber", &Error)); 93 EXPECT_EQ(Error, ""); 94 EXPECT_EQ("a\nber", Regex("[0-9]+").sub("\\n", "a1234ber", &Error)); 95 EXPECT_EQ(Error, ""); 96 EXPECT_EQ("a\tber", Regex("[0-9]+").sub("\\t", "a1234ber", &Error)); 97 EXPECT_EQ(Error, ""); 98 EXPECT_EQ("ajber", Regex("[0-9]+").sub("\\j", "a1234ber", &Error)); 99 EXPECT_EQ(Error, ""); 100 101 EXPECT_EQ("aber", Regex("[0-9]+").sub("\\", "a1234ber", &Error)); 102 EXPECT_EQ(Error, "replacement string contained trailing backslash"); 103 104 // Backreferences 105 EXPECT_EQ("aa1234bber", Regex("a[0-9]+b").sub("a\\0b", "a1234ber", &Error)); 106 EXPECT_EQ(Error, ""); 107 108 EXPECT_EQ("a1234ber", Regex("a([0-9]+)b").sub("a\\1b", "a1234ber", &Error)); 109 EXPECT_EQ(Error, ""); 110 111 EXPECT_EQ("aber", Regex("a[0-9]+b").sub("a\\100b", "a1234ber", &Error)); 112 EXPECT_EQ(Error, "invalid backreference string '100'"); 113 } 114 115 TEST_F(RegexTest, IsLiteralERE) { 116 EXPECT_TRUE(Regex::isLiteralERE("abc")); 117 EXPECT_FALSE(Regex::isLiteralERE("a(bc)")); 118 EXPECT_FALSE(Regex::isLiteralERE("^abc")); 119 EXPECT_FALSE(Regex::isLiteralERE("abc$")); 120 EXPECT_FALSE(Regex::isLiteralERE("a|bc")); 121 EXPECT_FALSE(Regex::isLiteralERE("abc*")); 122 EXPECT_FALSE(Regex::isLiteralERE("abc+")); 123 EXPECT_FALSE(Regex::isLiteralERE("abc?")); 124 EXPECT_FALSE(Regex::isLiteralERE("abc.")); 125 EXPECT_FALSE(Regex::isLiteralERE("a[bc]")); 126 EXPECT_FALSE(Regex::isLiteralERE("abc\\1")); 127 EXPECT_FALSE(Regex::isLiteralERE("abc{1,2}")); 128 } 129 130 TEST_F(RegexTest, IsValid) { 131 std::string Error; 132 EXPECT_FALSE(Regex("(foo").isValid(Error)); 133 EXPECT_EQ("parentheses not balanced", Error); 134 EXPECT_FALSE(Regex("a[b-").isValid(Error)); 135 EXPECT_EQ("invalid character range", Error); 136 } 137 138 } 139