Details | Last modification | View Log | RSS feed
| Rev | Author | Line No. | Line | 
|---|---|---|---|
| 14 | pmbaty | 1 | //==-- llvm/FileCheck/FileCheck.h --------------------------------*- C++ -*-==// | 
| 2 | // | ||
| 3 | // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. | ||
| 4 | // See https://llvm.org/LICENSE.txt for license information. | ||
| 5 | // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception | ||
| 6 | // | ||
| 7 | //===----------------------------------------------------------------------===// | ||
| 8 | // | ||
| 9 | /// \file This file has some utilities to use FileCheck as an API | ||
| 10 | // | ||
| 11 | //===----------------------------------------------------------------------===// | ||
| 12 | |||
| 13 | #ifndef LLVM_FILECHECK_FILECHECK_H | ||
| 14 | #define LLVM_FILECHECK_FILECHECK_H | ||
| 15 | |||
| 16 | #include "llvm/ADT/StringRef.h" | ||
| 17 | #include "llvm/Support/Regex.h" | ||
| 18 | #include "llvm/Support/SMLoc.h" | ||
| 19 | #include <bitset> | ||
| 20 | #include <memory> | ||
| 21 | #include <string> | ||
| 22 | #include <vector> | ||
| 23 | |||
| 24 | namespace llvm { | ||
| 25 | class MemoryBuffer; | ||
| 26 | class SourceMgr; | ||
| 27 | template <typename T> class SmallVectorImpl; | ||
| 28 | |||
| 29 | /// Contains info about various FileCheck options. | ||
| 30 | struct FileCheckRequest { | ||
| 31 | std::vector<StringRef> CheckPrefixes; | ||
| 32 | std::vector<StringRef> CommentPrefixes; | ||
| 33 | bool NoCanonicalizeWhiteSpace = false; | ||
| 34 | std::vector<StringRef> ImplicitCheckNot; | ||
| 35 | std::vector<StringRef> GlobalDefines; | ||
| 36 | bool AllowEmptyInput = false; | ||
| 37 | bool AllowUnusedPrefixes = false; | ||
| 38 | bool MatchFullLines = false; | ||
| 39 | bool IgnoreCase = false; | ||
| 40 | bool IsDefaultCheckPrefix = false; | ||
| 41 | bool EnableVarScope = false; | ||
| 42 | bool AllowDeprecatedDagOverlap = false; | ||
| 43 | bool Verbose = false; | ||
| 44 | bool VerboseVerbose = false; | ||
| 45 | }; | ||
| 46 | |||
| 47 | namespace Check { | ||
| 48 | |||
| 49 | enum FileCheckKind { | ||
| 50 | CheckNone = 0, | ||
| 51 | CheckMisspelled, | ||
| 52 | CheckPlain, | ||
| 53 | CheckNext, | ||
| 54 | CheckSame, | ||
| 55 | CheckNot, | ||
| 56 | CheckDAG, | ||
| 57 | CheckLabel, | ||
| 58 | CheckEmpty, | ||
| 59 | CheckComment, | ||
| 60 | |||
| 61 |   /// Indicates the pattern only matches the end of file. This is used for | ||
| 62 |   /// trailing CHECK-NOTs. | ||
| 63 | CheckEOF, | ||
| 64 | |||
| 65 |   /// Marks when parsing found a -NOT check combined with another CHECK suffix. | ||
| 66 | CheckBadNot, | ||
| 67 | |||
| 68 |   /// Marks when parsing found a -COUNT directive with invalid count value. | ||
| 69 | CheckBadCount | ||
| 70 | }; | ||
| 71 | |||
| 72 | enum FileCheckKindModifier { | ||
| 73 |   /// Modifies directive to perform literal match. | ||
| 74 | ModifierLiteral = 0, | ||
| 75 | |||
| 76 |   // The number of modifier. | ||
| 77 | Size | ||
| 78 | }; | ||
| 79 | |||
| 80 | class FileCheckType { | ||
| 81 |   FileCheckKind Kind; | ||
| 82 | int Count; ///< optional Count for some checks | ||
| 83 |   /// Modifers for the check directive. | ||
| 84 | std::bitset<FileCheckKindModifier::Size> Modifiers; | ||
| 85 | |||
| 86 | public: | ||
| 87 | FileCheckType(FileCheckKind Kind = CheckNone) : Kind(Kind), Count(1) {} | ||
| 88 | FileCheckType(const FileCheckType &) = default; | ||
| 89 | FileCheckType &operator=(const FileCheckType &) = default; | ||
| 90 | |||
| 91 | operator FileCheckKind() const { return Kind; } | ||
| 92 | |||
| 93 | int getCount() const { return Count; } | ||
| 94 | FileCheckType &setCount(int C); | ||
| 95 | |||
| 96 | bool isLiteralMatch() const { | ||
| 97 | return Modifiers[FileCheckKindModifier::ModifierLiteral]; | ||
| 98 |   } | ||
| 99 | FileCheckType &setLiteralMatch(bool Literal = true) { | ||
| 100 | Modifiers.set(FileCheckKindModifier::ModifierLiteral, Literal); | ||
| 101 | return *this; | ||
| 102 |   } | ||
| 103 | |||
| 104 |   // \returns a description of \p Prefix. | ||
| 105 | std::string getDescription(StringRef Prefix) const; | ||
| 106 | |||
| 107 |   // \returns a description of \p Modifiers. | ||
| 108 | std::string getModifiersDescription() const; | ||
| 109 | }; | ||
| 110 | } // namespace Check | ||
| 111 | |||
| 112 | /// Summary of a FileCheck diagnostic. | ||
| 113 | struct FileCheckDiag { | ||
| 114 |   /// What is the FileCheck directive for this diagnostic? | ||
| 115 | Check::FileCheckType CheckTy; | ||
| 116 |   /// Where is the FileCheck directive for this diagnostic? | ||
| 117 |   SMLoc CheckLoc; | ||
| 118 |   /// What type of match result does this diagnostic describe? | ||
| 119 |   /// | ||
| 120 |   /// A directive's supplied pattern is said to be either expected or excluded | ||
| 121 |   /// depending on whether the pattern must have or must not have a match in | ||
| 122 |   /// order for the directive to succeed.  For example, a CHECK directive's | ||
| 123 |   /// pattern is expected, and a CHECK-NOT directive's pattern is excluded. | ||
| 124 |   /// | ||
| 125 |   /// There might be more than one match result for a single pattern.  For | ||
| 126 |   /// example, there might be several discarded matches | ||
| 127 |   /// (MatchFoundButDiscarded) before either a good match | ||
| 128 |   /// (MatchFoundAndExpected) or a failure to match (MatchNoneButExpected), | ||
| 129 |   /// and there might be a fuzzy match (MatchFuzzy) after the latter. | ||
| 130 | enum MatchType { | ||
| 131 |     /// Indicates a good match for an expected pattern. | ||
| 132 | MatchFoundAndExpected, | ||
| 133 |     /// Indicates a match for an excluded pattern. | ||
| 134 | MatchFoundButExcluded, | ||
| 135 |     /// Indicates a match for an expected pattern, but the match is on the | ||
| 136 |     /// wrong line. | ||
| 137 | MatchFoundButWrongLine, | ||
| 138 |     /// Indicates a discarded match for an expected pattern. | ||
| 139 | MatchFoundButDiscarded, | ||
| 140 |     /// Indicates an error while processing a match after the match was found | ||
| 141 |     /// for an expected or excluded pattern.  The error is specified by \c Note, | ||
| 142 |     /// to which it should be appropriate to prepend "error: " later.  The full | ||
| 143 |     /// match itself should be recorded in a preceding diagnostic of a different | ||
| 144 |     /// \c MatchFound match type. | ||
| 145 | MatchFoundErrorNote, | ||
| 146 |     /// Indicates no match for an excluded pattern. | ||
| 147 | MatchNoneAndExcluded, | ||
| 148 |     /// Indicates no match for an expected pattern, but this might follow good | ||
| 149 |     /// matches when multiple matches are expected for the pattern, or it might | ||
| 150 |     /// follow discarded matches for the pattern. | ||
| 151 | MatchNoneButExpected, | ||
| 152 |     /// Indicates no match due to an expected or excluded pattern that has | ||
| 153 |     /// proven to be invalid at match time.  The exact problems are usually | ||
| 154 |     /// reported in subsequent diagnostics of the same match type but with | ||
| 155 |     /// \c Note set. | ||
| 156 | MatchNoneForInvalidPattern, | ||
| 157 |     /// Indicates a fuzzy match that serves as a suggestion for the next | ||
| 158 |     /// intended match for an expected pattern with too few or no good matches. | ||
| 159 | MatchFuzzy, | ||
| 160 | } MatchTy; | ||
| 161 |   /// The search range if MatchTy starts with MatchNone, or the match range | ||
| 162 |   /// otherwise. | ||
| 163 | unsigned InputStartLine; | ||
| 164 | unsigned InputStartCol; | ||
| 165 | unsigned InputEndLine; | ||
| 166 | unsigned InputEndCol; | ||
| 167 |   /// A note to replace the one normally indicated by MatchTy, or the empty | ||
| 168 |   /// string if none. | ||
| 169 | std::string Note; | ||
| 170 | FileCheckDiag(const SourceMgr &SM, const Check::FileCheckType &CheckTy, | ||
| 171 | SMLoc CheckLoc, MatchType MatchTy, SMRange InputRange, | ||
| 172 | StringRef Note = ""); | ||
| 173 | }; | ||
| 174 | |||
| 175 | class FileCheckPatternContext; | ||
| 176 | struct FileCheckString; | ||
| 177 | |||
| 178 | /// FileCheck class takes the request and exposes various methods that | ||
| 179 | /// use information from the request. | ||
| 180 | class FileCheck { | ||
| 181 |   FileCheckRequest Req; | ||
| 182 | std::unique_ptr<FileCheckPatternContext> PatternContext; | ||
| 183 |   // C++17 TODO: make this a plain std::vector. | ||
| 184 | std::unique_ptr<std::vector<FileCheckString>> CheckStrings; | ||
| 185 | |||
| 186 | public: | ||
| 187 | explicit FileCheck(FileCheckRequest Req); | ||
| 188 | ~FileCheck(); | ||
| 189 | |||
| 190 |   // Combines the check prefixes into a single regex so that we can efficiently | ||
| 191 |   // scan for any of the set. | ||
| 192 |   // | ||
| 193 |   // The semantics are that the longest-match wins which matches our regex | ||
| 194 |   // library. | ||
| 195 | Regex buildCheckPrefixRegex(); | ||
| 196 | |||
| 197 |   /// Reads the check file from \p Buffer and records the expected strings it | ||
| 198 |   /// contains. Errors are reported against \p SM. | ||
| 199 |   /// | ||
| 200 |   /// Only expected strings whose prefix is one of those listed in \p PrefixRE | ||
| 201 |   /// are recorded. \returns true in case of an error, false otherwise. | ||
| 202 |   /// | ||
| 203 |   /// If \p ImpPatBufferIDRange, then the range (inclusive start, exclusive end) | ||
| 204 |   /// of IDs for source buffers added to \p SM for implicit patterns are | ||
| 205 |   /// recorded in it.  The range is empty if there are none. | ||
| 206 |   bool | ||
| 207 | readCheckFile(SourceMgr &SM, StringRef Buffer, Regex &PrefixRE, | ||
| 208 | std::pair<unsigned, unsigned> *ImpPatBufferIDRange = nullptr); | ||
| 209 | |||
| 210 | bool ValidateCheckPrefixes(); | ||
| 211 | |||
| 212 |   /// Canonicalizes whitespaces in the file. Line endings are replaced with | ||
| 213 |   /// UNIX-style '\n'. | ||
| 214 | StringRef CanonicalizeFile(MemoryBuffer &MB, | ||
| 215 | SmallVectorImpl<char> &OutputBuffer); | ||
| 216 | |||
| 217 |   /// Checks the input to FileCheck provided in the \p Buffer against the | ||
| 218 |   /// expected strings read from the check file and record diagnostics emitted | ||
| 219 |   /// in \p Diags. Errors are recorded against \p SM. | ||
| 220 |   /// | ||
| 221 |   /// \returns false if the input fails to satisfy the checks. | ||
| 222 | bool checkInput(SourceMgr &SM, StringRef Buffer, | ||
| 223 | std::vector<FileCheckDiag> *Diags = nullptr); | ||
| 224 | }; | ||
| 225 | |||
| 226 | } // namespace llvm | ||
| 227 | |||
| 228 | #endif |