Pragma.cpp revision 374b3837d676133fcc1eb70a25c8baf8ec4a5c4a
15f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer//===--- Pragma.cpp - Pragma registration and handling --------------------===// 25f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer// 35f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer// The LLVM Compiler Infrastructure 45f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer// 50bc735ffcfb223c0186419547abaa5c84482663eChris Lattner// This file is distributed under the University of Illinois Open Source 60bc735ffcfb223c0186419547abaa5c84482663eChris Lattner// License. See LICENSE.TXT for details. 75f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer// 85f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer//===----------------------------------------------------------------------===// 95f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer// 105f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer// This file implements the PragmaHandler/PragmaTable interfaces and implements 115f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer// pragma related methods of the Preprocessor class. 125f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer// 135f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer//===----------------------------------------------------------------------===// 145f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 155f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer#include "clang/Lex/Pragma.h" 165f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer#include "clang/Lex/HeaderSearch.h" 17a9d9145741ef77db45890911674705b81605b10bChris Lattner#include "clang/Lex/LiteralSupport.h" 185f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer#include "clang/Lex/Preprocessor.h" 19f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner#include "clang/Lex/MacroInfo.h" 20500d3297d2a21edeac4d46cbcbe21bc2352c2a28Chris Lattner#include "clang/Lex/LexDiagnostic.h" 215f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer#include "clang/Basic/FileManager.h" 225f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer#include "clang/Basic/SourceManager.h" 23ff759a66526d6a985fb4efae69c6b798d83364dfDaniel Dunbar#include "llvm/Support/CrashRecoveryContext.h" 245505413ee8e4e2924f52ba81181071f3a492e7d9Daniel Dunbar#include "llvm/Support/ErrorHandling.h" 252e22253e03e175144aeb9d13350a12fd83f858beDouglas Gregor#include <algorithm> 265f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerusing namespace clang; 275f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 285f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer// Out-of-line destructor to provide a home for the class. 295f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid SpencerPragmaHandler::~PragmaHandler() { 305f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer} 315f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 325f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer//===----------------------------------------------------------------------===// 33c72cc5072cdc1a1a6e05f9d0f962f293a69248c4Daniel Dunbar// EmptyPragmaHandler Implementation. 34c72cc5072cdc1a1a6e05f9d0f962f293a69248c4Daniel Dunbar//===----------------------------------------------------------------------===// 35c72cc5072cdc1a1a6e05f9d0f962f293a69248c4Daniel Dunbar 369b36c3f0de0105e903130bbda3c4aea7d792c0afArgyrios KyrtzidisEmptyPragmaHandler::EmptyPragmaHandler() {} 37c72cc5072cdc1a1a6e05f9d0f962f293a69248c4Daniel Dunbar 3880c60f72848896f867f6b7e664e7060d9e78f019Douglas Gregorvoid EmptyPragmaHandler::HandlePragma(Preprocessor &PP, 3980c60f72848896f867f6b7e664e7060d9e78f019Douglas Gregor PragmaIntroducerKind Introducer, 4080c60f72848896f867f6b7e664e7060d9e78f019Douglas Gregor Token &FirstToken) {} 41c72cc5072cdc1a1a6e05f9d0f962f293a69248c4Daniel Dunbar 42c72cc5072cdc1a1a6e05f9d0f962f293a69248c4Daniel Dunbar//===----------------------------------------------------------------------===// 435f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer// PragmaNamespace Implementation. 445f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer//===----------------------------------------------------------------------===// 455f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 465f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 475f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid SpencerPragmaNamespace::~PragmaNamespace() { 489b36c3f0de0105e903130bbda3c4aea7d792c0afArgyrios Kyrtzidis for (llvm::StringMap<PragmaHandler*>::iterator 499b36c3f0de0105e903130bbda3c4aea7d792c0afArgyrios Kyrtzidis I = Handlers.begin(), E = Handlers.end(); I != E; ++I) 509b36c3f0de0105e903130bbda3c4aea7d792c0afArgyrios Kyrtzidis delete I->second; 515f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer} 525f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 535f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// FindHandler - Check to see if there is already a handler for the 545f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// specified name. If not, return the handler for the null identifier if it 555f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// exists, otherwise return null. If IgnoreNull is true (the default) then 565f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// the null handler isn't returned on failure to match. 575f9e272e632e951b1efe824cd16acb4d96077930Chris LattnerPragmaHandler *PragmaNamespace::FindHandler(StringRef Name, 585f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer bool IgnoreNull) const { 599b36c3f0de0105e903130bbda3c4aea7d792c0afArgyrios Kyrtzidis if (PragmaHandler *Handler = Handlers.lookup(Name)) 609b36c3f0de0105e903130bbda3c4aea7d792c0afArgyrios Kyrtzidis return Handler; 615f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner return IgnoreNull ? 0 : Handlers.lookup(StringRef()); 629b36c3f0de0105e903130bbda3c4aea7d792c0afArgyrios Kyrtzidis} 631eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 649b36c3f0de0105e903130bbda3c4aea7d792c0afArgyrios Kyrtzidisvoid PragmaNamespace::AddPragma(PragmaHandler *Handler) { 659b36c3f0de0105e903130bbda3c4aea7d792c0afArgyrios Kyrtzidis assert(!Handlers.lookup(Handler->getName()) && 669b36c3f0de0105e903130bbda3c4aea7d792c0afArgyrios Kyrtzidis "A handler with this name is already registered in this namespace"); 679b36c3f0de0105e903130bbda3c4aea7d792c0afArgyrios Kyrtzidis llvm::StringMapEntry<PragmaHandler *> &Entry = 689b36c3f0de0105e903130bbda3c4aea7d792c0afArgyrios Kyrtzidis Handlers.GetOrCreateValue(Handler->getName()); 699b36c3f0de0105e903130bbda3c4aea7d792c0afArgyrios Kyrtzidis Entry.setValue(Handler); 705f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer} 715f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 724095080aff204008eefb26b100906c6ca2bc4bb6Daniel Dunbarvoid PragmaNamespace::RemovePragmaHandler(PragmaHandler *Handler) { 739b36c3f0de0105e903130bbda3c4aea7d792c0afArgyrios Kyrtzidis assert(Handlers.lookup(Handler->getName()) && 749b36c3f0de0105e903130bbda3c4aea7d792c0afArgyrios Kyrtzidis "Handler not registered in this namespace"); 759b36c3f0de0105e903130bbda3c4aea7d792c0afArgyrios Kyrtzidis Handlers.erase(Handler->getName()); 764095080aff204008eefb26b100906c6ca2bc4bb6Daniel Dunbar} 774095080aff204008eefb26b100906c6ca2bc4bb6Daniel Dunbar 7880c60f72848896f867f6b7e664e7060d9e78f019Douglas Gregorvoid PragmaNamespace::HandlePragma(Preprocessor &PP, 7980c60f72848896f867f6b7e664e7060d9e78f019Douglas Gregor PragmaIntroducerKind Introducer, 8080c60f72848896f867f6b7e664e7060d9e78f019Douglas Gregor Token &Tok) { 815f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Read the 'namespace' that the directive is in, e.g. STDC. Do not macro 825f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // expand it, the user can have a STDC #define, that should not affect this. 835f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer PP.LexUnexpandedToken(Tok); 841eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 855f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Get the handler for this token. If there is no handler, ignore the pragma. 869b36c3f0de0105e903130bbda3c4aea7d792c0afArgyrios Kyrtzidis PragmaHandler *Handler 879b36c3f0de0105e903130bbda3c4aea7d792c0afArgyrios Kyrtzidis = FindHandler(Tok.getIdentifierInfo() ? Tok.getIdentifierInfo()->getName() 885f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner : StringRef(), 899b36c3f0de0105e903130bbda3c4aea7d792c0afArgyrios Kyrtzidis /*IgnoreNull=*/false); 90af7cdf45da4925f788e87a4c318ee67404646088Chris Lattner if (Handler == 0) { 91af7cdf45da4925f788e87a4c318ee67404646088Chris Lattner PP.Diag(Tok, diag::warn_pragma_ignored); 92af7cdf45da4925f788e87a4c318ee67404646088Chris Lattner return; 93af7cdf45da4925f788e87a4c318ee67404646088Chris Lattner } 941eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 955f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Otherwise, pass it down. 9680c60f72848896f867f6b7e664e7060d9e78f019Douglas Gregor Handler->HandlePragma(PP, Introducer, Tok); 975f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer} 985f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 995f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer//===----------------------------------------------------------------------===// 1005f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer// Preprocessor Pragma Directive Handling. 1015f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer//===----------------------------------------------------------------------===// 1025f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 103b6e95b7752893f66e9d067b4287dff2dbf4a5802James Dennett/// HandlePragmaDirective - The "\#pragma" directive has been parsed. Lex the 1045f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// rest of the pragma, passing it to the registered pragma handlers. 10580c60f72848896f867f6b7e664e7060d9e78f019Douglas Gregorvoid Preprocessor::HandlePragmaDirective(unsigned Introducer) { 1066fe6a49c4058211ff4489023c78615ec0266c5ffJordan Rose if (!PragmasEnabled) 1076fe6a49c4058211ff4489023c78615ec0266c5ffJordan Rose return; 1086fe6a49c4058211ff4489023c78615ec0266c5ffJordan Rose 1095f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer ++NumPragma; 1101eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 1115f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Invoke the first level of pragma handlers which reads the namespace id. 112d217773f106856a11879ec79dc468efefaf2ee75Chris Lattner Token Tok; 11380c60f72848896f867f6b7e664e7060d9e78f019Douglas Gregor PragmaHandlers->HandlePragma(*this, PragmaIntroducerKind(Introducer), Tok); 1141eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 1155f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // If the pragma handler didn't read the rest of the line, consume it now. 116b2eb53d9fd973a1a02e05e67a3307b3efd12eff2Peter Collingbourne if ((CurTokenLexer && CurTokenLexer->isParsingPreprocessorDirective()) 117b2eb53d9fd973a1a02e05e67a3307b3efd12eff2Peter Collingbourne || (CurPPLexer && CurPPLexer->ParsingPreprocessorDirective)) 1185f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer DiscardUntilEndOfDirective(); 1195f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer} 1205f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 12114e645557ae91c6770d62beb00a1c522e0bfd5d6Argyrios Kyrtzidisnamespace { 12214e645557ae91c6770d62beb00a1c522e0bfd5d6Argyrios Kyrtzidis/// \brief Helper class for \see Preprocessor::Handle_Pragma. 12314e645557ae91c6770d62beb00a1c522e0bfd5d6Argyrios Kyrtzidisclass LexingFor_PragmaRAII { 12414e645557ae91c6770d62beb00a1c522e0bfd5d6Argyrios Kyrtzidis Preprocessor &PP; 12514e645557ae91c6770d62beb00a1c522e0bfd5d6Argyrios Kyrtzidis bool InMacroArgPreExpansion; 12614e645557ae91c6770d62beb00a1c522e0bfd5d6Argyrios Kyrtzidis bool Failed; 12714e645557ae91c6770d62beb00a1c522e0bfd5d6Argyrios Kyrtzidis Token &OutTok; 12814e645557ae91c6770d62beb00a1c522e0bfd5d6Argyrios Kyrtzidis Token PragmaTok; 12914e645557ae91c6770d62beb00a1c522e0bfd5d6Argyrios Kyrtzidis 13014e645557ae91c6770d62beb00a1c522e0bfd5d6Argyrios Kyrtzidispublic: 13114e645557ae91c6770d62beb00a1c522e0bfd5d6Argyrios Kyrtzidis LexingFor_PragmaRAII(Preprocessor &PP, bool InMacroArgPreExpansion, 13214e645557ae91c6770d62beb00a1c522e0bfd5d6Argyrios Kyrtzidis Token &Tok) 13314e645557ae91c6770d62beb00a1c522e0bfd5d6Argyrios Kyrtzidis : PP(PP), InMacroArgPreExpansion(InMacroArgPreExpansion), 13414e645557ae91c6770d62beb00a1c522e0bfd5d6Argyrios Kyrtzidis Failed(false), OutTok(Tok) { 13514e645557ae91c6770d62beb00a1c522e0bfd5d6Argyrios Kyrtzidis if (InMacroArgPreExpansion) { 13614e645557ae91c6770d62beb00a1c522e0bfd5d6Argyrios Kyrtzidis PragmaTok = OutTok; 13714e645557ae91c6770d62beb00a1c522e0bfd5d6Argyrios Kyrtzidis PP.EnableBacktrackAtThisPos(); 13814e645557ae91c6770d62beb00a1c522e0bfd5d6Argyrios Kyrtzidis } 13914e645557ae91c6770d62beb00a1c522e0bfd5d6Argyrios Kyrtzidis } 14014e645557ae91c6770d62beb00a1c522e0bfd5d6Argyrios Kyrtzidis 14114e645557ae91c6770d62beb00a1c522e0bfd5d6Argyrios Kyrtzidis ~LexingFor_PragmaRAII() { 14214e645557ae91c6770d62beb00a1c522e0bfd5d6Argyrios Kyrtzidis if (InMacroArgPreExpansion) { 14314e645557ae91c6770d62beb00a1c522e0bfd5d6Argyrios Kyrtzidis if (Failed) { 14414e645557ae91c6770d62beb00a1c522e0bfd5d6Argyrios Kyrtzidis PP.CommitBacktrackedTokens(); 14514e645557ae91c6770d62beb00a1c522e0bfd5d6Argyrios Kyrtzidis } else { 14614e645557ae91c6770d62beb00a1c522e0bfd5d6Argyrios Kyrtzidis PP.Backtrack(); 14714e645557ae91c6770d62beb00a1c522e0bfd5d6Argyrios Kyrtzidis OutTok = PragmaTok; 14814e645557ae91c6770d62beb00a1c522e0bfd5d6Argyrios Kyrtzidis } 14914e645557ae91c6770d62beb00a1c522e0bfd5d6Argyrios Kyrtzidis } 15014e645557ae91c6770d62beb00a1c522e0bfd5d6Argyrios Kyrtzidis } 15114e645557ae91c6770d62beb00a1c522e0bfd5d6Argyrios Kyrtzidis 15214e645557ae91c6770d62beb00a1c522e0bfd5d6Argyrios Kyrtzidis void failed() { 15314e645557ae91c6770d62beb00a1c522e0bfd5d6Argyrios Kyrtzidis Failed = true; 15414e645557ae91c6770d62beb00a1c522e0bfd5d6Argyrios Kyrtzidis } 15514e645557ae91c6770d62beb00a1c522e0bfd5d6Argyrios Kyrtzidis}; 15614e645557ae91c6770d62beb00a1c522e0bfd5d6Argyrios Kyrtzidis} 15714e645557ae91c6770d62beb00a1c522e0bfd5d6Argyrios Kyrtzidis 1585f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// Handle_Pragma - Read a _Pragma directive, slice it up, process it, then 1595f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// return the first token after the directive. The _Pragma token has just 1605f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// been read into 'Tok'. 161d217773f106856a11879ec79dc468efefaf2ee75Chris Lattnervoid Preprocessor::Handle_Pragma(Token &Tok) { 16214e645557ae91c6770d62beb00a1c522e0bfd5d6Argyrios Kyrtzidis 16314e645557ae91c6770d62beb00a1c522e0bfd5d6Argyrios Kyrtzidis // This works differently if we are pre-expanding a macro argument. 16414e645557ae91c6770d62beb00a1c522e0bfd5d6Argyrios Kyrtzidis // In that case we don't actually "activate" the pragma now, we only lex it 16514e645557ae91c6770d62beb00a1c522e0bfd5d6Argyrios Kyrtzidis // until we are sure it is lexically correct and then we backtrack so that 16614e645557ae91c6770d62beb00a1c522e0bfd5d6Argyrios Kyrtzidis // we activate the pragma whenever we encounter the tokens again in the token 16714e645557ae91c6770d62beb00a1c522e0bfd5d6Argyrios Kyrtzidis // stream. This ensures that we will activate it in the correct location 16814e645557ae91c6770d62beb00a1c522e0bfd5d6Argyrios Kyrtzidis // or that we will ignore it if it never enters the token stream, e.g: 16914e645557ae91c6770d62beb00a1c522e0bfd5d6Argyrios Kyrtzidis // 17014e645557ae91c6770d62beb00a1c522e0bfd5d6Argyrios Kyrtzidis // #define EMPTY(x) 17114e645557ae91c6770d62beb00a1c522e0bfd5d6Argyrios Kyrtzidis // #define INACTIVE(x) EMPTY(x) 17214e645557ae91c6770d62beb00a1c522e0bfd5d6Argyrios Kyrtzidis // INACTIVE(_Pragma("clang diagnostic ignored \"-Wconversion\"")) 17314e645557ae91c6770d62beb00a1c522e0bfd5d6Argyrios Kyrtzidis 17414e645557ae91c6770d62beb00a1c522e0bfd5d6Argyrios Kyrtzidis LexingFor_PragmaRAII _PragmaLexing(*this, InMacroArgPreExpansion, Tok); 17514e645557ae91c6770d62beb00a1c522e0bfd5d6Argyrios Kyrtzidis 1765f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Remember the pragma token location. 1775f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer SourceLocation PragmaLoc = Tok.getLocation(); 1781eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 1795f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Read the '('. 1805f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer Lex(Tok); 1813692b09faa9fe346f39bc922db6dce48cdcc3f63Chris Lattner if (Tok.isNot(tok::l_paren)) { 1823692b09faa9fe346f39bc922db6dce48cdcc3f63Chris Lattner Diag(PragmaLoc, diag::err__Pragma_malformed); 18314e645557ae91c6770d62beb00a1c522e0bfd5d6Argyrios Kyrtzidis return _PragmaLexing.failed(); 1843692b09faa9fe346f39bc922db6dce48cdcc3f63Chris Lattner } 1855f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 1865f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Read the '"..."'. 1875f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer Lex(Tok); 1883692b09faa9fe346f39bc922db6dce48cdcc3f63Chris Lattner if (Tok.isNot(tok::string_literal) && Tok.isNot(tok::wide_string_literal)) { 1893692b09faa9fe346f39bc922db6dce48cdcc3f63Chris Lattner Diag(PragmaLoc, diag::err__Pragma_malformed); 19099831e4677a7e2e051af636221694d60ba31fcdbRichard Smith // Skip this token, and the ')', if present. 19199831e4677a7e2e051af636221694d60ba31fcdbRichard Smith if (Tok.isNot(tok::r_paren)) 19299831e4677a7e2e051af636221694d60ba31fcdbRichard Smith Lex(Tok); 19399831e4677a7e2e051af636221694d60ba31fcdbRichard Smith if (Tok.is(tok::r_paren)) 19499831e4677a7e2e051af636221694d60ba31fcdbRichard Smith Lex(Tok); 19514e645557ae91c6770d62beb00a1c522e0bfd5d6Argyrios Kyrtzidis return _PragmaLexing.failed(); 19699831e4677a7e2e051af636221694d60ba31fcdbRichard Smith } 19799831e4677a7e2e051af636221694d60ba31fcdbRichard Smith 19899831e4677a7e2e051af636221694d60ba31fcdbRichard Smith if (Tok.hasUDSuffix()) { 19999831e4677a7e2e051af636221694d60ba31fcdbRichard Smith Diag(Tok, diag::err_invalid_string_udl); 20099831e4677a7e2e051af636221694d60ba31fcdbRichard Smith // Skip this token, and the ')', if present. 20199831e4677a7e2e051af636221694d60ba31fcdbRichard Smith Lex(Tok); 20299831e4677a7e2e051af636221694d60ba31fcdbRichard Smith if (Tok.is(tok::r_paren)) 20399831e4677a7e2e051af636221694d60ba31fcdbRichard Smith Lex(Tok); 20414e645557ae91c6770d62beb00a1c522e0bfd5d6Argyrios Kyrtzidis return _PragmaLexing.failed(); 2053692b09faa9fe346f39bc922db6dce48cdcc3f63Chris Lattner } 2061eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 2075f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Remember the string. 20814e645557ae91c6770d62beb00a1c522e0bfd5d6Argyrios Kyrtzidis Token StrTok = Tok; 2095f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 2105f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Read the ')'. 2115f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer Lex(Tok); 2123692b09faa9fe346f39bc922db6dce48cdcc3f63Chris Lattner if (Tok.isNot(tok::r_paren)) { 2133692b09faa9fe346f39bc922db6dce48cdcc3f63Chris Lattner Diag(PragmaLoc, diag::err__Pragma_malformed); 21414e645557ae91c6770d62beb00a1c522e0bfd5d6Argyrios Kyrtzidis return _PragmaLexing.failed(); 2153692b09faa9fe346f39bc922db6dce48cdcc3f63Chris Lattner } 2161eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 21714e645557ae91c6770d62beb00a1c522e0bfd5d6Argyrios Kyrtzidis if (InMacroArgPreExpansion) 21814e645557ae91c6770d62beb00a1c522e0bfd5d6Argyrios Kyrtzidis return; 21914e645557ae91c6770d62beb00a1c522e0bfd5d6Argyrios Kyrtzidis 220e7fb48466afcbf2c4ccdfa658824282fdc3c512cChris Lattner SourceLocation RParenLoc = Tok.getLocation(); 22114e645557ae91c6770d62beb00a1c522e0bfd5d6Argyrios Kyrtzidis std::string StrVal = getSpelling(StrTok); 2221eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 223a9d9145741ef77db45890911674705b81605b10bChris Lattner // The _Pragma is lexically sound. Destringize according to C99 6.10.9.1: 224a9d9145741ef77db45890911674705b81605b10bChris Lattner // "The string literal is destringized by deleting the L prefix, if present, 225a9d9145741ef77db45890911674705b81605b10bChris Lattner // deleting the leading and trailing double-quotes, replacing each escape 226a9d9145741ef77db45890911674705b81605b10bChris Lattner // sequence \" by a double-quote, and replacing each escape sequence \\ by a 227a9d9145741ef77db45890911674705b81605b10bChris Lattner // single backslash." 2285f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer if (StrVal[0] == 'L') // Remove L prefix. 2295f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer StrVal.erase(StrVal.begin()); 2305f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer assert(StrVal[0] == '"' && StrVal[StrVal.size()-1] == '"' && 2315f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer "Invalid string token!"); 2321eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 2335f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Remove the front quote, replacing it with a space, so that the pragma 2345f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // contents appear to have a space before them. 2355f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer StrVal[0] = ' '; 2361eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 2371fa495304c81e03f07f278a47b5efe9317104aabChris Lattner // Replace the terminating quote with a \n. 2385f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer StrVal[StrVal.size()-1] = '\n'; 2391eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 2405f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Remove escaped quotes and escapes. 2415f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer for (unsigned i = 0, e = StrVal.size(); i != e-1; ++i) { 2425f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer if (StrVal[i] == '\\' && 2435f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer (StrVal[i+1] == '\\' || StrVal[i+1] == '"')) { 2445f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // \\ -> '\' and \" -> '"'. 2455f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer StrVal.erase(StrVal.begin()+i); 2465f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer --e; 2475f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 2485f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 2491ef8a2e7675f3d8b6e8d9963b00378086e1dcdc7John McCall 250a5ef584fd3d18da0c98342b4b6453948b7eb30d3Peter Collingbourne // Plop the string (including the newline and trailing null) into a buffer 251a5ef584fd3d18da0c98342b4b6453948b7eb30d3Peter Collingbourne // where we can lex it. 252a5ef584fd3d18da0c98342b4b6453948b7eb30d3Peter Collingbourne Token TmpTok; 253a5ef584fd3d18da0c98342b4b6453948b7eb30d3Peter Collingbourne TmpTok.startToken(); 254374b3837d676133fcc1eb70a25c8baf8ec4a5c4aDmitri Gribenko CreateString(StrVal, TmpTok); 255a5ef584fd3d18da0c98342b4b6453948b7eb30d3Peter Collingbourne SourceLocation TokLoc = TmpTok.getLocation(); 256a5ef584fd3d18da0c98342b4b6453948b7eb30d3Peter Collingbourne 257a5ef584fd3d18da0c98342b4b6453948b7eb30d3Peter Collingbourne // Make and enter a lexer object so that we lex and expand the tokens just 258a5ef584fd3d18da0c98342b4b6453948b7eb30d3Peter Collingbourne // like any others. 259a5ef584fd3d18da0c98342b4b6453948b7eb30d3Peter Collingbourne Lexer *TL = Lexer::Create_PragmaLexer(TokLoc, PragmaLoc, RParenLoc, 260a5ef584fd3d18da0c98342b4b6453948b7eb30d3Peter Collingbourne StrVal.size(), *this); 261a5ef584fd3d18da0c98342b4b6453948b7eb30d3Peter Collingbourne 262a5ef584fd3d18da0c98342b4b6453948b7eb30d3Peter Collingbourne EnterSourceFileWithLexer(TL, 0); 263a5ef584fd3d18da0c98342b4b6453948b7eb30d3Peter Collingbourne 264a5ef584fd3d18da0c98342b4b6453948b7eb30d3Peter Collingbourne // With everything set up, lex this as a #pragma directive. 265a5ef584fd3d18da0c98342b4b6453948b7eb30d3Peter Collingbourne HandlePragmaDirective(PIK__Pragma); 2661ef8a2e7675f3d8b6e8d9963b00378086e1dcdc7John McCall 2671ef8a2e7675f3d8b6e8d9963b00378086e1dcdc7John McCall // Finally, return whatever came after the pragma directive. 2681ef8a2e7675f3d8b6e8d9963b00378086e1dcdc7John McCall return Lex(Tok); 2691ef8a2e7675f3d8b6e8d9963b00378086e1dcdc7John McCall} 2701ef8a2e7675f3d8b6e8d9963b00378086e1dcdc7John McCall 2711ef8a2e7675f3d8b6e8d9963b00378086e1dcdc7John McCall/// HandleMicrosoft__pragma - Like Handle_Pragma except the pragma text 2721ef8a2e7675f3d8b6e8d9963b00378086e1dcdc7John McCall/// is not enclosed within a string literal. 2731ef8a2e7675f3d8b6e8d9963b00378086e1dcdc7John McCallvoid Preprocessor::HandleMicrosoft__pragma(Token &Tok) { 2741ef8a2e7675f3d8b6e8d9963b00378086e1dcdc7John McCall // Remember the pragma token location. 2751ef8a2e7675f3d8b6e8d9963b00378086e1dcdc7John McCall SourceLocation PragmaLoc = Tok.getLocation(); 2761ef8a2e7675f3d8b6e8d9963b00378086e1dcdc7John McCall 2771ef8a2e7675f3d8b6e8d9963b00378086e1dcdc7John McCall // Read the '('. 2781ef8a2e7675f3d8b6e8d9963b00378086e1dcdc7John McCall Lex(Tok); 2791ef8a2e7675f3d8b6e8d9963b00378086e1dcdc7John McCall if (Tok.isNot(tok::l_paren)) { 2801ef8a2e7675f3d8b6e8d9963b00378086e1dcdc7John McCall Diag(PragmaLoc, diag::err__Pragma_malformed); 2811ef8a2e7675f3d8b6e8d9963b00378086e1dcdc7John McCall return; 2821ef8a2e7675f3d8b6e8d9963b00378086e1dcdc7John McCall } 2831ef8a2e7675f3d8b6e8d9963b00378086e1dcdc7John McCall 284a5ef584fd3d18da0c98342b4b6453948b7eb30d3Peter Collingbourne // Get the tokens enclosed within the __pragma(), as well as the final ')'. 2855f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner SmallVector<Token, 32> PragmaToks; 2861ef8a2e7675f3d8b6e8d9963b00378086e1dcdc7John McCall int NumParens = 0; 2871ef8a2e7675f3d8b6e8d9963b00378086e1dcdc7John McCall Lex(Tok); 2881ef8a2e7675f3d8b6e8d9963b00378086e1dcdc7John McCall while (Tok.isNot(tok::eof)) { 289a5ef584fd3d18da0c98342b4b6453948b7eb30d3Peter Collingbourne PragmaToks.push_back(Tok); 2901ef8a2e7675f3d8b6e8d9963b00378086e1dcdc7John McCall if (Tok.is(tok::l_paren)) 2911ef8a2e7675f3d8b6e8d9963b00378086e1dcdc7John McCall NumParens++; 2921ef8a2e7675f3d8b6e8d9963b00378086e1dcdc7John McCall else if (Tok.is(tok::r_paren) && NumParens-- == 0) 2931ef8a2e7675f3d8b6e8d9963b00378086e1dcdc7John McCall break; 2941ef8a2e7675f3d8b6e8d9963b00378086e1dcdc7John McCall Lex(Tok); 2951ef8a2e7675f3d8b6e8d9963b00378086e1dcdc7John McCall } 2961ef8a2e7675f3d8b6e8d9963b00378086e1dcdc7John McCall 2973da92a9d21f707c164797dd967ba894b2282b343John McCall if (Tok.is(tok::eof)) { 2983da92a9d21f707c164797dd967ba894b2282b343John McCall Diag(PragmaLoc, diag::err_unterminated___pragma); 2993da92a9d21f707c164797dd967ba894b2282b343John McCall return; 3003da92a9d21f707c164797dd967ba894b2282b343John McCall } 3013da92a9d21f707c164797dd967ba894b2282b343John McCall 302a5ef584fd3d18da0c98342b4b6453948b7eb30d3Peter Collingbourne PragmaToks.front().setFlag(Token::LeadingSpace); 3031ef8a2e7675f3d8b6e8d9963b00378086e1dcdc7John McCall 30484021556baceb76eedf7d44be8ba71d9b8cfaccePeter Collingbourne // Replace the ')' with an EOD to mark the end of the pragma. 30584021556baceb76eedf7d44be8ba71d9b8cfaccePeter Collingbourne PragmaToks.back().setKind(tok::eod); 3061ef8a2e7675f3d8b6e8d9963b00378086e1dcdc7John McCall 307a5ef584fd3d18da0c98342b4b6453948b7eb30d3Peter Collingbourne Token *TokArray = new Token[PragmaToks.size()]; 308a5ef584fd3d18da0c98342b4b6453948b7eb30d3Peter Collingbourne std::copy(PragmaToks.begin(), PragmaToks.end(), TokArray); 3091eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 310a5ef584fd3d18da0c98342b4b6453948b7eb30d3Peter Collingbourne // Push the tokens onto the stack. 311a5ef584fd3d18da0c98342b4b6453948b7eb30d3Peter Collingbourne EnterTokenStream(TokArray, PragmaToks.size(), true, true); 3125f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 3135f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // With everything set up, lex this as a #pragma directive. 314a5ef584fd3d18da0c98342b4b6453948b7eb30d3Peter Collingbourne HandlePragmaDirective(PIK___pragma); 3155f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 316a5ef584fd3d18da0c98342b4b6453948b7eb30d3Peter Collingbourne // Finally, return whatever came after the pragma directive. 317a5ef584fd3d18da0c98342b4b6453948b7eb30d3Peter Collingbourne return Lex(Tok); 318a5ef584fd3d18da0c98342b4b6453948b7eb30d3Peter Collingbourne} 3195f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 320b6e95b7752893f66e9d067b4287dff2dbf4a5802James Dennett/// HandlePragmaOnce - Handle \#pragma once. OnceTok is the 'once'. 3215f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// 322d217773f106856a11879ec79dc468efefaf2ee75Chris Lattnervoid Preprocessor::HandlePragmaOnce(Token &OnceTok) { 3235f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer if (isInPrimaryFile()) { 3245f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer Diag(OnceTok, diag::pp_pragma_once_in_main_file); 3255f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer return; 3265f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 3271eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 3285f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Get the current file lexer we're looking at. Ignore _Pragma 'files' etc. 3295f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Mark the file as a once-only file now. 3302b2453a7d8fe732561795431f39ceb2b2a832d84Chris Lattner HeaderInfo.MarkFileIncludeOnce(getCurrentFileLexer()->getFileEntry()); 3315f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer} 3325f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 3332243449253475574fc6f14986ff8f7fce5d46799Chris Lattnervoid Preprocessor::HandlePragmaMark() { 33417ff58a63197b398ae52697b088dc0fb8b255519Ted Kremenek assert(CurPPLexer && "No current lexer?"); 3356896a371e302c63ce3d4a99ad2c32e42dde6a9e4Chris Lattner if (CurLexer) 3366896a371e302c63ce3d4a99ad2c32e42dde6a9e4Chris Lattner CurLexer->ReadToEndOfLine(); 3376896a371e302c63ce3d4a99ad2c32e42dde6a9e4Chris Lattner else 3386896a371e302c63ce3d4a99ad2c32e42dde6a9e4Chris Lattner CurPTHLexer->DiscardToEndOfLine(); 3392243449253475574fc6f14986ff8f7fce5d46799Chris Lattner} 3402243449253475574fc6f14986ff8f7fce5d46799Chris Lattner 3412243449253475574fc6f14986ff8f7fce5d46799Chris Lattner 342b6e95b7752893f66e9d067b4287dff2dbf4a5802James Dennett/// HandlePragmaPoison - Handle \#pragma GCC poison. PoisonTok is the 'poison'. 3435f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// 344d217773f106856a11879ec79dc468efefaf2ee75Chris Lattnervoid Preprocessor::HandlePragmaPoison(Token &PoisonTok) { 345d217773f106856a11879ec79dc468efefaf2ee75Chris Lattner Token Tok; 3465f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 3475f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer while (1) { 3485f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Read the next token to poison. While doing this, pretend that we are 3495f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // skipping while reading the identifier to poison. 3505f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // This avoids errors on code like: 3515f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // #pragma GCC poison X 3525f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // #pragma GCC poison X 35368a91d5736c2c03d60a9d1c59d08191b8e0d6c52Ted Kremenek if (CurPPLexer) CurPPLexer->LexingRawMode = true; 3545f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer LexUnexpandedToken(Tok); 35568a91d5736c2c03d60a9d1c59d08191b8e0d6c52Ted Kremenek if (CurPPLexer) CurPPLexer->LexingRawMode = false; 3561eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 3575f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // If we reached the end of line, we're done. 35884021556baceb76eedf7d44be8ba71d9b8cfaccePeter Collingbourne if (Tok.is(tok::eod)) return; 3591eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 3605f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Can only poison identifiers. 361c4bf2b9afb7d47445a9dc6bc848657098a4e3851Abramo Bagnara if (Tok.isNot(tok::raw_identifier)) { 3625f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer Diag(Tok, diag::err_pp_invalid_poison); 3635f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer return; 3645f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 3651eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 3665f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Look up the identifier info for the token. We disabled identifier lookup 3675f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // by saying we're skipping contents, so we need to do this manually. 3685f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer IdentifierInfo *II = LookUpIdentifierInfo(Tok); 3691eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 3705f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Already poisoned. 3715f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer if (II->isPoisoned()) continue; 3721eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 3735f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // If this is a macro identifier, emit a warning. 3740edde55077cc3cb9fffeba19f5936f05a68c8e2bChris Lattner if (II->hasMacroDefinition()) 3755f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer Diag(Tok, diag::pp_poisoning_existing_macro); 3761eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 3775f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Finally, poison it! 3785f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer II->setIsPoisoned(); 379eee242ff426bf79149f221798966e58688383c1eDouglas Gregor if (II->isFromAST()) 380eee242ff426bf79149f221798966e58688383c1eDouglas Gregor II->setChangedSinceDeserialization(); 3815f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 3825f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer} 3835f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 384b6e95b7752893f66e9d067b4287dff2dbf4a5802James Dennett/// HandlePragmaSystemHeader - Implement \#pragma GCC system_header. We know 3855f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// that the whole directive has been parsed. 386d217773f106856a11879ec79dc468efefaf2ee75Chris Lattnervoid Preprocessor::HandlePragmaSystemHeader(Token &SysHeaderTok) { 3875f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer if (isInPrimaryFile()) { 3885f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer Diag(SysHeaderTok, diag::pp_pragma_sysheader_in_main_file); 3895f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer return; 3905f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 3911eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 3925f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Get the current file lexer we're looking at. Ignore _Pragma 'files' etc. 39335c10c25ddec4effbd26dead23ea5b04ee32f45aTed Kremenek PreprocessorLexer *TheLexer = getCurrentFileLexer(); 3941eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 3955f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Mark the file as a system header. 3962b2453a7d8fe732561795431f39ceb2b2a832d84Chris Lattner HeaderInfo.MarkFileSystemHeader(TheLexer->getFileEntry()); 3971eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 3981eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 3996896a371e302c63ce3d4a99ad2c32e42dde6a9e4Chris Lattner PresumedLoc PLoc = SourceMgr.getPresumedLoc(SysHeaderTok.getLocation()); 400cb7b1e17b63967317ab5cc55682168cf0380519aDouglas Gregor if (PLoc.isInvalid()) 401cb7b1e17b63967317ab5cc55682168cf0380519aDouglas Gregor return; 402cb7b1e17b63967317ab5cc55682168cf0380519aDouglas Gregor 40365aa6885818d4b4eea2e5a9d12085b2398148662Jay Foad unsigned FilenameID = SourceMgr.getLineTableFilenameID(PLoc.getFilename()); 4041eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 405784c257c8f5c3a737b0ae1dceb5d54d29b6637cfChris Lattner // Notify the client, if desired, that we are in a new source file. 406784c257c8f5c3a737b0ae1dceb5d54d29b6637cfChris Lattner if (Callbacks) 407784c257c8f5c3a737b0ae1dceb5d54d29b6637cfChris Lattner Callbacks->FileChanged(SysHeaderTok.getLocation(), 408784c257c8f5c3a737b0ae1dceb5d54d29b6637cfChris Lattner PPCallbacks::SystemHeaderPragma, SrcMgr::C_System); 409784c257c8f5c3a737b0ae1dceb5d54d29b6637cfChris Lattner 4106896a371e302c63ce3d4a99ad2c32e42dde6a9e4Chris Lattner // Emit a line marker. This will change any source locations from this point 4116896a371e302c63ce3d4a99ad2c32e42dde6a9e4Chris Lattner // forward to realize they are in a system header. 4126896a371e302c63ce3d4a99ad2c32e42dde6a9e4Chris Lattner // Create a line note with this information. 4136896a371e302c63ce3d4a99ad2c32e42dde6a9e4Chris Lattner SourceMgr.AddLineNote(SysHeaderTok.getLocation(), PLoc.getLine(), FilenameID, 4146896a371e302c63ce3d4a99ad2c32e42dde6a9e4Chris Lattner false, false, true, false); 4155f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer} 4165f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 417b6e95b7752893f66e9d067b4287dff2dbf4a5802James Dennett/// HandlePragmaDependency - Handle \#pragma GCC dependency "foo" blah. 4185f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// 419d217773f106856a11879ec79dc468efefaf2ee75Chris Lattnervoid Preprocessor::HandlePragmaDependency(Token &DependencyTok) { 420d217773f106856a11879ec79dc468efefaf2ee75Chris Lattner Token FilenameTok; 42168a91d5736c2c03d60a9d1c59d08191b8e0d6c52Ted Kremenek CurPPLexer->LexIncludeFilename(FilenameTok); 4225f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 42384021556baceb76eedf7d44be8ba71d9b8cfaccePeter Collingbourne // If the token kind is EOD, the error has already been diagnosed. 42484021556baceb76eedf7d44be8ba71d9b8cfaccePeter Collingbourne if (FilenameTok.is(tok::eod)) 4255f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer return; 4261eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 4275f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Reserve a buffer to get the spelling. 428f7ccbad5d9949e7ddd1cbef43d482553b811e026Dylan Noblesmith SmallString<128> FilenameBuffer; 429453091cc2082e207ea2c2dda645a9bc01b37fb0cDouglas Gregor bool Invalid = false; 4305f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner StringRef Filename = getSpelling(FilenameTok, FilenameBuffer, &Invalid); 431453091cc2082e207ea2c2dda645a9bc01b37fb0cDouglas Gregor if (Invalid) 432453091cc2082e207ea2c2dda645a9bc01b37fb0cDouglas Gregor return; 4331eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 434a139481e62fdb209d9d87a54a5733f989d2e8d51Chris Lattner bool isAngled = 435a139481e62fdb209d9d87a54a5733f989d2e8d51Chris Lattner GetIncludeFilenameSpelling(FilenameTok.getLocation(), Filename); 4365f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // If GetIncludeFilenameSpelling set the start ptr to null, there was an 4375f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // error. 438a139481e62fdb209d9d87a54a5733f989d2e8d51Chris Lattner if (Filename.empty()) 4395f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer return; 4401eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 4415f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Search include directories for this file. 4425f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer const DirectoryLookup *CurDir; 443fba18aa8f2cd1994dc65e8cb9f4be201c560dc0bDouglas Gregor const FileEntry *File = LookupFile(Filename, isAngled, 0, CurDir, NULL, NULL, 444fba18aa8f2cd1994dc65e8cb9f4be201c560dc0bDouglas Gregor NULL); 44556b05c8829efd13b7e5b333f8f587c71d025c67dChris Lattner if (File == 0) { 446f84139a1331c63c998e8b7d54148c75ac0b48ccdEli Friedman if (!SuppressIncludeNotFoundError) 447f84139a1331c63c998e8b7d54148c75ac0b48ccdEli Friedman Diag(FilenameTok, diag::err_pp_file_not_found) << Filename; 44856b05c8829efd13b7e5b333f8f587c71d025c67dChris Lattner return; 44956b05c8829efd13b7e5b333f8f587c71d025c67dChris Lattner } 4501eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 4512b2453a7d8fe732561795431f39ceb2b2a832d84Chris Lattner const FileEntry *CurFile = getCurrentFileLexer()->getFileEntry(); 4525f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 4535f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // If this file is older than the file it depends on, emit a diagnostic. 4545f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer if (CurFile && CurFile->getModificationTime() < File->getModificationTime()) { 4555f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Lex tokens at the end of the message and include them in the message. 4565f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer std::string Message; 4575f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer Lex(DependencyTok); 45884021556baceb76eedf7d44be8ba71d9b8cfaccePeter Collingbourne while (DependencyTok.isNot(tok::eod)) { 4595f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer Message += getSpelling(DependencyTok) + " "; 4605f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer Lex(DependencyTok); 4615f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 4621eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 46396de25986d28171945db96a6c6a54f91fd12ae4bChris Lattner // Remove the trailing ' ' if present. 46496de25986d28171945db96a6c6a54f91fd12ae4bChris Lattner if (!Message.empty()) 46596de25986d28171945db96a6c6a54f91fd12ae4bChris Lattner Message.erase(Message.end()-1); 46656b05c8829efd13b7e5b333f8f587c71d025c67dChris Lattner Diag(FilenameTok, diag::pp_out_of_date_dependency) << Message; 4675f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 4685f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer} 4695f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 470b6e95b7752893f66e9d067b4287dff2dbf4a5802James Dennett/// \brief Handle the microsoft \#pragma comment extension. 471b6e95b7752893f66e9d067b4287dff2dbf4a5802James Dennett/// 472b6e95b7752893f66e9d067b4287dff2dbf4a5802James Dennett/// The syntax is: 473b6e95b7752893f66e9d067b4287dff2dbf4a5802James Dennett/// \code 474b6e95b7752893f66e9d067b4287dff2dbf4a5802James Dennett/// \#pragma comment(linker, "foo") 475b6e95b7752893f66e9d067b4287dff2dbf4a5802James Dennett/// \endcode 476636c5ef6572e899d36cec1b0023fb28ba65189e1Chris Lattner/// 'linker' is one of five identifiers: compiler, exestr, lib, linker, user. 477636c5ef6572e899d36cec1b0023fb28ba65189e1Chris Lattner/// "foo" is a string, which is fully macro expanded, and permits string 478d7ee349f1dd84e023d09a35fb4d40c15888a55acGabor Greif/// concatenation, embedded escape characters etc. See MSDN for more details. 479636c5ef6572e899d36cec1b0023fb28ba65189e1Chris Lattnervoid Preprocessor::HandlePragmaComment(Token &Tok) { 480636c5ef6572e899d36cec1b0023fb28ba65189e1Chris Lattner SourceLocation CommentLoc = Tok.getLocation(); 481636c5ef6572e899d36cec1b0023fb28ba65189e1Chris Lattner Lex(Tok); 482636c5ef6572e899d36cec1b0023fb28ba65189e1Chris Lattner if (Tok.isNot(tok::l_paren)) { 483636c5ef6572e899d36cec1b0023fb28ba65189e1Chris Lattner Diag(CommentLoc, diag::err_pragma_comment_malformed); 484636c5ef6572e899d36cec1b0023fb28ba65189e1Chris Lattner return; 485636c5ef6572e899d36cec1b0023fb28ba65189e1Chris Lattner } 4861eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 487636c5ef6572e899d36cec1b0023fb28ba65189e1Chris Lattner // Read the identifier. 488636c5ef6572e899d36cec1b0023fb28ba65189e1Chris Lattner Lex(Tok); 489636c5ef6572e899d36cec1b0023fb28ba65189e1Chris Lattner if (Tok.isNot(tok::identifier)) { 490636c5ef6572e899d36cec1b0023fb28ba65189e1Chris Lattner Diag(CommentLoc, diag::err_pragma_comment_malformed); 491636c5ef6572e899d36cec1b0023fb28ba65189e1Chris Lattner return; 492636c5ef6572e899d36cec1b0023fb28ba65189e1Chris Lattner } 4931eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 494636c5ef6572e899d36cec1b0023fb28ba65189e1Chris Lattner // Verify that this is one of the 5 whitelisted options. 495636c5ef6572e899d36cec1b0023fb28ba65189e1Chris Lattner // FIXME: warn that 'exestr' is deprecated. 496636c5ef6572e899d36cec1b0023fb28ba65189e1Chris Lattner const IdentifierInfo *II = Tok.getIdentifierInfo(); 4971eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump if (!II->isStr("compiler") && !II->isStr("exestr") && !II->isStr("lib") && 498636c5ef6572e899d36cec1b0023fb28ba65189e1Chris Lattner !II->isStr("linker") && !II->isStr("user")) { 499636c5ef6572e899d36cec1b0023fb28ba65189e1Chris Lattner Diag(Tok.getLocation(), diag::err_pragma_comment_unknown_kind); 500636c5ef6572e899d36cec1b0023fb28ba65189e1Chris Lattner return; 501636c5ef6572e899d36cec1b0023fb28ba65189e1Chris Lattner } 5021eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 503a9d9145741ef77db45890911674705b81605b10bChris Lattner // Read the optional string if present. 504636c5ef6572e899d36cec1b0023fb28ba65189e1Chris Lattner Lex(Tok); 505a9d9145741ef77db45890911674705b81605b10bChris Lattner std::string ArgumentString; 506636c5ef6572e899d36cec1b0023fb28ba65189e1Chris Lattner if (Tok.is(tok::comma)) { 507a9d9145741ef77db45890911674705b81605b10bChris Lattner Lex(Tok); // eat the comma. 508636c5ef6572e899d36cec1b0023fb28ba65189e1Chris Lattner 509636c5ef6572e899d36cec1b0023fb28ba65189e1Chris Lattner // We need at least one string. 510edaf877d2469147ba818187a3e57a00b3f73b123Chris Lattner if (Tok.isNot(tok::string_literal)) { 511636c5ef6572e899d36cec1b0023fb28ba65189e1Chris Lattner Diag(Tok.getLocation(), diag::err_pragma_comment_malformed); 512636c5ef6572e899d36cec1b0023fb28ba65189e1Chris Lattner return; 513636c5ef6572e899d36cec1b0023fb28ba65189e1Chris Lattner } 514636c5ef6572e899d36cec1b0023fb28ba65189e1Chris Lattner 515636c5ef6572e899d36cec1b0023fb28ba65189e1Chris Lattner // String concatenation allows multiple strings, which can even come from 516636c5ef6572e899d36cec1b0023fb28ba65189e1Chris Lattner // macro expansion. 517636c5ef6572e899d36cec1b0023fb28ba65189e1Chris Lattner // "foo " "bar" "Baz" 5185f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner SmallVector<Token, 4> StrToks; 519edaf877d2469147ba818187a3e57a00b3f73b123Chris Lattner while (Tok.is(tok::string_literal)) { 52099831e4677a7e2e051af636221694d60ba31fcdbRichard Smith if (Tok.hasUDSuffix()) 52199831e4677a7e2e051af636221694d60ba31fcdbRichard Smith Diag(Tok, diag::err_invalid_string_udl); 522a9d9145741ef77db45890911674705b81605b10bChris Lattner StrToks.push_back(Tok); 523636c5ef6572e899d36cec1b0023fb28ba65189e1Chris Lattner Lex(Tok); 524a9d9145741ef77db45890911674705b81605b10bChris Lattner } 525a9d9145741ef77db45890911674705b81605b10bChris Lattner 526a9d9145741ef77db45890911674705b81605b10bChris Lattner // Concatenate and parse the strings. 527a9d9145741ef77db45890911674705b81605b10bChris Lattner StringLiteralParser Literal(&StrToks[0], StrToks.size(), *this); 5285cee1195584fa8672253139c86e922daeda69b9eDouglas Gregor assert(Literal.isAscii() && "Didn't allow wide strings in"); 529a9d9145741ef77db45890911674705b81605b10bChris Lattner if (Literal.hadError) 530a9d9145741ef77db45890911674705b81605b10bChris Lattner return; 531a9d9145741ef77db45890911674705b81605b10bChris Lattner if (Literal.Pascal) { 532a9d9145741ef77db45890911674705b81605b10bChris Lattner Diag(StrToks[0].getLocation(), diag::err_pragma_comment_malformed); 533a9d9145741ef77db45890911674705b81605b10bChris Lattner return; 534a9d9145741ef77db45890911674705b81605b10bChris Lattner } 535a9d9145741ef77db45890911674705b81605b10bChris Lattner 53665aa6885818d4b4eea2e5a9d12085b2398148662Jay Foad ArgumentString = Literal.GetString(); 537636c5ef6572e899d36cec1b0023fb28ba65189e1Chris Lattner } 5381eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 539a9d9145741ef77db45890911674705b81605b10bChris Lattner // FIXME: If the kind is "compiler" warn if the string is present (it is 540a9d9145741ef77db45890911674705b81605b10bChris Lattner // ignored). 541a9d9145741ef77db45890911674705b81605b10bChris Lattner // FIXME: 'lib' requires a comment string. 542a9d9145741ef77db45890911674705b81605b10bChris Lattner // FIXME: 'linker' requires a comment string, and has a specific list of 543a9d9145741ef77db45890911674705b81605b10bChris Lattner // things that are allowable. 5441eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 545636c5ef6572e899d36cec1b0023fb28ba65189e1Chris Lattner if (Tok.isNot(tok::r_paren)) { 546636c5ef6572e899d36cec1b0023fb28ba65189e1Chris Lattner Diag(Tok.getLocation(), diag::err_pragma_comment_malformed); 547636c5ef6572e899d36cec1b0023fb28ba65189e1Chris Lattner return; 548636c5ef6572e899d36cec1b0023fb28ba65189e1Chris Lattner } 549a9d9145741ef77db45890911674705b81605b10bChris Lattner Lex(Tok); // eat the r_paren. 550636c5ef6572e899d36cec1b0023fb28ba65189e1Chris Lattner 55184021556baceb76eedf7d44be8ba71d9b8cfaccePeter Collingbourne if (Tok.isNot(tok::eod)) { 552636c5ef6572e899d36cec1b0023fb28ba65189e1Chris Lattner Diag(Tok.getLocation(), diag::err_pragma_comment_malformed); 553636c5ef6572e899d36cec1b0023fb28ba65189e1Chris Lattner return; 554636c5ef6572e899d36cec1b0023fb28ba65189e1Chris Lattner } 5551eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 556a9d9145741ef77db45890911674705b81605b10bChris Lattner // If the pragma is lexically sound, notify any interested PPCallbacks. 557172e336b12fa32a1737f66d77a3f2de86f9cc8e8Chris Lattner if (Callbacks) 558172e336b12fa32a1737f66d77a3f2de86f9cc8e8Chris Lattner Callbacks->PragmaComment(CommentLoc, II, ArgumentString); 559636c5ef6572e899d36cec1b0023fb28ba65189e1Chris Lattner} 560636c5ef6572e899d36cec1b0023fb28ba65189e1Chris Lattner 561b6e95b7752893f66e9d067b4287dff2dbf4a5802James Dennett/// HandlePragmaMessage - Handle the microsoft and gcc \#pragma message 562301669b381dfead44bda5ea3211d69ffec5a530bMichael J. Spencer/// extension. The syntax is: 563b6e95b7752893f66e9d067b4287dff2dbf4a5802James Dennett/// \code 564b6e95b7752893f66e9d067b4287dff2dbf4a5802James Dennett/// \#pragma message(string) 565b6e95b7752893f66e9d067b4287dff2dbf4a5802James Dennett/// \endcode 566301669b381dfead44bda5ea3211d69ffec5a530bMichael J. Spencer/// OR, in GCC mode: 567b6e95b7752893f66e9d067b4287dff2dbf4a5802James Dennett/// \code 568b6e95b7752893f66e9d067b4287dff2dbf4a5802James Dennett/// \#pragma message string 569b6e95b7752893f66e9d067b4287dff2dbf4a5802James Dennett/// \endcode 570301669b381dfead44bda5ea3211d69ffec5a530bMichael J. Spencer/// string is a string, which is fully macro expanded, and permits string 571301669b381dfead44bda5ea3211d69ffec5a530bMichael J. Spencer/// concatenation, embedded escape characters, etc... See MSDN for more details. 572abfe094ce71c42656dcb84a3bdc3e79cb3c16fc3Chris Lattnervoid Preprocessor::HandlePragmaMessage(Token &Tok) { 573abfe094ce71c42656dcb84a3bdc3e79cb3c16fc3Chris Lattner SourceLocation MessageLoc = Tok.getLocation(); 574abfe094ce71c42656dcb84a3bdc3e79cb3c16fc3Chris Lattner Lex(Tok); 575301669b381dfead44bda5ea3211d69ffec5a530bMichael J. Spencer bool ExpectClosingParen = false; 576d83fc546a4351df49a8d106daaecb959e7278a86Michael J. Spencer switch (Tok.getKind()) { 577301669b381dfead44bda5ea3211d69ffec5a530bMichael J. Spencer case tok::l_paren: 578301669b381dfead44bda5ea3211d69ffec5a530bMichael J. Spencer // We have a MSVC style pragma message. 579301669b381dfead44bda5ea3211d69ffec5a530bMichael J. Spencer ExpectClosingParen = true; 580301669b381dfead44bda5ea3211d69ffec5a530bMichael J. Spencer // Read the string. 581301669b381dfead44bda5ea3211d69ffec5a530bMichael J. Spencer Lex(Tok); 582301669b381dfead44bda5ea3211d69ffec5a530bMichael J. Spencer break; 583301669b381dfead44bda5ea3211d69ffec5a530bMichael J. Spencer case tok::string_literal: 584301669b381dfead44bda5ea3211d69ffec5a530bMichael J. Spencer // We have a GCC style pragma message, and we just read the string. 585301669b381dfead44bda5ea3211d69ffec5a530bMichael J. Spencer break; 586301669b381dfead44bda5ea3211d69ffec5a530bMichael J. Spencer default: 587abfe094ce71c42656dcb84a3bdc3e79cb3c16fc3Chris Lattner Diag(MessageLoc, diag::err_pragma_message_malformed); 588abfe094ce71c42656dcb84a3bdc3e79cb3c16fc3Chris Lattner return; 589abfe094ce71c42656dcb84a3bdc3e79cb3c16fc3Chris Lattner } 590abfe094ce71c42656dcb84a3bdc3e79cb3c16fc3Chris Lattner 591abfe094ce71c42656dcb84a3bdc3e79cb3c16fc3Chris Lattner // We need at least one string. 592abfe094ce71c42656dcb84a3bdc3e79cb3c16fc3Chris Lattner if (Tok.isNot(tok::string_literal)) { 593abfe094ce71c42656dcb84a3bdc3e79cb3c16fc3Chris Lattner Diag(Tok.getLocation(), diag::err_pragma_message_malformed); 594abfe094ce71c42656dcb84a3bdc3e79cb3c16fc3Chris Lattner return; 595abfe094ce71c42656dcb84a3bdc3e79cb3c16fc3Chris Lattner } 596abfe094ce71c42656dcb84a3bdc3e79cb3c16fc3Chris Lattner 597abfe094ce71c42656dcb84a3bdc3e79cb3c16fc3Chris Lattner // String concatenation allows multiple strings, which can even come from 598abfe094ce71c42656dcb84a3bdc3e79cb3c16fc3Chris Lattner // macro expansion. 599abfe094ce71c42656dcb84a3bdc3e79cb3c16fc3Chris Lattner // "foo " "bar" "Baz" 6005f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner SmallVector<Token, 4> StrToks; 601abfe094ce71c42656dcb84a3bdc3e79cb3c16fc3Chris Lattner while (Tok.is(tok::string_literal)) { 60299831e4677a7e2e051af636221694d60ba31fcdbRichard Smith if (Tok.hasUDSuffix()) 60399831e4677a7e2e051af636221694d60ba31fcdbRichard Smith Diag(Tok, diag::err_invalid_string_udl); 604abfe094ce71c42656dcb84a3bdc3e79cb3c16fc3Chris Lattner StrToks.push_back(Tok); 605abfe094ce71c42656dcb84a3bdc3e79cb3c16fc3Chris Lattner Lex(Tok); 606abfe094ce71c42656dcb84a3bdc3e79cb3c16fc3Chris Lattner } 607abfe094ce71c42656dcb84a3bdc3e79cb3c16fc3Chris Lattner 608abfe094ce71c42656dcb84a3bdc3e79cb3c16fc3Chris Lattner // Concatenate and parse the strings. 609abfe094ce71c42656dcb84a3bdc3e79cb3c16fc3Chris Lattner StringLiteralParser Literal(&StrToks[0], StrToks.size(), *this); 6105cee1195584fa8672253139c86e922daeda69b9eDouglas Gregor assert(Literal.isAscii() && "Didn't allow wide strings in"); 611abfe094ce71c42656dcb84a3bdc3e79cb3c16fc3Chris Lattner if (Literal.hadError) 612abfe094ce71c42656dcb84a3bdc3e79cb3c16fc3Chris Lattner return; 613abfe094ce71c42656dcb84a3bdc3e79cb3c16fc3Chris Lattner if (Literal.Pascal) { 614abfe094ce71c42656dcb84a3bdc3e79cb3c16fc3Chris Lattner Diag(StrToks[0].getLocation(), diag::err_pragma_message_malformed); 615abfe094ce71c42656dcb84a3bdc3e79cb3c16fc3Chris Lattner return; 616abfe094ce71c42656dcb84a3bdc3e79cb3c16fc3Chris Lattner } 617abfe094ce71c42656dcb84a3bdc3e79cb3c16fc3Chris Lattner 6185f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner StringRef MessageString(Literal.GetString()); 619636c5ef6572e899d36cec1b0023fb28ba65189e1Chris Lattner 620301669b381dfead44bda5ea3211d69ffec5a530bMichael J. Spencer if (ExpectClosingParen) { 621301669b381dfead44bda5ea3211d69ffec5a530bMichael J. Spencer if (Tok.isNot(tok::r_paren)) { 622301669b381dfead44bda5ea3211d69ffec5a530bMichael J. Spencer Diag(Tok.getLocation(), diag::err_pragma_message_malformed); 623301669b381dfead44bda5ea3211d69ffec5a530bMichael J. Spencer return; 624301669b381dfead44bda5ea3211d69ffec5a530bMichael J. Spencer } 625301669b381dfead44bda5ea3211d69ffec5a530bMichael J. Spencer Lex(Tok); // eat the r_paren. 626abfe094ce71c42656dcb84a3bdc3e79cb3c16fc3Chris Lattner } 627abfe094ce71c42656dcb84a3bdc3e79cb3c16fc3Chris Lattner 62884021556baceb76eedf7d44be8ba71d9b8cfaccePeter Collingbourne if (Tok.isNot(tok::eod)) { 629abfe094ce71c42656dcb84a3bdc3e79cb3c16fc3Chris Lattner Diag(Tok.getLocation(), diag::err_pragma_message_malformed); 630abfe094ce71c42656dcb84a3bdc3e79cb3c16fc3Chris Lattner return; 631abfe094ce71c42656dcb84a3bdc3e79cb3c16fc3Chris Lattner } 632abfe094ce71c42656dcb84a3bdc3e79cb3c16fc3Chris Lattner 633abfe094ce71c42656dcb84a3bdc3e79cb3c16fc3Chris Lattner // Output the message. 634abfe094ce71c42656dcb84a3bdc3e79cb3c16fc3Chris Lattner Diag(MessageLoc, diag::warn_pragma_message) << MessageString; 635abfe094ce71c42656dcb84a3bdc3e79cb3c16fc3Chris Lattner 636abfe094ce71c42656dcb84a3bdc3e79cb3c16fc3Chris Lattner // If the pragma is lexically sound, notify any interested PPCallbacks. 637abfe094ce71c42656dcb84a3bdc3e79cb3c16fc3Chris Lattner if (Callbacks) 638abfe094ce71c42656dcb84a3bdc3e79cb3c16fc3Chris Lattner Callbacks->PragmaMessage(MessageLoc, MessageString); 639abfe094ce71c42656dcb84a3bdc3e79cb3c16fc3Chris Lattner} 640636c5ef6572e899d36cec1b0023fb28ba65189e1Chris Lattner 641f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner/// ParsePragmaPushOrPopMacro - Handle parsing of pragma push_macro/pop_macro. 642f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner/// Return the IdentifierInfo* associated with the macro to push or pop. 643f47724bf78299c7a50f008e0443c5f9f9f279ddcChris LattnerIdentifierInfo *Preprocessor::ParsePragmaPushOrPopMacro(Token &Tok) { 644f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner // Remember the pragma token location. 645f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner Token PragmaTok = Tok; 646f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner 647f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner // Read the '('. 648f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner Lex(Tok); 649f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner if (Tok.isNot(tok::l_paren)) { 650f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner Diag(PragmaTok.getLocation(), diag::err_pragma_push_pop_macro_malformed) 651f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner << getSpelling(PragmaTok); 652f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner return 0; 653f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner } 654f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner 655f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner // Read the macro name string. 656f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner Lex(Tok); 657f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner if (Tok.isNot(tok::string_literal)) { 658f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner Diag(PragmaTok.getLocation(), diag::err_pragma_push_pop_macro_malformed) 659f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner << getSpelling(PragmaTok); 660f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner return 0; 661f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner } 662f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner 66399831e4677a7e2e051af636221694d60ba31fcdbRichard Smith if (Tok.hasUDSuffix()) { 66499831e4677a7e2e051af636221694d60ba31fcdbRichard Smith Diag(Tok, diag::err_invalid_string_udl); 66599831e4677a7e2e051af636221694d60ba31fcdbRichard Smith return 0; 66699831e4677a7e2e051af636221694d60ba31fcdbRichard Smith } 66799831e4677a7e2e051af636221694d60ba31fcdbRichard Smith 668f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner // Remember the macro string. 669f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner std::string StrVal = getSpelling(Tok); 670f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner 671f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner // Read the ')'. 672f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner Lex(Tok); 673f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner if (Tok.isNot(tok::r_paren)) { 674f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner Diag(PragmaTok.getLocation(), diag::err_pragma_push_pop_macro_malformed) 675f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner << getSpelling(PragmaTok); 676f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner return 0; 677f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner } 678f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner 679f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner assert(StrVal[0] == '"' && StrVal[StrVal.size()-1] == '"' && 680f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner "Invalid string token!"); 681f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner 682f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner // Create a Token from the string. 683f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner Token MacroTok; 684f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner MacroTok.startToken(); 685c4bf2b9afb7d47445a9dc6bc848657098a4e3851Abramo Bagnara MacroTok.setKind(tok::raw_identifier); 686374b3837d676133fcc1eb70a25c8baf8ec4a5c4aDmitri Gribenko CreateString(StringRef(&StrVal[1], StrVal.size() - 2), MacroTok); 687f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner 688f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner // Get the IdentifierInfo of MacroToPushTok. 689f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner return LookUpIdentifierInfo(MacroTok); 690f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner} 691f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner 692b6e95b7752893f66e9d067b4287dff2dbf4a5802James Dennett/// \brief Handle \#pragma push_macro. 693b6e95b7752893f66e9d067b4287dff2dbf4a5802James Dennett/// 694f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner/// The syntax is: 695b6e95b7752893f66e9d067b4287dff2dbf4a5802James Dennett/// \code 696b6e95b7752893f66e9d067b4287dff2dbf4a5802James Dennett/// \#pragma push_macro("macro") 697b6e95b7752893f66e9d067b4287dff2dbf4a5802James Dennett/// \endcode 698f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattnervoid Preprocessor::HandlePragmaPushMacro(Token &PushMacroTok) { 699f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner // Parse the pragma directive and get the macro IdentifierInfo*. 700f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner IdentifierInfo *IdentInfo = ParsePragmaPushOrPopMacro(PushMacroTok); 701f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner if (!IdentInfo) return; 702f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner 703f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner // Get the MacroInfo associated with IdentInfo. 704f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner MacroInfo *MI = getMacroInfo(IdentInfo); 705f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner 706f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner MacroInfo *MacroCopyToPush = 0; 707f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner if (MI) { 708f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner // Make a clone of MI. 709f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner MacroCopyToPush = CloneMacroInfo(*MI); 710f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner 711f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner // Allow the original MacroInfo to be redefined later. 712f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner MI->setIsAllowRedefinitionsWithoutWarning(true); 713f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner } 714f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner 715f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner // Push the cloned MacroInfo so we can retrieve it later. 716f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner PragmaPushMacroInfo[IdentInfo].push_back(MacroCopyToPush); 717f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner} 718f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner 719b6e95b7752893f66e9d067b4287dff2dbf4a5802James Dennett/// \brief Handle \#pragma pop_macro. 720b6e95b7752893f66e9d067b4287dff2dbf4a5802James Dennett/// 721f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner/// The syntax is: 722b6e95b7752893f66e9d067b4287dff2dbf4a5802James Dennett/// \code 723f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner/// #pragma pop_macro("macro") 724b6e95b7752893f66e9d067b4287dff2dbf4a5802James Dennett/// \endcode 725f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattnervoid Preprocessor::HandlePragmaPopMacro(Token &PopMacroTok) { 726f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner SourceLocation MessageLoc = PopMacroTok.getLocation(); 727f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner 728f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner // Parse the pragma directive and get the macro IdentifierInfo*. 729f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner IdentifierInfo *IdentInfo = ParsePragmaPushOrPopMacro(PopMacroTok); 730f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner if (!IdentInfo) return; 731f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner 732f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner // Find the vector<MacroInfo*> associated with the macro. 733f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner llvm::DenseMap<IdentifierInfo*, std::vector<MacroInfo*> >::iterator iter = 734f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner PragmaPushMacroInfo.find(IdentInfo); 735f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner if (iter != PragmaPushMacroInfo.end()) { 7368a64bb58c3b24d7d97895e435bbc0965c99bd3beAlexander Kornienko // Forget the MacroInfo currently associated with IdentInfo. 7378a64bb58c3b24d7d97895e435bbc0965c99bd3beAlexander Kornienko if (MacroInfo *CurrentMI = getMacroInfo(IdentInfo)) { 7380827408865e32789e0ec4b8113a302ccdc531423Argyrios Kyrtzidis if (CurrentMI->isWarnIfUnused()) 7390827408865e32789e0ec4b8113a302ccdc531423Argyrios Kyrtzidis WarnUnusedMacroLocs.erase(CurrentMI->getDefinitionLoc()); 740e40c4238a572bf8241a04e0005f70550cbfc1cfbAlexander Kornienko CurrentMI->setUndefLoc(MessageLoc); 7410827408865e32789e0ec4b8113a302ccdc531423Argyrios Kyrtzidis } 742f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner 743f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner // Get the MacroInfo we want to reinstall. 744f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner MacroInfo *MacroToReInstall = iter->second.back(); 745f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner 746e40c4238a572bf8241a04e0005f70550cbfc1cfbAlexander Kornienko if (MacroToReInstall) { 747e40c4238a572bf8241a04e0005f70550cbfc1cfbAlexander Kornienko // Reinstall the previously pushed macro. 748e40c4238a572bf8241a04e0005f70550cbfc1cfbAlexander Kornienko setMacroInfo(IdentInfo, MacroToReInstall); 749e40c4238a572bf8241a04e0005f70550cbfc1cfbAlexander Kornienko } else if (IdentInfo->hasMacroDefinition()) { 750e40c4238a572bf8241a04e0005f70550cbfc1cfbAlexander Kornienko clearMacroInfo(IdentInfo); 751e40c4238a572bf8241a04e0005f70550cbfc1cfbAlexander Kornienko } 752f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner 753f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner // Pop PragmaPushMacroInfo stack. 754f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner iter->second.pop_back(); 755f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner if (iter->second.size() == 0) 756f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner PragmaPushMacroInfo.erase(iter); 757f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner } else { 758f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner Diag(MessageLoc, diag::warn_pragma_pop_macro_no_push) 759f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner << IdentInfo->getName(); 760f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner } 761f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner} 7625f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 7634c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballmanvoid Preprocessor::HandlePragmaIncludeAlias(Token &Tok) { 7644c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman // We will either get a quoted filename or a bracketed filename, and we 7654c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman // have to track which we got. The first filename is the source name, 7664c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman // and the second name is the mapped filename. If the first is quoted, 7674c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman // the second must be as well (cannot mix and match quotes and brackets). 7684c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman 7694c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman // Get the open paren 7704c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman Lex(Tok); 7714c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman if (Tok.isNot(tok::l_paren)) { 7724c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman Diag(Tok, diag::warn_pragma_include_alias_expected) << "("; 7734c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman return; 7744c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman } 7754c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman 7764c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman // We expect either a quoted string literal, or a bracketed name 7774c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman Token SourceFilenameTok; 7784c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman CurPPLexer->LexIncludeFilename(SourceFilenameTok); 7794c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman if (SourceFilenameTok.is(tok::eod)) { 7804c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman // The diagnostic has already been handled 7814c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman return; 7824c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman } 7834c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman 7844c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman StringRef SourceFileName; 7854c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman SmallString<128> FileNameBuffer; 7864c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman if (SourceFilenameTok.is(tok::string_literal) || 7874c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman SourceFilenameTok.is(tok::angle_string_literal)) { 7884c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman SourceFileName = getSpelling(SourceFilenameTok, FileNameBuffer); 7894c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman } else if (SourceFilenameTok.is(tok::less)) { 7904c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman // This could be a path instead of just a name 7914c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman FileNameBuffer.push_back('<'); 7924c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman SourceLocation End; 7934c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman if (ConcatenateIncludeName(FileNameBuffer, End)) 7944c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman return; // Diagnostic already emitted 7954c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman SourceFileName = FileNameBuffer.str(); 7964c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman } else { 7974c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman Diag(Tok, diag::warn_pragma_include_alias_expected_filename); 7984c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman return; 7994c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman } 8004c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman FileNameBuffer.clear(); 8014c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman 8024c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman // Now we expect a comma, followed by another include name 8034c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman Lex(Tok); 8044c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman if (Tok.isNot(tok::comma)) { 8054c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman Diag(Tok, diag::warn_pragma_include_alias_expected) << ","; 8064c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman return; 8074c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman } 8084c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman 8094c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman Token ReplaceFilenameTok; 8104c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman CurPPLexer->LexIncludeFilename(ReplaceFilenameTok); 8114c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman if (ReplaceFilenameTok.is(tok::eod)) { 8124c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman // The diagnostic has already been handled 8134c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman return; 8144c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman } 8154c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman 8164c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman StringRef ReplaceFileName; 8174c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman if (ReplaceFilenameTok.is(tok::string_literal) || 8184c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman ReplaceFilenameTok.is(tok::angle_string_literal)) { 8194c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman ReplaceFileName = getSpelling(ReplaceFilenameTok, FileNameBuffer); 8204c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman } else if (ReplaceFilenameTok.is(tok::less)) { 8214c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman // This could be a path instead of just a name 8224c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman FileNameBuffer.push_back('<'); 8234c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman SourceLocation End; 8244c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman if (ConcatenateIncludeName(FileNameBuffer, End)) 8254c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman return; // Diagnostic already emitted 8264c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman ReplaceFileName = FileNameBuffer.str(); 8274c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman } else { 8284c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman Diag(Tok, diag::warn_pragma_include_alias_expected_filename); 8294c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman return; 8304c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman } 8314c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman 8324c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman // Finally, we expect the closing paren 8334c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman Lex(Tok); 8344c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman if (Tok.isNot(tok::r_paren)) { 8354c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman Diag(Tok, diag::warn_pragma_include_alias_expected) << ")"; 8364c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman return; 8374c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman } 8384c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman 8394c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman // Now that we have the source and target filenames, we need to make sure 8404c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman // they're both of the same type (angled vs non-angled) 8414c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman StringRef OriginalSource = SourceFileName; 8424c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman 8434c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman bool SourceIsAngled = 8444c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman GetIncludeFilenameSpelling(SourceFilenameTok.getLocation(), 8454c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman SourceFileName); 8464c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman bool ReplaceIsAngled = 8474c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman GetIncludeFilenameSpelling(ReplaceFilenameTok.getLocation(), 8484c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman ReplaceFileName); 8494c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman if (!SourceFileName.empty() && !ReplaceFileName.empty() && 8504c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman (SourceIsAngled != ReplaceIsAngled)) { 8514c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman unsigned int DiagID; 8524c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman if (SourceIsAngled) 8534c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman DiagID = diag::warn_pragma_include_alias_mismatch_angle; 8544c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman else 8554c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman DiagID = diag::warn_pragma_include_alias_mismatch_quote; 8564c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman 8574c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman Diag(SourceFilenameTok.getLocation(), DiagID) 8584c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman << SourceFileName 8594c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman << ReplaceFileName; 8604c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman 8614c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman return; 8624c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman } 8634c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman 8644c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman // Now we can let the include handler know about this mapping 8654c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman getHeaderSearchInfo().AddIncludeAlias(OriginalSource, ReplaceFileName); 8664c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman} 8674c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman 8685f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// AddPragmaHandler - Add the specified pragma handler to the preprocessor. 8695f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// If 'Namespace' is non-null, then it is a token required to exist on the 8705f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// pragma line before the pragma string starts, e.g. "STDC" or "GCC". 8715f9e272e632e951b1efe824cd16acb4d96077930Chris Lattnervoid Preprocessor::AddPragmaHandler(StringRef Namespace, 8725f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer PragmaHandler *Handler) { 8735f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer PragmaNamespace *InsertNS = PragmaHandlers; 8741eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 8755f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // If this is specified to be in a namespace, step down into it. 8769b36c3f0de0105e903130bbda3c4aea7d792c0afArgyrios Kyrtzidis if (!Namespace.empty()) { 8775f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // If there is already a pragma handler with the name of this namespace, 8785f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // we either have an error (directive with the same name as a namespace) or 8795f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // we already have the namespace to insert into. 8809b36c3f0de0105e903130bbda3c4aea7d792c0afArgyrios Kyrtzidis if (PragmaHandler *Existing = PragmaHandlers->FindHandler(Namespace)) { 8815f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer InsertNS = Existing->getIfNamespace(); 8825f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer assert(InsertNS != 0 && "Cannot have a pragma namespace and pragma" 8835f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer " handler with the same name!"); 8845f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } else { 8855f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Otherwise, this namespace doesn't exist yet, create and insert the 8865f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // handler for it. 8879b36c3f0de0105e903130bbda3c4aea7d792c0afArgyrios Kyrtzidis InsertNS = new PragmaNamespace(Namespace); 8885f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer PragmaHandlers->AddPragma(InsertNS); 8895f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 8905f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 8911eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 8925f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Check to make sure we don't already have a pragma for this identifier. 8935f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer assert(!InsertNS->FindHandler(Handler->getName()) && 8945f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer "Pragma handler already exists for this identifier!"); 8955f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer InsertNS->AddPragma(Handler); 8965f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer} 8975f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 8984095080aff204008eefb26b100906c6ca2bc4bb6Daniel Dunbar/// RemovePragmaHandler - Remove the specific pragma handler from the 8994095080aff204008eefb26b100906c6ca2bc4bb6Daniel Dunbar/// preprocessor. If \arg Namespace is non-null, then it should be the 9004095080aff204008eefb26b100906c6ca2bc4bb6Daniel Dunbar/// namespace that \arg Handler was added to. It is an error to remove 9014095080aff204008eefb26b100906c6ca2bc4bb6Daniel Dunbar/// a handler that has not been registered. 9025f9e272e632e951b1efe824cd16acb4d96077930Chris Lattnervoid Preprocessor::RemovePragmaHandler(StringRef Namespace, 9034095080aff204008eefb26b100906c6ca2bc4bb6Daniel Dunbar PragmaHandler *Handler) { 9044095080aff204008eefb26b100906c6ca2bc4bb6Daniel Dunbar PragmaNamespace *NS = PragmaHandlers; 9051eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 9064095080aff204008eefb26b100906c6ca2bc4bb6Daniel Dunbar // If this is specified to be in a namespace, step down into it. 9079b36c3f0de0105e903130bbda3c4aea7d792c0afArgyrios Kyrtzidis if (!Namespace.empty()) { 9089b36c3f0de0105e903130bbda3c4aea7d792c0afArgyrios Kyrtzidis PragmaHandler *Existing = PragmaHandlers->FindHandler(Namespace); 9094095080aff204008eefb26b100906c6ca2bc4bb6Daniel Dunbar assert(Existing && "Namespace containing handler does not exist!"); 9104095080aff204008eefb26b100906c6ca2bc4bb6Daniel Dunbar 9114095080aff204008eefb26b100906c6ca2bc4bb6Daniel Dunbar NS = Existing->getIfNamespace(); 9124095080aff204008eefb26b100906c6ca2bc4bb6Daniel Dunbar assert(NS && "Invalid namespace, registered as a regular pragma handler!"); 9134095080aff204008eefb26b100906c6ca2bc4bb6Daniel Dunbar } 9144095080aff204008eefb26b100906c6ca2bc4bb6Daniel Dunbar 9154095080aff204008eefb26b100906c6ca2bc4bb6Daniel Dunbar NS->RemovePragmaHandler(Handler); 9161eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 9174095080aff204008eefb26b100906c6ca2bc4bb6Daniel Dunbar // If this is a non-default namespace and it is now empty, remove 9184095080aff204008eefb26b100906c6ca2bc4bb6Daniel Dunbar // it. 919ce52bb3dec520ef94574280ac3ef8ad63f4f1ce2Argyrios Kyrtzidis if (NS != PragmaHandlers && NS->IsEmpty()) { 9204095080aff204008eefb26b100906c6ca2bc4bb6Daniel Dunbar PragmaHandlers->RemovePragmaHandler(NS); 921ce52bb3dec520ef94574280ac3ef8ad63f4f1ce2Argyrios Kyrtzidis delete NS; 922ce52bb3dec520ef94574280ac3ef8ad63f4f1ce2Argyrios Kyrtzidis } 9234095080aff204008eefb26b100906c6ca2bc4bb6Daniel Dunbar} 9244095080aff204008eefb26b100906c6ca2bc4bb6Daniel Dunbar 9259d3f5f7550a2fab4178ed01425758c349b73a609Peter Collingbournebool Preprocessor::LexOnOffSwitch(tok::OnOffSwitch &Result) { 9269d3f5f7550a2fab4178ed01425758c349b73a609Peter Collingbourne Token Tok; 9279d3f5f7550a2fab4178ed01425758c349b73a609Peter Collingbourne LexUnexpandedToken(Tok); 9289d3f5f7550a2fab4178ed01425758c349b73a609Peter Collingbourne 9299d3f5f7550a2fab4178ed01425758c349b73a609Peter Collingbourne if (Tok.isNot(tok::identifier)) { 9309d3f5f7550a2fab4178ed01425758c349b73a609Peter Collingbourne Diag(Tok, diag::ext_on_off_switch_syntax); 9319d3f5f7550a2fab4178ed01425758c349b73a609Peter Collingbourne return true; 9329d3f5f7550a2fab4178ed01425758c349b73a609Peter Collingbourne } 9339d3f5f7550a2fab4178ed01425758c349b73a609Peter Collingbourne IdentifierInfo *II = Tok.getIdentifierInfo(); 9349d3f5f7550a2fab4178ed01425758c349b73a609Peter Collingbourne if (II->isStr("ON")) 9359d3f5f7550a2fab4178ed01425758c349b73a609Peter Collingbourne Result = tok::OOS_ON; 9369d3f5f7550a2fab4178ed01425758c349b73a609Peter Collingbourne else if (II->isStr("OFF")) 9379d3f5f7550a2fab4178ed01425758c349b73a609Peter Collingbourne Result = tok::OOS_OFF; 9389d3f5f7550a2fab4178ed01425758c349b73a609Peter Collingbourne else if (II->isStr("DEFAULT")) 9399d3f5f7550a2fab4178ed01425758c349b73a609Peter Collingbourne Result = tok::OOS_DEFAULT; 9409d3f5f7550a2fab4178ed01425758c349b73a609Peter Collingbourne else { 9419d3f5f7550a2fab4178ed01425758c349b73a609Peter Collingbourne Diag(Tok, diag::ext_on_off_switch_syntax); 9429d3f5f7550a2fab4178ed01425758c349b73a609Peter Collingbourne return true; 9439d3f5f7550a2fab4178ed01425758c349b73a609Peter Collingbourne } 9449d3f5f7550a2fab4178ed01425758c349b73a609Peter Collingbourne 94584021556baceb76eedf7d44be8ba71d9b8cfaccePeter Collingbourne // Verify that this is followed by EOD. 9469d3f5f7550a2fab4178ed01425758c349b73a609Peter Collingbourne LexUnexpandedToken(Tok); 94784021556baceb76eedf7d44be8ba71d9b8cfaccePeter Collingbourne if (Tok.isNot(tok::eod)) 94884021556baceb76eedf7d44be8ba71d9b8cfaccePeter Collingbourne Diag(Tok, diag::ext_pragma_syntax_eod); 9499d3f5f7550a2fab4178ed01425758c349b73a609Peter Collingbourne return false; 9509d3f5f7550a2fab4178ed01425758c349b73a609Peter Collingbourne} 9519d3f5f7550a2fab4178ed01425758c349b73a609Peter Collingbourne 9525f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencernamespace { 953b6e95b7752893f66e9d067b4287dff2dbf4a5802James Dennett/// PragmaOnceHandler - "\#pragma once" marks the file as atomically included. 9545f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerstruct PragmaOnceHandler : public PragmaHandler { 9559b36c3f0de0105e903130bbda3c4aea7d792c0afArgyrios Kyrtzidis PragmaOnceHandler() : PragmaHandler("once") {} 95680c60f72848896f867f6b7e664e7060d9e78f019Douglas Gregor virtual void HandlePragma(Preprocessor &PP, PragmaIntroducerKind Introducer, 95780c60f72848896f867f6b7e664e7060d9e78f019Douglas Gregor Token &OnceTok) { 95835410d5cbbb18735d76e00496540d416dc49b577Chris Lattner PP.CheckEndOfDirective("pragma once"); 9595f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer PP.HandlePragmaOnce(OnceTok); 9605f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 9615f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer}; 9625f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 963b6e95b7752893f66e9d067b4287dff2dbf4a5802James Dennett/// PragmaMarkHandler - "\#pragma mark ..." is ignored by the compiler, and the 9642243449253475574fc6f14986ff8f7fce5d46799Chris Lattner/// rest of the line is not lexed. 9652243449253475574fc6f14986ff8f7fce5d46799Chris Lattnerstruct PragmaMarkHandler : public PragmaHandler { 9669b36c3f0de0105e903130bbda3c4aea7d792c0afArgyrios Kyrtzidis PragmaMarkHandler() : PragmaHandler("mark") {} 96780c60f72848896f867f6b7e664e7060d9e78f019Douglas Gregor virtual void HandlePragma(Preprocessor &PP, PragmaIntroducerKind Introducer, 96880c60f72848896f867f6b7e664e7060d9e78f019Douglas Gregor Token &MarkTok) { 9692243449253475574fc6f14986ff8f7fce5d46799Chris Lattner PP.HandlePragmaMark(); 9702243449253475574fc6f14986ff8f7fce5d46799Chris Lattner } 9712243449253475574fc6f14986ff8f7fce5d46799Chris Lattner}; 9722243449253475574fc6f14986ff8f7fce5d46799Chris Lattner 973b6e95b7752893f66e9d067b4287dff2dbf4a5802James Dennett/// PragmaPoisonHandler - "\#pragma poison x" marks x as not usable. 9745f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerstruct PragmaPoisonHandler : public PragmaHandler { 9759b36c3f0de0105e903130bbda3c4aea7d792c0afArgyrios Kyrtzidis PragmaPoisonHandler() : PragmaHandler("poison") {} 97680c60f72848896f867f6b7e664e7060d9e78f019Douglas Gregor virtual void HandlePragma(Preprocessor &PP, PragmaIntroducerKind Introducer, 97780c60f72848896f867f6b7e664e7060d9e78f019Douglas Gregor Token &PoisonTok) { 9785f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer PP.HandlePragmaPoison(PoisonTok); 9795f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 9805f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer}; 9815f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 982b6e95b7752893f66e9d067b4287dff2dbf4a5802James Dennett/// PragmaSystemHeaderHandler - "\#pragma system_header" marks the current file 9832243449253475574fc6f14986ff8f7fce5d46799Chris Lattner/// as a system header, which silences warnings in it. 9845f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerstruct PragmaSystemHeaderHandler : public PragmaHandler { 9859b36c3f0de0105e903130bbda3c4aea7d792c0afArgyrios Kyrtzidis PragmaSystemHeaderHandler() : PragmaHandler("system_header") {} 98680c60f72848896f867f6b7e664e7060d9e78f019Douglas Gregor virtual void HandlePragma(Preprocessor &PP, PragmaIntroducerKind Introducer, 98780c60f72848896f867f6b7e664e7060d9e78f019Douglas Gregor Token &SHToken) { 9885f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer PP.HandlePragmaSystemHeader(SHToken); 98935410d5cbbb18735d76e00496540d416dc49b577Chris Lattner PP.CheckEndOfDirective("pragma"); 9905f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 9915f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer}; 9925f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerstruct PragmaDependencyHandler : public PragmaHandler { 9939b36c3f0de0105e903130bbda3c4aea7d792c0afArgyrios Kyrtzidis PragmaDependencyHandler() : PragmaHandler("dependency") {} 99480c60f72848896f867f6b7e664e7060d9e78f019Douglas Gregor virtual void HandlePragma(Preprocessor &PP, PragmaIntroducerKind Introducer, 99580c60f72848896f867f6b7e664e7060d9e78f019Douglas Gregor Token &DepToken) { 9965f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer PP.HandlePragmaDependency(DepToken); 9975f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 9985f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer}; 9991eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 1000abf7b72a965efba4a5ac6f85c66d5aba0c83f972Daniel Dunbarstruct PragmaDebugHandler : public PragmaHandler { 1001abf7b72a965efba4a5ac6f85c66d5aba0c83f972Daniel Dunbar PragmaDebugHandler() : PragmaHandler("__debug") {} 100280c60f72848896f867f6b7e664e7060d9e78f019Douglas Gregor virtual void HandlePragma(Preprocessor &PP, PragmaIntroducerKind Introducer, 100380c60f72848896f867f6b7e664e7060d9e78f019Douglas Gregor Token &DepToken) { 1004abf7b72a965efba4a5ac6f85c66d5aba0c83f972Daniel Dunbar Token Tok; 1005abf7b72a965efba4a5ac6f85c66d5aba0c83f972Daniel Dunbar PP.LexUnexpandedToken(Tok); 1006abf7b72a965efba4a5ac6f85c66d5aba0c83f972Daniel Dunbar if (Tok.isNot(tok::identifier)) { 10076493a4d7129673f7878da2382dedf4f9abc57e4cDouglas Gregor PP.Diag(Tok, diag::warn_pragma_diagnostic_invalid); 1008abf7b72a965efba4a5ac6f85c66d5aba0c83f972Daniel Dunbar return; 1009abf7b72a965efba4a5ac6f85c66d5aba0c83f972Daniel Dunbar } 1010abf7b72a965efba4a5ac6f85c66d5aba0c83f972Daniel Dunbar IdentifierInfo *II = Tok.getIdentifierInfo(); 1011abf7b72a965efba4a5ac6f85c66d5aba0c83f972Daniel Dunbar 10125505413ee8e4e2924f52ba81181071f3a492e7d9Daniel Dunbar if (II->isStr("assert")) { 1013b219cfc4d75f0a03630b7c4509ef791b7e97b2c8David Blaikie llvm_unreachable("This is an assertion!"); 1014abf7b72a965efba4a5ac6f85c66d5aba0c83f972Daniel Dunbar } else if (II->isStr("crash")) { 1015377da4c6b78c0130fb6141313636c8fda7b60b72David Blaikie LLVM_BUILTIN_TRAP; 1016e75d9cfbf41a0ee9e456a665776f91fdd9773b36David Blaikie } else if (II->isStr("parser_crash")) { 1017e75d9cfbf41a0ee9e456a665776f91fdd9773b36David Blaikie Token Crasher; 1018e75d9cfbf41a0ee9e456a665776f91fdd9773b36David Blaikie Crasher.setKind(tok::annot_pragma_parser_crash); 1019e75d9cfbf41a0ee9e456a665776f91fdd9773b36David Blaikie PP.EnterToken(Crasher); 10205505413ee8e4e2924f52ba81181071f3a492e7d9Daniel Dunbar } else if (II->isStr("llvm_fatal_error")) { 10215505413ee8e4e2924f52ba81181071f3a492e7d9Daniel Dunbar llvm::report_fatal_error("#pragma clang __debug llvm_fatal_error"); 10225505413ee8e4e2924f52ba81181071f3a492e7d9Daniel Dunbar } else if (II->isStr("llvm_unreachable")) { 10235505413ee8e4e2924f52ba81181071f3a492e7d9Daniel Dunbar llvm_unreachable("#pragma clang __debug llvm_unreachable"); 10245505413ee8e4e2924f52ba81181071f3a492e7d9Daniel Dunbar } else if (II->isStr("overflow_stack")) { 10255505413ee8e4e2924f52ba81181071f3a492e7d9Daniel Dunbar DebugOverflowStack(); 1026ff759a66526d6a985fb4efae69c6b798d83364dfDaniel Dunbar } else if (II->isStr("handle_crash")) { 1027ff759a66526d6a985fb4efae69c6b798d83364dfDaniel Dunbar llvm::CrashRecoveryContext *CRC =llvm::CrashRecoveryContext::GetCurrent(); 1028ff759a66526d6a985fb4efae69c6b798d83364dfDaniel Dunbar if (CRC) 1029ff759a66526d6a985fb4efae69c6b798d83364dfDaniel Dunbar CRC->HandleCrash(); 10305505413ee8e4e2924f52ba81181071f3a492e7d9Daniel Dunbar } else { 10315505413ee8e4e2924f52ba81181071f3a492e7d9Daniel Dunbar PP.Diag(Tok, diag::warn_pragma_debug_unexpected_command) 10325505413ee8e4e2924f52ba81181071f3a492e7d9Daniel Dunbar << II->getName(); 1033abf7b72a965efba4a5ac6f85c66d5aba0c83f972Daniel Dunbar } 1034abf7b72a965efba4a5ac6f85c66d5aba0c83f972Daniel Dunbar } 1035abf7b72a965efba4a5ac6f85c66d5aba0c83f972Daniel Dunbar 10361066c6c7c4d5186c2434b90b054c302564f98080Francois Pichet// Disable MSVC warning about runtime stack overflow. 10371066c6c7c4d5186c2434b90b054c302564f98080Francois Pichet#ifdef _MSC_VER 10381066c6c7c4d5186c2434b90b054c302564f98080Francois Pichet #pragma warning(disable : 4717) 10391066c6c7c4d5186c2434b90b054c302564f98080Francois Pichet#endif 1040abf7b72a965efba4a5ac6f85c66d5aba0c83f972Daniel Dunbar void DebugOverflowStack() { 1041abf7b72a965efba4a5ac6f85c66d5aba0c83f972Daniel Dunbar DebugOverflowStack(); 1042abf7b72a965efba4a5ac6f85c66d5aba0c83f972Daniel Dunbar } 10431066c6c7c4d5186c2434b90b054c302564f98080Francois Pichet#ifdef _MSC_VER 10441066c6c7c4d5186c2434b90b054c302564f98080Francois Pichet #pragma warning(default : 4717) 10451066c6c7c4d5186c2434b90b054c302564f98080Francois Pichet#endif 10461066c6c7c4d5186c2434b90b054c302564f98080Francois Pichet 1047abf7b72a965efba4a5ac6f85c66d5aba0c83f972Daniel Dunbar}; 1048abf7b72a965efba4a5ac6f85c66d5aba0c83f972Daniel Dunbar 1049b6e95b7752893f66e9d067b4287dff2dbf4a5802James Dennett/// PragmaDiagnosticHandler - e.g. '\#pragma GCC diagnostic ignored "-Wformat"' 1050edaf877d2469147ba818187a3e57a00b3f73b123Chris Lattnerstruct PragmaDiagnosticHandler : public PragmaHandler { 1051c09ce1224dedc470fce9747e5936ff83cc6762ebDouglas Gregorprivate: 1052c09ce1224dedc470fce9747e5936ff83cc6762ebDouglas Gregor const char *Namespace; 105304ae2df026b275aae5dddfc0db5ca55ff4e62179Chris Lattnerpublic: 1054c09ce1224dedc470fce9747e5936ff83cc6762ebDouglas Gregor explicit PragmaDiagnosticHandler(const char *NS) : 1055c09ce1224dedc470fce9747e5936ff83cc6762ebDouglas Gregor PragmaHandler("diagnostic"), Namespace(NS) {} 105680c60f72848896f867f6b7e664e7060d9e78f019Douglas Gregor virtual void HandlePragma(Preprocessor &PP, PragmaIntroducerKind Introducer, 105780c60f72848896f867f6b7e664e7060d9e78f019Douglas Gregor Token &DiagToken) { 10580827408865e32789e0ec4b8113a302ccdc531423Argyrios Kyrtzidis SourceLocation DiagLoc = DiagToken.getLocation(); 1059edaf877d2469147ba818187a3e57a00b3f73b123Chris Lattner Token Tok; 1060edaf877d2469147ba818187a3e57a00b3f73b123Chris Lattner PP.LexUnexpandedToken(Tok); 1061edaf877d2469147ba818187a3e57a00b3f73b123Chris Lattner if (Tok.isNot(tok::identifier)) { 10626493a4d7129673f7878da2382dedf4f9abc57e4cDouglas Gregor PP.Diag(Tok, diag::warn_pragma_diagnostic_invalid); 1063edaf877d2469147ba818187a3e57a00b3f73b123Chris Lattner return; 1064edaf877d2469147ba818187a3e57a00b3f73b123Chris Lattner } 1065edaf877d2469147ba818187a3e57a00b3f73b123Chris Lattner IdentifierInfo *II = Tok.getIdentifierInfo(); 1066c09ce1224dedc470fce9747e5936ff83cc6762ebDouglas Gregor PPCallbacks *Callbacks = PP.getPPCallbacks(); 10671eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 1068edaf877d2469147ba818187a3e57a00b3f73b123Chris Lattner diag::Mapping Map; 1069edaf877d2469147ba818187a3e57a00b3f73b123Chris Lattner if (II->isStr("warning")) 1070edaf877d2469147ba818187a3e57a00b3f73b123Chris Lattner Map = diag::MAP_WARNING; 1071edaf877d2469147ba818187a3e57a00b3f73b123Chris Lattner else if (II->isStr("error")) 1072edaf877d2469147ba818187a3e57a00b3f73b123Chris Lattner Map = diag::MAP_ERROR; 1073edaf877d2469147ba818187a3e57a00b3f73b123Chris Lattner else if (II->isStr("ignored")) 1074edaf877d2469147ba818187a3e57a00b3f73b123Chris Lattner Map = diag::MAP_IGNORE; 1075edaf877d2469147ba818187a3e57a00b3f73b123Chris Lattner else if (II->isStr("fatal")) 1076edaf877d2469147ba818187a3e57a00b3f73b123Chris Lattner Map = diag::MAP_FATAL; 10776493a4d7129673f7878da2382dedf4f9abc57e4cDouglas Gregor else if (II->isStr("pop")) { 10780827408865e32789e0ec4b8113a302ccdc531423Argyrios Kyrtzidis if (!PP.getDiagnostics().popMappings(DiagLoc)) 10796493a4d7129673f7878da2382dedf4f9abc57e4cDouglas Gregor PP.Diag(Tok, diag::warn_pragma_diagnostic_cannot_pop); 1080c09ce1224dedc470fce9747e5936ff83cc6762ebDouglas Gregor else if (Callbacks) 1081c09ce1224dedc470fce9747e5936ff83cc6762ebDouglas Gregor Callbacks->PragmaDiagnosticPop(DiagLoc, Namespace); 10826493a4d7129673f7878da2382dedf4f9abc57e4cDouglas Gregor return; 10836493a4d7129673f7878da2382dedf4f9abc57e4cDouglas Gregor } else if (II->isStr("push")) { 10840827408865e32789e0ec4b8113a302ccdc531423Argyrios Kyrtzidis PP.getDiagnostics().pushMappings(DiagLoc); 1085c09ce1224dedc470fce9747e5936ff83cc6762ebDouglas Gregor if (Callbacks) 1086c09ce1224dedc470fce9747e5936ff83cc6762ebDouglas Gregor Callbacks->PragmaDiagnosticPush(DiagLoc, Namespace); 108704ae2df026b275aae5dddfc0db5ca55ff4e62179Chris Lattner return; 108804ae2df026b275aae5dddfc0db5ca55ff4e62179Chris Lattner } else { 10896493a4d7129673f7878da2382dedf4f9abc57e4cDouglas Gregor PP.Diag(Tok, diag::warn_pragma_diagnostic_invalid); 1090edaf877d2469147ba818187a3e57a00b3f73b123Chris Lattner return; 1091edaf877d2469147ba818187a3e57a00b3f73b123Chris Lattner } 10921eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 1093edaf877d2469147ba818187a3e57a00b3f73b123Chris Lattner PP.LexUnexpandedToken(Tok); 1094edaf877d2469147ba818187a3e57a00b3f73b123Chris Lattner 1095edaf877d2469147ba818187a3e57a00b3f73b123Chris Lattner // We need at least one string. 1096edaf877d2469147ba818187a3e57a00b3f73b123Chris Lattner if (Tok.isNot(tok::string_literal)) { 1097edaf877d2469147ba818187a3e57a00b3f73b123Chris Lattner PP.Diag(Tok.getLocation(), diag::warn_pragma_diagnostic_invalid_token); 1098edaf877d2469147ba818187a3e57a00b3f73b123Chris Lattner return; 1099edaf877d2469147ba818187a3e57a00b3f73b123Chris Lattner } 11001eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 1101edaf877d2469147ba818187a3e57a00b3f73b123Chris Lattner // String concatenation allows multiple strings, which can even come from 1102edaf877d2469147ba818187a3e57a00b3f73b123Chris Lattner // macro expansion. 1103edaf877d2469147ba818187a3e57a00b3f73b123Chris Lattner // "foo " "bar" "Baz" 11045f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner SmallVector<Token, 4> StrToks; 1105edaf877d2469147ba818187a3e57a00b3f73b123Chris Lattner while (Tok.is(tok::string_literal)) { 1106edaf877d2469147ba818187a3e57a00b3f73b123Chris Lattner StrToks.push_back(Tok); 1107edaf877d2469147ba818187a3e57a00b3f73b123Chris Lattner PP.LexUnexpandedToken(Tok); 1108edaf877d2469147ba818187a3e57a00b3f73b123Chris Lattner } 11091eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 111084021556baceb76eedf7d44be8ba71d9b8cfaccePeter Collingbourne if (Tok.isNot(tok::eod)) { 1111edaf877d2469147ba818187a3e57a00b3f73b123Chris Lattner PP.Diag(Tok.getLocation(), diag::warn_pragma_diagnostic_invalid_token); 1112edaf877d2469147ba818187a3e57a00b3f73b123Chris Lattner return; 1113edaf877d2469147ba818187a3e57a00b3f73b123Chris Lattner } 11141eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 1115edaf877d2469147ba818187a3e57a00b3f73b123Chris Lattner // Concatenate and parse the strings. 1116edaf877d2469147ba818187a3e57a00b3f73b123Chris Lattner StringLiteralParser Literal(&StrToks[0], StrToks.size(), PP); 11175cee1195584fa8672253139c86e922daeda69b9eDouglas Gregor assert(Literal.isAscii() && "Didn't allow wide strings in"); 1118edaf877d2469147ba818187a3e57a00b3f73b123Chris Lattner if (Literal.hadError) 1119edaf877d2469147ba818187a3e57a00b3f73b123Chris Lattner return; 1120edaf877d2469147ba818187a3e57a00b3f73b123Chris Lattner if (Literal.Pascal) { 11216493a4d7129673f7878da2382dedf4f9abc57e4cDouglas Gregor PP.Diag(Tok, diag::warn_pragma_diagnostic_invalid); 1122edaf877d2469147ba818187a3e57a00b3f73b123Chris Lattner return; 1123edaf877d2469147ba818187a3e57a00b3f73b123Chris Lattner } 112404ae2df026b275aae5dddfc0db5ca55ff4e62179Chris Lattner 11255f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner StringRef WarningName(Literal.GetString()); 1126edaf877d2469147ba818187a3e57a00b3f73b123Chris Lattner 1127edaf877d2469147ba818187a3e57a00b3f73b123Chris Lattner if (WarningName.size() < 3 || WarningName[0] != '-' || 1128edaf877d2469147ba818187a3e57a00b3f73b123Chris Lattner WarningName[1] != 'W') { 1129edaf877d2469147ba818187a3e57a00b3f73b123Chris Lattner PP.Diag(StrToks[0].getLocation(), 1130edaf877d2469147ba818187a3e57a00b3f73b123Chris Lattner diag::warn_pragma_diagnostic_invalid_option); 1131edaf877d2469147ba818187a3e57a00b3f73b123Chris Lattner return; 1132edaf877d2469147ba818187a3e57a00b3f73b123Chris Lattner } 11331eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 1134477aab6782795e7472055a54108d2df270ce1a89Argyrios Kyrtzidis if (PP.getDiagnostics().setDiagnosticGroupMapping(WarningName.substr(2), 11350827408865e32789e0ec4b8113a302ccdc531423Argyrios Kyrtzidis Map, DiagLoc)) 1136edaf877d2469147ba818187a3e57a00b3f73b123Chris Lattner PP.Diag(StrToks[0].getLocation(), 1137edaf877d2469147ba818187a3e57a00b3f73b123Chris Lattner diag::warn_pragma_diagnostic_unknown_warning) << WarningName; 1138c09ce1224dedc470fce9747e5936ff83cc6762ebDouglas Gregor else if (Callbacks) 1139c09ce1224dedc470fce9747e5936ff83cc6762ebDouglas Gregor Callbacks->PragmaDiagnostic(DiagLoc, Namespace, Map, WarningName); 1140edaf877d2469147ba818187a3e57a00b3f73b123Chris Lattner } 1141edaf877d2469147ba818187a3e57a00b3f73b123Chris Lattner}; 11421eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 1143b6e95b7752893f66e9d067b4287dff2dbf4a5802James Dennett/// PragmaCommentHandler - "\#pragma comment ...". 1144636c5ef6572e899d36cec1b0023fb28ba65189e1Chris Lattnerstruct PragmaCommentHandler : public PragmaHandler { 11459b36c3f0de0105e903130bbda3c4aea7d792c0afArgyrios Kyrtzidis PragmaCommentHandler() : PragmaHandler("comment") {} 114680c60f72848896f867f6b7e664e7060d9e78f019Douglas Gregor virtual void HandlePragma(Preprocessor &PP, PragmaIntroducerKind Introducer, 114780c60f72848896f867f6b7e664e7060d9e78f019Douglas Gregor Token &CommentTok) { 1148636c5ef6572e899d36cec1b0023fb28ba65189e1Chris Lattner PP.HandlePragmaComment(CommentTok); 1149636c5ef6572e899d36cec1b0023fb28ba65189e1Chris Lattner } 1150636c5ef6572e899d36cec1b0023fb28ba65189e1Chris Lattner}; 11511eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 1152b6e95b7752893f66e9d067b4287dff2dbf4a5802James Dennett/// PragmaIncludeAliasHandler - "\#pragma include_alias("...")". 11534c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballmanstruct PragmaIncludeAliasHandler : public PragmaHandler { 11544c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman PragmaIncludeAliasHandler() : PragmaHandler("include_alias") {} 11554c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman virtual void HandlePragma(Preprocessor &PP, PragmaIntroducerKind Introducer, 11564c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman Token &IncludeAliasTok) { 11574c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman PP.HandlePragmaIncludeAlias(IncludeAliasTok); 11584c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman } 11594c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman}; 11604c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman 1161b6e95b7752893f66e9d067b4287dff2dbf4a5802James Dennett/// PragmaMessageHandler - "\#pragma message("...")". 1162abfe094ce71c42656dcb84a3bdc3e79cb3c16fc3Chris Lattnerstruct PragmaMessageHandler : public PragmaHandler { 11639b36c3f0de0105e903130bbda3c4aea7d792c0afArgyrios Kyrtzidis PragmaMessageHandler() : PragmaHandler("message") {} 116480c60f72848896f867f6b7e664e7060d9e78f019Douglas Gregor virtual void HandlePragma(Preprocessor &PP, PragmaIntroducerKind Introducer, 116580c60f72848896f867f6b7e664e7060d9e78f019Douglas Gregor Token &CommentTok) { 1166abfe094ce71c42656dcb84a3bdc3e79cb3c16fc3Chris Lattner PP.HandlePragmaMessage(CommentTok); 1167abfe094ce71c42656dcb84a3bdc3e79cb3c16fc3Chris Lattner } 1168abfe094ce71c42656dcb84a3bdc3e79cb3c16fc3Chris Lattner}; 1169abfe094ce71c42656dcb84a3bdc3e79cb3c16fc3Chris Lattner 1170b6e95b7752893f66e9d067b4287dff2dbf4a5802James Dennett/// PragmaPushMacroHandler - "\#pragma push_macro" saves the value of the 1171f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner/// macro on the top of the stack. 1172f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattnerstruct PragmaPushMacroHandler : public PragmaHandler { 1173f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner PragmaPushMacroHandler() : PragmaHandler("push_macro") {} 117480c60f72848896f867f6b7e664e7060d9e78f019Douglas Gregor virtual void HandlePragma(Preprocessor &PP, PragmaIntroducerKind Introducer, 117580c60f72848896f867f6b7e664e7060d9e78f019Douglas Gregor Token &PushMacroTok) { 1176f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner PP.HandlePragmaPushMacro(PushMacroTok); 1177f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner } 1178f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner}; 1179f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner 1180f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner 1181b6e95b7752893f66e9d067b4287dff2dbf4a5802James Dennett/// PragmaPopMacroHandler - "\#pragma pop_macro" sets the value of the 1182f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner/// macro to the value on the top of the stack. 1183f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattnerstruct PragmaPopMacroHandler : public PragmaHandler { 1184f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner PragmaPopMacroHandler() : PragmaHandler("pop_macro") {} 118580c60f72848896f867f6b7e664e7060d9e78f019Douglas Gregor virtual void HandlePragma(Preprocessor &PP, PragmaIntroducerKind Introducer, 118680c60f72848896f867f6b7e664e7060d9e78f019Douglas Gregor Token &PopMacroTok) { 1187f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner PP.HandlePragmaPopMacro(PopMacroTok); 1188f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner } 1189f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner}; 1190f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner 1191062f23246510393c19b537b68ec88b6a08ee8996Chris Lattner// Pragma STDC implementations. 11926c5cf4a2e234923ab66127de0874a71cb6bfdd83Chris Lattner 1193b6e95b7752893f66e9d067b4287dff2dbf4a5802James Dennett/// PragmaSTDC_FENV_ACCESSHandler - "\#pragma STDC FENV_ACCESS ...". 1194062f23246510393c19b537b68ec88b6a08ee8996Chris Lattnerstruct PragmaSTDC_FENV_ACCESSHandler : public PragmaHandler { 11959b36c3f0de0105e903130bbda3c4aea7d792c0afArgyrios Kyrtzidis PragmaSTDC_FENV_ACCESSHandler() : PragmaHandler("FENV_ACCESS") {} 119680c60f72848896f867f6b7e664e7060d9e78f019Douglas Gregor virtual void HandlePragma(Preprocessor &PP, PragmaIntroducerKind Introducer, 119780c60f72848896f867f6b7e664e7060d9e78f019Douglas Gregor Token &Tok) { 11989d3f5f7550a2fab4178ed01425758c349b73a609Peter Collingbourne tok::OnOffSwitch OOS; 11999d3f5f7550a2fab4178ed01425758c349b73a609Peter Collingbourne if (PP.LexOnOffSwitch(OOS)) 12009d3f5f7550a2fab4178ed01425758c349b73a609Peter Collingbourne return; 12019d3f5f7550a2fab4178ed01425758c349b73a609Peter Collingbourne if (OOS == tok::OOS_ON) 12024d8aac3778b40c161bed9964125948ee01c08821Chris Lattner PP.Diag(Tok, diag::warn_stdc_fenv_access_not_supported); 1203062f23246510393c19b537b68ec88b6a08ee8996Chris Lattner } 1204062f23246510393c19b537b68ec88b6a08ee8996Chris Lattner}; 12051eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 1206b6e95b7752893f66e9d067b4287dff2dbf4a5802James Dennett/// PragmaSTDC_CX_LIMITED_RANGEHandler - "\#pragma STDC CX_LIMITED_RANGE ...". 1207062f23246510393c19b537b68ec88b6a08ee8996Chris Lattnerstruct PragmaSTDC_CX_LIMITED_RANGEHandler : public PragmaHandler { 12089b36c3f0de0105e903130bbda3c4aea7d792c0afArgyrios Kyrtzidis PragmaSTDC_CX_LIMITED_RANGEHandler() 12099b36c3f0de0105e903130bbda3c4aea7d792c0afArgyrios Kyrtzidis : PragmaHandler("CX_LIMITED_RANGE") {} 121080c60f72848896f867f6b7e664e7060d9e78f019Douglas Gregor virtual void HandlePragma(Preprocessor &PP, PragmaIntroducerKind Introducer, 121180c60f72848896f867f6b7e664e7060d9e78f019Douglas Gregor Token &Tok) { 12129d3f5f7550a2fab4178ed01425758c349b73a609Peter Collingbourne tok::OnOffSwitch OOS; 12139d3f5f7550a2fab4178ed01425758c349b73a609Peter Collingbourne PP.LexOnOffSwitch(OOS); 1214062f23246510393c19b537b68ec88b6a08ee8996Chris Lattner } 1215062f23246510393c19b537b68ec88b6a08ee8996Chris Lattner}; 12161eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 1217b6e95b7752893f66e9d067b4287dff2dbf4a5802James Dennett/// PragmaSTDC_UnknownHandler - "\#pragma STDC ...". 1218062f23246510393c19b537b68ec88b6a08ee8996Chris Lattnerstruct PragmaSTDC_UnknownHandler : public PragmaHandler { 12199b36c3f0de0105e903130bbda3c4aea7d792c0afArgyrios Kyrtzidis PragmaSTDC_UnknownHandler() {} 122080c60f72848896f867f6b7e664e7060d9e78f019Douglas Gregor virtual void HandlePragma(Preprocessor &PP, PragmaIntroducerKind Introducer, 122180c60f72848896f867f6b7e664e7060d9e78f019Douglas Gregor Token &UnknownTok) { 12226c5cf4a2e234923ab66127de0874a71cb6bfdd83Chris Lattner // C99 6.10.6p2, unknown forms are not allowed. 1223f545be5552b6fd40a4c766fbf82dab0ab5305790Chris Lattner PP.Diag(UnknownTok, diag::ext_stdc_pragma_ignored); 1224062f23246510393c19b537b68ec88b6a08ee8996Chris Lattner } 1225062f23246510393c19b537b68ec88b6a08ee8996Chris Lattner}; 12261eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 12278dfac0baaf0f81d3945bcb306480e358ba8d1f08John McCall/// PragmaARCCFCodeAuditedHandler - 1228b6e95b7752893f66e9d067b4287dff2dbf4a5802James Dennett/// \#pragma clang arc_cf_code_audited begin/end 12298dfac0baaf0f81d3945bcb306480e358ba8d1f08John McCallstruct PragmaARCCFCodeAuditedHandler : public PragmaHandler { 12308dfac0baaf0f81d3945bcb306480e358ba8d1f08John McCall PragmaARCCFCodeAuditedHandler() : PragmaHandler("arc_cf_code_audited") {} 12318dfac0baaf0f81d3945bcb306480e358ba8d1f08John McCall virtual void HandlePragma(Preprocessor &PP, PragmaIntroducerKind Introducer, 12328dfac0baaf0f81d3945bcb306480e358ba8d1f08John McCall Token &NameTok) { 12338dfac0baaf0f81d3945bcb306480e358ba8d1f08John McCall SourceLocation Loc = NameTok.getLocation(); 12348dfac0baaf0f81d3945bcb306480e358ba8d1f08John McCall bool IsBegin; 12358dfac0baaf0f81d3945bcb306480e358ba8d1f08John McCall 12368dfac0baaf0f81d3945bcb306480e358ba8d1f08John McCall Token Tok; 12378dfac0baaf0f81d3945bcb306480e358ba8d1f08John McCall 12388dfac0baaf0f81d3945bcb306480e358ba8d1f08John McCall // Lex the 'begin' or 'end'. 12398dfac0baaf0f81d3945bcb306480e358ba8d1f08John McCall PP.LexUnexpandedToken(Tok); 12408dfac0baaf0f81d3945bcb306480e358ba8d1f08John McCall const IdentifierInfo *BeginEnd = Tok.getIdentifierInfo(); 12418dfac0baaf0f81d3945bcb306480e358ba8d1f08John McCall if (BeginEnd && BeginEnd->isStr("begin")) { 12428dfac0baaf0f81d3945bcb306480e358ba8d1f08John McCall IsBegin = true; 12438dfac0baaf0f81d3945bcb306480e358ba8d1f08John McCall } else if (BeginEnd && BeginEnd->isStr("end")) { 12448dfac0baaf0f81d3945bcb306480e358ba8d1f08John McCall IsBegin = false; 12458dfac0baaf0f81d3945bcb306480e358ba8d1f08John McCall } else { 12468dfac0baaf0f81d3945bcb306480e358ba8d1f08John McCall PP.Diag(Tok.getLocation(), diag::err_pp_arc_cf_code_audited_syntax); 12478dfac0baaf0f81d3945bcb306480e358ba8d1f08John McCall return; 12488dfac0baaf0f81d3945bcb306480e358ba8d1f08John McCall } 12498dfac0baaf0f81d3945bcb306480e358ba8d1f08John McCall 12508dfac0baaf0f81d3945bcb306480e358ba8d1f08John McCall // Verify that this is followed by EOD. 12518dfac0baaf0f81d3945bcb306480e358ba8d1f08John McCall PP.LexUnexpandedToken(Tok); 12528dfac0baaf0f81d3945bcb306480e358ba8d1f08John McCall if (Tok.isNot(tok::eod)) 12538dfac0baaf0f81d3945bcb306480e358ba8d1f08John McCall PP.Diag(Tok, diag::ext_pp_extra_tokens_at_eol) << "pragma"; 12548dfac0baaf0f81d3945bcb306480e358ba8d1f08John McCall 12558dfac0baaf0f81d3945bcb306480e358ba8d1f08John McCall // The start location of the active audit. 12568dfac0baaf0f81d3945bcb306480e358ba8d1f08John McCall SourceLocation BeginLoc = PP.getPragmaARCCFCodeAuditedLoc(); 12578dfac0baaf0f81d3945bcb306480e358ba8d1f08John McCall 12588dfac0baaf0f81d3945bcb306480e358ba8d1f08John McCall // The start location we want after processing this. 12598dfac0baaf0f81d3945bcb306480e358ba8d1f08John McCall SourceLocation NewLoc; 12608dfac0baaf0f81d3945bcb306480e358ba8d1f08John McCall 12618dfac0baaf0f81d3945bcb306480e358ba8d1f08John McCall if (IsBegin) { 12628dfac0baaf0f81d3945bcb306480e358ba8d1f08John McCall // Complain about attempts to re-enter an audit. 12638dfac0baaf0f81d3945bcb306480e358ba8d1f08John McCall if (BeginLoc.isValid()) { 12648dfac0baaf0f81d3945bcb306480e358ba8d1f08John McCall PP.Diag(Loc, diag::err_pp_double_begin_of_arc_cf_code_audited); 12658dfac0baaf0f81d3945bcb306480e358ba8d1f08John McCall PP.Diag(BeginLoc, diag::note_pragma_entered_here); 12668dfac0baaf0f81d3945bcb306480e358ba8d1f08John McCall } 12678dfac0baaf0f81d3945bcb306480e358ba8d1f08John McCall NewLoc = Loc; 12688dfac0baaf0f81d3945bcb306480e358ba8d1f08John McCall } else { 12698dfac0baaf0f81d3945bcb306480e358ba8d1f08John McCall // Complain about attempts to leave an audit that doesn't exist. 12708dfac0baaf0f81d3945bcb306480e358ba8d1f08John McCall if (!BeginLoc.isValid()) { 12718dfac0baaf0f81d3945bcb306480e358ba8d1f08John McCall PP.Diag(Loc, diag::err_pp_unmatched_end_of_arc_cf_code_audited); 12728dfac0baaf0f81d3945bcb306480e358ba8d1f08John McCall return; 12738dfac0baaf0f81d3945bcb306480e358ba8d1f08John McCall } 12748dfac0baaf0f81d3945bcb306480e358ba8d1f08John McCall NewLoc = SourceLocation(); 12758dfac0baaf0f81d3945bcb306480e358ba8d1f08John McCall } 12768dfac0baaf0f81d3945bcb306480e358ba8d1f08John McCall 12778dfac0baaf0f81d3945bcb306480e358ba8d1f08John McCall PP.setPragmaARCCFCodeAuditedLoc(NewLoc); 12788dfac0baaf0f81d3945bcb306480e358ba8d1f08John McCall } 12798dfac0baaf0f81d3945bcb306480e358ba8d1f08John McCall}; 12808dfac0baaf0f81d3945bcb306480e358ba8d1f08John McCall 12815f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer} // end anonymous namespace 12825f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 12835f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 12845f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// RegisterBuiltinPragmas - Install the standard preprocessor pragmas: 1285b6e95b7752893f66e9d067b4287dff2dbf4a5802James Dennett/// \#pragma GCC poison/system_header/dependency and \#pragma once. 12865f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencervoid Preprocessor::RegisterBuiltinPragmas() { 12879b36c3f0de0105e903130bbda3c4aea7d792c0afArgyrios Kyrtzidis AddPragmaHandler(new PragmaOnceHandler()); 12889b36c3f0de0105e903130bbda3c4aea7d792c0afArgyrios Kyrtzidis AddPragmaHandler(new PragmaMarkHandler()); 1289f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner AddPragmaHandler(new PragmaPushMacroHandler()); 1290f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner AddPragmaHandler(new PragmaPopMacroHandler()); 1291301669b381dfead44bda5ea3211d69ffec5a530bMichael J. Spencer AddPragmaHandler(new PragmaMessageHandler()); 12921eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 1293e8fa06e07363b6d5e6c371bbd454d51bab78d01dChris Lattner // #pragma GCC ... 12949b36c3f0de0105e903130bbda3c4aea7d792c0afArgyrios Kyrtzidis AddPragmaHandler("GCC", new PragmaPoisonHandler()); 12959b36c3f0de0105e903130bbda3c4aea7d792c0afArgyrios Kyrtzidis AddPragmaHandler("GCC", new PragmaSystemHeaderHandler()); 12969b36c3f0de0105e903130bbda3c4aea7d792c0afArgyrios Kyrtzidis AddPragmaHandler("GCC", new PragmaDependencyHandler()); 1297c09ce1224dedc470fce9747e5936ff83cc6762ebDouglas Gregor AddPragmaHandler("GCC", new PragmaDiagnosticHandler("GCC")); 1298e8fa06e07363b6d5e6c371bbd454d51bab78d01dChris Lattner // #pragma clang ... 12999b36c3f0de0105e903130bbda3c4aea7d792c0afArgyrios Kyrtzidis AddPragmaHandler("clang", new PragmaPoisonHandler()); 13009b36c3f0de0105e903130bbda3c4aea7d792c0afArgyrios Kyrtzidis AddPragmaHandler("clang", new PragmaSystemHeaderHandler()); 1301abf7b72a965efba4a5ac6f85c66d5aba0c83f972Daniel Dunbar AddPragmaHandler("clang", new PragmaDebugHandler()); 13029b36c3f0de0105e903130bbda3c4aea7d792c0afArgyrios Kyrtzidis AddPragmaHandler("clang", new PragmaDependencyHandler()); 1303c09ce1224dedc470fce9747e5936ff83cc6762ebDouglas Gregor AddPragmaHandler("clang", new PragmaDiagnosticHandler("clang")); 13048dfac0baaf0f81d3945bcb306480e358ba8d1f08John McCall AddPragmaHandler("clang", new PragmaARCCFCodeAuditedHandler()); 13059b36c3f0de0105e903130bbda3c4aea7d792c0afArgyrios Kyrtzidis 13069b36c3f0de0105e903130bbda3c4aea7d792c0afArgyrios Kyrtzidis AddPragmaHandler("STDC", new PragmaSTDC_FENV_ACCESSHandler()); 13079b36c3f0de0105e903130bbda3c4aea7d792c0afArgyrios Kyrtzidis AddPragmaHandler("STDC", new PragmaSTDC_CX_LIMITED_RANGEHandler()); 1308062f23246510393c19b537b68ec88b6a08ee8996Chris Lattner AddPragmaHandler("STDC", new PragmaSTDC_UnknownHandler()); 13091eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 1310636c5ef6572e899d36cec1b0023fb28ba65189e1Chris Lattner // MS extensions. 13114e4d08403ca5cfd4d558fa2936215d3a4e5a528dDavid Blaikie if (LangOpts.MicrosoftExt) { 13129b36c3f0de0105e903130bbda3c4aea7d792c0afArgyrios Kyrtzidis AddPragmaHandler(new PragmaCommentHandler()); 13134c55c54db8e676aa3e042188773d9d82d59fff91Aaron Ballman AddPragmaHandler(new PragmaIncludeAliasHandler()); 1314abfe094ce71c42656dcb84a3bdc3e79cb3c16fc3Chris Lattner } 13155f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer} 1316