Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 1 | //===--- PrintPreprocessedOutput.cpp - Implement the -E mode --------------===// |
| 2 | // |
| 3 | // The LLVM Compiler Infrastructure |
| 4 | // |
Chris Lattner | 0bc735f | 2007-12-29 19:59:25 +0000 | [diff] [blame] | 5 | // This file is distributed under the University of Illinois Open Source |
| 6 | // License. See LICENSE.TXT for details. |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 7 | // |
| 8 | //===----------------------------------------------------------------------===// |
| 9 | // |
| 10 | // This code simply runs the preprocessor on the input file and prints out the |
| 11 | // result. This is the traditional behavior of the -E option. |
| 12 | // |
| 13 | //===----------------------------------------------------------------------===// |
| 14 | |
Eli Friedman | b09f6e1 | 2009-05-19 04:14:29 +0000 | [diff] [blame] | 15 | #include "clang/Frontend/Utils.h" |
Daniel Dunbar | 775bee7 | 2009-11-11 10:07:22 +0000 | [diff] [blame] | 16 | #include "clang/Basic/Diagnostic.h" |
| 17 | #include "clang/Basic/SourceManager.h" |
| 18 | #include "clang/Frontend/PreprocessorOutputOptions.h" |
Chris Lattner | f73903a | 2009-02-06 06:45:26 +0000 | [diff] [blame] | 19 | #include "clang/Lex/MacroInfo.h" |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 20 | #include "clang/Lex/PPCallbacks.h" |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 21 | #include "clang/Lex/Pragma.h" |
Daniel Dunbar | 775bee7 | 2009-11-11 10:07:22 +0000 | [diff] [blame] | 22 | #include "clang/Lex/Preprocessor.h" |
Chris Lattner | d7038e1 | 2009-02-13 00:46:04 +0000 | [diff] [blame] | 23 | #include "clang/Lex/TokenConcatenation.h" |
Chris Lattner | d8e3083 | 2007-07-24 06:57:14 +0000 | [diff] [blame] | 24 | #include "llvm/ADT/SmallString.h" |
Benjamin Kramer | 3ff2a4b | 2010-01-19 17:42:20 +0000 | [diff] [blame] | 25 | #include "llvm/ADT/STLExtras.h" |
Chris Lattner | abfe094 | 2010-06-26 17:11:39 +0000 | [diff] [blame] | 26 | #include "llvm/ADT/StringRef.h" |
Chris Lattner | dceb6a7 | 2008-08-17 01:47:12 +0000 | [diff] [blame] | 27 | #include "llvm/Support/raw_ostream.h" |
Douglas Gregor | c09ce12 | 2011-06-22 19:41:48 +0000 | [diff] [blame] | 28 | #include "llvm/Support/ErrorHandling.h" |
Joerg Sonnenberger | 7094dee | 2012-08-10 10:58:18 +0000 | [diff] [blame] | 29 | #include <cctype> |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 30 | #include <cstdio> |
| 31 | using namespace clang; |
| 32 | |
Chris Lattner | d82df3a | 2009-04-12 01:56:53 +0000 | [diff] [blame] | 33 | /// PrintMacroDefinition - Print a macro definition in a form that will be |
| 34 | /// properly accepted back as a definition. |
| 35 | static void PrintMacroDefinition(const IdentifierInfo &II, const MacroInfo &MI, |
Chris Lattner | 5f9e272 | 2011-07-23 10:55:15 +0000 | [diff] [blame] | 36 | Preprocessor &PP, raw_ostream &OS) { |
Chris Lattner | d82df3a | 2009-04-12 01:56:53 +0000 | [diff] [blame] | 37 | OS << "#define " << II.getName(); |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 38 | |
Chris Lattner | d82df3a | 2009-04-12 01:56:53 +0000 | [diff] [blame] | 39 | if (MI.isFunctionLike()) { |
| 40 | OS << '('; |
Chris Lattner | 13d5558 | 2009-12-09 02:08:14 +0000 | [diff] [blame] | 41 | if (!MI.arg_empty()) { |
Chris Lattner | d82df3a | 2009-04-12 01:56:53 +0000 | [diff] [blame] | 42 | MacroInfo::arg_iterator AI = MI.arg_begin(), E = MI.arg_end(); |
Chris Lattner | 13d5558 | 2009-12-09 02:08:14 +0000 | [diff] [blame] | 43 | for (; AI+1 != E; ++AI) { |
| 44 | OS << (*AI)->getName(); |
Chris Lattner | 807b93e | 2009-12-07 01:58:34 +0000 | [diff] [blame] | 45 | OS << ','; |
Chris Lattner | 807b93e | 2009-12-07 01:58:34 +0000 | [diff] [blame] | 46 | } |
Chris Lattner | 13d5558 | 2009-12-09 02:08:14 +0000 | [diff] [blame] | 47 | |
| 48 | // Last argument. |
| 49 | if ((*AI)->getName() == "__VA_ARGS__") |
| 50 | OS << "..."; |
| 51 | else |
| 52 | OS << (*AI)->getName(); |
Chris Lattner | d82df3a | 2009-04-12 01:56:53 +0000 | [diff] [blame] | 53 | } |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 54 | |
Chris Lattner | 807b93e | 2009-12-07 01:58:34 +0000 | [diff] [blame] | 55 | if (MI.isGNUVarargs()) |
| 56 | OS << "..."; // #define foo(x...) |
Kovarththanan Rajaratnam | a5d10c4 | 2010-03-07 07:30:06 +0000 | [diff] [blame] | 57 | |
Chris Lattner | d82df3a | 2009-04-12 01:56:53 +0000 | [diff] [blame] | 58 | OS << ')'; |
| 59 | } |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 60 | |
Chris Lattner | d82df3a | 2009-04-12 01:56:53 +0000 | [diff] [blame] | 61 | // GCC always emits a space, even if the macro body is empty. However, do not |
| 62 | // want to emit two spaces if the first token has a leading space. |
| 63 | if (MI.tokens_empty() || !MI.tokens_begin()->hasLeadingSpace()) |
| 64 | OS << ' '; |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 65 | |
Dylan Noblesmith | f7ccbad | 2012-02-05 02:13:05 +0000 | [diff] [blame] | 66 | SmallString<128> SpellingBuffer; |
Chris Lattner | d82df3a | 2009-04-12 01:56:53 +0000 | [diff] [blame] | 67 | for (MacroInfo::tokens_iterator I = MI.tokens_begin(), E = MI.tokens_end(); |
| 68 | I != E; ++I) { |
| 69 | if (I->hasLeadingSpace()) |
| 70 | OS << ' '; |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 71 | |
Benjamin Kramer | ddeea56 | 2010-02-27 13:44:12 +0000 | [diff] [blame] | 72 | OS << PP.getSpelling(*I, SpellingBuffer); |
Chris Lattner | d82df3a | 2009-04-12 01:56:53 +0000 | [diff] [blame] | 73 | } |
| 74 | } |
| 75 | |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 76 | //===----------------------------------------------------------------------===// |
| 77 | // Preprocessed token printer |
| 78 | //===----------------------------------------------------------------------===// |
| 79 | |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 80 | namespace { |
| 81 | class PrintPPOutputPPCallbacks : public PPCallbacks { |
| 82 | Preprocessor &PP; |
Chris Lattner | df72140 | 2010-04-13 00:06:42 +0000 | [diff] [blame] | 83 | SourceManager &SM; |
Chris Lattner | d7038e1 | 2009-02-13 00:46:04 +0000 | [diff] [blame] | 84 | TokenConcatenation ConcatInfo; |
Chris Lattner | e96de3e | 2008-08-17 03:12:02 +0000 | [diff] [blame] | 85 | public: |
Chris Lattner | 5f9e272 | 2011-07-23 10:55:15 +0000 | [diff] [blame] | 86 | raw_ostream &OS; |
Chris Lattner | e96de3e | 2008-08-17 03:12:02 +0000 | [diff] [blame] | 87 | private: |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 88 | unsigned CurLine; |
Daniel Dunbar | f7c16d9 | 2010-08-24 22:44:13 +0000 | [diff] [blame] | 89 | |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 90 | bool EmittedTokensOnThisLine; |
Jordan Rose | 0cdd1fe | 2012-06-15 23:33:51 +0000 | [diff] [blame] | 91 | bool EmittedDirectiveOnThisLine; |
Chris Lattner | 9d72851 | 2008-10-27 01:19:25 +0000 | [diff] [blame] | 92 | SrcMgr::CharacteristicKind FileType; |
Dylan Noblesmith | f7ccbad | 2012-02-05 02:13:05 +0000 | [diff] [blame] | 93 | SmallString<512> CurFilename; |
Daniel Dunbar | 737bdb4 | 2008-09-05 03:22:57 +0000 | [diff] [blame] | 94 | bool Initialized; |
Eli Friedman | 12d3b1d | 2009-05-19 03:06:47 +0000 | [diff] [blame] | 95 | bool DisableLineMarkers; |
| 96 | bool DumpDefines; |
Chris Lattner | f744934 | 2009-12-07 01:42:56 +0000 | [diff] [blame] | 97 | bool UseLineDirective; |
Daniel Dunbar | 2a9be3e | 2012-11-16 01:51:11 +0000 | [diff] [blame^] | 98 | bool IsFirstFileEntered; |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 99 | public: |
Chris Lattner | 5f9e272 | 2011-07-23 10:55:15 +0000 | [diff] [blame] | 100 | PrintPPOutputPPCallbacks(Preprocessor &pp, raw_ostream &os, |
Daniel Dunbar | eef63e0 | 2011-02-02 15:41:17 +0000 | [diff] [blame] | 101 | bool lineMarkers, bool defines) |
Chris Lattner | df72140 | 2010-04-13 00:06:42 +0000 | [diff] [blame] | 102 | : PP(pp), SM(PP.getSourceManager()), |
| 103 | ConcatInfo(PP), OS(os), DisableLineMarkers(lineMarkers), |
Daniel Dunbar | eef63e0 | 2011-02-02 15:41:17 +0000 | [diff] [blame] | 104 | DumpDefines(defines) { |
| 105 | CurLine = 0; |
Chris Lattner | d8e3083 | 2007-07-24 06:57:14 +0000 | [diff] [blame] | 106 | CurFilename += "<uninit>"; |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 107 | EmittedTokensOnThisLine = false; |
Jordan Rose | 0cdd1fe | 2012-06-15 23:33:51 +0000 | [diff] [blame] | 108 | EmittedDirectiveOnThisLine = false; |
Chris Lattner | 0b9e736 | 2008-09-26 21:18:42 +0000 | [diff] [blame] | 109 | FileType = SrcMgr::C_User; |
Daniel Dunbar | 737bdb4 | 2008-09-05 03:22:57 +0000 | [diff] [blame] | 110 | Initialized = false; |
Daniel Dunbar | 2a9be3e | 2012-11-16 01:51:11 +0000 | [diff] [blame^] | 111 | IsFirstFileEntered = false; |
Kovarththanan Rajaratnam | a5d10c4 | 2010-03-07 07:30:06 +0000 | [diff] [blame] | 112 | |
Chris Lattner | f744934 | 2009-12-07 01:42:56 +0000 | [diff] [blame] | 113 | // If we're in microsoft mode, use normal #line instead of line markers. |
David Blaikie | 4e4d084 | 2012-03-11 07:00:24 +0000 | [diff] [blame] | 114 | UseLineDirective = PP.getLangOpts().MicrosoftExt; |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 115 | } |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 116 | |
Jordan Rose | 0cdd1fe | 2012-06-15 23:33:51 +0000 | [diff] [blame] | 117 | void setEmittedTokensOnThisLine() { EmittedTokensOnThisLine = true; } |
Chris Lattner | f0f2b29 | 2007-07-23 06:09:34 +0000 | [diff] [blame] | 118 | bool hasEmittedTokensOnThisLine() const { return EmittedTokensOnThisLine; } |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 119 | |
Jordan Rose | 0cdd1fe | 2012-06-15 23:33:51 +0000 | [diff] [blame] | 120 | void setEmittedDirectiveOnThisLine() { EmittedDirectiveOnThisLine = true; } |
| 121 | bool hasEmittedDirectiveOnThisLine() const { |
| 122 | return EmittedDirectiveOnThisLine; |
| 123 | } |
| 124 | |
| 125 | bool startNewLineIfNeeded(bool ShouldUpdateCurrentLine = true); |
Douglas Gregor | 80c60f7 | 2010-09-09 22:45:38 +0000 | [diff] [blame] | 126 | |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 127 | virtual void FileChanged(SourceLocation Loc, FileChangeReason Reason, |
Argyrios Kyrtzidis | c892c5f | 2011-10-11 17:29:44 +0000 | [diff] [blame] | 128 | SrcMgr::CharacteristicKind FileType, |
| 129 | FileID PrevFID); |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 130 | virtual void Ident(SourceLocation Loc, const std::string &str); |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 131 | virtual void PragmaComment(SourceLocation Loc, const IdentifierInfo *Kind, |
Chris Lattner | c7d945d | 2009-01-16 19:25:54 +0000 | [diff] [blame] | 132 | const std::string &Str); |
Chris Lattner | 5f9e272 | 2011-07-23 10:55:15 +0000 | [diff] [blame] | 133 | virtual void PragmaMessage(SourceLocation Loc, StringRef Str); |
Douglas Gregor | c09ce12 | 2011-06-22 19:41:48 +0000 | [diff] [blame] | 134 | virtual void PragmaDiagnosticPush(SourceLocation Loc, |
Chris Lattner | 5f9e272 | 2011-07-23 10:55:15 +0000 | [diff] [blame] | 135 | StringRef Namespace); |
Douglas Gregor | c09ce12 | 2011-06-22 19:41:48 +0000 | [diff] [blame] | 136 | virtual void PragmaDiagnosticPop(SourceLocation Loc, |
Chris Lattner | 5f9e272 | 2011-07-23 10:55:15 +0000 | [diff] [blame] | 137 | StringRef Namespace); |
| 138 | virtual void PragmaDiagnostic(SourceLocation Loc, StringRef Namespace, |
Benjamin Kramer | d7a3e2c | 2012-02-07 22:29:24 +0000 | [diff] [blame] | 139 | diag::Mapping Map, StringRef Str); |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 140 | |
Chris Lattner | 5f18032 | 2007-12-09 21:11:08 +0000 | [diff] [blame] | 141 | bool HandleFirstTokOnLine(Token &Tok); |
Chris Lattner | 88aae91 | 2010-04-13 00:01:41 +0000 | [diff] [blame] | 142 | bool MoveToLine(SourceLocation Loc) { |
Douglas Gregor | cb7b1e1 | 2010-11-12 07:15:47 +0000 | [diff] [blame] | 143 | PresumedLoc PLoc = SM.getPresumedLoc(Loc); |
| 144 | if (PLoc.isInvalid()) |
| 145 | return false; |
| 146 | return MoveToLine(PLoc.getLine()); |
Chris Lattner | 88aae91 | 2010-04-13 00:01:41 +0000 | [diff] [blame] | 147 | } |
| 148 | bool MoveToLine(unsigned LineNo); |
| 149 | |
Chris Lattner | 8877321 | 2010-04-14 03:57:19 +0000 | [diff] [blame] | 150 | bool AvoidConcat(const Token &PrevPrevTok, const Token &PrevTok, |
| 151 | const Token &Tok) { |
| 152 | return ConcatInfo.AvoidConcat(PrevPrevTok, PrevTok, Tok); |
Chris Lattner | d7038e1 | 2009-02-13 00:46:04 +0000 | [diff] [blame] | 153 | } |
Daniel Dunbar | 737bdb4 | 2008-09-05 03:22:57 +0000 | [diff] [blame] | 154 | void WriteLineInfo(unsigned LineNo, const char *Extra=0, unsigned ExtraLen=0); |
Douglas Gregor | 80c60f7 | 2010-09-09 22:45:38 +0000 | [diff] [blame] | 155 | bool LineMarkersAreDisabled() const { return DisableLineMarkers; } |
Chris Lattner | 3ee211f | 2009-06-15 01:25:23 +0000 | [diff] [blame] | 156 | void HandleNewlinesInToken(const char *TokStr, unsigned Len); |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 157 | |
Chris Lattner | d82df3a | 2009-04-12 01:56:53 +0000 | [diff] [blame] | 158 | /// MacroDefined - This hook is called whenever a macro definition is seen. |
Craig Silverstein | 2aa9267 | 2010-11-19 21:33:15 +0000 | [diff] [blame] | 159 | void MacroDefined(const Token &MacroNameTok, const MacroInfo *MI); |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 160 | |
Benjamin Kramer | 2f05449 | 2010-08-07 22:27:00 +0000 | [diff] [blame] | 161 | /// MacroUndefined - This hook is called whenever a macro #undef is seen. |
Craig Silverstein | 2aa9267 | 2010-11-19 21:33:15 +0000 | [diff] [blame] | 162 | void MacroUndefined(const Token &MacroNameTok, const MacroInfo *MI); |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 163 | }; |
Chris Lattner | 5db17c9 | 2008-04-08 04:16:20 +0000 | [diff] [blame] | 164 | } // end anonymous namespace |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 165 | |
Daniel Dunbar | 737bdb4 | 2008-09-05 03:22:57 +0000 | [diff] [blame] | 166 | void PrintPPOutputPPCallbacks::WriteLineInfo(unsigned LineNo, |
| 167 | const char *Extra, |
| 168 | unsigned ExtraLen) { |
Jordan Rose | 0cdd1fe | 2012-06-15 23:33:51 +0000 | [diff] [blame] | 169 | startNewLineIfNeeded(/*ShouldUpdateCurrentLine=*/false); |
Daniel Dunbar | 737bdb4 | 2008-09-05 03:22:57 +0000 | [diff] [blame] | 170 | |
Chris Lattner | f744934 | 2009-12-07 01:42:56 +0000 | [diff] [blame] | 171 | // Emit #line directives or GNU line markers depending on what mode we're in. |
| 172 | if (UseLineDirective) { |
| 173 | OS << "#line" << ' ' << LineNo << ' ' << '"'; |
Ted Kremenek | 2346513 | 2010-09-17 00:41:18 +0000 | [diff] [blame] | 174 | OS.write(CurFilename.data(), CurFilename.size()); |
Chris Lattner | f744934 | 2009-12-07 01:42:56 +0000 | [diff] [blame] | 175 | OS << '"'; |
| 176 | } else { |
| 177 | OS << '#' << ' ' << LineNo << ' ' << '"'; |
Ted Kremenek | 2346513 | 2010-09-17 00:41:18 +0000 | [diff] [blame] | 178 | OS.write(CurFilename.data(), CurFilename.size()); |
Chris Lattner | f744934 | 2009-12-07 01:42:56 +0000 | [diff] [blame] | 179 | OS << '"'; |
Kovarththanan Rajaratnam | a5d10c4 | 2010-03-07 07:30:06 +0000 | [diff] [blame] | 180 | |
Steve Naroff | d76fbda | 2009-12-06 01:02:14 +0000 | [diff] [blame] | 181 | if (ExtraLen) |
| 182 | OS.write(Extra, ExtraLen); |
Daniel Dunbar | 737bdb4 | 2008-09-05 03:22:57 +0000 | [diff] [blame] | 183 | |
Steve Naroff | d76fbda | 2009-12-06 01:02:14 +0000 | [diff] [blame] | 184 | if (FileType == SrcMgr::C_System) |
| 185 | OS.write(" 3", 2); |
| 186 | else if (FileType == SrcMgr::C_ExternCSystem) |
| 187 | OS.write(" 3 4", 4); |
| 188 | } |
Daniel Dunbar | 737bdb4 | 2008-09-05 03:22:57 +0000 | [diff] [blame] | 189 | OS << '\n'; |
| 190 | } |
| 191 | |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 192 | /// MoveToLine - Move the output to the source line specified by the location |
| 193 | /// object. We can do this by emitting some number of \n's, or be emitting a |
Chris Lattner | 5f18032 | 2007-12-09 21:11:08 +0000 | [diff] [blame] | 194 | /// #line directive. This returns false if already at the specified line, true |
| 195 | /// if some newlines were emitted. |
Chris Lattner | 88aae91 | 2010-04-13 00:01:41 +0000 | [diff] [blame] | 196 | bool PrintPPOutputPPCallbacks::MoveToLine(unsigned LineNo) { |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 197 | // If this line is "close enough" to the original line, just print newlines, |
| 198 | // otherwise print a #line directive. |
Daniel Dunbar | fd96684 | 2008-09-26 01:13:35 +0000 | [diff] [blame] | 199 | if (LineNo-CurLine <= 8) { |
Chris Lattner | 822f940 | 2007-07-23 05:14:05 +0000 | [diff] [blame] | 200 | if (LineNo-CurLine == 1) |
Chris Lattner | e96de3e | 2008-08-17 03:12:02 +0000 | [diff] [blame] | 201 | OS << '\n'; |
Chris Lattner | 5f18032 | 2007-12-09 21:11:08 +0000 | [diff] [blame] | 202 | else if (LineNo == CurLine) |
Chandler Carruth | 9f79a1f | 2011-07-14 16:14:52 +0000 | [diff] [blame] | 203 | return false; // Spelling line moved, but expansion line didn't. |
Chris Lattner | 822f940 | 2007-07-23 05:14:05 +0000 | [diff] [blame] | 204 | else { |
| 205 | const char *NewLines = "\n\n\n\n\n\n\n\n"; |
Chris Lattner | e96de3e | 2008-08-17 03:12:02 +0000 | [diff] [blame] | 206 | OS.write(NewLines, LineNo-CurLine); |
Chris Lattner | 822f940 | 2007-07-23 05:14:05 +0000 | [diff] [blame] | 207 | } |
Chris Lattner | 6133aeb | 2010-06-12 16:20:56 +0000 | [diff] [blame] | 208 | } else if (!DisableLineMarkers) { |
| 209 | // Emit a #line or line marker. |
Daniel Dunbar | 737bdb4 | 2008-09-05 03:22:57 +0000 | [diff] [blame] | 210 | WriteLineInfo(LineNo, 0, 0); |
Chris Lattner | 6133aeb | 2010-06-12 16:20:56 +0000 | [diff] [blame] | 211 | } else { |
| 212 | // Okay, we're in -P mode, which turns off line markers. However, we still |
| 213 | // need to emit a newline between tokens on different lines. |
Jordan Rose | 0cdd1fe | 2012-06-15 23:33:51 +0000 | [diff] [blame] | 214 | startNewLineIfNeeded(/*ShouldUpdateCurrentLine=*/false); |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 215 | } |
Daniel Dunbar | 737bdb4 | 2008-09-05 03:22:57 +0000 | [diff] [blame] | 216 | |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 217 | CurLine = LineNo; |
Chris Lattner | 5f18032 | 2007-12-09 21:11:08 +0000 | [diff] [blame] | 218 | return true; |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 219 | } |
| 220 | |
Jordan Rose | 0cdd1fe | 2012-06-15 23:33:51 +0000 | [diff] [blame] | 221 | bool |
| 222 | PrintPPOutputPPCallbacks::startNewLineIfNeeded(bool ShouldUpdateCurrentLine) { |
| 223 | if (EmittedTokensOnThisLine || EmittedDirectiveOnThisLine) { |
Douglas Gregor | 80c60f7 | 2010-09-09 22:45:38 +0000 | [diff] [blame] | 224 | OS << '\n'; |
| 225 | EmittedTokensOnThisLine = false; |
Jordan Rose | 0cdd1fe | 2012-06-15 23:33:51 +0000 | [diff] [blame] | 226 | EmittedDirectiveOnThisLine = false; |
| 227 | if (ShouldUpdateCurrentLine) |
| 228 | ++CurLine; |
Douglas Gregor | 80c60f7 | 2010-09-09 22:45:38 +0000 | [diff] [blame] | 229 | return true; |
| 230 | } |
| 231 | |
| 232 | return false; |
| 233 | } |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 234 | |
| 235 | /// FileChanged - Whenever the preprocessor enters or exits a #include file |
| 236 | /// it invokes this handler. Update our conception of the current source |
| 237 | /// position. |
| 238 | void PrintPPOutputPPCallbacks::FileChanged(SourceLocation Loc, |
| 239 | FileChangeReason Reason, |
Argyrios Kyrtzidis | c892c5f | 2011-10-11 17:29:44 +0000 | [diff] [blame] | 240 | SrcMgr::CharacteristicKind NewFileType, |
| 241 | FileID PrevFID) { |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 242 | // Unless we are exiting a #include, make sure to skip ahead to the line the |
| 243 | // #include directive was at. |
Chris Lattner | df72140 | 2010-04-13 00:06:42 +0000 | [diff] [blame] | 244 | SourceManager &SourceMgr = SM; |
Chris Lattner | 88aae91 | 2010-04-13 00:01:41 +0000 | [diff] [blame] | 245 | |
| 246 | PresumedLoc UserLoc = SourceMgr.getPresumedLoc(Loc); |
Douglas Gregor | cb7b1e1 | 2010-11-12 07:15:47 +0000 | [diff] [blame] | 247 | if (UserLoc.isInvalid()) |
| 248 | return; |
| 249 | |
Chris Lattner | 86d0ef7 | 2010-04-14 04:28:50 +0000 | [diff] [blame] | 250 | unsigned NewLine = UserLoc.getLine(); |
Daniel Dunbar | f7c16d9 | 2010-08-24 22:44:13 +0000 | [diff] [blame] | 251 | |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 252 | if (Reason == PPCallbacks::EnterFile) { |
Douglas Gregor | cb7b1e1 | 2010-11-12 07:15:47 +0000 | [diff] [blame] | 253 | SourceLocation IncludeLoc = UserLoc.getIncludeLoc(); |
Chris Lattner | 71d8bfb | 2009-01-30 18:44:17 +0000 | [diff] [blame] | 254 | if (IncludeLoc.isValid()) |
| 255 | MoveToLine(IncludeLoc); |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 256 | } else if (Reason == PPCallbacks::SystemHeaderPragma) { |
Chris Lattner | 88aae91 | 2010-04-13 00:01:41 +0000 | [diff] [blame] | 257 | MoveToLine(NewLine); |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 258 | |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 259 | // TODO GCC emits the # directive for this directive on the line AFTER the |
| 260 | // directive and emits a bunch of spaces that aren't needed. Emulate this |
| 261 | // strange behavior. |
| 262 | } |
Chris Lattner | 88aae91 | 2010-04-13 00:01:41 +0000 | [diff] [blame] | 263 | |
| 264 | CurLine = NewLine; |
Daniel Dunbar | 737bdb4 | 2008-09-05 03:22:57 +0000 | [diff] [blame] | 265 | |
Chris Lattner | d8e3083 | 2007-07-24 06:57:14 +0000 | [diff] [blame] | 266 | CurFilename.clear(); |
Chris Lattner | 88aae91 | 2010-04-13 00:01:41 +0000 | [diff] [blame] | 267 | CurFilename += UserLoc.getFilename(); |
Chris Lattner | d8e3083 | 2007-07-24 06:57:14 +0000 | [diff] [blame] | 268 | Lexer::Stringify(CurFilename); |
Daniel Dunbar | 737bdb4 | 2008-09-05 03:22:57 +0000 | [diff] [blame] | 269 | FileType = NewFileType; |
| 270 | |
Daniel Dunbar | f7c16d9 | 2010-08-24 22:44:13 +0000 | [diff] [blame] | 271 | if (DisableLineMarkers) return; |
| 272 | |
Daniel Dunbar | 737bdb4 | 2008-09-05 03:22:57 +0000 | [diff] [blame] | 273 | if (!Initialized) { |
| 274 | WriteLineInfo(CurLine); |
| 275 | Initialized = true; |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 276 | } |
Daniel Dunbar | 737bdb4 | 2008-09-05 03:22:57 +0000 | [diff] [blame] | 277 | |
Daniel Dunbar | 2a9be3e | 2012-11-16 01:51:11 +0000 | [diff] [blame^] | 278 | // Do not emit an enter marker for the main file (which we expect is the first |
| 279 | // entered file). This matches gcc, and improves compatibility with some tools |
| 280 | // which track the # line markers as a way to determine when the preprocessed |
| 281 | // output is in the context of the main file. |
| 282 | if (Reason == PPCallbacks::EnterFile && !IsFirstFileEntered) { |
| 283 | IsFirstFileEntered = true; |
| 284 | return; |
| 285 | } |
| 286 | |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 287 | switch (Reason) { |
| 288 | case PPCallbacks::EnterFile: |
Daniel Dunbar | 737bdb4 | 2008-09-05 03:22:57 +0000 | [diff] [blame] | 289 | WriteLineInfo(CurLine, " 1", 2); |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 290 | break; |
| 291 | case PPCallbacks::ExitFile: |
Daniel Dunbar | 737bdb4 | 2008-09-05 03:22:57 +0000 | [diff] [blame] | 292 | WriteLineInfo(CurLine, " 2", 2); |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 293 | break; |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 294 | case PPCallbacks::SystemHeaderPragma: |
| 295 | case PPCallbacks::RenameFile: |
Daniel Dunbar | 737bdb4 | 2008-09-05 03:22:57 +0000 | [diff] [blame] | 296 | WriteLineInfo(CurLine); |
| 297 | break; |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 298 | } |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 299 | } |
| 300 | |
Chris Lattner | c7d945d | 2009-01-16 19:25:54 +0000 | [diff] [blame] | 301 | /// Ident - Handle #ident directives when read by the preprocessor. |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 302 | /// |
| 303 | void PrintPPOutputPPCallbacks::Ident(SourceLocation Loc, const std::string &S) { |
| 304 | MoveToLine(Loc); |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 305 | |
Chris Lattner | e96de3e | 2008-08-17 03:12:02 +0000 | [diff] [blame] | 306 | OS.write("#ident ", strlen("#ident ")); |
| 307 | OS.write(&S[0], S.size()); |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 308 | EmittedTokensOnThisLine = true; |
| 309 | } |
| 310 | |
Chris Lattner | d82df3a | 2009-04-12 01:56:53 +0000 | [diff] [blame] | 311 | /// MacroDefined - This hook is called whenever a macro definition is seen. |
Craig Silverstein | 2aa9267 | 2010-11-19 21:33:15 +0000 | [diff] [blame] | 312 | void PrintPPOutputPPCallbacks::MacroDefined(const Token &MacroNameTok, |
Chris Lattner | d82df3a | 2009-04-12 01:56:53 +0000 | [diff] [blame] | 313 | const MacroInfo *MI) { |
| 314 | // Only print out macro definitions in -dD mode. |
| 315 | if (!DumpDefines || |
| 316 | // Ignore __FILE__ etc. |
| 317 | MI->isBuiltinMacro()) return; |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 318 | |
Chris Lattner | d82df3a | 2009-04-12 01:56:53 +0000 | [diff] [blame] | 319 | MoveToLine(MI->getDefinitionLoc()); |
Craig Silverstein | 2aa9267 | 2010-11-19 21:33:15 +0000 | [diff] [blame] | 320 | PrintMacroDefinition(*MacroNameTok.getIdentifierInfo(), *MI, PP, OS); |
Jordan Rose | 0cdd1fe | 2012-06-15 23:33:51 +0000 | [diff] [blame] | 321 | setEmittedDirectiveOnThisLine(); |
Chris Lattner | d82df3a | 2009-04-12 01:56:53 +0000 | [diff] [blame] | 322 | } |
| 323 | |
Craig Silverstein | 2aa9267 | 2010-11-19 21:33:15 +0000 | [diff] [blame] | 324 | void PrintPPOutputPPCallbacks::MacroUndefined(const Token &MacroNameTok, |
Benjamin Kramer | 2f05449 | 2010-08-07 22:27:00 +0000 | [diff] [blame] | 325 | const MacroInfo *MI) { |
| 326 | // Only print out macro definitions in -dD mode. |
| 327 | if (!DumpDefines) return; |
| 328 | |
Craig Silverstein | 2aa9267 | 2010-11-19 21:33:15 +0000 | [diff] [blame] | 329 | MoveToLine(MacroNameTok.getLocation()); |
| 330 | OS << "#undef " << MacroNameTok.getIdentifierInfo()->getName(); |
Jordan Rose | 0cdd1fe | 2012-06-15 23:33:51 +0000 | [diff] [blame] | 331 | setEmittedDirectiveOnThisLine(); |
Benjamin Kramer | 2f05449 | 2010-08-07 22:27:00 +0000 | [diff] [blame] | 332 | } |
Chris Lattner | d82df3a | 2009-04-12 01:56:53 +0000 | [diff] [blame] | 333 | |
Chris Lattner | c7d945d | 2009-01-16 19:25:54 +0000 | [diff] [blame] | 334 | void PrintPPOutputPPCallbacks::PragmaComment(SourceLocation Loc, |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 335 | const IdentifierInfo *Kind, |
Chris Lattner | c7d945d | 2009-01-16 19:25:54 +0000 | [diff] [blame] | 336 | const std::string &Str) { |
Jordan Rose | 0cdd1fe | 2012-06-15 23:33:51 +0000 | [diff] [blame] | 337 | startNewLineIfNeeded(); |
Chris Lattner | c7d945d | 2009-01-16 19:25:54 +0000 | [diff] [blame] | 338 | MoveToLine(Loc); |
| 339 | OS << "#pragma comment(" << Kind->getName(); |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 340 | |
Chris Lattner | c7d945d | 2009-01-16 19:25:54 +0000 | [diff] [blame] | 341 | if (!Str.empty()) { |
| 342 | OS << ", \""; |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 343 | |
Chris Lattner | c7d945d | 2009-01-16 19:25:54 +0000 | [diff] [blame] | 344 | for (unsigned i = 0, e = Str.size(); i != e; ++i) { |
| 345 | unsigned char Char = Str[i]; |
Chris Lattner | 52a3e9e | 2009-01-16 22:13:37 +0000 | [diff] [blame] | 346 | if (isprint(Char) && Char != '\\' && Char != '"') |
Chris Lattner | c7d945d | 2009-01-16 19:25:54 +0000 | [diff] [blame] | 347 | OS << (char)Char; |
| 348 | else // Output anything hard as an octal escape. |
| 349 | OS << '\\' |
| 350 | << (char)('0'+ ((Char >> 6) & 7)) |
| 351 | << (char)('0'+ ((Char >> 3) & 7)) |
| 352 | << (char)('0'+ ((Char >> 0) & 7)); |
| 353 | } |
| 354 | OS << '"'; |
| 355 | } |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 356 | |
Chris Lattner | c7d945d | 2009-01-16 19:25:54 +0000 | [diff] [blame] | 357 | OS << ')'; |
Jordan Rose | 0cdd1fe | 2012-06-15 23:33:51 +0000 | [diff] [blame] | 358 | setEmittedDirectiveOnThisLine(); |
Chris Lattner | c7d945d | 2009-01-16 19:25:54 +0000 | [diff] [blame] | 359 | } |
| 360 | |
Chris Lattner | abfe094 | 2010-06-26 17:11:39 +0000 | [diff] [blame] | 361 | void PrintPPOutputPPCallbacks::PragmaMessage(SourceLocation Loc, |
Chris Lattner | 5f9e272 | 2011-07-23 10:55:15 +0000 | [diff] [blame] | 362 | StringRef Str) { |
Jordan Rose | 0cdd1fe | 2012-06-15 23:33:51 +0000 | [diff] [blame] | 363 | startNewLineIfNeeded(); |
Chris Lattner | abfe094 | 2010-06-26 17:11:39 +0000 | [diff] [blame] | 364 | MoveToLine(Loc); |
| 365 | OS << "#pragma message("; |
| 366 | |
| 367 | OS << '"'; |
| 368 | |
| 369 | for (unsigned i = 0, e = Str.size(); i != e; ++i) { |
| 370 | unsigned char Char = Str[i]; |
| 371 | if (isprint(Char) && Char != '\\' && Char != '"') |
| 372 | OS << (char)Char; |
| 373 | else // Output anything hard as an octal escape. |
| 374 | OS << '\\' |
| 375 | << (char)('0'+ ((Char >> 6) & 7)) |
| 376 | << (char)('0'+ ((Char >> 3) & 7)) |
| 377 | << (char)('0'+ ((Char >> 0) & 7)); |
| 378 | } |
| 379 | OS << '"'; |
| 380 | |
| 381 | OS << ')'; |
Jordan Rose | 0cdd1fe | 2012-06-15 23:33:51 +0000 | [diff] [blame] | 382 | setEmittedDirectiveOnThisLine(); |
Chris Lattner | abfe094 | 2010-06-26 17:11:39 +0000 | [diff] [blame] | 383 | } |
| 384 | |
Douglas Gregor | c09ce12 | 2011-06-22 19:41:48 +0000 | [diff] [blame] | 385 | void PrintPPOutputPPCallbacks:: |
Chris Lattner | 5f9e272 | 2011-07-23 10:55:15 +0000 | [diff] [blame] | 386 | PragmaDiagnosticPush(SourceLocation Loc, StringRef Namespace) { |
Jordan Rose | 0cdd1fe | 2012-06-15 23:33:51 +0000 | [diff] [blame] | 387 | startNewLineIfNeeded(); |
Douglas Gregor | c09ce12 | 2011-06-22 19:41:48 +0000 | [diff] [blame] | 388 | MoveToLine(Loc); |
| 389 | OS << "#pragma " << Namespace << " diagnostic push"; |
Jordan Rose | 0cdd1fe | 2012-06-15 23:33:51 +0000 | [diff] [blame] | 390 | setEmittedDirectiveOnThisLine(); |
Douglas Gregor | c09ce12 | 2011-06-22 19:41:48 +0000 | [diff] [blame] | 391 | } |
| 392 | |
| 393 | void PrintPPOutputPPCallbacks:: |
Chris Lattner | 5f9e272 | 2011-07-23 10:55:15 +0000 | [diff] [blame] | 394 | PragmaDiagnosticPop(SourceLocation Loc, StringRef Namespace) { |
Jordan Rose | 0cdd1fe | 2012-06-15 23:33:51 +0000 | [diff] [blame] | 395 | startNewLineIfNeeded(); |
Douglas Gregor | c09ce12 | 2011-06-22 19:41:48 +0000 | [diff] [blame] | 396 | MoveToLine(Loc); |
| 397 | OS << "#pragma " << Namespace << " diagnostic pop"; |
Jordan Rose | 0cdd1fe | 2012-06-15 23:33:51 +0000 | [diff] [blame] | 398 | setEmittedDirectiveOnThisLine(); |
Douglas Gregor | c09ce12 | 2011-06-22 19:41:48 +0000 | [diff] [blame] | 399 | } |
| 400 | |
| 401 | void PrintPPOutputPPCallbacks:: |
Chris Lattner | 5f9e272 | 2011-07-23 10:55:15 +0000 | [diff] [blame] | 402 | PragmaDiagnostic(SourceLocation Loc, StringRef Namespace, |
Benjamin Kramer | d7a3e2c | 2012-02-07 22:29:24 +0000 | [diff] [blame] | 403 | diag::Mapping Map, StringRef Str) { |
Jordan Rose | 0cdd1fe | 2012-06-15 23:33:51 +0000 | [diff] [blame] | 404 | startNewLineIfNeeded(); |
Douglas Gregor | c09ce12 | 2011-06-22 19:41:48 +0000 | [diff] [blame] | 405 | MoveToLine(Loc); |
| 406 | OS << "#pragma " << Namespace << " diagnostic "; |
Benjamin Kramer | d7a3e2c | 2012-02-07 22:29:24 +0000 | [diff] [blame] | 407 | switch (Map) { |
Douglas Gregor | c09ce12 | 2011-06-22 19:41:48 +0000 | [diff] [blame] | 408 | case diag::MAP_WARNING: |
| 409 | OS << "warning"; |
| 410 | break; |
| 411 | case diag::MAP_ERROR: |
| 412 | OS << "error"; |
| 413 | break; |
| 414 | case diag::MAP_IGNORE: |
| 415 | OS << "ignored"; |
| 416 | break; |
| 417 | case diag::MAP_FATAL: |
| 418 | OS << "fatal"; |
| 419 | break; |
| 420 | } |
| 421 | OS << " \"" << Str << '"'; |
Jordan Rose | 0cdd1fe | 2012-06-15 23:33:51 +0000 | [diff] [blame] | 422 | setEmittedDirectiveOnThisLine(); |
Douglas Gregor | c09ce12 | 2011-06-22 19:41:48 +0000 | [diff] [blame] | 423 | } |
Chris Lattner | c7d945d | 2009-01-16 19:25:54 +0000 | [diff] [blame] | 424 | |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 425 | /// HandleFirstTokOnLine - When emitting a preprocessed file in -E mode, this |
Chris Lattner | 5f18032 | 2007-12-09 21:11:08 +0000 | [diff] [blame] | 426 | /// is called for the first token on each new line. If this really is the start |
| 427 | /// of a new logical line, handle it and return true, otherwise return false. |
| 428 | /// This may not be the start of a logical line because the "start of line" |
Chandler Carruth | 9f79a1f | 2011-07-14 16:14:52 +0000 | [diff] [blame] | 429 | /// marker is set for spelling lines, not expansion ones. |
Chris Lattner | 5f18032 | 2007-12-09 21:11:08 +0000 | [diff] [blame] | 430 | bool PrintPPOutputPPCallbacks::HandleFirstTokOnLine(Token &Tok) { |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 431 | // Figure out what line we went to and insert the appropriate number of |
| 432 | // newline characters. |
Chris Lattner | 5f18032 | 2007-12-09 21:11:08 +0000 | [diff] [blame] | 433 | if (!MoveToLine(Tok.getLocation())) |
| 434 | return false; |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 435 | |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 436 | // Print out space characters so that the first token on a line is |
| 437 | // indented for easy reading. |
Chandler Carruth | a77c031 | 2011-07-25 20:57:57 +0000 | [diff] [blame] | 438 | unsigned ColNo = SM.getExpansionColumnNumber(Tok.getLocation()); |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 439 | |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 440 | // This hack prevents stuff like: |
| 441 | // #define HASH # |
| 442 | // HASH define foo bar |
| 443 | // From having the # character end up at column 1, which makes it so it |
| 444 | // is not handled as a #define next time through the preprocessor if in |
| 445 | // -fpreprocessed mode. |
Chris Lattner | 057aaf6 | 2007-10-09 18:03:42 +0000 | [diff] [blame] | 446 | if (ColNo <= 1 && Tok.is(tok::hash)) |
Chris Lattner | e96de3e | 2008-08-17 03:12:02 +0000 | [diff] [blame] | 447 | OS << ' '; |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 448 | |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 449 | // Otherwise, indent the appropriate number of spaces. |
| 450 | for (; ColNo > 1; --ColNo) |
Chris Lattner | e96de3e | 2008-08-17 03:12:02 +0000 | [diff] [blame] | 451 | OS << ' '; |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 452 | |
Chris Lattner | 5f18032 | 2007-12-09 21:11:08 +0000 | [diff] [blame] | 453 | return true; |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 454 | } |
| 455 | |
Chris Lattner | 3ee211f | 2009-06-15 01:25:23 +0000 | [diff] [blame] | 456 | void PrintPPOutputPPCallbacks::HandleNewlinesInToken(const char *TokStr, |
| 457 | unsigned Len) { |
| 458 | unsigned NumNewlines = 0; |
| 459 | for (; Len; --Len, ++TokStr) { |
| 460 | if (*TokStr != '\n' && |
| 461 | *TokStr != '\r') |
| 462 | continue; |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 463 | |
Chris Lattner | 3ee211f | 2009-06-15 01:25:23 +0000 | [diff] [blame] | 464 | ++NumNewlines; |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 465 | |
Chris Lattner | 3ee211f | 2009-06-15 01:25:23 +0000 | [diff] [blame] | 466 | // If we have \n\r or \r\n, skip both and count as one line. |
| 467 | if (Len != 1 && |
| 468 | (TokStr[1] == '\n' || TokStr[1] == '\r') && |
| 469 | TokStr[0] != TokStr[1]) |
| 470 | ++TokStr, --Len; |
| 471 | } |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 472 | |
Chris Lattner | 3ee211f | 2009-06-15 01:25:23 +0000 | [diff] [blame] | 473 | if (NumNewlines == 0) return; |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 474 | |
Chris Lattner | ed2d7c4 | 2009-06-15 04:08:28 +0000 | [diff] [blame] | 475 | CurLine += NumNewlines; |
Chris Lattner | 3ee211f | 2009-06-15 01:25:23 +0000 | [diff] [blame] | 476 | } |
| 477 | |
| 478 | |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 479 | namespace { |
| 480 | struct UnknownPragmaHandler : public PragmaHandler { |
| 481 | const char *Prefix; |
| 482 | PrintPPOutputPPCallbacks *Callbacks; |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 483 | |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 484 | UnknownPragmaHandler(const char *prefix, PrintPPOutputPPCallbacks *callbacks) |
Argyrios Kyrtzidis | 9b36c3f | 2010-07-13 09:07:17 +0000 | [diff] [blame] | 485 | : Prefix(prefix), Callbacks(callbacks) {} |
Douglas Gregor | 80c60f7 | 2010-09-09 22:45:38 +0000 | [diff] [blame] | 486 | virtual void HandlePragma(Preprocessor &PP, PragmaIntroducerKind Introducer, |
| 487 | Token &PragmaTok) { |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 488 | // Figure out what line we went to and insert the appropriate number of |
| 489 | // newline characters. |
Jordan Rose | 0cdd1fe | 2012-06-15 23:33:51 +0000 | [diff] [blame] | 490 | Callbacks->startNewLineIfNeeded(); |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 491 | Callbacks->MoveToLine(PragmaTok.getLocation()); |
Chris Lattner | e96de3e | 2008-08-17 03:12:02 +0000 | [diff] [blame] | 492 | Callbacks->OS.write(Prefix, strlen(Prefix)); |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 493 | // Read and print all of the pragma tokens. |
Peter Collingbourne | 8402155 | 2011-02-28 02:37:51 +0000 | [diff] [blame] | 494 | while (PragmaTok.isNot(tok::eod)) { |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 495 | if (PragmaTok.hasLeadingSpace()) |
Chris Lattner | e96de3e | 2008-08-17 03:12:02 +0000 | [diff] [blame] | 496 | Callbacks->OS << ' '; |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 497 | std::string TokSpell = PP.getSpelling(PragmaTok); |
Chris Lattner | e96de3e | 2008-08-17 03:12:02 +0000 | [diff] [blame] | 498 | Callbacks->OS.write(&TokSpell[0], TokSpell.size()); |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 499 | PP.LexUnexpandedToken(PragmaTok); |
| 500 | } |
Jordan Rose | 0cdd1fe | 2012-06-15 23:33:51 +0000 | [diff] [blame] | 501 | Callbacks->setEmittedDirectiveOnThisLine(); |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 502 | } |
| 503 | }; |
| 504 | } // end anonymous namespace |
| 505 | |
Chris Lattner | f0f2b29 | 2007-07-23 06:09:34 +0000 | [diff] [blame] | 506 | |
Chris Lattner | 59076ab | 2009-02-06 05:56:11 +0000 | [diff] [blame] | 507 | static void PrintPreprocessedTokens(Preprocessor &PP, Token &Tok, |
| 508 | PrintPPOutputPPCallbacks *Callbacks, |
Chris Lattner | 5f9e272 | 2011-07-23 10:55:15 +0000 | [diff] [blame] | 509 | raw_ostream &OS) { |
Benjamin Kramer | e396007 | 2010-02-27 18:02:51 +0000 | [diff] [blame] | 510 | char Buffer[256]; |
Chris Lattner | c54539c | 2010-06-15 21:06:38 +0000 | [diff] [blame] | 511 | Token PrevPrevTok, PrevTok; |
| 512 | PrevPrevTok.startToken(); |
| 513 | PrevTok.startToken(); |
Chris Lattner | 6f688e1 | 2007-10-10 20:45:16 +0000 | [diff] [blame] | 514 | while (1) { |
Jordan Rose | 0cdd1fe | 2012-06-15 23:33:51 +0000 | [diff] [blame] | 515 | if (Callbacks->hasEmittedDirectiveOnThisLine()) { |
| 516 | Callbacks->startNewLineIfNeeded(); |
| 517 | Callbacks->MoveToLine(Tok.getLocation()); |
| 518 | } |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 519 | |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 520 | // If this token is at the start of a line, emit newlines if needed. |
Chris Lattner | 5f18032 | 2007-12-09 21:11:08 +0000 | [diff] [blame] | 521 | if (Tok.isAtStartOfLine() && Callbacks->HandleFirstTokOnLine(Tok)) { |
| 522 | // done. |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 523 | } else if (Tok.hasLeadingSpace() || |
Chris Lattner | f0f2b29 | 2007-07-23 06:09:34 +0000 | [diff] [blame] | 524 | // If we haven't emitted a token on this line yet, PrevTok isn't |
| 525 | // useful to look at and no concatenation could happen anyway. |
Chris Lattner | b638a30 | 2007-07-23 23:21:34 +0000 | [diff] [blame] | 526 | (Callbacks->hasEmittedTokensOnThisLine() && |
Chris Lattner | f0f2b29 | 2007-07-23 06:09:34 +0000 | [diff] [blame] | 527 | // Don't print "-" next to "-", it would form "--". |
Chris Lattner | 8877321 | 2010-04-14 03:57:19 +0000 | [diff] [blame] | 528 | Callbacks->AvoidConcat(PrevPrevTok, PrevTok, Tok))) { |
Chris Lattner | e96de3e | 2008-08-17 03:12:02 +0000 | [diff] [blame] | 529 | OS << ' '; |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 530 | } |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 531 | |
Benjamin Kramer | e396007 | 2010-02-27 18:02:51 +0000 | [diff] [blame] | 532 | if (IdentifierInfo *II = Tok.getIdentifierInfo()) { |
| 533 | OS << II->getName(); |
| 534 | } else if (Tok.isLiteral() && !Tok.needsCleaning() && |
| 535 | Tok.getLiteralData()) { |
| 536 | OS.write(Tok.getLiteralData(), Tok.getLength()); |
| 537 | } else if (Tok.getLength() < 256) { |
| 538 | const char *TokPtr = Buffer; |
| 539 | unsigned Len = PP.getSpelling(Tok, TokPtr); |
| 540 | OS.write(TokPtr, Len); |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 541 | |
Benjamin Kramer | e396007 | 2010-02-27 18:02:51 +0000 | [diff] [blame] | 542 | // Tokens that can contain embedded newlines need to adjust our current |
| 543 | // line number. |
| 544 | if (Tok.getKind() == tok::comment) |
| 545 | Callbacks->HandleNewlinesInToken(TokPtr, Len); |
| 546 | } else { |
| 547 | std::string S = PP.getSpelling(Tok); |
| 548 | OS.write(&S[0], S.size()); |
| 549 | |
| 550 | // Tokens that can contain embedded newlines need to adjust our current |
| 551 | // line number. |
| 552 | if (Tok.getKind() == tok::comment) |
| 553 | Callbacks->HandleNewlinesInToken(&S[0], S.size()); |
| 554 | } |
Jordan Rose | 0cdd1fe | 2012-06-15 23:33:51 +0000 | [diff] [blame] | 555 | Callbacks->setEmittedTokensOnThisLine(); |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 556 | |
Chris Lattner | 6f688e1 | 2007-10-10 20:45:16 +0000 | [diff] [blame] | 557 | if (Tok.is(tok::eof)) break; |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 558 | |
Chris Lattner | 8877321 | 2010-04-14 03:57:19 +0000 | [diff] [blame] | 559 | PrevPrevTok = PrevTok; |
Chris Lattner | 6f688e1 | 2007-10-10 20:45:16 +0000 | [diff] [blame] | 560 | PrevTok = Tok; |
| 561 | PP.Lex(Tok); |
| 562 | } |
Chris Lattner | 59076ab | 2009-02-06 05:56:11 +0000 | [diff] [blame] | 563 | } |
| 564 | |
Benjamin Kramer | 3ff2a4b | 2010-01-19 17:42:20 +0000 | [diff] [blame] | 565 | typedef std::pair<IdentifierInfo*, MacroInfo*> id_macro_pair; |
| 566 | static int MacroIDCompare(const void* a, const void* b) { |
| 567 | const id_macro_pair *LHS = static_cast<const id_macro_pair*>(a); |
| 568 | const id_macro_pair *RHS = static_cast<const id_macro_pair*>(b); |
| 569 | return LHS->first->getName().compare(RHS->first->getName()); |
Chris Lattner | 2a2bb18 | 2009-02-10 22:28:19 +0000 | [diff] [blame] | 570 | } |
Chris Lattner | 59076ab | 2009-02-06 05:56:11 +0000 | [diff] [blame] | 571 | |
Chris Lattner | 5f9e272 | 2011-07-23 10:55:15 +0000 | [diff] [blame] | 572 | static void DoPrintMacros(Preprocessor &PP, raw_ostream *OS) { |
Daniel Dunbar | c72cc50 | 2010-06-11 20:10:12 +0000 | [diff] [blame] | 573 | // Ignore unknown pragmas. |
Argyrios Kyrtzidis | 9b36c3f | 2010-07-13 09:07:17 +0000 | [diff] [blame] | 574 | PP.AddPragmaHandler(new EmptyPragmaHandler()); |
Daniel Dunbar | c72cc50 | 2010-06-11 20:10:12 +0000 | [diff] [blame] | 575 | |
Eli Friedman | f1db585 | 2009-05-19 01:32:34 +0000 | [diff] [blame] | 576 | // -dM mode just scans and ignores all tokens in the files, then dumps out |
| 577 | // the macro table at the end. |
Chris Lattner | e127a0d | 2010-04-20 20:35:58 +0000 | [diff] [blame] | 578 | PP.EnterMainSourceFile(); |
Eli Friedman | f1db585 | 2009-05-19 01:32:34 +0000 | [diff] [blame] | 579 | |
| 580 | Token Tok; |
| 581 | do PP.Lex(Tok); |
| 582 | while (Tok.isNot(tok::eof)); |
| 583 | |
Alexander Kornienko | 8a64bb5 | 2012-08-29 00:20:03 +0000 | [diff] [blame] | 584 | SmallVector<id_macro_pair, 128> MacrosByID; |
| 585 | for (Preprocessor::macro_iterator I = PP.macro_begin(), E = PP.macro_end(); |
| 586 | I != E; ++I) { |
| 587 | if (I->first->hasMacroDefinition()) |
| 588 | MacrosByID.push_back(id_macro_pair(I->first, I->second)); |
| 589 | } |
Benjamin Kramer | 3ff2a4b | 2010-01-19 17:42:20 +0000 | [diff] [blame] | 590 | llvm::array_pod_sort(MacrosByID.begin(), MacrosByID.end(), MacroIDCompare); |
Eli Friedman | f1db585 | 2009-05-19 01:32:34 +0000 | [diff] [blame] | 591 | |
| 592 | for (unsigned i = 0, e = MacrosByID.size(); i != e; ++i) { |
| 593 | MacroInfo &MI = *MacrosByID[i].second; |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 594 | // Ignore computed macros like __LINE__ and friends. |
Eli Friedman | f1db585 | 2009-05-19 01:32:34 +0000 | [diff] [blame] | 595 | if (MI.isBuiltinMacro()) continue; |
| 596 | |
| 597 | PrintMacroDefinition(*MacrosByID[i].first, MI, PP, *OS); |
Benjamin Kramer | 3ff2a4b | 2010-01-19 17:42:20 +0000 | [diff] [blame] | 598 | *OS << '\n'; |
Eli Friedman | f1db585 | 2009-05-19 01:32:34 +0000 | [diff] [blame] | 599 | } |
| 600 | } |
| 601 | |
Chris Lattner | 59076ab | 2009-02-06 05:56:11 +0000 | [diff] [blame] | 602 | /// DoPrintPreprocessedInput - This implements -E mode. |
| 603 | /// |
Chris Lattner | 5f9e272 | 2011-07-23 10:55:15 +0000 | [diff] [blame] | 604 | void clang::DoPrintPreprocessedInput(Preprocessor &PP, raw_ostream *OS, |
Daniel Dunbar | 29cf746 | 2009-11-11 10:07:44 +0000 | [diff] [blame] | 605 | const PreprocessorOutputOptions &Opts) { |
Daniel Dunbar | 775bee7 | 2009-11-11 10:07:22 +0000 | [diff] [blame] | 606 | // Show macros with no output is handled specially. |
| 607 | if (!Opts.ShowCPP) { |
| 608 | assert(Opts.ShowMacros && "Not yet implemented!"); |
| 609 | DoPrintMacros(PP, OS); |
| 610 | return; |
| 611 | } |
| 612 | |
Chris Lattner | 59076ab | 2009-02-06 05:56:11 +0000 | [diff] [blame] | 613 | // Inform the preprocessor whether we want it to retain comments or not, due |
| 614 | // to -C or -CC. |
Daniel Dunbar | 775bee7 | 2009-11-11 10:07:22 +0000 | [diff] [blame] | 615 | PP.SetCommentRetentionState(Opts.ShowComments, Opts.ShowMacroComments); |
Eli Friedman | f54fce8 | 2009-05-19 01:02:07 +0000 | [diff] [blame] | 616 | |
Eli Friedman | 12d3b1d | 2009-05-19 03:06:47 +0000 | [diff] [blame] | 617 | PrintPPOutputPPCallbacks *Callbacks = |
Daniel Dunbar | 775bee7 | 2009-11-11 10:07:22 +0000 | [diff] [blame] | 618 | new PrintPPOutputPPCallbacks(PP, *OS, !Opts.ShowLineMarkers, |
Daniel Dunbar | eef63e0 | 2011-02-02 15:41:17 +0000 | [diff] [blame] | 619 | Opts.ShowMacros); |
Argyrios Kyrtzidis | 9b36c3f | 2010-07-13 09:07:17 +0000 | [diff] [blame] | 620 | PP.AddPragmaHandler(new UnknownPragmaHandler("#pragma", Callbacks)); |
Chris Lattner | 1397399 | 2010-11-10 01:00:49 +0000 | [diff] [blame] | 621 | PP.AddPragmaHandler("GCC", new UnknownPragmaHandler("#pragma GCC",Callbacks)); |
| 622 | PP.AddPragmaHandler("clang", |
| 623 | new UnknownPragmaHandler("#pragma clang", Callbacks)); |
Chris Lattner | 59076ab | 2009-02-06 05:56:11 +0000 | [diff] [blame] | 624 | |
Kovarththanan Rajaratnam | a5d10c4 | 2010-03-07 07:30:06 +0000 | [diff] [blame] | 625 | PP.addPPCallbacks(Callbacks); |
Chris Lattner | 59076ab | 2009-02-06 05:56:11 +0000 | [diff] [blame] | 626 | |
Eli Friedman | f1db585 | 2009-05-19 01:32:34 +0000 | [diff] [blame] | 627 | // After we have configured the preprocessor, enter the main file. |
Chris Lattner | e127a0d | 2010-04-20 20:35:58 +0000 | [diff] [blame] | 628 | PP.EnterMainSourceFile(); |
Chris Lattner | 59076ab | 2009-02-06 05:56:11 +0000 | [diff] [blame] | 629 | |
Eli Friedman | f1db585 | 2009-05-19 01:32:34 +0000 | [diff] [blame] | 630 | // Consume all of the tokens that come from the predefines buffer. Those |
| 631 | // should not be emitted into the output and are guaranteed to be at the |
| 632 | // start. |
| 633 | const SourceManager &SourceMgr = PP.getSourceManager(); |
| 634 | Token Tok; |
Douglas Gregor | cb7b1e1 | 2010-11-12 07:15:47 +0000 | [diff] [blame] | 635 | do { |
| 636 | PP.Lex(Tok); |
| 637 | if (Tok.is(tok::eof) || !Tok.getLocation().isFileID()) |
| 638 | break; |
| 639 | |
| 640 | PresumedLoc PLoc = SourceMgr.getPresumedLoc(Tok.getLocation()); |
| 641 | if (PLoc.isInvalid()) |
| 642 | break; |
| 643 | |
| 644 | if (strcmp(PLoc.getFilename(), "<built-in>")) |
| 645 | break; |
| 646 | } while (true); |
Chris Lattner | 59076ab | 2009-02-06 05:56:11 +0000 | [diff] [blame] | 647 | |
Eli Friedman | f1db585 | 2009-05-19 01:32:34 +0000 | [diff] [blame] | 648 | // Read all the preprocessed tokens, printing them out to the stream. |
| 649 | PrintPreprocessedTokens(PP, Tok, Callbacks, *OS); |
| 650 | *OS << '\n'; |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 651 | } |