1ee3c74fbSChris Lattner //===- FileCheck.cpp - Check that File's Contents match what is expected --===//
2ee3c74fbSChris Lattner //
32946cd70SChandler Carruth // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
42946cd70SChandler Carruth // See https://llvm.org/LICENSE.txt for license information.
52946cd70SChandler Carruth // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6ee3c74fbSChris Lattner //
7ee3c74fbSChris Lattner //===----------------------------------------------------------------------===//
8ee3c74fbSChris Lattner //
9ee3c74fbSChris Lattner // FileCheck does a line-by line check of a file that validates whether it
10ee3c74fbSChris Lattner // contains the expected content.  This is useful for regression tests etc.
11ee3c74fbSChris Lattner //
12b5ecceffSJames Henderson // This program exits with an exit status of 2 on error, exit status of 0 if
13ee3c74fbSChris Lattner // the file matched the expected contents, and exit status of 1 if it did not
14ee3c74fbSChris Lattner // contain the expected contents.
15ee3c74fbSChris Lattner //
16ee3c74fbSChris Lattner //===----------------------------------------------------------------------===//
17ee3c74fbSChris Lattner 
18ee3c74fbSChris Lattner #include "llvm/Support/CommandLine.h"
19197194b6SRui Ueyama #include "llvm/Support/InitLLVM.h"
203e66509fSJoel E. Denny #include "llvm/Support/Process.h"
213c5d267eSJoel E. Denny #include "llvm/Support/WithColor.h"
22ee3c74fbSChris Lattner #include "llvm/Support/raw_ostream.h"
23ffa9d2e4SAditya Nandakumar #include "llvm/Support/FileCheck.h"
24010982f7SRainer Orth #include <cmath>
25ee3c74fbSChris Lattner using namespace llvm;
26ee3c74fbSChris Lattner 
27dbb757f4SJoel E. Denny static cl::extrahelp FileCheckOptsEnv(
28dbb757f4SJoel E. Denny     "\nOptions are parsed from the environment variable FILECHECK_OPTS and\n"
29dbb757f4SJoel E. Denny     "from the command line.\n");
30dbb757f4SJoel E. Denny 
31ee3c74fbSChris Lattner static cl::opt<std::string>
323c5d267eSJoel E. Denny     CheckFilename(cl::Positional, cl::desc("<check-file>"), cl::Optional);
33ee3c74fbSChris Lattner 
34ee3c74fbSChris Lattner static cl::opt<std::string>
35ee3c74fbSChris Lattner     InputFilename("input-file", cl::desc("File to check (defaults to stdin)"),
36ee3c74fbSChris Lattner                   cl::init("-"), cl::value_desc("filename"));
37ee3c74fbSChris Lattner 
38e8f2fb20SChandler Carruth static cl::list<std::string> CheckPrefixes(
39e8f2fb20SChandler Carruth     "check-prefix",
40ee3c74fbSChris Lattner     cl::desc("Prefix to use from check file (defaults to 'CHECK')"));
41fd557cb0SDaniel Sanders static cl::alias CheckPrefixesAlias(
42fd557cb0SDaniel Sanders     "check-prefixes", cl::aliasopt(CheckPrefixes), cl::CommaSeparated,
43fd557cb0SDaniel Sanders     cl::NotHidden,
44fd557cb0SDaniel Sanders     cl::desc(
45fd557cb0SDaniel Sanders         "Alias for -check-prefix permitting multiple comma separated values"));
46ee3c74fbSChris Lattner 
47e8f2fb20SChandler Carruth static cl::opt<bool> NoCanonicalizeWhiteSpace(
48e8f2fb20SChandler Carruth     "strict-whitespace",
492c3e5cdfSChris Lattner     cl::desc("Do not treat all horizontal whitespace as equivalent"));
502c3e5cdfSChris Lattner 
51*5b5b2fd2SKai Nacke static cl::opt<bool> IgnoreCase(
52*5b5b2fd2SKai Nacke     "ignore-case",
53*5b5b2fd2SKai Nacke     cl::desc("Use case-insensitive matching"));
54*5b5b2fd2SKai Nacke 
5556ccdbbdSAlexander Kornienko static cl::list<std::string> ImplicitCheckNot(
5656ccdbbdSAlexander Kornienko     "implicit-check-not",
5756ccdbbdSAlexander Kornienko     cl::desc("Add an implicit negative check with this pattern to every\n"
5856ccdbbdSAlexander Kornienko              "positive check. This can be used to ensure that no instances of\n"
5956ccdbbdSAlexander Kornienko              "this pattern occur which are not matched by a positive pattern"),
6056ccdbbdSAlexander Kornienko     cl::value_desc("pattern"));
6156ccdbbdSAlexander Kornienko 
62a5e233bfSThomas Preud'homme static cl::list<std::string>
63a5e233bfSThomas Preud'homme     GlobalDefines("D", cl::AlwaysPrefix,
6446e1fd61SAlexander Richardson                   cl::desc("Define a variable to be used in capture patterns."),
6546e1fd61SAlexander Richardson                   cl::value_desc("VAR=VALUE"));
6646e1fd61SAlexander Richardson 
671b9f936fSJustin Bogner static cl::opt<bool> AllowEmptyInput(
681b9f936fSJustin Bogner     "allow-empty", cl::init(false),
691b9f936fSJustin Bogner     cl::desc("Allow the input file to be empty. This is useful when making\n"
701b9f936fSJustin Bogner              "checks that some error message does not occur, for example."));
711b9f936fSJustin Bogner 
7285913ccaSJames Y Knight static cl::opt<bool> MatchFullLines(
7385913ccaSJames Y Knight     "match-full-lines", cl::init(false),
7485913ccaSJames Y Knight     cl::desc("Require all positive matches to cover an entire input line.\n"
7585913ccaSJames Y Knight              "Allows leading and trailing whitespace if --strict-whitespace\n"
7685913ccaSJames Y Knight              "is not also passed."));
7785913ccaSJames Y Knight 
78f55e72a5SArtem Belevich static cl::opt<bool> EnableVarScope(
79f55e72a5SArtem Belevich     "enable-var-scope", cl::init(false),
80f55e72a5SArtem Belevich     cl::desc("Enables scope for regex variables. Variables with names that\n"
81f55e72a5SArtem Belevich              "do not start with '$' will be reset at the beginning of\n"
82f55e72a5SArtem Belevich              "each CHECK-LABEL block."));
83f55e72a5SArtem Belevich 
84bcf5b441SJoel E. Denny static cl::opt<bool> AllowDeprecatedDagOverlap(
85bcf5b441SJoel E. Denny     "allow-deprecated-dag-overlap", cl::init(false),
86bcf5b441SJoel E. Denny     cl::desc("Enable overlapping among matches in a group of consecutive\n"
87bcf5b441SJoel E. Denny              "CHECK-DAG directives.  This option is deprecated and is only\n"
88bcf5b441SJoel E. Denny              "provided for convenience as old tests are migrated to the new\n"
89bcf5b441SJoel E. Denny              "non-overlapping CHECK-DAG implementation.\n"));
90bcf5b441SJoel E. Denny 
91352695c3SJoel E. Denny static cl::opt<bool> Verbose(
92352695c3SJoel E. Denny     "v", cl::init(false),
93352695c3SJoel E. Denny     cl::desc("Print directive pattern matches, or add them to the input dump\n"
94352695c3SJoel E. Denny              "if enabled.\n"));
95dc5ba317SJoel E. Denny 
96dc5ba317SJoel E. Denny static cl::opt<bool> VerboseVerbose(
97dc5ba317SJoel E. Denny     "vv", cl::init(false),
98dc5ba317SJoel E. Denny     cl::desc("Print information helpful in diagnosing internal FileCheck\n"
99352695c3SJoel E. Denny              "issues, or add it to the input dump if enabled.  Implies\n"
100352695c3SJoel E. Denny              "-v.\n"));
101346dfbe2SGeorge Karpenkov static const char * DumpInputEnv = "FILECHECK_DUMP_INPUT_ON_FAILURE";
102346dfbe2SGeorge Karpenkov 
103346dfbe2SGeorge Karpenkov static cl::opt<bool> DumpInputOnFailure(
104ffc722a3SMichal Gorny     "dump-input-on-failure",
105ffc722a3SMichal Gorny     cl::init(std::getenv(DumpInputEnv) && *std::getenv(DumpInputEnv)),
106346dfbe2SGeorge Karpenkov     cl::desc("Dump original input to stderr before failing.\n"
107346dfbe2SGeorge Karpenkov              "The value can be also controlled using\n"
1083c5d267eSJoel E. Denny              "FILECHECK_DUMP_INPUT_ON_FAILURE environment variable.\n"
1093c5d267eSJoel E. Denny              "This option is deprecated in favor of -dump-input=fail.\n"));
1103c5d267eSJoel E. Denny 
1113c5d267eSJoel E. Denny enum DumpInputValue {
1123c5d267eSJoel E. Denny   DumpInputDefault,
1133c5d267eSJoel E. Denny   DumpInputHelp,
1143c5d267eSJoel E. Denny   DumpInputNever,
1153c5d267eSJoel E. Denny   DumpInputFail,
1163c5d267eSJoel E. Denny   DumpInputAlways
1173c5d267eSJoel E. Denny };
1183c5d267eSJoel E. Denny 
1193c5d267eSJoel E. Denny static cl::opt<DumpInputValue> DumpInput(
1203c5d267eSJoel E. Denny     "dump-input", cl::init(DumpInputDefault),
1213c5d267eSJoel E. Denny     cl::desc("Dump input to stderr, adding annotations representing\n"
1223c5d267eSJoel E. Denny              " currently enabled diagnostics\n"),
1233c5d267eSJoel E. Denny     cl::value_desc("mode"),
1243c5d267eSJoel E. Denny     cl::values(clEnumValN(DumpInputHelp, "help",
1253c5d267eSJoel E. Denny                           "Explain dump format and quit"),
1263c5d267eSJoel E. Denny                clEnumValN(DumpInputNever, "never", "Never dump input"),
1273c5d267eSJoel E. Denny                clEnumValN(DumpInputFail, "fail", "Dump input on failure"),
1283c5d267eSJoel E. Denny                clEnumValN(DumpInputAlways, "always", "Always dump input")));
129dc5ba317SJoel E. Denny 
13013df4626SMatt Arsenault typedef cl::list<std::string>::const_iterator prefix_iterator;
13113df4626SMatt Arsenault 
13213df4626SMatt Arsenault 
13313df4626SMatt Arsenault 
134726774cbSChandler Carruth 
135726774cbSChandler Carruth 
136726774cbSChandler Carruth 
137c2735158SRui Ueyama 
1382bd4f8b6SXinliang David Li static void DumpCommandLine(int argc, char **argv) {
1392bd4f8b6SXinliang David Li   errs() << "FileCheck command line: ";
1402bd4f8b6SXinliang David Li   for (int I = 0; I < argc; I++)
1412bd4f8b6SXinliang David Li     errs() << " " << argv[I];
1422bd4f8b6SXinliang David Li   errs() << "\n";
1432bd4f8b6SXinliang David Li }
1442bd4f8b6SXinliang David Li 
1453c5d267eSJoel E. Denny struct MarkerStyle {
1463c5d267eSJoel E. Denny   /// The starting char (before tildes) for marking the line.
1473c5d267eSJoel E. Denny   char Lead;
1483c5d267eSJoel E. Denny   /// What color to use for this annotation.
1494d41c332SRui Ueyama   raw_ostream::Colors Color;
1503c5d267eSJoel E. Denny   /// A note to follow the marker, or empty string if none.
1513c5d267eSJoel E. Denny   std::string Note;
1523c5d267eSJoel E. Denny   MarkerStyle() {}
1534d41c332SRui Ueyama   MarkerStyle(char Lead, raw_ostream::Colors Color,
1544d41c332SRui Ueyama               const std::string &Note = "")
1553c5d267eSJoel E. Denny       : Lead(Lead), Color(Color), Note(Note) {}
1563c5d267eSJoel E. Denny };
1573c5d267eSJoel E. Denny 
1583c5d267eSJoel E. Denny static MarkerStyle GetMarker(FileCheckDiag::MatchType MatchTy) {
1593c5d267eSJoel E. Denny   switch (MatchTy) {
160e2afb614SJoel E. Denny   case FileCheckDiag::MatchFoundAndExpected:
1617df86967SJoel E. Denny     return MarkerStyle('^', raw_ostream::GREEN);
162e2afb614SJoel E. Denny   case FileCheckDiag::MatchFoundButExcluded:
1630e7e3fa0SJoel E. Denny     return MarkerStyle('!', raw_ostream::RED, "error: no match expected");
164e2afb614SJoel E. Denny   case FileCheckDiag::MatchFoundButWrongLine:
165cadfcef4SJoel E. Denny     return MarkerStyle('!', raw_ostream::RED, "error: match on wrong line");
166e2afb614SJoel E. Denny   case FileCheckDiag::MatchFoundButDiscarded:
167f7c1c4d8SJoel E. Denny     return MarkerStyle('!', raw_ostream::CYAN,
168f7c1c4d8SJoel E. Denny                        "discard: overlaps earlier match");
16996f0e84cSJoel E. Denny   case FileCheckDiag::MatchNoneAndExcluded:
17096f0e84cSJoel E. Denny     return MarkerStyle('X', raw_ostream::GREEN);
1713c5d267eSJoel E. Denny   case FileCheckDiag::MatchNoneButExpected:
1723c5d267eSJoel E. Denny     return MarkerStyle('X', raw_ostream::RED, "error: no match found");
1732c007c80SJoel E. Denny   case FileCheckDiag::MatchFuzzy:
1742c007c80SJoel E. Denny     return MarkerStyle('?', raw_ostream::MAGENTA, "possible intended match");
1753c5d267eSJoel E. Denny   }
1763c5d267eSJoel E. Denny   llvm_unreachable_internal("unexpected match type");
1773c5d267eSJoel E. Denny }
1783c5d267eSJoel E. Denny 
1793c5d267eSJoel E. Denny static void DumpInputAnnotationHelp(raw_ostream &OS) {
1803c5d267eSJoel E. Denny   OS << "The following description was requested by -dump-input=help to\n"
1813c5d267eSJoel E. Denny      << "explain the input annotations printed by -dump-input=always and\n"
1823c5d267eSJoel E. Denny      << "-dump-input=fail:\n\n";
1833c5d267eSJoel E. Denny 
1843c5d267eSJoel E. Denny   // Labels for input lines.
1853c5d267eSJoel E. Denny   OS << "  - ";
1863c5d267eSJoel E. Denny   WithColor(OS, raw_ostream::SAVEDCOLOR, true) << "L:";
1873c5d267eSJoel E. Denny   OS << "     labels line number L of the input file\n";
1883c5d267eSJoel E. Denny 
1893c5d267eSJoel E. Denny   // Labels for annotation lines.
1903c5d267eSJoel E. Denny   OS << "  - ";
1913c5d267eSJoel E. Denny   WithColor(OS, raw_ostream::SAVEDCOLOR, true) << "T:L";
1922c007c80SJoel E. Denny   OS << "    labels the only match result for a pattern of type T from "
1933c5d267eSJoel E. Denny      << "line L of\n"
1943c5d267eSJoel E. Denny      << "           the check file\n";
1952c007c80SJoel E. Denny   OS << "  - ";
1962c007c80SJoel E. Denny   WithColor(OS, raw_ostream::SAVEDCOLOR, true) << "T:L'N";
1972c007c80SJoel E. Denny   OS << "  labels the Nth match result for a pattern of type T from line "
1982c007c80SJoel E. Denny      << "L of\n"
1992c007c80SJoel E. Denny      << "           the check file\n";
2003c5d267eSJoel E. Denny 
2013c5d267eSJoel E. Denny   // Markers on annotation lines.
2023c5d267eSJoel E. Denny   OS << "  - ";
2037df86967SJoel E. Denny   WithColor(OS, raw_ostream::SAVEDCOLOR, true) << "^~~";
2047df86967SJoel E. Denny   OS << "    marks good match (reported if -v)\n"
2057df86967SJoel E. Denny      << "  - ";
206cadfcef4SJoel E. Denny   WithColor(OS, raw_ostream::SAVEDCOLOR, true) << "!~~";
207cadfcef4SJoel E. Denny   OS << "    marks bad match, such as:\n"
208cadfcef4SJoel E. Denny      << "           - CHECK-NEXT on same line as previous match (error)\n"
2090e7e3fa0SJoel E. Denny      << "           - CHECK-NOT found (error)\n"
210f7c1c4d8SJoel E. Denny      << "           - CHECK-DAG overlapping match (discarded, reported if "
211f7c1c4d8SJoel E. Denny      << "-vv)\n"
212cadfcef4SJoel E. Denny      << "  - ";
2133c5d267eSJoel E. Denny   WithColor(OS, raw_ostream::SAVEDCOLOR, true) << "X~~";
214cadfcef4SJoel E. Denny   OS << "    marks search range when no match is found, such as:\n"
215cadfcef4SJoel E. Denny      << "           - CHECK-NEXT not found (error)\n"
21696f0e84cSJoel E. Denny      << "           - CHECK-NOT not found (success, reported if -vv)\n"
217f7c1c4d8SJoel E. Denny      << "           - CHECK-DAG not found after discarded matches (error)\n"
2182c007c80SJoel E. Denny      << "  - ";
2192c007c80SJoel E. Denny   WithColor(OS, raw_ostream::SAVEDCOLOR, true) << "?";
2202c007c80SJoel E. Denny   OS << "      marks fuzzy match when no match is found\n";
2213c5d267eSJoel E. Denny 
2223c5d267eSJoel E. Denny   // Colors.
2233c5d267eSJoel E. Denny   OS << "  - colors ";
2247df86967SJoel E. Denny   WithColor(OS, raw_ostream::GREEN, true) << "success";
2257df86967SJoel E. Denny   OS << ", ";
2263c5d267eSJoel E. Denny   WithColor(OS, raw_ostream::RED, true) << "error";
2272c007c80SJoel E. Denny   OS << ", ";
2282c007c80SJoel E. Denny   WithColor(OS, raw_ostream::MAGENTA, true) << "fuzzy match";
2297df86967SJoel E. Denny   OS << ", ";
230f7c1c4d8SJoel E. Denny   WithColor(OS, raw_ostream::CYAN, true, false) << "discarded match";
231f7c1c4d8SJoel E. Denny   OS << ", ";
2327df86967SJoel E. Denny   WithColor(OS, raw_ostream::CYAN, true, true) << "unmatched input";
2333c5d267eSJoel E. Denny   OS << "\n\n"
2343c5d267eSJoel E. Denny      << "If you are not seeing color above or in input dumps, try: -color\n";
2353c5d267eSJoel E. Denny }
2363c5d267eSJoel E. Denny 
2373c5d267eSJoel E. Denny /// An annotation for a single input line.
2383c5d267eSJoel E. Denny struct InputAnnotation {
2393c5d267eSJoel E. Denny   /// The check file line (one-origin indexing) where the directive that
2403c5d267eSJoel E. Denny   /// produced this annotation is located.
2413c5d267eSJoel E. Denny   unsigned CheckLine;
2422c007c80SJoel E. Denny   /// The index of the match result for this check.
2432c007c80SJoel E. Denny   unsigned CheckDiagIndex;
2443c5d267eSJoel E. Denny   /// The label for this annotation.
2453c5d267eSJoel E. Denny   std::string Label;
2463c5d267eSJoel E. Denny   /// What input line (one-origin indexing) this annotation marks.  This might
2473c5d267eSJoel E. Denny   /// be different from the starting line of the original diagnostic if this is
2483c5d267eSJoel E. Denny   /// a non-initial fragment of a diagnostic that has been broken across
2493c5d267eSJoel E. Denny   /// multiple lines.
2503c5d267eSJoel E. Denny   unsigned InputLine;
2513c5d267eSJoel E. Denny   /// The column range (one-origin indexing, open end) in which to to mark the
2523c5d267eSJoel E. Denny   /// input line.  If InputEndCol is UINT_MAX, treat it as the last column
2533c5d267eSJoel E. Denny   /// before the newline.
2543c5d267eSJoel E. Denny   unsigned InputStartCol, InputEndCol;
2553c5d267eSJoel E. Denny   /// The marker to use.
2563c5d267eSJoel E. Denny   MarkerStyle Marker;
257e2afb614SJoel E. Denny   /// Whether this annotation represents a good match for an expected pattern.
258e2afb614SJoel E. Denny   bool FoundAndExpectedMatch;
2593c5d267eSJoel E. Denny };
2603c5d267eSJoel E. Denny 
2613c5d267eSJoel E. Denny /// Get an abbreviation for the check type.
2623c5d267eSJoel E. Denny std::string GetCheckTypeAbbreviation(Check::FileCheckType Ty) {
2633c5d267eSJoel E. Denny   switch (Ty) {
2643c5d267eSJoel E. Denny   case Check::CheckPlain:
2653c5d267eSJoel E. Denny     if (Ty.getCount() > 1)
2663c5d267eSJoel E. Denny       return "count";
2673c5d267eSJoel E. Denny     return "check";
2683c5d267eSJoel E. Denny   case Check::CheckNext:
2693c5d267eSJoel E. Denny     return "next";
2703c5d267eSJoel E. Denny   case Check::CheckSame:
2713c5d267eSJoel E. Denny     return "same";
2723c5d267eSJoel E. Denny   case Check::CheckNot:
2733c5d267eSJoel E. Denny     return "not";
2743c5d267eSJoel E. Denny   case Check::CheckDAG:
2753c5d267eSJoel E. Denny     return "dag";
2763c5d267eSJoel E. Denny   case Check::CheckLabel:
2773c5d267eSJoel E. Denny     return "label";
2783c5d267eSJoel E. Denny   case Check::CheckEmpty:
2793c5d267eSJoel E. Denny     return "empty";
2803c5d267eSJoel E. Denny   case Check::CheckEOF:
2813c5d267eSJoel E. Denny     return "eof";
2823c5d267eSJoel E. Denny   case Check::CheckBadNot:
2833c5d267eSJoel E. Denny     return "bad-not";
2843c5d267eSJoel E. Denny   case Check::CheckBadCount:
2853c5d267eSJoel E. Denny     return "bad-count";
2863c5d267eSJoel E. Denny   case Check::CheckNone:
2873c5d267eSJoel E. Denny     llvm_unreachable("invalid FileCheckType");
2883c5d267eSJoel E. Denny   }
2893c5d267eSJoel E. Denny   llvm_unreachable("unknown FileCheckType");
2903c5d267eSJoel E. Denny }
2913c5d267eSJoel E. Denny 
2923c5d267eSJoel E. Denny static void BuildInputAnnotations(const std::vector<FileCheckDiag> &Diags,
2933c5d267eSJoel E. Denny                                   std::vector<InputAnnotation> &Annotations,
2943c5d267eSJoel E. Denny                                   unsigned &LabelWidth) {
2952c007c80SJoel E. Denny   // How many diagnostics has the current check seen so far?
2962c007c80SJoel E. Denny   unsigned CheckDiagCount = 0;
2973c5d267eSJoel E. Denny   // What's the widest label?
2983c5d267eSJoel E. Denny   LabelWidth = 0;
2993c5d267eSJoel E. Denny   for (auto DiagItr = Diags.begin(), DiagEnd = Diags.end(); DiagItr != DiagEnd;
3003c5d267eSJoel E. Denny        ++DiagItr) {
3013c5d267eSJoel E. Denny     InputAnnotation A;
3023c5d267eSJoel E. Denny 
3033c5d267eSJoel E. Denny     // Build label, which uniquely identifies this check result.
3043c5d267eSJoel E. Denny     A.CheckLine = DiagItr->CheckLine;
3053c5d267eSJoel E. Denny     llvm::raw_string_ostream Label(A.Label);
3063c5d267eSJoel E. Denny     Label << GetCheckTypeAbbreviation(DiagItr->CheckTy) << ":"
3073c5d267eSJoel E. Denny           << DiagItr->CheckLine;
3082c007c80SJoel E. Denny     A.CheckDiagIndex = UINT_MAX;
3092c007c80SJoel E. Denny     auto DiagNext = std::next(DiagItr);
3102c007c80SJoel E. Denny     if (DiagNext != DiagEnd && DiagItr->CheckTy == DiagNext->CheckTy &&
3112c007c80SJoel E. Denny         DiagItr->CheckLine == DiagNext->CheckLine)
3122c007c80SJoel E. Denny       A.CheckDiagIndex = CheckDiagCount++;
3132c007c80SJoel E. Denny     else if (CheckDiagCount) {
3142c007c80SJoel E. Denny       A.CheckDiagIndex = CheckDiagCount;
3152c007c80SJoel E. Denny       CheckDiagCount = 0;
3162c007c80SJoel E. Denny     }
3172c007c80SJoel E. Denny     if (A.CheckDiagIndex != UINT_MAX)
3182c007c80SJoel E. Denny       Label << "'" << A.CheckDiagIndex;
3192c007c80SJoel E. Denny     else
3202c007c80SJoel E. Denny       A.CheckDiagIndex = 0;
3213c5d267eSJoel E. Denny     Label.flush();
3223c5d267eSJoel E. Denny     LabelWidth = std::max((std::string::size_type)LabelWidth, A.Label.size());
3233c5d267eSJoel E. Denny 
324608f2bfdSJoel E. Denny     A.Marker = GetMarker(DiagItr->MatchTy);
325e2afb614SJoel E. Denny     A.FoundAndExpectedMatch =
326e2afb614SJoel E. Denny         DiagItr->MatchTy == FileCheckDiag::MatchFoundAndExpected;
3273c5d267eSJoel E. Denny 
3283c5d267eSJoel E. Denny     // Compute the mark location, and break annotation into multiple
3293c5d267eSJoel E. Denny     // annotations if it spans multiple lines.
3303c5d267eSJoel E. Denny     A.InputLine = DiagItr->InputStartLine;
3313c5d267eSJoel E. Denny     A.InputStartCol = DiagItr->InputStartCol;
3323c5d267eSJoel E. Denny     if (DiagItr->InputStartLine == DiagItr->InputEndLine) {
3333c5d267eSJoel E. Denny       // Sometimes ranges are empty in order to indicate a specific point, but
3343c5d267eSJoel E. Denny       // that would mean nothing would be marked, so adjust the range to
3353c5d267eSJoel E. Denny       // include the following character.
3363c5d267eSJoel E. Denny       A.InputEndCol =
3373c5d267eSJoel E. Denny           std::max(DiagItr->InputStartCol + 1, DiagItr->InputEndCol);
3383c5d267eSJoel E. Denny       Annotations.push_back(A);
3393c5d267eSJoel E. Denny     } else {
3403c5d267eSJoel E. Denny       assert(DiagItr->InputStartLine < DiagItr->InputEndLine &&
3413c5d267eSJoel E. Denny              "expected input range not to be inverted");
3423c5d267eSJoel E. Denny       A.InputEndCol = UINT_MAX;
3433c5d267eSJoel E. Denny       Annotations.push_back(A);
3443c5d267eSJoel E. Denny       for (unsigned L = DiagItr->InputStartLine + 1, E = DiagItr->InputEndLine;
3453c5d267eSJoel E. Denny            L <= E; ++L) {
3463c5d267eSJoel E. Denny         // If a range ends before the first column on a line, then it has no
3473c5d267eSJoel E. Denny         // characters on that line, so there's nothing to render.
348608f2bfdSJoel E. Denny         if (DiagItr->InputEndCol == 1 && L == E)
3493c5d267eSJoel E. Denny           break;
3503c5d267eSJoel E. Denny         InputAnnotation B;
3513c5d267eSJoel E. Denny         B.CheckLine = A.CheckLine;
3522c007c80SJoel E. Denny         B.CheckDiagIndex = A.CheckDiagIndex;
3533c5d267eSJoel E. Denny         B.Label = A.Label;
3543c5d267eSJoel E. Denny         B.InputLine = L;
355608f2bfdSJoel E. Denny         B.Marker = A.Marker;
3563c5d267eSJoel E. Denny         B.Marker.Lead = '~';
3573c5d267eSJoel E. Denny         B.Marker.Note = "";
358608f2bfdSJoel E. Denny         B.InputStartCol = 1;
359608f2bfdSJoel E. Denny         if (L != E)
360608f2bfdSJoel E. Denny           B.InputEndCol = UINT_MAX;
361608f2bfdSJoel E. Denny         else
3623c5d267eSJoel E. Denny           B.InputEndCol = DiagItr->InputEndCol;
363e2afb614SJoel E. Denny         B.FoundAndExpectedMatch = A.FoundAndExpectedMatch;
3643c5d267eSJoel E. Denny         Annotations.push_back(B);
3653c5d267eSJoel E. Denny       }
3663c5d267eSJoel E. Denny     }
3673c5d267eSJoel E. Denny   }
3683c5d267eSJoel E. Denny }
3693c5d267eSJoel E. Denny 
3707df86967SJoel E. Denny static void DumpAnnotatedInput(raw_ostream &OS, const FileCheckRequest &Req,
3717df86967SJoel E. Denny                                StringRef InputFileText,
3727df86967SJoel E. Denny                                std::vector<InputAnnotation> &Annotations,
3737df86967SJoel E. Denny                                unsigned LabelWidth) {
3743c5d267eSJoel E. Denny   OS << "Full input was:\n<<<<<<\n";
3753c5d267eSJoel E. Denny 
3763c5d267eSJoel E. Denny   // Sort annotations.
3773c5d267eSJoel E. Denny   //
3783c5d267eSJoel E. Denny   // First, sort in the order of input lines to make it easier to find relevant
3793c5d267eSJoel E. Denny   // annotations while iterating input lines in the implementation below.
3803c5d267eSJoel E. Denny   // FileCheck diagnostics are not always reported and recorded in the order of
3813c5d267eSJoel E. Denny   // input lines due to, for example, CHECK-DAG and CHECK-NOT.
3823c5d267eSJoel E. Denny   //
3833c5d267eSJoel E. Denny   // Second, for annotations for the same input line, sort in the order of the
3843c5d267eSJoel E. Denny   // FileCheck directive's line in the check file (where there's at most one
3852c007c80SJoel E. Denny   // directive per line) and then by the index of the match result for that
3862c007c80SJoel E. Denny   // directive.  The rationale of this choice is that, for any input line, this
3872c007c80SJoel E. Denny   // sort establishes a total order of annotations that, with respect to match
3882c007c80SJoel E. Denny   // results, is consistent across multiple lines, thus making match results
3892c007c80SJoel E. Denny   // easier to track from one line to the next when they span multiple lines.
3903c5d267eSJoel E. Denny   std::sort(Annotations.begin(), Annotations.end(),
3913c5d267eSJoel E. Denny             [](const InputAnnotation &A, const InputAnnotation &B) {
3923c5d267eSJoel E. Denny               if (A.InputLine != B.InputLine)
3933c5d267eSJoel E. Denny                 return A.InputLine < B.InputLine;
3942c007c80SJoel E. Denny               if (A.CheckLine != B.CheckLine)
3953c5d267eSJoel E. Denny                 return A.CheckLine < B.CheckLine;
3967df86967SJoel E. Denny               // FIXME: Sometimes CHECK-LABEL reports its match twice with
3977df86967SJoel E. Denny               // other diagnostics in between, and then diag index incrementing
3987df86967SJoel E. Denny               // fails to work properly, and then this assert fails.  We should
3997df86967SJoel E. Denny               // suppress one of those diagnostics or do a better job of
4007df86967SJoel E. Denny               // computing this index.  For now, we just produce a redundant
4017df86967SJoel E. Denny               // CHECK-LABEL annotation.
4027df86967SJoel E. Denny               // assert(A.CheckDiagIndex != B.CheckDiagIndex &&
4037df86967SJoel E. Denny               //        "expected diagnostic indices to be unique within a "
4047df86967SJoel E. Denny               //        " check line");
4052c007c80SJoel E. Denny               return A.CheckDiagIndex < B.CheckDiagIndex;
4063c5d267eSJoel E. Denny             });
4073c5d267eSJoel E. Denny 
4083c5d267eSJoel E. Denny   // Compute the width of the label column.
4093c5d267eSJoel E. Denny   const unsigned char *InputFilePtr = InputFileText.bytes_begin(),
4103c5d267eSJoel E. Denny                       *InputFileEnd = InputFileText.bytes_end();
4113c5d267eSJoel E. Denny   unsigned LineCount = InputFileText.count('\n');
4123c5d267eSJoel E. Denny   if (InputFileEnd[-1] != '\n')
4133c5d267eSJoel E. Denny     ++LineCount;
414010982f7SRainer Orth   unsigned LineNoWidth = std::log10(LineCount) + 1;
4153c5d267eSJoel E. Denny   // +3 below adds spaces (1) to the left of the (right-aligned) line numbers
4163c5d267eSJoel E. Denny   // on input lines and (2) to the right of the (left-aligned) labels on
4173c5d267eSJoel E. Denny   // annotation lines so that input lines and annotation lines are more
4183c5d267eSJoel E. Denny   // visually distinct.  For example, the spaces on the annotation lines ensure
4193c5d267eSJoel E. Denny   // that input line numbers and check directive line numbers never align
4203c5d267eSJoel E. Denny   // horizontally.  Those line numbers might not even be for the same file.
4213c5d267eSJoel E. Denny   // One space would be enough to achieve that, but more makes it even easier
4223c5d267eSJoel E. Denny   // to see.
4233c5d267eSJoel E. Denny   LabelWidth = std::max(LabelWidth, LineNoWidth) + 3;
4243c5d267eSJoel E. Denny 
4253c5d267eSJoel E. Denny   // Print annotated input lines.
4263c5d267eSJoel E. Denny   auto AnnotationItr = Annotations.begin(), AnnotationEnd = Annotations.end();
4273c5d267eSJoel E. Denny   for (unsigned Line = 1;
4283c5d267eSJoel E. Denny        InputFilePtr != InputFileEnd || AnnotationItr != AnnotationEnd;
4293c5d267eSJoel E. Denny        ++Line) {
4303c5d267eSJoel E. Denny     const unsigned char *InputFileLine = InputFilePtr;
4313c5d267eSJoel E. Denny 
4323c5d267eSJoel E. Denny     // Print right-aligned line number.
4333c5d267eSJoel E. Denny     WithColor(OS, raw_ostream::BLACK, true)
4343c5d267eSJoel E. Denny         << format_decimal(Line, LabelWidth) << ": ";
4353c5d267eSJoel E. Denny 
436e2afb614SJoel E. Denny     // For the case where -v and colors are enabled, find the annotations for
437e2afb614SJoel E. Denny     // good matches for expected patterns in order to highlight everything
438e2afb614SJoel E. Denny     // else in the line.  There are no such annotations if -v is disabled.
439e2afb614SJoel E. Denny     std::vector<InputAnnotation> FoundAndExpectedMatches;
4407df86967SJoel E. Denny     if (Req.Verbose && WithColor(OS).colorsEnabled()) {
4417df86967SJoel E. Denny       for (auto I = AnnotationItr; I != AnnotationEnd && I->InputLine == Line;
4427df86967SJoel E. Denny            ++I) {
443e2afb614SJoel E. Denny         if (I->FoundAndExpectedMatch)
444e2afb614SJoel E. Denny           FoundAndExpectedMatches.push_back(*I);
4457df86967SJoel E. Denny       }
4467df86967SJoel E. Denny     }
4477df86967SJoel E. Denny 
4487df86967SJoel E. Denny     // Print numbered line with highlighting where there are no matches for
4497df86967SJoel E. Denny     // expected patterns.
4503c5d267eSJoel E. Denny     bool Newline = false;
4517df86967SJoel E. Denny     {
4527df86967SJoel E. Denny       WithColor COS(OS);
4537df86967SJoel E. Denny       bool InMatch = false;
4547df86967SJoel E. Denny       if (Req.Verbose)
4557df86967SJoel E. Denny         COS.changeColor(raw_ostream::CYAN, true, true);
4567df86967SJoel E. Denny       for (unsigned Col = 1; InputFilePtr != InputFileEnd && !Newline; ++Col) {
4577df86967SJoel E. Denny         bool WasInMatch = InMatch;
4587df86967SJoel E. Denny         InMatch = false;
459e2afb614SJoel E. Denny         for (auto M : FoundAndExpectedMatches) {
4607df86967SJoel E. Denny           if (M.InputStartCol <= Col && Col < M.InputEndCol) {
4617df86967SJoel E. Denny             InMatch = true;
4627df86967SJoel E. Denny             break;
4637df86967SJoel E. Denny           }
4647df86967SJoel E. Denny         }
4657df86967SJoel E. Denny         if (!WasInMatch && InMatch)
4667df86967SJoel E. Denny           COS.resetColor();
4677df86967SJoel E. Denny         else if (WasInMatch && !InMatch)
4687df86967SJoel E. Denny           COS.changeColor(raw_ostream::CYAN, true, true);
4693c5d267eSJoel E. Denny         if (*InputFilePtr == '\n')
4703c5d267eSJoel E. Denny           Newline = true;
4713c5d267eSJoel E. Denny         else
4727df86967SJoel E. Denny           COS << *InputFilePtr;
4733c5d267eSJoel E. Denny         ++InputFilePtr;
4743c5d267eSJoel E. Denny       }
4757df86967SJoel E. Denny     }
4763c5d267eSJoel E. Denny     OS << '\n';
4773c5d267eSJoel E. Denny     unsigned InputLineWidth = InputFilePtr - InputFileLine - Newline;
4783c5d267eSJoel E. Denny 
4793c5d267eSJoel E. Denny     // Print any annotations.
4803c5d267eSJoel E. Denny     while (AnnotationItr != AnnotationEnd &&
4813c5d267eSJoel E. Denny            AnnotationItr->InputLine == Line) {
4823c5d267eSJoel E. Denny       WithColor COS(OS, AnnotationItr->Marker.Color, true);
4833c5d267eSJoel E. Denny       // The two spaces below are where the ": " appears on input lines.
4843c5d267eSJoel E. Denny       COS << left_justify(AnnotationItr->Label, LabelWidth) << "  ";
4853c5d267eSJoel E. Denny       unsigned Col;
4863c5d267eSJoel E. Denny       for (Col = 1; Col < AnnotationItr->InputStartCol; ++Col)
4873c5d267eSJoel E. Denny         COS << ' ';
4883c5d267eSJoel E. Denny       COS << AnnotationItr->Marker.Lead;
4893c5d267eSJoel E. Denny       // If InputEndCol=UINT_MAX, stop at InputLineWidth.
4903c5d267eSJoel E. Denny       for (++Col; Col < AnnotationItr->InputEndCol && Col <= InputLineWidth;
4913c5d267eSJoel E. Denny            ++Col)
4923c5d267eSJoel E. Denny         COS << '~';
4933c5d267eSJoel E. Denny       const std::string &Note = AnnotationItr->Marker.Note;
4943c5d267eSJoel E. Denny       if (!Note.empty()) {
4953c5d267eSJoel E. Denny         // Put the note at the end of the input line.  If we were to instead
4963c5d267eSJoel E. Denny         // put the note right after the marker, subsequent annotations for the
4973c5d267eSJoel E. Denny         // same input line might appear to mark this note instead of the input
4983c5d267eSJoel E. Denny         // line.
4993c5d267eSJoel E. Denny         for (; Col <= InputLineWidth; ++Col)
5003c5d267eSJoel E. Denny           COS << ' ';
5013c5d267eSJoel E. Denny         COS << ' ' << Note;
5023c5d267eSJoel E. Denny       }
5033c5d267eSJoel E. Denny       COS << '\n';
5043c5d267eSJoel E. Denny       ++AnnotationItr;
5053c5d267eSJoel E. Denny     }
5063c5d267eSJoel E. Denny   }
5073c5d267eSJoel E. Denny 
5083c5d267eSJoel E. Denny   OS << ">>>>>>\n";
5093c5d267eSJoel E. Denny }
5103c5d267eSJoel E. Denny 
511ee3c74fbSChris Lattner int main(int argc, char **argv) {
5123e66509fSJoel E. Denny   // Enable use of ANSI color codes because FileCheck is using them to
5133e66509fSJoel E. Denny   // highlight text.
5143e66509fSJoel E. Denny   llvm::sys::Process::UseANSIEscapeCodes(true);
5153e66509fSJoel E. Denny 
516197194b6SRui Ueyama   InitLLVM X(argc, argv);
51724994d77SJoel E. Denny   cl::ParseCommandLineOptions(argc, argv, /*Overview*/ "", /*Errs*/ nullptr,
51824994d77SJoel E. Denny                               "FILECHECK_OPTS");
5193c5d267eSJoel E. Denny   if (DumpInput == DumpInputHelp) {
5203c5d267eSJoel E. Denny     DumpInputAnnotationHelp(outs());
5213c5d267eSJoel E. Denny     return 0;
5223c5d267eSJoel E. Denny   }
5233c5d267eSJoel E. Denny   if (CheckFilename.empty()) {
5243c5d267eSJoel E. Denny     errs() << "<check-file> not specified\n";
5253c5d267eSJoel E. Denny     return 2;
5263c5d267eSJoel E. Denny   }
527ee3c74fbSChris Lattner 
528ffa9d2e4SAditya Nandakumar   FileCheckRequest Req;
529ffa9d2e4SAditya Nandakumar   for (auto Prefix : CheckPrefixes)
530ffa9d2e4SAditya Nandakumar     Req.CheckPrefixes.push_back(Prefix);
531ffa9d2e4SAditya Nandakumar 
532ffa9d2e4SAditya Nandakumar   for (auto CheckNot : ImplicitCheckNot)
533ffa9d2e4SAditya Nandakumar     Req.ImplicitCheckNot.push_back(CheckNot);
534ffa9d2e4SAditya Nandakumar 
535a5e233bfSThomas Preud'homme   bool GlobalDefineError = false;
536a5e233bfSThomas Preud'homme   for (auto G : GlobalDefines) {
537a5e233bfSThomas Preud'homme     size_t EqIdx = G.find('=');
538a5e233bfSThomas Preud'homme     if (EqIdx == std::string::npos) {
539a5e233bfSThomas Preud'homme       errs() << "Missing equal sign in command-line definition '-D" << G
540a5e233bfSThomas Preud'homme              << "'\n";
541a5e233bfSThomas Preud'homme       GlobalDefineError = true;
542a5e233bfSThomas Preud'homme       continue;
543a5e233bfSThomas Preud'homme     }
544a5e233bfSThomas Preud'homme     if (EqIdx == 0) {
5451a944d27SThomas Preud'homme       errs() << "Missing variable name in command-line definition '-D" << G
5461a944d27SThomas Preud'homme              << "'\n";
547a5e233bfSThomas Preud'homme       GlobalDefineError = true;
548a5e233bfSThomas Preud'homme       continue;
549a5e233bfSThomas Preud'homme     }
550ffa9d2e4SAditya Nandakumar     Req.GlobalDefines.push_back(G);
551a5e233bfSThomas Preud'homme   }
552a5e233bfSThomas Preud'homme   if (GlobalDefineError)
553a5e233bfSThomas Preud'homme     return 2;
554ffa9d2e4SAditya Nandakumar 
555ffa9d2e4SAditya Nandakumar   Req.AllowEmptyInput = AllowEmptyInput;
556ffa9d2e4SAditya Nandakumar   Req.EnableVarScope = EnableVarScope;
557ffa9d2e4SAditya Nandakumar   Req.AllowDeprecatedDagOverlap = AllowDeprecatedDagOverlap;
558ffa9d2e4SAditya Nandakumar   Req.Verbose = Verbose;
559ffa9d2e4SAditya Nandakumar   Req.VerboseVerbose = VerboseVerbose;
560ffa9d2e4SAditya Nandakumar   Req.NoCanonicalizeWhiteSpace = NoCanonicalizeWhiteSpace;
561ffa9d2e4SAditya Nandakumar   Req.MatchFullLines = MatchFullLines;
562*5b5b2fd2SKai Nacke   Req.IgnoreCase = IgnoreCase;
563ffa9d2e4SAditya Nandakumar 
564ffa9d2e4SAditya Nandakumar   if (VerboseVerbose)
565ffa9d2e4SAditya Nandakumar     Req.Verbose = true;
566ffa9d2e4SAditya Nandakumar 
567ffa9d2e4SAditya Nandakumar   FileCheck FC(Req);
568ffa9d2e4SAditya Nandakumar   if (!FC.ValidateCheckPrefixes()) {
56913df4626SMatt Arsenault     errs() << "Supplied check-prefix is invalid! Prefixes must be unique and "
57013df4626SMatt Arsenault               "start with a letter and contain only alphanumeric characters, "
57113df4626SMatt Arsenault               "hyphens and underscores\n";
572c2735158SRui Ueyama     return 2;
573c2735158SRui Ueyama   }
574c2735158SRui Ueyama 
575ffa9d2e4SAditya Nandakumar   Regex PrefixRE = FC.buildCheckPrefixRegex();
576726774cbSChandler Carruth   std::string REError;
577726774cbSChandler Carruth   if (!PrefixRE.isValid(REError)) {
578726774cbSChandler Carruth     errs() << "Unable to combine check-prefix strings into a prefix regular "
579726774cbSChandler Carruth               "expression! This is likely a bug in FileCheck's verification of "
580726774cbSChandler Carruth               "the check-prefix strings. Regular expression parsing failed "
581726774cbSChandler Carruth               "with the following error: "
582726774cbSChandler Carruth            << REError << "\n";
583726774cbSChandler Carruth     return 2;
584726774cbSChandler Carruth   }
58513df4626SMatt Arsenault 
586ee3c74fbSChris Lattner   SourceMgr SM;
587ee3c74fbSChris Lattner 
588ee3c74fbSChris Lattner   // Read the expected strings from the check file.
58920247900SChandler Carruth   ErrorOr<std::unique_ptr<MemoryBuffer>> CheckFileOrErr =
59020247900SChandler Carruth       MemoryBuffer::getFileOrSTDIN(CheckFilename);
59120247900SChandler Carruth   if (std::error_code EC = CheckFileOrErr.getError()) {
59220247900SChandler Carruth     errs() << "Could not open check file '" << CheckFilename
59320247900SChandler Carruth            << "': " << EC.message() << '\n';
59420247900SChandler Carruth     return 2;
59520247900SChandler Carruth   }
59620247900SChandler Carruth   MemoryBuffer &CheckFile = *CheckFileOrErr.get();
59720247900SChandler Carruth 
59820247900SChandler Carruth   SmallString<4096> CheckFileBuffer;
599ffa9d2e4SAditya Nandakumar   StringRef CheckFileText = FC.CanonicalizeFile(CheckFile, CheckFileBuffer);
60020247900SChandler Carruth 
60120247900SChandler Carruth   SM.AddNewSourceBuffer(MemoryBuffer::getMemBuffer(
60220247900SChandler Carruth                             CheckFileText, CheckFile.getBufferIdentifier()),
60320247900SChandler Carruth                         SMLoc());
60420247900SChandler Carruth 
60502ada9bdSThomas Preud'homme   if (FC.readCheckFile(SM, CheckFileText, PrefixRE))
606ee3c74fbSChris Lattner     return 2;
607ee3c74fbSChris Lattner 
608ee3c74fbSChris Lattner   // Open the file to check and add it to SourceMgr.
60920247900SChandler Carruth   ErrorOr<std::unique_ptr<MemoryBuffer>> InputFileOrErr =
610adf21f2aSRafael Espindola       MemoryBuffer::getFileOrSTDIN(InputFilename);
61120247900SChandler Carruth   if (std::error_code EC = InputFileOrErr.getError()) {
612adf21f2aSRafael Espindola     errs() << "Could not open input file '" << InputFilename
613adf21f2aSRafael Espindola            << "': " << EC.message() << '\n';
6148e1c6477SEli Bendersky     return 2;
615ee3c74fbSChris Lattner   }
61620247900SChandler Carruth   MemoryBuffer &InputFile = *InputFileOrErr.get();
6172c3e5cdfSChris Lattner 
61820247900SChandler Carruth   if (InputFile.getBufferSize() == 0 && !AllowEmptyInput) {
619b692bed7SChris Lattner     errs() << "FileCheck error: '" << InputFilename << "' is empty.\n";
6202bd4f8b6SXinliang David Li     DumpCommandLine(argc, argv);
6218e1c6477SEli Bendersky     return 2;
622b692bed7SChris Lattner   }
623b692bed7SChris Lattner 
62420247900SChandler Carruth   SmallString<4096> InputFileBuffer;
625ffa9d2e4SAditya Nandakumar   StringRef InputFileText = FC.CanonicalizeFile(InputFile, InputFileBuffer);
6262c3e5cdfSChris Lattner 
627e8f2fb20SChandler Carruth   SM.AddNewSourceBuffer(MemoryBuffer::getMemBuffer(
628e8f2fb20SChandler Carruth                             InputFileText, InputFile.getBufferIdentifier()),
629e8f2fb20SChandler Carruth                         SMLoc());
630ee3c74fbSChris Lattner 
6313c5d267eSJoel E. Denny   if (DumpInput == DumpInputDefault)
6323c5d267eSJoel E. Denny     DumpInput = DumpInputOnFailure ? DumpInputFail : DumpInputNever;
6333c5d267eSJoel E. Denny 
6343c5d267eSJoel E. Denny   std::vector<FileCheckDiag> Diags;
63502ada9bdSThomas Preud'homme   int ExitCode = FC.checkInput(SM, InputFileText,
6363c5d267eSJoel E. Denny                                DumpInput == DumpInputNever ? nullptr : &Diags)
6373c5d267eSJoel E. Denny                      ? EXIT_SUCCESS
6383c5d267eSJoel E. Denny                      : 1;
6393c5d267eSJoel E. Denny   if (DumpInput == DumpInputAlways ||
6403c5d267eSJoel E. Denny       (ExitCode == 1 && DumpInput == DumpInputFail)) {
6413c5d267eSJoel E. Denny     errs() << "\n"
6423c5d267eSJoel E. Denny            << "Input file: "
6433c5d267eSJoel E. Denny            << (InputFilename == "-" ? "<stdin>" : InputFilename.getValue())
6443c5d267eSJoel E. Denny            << "\n"
6453c5d267eSJoel E. Denny            << "Check file: " << CheckFilename << "\n"
6463c5d267eSJoel E. Denny            << "\n"
6473c5d267eSJoel E. Denny            << "-dump-input=help describes the format of the following dump.\n"
6483c5d267eSJoel E. Denny            << "\n";
6493c5d267eSJoel E. Denny     std::vector<InputAnnotation> Annotations;
6503c5d267eSJoel E. Denny     unsigned LabelWidth;
6513c5d267eSJoel E. Denny     BuildInputAnnotations(Diags, Annotations, LabelWidth);
6527df86967SJoel E. Denny     DumpAnnotatedInput(errs(), Req, InputFileText, Annotations, LabelWidth);
6533c5d267eSJoel E. Denny   }
654346dfbe2SGeorge Karpenkov 
655346dfbe2SGeorge Karpenkov   return ExitCode;
656ee3c74fbSChris Lattner }
657