diff --git a/clang/Driver/DiagChecker.cpp b/clang/Driver/DiagChecker.cpp index 6503214c5218..eeba859b19a0 100644 --- a/clang/Driver/DiagChecker.cpp +++ b/clang/Driver/DiagChecker.cpp @@ -118,12 +118,9 @@ static void FindExpectedDiags(Preprocessor &PP, // Create a raw lexer to pull all the comments out of the main file. We don't // want to look in #include'd headers for expected-error strings. FileID FID = PP.getSourceManager().getMainFileID(); - std::pair File = - PP.getSourceManager().getBufferData(FID); // Create a lexer to lex all the tokens of the main file in raw mode. - Lexer RawLex(PP.getSourceManager().getLocForStartOfFile(FID), - PP.getLangOptions(), File.first, File.second); + Lexer RawLex(FID, PP.getSourceManager(), PP.getLangOptions()); // Return comments as tokens, this is how we find expected diagnostics. RawLex.SetCommentRetentionState(true); diff --git a/clang/Driver/RewriteMacros.cpp b/clang/Driver/RewriteMacros.cpp index ecf4b416d8b5..6ad4a7160cdc 100644 --- a/clang/Driver/RewriteMacros.cpp +++ b/clang/Driver/RewriteMacros.cpp @@ -61,12 +61,10 @@ static const Token &GetNextRawTok(const std::vector &RawTokens, static void LexRawTokensFromMainFile(Preprocessor &PP, std::vector &RawTokens) { SourceManager &SM = PP.getSourceManager(); - std::pair File =SM.getBufferData(SM.getMainFileID()); // Create a lexer to lex all the tokens of the main file in raw mode. Even // though it is in raw mode, it will not return comments. - Lexer RawLex(SM.getLocForStartOfFile(SM.getMainFileID()), - PP.getLangOptions(), File.first, File.second); + Lexer RawLex(SM.getMainFileID(), SM, PP.getLangOptions()); // Switch on comment lexing because we really do want them. RawLex.SetCommentRetentionState(true);