1 //===---- VerifyDiagnosticConsumer.cpp - Verifying Diagnostic Client ------===//
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 // This is a concrete diagnostic client, which buffers the diagnostic messages.
11 //
12 //===----------------------------------------------------------------------===//
13 
14 #include "clang/Basic/FileManager.h"
15 #include "clang/Frontend/VerifyDiagnosticConsumer.h"
16 #include "clang/Frontend/FrontendDiagnostic.h"
17 #include "clang/Frontend/TextDiagnosticBuffer.h"
18 #include "clang/Lex/HeaderSearch.h"
19 #include "clang/Lex/Preprocessor.h"
20 #include "llvm/ADT/SmallString.h"
21 #include "llvm/Support/Regex.h"
22 #include "llvm/Support/raw_ostream.h"
23 #include <cctype>
24 
25 using namespace clang;
26 typedef VerifyDiagnosticConsumer::Directive Directive;
27 typedef VerifyDiagnosticConsumer::DirectiveList DirectiveList;
28 typedef VerifyDiagnosticConsumer::ExpectedData ExpectedData;
29 
30 VerifyDiagnosticConsumer::VerifyDiagnosticConsumer(DiagnosticsEngine &_Diags)
31   : Diags(_Diags),
32     PrimaryClient(Diags.getClient()), OwnsPrimaryClient(Diags.ownsClient()),
33     Buffer(new TextDiagnosticBuffer()), CurrentPreprocessor(0),
34     LangOpts(0), SrcManager(0), ActiveSourceFiles(0)
35 {
36   Diags.takeClient();
37   if (Diags.hasSourceManager())
38     setSourceManager(Diags.getSourceManager());
39 }
40 
41 VerifyDiagnosticConsumer::~VerifyDiagnosticConsumer() {
42   assert(!ActiveSourceFiles && "Incomplete parsing of source files!");
43   assert(!CurrentPreprocessor && "CurrentPreprocessor should be invalid!");
44   SrcManager = 0;
45   CheckDiagnostics();
46   Diags.takeClient();
47   if (OwnsPrimaryClient)
48     delete PrimaryClient;
49 }
50 
51 #ifndef NDEBUG
52 namespace {
53 class VerifyFileTracker : public PPCallbacks {
54   VerifyDiagnosticConsumer &Verify;
55   SourceManager &SM;
56 
57 public:
58   VerifyFileTracker(VerifyDiagnosticConsumer &Verify, SourceManager &SM)
59     : Verify(Verify), SM(SM) { }
60 
61   /// \brief Hook into the preprocessor and update the list of parsed
62   /// files when the preprocessor indicates a new file is entered.
63   virtual void FileChanged(SourceLocation Loc, FileChangeReason Reason,
64                            SrcMgr::CharacteristicKind FileType,
65                            FileID PrevFID) {
66     Verify.UpdateParsedFileStatus(SM, SM.getFileID(Loc),
67                                   VerifyDiagnosticConsumer::IsParsed);
68   }
69 };
70 } // End anonymous namespace.
71 #endif
72 
73 // DiagnosticConsumer interface.
74 
75 void VerifyDiagnosticConsumer::BeginSourceFile(const LangOptions &LangOpts,
76                                                const Preprocessor *PP) {
77   // Attach comment handler on first invocation.
78   if (++ActiveSourceFiles == 1) {
79     if (PP) {
80       CurrentPreprocessor = PP;
81       this->LangOpts = &LangOpts;
82       setSourceManager(PP->getSourceManager());
83       const_cast<Preprocessor*>(PP)->addCommentHandler(this);
84 #ifndef NDEBUG
85       // Debug build tracks parsed files.
86       VerifyFileTracker *V = new VerifyFileTracker(*this, *SrcManager);
87       const_cast<Preprocessor*>(PP)->addPPCallbacks(V);
88 #endif
89     }
90   }
91 
92   assert((!PP || CurrentPreprocessor == PP) && "Preprocessor changed!");
93   PrimaryClient->BeginSourceFile(LangOpts, PP);
94 }
95 
96 void VerifyDiagnosticConsumer::EndSourceFile() {
97   assert(ActiveSourceFiles && "No active source files!");
98   PrimaryClient->EndSourceFile();
99 
100   // Detach comment handler once last active source file completed.
101   if (--ActiveSourceFiles == 0) {
102     if (CurrentPreprocessor)
103       const_cast<Preprocessor*>(CurrentPreprocessor)->removeCommentHandler(this);
104 
105     // Check diagnostics once last file completed.
106     CheckDiagnostics();
107     CurrentPreprocessor = 0;
108     LangOpts = 0;
109   }
110 }
111 
112 void VerifyDiagnosticConsumer::HandleDiagnostic(
113       DiagnosticsEngine::Level DiagLevel, const Diagnostic &Info) {
114   if (Info.hasSourceManager())
115     setSourceManager(Info.getSourceManager());
116 
117 #ifndef NDEBUG
118   // Debug build tracks unparsed files for possible
119   // unparsed expected-* directives.
120   if (SrcManager) {
121     SourceLocation Loc = Info.getLocation();
122     if (Loc.isValid()) {
123       ParsedStatus PS = IsUnparsed;
124 
125       Loc = SrcManager->getExpansionLoc(Loc);
126       FileID FID = SrcManager->getFileID(Loc);
127 
128       const FileEntry *FE = SrcManager->getFileEntryForID(FID);
129       if (FE && CurrentPreprocessor && SrcManager->isLoadedFileID(FID)) {
130         // If the file is a modules header file it shall not be parsed
131         // for expected-* directives.
132         HeaderSearch &HS = CurrentPreprocessor->getHeaderSearchInfo();
133         if (HS.findModuleForHeader(FE))
134           PS = IsUnparsedNoDirectives;
135       }
136 
137       UpdateParsedFileStatus(*SrcManager, FID, PS);
138     }
139   }
140 #endif
141 
142   // Send the diagnostic to the buffer, we will check it once we reach the end
143   // of the source file (or are destructed).
144   Buffer->HandleDiagnostic(DiagLevel, Info);
145 }
146 
147 //===----------------------------------------------------------------------===//
148 // Checking diagnostics implementation.
149 //===----------------------------------------------------------------------===//
150 
151 typedef TextDiagnosticBuffer::DiagList DiagList;
152 typedef TextDiagnosticBuffer::const_iterator const_diag_iterator;
153 
154 namespace {
155 
156 /// StandardDirective - Directive with string matching.
157 ///
158 class StandardDirective : public Directive {
159 public:
160   StandardDirective(SourceLocation DirectiveLoc, SourceLocation DiagnosticLoc,
161                     StringRef Text, unsigned Min, unsigned Max)
162     : Directive(DirectiveLoc, DiagnosticLoc, Text, Min, Max) { }
163 
164   virtual bool isValid(std::string &Error) {
165     // all strings are considered valid; even empty ones
166     return true;
167   }
168 
169   virtual bool match(StringRef S) {
170     return S.find(Text) != StringRef::npos;
171   }
172 };
173 
174 /// RegexDirective - Directive with regular-expression matching.
175 ///
176 class RegexDirective : public Directive {
177 public:
178   RegexDirective(SourceLocation DirectiveLoc, SourceLocation DiagnosticLoc,
179                  StringRef Text, unsigned Min, unsigned Max)
180     : Directive(DirectiveLoc, DiagnosticLoc, Text, Min, Max), Regex(Text) { }
181 
182   virtual bool isValid(std::string &Error) {
183     if (Regex.isValid(Error))
184       return true;
185     return false;
186   }
187 
188   virtual bool match(StringRef S) {
189     return Regex.match(S);
190   }
191 
192 private:
193   llvm::Regex Regex;
194 };
195 
196 class ParseHelper
197 {
198 public:
199   ParseHelper(StringRef S)
200     : Begin(S.begin()), End(S.end()), C(Begin), P(Begin), PEnd(NULL) { }
201 
202   // Return true if string literal is next.
203   bool Next(StringRef S) {
204     P = C;
205     PEnd = C + S.size();
206     if (PEnd > End)
207       return false;
208     return !memcmp(P, S.data(), S.size());
209   }
210 
211   // Return true if number is next.
212   // Output N only if number is next.
213   bool Next(unsigned &N) {
214     unsigned TMP = 0;
215     P = C;
216     for (; P < End && P[0] >= '0' && P[0] <= '9'; ++P) {
217       TMP *= 10;
218       TMP += P[0] - '0';
219     }
220     if (P == C)
221       return false;
222     PEnd = P;
223     N = TMP;
224     return true;
225   }
226 
227   // Return true if string literal is found.
228   // When true, P marks begin-position of S in content.
229   bool Search(StringRef S) {
230     P = std::search(C, End, S.begin(), S.end());
231     PEnd = P + S.size();
232     return P != End;
233   }
234 
235   // Advance 1-past previous next/search.
236   // Behavior is undefined if previous next/search failed.
237   bool Advance() {
238     C = PEnd;
239     return C < End;
240   }
241 
242   // Skip zero or more whitespace.
243   void SkipWhitespace() {
244     for (; C < End && isspace(*C); ++C)
245       ;
246   }
247 
248   // Return true if EOF reached.
249   bool Done() {
250     return !(C < End);
251   }
252 
253   const char * const Begin; // beginning of expected content
254   const char * const End;   // end of expected content (1-past)
255   const char *C;            // position of next char in content
256   const char *P;
257 
258 private:
259   const char *PEnd; // previous next/search subject end (1-past)
260 };
261 
262 } // namespace anonymous
263 
264 /// ParseDirective - Go through the comment and see if it indicates expected
265 /// diagnostics. If so, then put them in the appropriate directive list.
266 ///
267 /// Returns true if any valid directives were found.
268 static bool ParseDirective(StringRef S, ExpectedData *ED, SourceManager &SM,
269                            SourceLocation Pos, DiagnosticsEngine &Diags) {
270   // A single comment may contain multiple directives.
271   bool FoundDirective = false;
272   for (ParseHelper PH(S); !PH.Done();) {
273     // Search for token: expected
274     if (!PH.Search("expected"))
275       break;
276     PH.Advance();
277 
278     // Next token: -
279     if (!PH.Next("-"))
280       continue;
281     PH.Advance();
282 
283     // Next token: { error | warning | note }
284     DirectiveList* DL = NULL;
285     if (PH.Next("error"))
286       DL = ED ? &ED->Errors : NULL;
287     else if (PH.Next("warning"))
288       DL = ED ? &ED->Warnings : NULL;
289     else if (PH.Next("note"))
290       DL = ED ? &ED->Notes : NULL;
291     else
292       continue;
293     PH.Advance();
294 
295     // If a directive has been found but we're not interested
296     // in storing the directive information, return now.
297     if (!DL)
298       return true;
299 
300     // Default directive kind.
301     bool RegexKind = false;
302     const char* KindStr = "string";
303 
304     // Next optional token: -
305     if (PH.Next("-re")) {
306       PH.Advance();
307       RegexKind = true;
308       KindStr = "regex";
309     }
310 
311     // Next optional token: @
312     SourceLocation ExpectedLoc;
313     if (!PH.Next("@")) {
314       ExpectedLoc = Pos;
315     } else {
316       PH.Advance();
317       unsigned Line = 0;
318       bool FoundPlus = PH.Next("+");
319       if (FoundPlus || PH.Next("-")) {
320         // Relative to current line.
321         PH.Advance();
322         bool Invalid = false;
323         unsigned ExpectedLine = SM.getSpellingLineNumber(Pos, &Invalid);
324         if (!Invalid && PH.Next(Line) && (FoundPlus || Line < ExpectedLine)) {
325           if (FoundPlus) ExpectedLine += Line;
326           else ExpectedLine -= Line;
327           ExpectedLoc = SM.translateLineCol(SM.getFileID(Pos), ExpectedLine, 1);
328         }
329       } else {
330         // Absolute line number.
331         if (PH.Next(Line) && Line > 0)
332           ExpectedLoc = SM.translateLineCol(SM.getFileID(Pos), Line, 1);
333       }
334 
335       if (ExpectedLoc.isInvalid()) {
336         Diags.Report(Pos.getLocWithOffset(PH.C-PH.Begin),
337                      diag::err_verify_missing_line) << KindStr;
338         continue;
339       }
340       PH.Advance();
341     }
342 
343     // Skip optional whitespace.
344     PH.SkipWhitespace();
345 
346     // Next optional token: positive integer or a '+'.
347     unsigned Min = 1;
348     unsigned Max = 1;
349     if (PH.Next(Min)) {
350       PH.Advance();
351       // A positive integer can be followed by a '+' meaning min
352       // or more, or by a '-' meaning a range from min to max.
353       if (PH.Next("+")) {
354         Max = Directive::MaxCount;
355         PH.Advance();
356       } else if (PH.Next("-")) {
357         PH.Advance();
358         if (!PH.Next(Max) || Max < Min) {
359           Diags.Report(Pos.getLocWithOffset(PH.C-PH.Begin),
360                        diag::err_verify_invalid_range) << KindStr;
361           continue;
362         }
363         PH.Advance();
364       } else {
365         Max = Min;
366       }
367     } else if (PH.Next("+")) {
368       // '+' on its own means "1 or more".
369       Max = Directive::MaxCount;
370       PH.Advance();
371     }
372 
373     // Skip optional whitespace.
374     PH.SkipWhitespace();
375 
376     // Next token: {{
377     if (!PH.Next("{{")) {
378       Diags.Report(Pos.getLocWithOffset(PH.C-PH.Begin),
379                    diag::err_verify_missing_start) << KindStr;
380       continue;
381     }
382     PH.Advance();
383     const char* const ContentBegin = PH.C; // mark content begin
384 
385     // Search for token: }}
386     if (!PH.Search("}}")) {
387       Diags.Report(Pos.getLocWithOffset(PH.C-PH.Begin),
388                    diag::err_verify_missing_end) << KindStr;
389       continue;
390     }
391     const char* const ContentEnd = PH.P; // mark content end
392     PH.Advance();
393 
394     // Build directive text; convert \n to newlines.
395     std::string Text;
396     StringRef NewlineStr = "\\n";
397     StringRef Content(ContentBegin, ContentEnd-ContentBegin);
398     size_t CPos = 0;
399     size_t FPos;
400     while ((FPos = Content.find(NewlineStr, CPos)) != StringRef::npos) {
401       Text += Content.substr(CPos, FPos-CPos);
402       Text += '\n';
403       CPos = FPos + NewlineStr.size();
404     }
405     if (Text.empty())
406       Text.assign(ContentBegin, ContentEnd);
407 
408     // Construct new directive.
409     Directive *D = Directive::create(RegexKind, Pos, ExpectedLoc, Text,
410                                      Min, Max);
411     std::string Error;
412     if (D->isValid(Error)) {
413       DL->push_back(D);
414       FoundDirective = true;
415     } else {
416       Diags.Report(Pos.getLocWithOffset(ContentBegin-PH.Begin),
417                    diag::err_verify_invalid_content)
418         << KindStr << Error;
419     }
420   }
421 
422   return FoundDirective;
423 }
424 
425 /// HandleComment - Hook into the preprocessor and extract comments containing
426 ///  expected errors and warnings.
427 bool VerifyDiagnosticConsumer::HandleComment(Preprocessor &PP,
428                                              SourceRange Comment) {
429   SourceManager &SM = PP.getSourceManager();
430   SourceLocation CommentBegin = Comment.getBegin();
431 
432   const char *CommentRaw = SM.getCharacterData(CommentBegin);
433   StringRef C(CommentRaw, SM.getCharacterData(Comment.getEnd()) - CommentRaw);
434 
435   if (C.empty())
436     return false;
437 
438   // Fold any "\<EOL>" sequences
439   size_t loc = C.find('\\');
440   if (loc == StringRef::npos) {
441     ParseDirective(C, &ED, SM, CommentBegin, PP.getDiagnostics());
442     return false;
443   }
444 
445   std::string C2;
446   C2.reserve(C.size());
447 
448   for (size_t last = 0;; loc = C.find('\\', last)) {
449     if (loc == StringRef::npos || loc == C.size()) {
450       C2 += C.substr(last);
451       break;
452     }
453     C2 += C.substr(last, loc-last);
454     last = loc + 1;
455 
456     if (C[last] == '\n' || C[last] == '\r') {
457       ++last;
458 
459       // Escape \r\n  or \n\r, but not \n\n.
460       if (last < C.size())
461         if (C[last] == '\n' || C[last] == '\r')
462           if (C[last] != C[last-1])
463             ++last;
464     } else {
465       // This was just a normal backslash.
466       C2 += '\\';
467     }
468   }
469 
470   if (!C2.empty())
471     ParseDirective(C2, &ED, SM, CommentBegin, PP.getDiagnostics());
472   return false;
473 }
474 
475 #ifndef NDEBUG
476 /// \brief Lex the specified source file to determine whether it contains
477 /// any expected-* directives.  As a Lexer is used rather than a full-blown
478 /// Preprocessor, directives inside skipped #if blocks will still be found.
479 ///
480 /// \return true if any directives were found.
481 static bool findDirectives(SourceManager &SM, FileID FID,
482                            const LangOptions &LangOpts) {
483   // Create a raw lexer to pull all the comments out of FID.
484   if (FID.isInvalid())
485     return false;
486 
487   // Create a lexer to lex all the tokens of the main file in raw mode.
488   const llvm::MemoryBuffer *FromFile = SM.getBuffer(FID);
489   Lexer RawLex(FID, FromFile, SM, LangOpts);
490 
491   // Return comments as tokens, this is how we find expected diagnostics.
492   RawLex.SetCommentRetentionState(true);
493 
494   Token Tok;
495   Tok.setKind(tok::comment);
496   while (Tok.isNot(tok::eof)) {
497     RawLex.Lex(Tok);
498     if (!Tok.is(tok::comment)) continue;
499 
500     std::string Comment = RawLex.getSpelling(Tok, SM, LangOpts);
501     if (Comment.empty()) continue;
502 
503     // Find first directive.
504     if (ParseDirective(Comment, 0, SM, Tok.getLocation(),
505                        SM.getDiagnostics()))
506       return true;
507   }
508   return false;
509 }
510 #endif // !NDEBUG
511 
512 /// \brief Takes a list of diagnostics that have been generated but not matched
513 /// by an expected-* directive and produces a diagnostic to the user from this.
514 static unsigned PrintUnexpected(DiagnosticsEngine &Diags, SourceManager *SourceMgr,
515                                 const_diag_iterator diag_begin,
516                                 const_diag_iterator diag_end,
517                                 const char *Kind) {
518   if (diag_begin == diag_end) return 0;
519 
520   SmallString<256> Fmt;
521   llvm::raw_svector_ostream OS(Fmt);
522   for (const_diag_iterator I = diag_begin, E = diag_end; I != E; ++I) {
523     if (I->first.isInvalid() || !SourceMgr)
524       OS << "\n  (frontend)";
525     else
526       OS << "\n  Line " << SourceMgr->getPresumedLineNumber(I->first);
527     OS << ": " << I->second;
528   }
529 
530   Diags.Report(diag::err_verify_inconsistent_diags).setForceEmit()
531     << Kind << /*Unexpected=*/true << OS.str();
532   return std::distance(diag_begin, diag_end);
533 }
534 
535 /// \brief Takes a list of diagnostics that were expected to have been generated
536 /// but were not and produces a diagnostic to the user from this.
537 static unsigned PrintExpected(DiagnosticsEngine &Diags, SourceManager &SourceMgr,
538                               DirectiveList &DL, const char *Kind) {
539   if (DL.empty())
540     return 0;
541 
542   SmallString<256> Fmt;
543   llvm::raw_svector_ostream OS(Fmt);
544   for (DirectiveList::iterator I = DL.begin(), E = DL.end(); I != E; ++I) {
545     Directive &D = **I;
546     OS << "\n  Line " << SourceMgr.getPresumedLineNumber(D.DiagnosticLoc);
547     if (D.DirectiveLoc != D.DiagnosticLoc)
548       OS << " (directive at "
549          << SourceMgr.getFilename(D.DirectiveLoc) << ":"
550          << SourceMgr.getPresumedLineNumber(D.DirectiveLoc) << ")";
551     OS << ": " << D.Text;
552   }
553 
554   Diags.Report(diag::err_verify_inconsistent_diags).setForceEmit()
555     << Kind << /*Unexpected=*/false << OS.str();
556   return DL.size();
557 }
558 
559 /// CheckLists - Compare expected to seen diagnostic lists and return the
560 /// the difference between them.
561 ///
562 static unsigned CheckLists(DiagnosticsEngine &Diags, SourceManager &SourceMgr,
563                            const char *Label,
564                            DirectiveList &Left,
565                            const_diag_iterator d2_begin,
566                            const_diag_iterator d2_end) {
567   DirectiveList LeftOnly;
568   DiagList Right(d2_begin, d2_end);
569 
570   for (DirectiveList::iterator I = Left.begin(), E = Left.end(); I != E; ++I) {
571     Directive& D = **I;
572     unsigned LineNo1 = SourceMgr.getPresumedLineNumber(D.DiagnosticLoc);
573 
574     for (unsigned i = 0; i < D.Max; ++i) {
575       DiagList::iterator II, IE;
576       for (II = Right.begin(), IE = Right.end(); II != IE; ++II) {
577         unsigned LineNo2 = SourceMgr.getPresumedLineNumber(II->first);
578         if (LineNo1 != LineNo2)
579           continue;
580 
581         const std::string &RightText = II->second;
582         if (D.match(RightText))
583           break;
584       }
585       if (II == IE) {
586         // Not found.
587         if (i >= D.Min) break;
588         LeftOnly.push_back(*I);
589       } else {
590         // Found. The same cannot be found twice.
591         Right.erase(II);
592       }
593     }
594   }
595   // Now all that's left in Right are those that were not matched.
596   unsigned num = PrintExpected(Diags, SourceMgr, LeftOnly, Label);
597   num += PrintUnexpected(Diags, &SourceMgr, Right.begin(), Right.end(), Label);
598   return num;
599 }
600 
601 /// CheckResults - This compares the expected results to those that
602 /// were actually reported. It emits any discrepencies. Return "true" if there
603 /// were problems. Return "false" otherwise.
604 ///
605 static unsigned CheckResults(DiagnosticsEngine &Diags, SourceManager &SourceMgr,
606                              const TextDiagnosticBuffer &Buffer,
607                              ExpectedData &ED) {
608   // We want to capture the delta between what was expected and what was
609   // seen.
610   //
611   //   Expected \ Seen - set expected but not seen
612   //   Seen \ Expected - set seen but not expected
613   unsigned NumProblems = 0;
614 
615   // See if there are error mismatches.
616   NumProblems += CheckLists(Diags, SourceMgr, "error", ED.Errors,
617                             Buffer.err_begin(), Buffer.err_end());
618 
619   // See if there are warning mismatches.
620   NumProblems += CheckLists(Diags, SourceMgr, "warning", ED.Warnings,
621                             Buffer.warn_begin(), Buffer.warn_end());
622 
623   // See if there are note mismatches.
624   NumProblems += CheckLists(Diags, SourceMgr, "note", ED.Notes,
625                             Buffer.note_begin(), Buffer.note_end());
626 
627   return NumProblems;
628 }
629 
630 void VerifyDiagnosticConsumer::UpdateParsedFileStatus(SourceManager &SM,
631                                                       FileID FID,
632                                                       ParsedStatus PS) {
633   // Check SourceManager hasn't changed.
634   setSourceManager(SM);
635 
636 #ifndef NDEBUG
637   if (FID.isInvalid())
638     return;
639 
640   const FileEntry *FE = SM.getFileEntryForID(FID);
641 
642   if (PS == IsParsed) {
643     // Move the FileID from the unparsed set to the parsed set.
644     UnparsedFiles.erase(FID);
645     ParsedFiles.insert(std::make_pair(FID, FE));
646   } else if (!ParsedFiles.count(FID) && !UnparsedFiles.count(FID)) {
647     // Add the FileID to the unparsed set if we haven't seen it before.
648 
649     // Check for directives.
650     bool FoundDirectives;
651     if (PS == IsUnparsedNoDirectives)
652       FoundDirectives = false;
653     else
654       FoundDirectives = !LangOpts || findDirectives(SM, FID, *LangOpts);
655 
656     // Add the FileID to the unparsed set.
657     UnparsedFiles.insert(std::make_pair(FID,
658                                       UnparsedFileStatus(FE, FoundDirectives)));
659   }
660 #endif
661 }
662 
663 void VerifyDiagnosticConsumer::CheckDiagnostics() {
664   // Ensure any diagnostics go to the primary client.
665   bool OwnsCurClient = Diags.ownsClient();
666   DiagnosticConsumer *CurClient = Diags.takeClient();
667   Diags.setClient(PrimaryClient, false);
668 
669 #ifndef NDEBUG
670   // In a debug build, scan through any files that may have been missed
671   // during parsing and issue a fatal error if directives are contained
672   // within these files.  If a fatal error occurs, this suggests that
673   // this file is being parsed separately from the main file, in which
674   // case consider moving the directives to the correct place, if this
675   // is applicable.
676   if (UnparsedFiles.size() > 0) {
677     // Generate a cache of parsed FileEntry pointers for alias lookups.
678     llvm::SmallPtrSet<const FileEntry *, 8> ParsedFileCache;
679     for (ParsedFilesMap::iterator I = ParsedFiles.begin(),
680                                 End = ParsedFiles.end(); I != End; ++I) {
681       if (const FileEntry *FE = I->second)
682         ParsedFileCache.insert(FE);
683     }
684 
685     // Iterate through list of unparsed files.
686     for (UnparsedFilesMap::iterator I = UnparsedFiles.begin(),
687                                   End = UnparsedFiles.end(); I != End; ++I) {
688       const UnparsedFileStatus &Status = I->second;
689       const FileEntry *FE = Status.getFile();
690 
691       // Skip files that have been parsed via an alias.
692       if (FE && ParsedFileCache.count(FE))
693         continue;
694 
695       // Report a fatal error if this file contained directives.
696       if (Status.foundDirectives()) {
697         llvm::report_fatal_error(Twine("-verify directives found after rather"
698                                        " than during normal parsing of ",
699                                  StringRef(FE ? FE->getName() : "(unknown)")));
700       }
701     }
702 
703     // UnparsedFiles has been processed now, so clear it.
704     UnparsedFiles.clear();
705   }
706 #endif // !NDEBUG
707 
708   if (SrcManager) {
709     // Check that the expected diagnostics occurred.
710     NumErrors += CheckResults(Diags, *SrcManager, *Buffer, ED);
711   } else {
712     NumErrors += (PrintUnexpected(Diags, 0, Buffer->err_begin(),
713                                   Buffer->err_end(), "error") +
714                   PrintUnexpected(Diags, 0, Buffer->warn_begin(),
715                                   Buffer->warn_end(), "warn") +
716                   PrintUnexpected(Diags, 0, Buffer->note_begin(),
717                                   Buffer->note_end(), "note"));
718   }
719 
720   Diags.takeClient();
721   Diags.setClient(CurClient, OwnsCurClient);
722 
723   // Reset the buffer, we have processed all the diagnostics in it.
724   Buffer.reset(new TextDiagnosticBuffer());
725   ED.Errors.clear();
726   ED.Warnings.clear();
727   ED.Notes.clear();
728 }
729 
730 DiagnosticConsumer *
731 VerifyDiagnosticConsumer::clone(DiagnosticsEngine &Diags) const {
732   if (!Diags.getClient())
733     Diags.setClient(PrimaryClient->clone(Diags));
734 
735   return new VerifyDiagnosticConsumer(Diags);
736 }
737 
738 Directive *Directive::create(bool RegexKind, SourceLocation DirectiveLoc,
739                              SourceLocation DiagnosticLoc, StringRef Text,
740                              unsigned Min, unsigned Max) {
741   if (RegexKind)
742     return new RegexDirective(DirectiveLoc, DiagnosticLoc, Text, Min, Max);
743   return new StandardDirective(DirectiveLoc, DiagnosticLoc, Text, Min, Max);
744 }
745