blob: 0805504ad17a8e0ac75111812e69cf0f2ec85ed2 [file] [log] [blame]
Chris Lattner81cb8ca2009-07-08 18:44:05 +00001//===- 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. Spencer3ff95632010-12-16 03:29:14 +000019#include "llvm/ADT/OwningPtr.h"
Chris Lattner81cb8ca2009-07-08 18:44:05 +000020#include "llvm/Support/CommandLine.h"
21#include "llvm/Support/MemoryBuffer.h"
22#include "llvm/Support/PrettyStackTrace.h"
Chris Lattner52870082009-09-24 21:47:32 +000023#include "llvm/Support/Regex.h"
Chris Lattner81cb8ca2009-07-08 18:44:05 +000024#include "llvm/Support/SourceMgr.h"
25#include "llvm/Support/raw_ostream.h"
Michael J. Spencer1f6efa32010-11-29 18:16:10 +000026#include "llvm/Support/Signals.h"
Michael J. Spencer333fb042010-12-09 17:36:48 +000027#include "llvm/Support/system_error.h"
Daniel Dunbarfafe93c2009-11-22 22:08:06 +000028#include "llvm/ADT/SmallString.h"
Chris Lattnereec96952009-09-27 07:56:52 +000029#include "llvm/ADT/StringMap.h"
30#include <algorithm>
Chris Lattner81cb8ca2009-07-08 18:44:05 +000031using namespace llvm;
32
33static cl::opt<std::string>
34CheckFilename(cl::Positional, cl::desc("<check-file>"), cl::Required);
35
36static cl::opt<std::string>
37InputFilename("input-file", cl::desc("File to check (defaults to stdin)"),
38 cl::init("-"), cl::value_desc("filename"));
39
40static cl::opt<std::string>
41CheckPrefix("check-prefix", cl::init("CHECK"),
42 cl::desc("Prefix to use from check file (defaults to 'CHECK')"));
43
Chris Lattner88a7e9e2009-07-11 18:58:15 +000044static cl::opt<bool>
45NoCanonicalizeWhiteSpace("strict-whitespace",
46 cl::desc("Do not treat all horizontal whitespace as equivalent"));
47
Ted Kremenek2e578382012-09-07 06:47:16 +000048static cl::opt<bool>
49NoRegex("exact-match",
50 cl::desc("Look for exact matches without using regular expressions"));
51
Chris Lattnera29703e2009-09-24 20:39:13 +000052//===----------------------------------------------------------------------===//
53// Pattern Handling Code.
54//===----------------------------------------------------------------------===//
55
Chris Lattner9fc66782009-09-24 20:25:55 +000056class Pattern {
Chris Lattner94638f02009-09-25 17:29:36 +000057 SMLoc PatternLoc;
Mikhail Glushenkov7112c862010-08-20 17:38:38 +000058
Jakob Stoklund Olesen824c10e2010-10-15 17:47:12 +000059 /// MatchEOF - When set, this pattern only matches the end of file. This is
60 /// used for trailing CHECK-NOTs.
61 bool MatchEOF;
62
Chris Lattner5d6a05f2009-09-25 17:23:43 +000063 /// FixedStr - If non-empty, this pattern is a fixed string match with the
64 /// specified fixed string.
Chris Lattner2702e6a2009-09-25 17:09:12 +000065 StringRef FixedStr;
Mikhail Glushenkov7112c862010-08-20 17:38:38 +000066
Chris Lattner5d6a05f2009-09-25 17:23:43 +000067 /// RegEx - If non-empty, this is a regex pattern.
68 std::string RegExStr;
Mikhail Glushenkov7112c862010-08-20 17:38:38 +000069
Chris Lattnereec96952009-09-27 07:56:52 +000070 /// VariableUses - Entries in this vector map to uses of a variable in the
71 /// pattern, e.g. "foo[[bar]]baz". In this case, the RegExStr will contain
72 /// "foobaz" and we'll get an entry in this vector that tells us to insert the
73 /// value of bar at offset 3.
74 std::vector<std::pair<StringRef, unsigned> > VariableUses;
Mikhail Glushenkov7112c862010-08-20 17:38:38 +000075
Chris Lattnereec96952009-09-27 07:56:52 +000076 /// VariableDefs - Entries in this vector map to definitions of a variable in
77 /// the pattern, e.g. "foo[[bar:.*]]baz". In this case, the RegExStr will
78 /// contain "foo(.*)baz" and VariableDefs will contain the pair "bar",1. The
79 /// index indicates what parenthesized value captures the variable value.
80 std::vector<std::pair<StringRef, unsigned> > VariableDefs;
Mikhail Glushenkov7112c862010-08-20 17:38:38 +000081
Chris Lattner9fc66782009-09-24 20:25:55 +000082public:
Mikhail Glushenkov7112c862010-08-20 17:38:38 +000083
Jakob Stoklund Olesen824c10e2010-10-15 17:47:12 +000084 Pattern(bool matchEOF = false) : MatchEOF(matchEOF) { }
Mikhail Glushenkov7112c862010-08-20 17:38:38 +000085
Chris Lattnera29703e2009-09-24 20:39:13 +000086 bool ParsePattern(StringRef PatternStr, SourceMgr &SM);
Mikhail Glushenkov7112c862010-08-20 17:38:38 +000087
Chris Lattner9fc66782009-09-24 20:25:55 +000088 /// Match - Match the pattern string against the input buffer Buffer. This
89 /// returns the position that is matched or npos if there is no match. If
90 /// there is a match, the size of the matched string is returned in MatchLen.
Chris Lattnereec96952009-09-27 07:56:52 +000091 ///
92 /// The VariableTable StringMap provides the current values of filecheck
93 /// variables and is updated if this match defines new values.
94 size_t Match(StringRef Buffer, size_t &MatchLen,
95 StringMap<StringRef> &VariableTable) const;
Daniel Dunbarfafe93c2009-11-22 22:08:06 +000096
97 /// PrintFailureInfo - Print additional information about a failure to match
98 /// involving this pattern.
99 void PrintFailureInfo(const SourceMgr &SM, StringRef Buffer,
100 const StringMap<StringRef> &VariableTable) const;
101
Chris Lattner5d6a05f2009-09-25 17:23:43 +0000102private:
Chris Lattnereec96952009-09-27 07:56:52 +0000103 static void AddFixedStringToRegEx(StringRef FixedStr, std::string &TheStr);
104 bool AddRegExToRegEx(StringRef RegExStr, unsigned &CurParen, SourceMgr &SM);
Daniel Dunbaread2dac2009-11-22 22:59:26 +0000105
106 /// ComputeMatchDistance - Compute an arbitrary estimate for the quality of
107 /// matching this pattern at the start of \arg Buffer; a distance of zero
108 /// should correspond to a perfect match.
109 unsigned ComputeMatchDistance(StringRef Buffer,
110 const StringMap<StringRef> &VariableTable) const;
Chris Lattner9fc66782009-09-24 20:25:55 +0000111};
112
Chris Lattnereec96952009-09-27 07:56:52 +0000113
Chris Lattnera29703e2009-09-24 20:39:13 +0000114bool Pattern::ParsePattern(StringRef PatternStr, SourceMgr &SM) {
Chris Lattner94638f02009-09-25 17:29:36 +0000115 PatternLoc = SMLoc::getFromPointer(PatternStr.data());
Mikhail Glushenkov7112c862010-08-20 17:38:38 +0000116
Chris Lattnera29703e2009-09-24 20:39:13 +0000117 // Ignore trailing whitespace.
118 while (!PatternStr.empty() &&
119 (PatternStr.back() == ' ' || PatternStr.back() == '\t'))
120 PatternStr = PatternStr.substr(0, PatternStr.size()-1);
Mikhail Glushenkov7112c862010-08-20 17:38:38 +0000121
Chris Lattnera29703e2009-09-24 20:39:13 +0000122 // Check that there is something on the line.
123 if (PatternStr.empty()) {
Chris Lattner3f2d5f62011-10-16 05:43:57 +0000124 SM.PrintMessage(PatternLoc, SourceMgr::DK_Error,
125 "found empty check string with prefix '" +
126 CheckPrefix+":'");
Chris Lattnera29703e2009-09-24 20:39:13 +0000127 return true;
128 }
Mikhail Glushenkov7112c862010-08-20 17:38:38 +0000129
Chris Lattner2702e6a2009-09-25 17:09:12 +0000130 // Check to see if this is a fixed string, or if it has regex pieces.
Ted Kremenek2e578382012-09-07 06:47:16 +0000131 if (PatternStr.size() < 2 || NoRegex ||
Chris Lattnereec96952009-09-27 07:56:52 +0000132 (PatternStr.find("{{") == StringRef::npos &&
133 PatternStr.find("[[") == StringRef::npos)) {
Chris Lattner2702e6a2009-09-25 17:09:12 +0000134 FixedStr = PatternStr;
135 return false;
136 }
Mikhail Glushenkov7112c862010-08-20 17:38:38 +0000137
Chris Lattnereec96952009-09-27 07:56:52 +0000138 // Paren value #0 is for the fully matched string. Any new parenthesized
Chris Lattner13a38c42011-04-09 06:18:02 +0000139 // values add from there.
Chris Lattnereec96952009-09-27 07:56:52 +0000140 unsigned CurParen = 1;
Mikhail Glushenkov7112c862010-08-20 17:38:38 +0000141
Chris Lattner5d6a05f2009-09-25 17:23:43 +0000142 // Otherwise, there is at least one regex piece. Build up the regex pattern
143 // by escaping scary characters in fixed strings, building up one big regex.
Chris Lattner52870082009-09-24 21:47:32 +0000144 while (!PatternStr.empty()) {
Chris Lattnereec96952009-09-27 07:56:52 +0000145 // RegEx matches.
Chris Lattner13a38c42011-04-09 06:18:02 +0000146 if (PatternStr.startswith("{{")) {
Mikhail Glushenkov7112c862010-08-20 17:38:38 +0000147
Chris Lattnereec96952009-09-27 07:56:52 +0000148 // Otherwise, this is the start of a regex match. Scan for the }}.
149 size_t End = PatternStr.find("}}");
150 if (End == StringRef::npos) {
151 SM.PrintMessage(SMLoc::getFromPointer(PatternStr.data()),
Chris Lattner3f2d5f62011-10-16 05:43:57 +0000152 SourceMgr::DK_Error,
153 "found start of regex string with no end '}}'");
Chris Lattnereec96952009-09-27 07:56:52 +0000154 return true;
155 }
Mikhail Glushenkov7112c862010-08-20 17:38:38 +0000156
Chris Lattner42e31df2011-04-09 06:37:03 +0000157 // Enclose {{}} patterns in parens just like [[]] even though we're not
158 // capturing the result for any purpose. This is required in case the
159 // expression contains an alternation like: CHECK: abc{{x|z}}def. We
160 // want this to turn into: "abc(x|z)def" not "abcx|zdef".
161 RegExStr += '(';
162 ++CurParen;
163
Chris Lattnereec96952009-09-27 07:56:52 +0000164 if (AddRegExToRegEx(PatternStr.substr(2, End-2), CurParen, SM))
165 return true;
Chris Lattner42e31df2011-04-09 06:37:03 +0000166 RegExStr += ')';
Chris Lattner13a38c42011-04-09 06:18:02 +0000167
Chris Lattnereec96952009-09-27 07:56:52 +0000168 PatternStr = PatternStr.substr(End+2);
Chris Lattner52870082009-09-24 21:47:32 +0000169 continue;
170 }
Mikhail Glushenkov7112c862010-08-20 17:38:38 +0000171
Chris Lattnereec96952009-09-27 07:56:52 +0000172 // Named RegEx matches. These are of two forms: [[foo:.*]] which matches .*
173 // (or some other regex) and assigns it to the FileCheck variable 'foo'. The
174 // second form is [[foo]] which is a reference to foo. The variable name
Daniel Dunbar964ac012009-11-22 22:07:50 +0000175 // itself must be of the form "[a-zA-Z_][0-9a-zA-Z_]*", otherwise we reject
Chris Lattnereec96952009-09-27 07:56:52 +0000176 // it. This is to catch some common errors.
Chris Lattner13a38c42011-04-09 06:18:02 +0000177 if (PatternStr.startswith("[[")) {
Chris Lattnereec96952009-09-27 07:56:52 +0000178 // Verify that it is terminated properly.
179 size_t End = PatternStr.find("]]");
180 if (End == StringRef::npos) {
181 SM.PrintMessage(SMLoc::getFromPointer(PatternStr.data()),
Chris Lattner3f2d5f62011-10-16 05:43:57 +0000182 SourceMgr::DK_Error,
183 "invalid named regex reference, no ]] found");
Chris Lattnereec96952009-09-27 07:56:52 +0000184 return true;
185 }
Mikhail Glushenkov7112c862010-08-20 17:38:38 +0000186
Chris Lattnereec96952009-09-27 07:56:52 +0000187 StringRef MatchStr = PatternStr.substr(2, End-2);
188 PatternStr = PatternStr.substr(End+2);
Mikhail Glushenkov7112c862010-08-20 17:38:38 +0000189
Chris Lattnereec96952009-09-27 07:56:52 +0000190 // Get the regex name (e.g. "foo").
191 size_t NameEnd = MatchStr.find(':');
192 StringRef Name = MatchStr.substr(0, NameEnd);
Mikhail Glushenkov7112c862010-08-20 17:38:38 +0000193
Chris Lattnereec96952009-09-27 07:56:52 +0000194 if (Name.empty()) {
Chris Lattner3f2d5f62011-10-16 05:43:57 +0000195 SM.PrintMessage(SMLoc::getFromPointer(Name.data()), SourceMgr::DK_Error,
196 "invalid name in named regex: empty name");
Chris Lattnereec96952009-09-27 07:56:52 +0000197 return true;
198 }
199
200 // Verify that the name is well formed.
201 for (unsigned i = 0, e = Name.size(); i != e; ++i)
Chris Lattner13a38c42011-04-09 06:18:02 +0000202 if (Name[i] != '_' && !isalnum(Name[i])) {
Chris Lattnereec96952009-09-27 07:56:52 +0000203 SM.PrintMessage(SMLoc::getFromPointer(Name.data()+i),
Chris Lattner3f2d5f62011-10-16 05:43:57 +0000204 SourceMgr::DK_Error, "invalid name in named regex");
Chris Lattnereec96952009-09-27 07:56:52 +0000205 return true;
206 }
Mikhail Glushenkov7112c862010-08-20 17:38:38 +0000207
Chris Lattnereec96952009-09-27 07:56:52 +0000208 // Name can't start with a digit.
209 if (isdigit(Name[0])) {
Chris Lattner3f2d5f62011-10-16 05:43:57 +0000210 SM.PrintMessage(SMLoc::getFromPointer(Name.data()), SourceMgr::DK_Error,
211 "invalid name in named regex");
Chris Lattnereec96952009-09-27 07:56:52 +0000212 return true;
213 }
Mikhail Glushenkov7112c862010-08-20 17:38:38 +0000214
Chris Lattnereec96952009-09-27 07:56:52 +0000215 // Handle [[foo]].
216 if (NameEnd == StringRef::npos) {
217 VariableUses.push_back(std::make_pair(Name, RegExStr.size()));
218 continue;
219 }
Mikhail Glushenkov7112c862010-08-20 17:38:38 +0000220
Chris Lattnereec96952009-09-27 07:56:52 +0000221 // Handle [[foo:.*]].
222 VariableDefs.push_back(std::make_pair(Name, CurParen));
223 RegExStr += '(';
224 ++CurParen;
Mikhail Glushenkov7112c862010-08-20 17:38:38 +0000225
Chris Lattnereec96952009-09-27 07:56:52 +0000226 if (AddRegExToRegEx(MatchStr.substr(NameEnd+1), CurParen, SM))
227 return true;
228
229 RegExStr += ')';
Chris Lattner52870082009-09-24 21:47:32 +0000230 }
Mikhail Glushenkov7112c862010-08-20 17:38:38 +0000231
Chris Lattnereec96952009-09-27 07:56:52 +0000232 // Handle fixed string matches.
233 // Find the end, which is the start of the next regex.
234 size_t FixedMatchEnd = PatternStr.find("{{");
235 FixedMatchEnd = std::min(FixedMatchEnd, PatternStr.find("[["));
236 AddFixedStringToRegEx(PatternStr.substr(0, FixedMatchEnd), RegExStr);
237 PatternStr = PatternStr.substr(FixedMatchEnd);
238 continue;
Chris Lattner52870082009-09-24 21:47:32 +0000239 }
Chris Lattneradea46e2009-09-24 20:45:07 +0000240
Chris Lattnera29703e2009-09-24 20:39:13 +0000241 return false;
242}
243
Chris Lattnereec96952009-09-27 07:56:52 +0000244void Pattern::AddFixedStringToRegEx(StringRef FixedStr, std::string &TheStr) {
Chris Lattner5d6a05f2009-09-25 17:23:43 +0000245 // Add the characters from FixedStr to the regex, escaping as needed. This
246 // avoids "leaning toothpicks" in common patterns.
247 for (unsigned i = 0, e = FixedStr.size(); i != e; ++i) {
248 switch (FixedStr[i]) {
249 // These are the special characters matched in "p_ere_exp".
250 case '(':
251 case ')':
252 case '^':
253 case '$':
254 case '|':
255 case '*':
256 case '+':
257 case '?':
258 case '.':
259 case '[':
260 case '\\':
261 case '{':
Chris Lattnereec96952009-09-27 07:56:52 +0000262 TheStr += '\\';
Chris Lattner5d6a05f2009-09-25 17:23:43 +0000263 // FALL THROUGH.
264 default:
Chris Lattnereec96952009-09-27 07:56:52 +0000265 TheStr += FixedStr[i];
Chris Lattner5d6a05f2009-09-25 17:23:43 +0000266 break;
267 }
268 }
269}
270
Chris Lattnereec96952009-09-27 07:56:52 +0000271bool Pattern::AddRegExToRegEx(StringRef RegexStr, unsigned &CurParen,
272 SourceMgr &SM) {
273 Regex R(RegexStr);
274 std::string Error;
275 if (!R.isValid(Error)) {
Chris Lattner3f2d5f62011-10-16 05:43:57 +0000276 SM.PrintMessage(SMLoc::getFromPointer(RegexStr.data()), SourceMgr::DK_Error,
277 "invalid regex: " + Error);
Chris Lattnereec96952009-09-27 07:56:52 +0000278 return true;
279 }
Mikhail Glushenkov7112c862010-08-20 17:38:38 +0000280
Chris Lattnereec96952009-09-27 07:56:52 +0000281 RegExStr += RegexStr.str();
282 CurParen += R.getNumMatches();
283 return false;
284}
Chris Lattner5d6a05f2009-09-25 17:23:43 +0000285
Chris Lattner52870082009-09-24 21:47:32 +0000286/// Match - Match the pattern string against the input buffer Buffer. This
287/// returns the position that is matched or npos if there is no match. If
288/// there is a match, the size of the matched string is returned in MatchLen.
Chris Lattnereec96952009-09-27 07:56:52 +0000289size_t Pattern::Match(StringRef Buffer, size_t &MatchLen,
290 StringMap<StringRef> &VariableTable) const {
Jakob Stoklund Olesen824c10e2010-10-15 17:47:12 +0000291 // If this is the EOF pattern, match it immediately.
292 if (MatchEOF) {
293 MatchLen = 0;
294 return Buffer.size();
295 }
296
Chris Lattner2702e6a2009-09-25 17:09:12 +0000297 // If this is a fixed string pattern, just match it now.
298 if (!FixedStr.empty()) {
299 MatchLen = FixedStr.size();
300 return Buffer.find(FixedStr);
301 }
Chris Lattnereec96952009-09-27 07:56:52 +0000302
Chris Lattner5d6a05f2009-09-25 17:23:43 +0000303 // Regex match.
Mikhail Glushenkov7112c862010-08-20 17:38:38 +0000304
Chris Lattnereec96952009-09-27 07:56:52 +0000305 // If there are variable uses, we need to create a temporary string with the
306 // actual value.
307 StringRef RegExToMatch = RegExStr;
308 std::string TmpStr;
309 if (!VariableUses.empty()) {
310 TmpStr = RegExStr;
Mikhail Glushenkov7112c862010-08-20 17:38:38 +0000311
Chris Lattnereec96952009-09-27 07:56:52 +0000312 unsigned InsertOffset = 0;
313 for (unsigned i = 0, e = VariableUses.size(); i != e; ++i) {
Daniel Dunbarfafe93c2009-11-22 22:08:06 +0000314 StringMap<StringRef>::iterator it =
315 VariableTable.find(VariableUses[i].first);
316 // If the variable is undefined, return an error.
317 if (it == VariableTable.end())
318 return StringRef::npos;
319
Chris Lattnereec96952009-09-27 07:56:52 +0000320 // Look up the value and escape it so that we can plop it into the regex.
321 std::string Value;
Daniel Dunbarfafe93c2009-11-22 22:08:06 +0000322 AddFixedStringToRegEx(it->second, Value);
Mikhail Glushenkov7112c862010-08-20 17:38:38 +0000323
Chris Lattnereec96952009-09-27 07:56:52 +0000324 // Plop it into the regex at the adjusted offset.
325 TmpStr.insert(TmpStr.begin()+VariableUses[i].second+InsertOffset,
326 Value.begin(), Value.end());
327 InsertOffset += Value.size();
328 }
Mikhail Glushenkov7112c862010-08-20 17:38:38 +0000329
Chris Lattnereec96952009-09-27 07:56:52 +0000330 // Match the newly constructed regex.
331 RegExToMatch = TmpStr;
332 }
Mikhail Glushenkov7112c862010-08-20 17:38:38 +0000333
334
Chris Lattner5d6a05f2009-09-25 17:23:43 +0000335 SmallVector<StringRef, 4> MatchInfo;
Chris Lattnereec96952009-09-27 07:56:52 +0000336 if (!Regex(RegExToMatch, Regex::Newline).match(Buffer, &MatchInfo))
Chris Lattner5d6a05f2009-09-25 17:23:43 +0000337 return StringRef::npos;
Mikhail Glushenkov7112c862010-08-20 17:38:38 +0000338
Chris Lattner5d6a05f2009-09-25 17:23:43 +0000339 // Successful regex match.
340 assert(!MatchInfo.empty() && "Didn't get any match");
341 StringRef FullMatch = MatchInfo[0];
Mikhail Glushenkov7112c862010-08-20 17:38:38 +0000342
Chris Lattnereec96952009-09-27 07:56:52 +0000343 // If this defines any variables, remember their values.
344 for (unsigned i = 0, e = VariableDefs.size(); i != e; ++i) {
345 assert(VariableDefs[i].second < MatchInfo.size() &&
346 "Internal paren error");
347 VariableTable[VariableDefs[i].first] = MatchInfo[VariableDefs[i].second];
Chris Lattner94638f02009-09-25 17:29:36 +0000348 }
Mikhail Glushenkov7112c862010-08-20 17:38:38 +0000349
Chris Lattner5d6a05f2009-09-25 17:23:43 +0000350 MatchLen = FullMatch.size();
351 return FullMatch.data()-Buffer.data();
Chris Lattner52870082009-09-24 21:47:32 +0000352}
353
Daniel Dunbaread2dac2009-11-22 22:59:26 +0000354unsigned Pattern::ComputeMatchDistance(StringRef Buffer,
355 const StringMap<StringRef> &VariableTable) const {
356 // Just compute the number of matching characters. For regular expressions, we
357 // just compare against the regex itself and hope for the best.
358 //
359 // FIXME: One easy improvement here is have the regex lib generate a single
360 // example regular expression which matches, and use that as the example
361 // string.
362 StringRef ExampleString(FixedStr);
363 if (ExampleString.empty())
364 ExampleString = RegExStr;
365
Daniel Dunbar0806f9f2010-01-30 00:24:06 +0000366 // Only compare up to the first line in the buffer, or the string size.
367 StringRef BufferPrefix = Buffer.substr(0, ExampleString.size());
368 BufferPrefix = BufferPrefix.split('\n').first;
369 return BufferPrefix.edit_distance(ExampleString);
Daniel Dunbaread2dac2009-11-22 22:59:26 +0000370}
371
Daniel Dunbarfafe93c2009-11-22 22:08:06 +0000372void Pattern::PrintFailureInfo(const SourceMgr &SM, StringRef Buffer,
373 const StringMap<StringRef> &VariableTable) const{
Daniel Dunbarfafe93c2009-11-22 22:08:06 +0000374 // If this was a regular expression using variables, print the current
375 // variable values.
376 if (!VariableUses.empty()) {
377 for (unsigned i = 0, e = VariableUses.size(); i != e; ++i) {
378 StringRef Var = VariableUses[i].first;
379 StringMap<StringRef>::const_iterator it = VariableTable.find(Var);
380 SmallString<256> Msg;
381 raw_svector_ostream OS(Msg);
382
383 // Check for undefined variable references.
384 if (it == VariableTable.end()) {
385 OS << "uses undefined variable \"";
386 OS.write_escaped(Var) << "\"";;
387 } else {
388 OS << "with variable \"";
389 OS.write_escaped(Var) << "\" equal to \"";
390 OS.write_escaped(it->second) << "\"";
391 }
392
Chris Lattner3f2d5f62011-10-16 05:43:57 +0000393 SM.PrintMessage(SMLoc::getFromPointer(Buffer.data()), SourceMgr::DK_Note,
394 OS.str());
Daniel Dunbarfafe93c2009-11-22 22:08:06 +0000395 }
396 }
Daniel Dunbaread2dac2009-11-22 22:59:26 +0000397
398 // Attempt to find the closest/best fuzzy match. Usually an error happens
399 // because some string in the output didn't exactly match. In these cases, we
400 // would like to show the user a best guess at what "should have" matched, to
401 // save them having to actually check the input manually.
402 size_t NumLinesForward = 0;
403 size_t Best = StringRef::npos;
404 double BestQuality = 0;
405
406 // Use an arbitrary 4k limit on how far we will search.
Dan Gohmane3a1e502010-01-29 21:57:46 +0000407 for (size_t i = 0, e = std::min(size_t(4096), Buffer.size()); i != e; ++i) {
Daniel Dunbaread2dac2009-11-22 22:59:26 +0000408 if (Buffer[i] == '\n')
409 ++NumLinesForward;
410
Dan Gohmand8a55412010-01-29 21:55:16 +0000411 // Patterns have leading whitespace stripped, so skip whitespace when
412 // looking for something which looks like a pattern.
413 if (Buffer[i] == ' ' || Buffer[i] == '\t')
414 continue;
415
Daniel Dunbaread2dac2009-11-22 22:59:26 +0000416 // Compute the "quality" of this match as an arbitrary combination of the
417 // match distance and the number of lines skipped to get to this match.
418 unsigned Distance = ComputeMatchDistance(Buffer.substr(i), VariableTable);
419 double Quality = Distance + (NumLinesForward / 100.);
420
421 if (Quality < BestQuality || Best == StringRef::npos) {
422 Best = i;
423 BestQuality = Quality;
424 }
425 }
426
Daniel Dunbar7a68e0d2010-03-19 18:07:43 +0000427 // Print the "possible intended match here" line if we found something
428 // reasonable and not equal to what we showed in the "scanning from here"
429 // line.
430 if (Best && Best != StringRef::npos && BestQuality < 50) {
431 SM.PrintMessage(SMLoc::getFromPointer(Buffer.data() + Best),
Chris Lattner3f2d5f62011-10-16 05:43:57 +0000432 SourceMgr::DK_Note, "possible intended match here");
Daniel Dunbaread2dac2009-11-22 22:59:26 +0000433
434 // FIXME: If we wanted to be really friendly we would show why the match
435 // failed, as it can be hard to spot simple one character differences.
436 }
Daniel Dunbarfafe93c2009-11-22 22:08:06 +0000437}
Chris Lattnera29703e2009-09-24 20:39:13 +0000438
439//===----------------------------------------------------------------------===//
440// Check Strings.
441//===----------------------------------------------------------------------===//
Chris Lattner9fc66782009-09-24 20:25:55 +0000442
443/// CheckString - This is a check that we found in the input file.
444struct CheckString {
445 /// Pat - The pattern to match.
446 Pattern Pat;
Mikhail Glushenkov7112c862010-08-20 17:38:38 +0000447
Chris Lattner207e1bc2009-08-15 17:41:04 +0000448 /// Loc - The location in the match file that the check string was specified.
449 SMLoc Loc;
Mikhail Glushenkov7112c862010-08-20 17:38:38 +0000450
Chris Lattner5dafafd2009-08-15 18:32:21 +0000451 /// IsCheckNext - This is true if this is a CHECK-NEXT: directive (as opposed
452 /// to a CHECK: directive.
453 bool IsCheckNext;
Mikhail Glushenkov7112c862010-08-20 17:38:38 +0000454
Chris Lattnerf15380b2009-09-20 22:35:26 +0000455 /// NotStrings - These are all of the strings that are disallowed from
456 /// occurring between this match string and the previous one (or start of
457 /// file).
Chris Lattnera29703e2009-09-24 20:39:13 +0000458 std::vector<std::pair<SMLoc, Pattern> > NotStrings;
Mikhail Glushenkov7112c862010-08-20 17:38:38 +0000459
Chris Lattner9fc66782009-09-24 20:25:55 +0000460 CheckString(const Pattern &P, SMLoc L, bool isCheckNext)
461 : Pat(P), Loc(L), IsCheckNext(isCheckNext) {}
Chris Lattner207e1bc2009-08-15 17:41:04 +0000462};
463
Chris Lattneradea46e2009-09-24 20:45:07 +0000464/// CanonicalizeInputFile - Remove duplicate horizontal space from the specified
465/// memory buffer, free it, and return a new one.
466static MemoryBuffer *CanonicalizeInputFile(MemoryBuffer *MB) {
Chris Lattner4c842dd2010-04-05 22:42:30 +0000467 SmallString<128> NewFile;
Chris Lattneradea46e2009-09-24 20:45:07 +0000468 NewFile.reserve(MB->getBufferSize());
Mikhail Glushenkov7112c862010-08-20 17:38:38 +0000469
Chris Lattneradea46e2009-09-24 20:45:07 +0000470 for (const char *Ptr = MB->getBufferStart(), *End = MB->getBufferEnd();
471 Ptr != End; ++Ptr) {
NAKAMURA Takumi9f6e03f2010-11-14 03:28:22 +0000472 // Eliminate trailing dosish \r.
473 if (Ptr <= End - 2 && Ptr[0] == '\r' && Ptr[1] == '\n') {
474 continue;
475 }
476
Chris Lattneradea46e2009-09-24 20:45:07 +0000477 // If C is not a horizontal whitespace, skip it.
478 if (*Ptr != ' ' && *Ptr != '\t') {
479 NewFile.push_back(*Ptr);
480 continue;
481 }
Mikhail Glushenkov7112c862010-08-20 17:38:38 +0000482
Chris Lattneradea46e2009-09-24 20:45:07 +0000483 // Otherwise, add one space and advance over neighboring space.
484 NewFile.push_back(' ');
485 while (Ptr+1 != End &&
486 (Ptr[1] == ' ' || Ptr[1] == '\t'))
487 ++Ptr;
488 }
Mikhail Glushenkov7112c862010-08-20 17:38:38 +0000489
Chris Lattneradea46e2009-09-24 20:45:07 +0000490 // Free the old buffer and return a new one.
491 MemoryBuffer *MB2 =
Chris Lattner4c842dd2010-04-05 22:42:30 +0000492 MemoryBuffer::getMemBufferCopy(NewFile.str(), MB->getBufferIdentifier());
Mikhail Glushenkov7112c862010-08-20 17:38:38 +0000493
Chris Lattneradea46e2009-09-24 20:45:07 +0000494 delete MB;
495 return MB2;
496}
497
Chris Lattner81cb8ca2009-07-08 18:44:05 +0000498
Chris Lattner81cb8ca2009-07-08 18:44:05 +0000499/// ReadCheckFile - Read the check file, which specifies the sequence of
500/// expected strings. The strings are added to the CheckStrings vector.
501static bool ReadCheckFile(SourceMgr &SM,
Chris Lattner207e1bc2009-08-15 17:41:04 +0000502 std::vector<CheckString> &CheckStrings) {
Chris Lattner81cb8ca2009-07-08 18:44:05 +0000503 // Open the check file, and tell SourceMgr about it.
Michael J. Spencer3ff95632010-12-16 03:29:14 +0000504 OwningPtr<MemoryBuffer> File;
505 if (error_code ec =
506 MemoryBuffer::getFileOrSTDIN(CheckFilename.c_str(), File)) {
Mikhail Glushenkov7112c862010-08-20 17:38:38 +0000507 errs() << "Could not open check file '" << CheckFilename << "': "
Michael J. Spencer333fb042010-12-09 17:36:48 +0000508 << ec.message() << '\n';
Chris Lattner81cb8ca2009-07-08 18:44:05 +0000509 return true;
510 }
Michael J. Spencer3ff95632010-12-16 03:29:14 +0000511 MemoryBuffer *F = File.take();
Mikhail Glushenkov7112c862010-08-20 17:38:38 +0000512
Chris Lattneradea46e2009-09-24 20:45:07 +0000513 // If we want to canonicalize whitespace, strip excess whitespace from the
514 // buffer containing the CHECK lines.
515 if (!NoCanonicalizeWhiteSpace)
516 F = CanonicalizeInputFile(F);
Mikhail Glushenkov7112c862010-08-20 17:38:38 +0000517
Chris Lattner81cb8ca2009-07-08 18:44:05 +0000518 SM.AddNewSourceBuffer(F, SMLoc());
519
Chris Lattnerd7e25052009-08-15 18:00:42 +0000520 // Find all instances of CheckPrefix followed by : in the file.
Chris Lattner96077032009-09-20 22:11:44 +0000521 StringRef Buffer = F->getBuffer();
Chris Lattner81cb8ca2009-07-08 18:44:05 +0000522
Chris Lattnera29703e2009-09-24 20:39:13 +0000523 std::vector<std::pair<SMLoc, Pattern> > NotMatches;
Mikhail Glushenkov7112c862010-08-20 17:38:38 +0000524
Chris Lattner81cb8ca2009-07-08 18:44:05 +0000525 while (1) {
526 // See if Prefix occurs in the memory buffer.
Chris Lattner96077032009-09-20 22:11:44 +0000527 Buffer = Buffer.substr(Buffer.find(CheckPrefix));
Mikhail Glushenkov7112c862010-08-20 17:38:38 +0000528
Chris Lattner81cb8ca2009-07-08 18:44:05 +0000529 // If we didn't find a match, we're done.
Chris Lattner96077032009-09-20 22:11:44 +0000530 if (Buffer.empty())
Chris Lattner81cb8ca2009-07-08 18:44:05 +0000531 break;
Mikhail Glushenkov7112c862010-08-20 17:38:38 +0000532
Chris Lattner96077032009-09-20 22:11:44 +0000533 const char *CheckPrefixStart = Buffer.data();
Mikhail Glushenkov7112c862010-08-20 17:38:38 +0000534
Chris Lattner5dafafd2009-08-15 18:32:21 +0000535 // When we find a check prefix, keep track of whether we find CHECK: or
536 // CHECK-NEXT:
Chris Lattnerf15380b2009-09-20 22:35:26 +0000537 bool IsCheckNext = false, IsCheckNot = false;
Mikhail Glushenkov7112c862010-08-20 17:38:38 +0000538
Chris Lattnerd7e25052009-08-15 18:00:42 +0000539 // Verify that the : is present after the prefix.
Chris Lattner96077032009-09-20 22:11:44 +0000540 if (Buffer[CheckPrefix.size()] == ':') {
541 Buffer = Buffer.substr(CheckPrefix.size()+1);
Chris Lattner96077032009-09-20 22:11:44 +0000542 } else if (Buffer.size() > CheckPrefix.size()+6 &&
543 memcmp(Buffer.data()+CheckPrefix.size(), "-NEXT:", 6) == 0) {
544 Buffer = Buffer.substr(CheckPrefix.size()+7);
Chris Lattner5dafafd2009-08-15 18:32:21 +0000545 IsCheckNext = true;
Chris Lattnerf15380b2009-09-20 22:35:26 +0000546 } else if (Buffer.size() > CheckPrefix.size()+5 &&
547 memcmp(Buffer.data()+CheckPrefix.size(), "-NOT:", 5) == 0) {
548 Buffer = Buffer.substr(CheckPrefix.size()+6);
549 IsCheckNot = true;
Chris Lattner5dafafd2009-08-15 18:32:21 +0000550 } else {
Chris Lattner96077032009-09-20 22:11:44 +0000551 Buffer = Buffer.substr(1);
Chris Lattnerd7e25052009-08-15 18:00:42 +0000552 continue;
553 }
Mikhail Glushenkov7112c862010-08-20 17:38:38 +0000554
Chris Lattner81cb8ca2009-07-08 18:44:05 +0000555 // Okay, we found the prefix, yay. Remember the rest of the line, but
556 // ignore leading and trailing whitespace.
Chris Lattnerf15380b2009-09-20 22:35:26 +0000557 Buffer = Buffer.substr(Buffer.find_first_not_of(" \t"));
Mikhail Glushenkov7112c862010-08-20 17:38:38 +0000558
Chris Lattner81cb8ca2009-07-08 18:44:05 +0000559 // Scan ahead to the end of line.
Chris Lattner96077032009-09-20 22:11:44 +0000560 size_t EOL = Buffer.find_first_of("\n\r");
Chris Lattnera29703e2009-09-24 20:39:13 +0000561
Dan Gohmane5463432010-01-29 21:53:18 +0000562 // Remember the location of the start of the pattern, for diagnostics.
563 SMLoc PatternLoc = SMLoc::getFromPointer(Buffer.data());
564
Chris Lattnera29703e2009-09-24 20:39:13 +0000565 // Parse the pattern.
566 Pattern P;
567 if (P.ParsePattern(Buffer.substr(0, EOL), SM))
Chris Lattner81cb8ca2009-07-08 18:44:05 +0000568 return true;
Mikhail Glushenkov7112c862010-08-20 17:38:38 +0000569
Chris Lattnera29703e2009-09-24 20:39:13 +0000570 Buffer = Buffer.substr(EOL);
571
Mikhail Glushenkov7112c862010-08-20 17:38:38 +0000572
Chris Lattner5dafafd2009-08-15 18:32:21 +0000573 // Verify that CHECK-NEXT lines have at least one CHECK line before them.
574 if (IsCheckNext && CheckStrings.empty()) {
575 SM.PrintMessage(SMLoc::getFromPointer(CheckPrefixStart),
Chris Lattner3f2d5f62011-10-16 05:43:57 +0000576 SourceMgr::DK_Error,
Chris Lattner5dafafd2009-08-15 18:32:21 +0000577 "found '"+CheckPrefix+"-NEXT:' without previous '"+
Chris Lattner3f2d5f62011-10-16 05:43:57 +0000578 CheckPrefix+ ": line");
Chris Lattner5dafafd2009-08-15 18:32:21 +0000579 return true;
580 }
Mikhail Glushenkov7112c862010-08-20 17:38:38 +0000581
Chris Lattnera29703e2009-09-24 20:39:13 +0000582 // Handle CHECK-NOT.
583 if (IsCheckNot) {
584 NotMatches.push_back(std::make_pair(SMLoc::getFromPointer(Buffer.data()),
585 P));
586 continue;
587 }
Mikhail Glushenkov7112c862010-08-20 17:38:38 +0000588
589
Chris Lattner81cb8ca2009-07-08 18:44:05 +0000590 // Okay, add the string we captured to the output vector and move on.
Chris Lattner9fc66782009-09-24 20:25:55 +0000591 CheckStrings.push_back(CheckString(P,
Dan Gohmane5463432010-01-29 21:53:18 +0000592 PatternLoc,
Chris Lattner5dafafd2009-08-15 18:32:21 +0000593 IsCheckNext));
Chris Lattnerf15380b2009-09-20 22:35:26 +0000594 std::swap(NotMatches, CheckStrings.back().NotStrings);
Chris Lattner81cb8ca2009-07-08 18:44:05 +0000595 }
Mikhail Glushenkov7112c862010-08-20 17:38:38 +0000596
Jakob Stoklund Olesen824c10e2010-10-15 17:47:12 +0000597 // Add an EOF pattern for any trailing CHECK-NOTs.
598 if (!NotMatches.empty()) {
599 CheckStrings.push_back(CheckString(Pattern(true),
600 SMLoc::getFromPointer(Buffer.data()),
601 false));
602 std::swap(NotMatches, CheckStrings.back().NotStrings);
603 }
604
Chris Lattner81cb8ca2009-07-08 18:44:05 +0000605 if (CheckStrings.empty()) {
Chris Lattnerd7e25052009-08-15 18:00:42 +0000606 errs() << "error: no check strings found with prefix '" << CheckPrefix
607 << ":'\n";
Chris Lattner81cb8ca2009-07-08 18:44:05 +0000608 return true;
609 }
Mikhail Glushenkov7112c862010-08-20 17:38:38 +0000610
Chris Lattner81cb8ca2009-07-08 18:44:05 +0000611 return false;
612}
613
Chris Lattner5dafafd2009-08-15 18:32:21 +0000614static void PrintCheckFailed(const SourceMgr &SM, const CheckString &CheckStr,
Daniel Dunbarfafe93c2009-11-22 22:08:06 +0000615 StringRef Buffer,
616 StringMap<StringRef> &VariableTable) {
Chris Lattner5dafafd2009-08-15 18:32:21 +0000617 // Otherwise, we have an error, emit an error message.
Chris Lattner3f2d5f62011-10-16 05:43:57 +0000618 SM.PrintMessage(CheckStr.Loc, SourceMgr::DK_Error,
619 "expected string not found in input");
Mikhail Glushenkov7112c862010-08-20 17:38:38 +0000620
Chris Lattner5dafafd2009-08-15 18:32:21 +0000621 // Print the "scanning from here" line. If the current position is at the
622 // end of a line, advance to the start of the next line.
Chris Lattner96077032009-09-20 22:11:44 +0000623 Buffer = Buffer.substr(Buffer.find_first_not_of(" \t\n\r"));
Mikhail Glushenkov7112c862010-08-20 17:38:38 +0000624
Chris Lattner3f2d5f62011-10-16 05:43:57 +0000625 SM.PrintMessage(SMLoc::getFromPointer(Buffer.data()), SourceMgr::DK_Note,
626 "scanning from here");
Daniel Dunbarfafe93c2009-11-22 22:08:06 +0000627
628 // Allow the pattern to print additional information if desired.
629 CheckStr.Pat.PrintFailureInfo(SM, Buffer, VariableTable);
Chris Lattner5dafafd2009-08-15 18:32:21 +0000630}
631
Chris Lattner3711b7a2009-09-20 22:42:44 +0000632/// CountNumNewlinesBetween - Count the number of newlines in the specified
633/// range.
634static unsigned CountNumNewlinesBetween(StringRef Range) {
Chris Lattner5dafafd2009-08-15 18:32:21 +0000635 unsigned NumNewLines = 0;
Chris Lattner3711b7a2009-09-20 22:42:44 +0000636 while (1) {
Chris Lattner5dafafd2009-08-15 18:32:21 +0000637 // Scan for newline.
Chris Lattner3711b7a2009-09-20 22:42:44 +0000638 Range = Range.substr(Range.find_first_of("\n\r"));
639 if (Range.empty()) return NumNewLines;
Mikhail Glushenkov7112c862010-08-20 17:38:38 +0000640
Chris Lattner5dafafd2009-08-15 18:32:21 +0000641 ++NumNewLines;
Mikhail Glushenkov7112c862010-08-20 17:38:38 +0000642
Chris Lattner5dafafd2009-08-15 18:32:21 +0000643 // Handle \n\r and \r\n as a single newline.
Chris Lattner3711b7a2009-09-20 22:42:44 +0000644 if (Range.size() > 1 &&
645 (Range[1] == '\n' || Range[1] == '\r') &&
646 (Range[0] != Range[1]))
647 Range = Range.substr(1);
648 Range = Range.substr(1);
Chris Lattner5dafafd2009-08-15 18:32:21 +0000649 }
Chris Lattner5dafafd2009-08-15 18:32:21 +0000650}
651
Chris Lattner81cb8ca2009-07-08 18:44:05 +0000652int main(int argc, char **argv) {
653 sys::PrintStackTraceOnErrorSignal();
654 PrettyStackTraceProgram X(argc, argv);
655 cl::ParseCommandLineOptions(argc, argv);
656
657 SourceMgr SM;
Mikhail Glushenkov7112c862010-08-20 17:38:38 +0000658
Chris Lattner81cb8ca2009-07-08 18:44:05 +0000659 // Read the expected strings from the check file.
Chris Lattner207e1bc2009-08-15 17:41:04 +0000660 std::vector<CheckString> CheckStrings;
Chris Lattner81cb8ca2009-07-08 18:44:05 +0000661 if (ReadCheckFile(SM, CheckStrings))
662 return 2;
663
664 // Open the file to check and add it to SourceMgr.
Michael J. Spencer3ff95632010-12-16 03:29:14 +0000665 OwningPtr<MemoryBuffer> File;
666 if (error_code ec =
667 MemoryBuffer::getFileOrSTDIN(InputFilename.c_str(), File)) {
Mikhail Glushenkov7112c862010-08-20 17:38:38 +0000668 errs() << "Could not open input file '" << InputFilename << "': "
Michael J. Spencer333fb042010-12-09 17:36:48 +0000669 << ec.message() << '\n';
Chris Lattner81cb8ca2009-07-08 18:44:05 +0000670 return true;
671 }
Michael J. Spencer3ff95632010-12-16 03:29:14 +0000672 MemoryBuffer *F = File.take();
Mikhail Glushenkov7112c862010-08-20 17:38:38 +0000673
Chris Lattner1aac1862011-02-09 16:46:02 +0000674 if (F->getBufferSize() == 0) {
675 errs() << "FileCheck error: '" << InputFilename << "' is empty.\n";
676 return 1;
677 }
678
Chris Lattner88a7e9e2009-07-11 18:58:15 +0000679 // Remove duplicate spaces in the input file if requested.
680 if (!NoCanonicalizeWhiteSpace)
681 F = CanonicalizeInputFile(F);
Mikhail Glushenkov7112c862010-08-20 17:38:38 +0000682
Chris Lattner81cb8ca2009-07-08 18:44:05 +0000683 SM.AddNewSourceBuffer(F, SMLoc());
Mikhail Glushenkov7112c862010-08-20 17:38:38 +0000684
Chris Lattnereec96952009-09-27 07:56:52 +0000685 /// VariableTable - This holds all the current filecheck variables.
686 StringMap<StringRef> VariableTable;
Mikhail Glushenkov7112c862010-08-20 17:38:38 +0000687
Chris Lattner81cb8ca2009-07-08 18:44:05 +0000688 // Check that we have all of the expected strings, in order, in the input
689 // file.
Chris Lattner96077032009-09-20 22:11:44 +0000690 StringRef Buffer = F->getBuffer();
Mikhail Glushenkov7112c862010-08-20 17:38:38 +0000691
Chris Lattnerf15380b2009-09-20 22:35:26 +0000692 const char *LastMatch = Buffer.data();
Mikhail Glushenkov7112c862010-08-20 17:38:38 +0000693
Chris Lattner81cb8ca2009-07-08 18:44:05 +0000694 for (unsigned StrNo = 0, e = CheckStrings.size(); StrNo != e; ++StrNo) {
Chris Lattner207e1bc2009-08-15 17:41:04 +0000695 const CheckString &CheckStr = CheckStrings[StrNo];
Mikhail Glushenkov7112c862010-08-20 17:38:38 +0000696
Chris Lattner96077032009-09-20 22:11:44 +0000697 StringRef SearchFrom = Buffer;
Mikhail Glushenkov7112c862010-08-20 17:38:38 +0000698
Chris Lattner81cb8ca2009-07-08 18:44:05 +0000699 // Find StrNo in the file.
Chris Lattner9fc66782009-09-24 20:25:55 +0000700 size_t MatchLen = 0;
Jakob Stoklund Olesen824c10e2010-10-15 17:47:12 +0000701 size_t MatchPos = CheckStr.Pat.Match(Buffer, MatchLen, VariableTable);
702 Buffer = Buffer.substr(MatchPos);
Mikhail Glushenkov7112c862010-08-20 17:38:38 +0000703
Chris Lattner5dafafd2009-08-15 18:32:21 +0000704 // If we didn't find a match, reject the input.
Jakob Stoklund Olesen824c10e2010-10-15 17:47:12 +0000705 if (MatchPos == StringRef::npos) {
Daniel Dunbarfafe93c2009-11-22 22:08:06 +0000706 PrintCheckFailed(SM, CheckStr, SearchFrom, VariableTable);
Chris Lattner5dafafd2009-08-15 18:32:21 +0000707 return 1;
Chris Lattner81cb8ca2009-07-08 18:44:05 +0000708 }
Chris Lattner3711b7a2009-09-20 22:42:44 +0000709
710 StringRef SkippedRegion(LastMatch, Buffer.data()-LastMatch);
711
Chris Lattner5dafafd2009-08-15 18:32:21 +0000712 // If this check is a "CHECK-NEXT", verify that the previous match was on
713 // the previous line (i.e. that there is one newline between them).
714 if (CheckStr.IsCheckNext) {
715 // Count the number of newlines between the previous match and this one.
Chris Lattnerf15380b2009-09-20 22:35:26 +0000716 assert(LastMatch != F->getBufferStart() &&
717 "CHECK-NEXT can't be the first check in a file");
Chris Lattner5dafafd2009-08-15 18:32:21 +0000718
Chris Lattner3711b7a2009-09-20 22:42:44 +0000719 unsigned NumNewLines = CountNumNewlinesBetween(SkippedRegion);
Chris Lattner5dafafd2009-08-15 18:32:21 +0000720 if (NumNewLines == 0) {
Chris Lattner3f2d5f62011-10-16 05:43:57 +0000721 SM.PrintMessage(CheckStr.Loc, SourceMgr::DK_Error,
722 CheckPrefix+"-NEXT: is on the same line as previous match");
Chris Lattner96077032009-09-20 22:11:44 +0000723 SM.PrintMessage(SMLoc::getFromPointer(Buffer.data()),
Chris Lattner3f2d5f62011-10-16 05:43:57 +0000724 SourceMgr::DK_Note, "'next' match was here");
725 SM.PrintMessage(SMLoc::getFromPointer(LastMatch), SourceMgr::DK_Note,
726 "previous match was here");
Chris Lattner5dafafd2009-08-15 18:32:21 +0000727 return 1;
728 }
Mikhail Glushenkov7112c862010-08-20 17:38:38 +0000729
Chris Lattner5dafafd2009-08-15 18:32:21 +0000730 if (NumNewLines != 1) {
Chris Lattner3f2d5f62011-10-16 05:43:57 +0000731 SM.PrintMessage(CheckStr.Loc, SourceMgr::DK_Error, CheckPrefix+
732 "-NEXT: is not on the line after the previous match");
Chris Lattner96077032009-09-20 22:11:44 +0000733 SM.PrintMessage(SMLoc::getFromPointer(Buffer.data()),
Chris Lattner3f2d5f62011-10-16 05:43:57 +0000734 SourceMgr::DK_Note, "'next' match was here");
735 SM.PrintMessage(SMLoc::getFromPointer(LastMatch), SourceMgr::DK_Note,
736 "previous match was here");
Chris Lattner5dafafd2009-08-15 18:32:21 +0000737 return 1;
738 }
739 }
Mikhail Glushenkov7112c862010-08-20 17:38:38 +0000740
Chris Lattnerf15380b2009-09-20 22:35:26 +0000741 // If this match had "not strings", verify that they don't exist in the
742 // skipped region.
Chris Lattnereec96952009-09-27 07:56:52 +0000743 for (unsigned ChunkNo = 0, e = CheckStr.NotStrings.size();
744 ChunkNo != e; ++ChunkNo) {
Chris Lattnera29703e2009-09-24 20:39:13 +0000745 size_t MatchLen = 0;
Chris Lattnereec96952009-09-27 07:56:52 +0000746 size_t Pos = CheckStr.NotStrings[ChunkNo].second.Match(SkippedRegion,
747 MatchLen,
748 VariableTable);
Chris Lattnerf15380b2009-09-20 22:35:26 +0000749 if (Pos == StringRef::npos) continue;
Mikhail Glushenkov7112c862010-08-20 17:38:38 +0000750
Chris Lattner3f2d5f62011-10-16 05:43:57 +0000751 SM.PrintMessage(SMLoc::getFromPointer(LastMatch+Pos), SourceMgr::DK_Error,
752 CheckPrefix+"-NOT: string occurred!");
753 SM.PrintMessage(CheckStr.NotStrings[ChunkNo].first, SourceMgr::DK_Note,
754 CheckPrefix+"-NOT: pattern specified here");
Chris Lattnerf15380b2009-09-20 22:35:26 +0000755 return 1;
756 }
Mikhail Glushenkov7112c862010-08-20 17:38:38 +0000757
Chris Lattner5dafafd2009-08-15 18:32:21 +0000758
Chris Lattner81115762009-09-21 02:30:42 +0000759 // Otherwise, everything is good. Step over the matched text and remember
760 // the position after the match as the end of the last match.
Chris Lattner9fc66782009-09-24 20:25:55 +0000761 Buffer = Buffer.substr(MatchLen);
Chris Lattner81115762009-09-21 02:30:42 +0000762 LastMatch = Buffer.data();
Chris Lattner81cb8ca2009-07-08 18:44:05 +0000763 }
Mikhail Glushenkov7112c862010-08-20 17:38:38 +0000764
Chris Lattner81cb8ca2009-07-08 18:44:05 +0000765 return 0;
766}