Chris Lattner | 81cb8ca | 2009-07-08 18:44:05 +0000 | [diff] [blame] | 1 | //===- FileCheck.cpp - Check that File's Contents match what is expected --===// |
| 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 | // FileCheck does a line-by line check of a file that validates whether it |
| 11 | // contains the expected content. This is useful for regression tests etc. |
| 12 | // |
| 13 | // This program exits with an error status of 2 on error, exit status of 0 if |
| 14 | // the file matched the expected contents, and exit status of 1 if it did not |
| 15 | // contain the expected contents. |
| 16 | // |
| 17 | //===----------------------------------------------------------------------===// |
| 18 | |
Michael J. Spencer | 3ff9563 | 2010-12-16 03:29:14 +0000 | [diff] [blame] | 19 | #include "llvm/ADT/OwningPtr.h" |
Chris Lattner | 81cb8ca | 2009-07-08 18:44:05 +0000 | [diff] [blame] | 20 | #include "llvm/Support/CommandLine.h" |
| 21 | #include "llvm/Support/MemoryBuffer.h" |
| 22 | #include "llvm/Support/PrettyStackTrace.h" |
Chris Lattner | 5287008 | 2009-09-24 21:47:32 +0000 | [diff] [blame] | 23 | #include "llvm/Support/Regex.h" |
Chris Lattner | 81cb8ca | 2009-07-08 18:44:05 +0000 | [diff] [blame] | 24 | #include "llvm/Support/SourceMgr.h" |
| 25 | #include "llvm/Support/raw_ostream.h" |
Michael J. Spencer | 1f6efa3 | 2010-11-29 18:16:10 +0000 | [diff] [blame] | 26 | #include "llvm/Support/Signals.h" |
Michael J. Spencer | 333fb04 | 2010-12-09 17:36:48 +0000 | [diff] [blame] | 27 | #include "llvm/Support/system_error.h" |
Daniel Dunbar | fafe93c | 2009-11-22 22:08:06 +0000 | [diff] [blame] | 28 | #include "llvm/ADT/SmallString.h" |
Chris Lattner | eec9695 | 2009-09-27 07:56:52 +0000 | [diff] [blame] | 29 | #include "llvm/ADT/StringMap.h" |
| 30 | #include <algorithm> |
Chris Lattner | 81cb8ca | 2009-07-08 18:44:05 +0000 | [diff] [blame] | 31 | using namespace llvm; |
| 32 | |
| 33 | static cl::opt<std::string> |
| 34 | CheckFilename(cl::Positional, cl::desc("<check-file>"), cl::Required); |
| 35 | |
| 36 | static cl::opt<std::string> |
| 37 | InputFilename("input-file", cl::desc("File to check (defaults to stdin)"), |
| 38 | cl::init("-"), cl::value_desc("filename")); |
| 39 | |
| 40 | static cl::opt<std::string> |
| 41 | CheckPrefix("check-prefix", cl::init("CHECK"), |
| 42 | cl::desc("Prefix to use from check file (defaults to 'CHECK')")); |
| 43 | |
Chris Lattner | 88a7e9e | 2009-07-11 18:58:15 +0000 | [diff] [blame] | 44 | static cl::opt<bool> |
| 45 | NoCanonicalizeWhiteSpace("strict-whitespace", |
| 46 | cl::desc("Do not treat all horizontal whitespace as equivalent")); |
| 47 | |
Chris Lattner | a29703e | 2009-09-24 20:39:13 +0000 | [diff] [blame] | 48 | //===----------------------------------------------------------------------===// |
| 49 | // Pattern Handling Code. |
| 50 | //===----------------------------------------------------------------------===// |
| 51 | |
Chris Lattner | 9fc6678 | 2009-09-24 20:25:55 +0000 | [diff] [blame] | 52 | class Pattern { |
Chris Lattner | 94638f0 | 2009-09-25 17:29:36 +0000 | [diff] [blame] | 53 | SMLoc PatternLoc; |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 54 | |
Jakob Stoklund Olesen | 824c10e | 2010-10-15 17:47:12 +0000 | [diff] [blame] | 55 | /// MatchEOF - When set, this pattern only matches the end of file. This is |
| 56 | /// used for trailing CHECK-NOTs. |
| 57 | bool MatchEOF; |
| 58 | |
Chris Lattner | 5d6a05f | 2009-09-25 17:23:43 +0000 | [diff] [blame] | 59 | /// FixedStr - If non-empty, this pattern is a fixed string match with the |
| 60 | /// specified fixed string. |
Chris Lattner | 2702e6a | 2009-09-25 17:09:12 +0000 | [diff] [blame] | 61 | StringRef FixedStr; |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 62 | |
Chris Lattner | 5d6a05f | 2009-09-25 17:23:43 +0000 | [diff] [blame] | 63 | /// RegEx - If non-empty, this is a regex pattern. |
| 64 | std::string RegExStr; |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 65 | |
Chris Lattner | eec9695 | 2009-09-27 07:56:52 +0000 | [diff] [blame] | 66 | /// VariableUses - Entries in this vector map to uses of a variable in the |
| 67 | /// pattern, e.g. "foo[[bar]]baz". In this case, the RegExStr will contain |
| 68 | /// "foobaz" and we'll get an entry in this vector that tells us to insert the |
| 69 | /// value of bar at offset 3. |
| 70 | std::vector<std::pair<StringRef, unsigned> > VariableUses; |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 71 | |
Chris Lattner | eec9695 | 2009-09-27 07:56:52 +0000 | [diff] [blame] | 72 | /// VariableDefs - Entries in this vector map to definitions of a variable in |
| 73 | /// the pattern, e.g. "foo[[bar:.*]]baz". In this case, the RegExStr will |
| 74 | /// contain "foo(.*)baz" and VariableDefs will contain the pair "bar",1. The |
| 75 | /// index indicates what parenthesized value captures the variable value. |
| 76 | std::vector<std::pair<StringRef, unsigned> > VariableDefs; |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 77 | |
Chris Lattner | 9fc6678 | 2009-09-24 20:25:55 +0000 | [diff] [blame] | 78 | public: |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 79 | |
Jakob Stoklund Olesen | 824c10e | 2010-10-15 17:47:12 +0000 | [diff] [blame] | 80 | Pattern(bool matchEOF = false) : MatchEOF(matchEOF) { } |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 81 | |
Chris Lattner | a29703e | 2009-09-24 20:39:13 +0000 | [diff] [blame] | 82 | bool ParsePattern(StringRef PatternStr, SourceMgr &SM); |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 83 | |
Chris Lattner | 9fc6678 | 2009-09-24 20:25:55 +0000 | [diff] [blame] | 84 | /// Match - Match the pattern string against the input buffer Buffer. This |
| 85 | /// returns the position that is matched or npos if there is no match. If |
| 86 | /// there is a match, the size of the matched string is returned in MatchLen. |
Chris Lattner | eec9695 | 2009-09-27 07:56:52 +0000 | [diff] [blame] | 87 | /// |
| 88 | /// The VariableTable StringMap provides the current values of filecheck |
| 89 | /// variables and is updated if this match defines new values. |
| 90 | size_t Match(StringRef Buffer, size_t &MatchLen, |
| 91 | StringMap<StringRef> &VariableTable) const; |
Daniel Dunbar | fafe93c | 2009-11-22 22:08:06 +0000 | [diff] [blame] | 92 | |
| 93 | /// PrintFailureInfo - Print additional information about a failure to match |
| 94 | /// involving this pattern. |
| 95 | void PrintFailureInfo(const SourceMgr &SM, StringRef Buffer, |
| 96 | const StringMap<StringRef> &VariableTable) const; |
| 97 | |
Chris Lattner | 5d6a05f | 2009-09-25 17:23:43 +0000 | [diff] [blame] | 98 | private: |
Chris Lattner | eec9695 | 2009-09-27 07:56:52 +0000 | [diff] [blame] | 99 | static void AddFixedStringToRegEx(StringRef FixedStr, std::string &TheStr); |
| 100 | bool AddRegExToRegEx(StringRef RegExStr, unsigned &CurParen, SourceMgr &SM); |
Daniel Dunbar | ead2dac | 2009-11-22 22:59:26 +0000 | [diff] [blame] | 101 | |
| 102 | /// ComputeMatchDistance - Compute an arbitrary estimate for the quality of |
| 103 | /// matching this pattern at the start of \arg Buffer; a distance of zero |
| 104 | /// should correspond to a perfect match. |
| 105 | unsigned ComputeMatchDistance(StringRef Buffer, |
| 106 | const StringMap<StringRef> &VariableTable) const; |
Chris Lattner | 9fc6678 | 2009-09-24 20:25:55 +0000 | [diff] [blame] | 107 | }; |
| 108 | |
Chris Lattner | eec9695 | 2009-09-27 07:56:52 +0000 | [diff] [blame] | 109 | |
Chris Lattner | a29703e | 2009-09-24 20:39:13 +0000 | [diff] [blame] | 110 | bool Pattern::ParsePattern(StringRef PatternStr, SourceMgr &SM) { |
Chris Lattner | 94638f0 | 2009-09-25 17:29:36 +0000 | [diff] [blame] | 111 | PatternLoc = SMLoc::getFromPointer(PatternStr.data()); |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 112 | |
Chris Lattner | a29703e | 2009-09-24 20:39:13 +0000 | [diff] [blame] | 113 | // Ignore trailing whitespace. |
| 114 | while (!PatternStr.empty() && |
| 115 | (PatternStr.back() == ' ' || PatternStr.back() == '\t')) |
| 116 | PatternStr = PatternStr.substr(0, PatternStr.size()-1); |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 117 | |
Chris Lattner | a29703e | 2009-09-24 20:39:13 +0000 | [diff] [blame] | 118 | // Check that there is something on the line. |
| 119 | if (PatternStr.empty()) { |
Chris Lattner | 94638f0 | 2009-09-25 17:29:36 +0000 | [diff] [blame] | 120 | SM.PrintMessage(PatternLoc, "found empty check string with prefix '" + |
| 121 | CheckPrefix+":'", "error"); |
Chris Lattner | a29703e | 2009-09-24 20:39:13 +0000 | [diff] [blame] | 122 | return true; |
| 123 | } |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 124 | |
Chris Lattner | 2702e6a | 2009-09-25 17:09:12 +0000 | [diff] [blame] | 125 | // Check to see if this is a fixed string, or if it has regex pieces. |
Chris Lattner | eec9695 | 2009-09-27 07:56:52 +0000 | [diff] [blame] | 126 | if (PatternStr.size() < 2 || |
| 127 | (PatternStr.find("{{") == StringRef::npos && |
| 128 | PatternStr.find("[[") == StringRef::npos)) { |
Chris Lattner | 2702e6a | 2009-09-25 17:09:12 +0000 | [diff] [blame] | 129 | FixedStr = PatternStr; |
| 130 | return false; |
| 131 | } |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 132 | |
Chris Lattner | eec9695 | 2009-09-27 07:56:52 +0000 | [diff] [blame] | 133 | // Paren value #0 is for the fully matched string. Any new parenthesized |
| 134 | // values add from their. |
| 135 | unsigned CurParen = 1; |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 136 | |
Chris Lattner | 5d6a05f | 2009-09-25 17:23:43 +0000 | [diff] [blame] | 137 | // Otherwise, there is at least one regex piece. Build up the regex pattern |
| 138 | // by escaping scary characters in fixed strings, building up one big regex. |
Chris Lattner | 5287008 | 2009-09-24 21:47:32 +0000 | [diff] [blame] | 139 | while (!PatternStr.empty()) { |
Chris Lattner | eec9695 | 2009-09-27 07:56:52 +0000 | [diff] [blame] | 140 | // RegEx matches. |
| 141 | if (PatternStr.size() >= 2 && |
| 142 | PatternStr[0] == '{' && PatternStr[1] == '{') { |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 143 | |
Chris Lattner | eec9695 | 2009-09-27 07:56:52 +0000 | [diff] [blame] | 144 | // Otherwise, this is the start of a regex match. Scan for the }}. |
| 145 | size_t End = PatternStr.find("}}"); |
| 146 | if (End == StringRef::npos) { |
| 147 | SM.PrintMessage(SMLoc::getFromPointer(PatternStr.data()), |
| 148 | "found start of regex string with no end '}}'", "error"); |
| 149 | return true; |
| 150 | } |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 151 | |
Chris Lattner | eec9695 | 2009-09-27 07:56:52 +0000 | [diff] [blame] | 152 | if (AddRegExToRegEx(PatternStr.substr(2, End-2), CurParen, SM)) |
| 153 | return true; |
| 154 | PatternStr = PatternStr.substr(End+2); |
Chris Lattner | 5287008 | 2009-09-24 21:47:32 +0000 | [diff] [blame] | 155 | continue; |
| 156 | } |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 157 | |
Chris Lattner | eec9695 | 2009-09-27 07:56:52 +0000 | [diff] [blame] | 158 | // Named RegEx matches. These are of two forms: [[foo:.*]] which matches .* |
| 159 | // (or some other regex) and assigns it to the FileCheck variable 'foo'. The |
| 160 | // second form is [[foo]] which is a reference to foo. The variable name |
Daniel Dunbar | 964ac01 | 2009-11-22 22:07:50 +0000 | [diff] [blame] | 161 | // itself must be of the form "[a-zA-Z_][0-9a-zA-Z_]*", otherwise we reject |
Chris Lattner | eec9695 | 2009-09-27 07:56:52 +0000 | [diff] [blame] | 162 | // it. This is to catch some common errors. |
| 163 | if (PatternStr.size() >= 2 && |
| 164 | PatternStr[0] == '[' && PatternStr[1] == '[') { |
| 165 | // Verify that it is terminated properly. |
| 166 | size_t End = PatternStr.find("]]"); |
| 167 | if (End == StringRef::npos) { |
| 168 | SM.PrintMessage(SMLoc::getFromPointer(PatternStr.data()), |
| 169 | "invalid named regex reference, no ]] found", "error"); |
| 170 | return true; |
| 171 | } |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 172 | |
Chris Lattner | eec9695 | 2009-09-27 07:56:52 +0000 | [diff] [blame] | 173 | StringRef MatchStr = PatternStr.substr(2, End-2); |
| 174 | PatternStr = PatternStr.substr(End+2); |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 175 | |
Chris Lattner | eec9695 | 2009-09-27 07:56:52 +0000 | [diff] [blame] | 176 | // Get the regex name (e.g. "foo"). |
| 177 | size_t NameEnd = MatchStr.find(':'); |
| 178 | StringRef Name = MatchStr.substr(0, NameEnd); |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 179 | |
Chris Lattner | eec9695 | 2009-09-27 07:56:52 +0000 | [diff] [blame] | 180 | if (Name.empty()) { |
| 181 | SM.PrintMessage(SMLoc::getFromPointer(Name.data()), |
| 182 | "invalid name in named regex: empty name", "error"); |
| 183 | return true; |
| 184 | } |
| 185 | |
| 186 | // Verify that the name is well formed. |
| 187 | for (unsigned i = 0, e = Name.size(); i != e; ++i) |
Daniel Dunbar | 964ac01 | 2009-11-22 22:07:50 +0000 | [diff] [blame] | 188 | if (Name[i] != '_' && |
| 189 | (Name[i] < 'a' || Name[i] > 'z') && |
Chris Lattner | eec9695 | 2009-09-27 07:56:52 +0000 | [diff] [blame] | 190 | (Name[i] < 'A' || Name[i] > 'Z') && |
| 191 | (Name[i] < '0' || Name[i] > '9')) { |
| 192 | SM.PrintMessage(SMLoc::getFromPointer(Name.data()+i), |
| 193 | "invalid name in named regex", "error"); |
| 194 | return true; |
| 195 | } |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 196 | |
Chris Lattner | eec9695 | 2009-09-27 07:56:52 +0000 | [diff] [blame] | 197 | // Name can't start with a digit. |
| 198 | if (isdigit(Name[0])) { |
| 199 | SM.PrintMessage(SMLoc::getFromPointer(Name.data()), |
| 200 | "invalid name in named regex", "error"); |
| 201 | return true; |
| 202 | } |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 203 | |
Chris Lattner | eec9695 | 2009-09-27 07:56:52 +0000 | [diff] [blame] | 204 | // Handle [[foo]]. |
| 205 | if (NameEnd == StringRef::npos) { |
| 206 | VariableUses.push_back(std::make_pair(Name, RegExStr.size())); |
| 207 | continue; |
| 208 | } |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 209 | |
Chris Lattner | eec9695 | 2009-09-27 07:56:52 +0000 | [diff] [blame] | 210 | // Handle [[foo:.*]]. |
| 211 | VariableDefs.push_back(std::make_pair(Name, CurParen)); |
| 212 | RegExStr += '('; |
| 213 | ++CurParen; |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 214 | |
Chris Lattner | eec9695 | 2009-09-27 07:56:52 +0000 | [diff] [blame] | 215 | if (AddRegExToRegEx(MatchStr.substr(NameEnd+1), CurParen, SM)) |
| 216 | return true; |
| 217 | |
| 218 | RegExStr += ')'; |
Chris Lattner | 5287008 | 2009-09-24 21:47:32 +0000 | [diff] [blame] | 219 | } |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 220 | |
Chris Lattner | eec9695 | 2009-09-27 07:56:52 +0000 | [diff] [blame] | 221 | // Handle fixed string matches. |
| 222 | // Find the end, which is the start of the next regex. |
| 223 | size_t FixedMatchEnd = PatternStr.find("{{"); |
| 224 | FixedMatchEnd = std::min(FixedMatchEnd, PatternStr.find("[[")); |
| 225 | AddFixedStringToRegEx(PatternStr.substr(0, FixedMatchEnd), RegExStr); |
| 226 | PatternStr = PatternStr.substr(FixedMatchEnd); |
| 227 | continue; |
Chris Lattner | 5287008 | 2009-09-24 21:47:32 +0000 | [diff] [blame] | 228 | } |
Chris Lattner | adea46e | 2009-09-24 20:45:07 +0000 | [diff] [blame] | 229 | |
Chris Lattner | a29703e | 2009-09-24 20:39:13 +0000 | [diff] [blame] | 230 | return false; |
| 231 | } |
| 232 | |
Chris Lattner | eec9695 | 2009-09-27 07:56:52 +0000 | [diff] [blame] | 233 | void Pattern::AddFixedStringToRegEx(StringRef FixedStr, std::string &TheStr) { |
Chris Lattner | 5d6a05f | 2009-09-25 17:23:43 +0000 | [diff] [blame] | 234 | // Add the characters from FixedStr to the regex, escaping as needed. This |
| 235 | // avoids "leaning toothpicks" in common patterns. |
| 236 | for (unsigned i = 0, e = FixedStr.size(); i != e; ++i) { |
| 237 | switch (FixedStr[i]) { |
| 238 | // These are the special characters matched in "p_ere_exp". |
| 239 | case '(': |
| 240 | case ')': |
| 241 | case '^': |
| 242 | case '$': |
| 243 | case '|': |
| 244 | case '*': |
| 245 | case '+': |
| 246 | case '?': |
| 247 | case '.': |
| 248 | case '[': |
| 249 | case '\\': |
| 250 | case '{': |
Chris Lattner | eec9695 | 2009-09-27 07:56:52 +0000 | [diff] [blame] | 251 | TheStr += '\\'; |
Chris Lattner | 5d6a05f | 2009-09-25 17:23:43 +0000 | [diff] [blame] | 252 | // FALL THROUGH. |
| 253 | default: |
Chris Lattner | eec9695 | 2009-09-27 07:56:52 +0000 | [diff] [blame] | 254 | TheStr += FixedStr[i]; |
Chris Lattner | 5d6a05f | 2009-09-25 17:23:43 +0000 | [diff] [blame] | 255 | break; |
| 256 | } |
| 257 | } |
| 258 | } |
| 259 | |
Chris Lattner | eec9695 | 2009-09-27 07:56:52 +0000 | [diff] [blame] | 260 | bool Pattern::AddRegExToRegEx(StringRef RegexStr, unsigned &CurParen, |
| 261 | SourceMgr &SM) { |
| 262 | Regex R(RegexStr); |
| 263 | std::string Error; |
| 264 | if (!R.isValid(Error)) { |
| 265 | SM.PrintMessage(SMLoc::getFromPointer(RegexStr.data()), |
| 266 | "invalid regex: " + Error, "error"); |
| 267 | return true; |
| 268 | } |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 269 | |
Chris Lattner | eec9695 | 2009-09-27 07:56:52 +0000 | [diff] [blame] | 270 | RegExStr += RegexStr.str(); |
| 271 | CurParen += R.getNumMatches(); |
| 272 | return false; |
| 273 | } |
Chris Lattner | 5d6a05f | 2009-09-25 17:23:43 +0000 | [diff] [blame] | 274 | |
Chris Lattner | 5287008 | 2009-09-24 21:47:32 +0000 | [diff] [blame] | 275 | /// Match - Match the pattern string against the input buffer Buffer. This |
| 276 | /// returns the position that is matched or npos if there is no match. If |
| 277 | /// there is a match, the size of the matched string is returned in MatchLen. |
Chris Lattner | eec9695 | 2009-09-27 07:56:52 +0000 | [diff] [blame] | 278 | size_t Pattern::Match(StringRef Buffer, size_t &MatchLen, |
| 279 | StringMap<StringRef> &VariableTable) const { |
Jakob Stoklund Olesen | 824c10e | 2010-10-15 17:47:12 +0000 | [diff] [blame] | 280 | // If this is the EOF pattern, match it immediately. |
| 281 | if (MatchEOF) { |
| 282 | MatchLen = 0; |
| 283 | return Buffer.size(); |
| 284 | } |
| 285 | |
Chris Lattner | 2702e6a | 2009-09-25 17:09:12 +0000 | [diff] [blame] | 286 | // If this is a fixed string pattern, just match it now. |
| 287 | if (!FixedStr.empty()) { |
| 288 | MatchLen = FixedStr.size(); |
| 289 | return Buffer.find(FixedStr); |
| 290 | } |
Chris Lattner | eec9695 | 2009-09-27 07:56:52 +0000 | [diff] [blame] | 291 | |
Chris Lattner | 5d6a05f | 2009-09-25 17:23:43 +0000 | [diff] [blame] | 292 | // Regex match. |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 293 | |
Chris Lattner | eec9695 | 2009-09-27 07:56:52 +0000 | [diff] [blame] | 294 | // If there are variable uses, we need to create a temporary string with the |
| 295 | // actual value. |
| 296 | StringRef RegExToMatch = RegExStr; |
| 297 | std::string TmpStr; |
| 298 | if (!VariableUses.empty()) { |
| 299 | TmpStr = RegExStr; |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 300 | |
Chris Lattner | eec9695 | 2009-09-27 07:56:52 +0000 | [diff] [blame] | 301 | unsigned InsertOffset = 0; |
| 302 | for (unsigned i = 0, e = VariableUses.size(); i != e; ++i) { |
Daniel Dunbar | fafe93c | 2009-11-22 22:08:06 +0000 | [diff] [blame] | 303 | StringMap<StringRef>::iterator it = |
| 304 | VariableTable.find(VariableUses[i].first); |
| 305 | // If the variable is undefined, return an error. |
| 306 | if (it == VariableTable.end()) |
| 307 | return StringRef::npos; |
| 308 | |
Chris Lattner | eec9695 | 2009-09-27 07:56:52 +0000 | [diff] [blame] | 309 | // Look up the value and escape it so that we can plop it into the regex. |
| 310 | std::string Value; |
Daniel Dunbar | fafe93c | 2009-11-22 22:08:06 +0000 | [diff] [blame] | 311 | AddFixedStringToRegEx(it->second, Value); |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 312 | |
Chris Lattner | eec9695 | 2009-09-27 07:56:52 +0000 | [diff] [blame] | 313 | // Plop it into the regex at the adjusted offset. |
| 314 | TmpStr.insert(TmpStr.begin()+VariableUses[i].second+InsertOffset, |
| 315 | Value.begin(), Value.end()); |
| 316 | InsertOffset += Value.size(); |
| 317 | } |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 318 | |
Chris Lattner | eec9695 | 2009-09-27 07:56:52 +0000 | [diff] [blame] | 319 | // Match the newly constructed regex. |
| 320 | RegExToMatch = TmpStr; |
| 321 | } |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 322 | |
| 323 | |
Chris Lattner | 5d6a05f | 2009-09-25 17:23:43 +0000 | [diff] [blame] | 324 | SmallVector<StringRef, 4> MatchInfo; |
Chris Lattner | eec9695 | 2009-09-27 07:56:52 +0000 | [diff] [blame] | 325 | if (!Regex(RegExToMatch, Regex::Newline).match(Buffer, &MatchInfo)) |
Chris Lattner | 5d6a05f | 2009-09-25 17:23:43 +0000 | [diff] [blame] | 326 | return StringRef::npos; |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 327 | |
Chris Lattner | 5d6a05f | 2009-09-25 17:23:43 +0000 | [diff] [blame] | 328 | // Successful regex match. |
| 329 | assert(!MatchInfo.empty() && "Didn't get any match"); |
| 330 | StringRef FullMatch = MatchInfo[0]; |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 331 | |
Chris Lattner | eec9695 | 2009-09-27 07:56:52 +0000 | [diff] [blame] | 332 | // If this defines any variables, remember their values. |
| 333 | for (unsigned i = 0, e = VariableDefs.size(); i != e; ++i) { |
| 334 | assert(VariableDefs[i].second < MatchInfo.size() && |
| 335 | "Internal paren error"); |
| 336 | VariableTable[VariableDefs[i].first] = MatchInfo[VariableDefs[i].second]; |
Chris Lattner | 94638f0 | 2009-09-25 17:29:36 +0000 | [diff] [blame] | 337 | } |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 338 | |
Chris Lattner | 5d6a05f | 2009-09-25 17:23:43 +0000 | [diff] [blame] | 339 | MatchLen = FullMatch.size(); |
| 340 | return FullMatch.data()-Buffer.data(); |
Chris Lattner | 5287008 | 2009-09-24 21:47:32 +0000 | [diff] [blame] | 341 | } |
| 342 | |
Daniel Dunbar | ead2dac | 2009-11-22 22:59:26 +0000 | [diff] [blame] | 343 | unsigned Pattern::ComputeMatchDistance(StringRef Buffer, |
| 344 | const StringMap<StringRef> &VariableTable) const { |
| 345 | // Just compute the number of matching characters. For regular expressions, we |
| 346 | // just compare against the regex itself and hope for the best. |
| 347 | // |
| 348 | // FIXME: One easy improvement here is have the regex lib generate a single |
| 349 | // example regular expression which matches, and use that as the example |
| 350 | // string. |
| 351 | StringRef ExampleString(FixedStr); |
| 352 | if (ExampleString.empty()) |
| 353 | ExampleString = RegExStr; |
| 354 | |
Daniel Dunbar | 0806f9f | 2010-01-30 00:24:06 +0000 | [diff] [blame] | 355 | // Only compare up to the first line in the buffer, or the string size. |
| 356 | StringRef BufferPrefix = Buffer.substr(0, ExampleString.size()); |
| 357 | BufferPrefix = BufferPrefix.split('\n').first; |
| 358 | return BufferPrefix.edit_distance(ExampleString); |
Daniel Dunbar | ead2dac | 2009-11-22 22:59:26 +0000 | [diff] [blame] | 359 | } |
| 360 | |
Daniel Dunbar | fafe93c | 2009-11-22 22:08:06 +0000 | [diff] [blame] | 361 | void Pattern::PrintFailureInfo(const SourceMgr &SM, StringRef Buffer, |
| 362 | const StringMap<StringRef> &VariableTable) const{ |
Daniel Dunbar | fafe93c | 2009-11-22 22:08:06 +0000 | [diff] [blame] | 363 | // If this was a regular expression using variables, print the current |
| 364 | // variable values. |
| 365 | if (!VariableUses.empty()) { |
| 366 | for (unsigned i = 0, e = VariableUses.size(); i != e; ++i) { |
| 367 | StringRef Var = VariableUses[i].first; |
| 368 | StringMap<StringRef>::const_iterator it = VariableTable.find(Var); |
| 369 | SmallString<256> Msg; |
| 370 | raw_svector_ostream OS(Msg); |
| 371 | |
| 372 | // Check for undefined variable references. |
| 373 | if (it == VariableTable.end()) { |
| 374 | OS << "uses undefined variable \""; |
| 375 | OS.write_escaped(Var) << "\"";; |
| 376 | } else { |
| 377 | OS << "with variable \""; |
| 378 | OS.write_escaped(Var) << "\" equal to \""; |
| 379 | OS.write_escaped(it->second) << "\""; |
| 380 | } |
| 381 | |
| 382 | SM.PrintMessage(SMLoc::getFromPointer(Buffer.data()), OS.str(), "note", |
| 383 | /*ShowLine=*/false); |
| 384 | } |
| 385 | } |
Daniel Dunbar | ead2dac | 2009-11-22 22:59:26 +0000 | [diff] [blame] | 386 | |
| 387 | // Attempt to find the closest/best fuzzy match. Usually an error happens |
| 388 | // because some string in the output didn't exactly match. In these cases, we |
| 389 | // would like to show the user a best guess at what "should have" matched, to |
| 390 | // save them having to actually check the input manually. |
| 391 | size_t NumLinesForward = 0; |
| 392 | size_t Best = StringRef::npos; |
| 393 | double BestQuality = 0; |
| 394 | |
| 395 | // Use an arbitrary 4k limit on how far we will search. |
Dan Gohman | e3a1e50 | 2010-01-29 21:57:46 +0000 | [diff] [blame] | 396 | for (size_t i = 0, e = std::min(size_t(4096), Buffer.size()); i != e; ++i) { |
Daniel Dunbar | ead2dac | 2009-11-22 22:59:26 +0000 | [diff] [blame] | 397 | if (Buffer[i] == '\n') |
| 398 | ++NumLinesForward; |
| 399 | |
Dan Gohman | d8a5541 | 2010-01-29 21:55:16 +0000 | [diff] [blame] | 400 | // Patterns have leading whitespace stripped, so skip whitespace when |
| 401 | // looking for something which looks like a pattern. |
| 402 | if (Buffer[i] == ' ' || Buffer[i] == '\t') |
| 403 | continue; |
| 404 | |
Daniel Dunbar | ead2dac | 2009-11-22 22:59:26 +0000 | [diff] [blame] | 405 | // Compute the "quality" of this match as an arbitrary combination of the |
| 406 | // match distance and the number of lines skipped to get to this match. |
| 407 | unsigned Distance = ComputeMatchDistance(Buffer.substr(i), VariableTable); |
| 408 | double Quality = Distance + (NumLinesForward / 100.); |
| 409 | |
| 410 | if (Quality < BestQuality || Best == StringRef::npos) { |
| 411 | Best = i; |
| 412 | BestQuality = Quality; |
| 413 | } |
| 414 | } |
| 415 | |
Daniel Dunbar | 7a68e0d | 2010-03-19 18:07:43 +0000 | [diff] [blame] | 416 | // Print the "possible intended match here" line if we found something |
| 417 | // reasonable and not equal to what we showed in the "scanning from here" |
| 418 | // line. |
| 419 | if (Best && Best != StringRef::npos && BestQuality < 50) { |
| 420 | SM.PrintMessage(SMLoc::getFromPointer(Buffer.data() + Best), |
| 421 | "possible intended match here", "note"); |
Daniel Dunbar | ead2dac | 2009-11-22 22:59:26 +0000 | [diff] [blame] | 422 | |
| 423 | // FIXME: If we wanted to be really friendly we would show why the match |
| 424 | // failed, as it can be hard to spot simple one character differences. |
| 425 | } |
Daniel Dunbar | fafe93c | 2009-11-22 22:08:06 +0000 | [diff] [blame] | 426 | } |
Chris Lattner | a29703e | 2009-09-24 20:39:13 +0000 | [diff] [blame] | 427 | |
| 428 | //===----------------------------------------------------------------------===// |
| 429 | // Check Strings. |
| 430 | //===----------------------------------------------------------------------===// |
Chris Lattner | 9fc6678 | 2009-09-24 20:25:55 +0000 | [diff] [blame] | 431 | |
| 432 | /// CheckString - This is a check that we found in the input file. |
| 433 | struct CheckString { |
| 434 | /// Pat - The pattern to match. |
| 435 | Pattern Pat; |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 436 | |
Chris Lattner | 207e1bc | 2009-08-15 17:41:04 +0000 | [diff] [blame] | 437 | /// Loc - The location in the match file that the check string was specified. |
| 438 | SMLoc Loc; |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 439 | |
Chris Lattner | 5dafafd | 2009-08-15 18:32:21 +0000 | [diff] [blame] | 440 | /// IsCheckNext - This is true if this is a CHECK-NEXT: directive (as opposed |
| 441 | /// to a CHECK: directive. |
| 442 | bool IsCheckNext; |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 443 | |
Chris Lattner | f15380b | 2009-09-20 22:35:26 +0000 | [diff] [blame] | 444 | /// NotStrings - These are all of the strings that are disallowed from |
| 445 | /// occurring between this match string and the previous one (or start of |
| 446 | /// file). |
Chris Lattner | a29703e | 2009-09-24 20:39:13 +0000 | [diff] [blame] | 447 | std::vector<std::pair<SMLoc, Pattern> > NotStrings; |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 448 | |
Chris Lattner | 9fc6678 | 2009-09-24 20:25:55 +0000 | [diff] [blame] | 449 | CheckString(const Pattern &P, SMLoc L, bool isCheckNext) |
| 450 | : Pat(P), Loc(L), IsCheckNext(isCheckNext) {} |
Chris Lattner | 207e1bc | 2009-08-15 17:41:04 +0000 | [diff] [blame] | 451 | }; |
| 452 | |
Chris Lattner | adea46e | 2009-09-24 20:45:07 +0000 | [diff] [blame] | 453 | /// CanonicalizeInputFile - Remove duplicate horizontal space from the specified |
| 454 | /// memory buffer, free it, and return a new one. |
| 455 | static MemoryBuffer *CanonicalizeInputFile(MemoryBuffer *MB) { |
Chris Lattner | 4c842dd | 2010-04-05 22:42:30 +0000 | [diff] [blame] | 456 | SmallString<128> NewFile; |
Chris Lattner | adea46e | 2009-09-24 20:45:07 +0000 | [diff] [blame] | 457 | NewFile.reserve(MB->getBufferSize()); |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 458 | |
Chris Lattner | adea46e | 2009-09-24 20:45:07 +0000 | [diff] [blame] | 459 | for (const char *Ptr = MB->getBufferStart(), *End = MB->getBufferEnd(); |
| 460 | Ptr != End; ++Ptr) { |
NAKAMURA Takumi | 9f6e03f | 2010-11-14 03:28:22 +0000 | [diff] [blame] | 461 | // Eliminate trailing dosish \r. |
| 462 | if (Ptr <= End - 2 && Ptr[0] == '\r' && Ptr[1] == '\n') { |
| 463 | continue; |
| 464 | } |
| 465 | |
Chris Lattner | adea46e | 2009-09-24 20:45:07 +0000 | [diff] [blame] | 466 | // If C is not a horizontal whitespace, skip it. |
| 467 | if (*Ptr != ' ' && *Ptr != '\t') { |
| 468 | NewFile.push_back(*Ptr); |
| 469 | continue; |
| 470 | } |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 471 | |
Chris Lattner | adea46e | 2009-09-24 20:45:07 +0000 | [diff] [blame] | 472 | // Otherwise, add one space and advance over neighboring space. |
| 473 | NewFile.push_back(' '); |
| 474 | while (Ptr+1 != End && |
| 475 | (Ptr[1] == ' ' || Ptr[1] == '\t')) |
| 476 | ++Ptr; |
| 477 | } |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 478 | |
Chris Lattner | adea46e | 2009-09-24 20:45:07 +0000 | [diff] [blame] | 479 | // Free the old buffer and return a new one. |
| 480 | MemoryBuffer *MB2 = |
Chris Lattner | 4c842dd | 2010-04-05 22:42:30 +0000 | [diff] [blame] | 481 | MemoryBuffer::getMemBufferCopy(NewFile.str(), MB->getBufferIdentifier()); |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 482 | |
Chris Lattner | adea46e | 2009-09-24 20:45:07 +0000 | [diff] [blame] | 483 | delete MB; |
| 484 | return MB2; |
| 485 | } |
| 486 | |
Chris Lattner | 81cb8ca | 2009-07-08 18:44:05 +0000 | [diff] [blame] | 487 | |
Chris Lattner | 81cb8ca | 2009-07-08 18:44:05 +0000 | [diff] [blame] | 488 | /// ReadCheckFile - Read the check file, which specifies the sequence of |
| 489 | /// expected strings. The strings are added to the CheckStrings vector. |
| 490 | static bool ReadCheckFile(SourceMgr &SM, |
Chris Lattner | 207e1bc | 2009-08-15 17:41:04 +0000 | [diff] [blame] | 491 | std::vector<CheckString> &CheckStrings) { |
Chris Lattner | 81cb8ca | 2009-07-08 18:44:05 +0000 | [diff] [blame] | 492 | // Open the check file, and tell SourceMgr about it. |
Michael J. Spencer | 3ff9563 | 2010-12-16 03:29:14 +0000 | [diff] [blame] | 493 | OwningPtr<MemoryBuffer> File; |
| 494 | if (error_code ec = |
| 495 | MemoryBuffer::getFileOrSTDIN(CheckFilename.c_str(), File)) { |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 496 | errs() << "Could not open check file '" << CheckFilename << "': " |
Michael J. Spencer | 333fb04 | 2010-12-09 17:36:48 +0000 | [diff] [blame] | 497 | << ec.message() << '\n'; |
Chris Lattner | 81cb8ca | 2009-07-08 18:44:05 +0000 | [diff] [blame] | 498 | return true; |
| 499 | } |
Michael J. Spencer | 3ff9563 | 2010-12-16 03:29:14 +0000 | [diff] [blame] | 500 | MemoryBuffer *F = File.take(); |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 501 | |
Chris Lattner | adea46e | 2009-09-24 20:45:07 +0000 | [diff] [blame] | 502 | // If we want to canonicalize whitespace, strip excess whitespace from the |
| 503 | // buffer containing the CHECK lines. |
| 504 | if (!NoCanonicalizeWhiteSpace) |
| 505 | F = CanonicalizeInputFile(F); |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 506 | |
Chris Lattner | 81cb8ca | 2009-07-08 18:44:05 +0000 | [diff] [blame] | 507 | SM.AddNewSourceBuffer(F, SMLoc()); |
| 508 | |
Chris Lattner | d7e2505 | 2009-08-15 18:00:42 +0000 | [diff] [blame] | 509 | // Find all instances of CheckPrefix followed by : in the file. |
Chris Lattner | 9607703 | 2009-09-20 22:11:44 +0000 | [diff] [blame] | 510 | StringRef Buffer = F->getBuffer(); |
Chris Lattner | 81cb8ca | 2009-07-08 18:44:05 +0000 | [diff] [blame] | 511 | |
Chris Lattner | a29703e | 2009-09-24 20:39:13 +0000 | [diff] [blame] | 512 | std::vector<std::pair<SMLoc, Pattern> > NotMatches; |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 513 | |
Chris Lattner | 81cb8ca | 2009-07-08 18:44:05 +0000 | [diff] [blame] | 514 | while (1) { |
| 515 | // See if Prefix occurs in the memory buffer. |
Chris Lattner | 9607703 | 2009-09-20 22:11:44 +0000 | [diff] [blame] | 516 | Buffer = Buffer.substr(Buffer.find(CheckPrefix)); |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 517 | |
Chris Lattner | 81cb8ca | 2009-07-08 18:44:05 +0000 | [diff] [blame] | 518 | // If we didn't find a match, we're done. |
Chris Lattner | 9607703 | 2009-09-20 22:11:44 +0000 | [diff] [blame] | 519 | if (Buffer.empty()) |
Chris Lattner | 81cb8ca | 2009-07-08 18:44:05 +0000 | [diff] [blame] | 520 | break; |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 521 | |
Chris Lattner | 9607703 | 2009-09-20 22:11:44 +0000 | [diff] [blame] | 522 | const char *CheckPrefixStart = Buffer.data(); |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 523 | |
Chris Lattner | 5dafafd | 2009-08-15 18:32:21 +0000 | [diff] [blame] | 524 | // When we find a check prefix, keep track of whether we find CHECK: or |
| 525 | // CHECK-NEXT: |
Chris Lattner | f15380b | 2009-09-20 22:35:26 +0000 | [diff] [blame] | 526 | bool IsCheckNext = false, IsCheckNot = false; |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 527 | |
Chris Lattner | d7e2505 | 2009-08-15 18:00:42 +0000 | [diff] [blame] | 528 | // Verify that the : is present after the prefix. |
Chris Lattner | 9607703 | 2009-09-20 22:11:44 +0000 | [diff] [blame] | 529 | if (Buffer[CheckPrefix.size()] == ':') { |
| 530 | Buffer = Buffer.substr(CheckPrefix.size()+1); |
Chris Lattner | 9607703 | 2009-09-20 22:11:44 +0000 | [diff] [blame] | 531 | } else if (Buffer.size() > CheckPrefix.size()+6 && |
| 532 | memcmp(Buffer.data()+CheckPrefix.size(), "-NEXT:", 6) == 0) { |
| 533 | Buffer = Buffer.substr(CheckPrefix.size()+7); |
Chris Lattner | 5dafafd | 2009-08-15 18:32:21 +0000 | [diff] [blame] | 534 | IsCheckNext = true; |
Chris Lattner | f15380b | 2009-09-20 22:35:26 +0000 | [diff] [blame] | 535 | } else if (Buffer.size() > CheckPrefix.size()+5 && |
| 536 | memcmp(Buffer.data()+CheckPrefix.size(), "-NOT:", 5) == 0) { |
| 537 | Buffer = Buffer.substr(CheckPrefix.size()+6); |
| 538 | IsCheckNot = true; |
Chris Lattner | 5dafafd | 2009-08-15 18:32:21 +0000 | [diff] [blame] | 539 | } else { |
Chris Lattner | 9607703 | 2009-09-20 22:11:44 +0000 | [diff] [blame] | 540 | Buffer = Buffer.substr(1); |
Chris Lattner | d7e2505 | 2009-08-15 18:00:42 +0000 | [diff] [blame] | 541 | continue; |
| 542 | } |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 543 | |
Chris Lattner | 81cb8ca | 2009-07-08 18:44:05 +0000 | [diff] [blame] | 544 | // Okay, we found the prefix, yay. Remember the rest of the line, but |
| 545 | // ignore leading and trailing whitespace. |
Chris Lattner | f15380b | 2009-09-20 22:35:26 +0000 | [diff] [blame] | 546 | Buffer = Buffer.substr(Buffer.find_first_not_of(" \t")); |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 547 | |
Chris Lattner | 81cb8ca | 2009-07-08 18:44:05 +0000 | [diff] [blame] | 548 | // Scan ahead to the end of line. |
Chris Lattner | 9607703 | 2009-09-20 22:11:44 +0000 | [diff] [blame] | 549 | size_t EOL = Buffer.find_first_of("\n\r"); |
Chris Lattner | a29703e | 2009-09-24 20:39:13 +0000 | [diff] [blame] | 550 | |
Dan Gohman | e546343 | 2010-01-29 21:53:18 +0000 | [diff] [blame] | 551 | // Remember the location of the start of the pattern, for diagnostics. |
| 552 | SMLoc PatternLoc = SMLoc::getFromPointer(Buffer.data()); |
| 553 | |
Chris Lattner | a29703e | 2009-09-24 20:39:13 +0000 | [diff] [blame] | 554 | // Parse the pattern. |
| 555 | Pattern P; |
| 556 | if (P.ParsePattern(Buffer.substr(0, EOL), SM)) |
Chris Lattner | 81cb8ca | 2009-07-08 18:44:05 +0000 | [diff] [blame] | 557 | return true; |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 558 | |
Chris Lattner | a29703e | 2009-09-24 20:39:13 +0000 | [diff] [blame] | 559 | Buffer = Buffer.substr(EOL); |
| 560 | |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 561 | |
Chris Lattner | 5dafafd | 2009-08-15 18:32:21 +0000 | [diff] [blame] | 562 | // Verify that CHECK-NEXT lines have at least one CHECK line before them. |
| 563 | if (IsCheckNext && CheckStrings.empty()) { |
| 564 | SM.PrintMessage(SMLoc::getFromPointer(CheckPrefixStart), |
| 565 | "found '"+CheckPrefix+"-NEXT:' without previous '"+ |
| 566 | CheckPrefix+ ": line", "error"); |
| 567 | return true; |
| 568 | } |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 569 | |
Chris Lattner | a29703e | 2009-09-24 20:39:13 +0000 | [diff] [blame] | 570 | // Handle CHECK-NOT. |
| 571 | if (IsCheckNot) { |
| 572 | NotMatches.push_back(std::make_pair(SMLoc::getFromPointer(Buffer.data()), |
| 573 | P)); |
| 574 | continue; |
| 575 | } |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 576 | |
| 577 | |
Chris Lattner | 81cb8ca | 2009-07-08 18:44:05 +0000 | [diff] [blame] | 578 | // Okay, add the string we captured to the output vector and move on. |
Chris Lattner | 9fc6678 | 2009-09-24 20:25:55 +0000 | [diff] [blame] | 579 | CheckStrings.push_back(CheckString(P, |
Dan Gohman | e546343 | 2010-01-29 21:53:18 +0000 | [diff] [blame] | 580 | PatternLoc, |
Chris Lattner | 5dafafd | 2009-08-15 18:32:21 +0000 | [diff] [blame] | 581 | IsCheckNext)); |
Chris Lattner | f15380b | 2009-09-20 22:35:26 +0000 | [diff] [blame] | 582 | std::swap(NotMatches, CheckStrings.back().NotStrings); |
Chris Lattner | 81cb8ca | 2009-07-08 18:44:05 +0000 | [diff] [blame] | 583 | } |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 584 | |
Jakob Stoklund Olesen | 824c10e | 2010-10-15 17:47:12 +0000 | [diff] [blame] | 585 | // Add an EOF pattern for any trailing CHECK-NOTs. |
| 586 | if (!NotMatches.empty()) { |
| 587 | CheckStrings.push_back(CheckString(Pattern(true), |
| 588 | SMLoc::getFromPointer(Buffer.data()), |
| 589 | false)); |
| 590 | std::swap(NotMatches, CheckStrings.back().NotStrings); |
| 591 | } |
| 592 | |
Chris Lattner | 81cb8ca | 2009-07-08 18:44:05 +0000 | [diff] [blame] | 593 | if (CheckStrings.empty()) { |
Chris Lattner | d7e2505 | 2009-08-15 18:00:42 +0000 | [diff] [blame] | 594 | errs() << "error: no check strings found with prefix '" << CheckPrefix |
| 595 | << ":'\n"; |
Chris Lattner | 81cb8ca | 2009-07-08 18:44:05 +0000 | [diff] [blame] | 596 | return true; |
| 597 | } |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 598 | |
Chris Lattner | 81cb8ca | 2009-07-08 18:44:05 +0000 | [diff] [blame] | 599 | return false; |
| 600 | } |
| 601 | |
Chris Lattner | 5dafafd | 2009-08-15 18:32:21 +0000 | [diff] [blame] | 602 | static void PrintCheckFailed(const SourceMgr &SM, const CheckString &CheckStr, |
Daniel Dunbar | fafe93c | 2009-11-22 22:08:06 +0000 | [diff] [blame] | 603 | StringRef Buffer, |
| 604 | StringMap<StringRef> &VariableTable) { |
Chris Lattner | 5dafafd | 2009-08-15 18:32:21 +0000 | [diff] [blame] | 605 | // Otherwise, we have an error, emit an error message. |
| 606 | SM.PrintMessage(CheckStr.Loc, "expected string not found in input", |
| 607 | "error"); |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 608 | |
Chris Lattner | 5dafafd | 2009-08-15 18:32:21 +0000 | [diff] [blame] | 609 | // Print the "scanning from here" line. If the current position is at the |
| 610 | // end of a line, advance to the start of the next line. |
Chris Lattner | 9607703 | 2009-09-20 22:11:44 +0000 | [diff] [blame] | 611 | Buffer = Buffer.substr(Buffer.find_first_not_of(" \t\n\r")); |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 612 | |
Chris Lattner | 9607703 | 2009-09-20 22:11:44 +0000 | [diff] [blame] | 613 | SM.PrintMessage(SMLoc::getFromPointer(Buffer.data()), "scanning from here", |
Chris Lattner | 5dafafd | 2009-08-15 18:32:21 +0000 | [diff] [blame] | 614 | "note"); |
Daniel Dunbar | fafe93c | 2009-11-22 22:08:06 +0000 | [diff] [blame] | 615 | |
| 616 | // Allow the pattern to print additional information if desired. |
| 617 | CheckStr.Pat.PrintFailureInfo(SM, Buffer, VariableTable); |
Chris Lattner | 5dafafd | 2009-08-15 18:32:21 +0000 | [diff] [blame] | 618 | } |
| 619 | |
Chris Lattner | 3711b7a | 2009-09-20 22:42:44 +0000 | [diff] [blame] | 620 | /// CountNumNewlinesBetween - Count the number of newlines in the specified |
| 621 | /// range. |
| 622 | static unsigned CountNumNewlinesBetween(StringRef Range) { |
Chris Lattner | 5dafafd | 2009-08-15 18:32:21 +0000 | [diff] [blame] | 623 | unsigned NumNewLines = 0; |
Chris Lattner | 3711b7a | 2009-09-20 22:42:44 +0000 | [diff] [blame] | 624 | while (1) { |
Chris Lattner | 5dafafd | 2009-08-15 18:32:21 +0000 | [diff] [blame] | 625 | // Scan for newline. |
Chris Lattner | 3711b7a | 2009-09-20 22:42:44 +0000 | [diff] [blame] | 626 | Range = Range.substr(Range.find_first_of("\n\r")); |
| 627 | if (Range.empty()) return NumNewLines; |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 628 | |
Chris Lattner | 5dafafd | 2009-08-15 18:32:21 +0000 | [diff] [blame] | 629 | ++NumNewLines; |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 630 | |
Chris Lattner | 5dafafd | 2009-08-15 18:32:21 +0000 | [diff] [blame] | 631 | // Handle \n\r and \r\n as a single newline. |
Chris Lattner | 3711b7a | 2009-09-20 22:42:44 +0000 | [diff] [blame] | 632 | if (Range.size() > 1 && |
| 633 | (Range[1] == '\n' || Range[1] == '\r') && |
| 634 | (Range[0] != Range[1])) |
| 635 | Range = Range.substr(1); |
| 636 | Range = Range.substr(1); |
Chris Lattner | 5dafafd | 2009-08-15 18:32:21 +0000 | [diff] [blame] | 637 | } |
Chris Lattner | 5dafafd | 2009-08-15 18:32:21 +0000 | [diff] [blame] | 638 | } |
| 639 | |
Chris Lattner | 81cb8ca | 2009-07-08 18:44:05 +0000 | [diff] [blame] | 640 | int main(int argc, char **argv) { |
| 641 | sys::PrintStackTraceOnErrorSignal(); |
| 642 | PrettyStackTraceProgram X(argc, argv); |
| 643 | cl::ParseCommandLineOptions(argc, argv); |
| 644 | |
| 645 | SourceMgr SM; |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 646 | |
Chris Lattner | 81cb8ca | 2009-07-08 18:44:05 +0000 | [diff] [blame] | 647 | // Read the expected strings from the check file. |
Chris Lattner | 207e1bc | 2009-08-15 17:41:04 +0000 | [diff] [blame] | 648 | std::vector<CheckString> CheckStrings; |
Chris Lattner | 81cb8ca | 2009-07-08 18:44:05 +0000 | [diff] [blame] | 649 | if (ReadCheckFile(SM, CheckStrings)) |
| 650 | return 2; |
| 651 | |
| 652 | // Open the file to check and add it to SourceMgr. |
Michael J. Spencer | 3ff9563 | 2010-12-16 03:29:14 +0000 | [diff] [blame] | 653 | OwningPtr<MemoryBuffer> File; |
| 654 | if (error_code ec = |
| 655 | MemoryBuffer::getFileOrSTDIN(InputFilename.c_str(), File)) { |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 656 | errs() << "Could not open input file '" << InputFilename << "': " |
Michael J. Spencer | 333fb04 | 2010-12-09 17:36:48 +0000 | [diff] [blame] | 657 | << ec.message() << '\n'; |
Chris Lattner | 81cb8ca | 2009-07-08 18:44:05 +0000 | [diff] [blame] | 658 | return true; |
| 659 | } |
Michael J. Spencer | 3ff9563 | 2010-12-16 03:29:14 +0000 | [diff] [blame] | 660 | MemoryBuffer *F = File.take(); |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 661 | |
Chris Lattner | 88a7e9e | 2009-07-11 18:58:15 +0000 | [diff] [blame] | 662 | // Remove duplicate spaces in the input file if requested. |
| 663 | if (!NoCanonicalizeWhiteSpace) |
| 664 | F = CanonicalizeInputFile(F); |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 665 | |
Chris Lattner | 81cb8ca | 2009-07-08 18:44:05 +0000 | [diff] [blame] | 666 | SM.AddNewSourceBuffer(F, SMLoc()); |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 667 | |
Chris Lattner | eec9695 | 2009-09-27 07:56:52 +0000 | [diff] [blame] | 668 | /// VariableTable - This holds all the current filecheck variables. |
| 669 | StringMap<StringRef> VariableTable; |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 670 | |
Chris Lattner | 81cb8ca | 2009-07-08 18:44:05 +0000 | [diff] [blame] | 671 | // Check that we have all of the expected strings, in order, in the input |
| 672 | // file. |
Chris Lattner | 9607703 | 2009-09-20 22:11:44 +0000 | [diff] [blame] | 673 | StringRef Buffer = F->getBuffer(); |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 674 | |
Chris Lattner | f15380b | 2009-09-20 22:35:26 +0000 | [diff] [blame] | 675 | const char *LastMatch = Buffer.data(); |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 676 | |
Chris Lattner | 81cb8ca | 2009-07-08 18:44:05 +0000 | [diff] [blame] | 677 | for (unsigned StrNo = 0, e = CheckStrings.size(); StrNo != e; ++StrNo) { |
Chris Lattner | 207e1bc | 2009-08-15 17:41:04 +0000 | [diff] [blame] | 678 | const CheckString &CheckStr = CheckStrings[StrNo]; |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 679 | |
Chris Lattner | 9607703 | 2009-09-20 22:11:44 +0000 | [diff] [blame] | 680 | StringRef SearchFrom = Buffer; |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 681 | |
Chris Lattner | 81cb8ca | 2009-07-08 18:44:05 +0000 | [diff] [blame] | 682 | // Find StrNo in the file. |
Chris Lattner | 9fc6678 | 2009-09-24 20:25:55 +0000 | [diff] [blame] | 683 | size_t MatchLen = 0; |
Jakob Stoklund Olesen | 824c10e | 2010-10-15 17:47:12 +0000 | [diff] [blame] | 684 | size_t MatchPos = CheckStr.Pat.Match(Buffer, MatchLen, VariableTable); |
| 685 | Buffer = Buffer.substr(MatchPos); |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 686 | |
Chris Lattner | 5dafafd | 2009-08-15 18:32:21 +0000 | [diff] [blame] | 687 | // If we didn't find a match, reject the input. |
Jakob Stoklund Olesen | 824c10e | 2010-10-15 17:47:12 +0000 | [diff] [blame] | 688 | if (MatchPos == StringRef::npos) { |
Daniel Dunbar | fafe93c | 2009-11-22 22:08:06 +0000 | [diff] [blame] | 689 | PrintCheckFailed(SM, CheckStr, SearchFrom, VariableTable); |
Chris Lattner | 5dafafd | 2009-08-15 18:32:21 +0000 | [diff] [blame] | 690 | return 1; |
Chris Lattner | 81cb8ca | 2009-07-08 18:44:05 +0000 | [diff] [blame] | 691 | } |
Chris Lattner | 3711b7a | 2009-09-20 22:42:44 +0000 | [diff] [blame] | 692 | |
| 693 | StringRef SkippedRegion(LastMatch, Buffer.data()-LastMatch); |
| 694 | |
Chris Lattner | 5dafafd | 2009-08-15 18:32:21 +0000 | [diff] [blame] | 695 | // If this check is a "CHECK-NEXT", verify that the previous match was on |
| 696 | // the previous line (i.e. that there is one newline between them). |
| 697 | if (CheckStr.IsCheckNext) { |
| 698 | // Count the number of newlines between the previous match and this one. |
Chris Lattner | f15380b | 2009-09-20 22:35:26 +0000 | [diff] [blame] | 699 | assert(LastMatch != F->getBufferStart() && |
| 700 | "CHECK-NEXT can't be the first check in a file"); |
Chris Lattner | 5dafafd | 2009-08-15 18:32:21 +0000 | [diff] [blame] | 701 | |
Chris Lattner | 3711b7a | 2009-09-20 22:42:44 +0000 | [diff] [blame] | 702 | unsigned NumNewLines = CountNumNewlinesBetween(SkippedRegion); |
Chris Lattner | 5dafafd | 2009-08-15 18:32:21 +0000 | [diff] [blame] | 703 | if (NumNewLines == 0) { |
Chris Lattner | 0b2353f | 2009-08-16 02:22:31 +0000 | [diff] [blame] | 704 | SM.PrintMessage(CheckStr.Loc, |
Chris Lattner | 5dafafd | 2009-08-15 18:32:21 +0000 | [diff] [blame] | 705 | CheckPrefix+"-NEXT: is on the same line as previous match", |
| 706 | "error"); |
Chris Lattner | 9607703 | 2009-09-20 22:11:44 +0000 | [diff] [blame] | 707 | SM.PrintMessage(SMLoc::getFromPointer(Buffer.data()), |
Chris Lattner | 0b2353f | 2009-08-16 02:22:31 +0000 | [diff] [blame] | 708 | "'next' match was here", "note"); |
Chris Lattner | 5dafafd | 2009-08-15 18:32:21 +0000 | [diff] [blame] | 709 | SM.PrintMessage(SMLoc::getFromPointer(LastMatch), |
| 710 | "previous match was here", "note"); |
| 711 | return 1; |
| 712 | } |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 713 | |
Chris Lattner | 5dafafd | 2009-08-15 18:32:21 +0000 | [diff] [blame] | 714 | if (NumNewLines != 1) { |
Chris Lattner | 0b2353f | 2009-08-16 02:22:31 +0000 | [diff] [blame] | 715 | SM.PrintMessage(CheckStr.Loc, |
Chris Lattner | 5dafafd | 2009-08-15 18:32:21 +0000 | [diff] [blame] | 716 | CheckPrefix+ |
| 717 | "-NEXT: is not on the line after the previous match", |
| 718 | "error"); |
Chris Lattner | 9607703 | 2009-09-20 22:11:44 +0000 | [diff] [blame] | 719 | SM.PrintMessage(SMLoc::getFromPointer(Buffer.data()), |
Chris Lattner | 0b2353f | 2009-08-16 02:22:31 +0000 | [diff] [blame] | 720 | "'next' match was here", "note"); |
Chris Lattner | 5dafafd | 2009-08-15 18:32:21 +0000 | [diff] [blame] | 721 | SM.PrintMessage(SMLoc::getFromPointer(LastMatch), |
| 722 | "previous match was here", "note"); |
| 723 | return 1; |
| 724 | } |
| 725 | } |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 726 | |
Chris Lattner | f15380b | 2009-09-20 22:35:26 +0000 | [diff] [blame] | 727 | // If this match had "not strings", verify that they don't exist in the |
| 728 | // skipped region. |
Chris Lattner | eec9695 | 2009-09-27 07:56:52 +0000 | [diff] [blame] | 729 | for (unsigned ChunkNo = 0, e = CheckStr.NotStrings.size(); |
| 730 | ChunkNo != e; ++ChunkNo) { |
Chris Lattner | a29703e | 2009-09-24 20:39:13 +0000 | [diff] [blame] | 731 | size_t MatchLen = 0; |
Chris Lattner | eec9695 | 2009-09-27 07:56:52 +0000 | [diff] [blame] | 732 | size_t Pos = CheckStr.NotStrings[ChunkNo].second.Match(SkippedRegion, |
| 733 | MatchLen, |
| 734 | VariableTable); |
Chris Lattner | f15380b | 2009-09-20 22:35:26 +0000 | [diff] [blame] | 735 | if (Pos == StringRef::npos) continue; |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 736 | |
Chris Lattner | f15380b | 2009-09-20 22:35:26 +0000 | [diff] [blame] | 737 | SM.PrintMessage(SMLoc::getFromPointer(LastMatch+Pos), |
| 738 | CheckPrefix+"-NOT: string occurred!", "error"); |
Chris Lattner | 5287008 | 2009-09-24 21:47:32 +0000 | [diff] [blame] | 739 | SM.PrintMessage(CheckStr.NotStrings[ChunkNo].first, |
Chris Lattner | f15380b | 2009-09-20 22:35:26 +0000 | [diff] [blame] | 740 | CheckPrefix+"-NOT: pattern specified here", "note"); |
| 741 | return 1; |
| 742 | } |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 743 | |
Chris Lattner | 5dafafd | 2009-08-15 18:32:21 +0000 | [diff] [blame] | 744 | |
Chris Lattner | 8111576 | 2009-09-21 02:30:42 +0000 | [diff] [blame] | 745 | // Otherwise, everything is good. Step over the matched text and remember |
| 746 | // the position after the match as the end of the last match. |
Chris Lattner | 9fc6678 | 2009-09-24 20:25:55 +0000 | [diff] [blame] | 747 | Buffer = Buffer.substr(MatchLen); |
Chris Lattner | 8111576 | 2009-09-21 02:30:42 +0000 | [diff] [blame] | 748 | LastMatch = Buffer.data(); |
Chris Lattner | 81cb8ca | 2009-07-08 18:44:05 +0000 | [diff] [blame] | 749 | } |
Mikhail Glushenkov | 7112c86 | 2010-08-20 17:38:38 +0000 | [diff] [blame] | 750 | |
Chris Lattner | 81cb8ca | 2009-07-08 18:44:05 +0000 | [diff] [blame] | 751 | return 0; |
| 752 | } |