1718292f2SDouglas Gregor //===--- ModuleMap.cpp - Describe the layout of modules ---------*- C++ -*-===// 2718292f2SDouglas Gregor // 3718292f2SDouglas Gregor // The LLVM Compiler Infrastructure 4718292f2SDouglas Gregor // 5718292f2SDouglas Gregor // This file is distributed under the University of Illinois Open Source 6718292f2SDouglas Gregor // License. See LICENSE.TXT for details. 7718292f2SDouglas Gregor // 8718292f2SDouglas Gregor //===----------------------------------------------------------------------===// 9718292f2SDouglas Gregor // 10718292f2SDouglas Gregor // This file defines the ModuleMap implementation, which describes the layout 11718292f2SDouglas Gregor // of a module as it relates to headers. 12718292f2SDouglas Gregor // 13718292f2SDouglas Gregor //===----------------------------------------------------------------------===// 14718292f2SDouglas Gregor #include "clang/Lex/ModuleMap.h" 15a7d03840SJordan Rose #include "clang/Basic/CharInfo.h" 16718292f2SDouglas Gregor #include "clang/Basic/Diagnostic.h" 17811db4eaSDouglas Gregor #include "clang/Basic/DiagnosticOptions.h" 18718292f2SDouglas Gregor #include "clang/Basic/FileManager.h" 19718292f2SDouglas Gregor #include "clang/Basic/TargetInfo.h" 20718292f2SDouglas Gregor #include "clang/Basic/TargetOptions.h" 21b146baabSArgyrios Kyrtzidis #include "clang/Lex/HeaderSearch.h" 223a02247dSChandler Carruth #include "clang/Lex/LexDiagnostic.h" 233a02247dSChandler Carruth #include "clang/Lex/Lexer.h" 243a02247dSChandler Carruth #include "clang/Lex/LiteralSupport.h" 253a02247dSChandler Carruth #include "llvm/ADT/StringRef.h" 263a02247dSChandler Carruth #include "llvm/ADT/StringSwitch.h" 27718292f2SDouglas Gregor #include "llvm/Support/Allocator.h" 28e89dbc1dSDouglas Gregor #include "llvm/Support/FileSystem.h" 29718292f2SDouglas Gregor #include "llvm/Support/Host.h" 30552c169eSRafael Espindola #include "llvm/Support/Path.h" 31718292f2SDouglas Gregor #include "llvm/Support/raw_ostream.h" 3207c22b78SDouglas Gregor #include <stdlib.h> 3301c7cfa2SDouglas Gregor #if defined(LLVM_ON_UNIX) 34eadae014SDmitri Gribenko #include <limits.h> 3501c7cfa2SDouglas Gregor #endif 36718292f2SDouglas Gregor using namespace clang; 37718292f2SDouglas Gregor 382b82c2a5SDouglas Gregor Module::ExportDecl 392b82c2a5SDouglas Gregor ModuleMap::resolveExport(Module *Mod, 402b82c2a5SDouglas Gregor const Module::UnresolvedExportDecl &Unresolved, 41e4412640SArgyrios Kyrtzidis bool Complain) const { 42f5eedd05SDouglas Gregor // We may have just a wildcard. 43f5eedd05SDouglas Gregor if (Unresolved.Id.empty()) { 44f5eedd05SDouglas Gregor assert(Unresolved.Wildcard && "Invalid unresolved export"); 45f5eedd05SDouglas Gregor return Module::ExportDecl(0, true); 46f5eedd05SDouglas Gregor } 47f5eedd05SDouglas Gregor 48fb912657SDouglas Gregor // Resolve the module-id. 49fb912657SDouglas Gregor Module *Context = resolveModuleId(Unresolved.Id, Mod, Complain); 50fb912657SDouglas Gregor if (!Context) 51fb912657SDouglas Gregor return Module::ExportDecl(); 52fb912657SDouglas Gregor 53fb912657SDouglas Gregor return Module::ExportDecl(Context, Unresolved.Wildcard); 54fb912657SDouglas Gregor } 55fb912657SDouglas Gregor 56fb912657SDouglas Gregor Module *ModuleMap::resolveModuleId(const ModuleId &Id, Module *Mod, 57fb912657SDouglas Gregor bool Complain) const { 582b82c2a5SDouglas Gregor // Find the starting module. 59fb912657SDouglas Gregor Module *Context = lookupModuleUnqualified(Id[0].first, Mod); 602b82c2a5SDouglas Gregor if (!Context) { 612b82c2a5SDouglas Gregor if (Complain) 620761a8a0SDaniel Jasper Diags.Report(Id[0].second, diag::err_mmap_missing_module_unqualified) 63fb912657SDouglas Gregor << Id[0].first << Mod->getFullModuleName(); 642b82c2a5SDouglas Gregor 65fb912657SDouglas Gregor return 0; 662b82c2a5SDouglas Gregor } 672b82c2a5SDouglas Gregor 682b82c2a5SDouglas Gregor // Dig into the module path. 69fb912657SDouglas Gregor for (unsigned I = 1, N = Id.size(); I != N; ++I) { 70fb912657SDouglas Gregor Module *Sub = lookupModuleQualified(Id[I].first, Context); 712b82c2a5SDouglas Gregor if (!Sub) { 722b82c2a5SDouglas Gregor if (Complain) 730761a8a0SDaniel Jasper Diags.Report(Id[I].second, diag::err_mmap_missing_module_qualified) 74fb912657SDouglas Gregor << Id[I].first << Context->getFullModuleName() 75fb912657SDouglas Gregor << SourceRange(Id[0].second, Id[I-1].second); 762b82c2a5SDouglas Gregor 77fb912657SDouglas Gregor return 0; 782b82c2a5SDouglas Gregor } 792b82c2a5SDouglas Gregor 802b82c2a5SDouglas Gregor Context = Sub; 812b82c2a5SDouglas Gregor } 822b82c2a5SDouglas Gregor 83fb912657SDouglas Gregor return Context; 842b82c2a5SDouglas Gregor } 852b82c2a5SDouglas Gregor 860761a8a0SDaniel Jasper ModuleMap::ModuleMap(SourceManager &SourceMgr, DiagnosticsEngine &Diags, 87b146baabSArgyrios Kyrtzidis const LangOptions &LangOpts, const TargetInfo *Target, 88b146baabSArgyrios Kyrtzidis HeaderSearch &HeaderInfo) 890761a8a0SDaniel Jasper : SourceMgr(SourceMgr), Diags(Diags), LangOpts(LangOpts), Target(Target), 901f76c4e8SManuel Klimek HeaderInfo(HeaderInfo), BuiltinIncludeDir(0), CompilingModule(0), 910761a8a0SDaniel Jasper SourceModule(0) {} 92718292f2SDouglas Gregor 93718292f2SDouglas Gregor ModuleMap::~ModuleMap() { 945acdf59eSDouglas Gregor for (llvm::StringMap<Module *>::iterator I = Modules.begin(), 955acdf59eSDouglas Gregor IEnd = Modules.end(); 965acdf59eSDouglas Gregor I != IEnd; ++I) { 975acdf59eSDouglas Gregor delete I->getValue(); 985acdf59eSDouglas Gregor } 99718292f2SDouglas Gregor } 100718292f2SDouglas Gregor 10189929282SDouglas Gregor void ModuleMap::setTarget(const TargetInfo &Target) { 10289929282SDouglas Gregor assert((!this->Target || this->Target == &Target) && 10389929282SDouglas Gregor "Improper target override"); 10489929282SDouglas Gregor this->Target = &Target; 10589929282SDouglas Gregor } 10689929282SDouglas Gregor 107056396aeSDouglas Gregor /// \brief "Sanitize" a filename so that it can be used as an identifier. 108056396aeSDouglas Gregor static StringRef sanitizeFilenameAsIdentifier(StringRef Name, 109056396aeSDouglas Gregor SmallVectorImpl<char> &Buffer) { 110056396aeSDouglas Gregor if (Name.empty()) 111056396aeSDouglas Gregor return Name; 112056396aeSDouglas Gregor 113a7d03840SJordan Rose if (!isValidIdentifier(Name)) { 114056396aeSDouglas Gregor // If we don't already have something with the form of an identifier, 115056396aeSDouglas Gregor // create a buffer with the sanitized name. 116056396aeSDouglas Gregor Buffer.clear(); 117a7d03840SJordan Rose if (isDigit(Name[0])) 118056396aeSDouglas Gregor Buffer.push_back('_'); 119056396aeSDouglas Gregor Buffer.reserve(Buffer.size() + Name.size()); 120056396aeSDouglas Gregor for (unsigned I = 0, N = Name.size(); I != N; ++I) { 121a7d03840SJordan Rose if (isIdentifierBody(Name[I])) 122056396aeSDouglas Gregor Buffer.push_back(Name[I]); 123056396aeSDouglas Gregor else 124056396aeSDouglas Gregor Buffer.push_back('_'); 125056396aeSDouglas Gregor } 126056396aeSDouglas Gregor 127056396aeSDouglas Gregor Name = StringRef(Buffer.data(), Buffer.size()); 128056396aeSDouglas Gregor } 129056396aeSDouglas Gregor 130056396aeSDouglas Gregor while (llvm::StringSwitch<bool>(Name) 131056396aeSDouglas Gregor #define KEYWORD(Keyword,Conditions) .Case(#Keyword, true) 132056396aeSDouglas Gregor #define ALIAS(Keyword, AliasOf, Conditions) .Case(Keyword, true) 133056396aeSDouglas Gregor #include "clang/Basic/TokenKinds.def" 134056396aeSDouglas Gregor .Default(false)) { 135056396aeSDouglas Gregor if (Name.data() != Buffer.data()) 136056396aeSDouglas Gregor Buffer.append(Name.begin(), Name.end()); 137056396aeSDouglas Gregor Buffer.push_back('_'); 138056396aeSDouglas Gregor Name = StringRef(Buffer.data(), Buffer.size()); 139056396aeSDouglas Gregor } 140056396aeSDouglas Gregor 141056396aeSDouglas Gregor return Name; 142056396aeSDouglas Gregor } 143056396aeSDouglas Gregor 14434d52749SDouglas Gregor /// \brief Determine whether the given file name is the name of a builtin 14534d52749SDouglas Gregor /// header, supplied by Clang to replace, override, or augment existing system 14634d52749SDouglas Gregor /// headers. 14734d52749SDouglas Gregor static bool isBuiltinHeader(StringRef FileName) { 14834d52749SDouglas Gregor return llvm::StringSwitch<bool>(FileName) 14934d52749SDouglas Gregor .Case("float.h", true) 15034d52749SDouglas Gregor .Case("iso646.h", true) 15134d52749SDouglas Gregor .Case("limits.h", true) 15234d52749SDouglas Gregor .Case("stdalign.h", true) 15334d52749SDouglas Gregor .Case("stdarg.h", true) 15434d52749SDouglas Gregor .Case("stdbool.h", true) 15534d52749SDouglas Gregor .Case("stddef.h", true) 15634d52749SDouglas Gregor .Case("stdint.h", true) 15734d52749SDouglas Gregor .Case("tgmath.h", true) 15834d52749SDouglas Gregor .Case("unwind.h", true) 15934d52749SDouglas Gregor .Default(false); 16034d52749SDouglas Gregor } 16134d52749SDouglas Gregor 16292669ee4SDaniel Jasper ModuleMap::HeadersMap::iterator 16392669ee4SDaniel Jasper ModuleMap::findKnownHeader(const FileEntry *File) { 16459527666SDouglas Gregor HeadersMap::iterator Known = Headers.find(File); 1654eaf0a6cSDaniel Jasper if (Known == Headers.end() && File->getDir() == BuiltinIncludeDir && 1664eaf0a6cSDaniel Jasper isBuiltinHeader(llvm::sys::path::filename(File->getName()))) { 1674eaf0a6cSDaniel Jasper HeaderInfo.loadTopLevelSystemModules(); 16892669ee4SDaniel Jasper return Headers.find(File); 1694eaf0a6cSDaniel Jasper } 17092669ee4SDaniel Jasper return Known; 17192669ee4SDaniel Jasper } 17292669ee4SDaniel Jasper 1734469138eSBen Langmuir ModuleMap::KnownHeader 1744469138eSBen Langmuir ModuleMap::findHeaderInUmbrellaDirs(const FileEntry *File, 1754469138eSBen Langmuir SmallVectorImpl<const DirectoryEntry *> &IntermediateDirs) { 1764469138eSBen Langmuir const DirectoryEntry *Dir = File->getDir(); 1774469138eSBen Langmuir assert(Dir && "file in no directory"); 1784469138eSBen Langmuir 1794469138eSBen Langmuir // Note: as an egregious but useful hack we use the real path here, because 1804469138eSBen Langmuir // frameworks moving from top-level frameworks to embedded frameworks tend 1814469138eSBen Langmuir // to be symlinked from the top-level location to the embedded location, 1824469138eSBen Langmuir // and we need to resolve lookups as if we had found the embedded location. 1834469138eSBen Langmuir StringRef DirName = SourceMgr.getFileManager().getCanonicalName(Dir); 1844469138eSBen Langmuir 1854469138eSBen Langmuir // Keep walking up the directory hierarchy, looking for a directory with 1864469138eSBen Langmuir // an umbrella header. 1874469138eSBen Langmuir do { 1884469138eSBen Langmuir auto KnownDir = UmbrellaDirs.find(Dir); 1894469138eSBen Langmuir if (KnownDir != UmbrellaDirs.end()) 1904469138eSBen Langmuir return KnownHeader(KnownDir->second, NormalHeader); 1914469138eSBen Langmuir 1924469138eSBen Langmuir IntermediateDirs.push_back(Dir); 1934469138eSBen Langmuir 1944469138eSBen Langmuir // Retrieve our parent path. 1954469138eSBen Langmuir DirName = llvm::sys::path::parent_path(DirName); 1964469138eSBen Langmuir if (DirName.empty()) 1974469138eSBen Langmuir break; 1984469138eSBen Langmuir 1994469138eSBen Langmuir // Resolve the parent path to a directory entry. 2004469138eSBen Langmuir Dir = SourceMgr.getFileManager().getDirectory(DirName); 2014469138eSBen Langmuir } while (Dir); 2024469138eSBen Langmuir return KnownHeader(); 2034469138eSBen Langmuir } 2044469138eSBen Langmuir 20592669ee4SDaniel Jasper // Returns 'true' if 'RequestingModule directly uses 'RequestedModule'. 20692669ee4SDaniel Jasper static bool directlyUses(const Module *RequestingModule, 20792669ee4SDaniel Jasper const Module *RequestedModule) { 20892669ee4SDaniel Jasper return std::find(RequestingModule->DirectUses.begin(), 20992669ee4SDaniel Jasper RequestingModule->DirectUses.end(), 21092669ee4SDaniel Jasper RequestedModule) != RequestingModule->DirectUses.end(); 21192669ee4SDaniel Jasper } 21292669ee4SDaniel Jasper 21392669ee4SDaniel Jasper static bool violatesPrivateInclude(Module *RequestingModule, 21492669ee4SDaniel Jasper const FileEntry *IncFileEnt, 21592669ee4SDaniel Jasper ModuleMap::ModuleHeaderRole Role, 21692669ee4SDaniel Jasper Module *RequestedModule) { 21792669ee4SDaniel Jasper #ifndef NDEBUG 21892669ee4SDaniel Jasper // Check for consistency between the module header role 21992669ee4SDaniel Jasper // as obtained from the lookup and as obtained from the module. 22092669ee4SDaniel Jasper // This check is not cheap, so enable it only for debugging. 22192669ee4SDaniel Jasper SmallVectorImpl<const FileEntry *> &PvtHdrs 22292669ee4SDaniel Jasper = RequestedModule->PrivateHeaders; 22392669ee4SDaniel Jasper SmallVectorImpl<const FileEntry *>::iterator Look 22492669ee4SDaniel Jasper = std::find(PvtHdrs.begin(), PvtHdrs.end(), IncFileEnt); 22592669ee4SDaniel Jasper bool IsPrivate = Look != PvtHdrs.end(); 22692669ee4SDaniel Jasper assert((IsPrivate && Role == ModuleMap::PrivateHeader) 22792669ee4SDaniel Jasper || (!IsPrivate && Role != ModuleMap::PrivateHeader)); 22892669ee4SDaniel Jasper #endif 22992669ee4SDaniel Jasper return Role == ModuleMap::PrivateHeader && 23092669ee4SDaniel Jasper RequestedModule->getTopLevelModule() != RequestingModule; 23192669ee4SDaniel Jasper } 23292669ee4SDaniel Jasper 233*71e1a64fSBen Langmuir static Module *getTopLevelOrNull(Module *M) { 234*71e1a64fSBen Langmuir return M ? M->getTopLevelModule() : nullptr; 235*71e1a64fSBen Langmuir } 236*71e1a64fSBen Langmuir 23792669ee4SDaniel Jasper void ModuleMap::diagnoseHeaderInclusion(Module *RequestingModule, 23892669ee4SDaniel Jasper SourceLocation FilenameLoc, 23992669ee4SDaniel Jasper StringRef Filename, 24092669ee4SDaniel Jasper const FileEntry *File) { 24192669ee4SDaniel Jasper // No errors for indirect modules. This may be a bit of a problem for modules 24292669ee4SDaniel Jasper // with no source files. 243*71e1a64fSBen Langmuir if (getTopLevelOrNull(RequestingModule) != getTopLevelOrNull(SourceModule)) 24492669ee4SDaniel Jasper return; 24592669ee4SDaniel Jasper 24692669ee4SDaniel Jasper if (RequestingModule) 24792669ee4SDaniel Jasper resolveUses(RequestingModule, /*Complain=*/false); 24892669ee4SDaniel Jasper 249*71e1a64fSBen Langmuir bool Excluded = false; 25092669ee4SDaniel Jasper Module *Private = NULL; 25192669ee4SDaniel Jasper Module *NotUsed = NULL; 252*71e1a64fSBen Langmuir 253*71e1a64fSBen Langmuir HeadersMap::iterator Known = findKnownHeader(File); 254*71e1a64fSBen Langmuir if (Known != Headers.end()) { 255*71e1a64fSBen Langmuir for (const KnownHeader &Header : Known->second) { 25692669ee4SDaniel Jasper // Excluded headers don't really belong to a module. 257*71e1a64fSBen Langmuir if (Header.getRole() == ModuleMap::ExcludedHeader) { 258*71e1a64fSBen Langmuir Excluded = true; 25992669ee4SDaniel Jasper continue; 260*71e1a64fSBen Langmuir } 26192669ee4SDaniel Jasper 26292669ee4SDaniel Jasper // If 'File' is part of 'RequestingModule' we can definitely include it. 263*71e1a64fSBen Langmuir if (Header.getModule() == RequestingModule) 26492669ee4SDaniel Jasper return; 26592669ee4SDaniel Jasper 26692669ee4SDaniel Jasper // Remember private headers for later printing of a diagnostic. 267*71e1a64fSBen Langmuir if (violatesPrivateInclude(RequestingModule, File, Header.getRole(), 268*71e1a64fSBen Langmuir Header.getModule())) { 269*71e1a64fSBen Langmuir Private = Header.getModule(); 27092669ee4SDaniel Jasper continue; 27192669ee4SDaniel Jasper } 27292669ee4SDaniel Jasper 27392669ee4SDaniel Jasper // If uses need to be specified explicitly, we are only allowed to return 27492669ee4SDaniel Jasper // modules that are explicitly used by the requesting module. 27592669ee4SDaniel Jasper if (RequestingModule && LangOpts.ModulesDeclUse && 276*71e1a64fSBen Langmuir !directlyUses(RequestingModule, Header.getModule())) { 277*71e1a64fSBen Langmuir NotUsed = Header.getModule(); 27892669ee4SDaniel Jasper continue; 27992669ee4SDaniel Jasper } 28092669ee4SDaniel Jasper 28192669ee4SDaniel Jasper // We have found a module that we can happily use. 28292669ee4SDaniel Jasper return; 28392669ee4SDaniel Jasper } 284*71e1a64fSBen Langmuir } 28592669ee4SDaniel Jasper 28692669ee4SDaniel Jasper // We have found a header, but it is private. 28792669ee4SDaniel Jasper if (Private != NULL) { 28892669ee4SDaniel Jasper Diags.Report(FilenameLoc, diag::error_use_of_private_header_outside_module) 28992669ee4SDaniel Jasper << Filename; 29092669ee4SDaniel Jasper return; 29192669ee4SDaniel Jasper } 29292669ee4SDaniel Jasper 29392669ee4SDaniel Jasper // We have found a module, but we don't use it. 29492669ee4SDaniel Jasper if (NotUsed != NULL) { 29592669ee4SDaniel Jasper Diags.Report(FilenameLoc, diag::error_undeclared_use_of_module) 29692669ee4SDaniel Jasper << RequestingModule->getFullModuleName() << Filename; 29792669ee4SDaniel Jasper return; 29892669ee4SDaniel Jasper } 29992669ee4SDaniel Jasper 300*71e1a64fSBen Langmuir if (Excluded || isHeaderInUmbrellaDirs(File)) 301*71e1a64fSBen Langmuir return; 302*71e1a64fSBen Langmuir 303*71e1a64fSBen Langmuir // At this point, only non-modular includes remain. 304*71e1a64fSBen Langmuir 305*71e1a64fSBen Langmuir if (LangOpts.ModulesStrictDeclUse) { 306*71e1a64fSBen Langmuir Diags.Report(FilenameLoc, diag::error_undeclared_use_of_module) 307*71e1a64fSBen Langmuir << RequestingModule->getFullModuleName() << Filename; 308*71e1a64fSBen Langmuir } else if (RequestingModule) { 309*71e1a64fSBen Langmuir diag::kind DiagID = RequestingModule->getTopLevelModule()->IsFramework ? 310*71e1a64fSBen Langmuir diag::warn_non_modular_include_in_framework_module : 311*71e1a64fSBen Langmuir diag::warn_non_modular_include_in_module; 312*71e1a64fSBen Langmuir Diags.Report(FilenameLoc, DiagID) << RequestingModule->getFullModuleName(); 313*71e1a64fSBen Langmuir } else { 314*71e1a64fSBen Langmuir Diags.Report(FilenameLoc, diag::warn_non_modular_include); 315*71e1a64fSBen Langmuir } 31692669ee4SDaniel Jasper } 31792669ee4SDaniel Jasper 31892669ee4SDaniel Jasper ModuleMap::KnownHeader 31992669ee4SDaniel Jasper ModuleMap::findModuleForHeader(const FileEntry *File, 32092669ee4SDaniel Jasper Module *RequestingModule) { 32192669ee4SDaniel Jasper HeadersMap::iterator Known = findKnownHeader(File); 3224eaf0a6cSDaniel Jasper 3231fb5c3a6SDouglas Gregor if (Known != Headers.end()) { 32497da9178SDaniel Jasper ModuleMap::KnownHeader Result = KnownHeader(); 3251fb5c3a6SDouglas Gregor 32697da9178SDaniel Jasper // Iterate over all modules that 'File' is part of to find the best fit. 32797da9178SDaniel Jasper for (SmallVectorImpl<KnownHeader>::iterator I = Known->second.begin(), 32897da9178SDaniel Jasper E = Known->second.end(); 32997da9178SDaniel Jasper I != E; ++I) { 3304eaf0a6cSDaniel Jasper // Cannot use a module if the header is excluded in it. 3314eaf0a6cSDaniel Jasper if (I->getRole() == ModuleMap::ExcludedHeader) 3324eaf0a6cSDaniel Jasper continue; 3334eaf0a6cSDaniel Jasper 3344eaf0a6cSDaniel Jasper // Cannot use a module if it is unavailable. 3354eaf0a6cSDaniel Jasper if (!I->getModule()->isAvailable()) 33697da9178SDaniel Jasper continue; 33797da9178SDaniel Jasper 33897da9178SDaniel Jasper // If 'File' is part of 'RequestingModule', 'RequestingModule' is the 33997da9178SDaniel Jasper // module we are looking for. 34097da9178SDaniel Jasper if (I->getModule() == RequestingModule) 34197da9178SDaniel Jasper return *I; 34297da9178SDaniel Jasper 34397da9178SDaniel Jasper // If uses need to be specified explicitly, we are only allowed to return 34497da9178SDaniel Jasper // modules that are explicitly used by the requesting module. 34597da9178SDaniel Jasper if (RequestingModule && LangOpts.ModulesDeclUse && 34692669ee4SDaniel Jasper !directlyUses(RequestingModule, I->getModule())) 34797da9178SDaniel Jasper continue; 3484eaf0a6cSDaniel Jasper 34997da9178SDaniel Jasper Result = *I; 35097da9178SDaniel Jasper // If 'File' is a public header of this module, this is as good as we 35197da9178SDaniel Jasper // are going to get. 3528c71eba1SRichard Smith // FIXME: If we have a RequestingModule, we should prefer the header from 3538c71eba1SRichard Smith // that module. 35497da9178SDaniel Jasper if (I->getRole() == ModuleMap::NormalHeader) 35597da9178SDaniel Jasper break; 35697da9178SDaniel Jasper } 35797da9178SDaniel Jasper return Result; 3581fb5c3a6SDouglas Gregor } 359ab0c8a84SDouglas Gregor 360f857950dSDmitri Gribenko SmallVector<const DirectoryEntry *, 2> SkippedDirs; 3614469138eSBen Langmuir KnownHeader H = findHeaderInUmbrellaDirs(File, SkippedDirs); 3624469138eSBen Langmuir if (H) { 3634469138eSBen Langmuir Module *Result = H.getModule(); 364930a85ccSDouglas Gregor 365930a85ccSDouglas Gregor // Search up the module stack until we find a module with an umbrella 36673141fa9SDouglas Gregor // directory. 367930a85ccSDouglas Gregor Module *UmbrellaModule = Result; 36873141fa9SDouglas Gregor while (!UmbrellaModule->getUmbrellaDir() && UmbrellaModule->Parent) 369930a85ccSDouglas Gregor UmbrellaModule = UmbrellaModule->Parent; 370930a85ccSDouglas Gregor 371930a85ccSDouglas Gregor if (UmbrellaModule->InferSubmodules) { 372a89c5ac4SDouglas Gregor // Infer submodules for each of the directories we found between 373a89c5ac4SDouglas Gregor // the directory of the umbrella header and the directory where 374a89c5ac4SDouglas Gregor // the actual header is located. 3759458f82dSDouglas Gregor bool Explicit = UmbrellaModule->InferExplicitSubmodules; 3769458f82dSDouglas Gregor 3777033127bSDouglas Gregor for (unsigned I = SkippedDirs.size(); I != 0; --I) { 378a89c5ac4SDouglas Gregor // Find or create the module that corresponds to this directory name. 379056396aeSDouglas Gregor SmallString<32> NameBuf; 380056396aeSDouglas Gregor StringRef Name = sanitizeFilenameAsIdentifier( 3814469138eSBen Langmuir llvm::sys::path::stem(SkippedDirs[I-1]->getName()), NameBuf); 382beee15e7SBen Langmuir Result = findOrCreateModule(Name, Result, UmbrellaModule->ModuleMap, 383beee15e7SBen Langmuir /*IsFramework=*/false, Explicit).first; 384ffbafa2aSBen Langmuir Result->IsInferred = true; 385a89c5ac4SDouglas Gregor 386a89c5ac4SDouglas Gregor // Associate the module and the directory. 387a89c5ac4SDouglas Gregor UmbrellaDirs[SkippedDirs[I-1]] = Result; 388a89c5ac4SDouglas Gregor 389a89c5ac4SDouglas Gregor // If inferred submodules export everything they import, add a 390a89c5ac4SDouglas Gregor // wildcard to the set of exports. 391930a85ccSDouglas Gregor if (UmbrellaModule->InferExportWildcard && Result->Exports.empty()) 392a89c5ac4SDouglas Gregor Result->Exports.push_back(Module::ExportDecl(0, true)); 393a89c5ac4SDouglas Gregor } 394a89c5ac4SDouglas Gregor 395a89c5ac4SDouglas Gregor // Infer a submodule with the same name as this header file. 396056396aeSDouglas Gregor SmallString<32> NameBuf; 397056396aeSDouglas Gregor StringRef Name = sanitizeFilenameAsIdentifier( 398056396aeSDouglas Gregor llvm::sys::path::stem(File->getName()), NameBuf); 399beee15e7SBen Langmuir Result = findOrCreateModule(Name, Result, UmbrellaModule->ModuleMap, 400beee15e7SBen Langmuir /*IsFramework=*/false, Explicit).first; 401ffbafa2aSBen Langmuir Result->IsInferred = true; 4023c5305c1SArgyrios Kyrtzidis Result->addTopHeader(File); 403a89c5ac4SDouglas Gregor 404a89c5ac4SDouglas Gregor // If inferred submodules export everything they import, add a 405a89c5ac4SDouglas Gregor // wildcard to the set of exports. 406930a85ccSDouglas Gregor if (UmbrellaModule->InferExportWildcard && Result->Exports.empty()) 407a89c5ac4SDouglas Gregor Result->Exports.push_back(Module::ExportDecl(0, true)); 408a89c5ac4SDouglas Gregor } else { 409a89c5ac4SDouglas Gregor // Record each of the directories we stepped through as being part of 410a89c5ac4SDouglas Gregor // the module we found, since the umbrella header covers them all. 411a89c5ac4SDouglas Gregor for (unsigned I = 0, N = SkippedDirs.size(); I != N; ++I) 412a89c5ac4SDouglas Gregor UmbrellaDirs[SkippedDirs[I]] = Result; 413a89c5ac4SDouglas Gregor } 414a89c5ac4SDouglas Gregor 41597da9178SDaniel Jasper Headers[File].push_back(KnownHeader(Result, NormalHeader)); 4161fb5c3a6SDouglas Gregor 4171fb5c3a6SDouglas Gregor // If a header corresponds to an unavailable module, don't report 4181fb5c3a6SDouglas Gregor // that it maps to anything. 4191fb5c3a6SDouglas Gregor if (!Result->isAvailable()) 420b53e5483SLawrence Crowl return KnownHeader(); 4211fb5c3a6SDouglas Gregor 42297da9178SDaniel Jasper return Headers[File].back(); 423a89c5ac4SDouglas Gregor } 424a89c5ac4SDouglas Gregor 425b53e5483SLawrence Crowl return KnownHeader(); 426ab0c8a84SDouglas Gregor } 427ab0c8a84SDouglas Gregor 428e4412640SArgyrios Kyrtzidis bool ModuleMap::isHeaderInUnavailableModule(const FileEntry *Header) const { 42950996ce1SRichard Smith return isHeaderUnavailableInModule(Header, 0); 43050996ce1SRichard Smith } 43150996ce1SRichard Smith 43262bcd925SDmitri Gribenko bool 43362bcd925SDmitri Gribenko ModuleMap::isHeaderUnavailableInModule(const FileEntry *Header, 43462bcd925SDmitri Gribenko const Module *RequestingModule) const { 435e4412640SArgyrios Kyrtzidis HeadersMap::const_iterator Known = Headers.find(Header); 43697da9178SDaniel Jasper if (Known != Headers.end()) { 43797da9178SDaniel Jasper for (SmallVectorImpl<KnownHeader>::const_iterator 43897da9178SDaniel Jasper I = Known->second.begin(), 43997da9178SDaniel Jasper E = Known->second.end(); 44097da9178SDaniel Jasper I != E; ++I) { 44150996ce1SRichard Smith if (I->isAvailable() && (!RequestingModule || 44250996ce1SRichard Smith I->getModule()->isSubModuleOf(RequestingModule))) 44397da9178SDaniel Jasper return false; 44497da9178SDaniel Jasper } 44597da9178SDaniel Jasper return true; 44697da9178SDaniel Jasper } 4471fb5c3a6SDouglas Gregor 4481fb5c3a6SDouglas Gregor const DirectoryEntry *Dir = Header->getDir(); 449f857950dSDmitri Gribenko SmallVector<const DirectoryEntry *, 2> SkippedDirs; 4501fb5c3a6SDouglas Gregor StringRef DirName = Dir->getName(); 4511fb5c3a6SDouglas Gregor 45250996ce1SRichard Smith auto IsUnavailable = [&](const Module *M) { 45350996ce1SRichard Smith return !M->isAvailable() && (!RequestingModule || 45450996ce1SRichard Smith M->isSubModuleOf(RequestingModule)); 45550996ce1SRichard Smith }; 45650996ce1SRichard Smith 4571fb5c3a6SDouglas Gregor // Keep walking up the directory hierarchy, looking for a directory with 4581fb5c3a6SDouglas Gregor // an umbrella header. 4591fb5c3a6SDouglas Gregor do { 460e4412640SArgyrios Kyrtzidis llvm::DenseMap<const DirectoryEntry *, Module *>::const_iterator KnownDir 4611fb5c3a6SDouglas Gregor = UmbrellaDirs.find(Dir); 4621fb5c3a6SDouglas Gregor if (KnownDir != UmbrellaDirs.end()) { 4631fb5c3a6SDouglas Gregor Module *Found = KnownDir->second; 46450996ce1SRichard Smith if (IsUnavailable(Found)) 4651fb5c3a6SDouglas Gregor return true; 4661fb5c3a6SDouglas Gregor 4671fb5c3a6SDouglas Gregor // Search up the module stack until we find a module with an umbrella 4681fb5c3a6SDouglas Gregor // directory. 4691fb5c3a6SDouglas Gregor Module *UmbrellaModule = Found; 4701fb5c3a6SDouglas Gregor while (!UmbrellaModule->getUmbrellaDir() && UmbrellaModule->Parent) 4711fb5c3a6SDouglas Gregor UmbrellaModule = UmbrellaModule->Parent; 4721fb5c3a6SDouglas Gregor 4731fb5c3a6SDouglas Gregor if (UmbrellaModule->InferSubmodules) { 4741fb5c3a6SDouglas Gregor for (unsigned I = SkippedDirs.size(); I != 0; --I) { 4751fb5c3a6SDouglas Gregor // Find or create the module that corresponds to this directory name. 476056396aeSDouglas Gregor SmallString<32> NameBuf; 477056396aeSDouglas Gregor StringRef Name = sanitizeFilenameAsIdentifier( 478056396aeSDouglas Gregor llvm::sys::path::stem(SkippedDirs[I-1]->getName()), 479056396aeSDouglas Gregor NameBuf); 4801fb5c3a6SDouglas Gregor Found = lookupModuleQualified(Name, Found); 4811fb5c3a6SDouglas Gregor if (!Found) 4821fb5c3a6SDouglas Gregor return false; 48350996ce1SRichard Smith if (IsUnavailable(Found)) 4841fb5c3a6SDouglas Gregor return true; 4851fb5c3a6SDouglas Gregor } 4861fb5c3a6SDouglas Gregor 4871fb5c3a6SDouglas Gregor // Infer a submodule with the same name as this header file. 488056396aeSDouglas Gregor SmallString<32> NameBuf; 489056396aeSDouglas Gregor StringRef Name = sanitizeFilenameAsIdentifier( 490056396aeSDouglas Gregor llvm::sys::path::stem(Header->getName()), 491056396aeSDouglas Gregor NameBuf); 4921fb5c3a6SDouglas Gregor Found = lookupModuleQualified(Name, Found); 4931fb5c3a6SDouglas Gregor if (!Found) 4941fb5c3a6SDouglas Gregor return false; 4951fb5c3a6SDouglas Gregor } 4961fb5c3a6SDouglas Gregor 49750996ce1SRichard Smith return IsUnavailable(Found); 4981fb5c3a6SDouglas Gregor } 4991fb5c3a6SDouglas Gregor 5001fb5c3a6SDouglas Gregor SkippedDirs.push_back(Dir); 5011fb5c3a6SDouglas Gregor 5021fb5c3a6SDouglas Gregor // Retrieve our parent path. 5031fb5c3a6SDouglas Gregor DirName = llvm::sys::path::parent_path(DirName); 5041fb5c3a6SDouglas Gregor if (DirName.empty()) 5051fb5c3a6SDouglas Gregor break; 5061fb5c3a6SDouglas Gregor 5071fb5c3a6SDouglas Gregor // Resolve the parent path to a directory entry. 5081f76c4e8SManuel Klimek Dir = SourceMgr.getFileManager().getDirectory(DirName); 5091fb5c3a6SDouglas Gregor } while (Dir); 5101fb5c3a6SDouglas Gregor 5111fb5c3a6SDouglas Gregor return false; 5121fb5c3a6SDouglas Gregor } 5131fb5c3a6SDouglas Gregor 514e4412640SArgyrios Kyrtzidis Module *ModuleMap::findModule(StringRef Name) const { 515e4412640SArgyrios Kyrtzidis llvm::StringMap<Module *>::const_iterator Known = Modules.find(Name); 51688bdfb0eSDouglas Gregor if (Known != Modules.end()) 51788bdfb0eSDouglas Gregor return Known->getValue(); 51888bdfb0eSDouglas Gregor 51988bdfb0eSDouglas Gregor return 0; 52088bdfb0eSDouglas Gregor } 52188bdfb0eSDouglas Gregor 522e4412640SArgyrios Kyrtzidis Module *ModuleMap::lookupModuleUnqualified(StringRef Name, 523e4412640SArgyrios Kyrtzidis Module *Context) const { 5242b82c2a5SDouglas Gregor for(; Context; Context = Context->Parent) { 5252b82c2a5SDouglas Gregor if (Module *Sub = lookupModuleQualified(Name, Context)) 5262b82c2a5SDouglas Gregor return Sub; 5272b82c2a5SDouglas Gregor } 5282b82c2a5SDouglas Gregor 5292b82c2a5SDouglas Gregor return findModule(Name); 5302b82c2a5SDouglas Gregor } 5312b82c2a5SDouglas Gregor 532e4412640SArgyrios Kyrtzidis Module *ModuleMap::lookupModuleQualified(StringRef Name, Module *Context) const{ 5332b82c2a5SDouglas Gregor if (!Context) 5342b82c2a5SDouglas Gregor return findModule(Name); 5352b82c2a5SDouglas Gregor 536eb90e830SDouglas Gregor return Context->findSubmodule(Name); 5372b82c2a5SDouglas Gregor } 5382b82c2a5SDouglas Gregor 539de3ef502SDouglas Gregor std::pair<Module *, bool> 540beee15e7SBen Langmuir ModuleMap::findOrCreateModule(StringRef Name, Module *Parent, 541beee15e7SBen Langmuir const FileEntry *ModuleMap, bool IsFramework, 54269021974SDouglas Gregor bool IsExplicit) { 54369021974SDouglas Gregor // Try to find an existing module with this name. 544eb90e830SDouglas Gregor if (Module *Sub = lookupModuleQualified(Name, Parent)) 545eb90e830SDouglas Gregor return std::make_pair(Sub, false); 54669021974SDouglas Gregor 54769021974SDouglas Gregor // Create a new module with this name. 548beee15e7SBen Langmuir Module *Result = new Module(Name, SourceLocation(), Parent, ModuleMap, 549beee15e7SBen Langmuir IsFramework, IsExplicit); 550ba7f2f71SDaniel Jasper if (LangOpts.CurrentModule == Name) { 551ba7f2f71SDaniel Jasper SourceModule = Result; 552ba7f2f71SDaniel Jasper SourceModuleName = Name; 553ba7f2f71SDaniel Jasper } 5546f722b4eSArgyrios Kyrtzidis if (!Parent) { 55569021974SDouglas Gregor Modules[Name] = Result; 5566f722b4eSArgyrios Kyrtzidis if (!LangOpts.CurrentModule.empty() && !CompilingModule && 5576f722b4eSArgyrios Kyrtzidis Name == LangOpts.CurrentModule) { 5586f722b4eSArgyrios Kyrtzidis CompilingModule = Result; 5596f722b4eSArgyrios Kyrtzidis } 5606f722b4eSArgyrios Kyrtzidis } 56169021974SDouglas Gregor return std::make_pair(Result, true); 56269021974SDouglas Gregor } 56369021974SDouglas Gregor 5649194a91dSDouglas Gregor bool ModuleMap::canInferFrameworkModule(const DirectoryEntry *ParentDir, 565e4412640SArgyrios Kyrtzidis StringRef Name, bool &IsSystem) const { 5669194a91dSDouglas Gregor // Check whether we have already looked into the parent directory 5679194a91dSDouglas Gregor // for a module map. 568e4412640SArgyrios Kyrtzidis llvm::DenseMap<const DirectoryEntry *, InferredDirectory>::const_iterator 5699194a91dSDouglas Gregor inferred = InferredDirectories.find(ParentDir); 5709194a91dSDouglas Gregor if (inferred == InferredDirectories.end()) 5719194a91dSDouglas Gregor return false; 5729194a91dSDouglas Gregor 5739194a91dSDouglas Gregor if (!inferred->second.InferModules) 5749194a91dSDouglas Gregor return false; 5759194a91dSDouglas Gregor 5769194a91dSDouglas Gregor // We're allowed to infer for this directory, but make sure it's okay 5779194a91dSDouglas Gregor // to infer this particular module. 5789194a91dSDouglas Gregor bool canInfer = std::find(inferred->second.ExcludedModules.begin(), 5799194a91dSDouglas Gregor inferred->second.ExcludedModules.end(), 5809194a91dSDouglas Gregor Name) == inferred->second.ExcludedModules.end(); 5819194a91dSDouglas Gregor 5829194a91dSDouglas Gregor if (canInfer && inferred->second.InferSystemModules) 5839194a91dSDouglas Gregor IsSystem = true; 5849194a91dSDouglas Gregor 5859194a91dSDouglas Gregor return canInfer; 5869194a91dSDouglas Gregor } 5879194a91dSDouglas Gregor 58811dfe6feSDouglas Gregor /// \brief For a framework module, infer the framework against which we 58911dfe6feSDouglas Gregor /// should link. 59011dfe6feSDouglas Gregor static void inferFrameworkLink(Module *Mod, const DirectoryEntry *FrameworkDir, 59111dfe6feSDouglas Gregor FileManager &FileMgr) { 59211dfe6feSDouglas Gregor assert(Mod->IsFramework && "Can only infer linking for framework modules"); 59311dfe6feSDouglas Gregor assert(!Mod->isSubFramework() && 59411dfe6feSDouglas Gregor "Can only infer linking for top-level frameworks"); 59511dfe6feSDouglas Gregor 59611dfe6feSDouglas Gregor SmallString<128> LibName; 59711dfe6feSDouglas Gregor LibName += FrameworkDir->getName(); 59811dfe6feSDouglas Gregor llvm::sys::path::append(LibName, Mod->Name); 59911dfe6feSDouglas Gregor if (FileMgr.getFile(LibName)) { 60011dfe6feSDouglas Gregor Mod->LinkLibraries.push_back(Module::LinkLibrary(Mod->Name, 60111dfe6feSDouglas Gregor /*IsFramework=*/true)); 60211dfe6feSDouglas Gregor } 60311dfe6feSDouglas Gregor } 60411dfe6feSDouglas Gregor 605de3ef502SDouglas Gregor Module * 60656c64013SDouglas Gregor ModuleMap::inferFrameworkModule(StringRef ModuleName, 607e89dbc1dSDouglas Gregor const DirectoryEntry *FrameworkDir, 608a686e1b0SDouglas Gregor bool IsSystem, 609e89dbc1dSDouglas Gregor Module *Parent) { 61056c64013SDouglas Gregor // Check whether we've already found this module. 611e89dbc1dSDouglas Gregor if (Module *Mod = lookupModuleQualified(ModuleName, Parent)) 612e89dbc1dSDouglas Gregor return Mod; 613e89dbc1dSDouglas Gregor 6141f76c4e8SManuel Klimek FileManager &FileMgr = SourceMgr.getFileManager(); 61556c64013SDouglas Gregor 6169194a91dSDouglas Gregor // If the framework has a parent path from which we're allowed to infer 6179194a91dSDouglas Gregor // a framework module, do so. 618beee15e7SBen Langmuir const FileEntry *ModuleMapFile = nullptr; 6199194a91dSDouglas Gregor if (!Parent) { 6204ddf2221SDouglas Gregor // Determine whether we're allowed to infer a module map. 621e00c8b20SDouglas Gregor 6224ddf2221SDouglas Gregor // Note: as an egregious but useful hack we use the real path here, because 6234ddf2221SDouglas Gregor // we might be looking at an embedded framework that symlinks out to a 6244ddf2221SDouglas Gregor // top-level framework, and we need to infer as if we were naming the 6254ddf2221SDouglas Gregor // top-level framework. 626e00c8b20SDouglas Gregor StringRef FrameworkDirName 6271f76c4e8SManuel Klimek = SourceMgr.getFileManager().getCanonicalName(FrameworkDir); 6284ddf2221SDouglas Gregor 6299194a91dSDouglas Gregor bool canInfer = false; 6304ddf2221SDouglas Gregor if (llvm::sys::path::has_parent_path(FrameworkDirName)) { 6319194a91dSDouglas Gregor // Figure out the parent path. 6324ddf2221SDouglas Gregor StringRef Parent = llvm::sys::path::parent_path(FrameworkDirName); 6339194a91dSDouglas Gregor if (const DirectoryEntry *ParentDir = FileMgr.getDirectory(Parent)) { 6349194a91dSDouglas Gregor // Check whether we have already looked into the parent directory 6359194a91dSDouglas Gregor // for a module map. 636e4412640SArgyrios Kyrtzidis llvm::DenseMap<const DirectoryEntry *, InferredDirectory>::const_iterator 6379194a91dSDouglas Gregor inferred = InferredDirectories.find(ParentDir); 6389194a91dSDouglas Gregor if (inferred == InferredDirectories.end()) { 6399194a91dSDouglas Gregor // We haven't looked here before. Load a module map, if there is 6409194a91dSDouglas Gregor // one. 641984e1df7SBen Langmuir bool IsFrameworkDir = Parent.endswith(".framework"); 642984e1df7SBen Langmuir if (const FileEntry *ModMapFile = 643984e1df7SBen Langmuir HeaderInfo.lookupModuleMapFile(ParentDir, IsFrameworkDir)) { 644963c5535SDouglas Gregor parseModuleMapFile(ModMapFile, IsSystem); 6459194a91dSDouglas Gregor inferred = InferredDirectories.find(ParentDir); 6469194a91dSDouglas Gregor } 6479194a91dSDouglas Gregor 6489194a91dSDouglas Gregor if (inferred == InferredDirectories.end()) 6499194a91dSDouglas Gregor inferred = InferredDirectories.insert( 6509194a91dSDouglas Gregor std::make_pair(ParentDir, InferredDirectory())).first; 6519194a91dSDouglas Gregor } 6529194a91dSDouglas Gregor 6539194a91dSDouglas Gregor if (inferred->second.InferModules) { 6549194a91dSDouglas Gregor // We're allowed to infer for this directory, but make sure it's okay 6559194a91dSDouglas Gregor // to infer this particular module. 6564ddf2221SDouglas Gregor StringRef Name = llvm::sys::path::stem(FrameworkDirName); 6579194a91dSDouglas Gregor canInfer = std::find(inferred->second.ExcludedModules.begin(), 6589194a91dSDouglas Gregor inferred->second.ExcludedModules.end(), 6599194a91dSDouglas Gregor Name) == inferred->second.ExcludedModules.end(); 6609194a91dSDouglas Gregor 6619194a91dSDouglas Gregor if (inferred->second.InferSystemModules) 6629194a91dSDouglas Gregor IsSystem = true; 663beee15e7SBen Langmuir ModuleMapFile = inferred->second.ModuleMapFile; 6649194a91dSDouglas Gregor } 6659194a91dSDouglas Gregor } 6669194a91dSDouglas Gregor } 6679194a91dSDouglas Gregor 6689194a91dSDouglas Gregor // If we're not allowed to infer a framework module, don't. 6699194a91dSDouglas Gregor if (!canInfer) 6709194a91dSDouglas Gregor return 0; 671beee15e7SBen Langmuir } else 672beee15e7SBen Langmuir ModuleMapFile = Parent->ModuleMap; 6739194a91dSDouglas Gregor 6749194a91dSDouglas Gregor 67556c64013SDouglas Gregor // Look for an umbrella header. 6762c1dd271SDylan Noblesmith SmallString<128> UmbrellaName = StringRef(FrameworkDir->getName()); 67717381a06SBenjamin Kramer llvm::sys::path::append(UmbrellaName, "Headers", ModuleName + ".h"); 678e89dbc1dSDouglas Gregor const FileEntry *UmbrellaHeader = FileMgr.getFile(UmbrellaName); 67956c64013SDouglas Gregor 68056c64013SDouglas Gregor // FIXME: If there's no umbrella header, we could probably scan the 68156c64013SDouglas Gregor // framework to load *everything*. But, it's not clear that this is a good 68256c64013SDouglas Gregor // idea. 68356c64013SDouglas Gregor if (!UmbrellaHeader) 68456c64013SDouglas Gregor return 0; 68556c64013SDouglas Gregor 686beee15e7SBen Langmuir Module *Result = new Module(ModuleName, SourceLocation(), Parent, ModuleMapFile, 687e89dbc1dSDouglas Gregor /*IsFramework=*/true, /*IsExplicit=*/false); 688ba7f2f71SDaniel Jasper if (LangOpts.CurrentModule == ModuleName) { 689ba7f2f71SDaniel Jasper SourceModule = Result; 690ba7f2f71SDaniel Jasper SourceModuleName = ModuleName; 691ba7f2f71SDaniel Jasper } 692a686e1b0SDouglas Gregor if (IsSystem) 693a686e1b0SDouglas Gregor Result->IsSystem = IsSystem; 694a686e1b0SDouglas Gregor 695eb90e830SDouglas Gregor if (!Parent) 696e89dbc1dSDouglas Gregor Modules[ModuleName] = Result; 697e89dbc1dSDouglas Gregor 698322f633cSDouglas Gregor // umbrella header "umbrella-header-name" 69973141fa9SDouglas Gregor Result->Umbrella = UmbrellaHeader; 70097da9178SDaniel Jasper Headers[UmbrellaHeader].push_back(KnownHeader(Result, NormalHeader)); 7014dc71835SDouglas Gregor UmbrellaDirs[UmbrellaHeader->getDir()] = Result; 702d8bd7537SDouglas Gregor 703d8bd7537SDouglas Gregor // export * 704d8bd7537SDouglas Gregor Result->Exports.push_back(Module::ExportDecl(0, true)); 705d8bd7537SDouglas Gregor 706a89c5ac4SDouglas Gregor // module * { export * } 707a89c5ac4SDouglas Gregor Result->InferSubmodules = true; 708a89c5ac4SDouglas Gregor Result->InferExportWildcard = true; 709a89c5ac4SDouglas Gregor 710e89dbc1dSDouglas Gregor // Look for subframeworks. 711e89dbc1dSDouglas Gregor llvm::error_code EC; 7122c1dd271SDylan Noblesmith SmallString<128> SubframeworksDirName 713ddaa69cbSDouglas Gregor = StringRef(FrameworkDir->getName()); 714e89dbc1dSDouglas Gregor llvm::sys::path::append(SubframeworksDirName, "Frameworks"); 7152d4d8cb3SBenjamin Kramer llvm::sys::path::native(SubframeworksDirName); 716ddaa69cbSDouglas Gregor for (llvm::sys::fs::directory_iterator 7172d4d8cb3SBenjamin Kramer Dir(SubframeworksDirName.str(), EC), DirEnd; 718e89dbc1dSDouglas Gregor Dir != DirEnd && !EC; Dir.increment(EC)) { 719e89dbc1dSDouglas Gregor if (!StringRef(Dir->path()).endswith(".framework")) 720e89dbc1dSDouglas Gregor continue; 721f2161a70SDouglas Gregor 722e89dbc1dSDouglas Gregor if (const DirectoryEntry *SubframeworkDir 723e89dbc1dSDouglas Gregor = FileMgr.getDirectory(Dir->path())) { 72407c22b78SDouglas Gregor // Note: as an egregious but useful hack, we use the real path here and 72507c22b78SDouglas Gregor // check whether it is actually a subdirectory of the parent directory. 72607c22b78SDouglas Gregor // This will not be the case if the 'subframework' is actually a symlink 72707c22b78SDouglas Gregor // out to a top-level framework. 728e00c8b20SDouglas Gregor StringRef SubframeworkDirName = FileMgr.getCanonicalName(SubframeworkDir); 72907c22b78SDouglas Gregor bool FoundParent = false; 73007c22b78SDouglas Gregor do { 73107c22b78SDouglas Gregor // Get the parent directory name. 73207c22b78SDouglas Gregor SubframeworkDirName 73307c22b78SDouglas Gregor = llvm::sys::path::parent_path(SubframeworkDirName); 73407c22b78SDouglas Gregor if (SubframeworkDirName.empty()) 73507c22b78SDouglas Gregor break; 73607c22b78SDouglas Gregor 73707c22b78SDouglas Gregor if (FileMgr.getDirectory(SubframeworkDirName) == FrameworkDir) { 73807c22b78SDouglas Gregor FoundParent = true; 73907c22b78SDouglas Gregor break; 74007c22b78SDouglas Gregor } 74107c22b78SDouglas Gregor } while (true); 74207c22b78SDouglas Gregor 74307c22b78SDouglas Gregor if (!FoundParent) 74407c22b78SDouglas Gregor continue; 74507c22b78SDouglas Gregor 746e89dbc1dSDouglas Gregor // FIXME: Do we want to warn about subframeworks without umbrella headers? 747056396aeSDouglas Gregor SmallString<32> NameBuf; 748056396aeSDouglas Gregor inferFrameworkModule(sanitizeFilenameAsIdentifier( 749056396aeSDouglas Gregor llvm::sys::path::stem(Dir->path()), NameBuf), 750056396aeSDouglas Gregor SubframeworkDir, IsSystem, Result); 751e89dbc1dSDouglas Gregor } 752e89dbc1dSDouglas Gregor } 753e89dbc1dSDouglas Gregor 75411dfe6feSDouglas Gregor // If the module is a top-level framework, automatically link against the 75511dfe6feSDouglas Gregor // framework. 75611dfe6feSDouglas Gregor if (!Result->isSubFramework()) { 75711dfe6feSDouglas Gregor inferFrameworkLink(Result, FrameworkDir, FileMgr); 75811dfe6feSDouglas Gregor } 75911dfe6feSDouglas Gregor 76056c64013SDouglas Gregor return Result; 76156c64013SDouglas Gregor } 76256c64013SDouglas Gregor 763a89c5ac4SDouglas Gregor void ModuleMap::setUmbrellaHeader(Module *Mod, const FileEntry *UmbrellaHeader){ 76497da9178SDaniel Jasper Headers[UmbrellaHeader].push_back(KnownHeader(Mod, NormalHeader)); 76573141fa9SDouglas Gregor Mod->Umbrella = UmbrellaHeader; 7667033127bSDouglas Gregor UmbrellaDirs[UmbrellaHeader->getDir()] = Mod; 767a89c5ac4SDouglas Gregor } 768a89c5ac4SDouglas Gregor 769524e33e1SDouglas Gregor void ModuleMap::setUmbrellaDir(Module *Mod, const DirectoryEntry *UmbrellaDir) { 770524e33e1SDouglas Gregor Mod->Umbrella = UmbrellaDir; 771524e33e1SDouglas Gregor UmbrellaDirs[UmbrellaDir] = Mod; 772524e33e1SDouglas Gregor } 773524e33e1SDouglas Gregor 77459527666SDouglas Gregor void ModuleMap::addHeader(Module *Mod, const FileEntry *Header, 775b53e5483SLawrence Crowl ModuleHeaderRole Role) { 776b53e5483SLawrence Crowl if (Role == ExcludedHeader) { 77759527666SDouglas Gregor Mod->ExcludedHeaders.push_back(Header); 778b146baabSArgyrios Kyrtzidis } else { 779b53e5483SLawrence Crowl if (Role == PrivateHeader) 780b53e5483SLawrence Crowl Mod->PrivateHeaders.push_back(Header); 781b53e5483SLawrence Crowl else 782b53e5483SLawrence Crowl Mod->NormalHeaders.push_back(Header); 7836f722b4eSArgyrios Kyrtzidis bool isCompilingModuleHeader = Mod->getTopLevelModule() == CompilingModule; 784b53e5483SLawrence Crowl HeaderInfo.MarkFileModuleHeader(Header, Role, isCompilingModuleHeader); 785b146baabSArgyrios Kyrtzidis } 78697da9178SDaniel Jasper Headers[Header].push_back(KnownHeader(Mod, Role)); 787a89c5ac4SDouglas Gregor } 788a89c5ac4SDouglas Gregor 789514b636aSDouglas Gregor const FileEntry * 790e4412640SArgyrios Kyrtzidis ModuleMap::getContainingModuleMapFile(Module *Module) const { 7911f76c4e8SManuel Klimek if (Module->DefinitionLoc.isInvalid()) 792514b636aSDouglas Gregor return 0; 793514b636aSDouglas Gregor 7941f76c4e8SManuel Klimek return SourceMgr.getFileEntryForID( 7951f76c4e8SManuel Klimek SourceMgr.getFileID(Module->DefinitionLoc)); 796514b636aSDouglas Gregor } 797514b636aSDouglas Gregor 798718292f2SDouglas Gregor void ModuleMap::dump() { 799718292f2SDouglas Gregor llvm::errs() << "Modules:"; 800718292f2SDouglas Gregor for (llvm::StringMap<Module *>::iterator M = Modules.begin(), 801718292f2SDouglas Gregor MEnd = Modules.end(); 802718292f2SDouglas Gregor M != MEnd; ++M) 803d28d1b8dSDouglas Gregor M->getValue()->print(llvm::errs(), 2); 804718292f2SDouglas Gregor 805718292f2SDouglas Gregor llvm::errs() << "Headers:"; 80659527666SDouglas Gregor for (HeadersMap::iterator H = Headers.begin(), HEnd = Headers.end(); 807718292f2SDouglas Gregor H != HEnd; ++H) { 80897da9178SDaniel Jasper llvm::errs() << " \"" << H->first->getName() << "\" -> "; 80997da9178SDaniel Jasper for (SmallVectorImpl<KnownHeader>::const_iterator I = H->second.begin(), 81097da9178SDaniel Jasper E = H->second.end(); 81197da9178SDaniel Jasper I != E; ++I) { 81297da9178SDaniel Jasper if (I != H->second.begin()) 81397da9178SDaniel Jasper llvm::errs() << ","; 81497da9178SDaniel Jasper llvm::errs() << I->getModule()->getFullModuleName(); 81597da9178SDaniel Jasper } 81697da9178SDaniel Jasper llvm::errs() << "\n"; 817718292f2SDouglas Gregor } 818718292f2SDouglas Gregor } 819718292f2SDouglas Gregor 8202b82c2a5SDouglas Gregor bool ModuleMap::resolveExports(Module *Mod, bool Complain) { 8212b82c2a5SDouglas Gregor bool HadError = false; 8222b82c2a5SDouglas Gregor for (unsigned I = 0, N = Mod->UnresolvedExports.size(); I != N; ++I) { 8232b82c2a5SDouglas Gregor Module::ExportDecl Export = resolveExport(Mod, Mod->UnresolvedExports[I], 8242b82c2a5SDouglas Gregor Complain); 825f5eedd05SDouglas Gregor if (Export.getPointer() || Export.getInt()) 8262b82c2a5SDouglas Gregor Mod->Exports.push_back(Export); 8272b82c2a5SDouglas Gregor else 8282b82c2a5SDouglas Gregor HadError = true; 8292b82c2a5SDouglas Gregor } 8302b82c2a5SDouglas Gregor Mod->UnresolvedExports.clear(); 8312b82c2a5SDouglas Gregor return HadError; 8322b82c2a5SDouglas Gregor } 8332b82c2a5SDouglas Gregor 834ba7f2f71SDaniel Jasper bool ModuleMap::resolveUses(Module *Mod, bool Complain) { 835ba7f2f71SDaniel Jasper bool HadError = false; 836ba7f2f71SDaniel Jasper for (unsigned I = 0, N = Mod->UnresolvedDirectUses.size(); I != N; ++I) { 837ba7f2f71SDaniel Jasper Module *DirectUse = 838ba7f2f71SDaniel Jasper resolveModuleId(Mod->UnresolvedDirectUses[I], Mod, Complain); 839ba7f2f71SDaniel Jasper if (DirectUse) 840ba7f2f71SDaniel Jasper Mod->DirectUses.push_back(DirectUse); 841ba7f2f71SDaniel Jasper else 842ba7f2f71SDaniel Jasper HadError = true; 843ba7f2f71SDaniel Jasper } 844ba7f2f71SDaniel Jasper Mod->UnresolvedDirectUses.clear(); 845ba7f2f71SDaniel Jasper return HadError; 846ba7f2f71SDaniel Jasper } 847ba7f2f71SDaniel Jasper 848fb912657SDouglas Gregor bool ModuleMap::resolveConflicts(Module *Mod, bool Complain) { 849fb912657SDouglas Gregor bool HadError = false; 850fb912657SDouglas Gregor for (unsigned I = 0, N = Mod->UnresolvedConflicts.size(); I != N; ++I) { 851fb912657SDouglas Gregor Module *OtherMod = resolveModuleId(Mod->UnresolvedConflicts[I].Id, 852fb912657SDouglas Gregor Mod, Complain); 853fb912657SDouglas Gregor if (!OtherMod) { 854fb912657SDouglas Gregor HadError = true; 855fb912657SDouglas Gregor continue; 856fb912657SDouglas Gregor } 857fb912657SDouglas Gregor 858fb912657SDouglas Gregor Module::Conflict Conflict; 859fb912657SDouglas Gregor Conflict.Other = OtherMod; 860fb912657SDouglas Gregor Conflict.Message = Mod->UnresolvedConflicts[I].Message; 861fb912657SDouglas Gregor Mod->Conflicts.push_back(Conflict); 862fb912657SDouglas Gregor } 863fb912657SDouglas Gregor Mod->UnresolvedConflicts.clear(); 864fb912657SDouglas Gregor return HadError; 865fb912657SDouglas Gregor } 866fb912657SDouglas Gregor 8670093b3c7SDouglas Gregor Module *ModuleMap::inferModuleFromLocation(FullSourceLoc Loc) { 8680093b3c7SDouglas Gregor if (Loc.isInvalid()) 8690093b3c7SDouglas Gregor return 0; 8700093b3c7SDouglas Gregor 8710093b3c7SDouglas Gregor // Use the expansion location to determine which module we're in. 8720093b3c7SDouglas Gregor FullSourceLoc ExpansionLoc = Loc.getExpansionLoc(); 8730093b3c7SDouglas Gregor if (!ExpansionLoc.isFileID()) 8740093b3c7SDouglas Gregor return 0; 8750093b3c7SDouglas Gregor 8760093b3c7SDouglas Gregor 8770093b3c7SDouglas Gregor const SourceManager &SrcMgr = Loc.getManager(); 8780093b3c7SDouglas Gregor FileID ExpansionFileID = ExpansionLoc.getFileID(); 879224d8a74SDouglas Gregor 880224d8a74SDouglas Gregor while (const FileEntry *ExpansionFile 881224d8a74SDouglas Gregor = SrcMgr.getFileEntryForID(ExpansionFileID)) { 882224d8a74SDouglas Gregor // Find the module that owns this header (if any). 883b53e5483SLawrence Crowl if (Module *Mod = findModuleForHeader(ExpansionFile).getModule()) 884224d8a74SDouglas Gregor return Mod; 885224d8a74SDouglas Gregor 886224d8a74SDouglas Gregor // No module owns this header, so look up the inclusion chain to see if 887224d8a74SDouglas Gregor // any included header has an associated module. 888224d8a74SDouglas Gregor SourceLocation IncludeLoc = SrcMgr.getIncludeLoc(ExpansionFileID); 889224d8a74SDouglas Gregor if (IncludeLoc.isInvalid()) 8900093b3c7SDouglas Gregor return 0; 8910093b3c7SDouglas Gregor 892224d8a74SDouglas Gregor ExpansionFileID = SrcMgr.getFileID(IncludeLoc); 893224d8a74SDouglas Gregor } 894224d8a74SDouglas Gregor 895224d8a74SDouglas Gregor return 0; 8960093b3c7SDouglas Gregor } 8970093b3c7SDouglas Gregor 898718292f2SDouglas Gregor //----------------------------------------------------------------------------// 899718292f2SDouglas Gregor // Module map file parser 900718292f2SDouglas Gregor //----------------------------------------------------------------------------// 901718292f2SDouglas Gregor 902718292f2SDouglas Gregor namespace clang { 903718292f2SDouglas Gregor /// \brief A token in a module map file. 904718292f2SDouglas Gregor struct MMToken { 905718292f2SDouglas Gregor enum TokenKind { 9061fb5c3a6SDouglas Gregor Comma, 90735b13eceSDouglas Gregor ConfigMacros, 908fb912657SDouglas Gregor Conflict, 909718292f2SDouglas Gregor EndOfFile, 910718292f2SDouglas Gregor HeaderKeyword, 911718292f2SDouglas Gregor Identifier, 912a3feee2aSRichard Smith Exclaim, 91359527666SDouglas Gregor ExcludeKeyword, 914718292f2SDouglas Gregor ExplicitKeyword, 9152b82c2a5SDouglas Gregor ExportKeyword, 91697292843SDaniel Jasper ExternKeyword, 917755b2055SDouglas Gregor FrameworkKeyword, 9186ddfca91SDouglas Gregor LinkKeyword, 919718292f2SDouglas Gregor ModuleKeyword, 9202b82c2a5SDouglas Gregor Period, 921b53e5483SLawrence Crowl PrivateKeyword, 922718292f2SDouglas Gregor UmbrellaKeyword, 923ba7f2f71SDaniel Jasper UseKeyword, 9241fb5c3a6SDouglas Gregor RequiresKeyword, 9252b82c2a5SDouglas Gregor Star, 926718292f2SDouglas Gregor StringLiteral, 927718292f2SDouglas Gregor LBrace, 928a686e1b0SDouglas Gregor RBrace, 929a686e1b0SDouglas Gregor LSquare, 930a686e1b0SDouglas Gregor RSquare 931718292f2SDouglas Gregor } Kind; 932718292f2SDouglas Gregor 933718292f2SDouglas Gregor unsigned Location; 934718292f2SDouglas Gregor unsigned StringLength; 935718292f2SDouglas Gregor const char *StringData; 936718292f2SDouglas Gregor 937718292f2SDouglas Gregor void clear() { 938718292f2SDouglas Gregor Kind = EndOfFile; 939718292f2SDouglas Gregor Location = 0; 940718292f2SDouglas Gregor StringLength = 0; 941718292f2SDouglas Gregor StringData = 0; 942718292f2SDouglas Gregor } 943718292f2SDouglas Gregor 944718292f2SDouglas Gregor bool is(TokenKind K) const { return Kind == K; } 945718292f2SDouglas Gregor 946718292f2SDouglas Gregor SourceLocation getLocation() const { 947718292f2SDouglas Gregor return SourceLocation::getFromRawEncoding(Location); 948718292f2SDouglas Gregor } 949718292f2SDouglas Gregor 950718292f2SDouglas Gregor StringRef getString() const { 951718292f2SDouglas Gregor return StringRef(StringData, StringLength); 952718292f2SDouglas Gregor } 953718292f2SDouglas Gregor }; 954718292f2SDouglas Gregor 9559194a91dSDouglas Gregor /// \brief The set of attributes that can be attached to a module. 9564442605fSBill Wendling struct Attributes { 95777944868SRichard Smith Attributes() : IsSystem(), IsExternC(), IsExhaustive() { } 9589194a91dSDouglas Gregor 9599194a91dSDouglas Gregor /// \brief Whether this is a system module. 9609194a91dSDouglas Gregor unsigned IsSystem : 1; 96135b13eceSDouglas Gregor 96277944868SRichard Smith /// \brief Whether this is an extern "C" module. 96377944868SRichard Smith unsigned IsExternC : 1; 96477944868SRichard Smith 96535b13eceSDouglas Gregor /// \brief Whether this is an exhaustive set of configuration macros. 96635b13eceSDouglas Gregor unsigned IsExhaustive : 1; 9679194a91dSDouglas Gregor }; 9689194a91dSDouglas Gregor 9699194a91dSDouglas Gregor 970718292f2SDouglas Gregor class ModuleMapParser { 971718292f2SDouglas Gregor Lexer &L; 972718292f2SDouglas Gregor SourceManager &SourceMgr; 973bc10b9fbSDouglas Gregor 974bc10b9fbSDouglas Gregor /// \brief Default target information, used only for string literal 975bc10b9fbSDouglas Gregor /// parsing. 976bc10b9fbSDouglas Gregor const TargetInfo *Target; 977bc10b9fbSDouglas Gregor 978718292f2SDouglas Gregor DiagnosticsEngine &Diags; 979718292f2SDouglas Gregor ModuleMap ⤅ 980718292f2SDouglas Gregor 981beee15e7SBen Langmuir /// \brief The current module map file. 982beee15e7SBen Langmuir const FileEntry *ModuleMapFile; 983beee15e7SBen Langmuir 9845257fc63SDouglas Gregor /// \brief The directory that this module map resides in. 9855257fc63SDouglas Gregor const DirectoryEntry *Directory; 9865257fc63SDouglas Gregor 9873ec6663bSDouglas Gregor /// \brief The directory containing Clang-supplied headers. 9883ec6663bSDouglas Gregor const DirectoryEntry *BuiltinIncludeDir; 9893ec6663bSDouglas Gregor 990963c5535SDouglas Gregor /// \brief Whether this module map is in a system header directory. 991963c5535SDouglas Gregor bool IsSystem; 992963c5535SDouglas Gregor 993718292f2SDouglas Gregor /// \brief Whether an error occurred. 994718292f2SDouglas Gregor bool HadError; 995718292f2SDouglas Gregor 996718292f2SDouglas Gregor /// \brief Stores string data for the various string literals referenced 997718292f2SDouglas Gregor /// during parsing. 998718292f2SDouglas Gregor llvm::BumpPtrAllocator StringData; 999718292f2SDouglas Gregor 1000718292f2SDouglas Gregor /// \brief The current token. 1001718292f2SDouglas Gregor MMToken Tok; 1002718292f2SDouglas Gregor 1003718292f2SDouglas Gregor /// \brief The active module. 1004de3ef502SDouglas Gregor Module *ActiveModule; 1005718292f2SDouglas Gregor 1006718292f2SDouglas Gregor /// \brief Consume the current token and return its location. 1007718292f2SDouglas Gregor SourceLocation consumeToken(); 1008718292f2SDouglas Gregor 1009718292f2SDouglas Gregor /// \brief Skip tokens until we reach the a token with the given kind 1010718292f2SDouglas Gregor /// (or the end of the file). 1011718292f2SDouglas Gregor void skipUntil(MMToken::TokenKind K); 1012718292f2SDouglas Gregor 1013f857950dSDmitri Gribenko typedef SmallVector<std::pair<std::string, SourceLocation>, 2> ModuleId; 1014e7ab3669SDouglas Gregor bool parseModuleId(ModuleId &Id); 1015718292f2SDouglas Gregor void parseModuleDecl(); 101697292843SDaniel Jasper void parseExternModuleDecl(); 10171fb5c3a6SDouglas Gregor void parseRequiresDecl(); 1018b53e5483SLawrence Crowl void parseHeaderDecl(clang::MMToken::TokenKind, 1019b53e5483SLawrence Crowl SourceLocation LeadingLoc); 1020524e33e1SDouglas Gregor void parseUmbrellaDirDecl(SourceLocation UmbrellaLoc); 10212b82c2a5SDouglas Gregor void parseExportDecl(); 1022ba7f2f71SDaniel Jasper void parseUseDecl(); 10236ddfca91SDouglas Gregor void parseLinkDecl(); 102435b13eceSDouglas Gregor void parseConfigMacros(); 1025fb912657SDouglas Gregor void parseConflict(); 10269194a91dSDouglas Gregor void parseInferredModuleDecl(bool Framework, bool Explicit); 10274442605fSBill Wendling bool parseOptionalAttributes(Attributes &Attrs); 1028718292f2SDouglas Gregor 1029718292f2SDouglas Gregor public: 1030718292f2SDouglas Gregor explicit ModuleMapParser(Lexer &L, SourceManager &SourceMgr, 1031bc10b9fbSDouglas Gregor const TargetInfo *Target, 1032718292f2SDouglas Gregor DiagnosticsEngine &Diags, 10335257fc63SDouglas Gregor ModuleMap &Map, 1034beee15e7SBen Langmuir const FileEntry *ModuleMapFile, 10353ec6663bSDouglas Gregor const DirectoryEntry *Directory, 1036963c5535SDouglas Gregor const DirectoryEntry *BuiltinIncludeDir, 1037963c5535SDouglas Gregor bool IsSystem) 1038bc10b9fbSDouglas Gregor : L(L), SourceMgr(SourceMgr), Target(Target), Diags(Diags), Map(Map), 1039beee15e7SBen Langmuir ModuleMapFile(ModuleMapFile), Directory(Directory), 1040beee15e7SBen Langmuir BuiltinIncludeDir(BuiltinIncludeDir), IsSystem(IsSystem), 1041beee15e7SBen Langmuir HadError(false), ActiveModule(0) 1042718292f2SDouglas Gregor { 1043718292f2SDouglas Gregor Tok.clear(); 1044718292f2SDouglas Gregor consumeToken(); 1045718292f2SDouglas Gregor } 1046718292f2SDouglas Gregor 1047718292f2SDouglas Gregor bool parseModuleMapFile(); 1048718292f2SDouglas Gregor }; 1049718292f2SDouglas Gregor } 1050718292f2SDouglas Gregor 1051718292f2SDouglas Gregor SourceLocation ModuleMapParser::consumeToken() { 1052718292f2SDouglas Gregor retry: 1053718292f2SDouglas Gregor SourceLocation Result = Tok.getLocation(); 1054718292f2SDouglas Gregor Tok.clear(); 1055718292f2SDouglas Gregor 1056718292f2SDouglas Gregor Token LToken; 1057718292f2SDouglas Gregor L.LexFromRawLexer(LToken); 1058718292f2SDouglas Gregor Tok.Location = LToken.getLocation().getRawEncoding(); 1059718292f2SDouglas Gregor switch (LToken.getKind()) { 1060718292f2SDouglas Gregor case tok::raw_identifier: 1061718292f2SDouglas Gregor Tok.StringData = LToken.getRawIdentifierData(); 1062718292f2SDouglas Gregor Tok.StringLength = LToken.getLength(); 1063718292f2SDouglas Gregor Tok.Kind = llvm::StringSwitch<MMToken::TokenKind>(Tok.getString()) 106435b13eceSDouglas Gregor .Case("config_macros", MMToken::ConfigMacros) 1065fb912657SDouglas Gregor .Case("conflict", MMToken::Conflict) 106659527666SDouglas Gregor .Case("exclude", MMToken::ExcludeKeyword) 1067718292f2SDouglas Gregor .Case("explicit", MMToken::ExplicitKeyword) 10682b82c2a5SDouglas Gregor .Case("export", MMToken::ExportKeyword) 106997292843SDaniel Jasper .Case("extern", MMToken::ExternKeyword) 1070755b2055SDouglas Gregor .Case("framework", MMToken::FrameworkKeyword) 107135b13eceSDouglas Gregor .Case("header", MMToken::HeaderKeyword) 10726ddfca91SDouglas Gregor .Case("link", MMToken::LinkKeyword) 1073718292f2SDouglas Gregor .Case("module", MMToken::ModuleKeyword) 1074b53e5483SLawrence Crowl .Case("private", MMToken::PrivateKeyword) 10751fb5c3a6SDouglas Gregor .Case("requires", MMToken::RequiresKeyword) 1076718292f2SDouglas Gregor .Case("umbrella", MMToken::UmbrellaKeyword) 1077ba7f2f71SDaniel Jasper .Case("use", MMToken::UseKeyword) 1078718292f2SDouglas Gregor .Default(MMToken::Identifier); 1079718292f2SDouglas Gregor break; 1080718292f2SDouglas Gregor 10811fb5c3a6SDouglas Gregor case tok::comma: 10821fb5c3a6SDouglas Gregor Tok.Kind = MMToken::Comma; 10831fb5c3a6SDouglas Gregor break; 10841fb5c3a6SDouglas Gregor 1085718292f2SDouglas Gregor case tok::eof: 1086718292f2SDouglas Gregor Tok.Kind = MMToken::EndOfFile; 1087718292f2SDouglas Gregor break; 1088718292f2SDouglas Gregor 1089718292f2SDouglas Gregor case tok::l_brace: 1090718292f2SDouglas Gregor Tok.Kind = MMToken::LBrace; 1091718292f2SDouglas Gregor break; 1092718292f2SDouglas Gregor 1093a686e1b0SDouglas Gregor case tok::l_square: 1094a686e1b0SDouglas Gregor Tok.Kind = MMToken::LSquare; 1095a686e1b0SDouglas Gregor break; 1096a686e1b0SDouglas Gregor 10972b82c2a5SDouglas Gregor case tok::period: 10982b82c2a5SDouglas Gregor Tok.Kind = MMToken::Period; 10992b82c2a5SDouglas Gregor break; 11002b82c2a5SDouglas Gregor 1101718292f2SDouglas Gregor case tok::r_brace: 1102718292f2SDouglas Gregor Tok.Kind = MMToken::RBrace; 1103718292f2SDouglas Gregor break; 1104718292f2SDouglas Gregor 1105a686e1b0SDouglas Gregor case tok::r_square: 1106a686e1b0SDouglas Gregor Tok.Kind = MMToken::RSquare; 1107a686e1b0SDouglas Gregor break; 1108a686e1b0SDouglas Gregor 11092b82c2a5SDouglas Gregor case tok::star: 11102b82c2a5SDouglas Gregor Tok.Kind = MMToken::Star; 11112b82c2a5SDouglas Gregor break; 11122b82c2a5SDouglas Gregor 1113a3feee2aSRichard Smith case tok::exclaim: 1114a3feee2aSRichard Smith Tok.Kind = MMToken::Exclaim; 1115a3feee2aSRichard Smith break; 1116a3feee2aSRichard Smith 1117718292f2SDouglas Gregor case tok::string_literal: { 1118d67aea28SRichard Smith if (LToken.hasUDSuffix()) { 1119d67aea28SRichard Smith Diags.Report(LToken.getLocation(), diag::err_invalid_string_udl); 1120d67aea28SRichard Smith HadError = true; 1121d67aea28SRichard Smith goto retry; 1122d67aea28SRichard Smith } 1123d67aea28SRichard Smith 1124718292f2SDouglas Gregor // Parse the string literal. 1125718292f2SDouglas Gregor LangOptions LangOpts; 1126718292f2SDouglas Gregor StringLiteralParser StringLiteral(<oken, 1, SourceMgr, LangOpts, *Target); 1127718292f2SDouglas Gregor if (StringLiteral.hadError) 1128718292f2SDouglas Gregor goto retry; 1129718292f2SDouglas Gregor 1130718292f2SDouglas Gregor // Copy the string literal into our string data allocator. 1131718292f2SDouglas Gregor unsigned Length = StringLiteral.GetStringLength(); 1132718292f2SDouglas Gregor char *Saved = StringData.Allocate<char>(Length + 1); 1133718292f2SDouglas Gregor memcpy(Saved, StringLiteral.GetString().data(), Length); 1134718292f2SDouglas Gregor Saved[Length] = 0; 1135718292f2SDouglas Gregor 1136718292f2SDouglas Gregor // Form the token. 1137718292f2SDouglas Gregor Tok.Kind = MMToken::StringLiteral; 1138718292f2SDouglas Gregor Tok.StringData = Saved; 1139718292f2SDouglas Gregor Tok.StringLength = Length; 1140718292f2SDouglas Gregor break; 1141718292f2SDouglas Gregor } 1142718292f2SDouglas Gregor 1143718292f2SDouglas Gregor case tok::comment: 1144718292f2SDouglas Gregor goto retry; 1145718292f2SDouglas Gregor 1146718292f2SDouglas Gregor default: 1147718292f2SDouglas Gregor Diags.Report(LToken.getLocation(), diag::err_mmap_unknown_token); 1148718292f2SDouglas Gregor HadError = true; 1149718292f2SDouglas Gregor goto retry; 1150718292f2SDouglas Gregor } 1151718292f2SDouglas Gregor 1152718292f2SDouglas Gregor return Result; 1153718292f2SDouglas Gregor } 1154718292f2SDouglas Gregor 1155718292f2SDouglas Gregor void ModuleMapParser::skipUntil(MMToken::TokenKind K) { 1156718292f2SDouglas Gregor unsigned braceDepth = 0; 1157a686e1b0SDouglas Gregor unsigned squareDepth = 0; 1158718292f2SDouglas Gregor do { 1159718292f2SDouglas Gregor switch (Tok.Kind) { 1160718292f2SDouglas Gregor case MMToken::EndOfFile: 1161718292f2SDouglas Gregor return; 1162718292f2SDouglas Gregor 1163718292f2SDouglas Gregor case MMToken::LBrace: 1164a686e1b0SDouglas Gregor if (Tok.is(K) && braceDepth == 0 && squareDepth == 0) 1165718292f2SDouglas Gregor return; 1166718292f2SDouglas Gregor 1167718292f2SDouglas Gregor ++braceDepth; 1168718292f2SDouglas Gregor break; 1169718292f2SDouglas Gregor 1170a686e1b0SDouglas Gregor case MMToken::LSquare: 1171a686e1b0SDouglas Gregor if (Tok.is(K) && braceDepth == 0 && squareDepth == 0) 1172a686e1b0SDouglas Gregor return; 1173a686e1b0SDouglas Gregor 1174a686e1b0SDouglas Gregor ++squareDepth; 1175a686e1b0SDouglas Gregor break; 1176a686e1b0SDouglas Gregor 1177718292f2SDouglas Gregor case MMToken::RBrace: 1178718292f2SDouglas Gregor if (braceDepth > 0) 1179718292f2SDouglas Gregor --braceDepth; 1180718292f2SDouglas Gregor else if (Tok.is(K)) 1181718292f2SDouglas Gregor return; 1182718292f2SDouglas Gregor break; 1183718292f2SDouglas Gregor 1184a686e1b0SDouglas Gregor case MMToken::RSquare: 1185a686e1b0SDouglas Gregor if (squareDepth > 0) 1186a686e1b0SDouglas Gregor --squareDepth; 1187a686e1b0SDouglas Gregor else if (Tok.is(K)) 1188a686e1b0SDouglas Gregor return; 1189a686e1b0SDouglas Gregor break; 1190a686e1b0SDouglas Gregor 1191718292f2SDouglas Gregor default: 1192a686e1b0SDouglas Gregor if (braceDepth == 0 && squareDepth == 0 && Tok.is(K)) 1193718292f2SDouglas Gregor return; 1194718292f2SDouglas Gregor break; 1195718292f2SDouglas Gregor } 1196718292f2SDouglas Gregor 1197718292f2SDouglas Gregor consumeToken(); 1198718292f2SDouglas Gregor } while (true); 1199718292f2SDouglas Gregor } 1200718292f2SDouglas Gregor 1201e7ab3669SDouglas Gregor /// \brief Parse a module-id. 1202e7ab3669SDouglas Gregor /// 1203e7ab3669SDouglas Gregor /// module-id: 1204e7ab3669SDouglas Gregor /// identifier 1205e7ab3669SDouglas Gregor /// identifier '.' module-id 1206e7ab3669SDouglas Gregor /// 1207e7ab3669SDouglas Gregor /// \returns true if an error occurred, false otherwise. 1208e7ab3669SDouglas Gregor bool ModuleMapParser::parseModuleId(ModuleId &Id) { 1209e7ab3669SDouglas Gregor Id.clear(); 1210e7ab3669SDouglas Gregor do { 12113cd34c76SDaniel Jasper if (Tok.is(MMToken::Identifier) || Tok.is(MMToken::StringLiteral)) { 1212e7ab3669SDouglas Gregor Id.push_back(std::make_pair(Tok.getString(), Tok.getLocation())); 1213e7ab3669SDouglas Gregor consumeToken(); 1214e7ab3669SDouglas Gregor } else { 1215e7ab3669SDouglas Gregor Diags.Report(Tok.getLocation(), diag::err_mmap_expected_module_name); 1216e7ab3669SDouglas Gregor return true; 1217e7ab3669SDouglas Gregor } 1218e7ab3669SDouglas Gregor 1219e7ab3669SDouglas Gregor if (!Tok.is(MMToken::Period)) 1220e7ab3669SDouglas Gregor break; 1221e7ab3669SDouglas Gregor 1222e7ab3669SDouglas Gregor consumeToken(); 1223e7ab3669SDouglas Gregor } while (true); 1224e7ab3669SDouglas Gregor 1225e7ab3669SDouglas Gregor return false; 1226e7ab3669SDouglas Gregor } 1227e7ab3669SDouglas Gregor 1228a686e1b0SDouglas Gregor namespace { 1229a686e1b0SDouglas Gregor /// \brief Enumerates the known attributes. 1230a686e1b0SDouglas Gregor enum AttributeKind { 1231a686e1b0SDouglas Gregor /// \brief An unknown attribute. 1232a686e1b0SDouglas Gregor AT_unknown, 1233a686e1b0SDouglas Gregor /// \brief The 'system' attribute. 123435b13eceSDouglas Gregor AT_system, 123577944868SRichard Smith /// \brief The 'extern_c' attribute. 123677944868SRichard Smith AT_extern_c, 123735b13eceSDouglas Gregor /// \brief The 'exhaustive' attribute. 123835b13eceSDouglas Gregor AT_exhaustive 1239a686e1b0SDouglas Gregor }; 1240a686e1b0SDouglas Gregor } 1241a686e1b0SDouglas Gregor 1242718292f2SDouglas Gregor /// \brief Parse a module declaration. 1243718292f2SDouglas Gregor /// 1244718292f2SDouglas Gregor /// module-declaration: 124597292843SDaniel Jasper /// 'extern' 'module' module-id string-literal 1246a686e1b0SDouglas Gregor /// 'explicit'[opt] 'framework'[opt] 'module' module-id attributes[opt] 1247a686e1b0SDouglas Gregor /// { module-member* } 1248a686e1b0SDouglas Gregor /// 1249718292f2SDouglas Gregor /// module-member: 12501fb5c3a6SDouglas Gregor /// requires-declaration 1251718292f2SDouglas Gregor /// header-declaration 1252e7ab3669SDouglas Gregor /// submodule-declaration 12532b82c2a5SDouglas Gregor /// export-declaration 12546ddfca91SDouglas Gregor /// link-declaration 125573441091SDouglas Gregor /// 125673441091SDouglas Gregor /// submodule-declaration: 125773441091SDouglas Gregor /// module-declaration 125873441091SDouglas Gregor /// inferred-submodule-declaration 1259718292f2SDouglas Gregor void ModuleMapParser::parseModuleDecl() { 1260755b2055SDouglas Gregor assert(Tok.is(MMToken::ExplicitKeyword) || Tok.is(MMToken::ModuleKeyword) || 126197292843SDaniel Jasper Tok.is(MMToken::FrameworkKeyword) || Tok.is(MMToken::ExternKeyword)); 126297292843SDaniel Jasper if (Tok.is(MMToken::ExternKeyword)) { 126397292843SDaniel Jasper parseExternModuleDecl(); 126497292843SDaniel Jasper return; 126597292843SDaniel Jasper } 126697292843SDaniel Jasper 1267f2161a70SDouglas Gregor // Parse 'explicit' or 'framework' keyword, if present. 1268e7ab3669SDouglas Gregor SourceLocation ExplicitLoc; 1269718292f2SDouglas Gregor bool Explicit = false; 1270f2161a70SDouglas Gregor bool Framework = false; 1271755b2055SDouglas Gregor 1272f2161a70SDouglas Gregor // Parse 'explicit' keyword, if present. 1273f2161a70SDouglas Gregor if (Tok.is(MMToken::ExplicitKeyword)) { 1274e7ab3669SDouglas Gregor ExplicitLoc = consumeToken(); 1275f2161a70SDouglas Gregor Explicit = true; 1276f2161a70SDouglas Gregor } 1277f2161a70SDouglas Gregor 1278f2161a70SDouglas Gregor // Parse 'framework' keyword, if present. 1279755b2055SDouglas Gregor if (Tok.is(MMToken::FrameworkKeyword)) { 1280755b2055SDouglas Gregor consumeToken(); 1281755b2055SDouglas Gregor Framework = true; 1282755b2055SDouglas Gregor } 1283718292f2SDouglas Gregor 1284718292f2SDouglas Gregor // Parse 'module' keyword. 1285718292f2SDouglas Gregor if (!Tok.is(MMToken::ModuleKeyword)) { 1286d6343c99SDouglas Gregor Diags.Report(Tok.getLocation(), diag::err_mmap_expected_module); 1287718292f2SDouglas Gregor consumeToken(); 1288718292f2SDouglas Gregor HadError = true; 1289718292f2SDouglas Gregor return; 1290718292f2SDouglas Gregor } 1291718292f2SDouglas Gregor consumeToken(); // 'module' keyword 1292718292f2SDouglas Gregor 129373441091SDouglas Gregor // If we have a wildcard for the module name, this is an inferred submodule. 129473441091SDouglas Gregor // Parse it. 129573441091SDouglas Gregor if (Tok.is(MMToken::Star)) 12969194a91dSDouglas Gregor return parseInferredModuleDecl(Framework, Explicit); 129773441091SDouglas Gregor 1298718292f2SDouglas Gregor // Parse the module name. 1299e7ab3669SDouglas Gregor ModuleId Id; 1300e7ab3669SDouglas Gregor if (parseModuleId(Id)) { 1301718292f2SDouglas Gregor HadError = true; 1302718292f2SDouglas Gregor return; 1303718292f2SDouglas Gregor } 1304e7ab3669SDouglas Gregor 1305e7ab3669SDouglas Gregor if (ActiveModule) { 1306e7ab3669SDouglas Gregor if (Id.size() > 1) { 1307e7ab3669SDouglas Gregor Diags.Report(Id.front().second, diag::err_mmap_nested_submodule_id) 1308e7ab3669SDouglas Gregor << SourceRange(Id.front().second, Id.back().second); 1309e7ab3669SDouglas Gregor 1310e7ab3669SDouglas Gregor HadError = true; 1311e7ab3669SDouglas Gregor return; 1312e7ab3669SDouglas Gregor } 1313e7ab3669SDouglas Gregor } else if (Id.size() == 1 && Explicit) { 1314e7ab3669SDouglas Gregor // Top-level modules can't be explicit. 1315e7ab3669SDouglas Gregor Diags.Report(ExplicitLoc, diag::err_mmap_explicit_top_level); 1316e7ab3669SDouglas Gregor Explicit = false; 1317e7ab3669SDouglas Gregor ExplicitLoc = SourceLocation(); 1318e7ab3669SDouglas Gregor HadError = true; 1319e7ab3669SDouglas Gregor } 1320e7ab3669SDouglas Gregor 1321e7ab3669SDouglas Gregor Module *PreviousActiveModule = ActiveModule; 1322e7ab3669SDouglas Gregor if (Id.size() > 1) { 1323e7ab3669SDouglas Gregor // This module map defines a submodule. Go find the module of which it 1324e7ab3669SDouglas Gregor // is a submodule. 1325e7ab3669SDouglas Gregor ActiveModule = 0; 1326e7ab3669SDouglas Gregor for (unsigned I = 0, N = Id.size() - 1; I != N; ++I) { 1327e7ab3669SDouglas Gregor if (Module *Next = Map.lookupModuleQualified(Id[I].first, ActiveModule)) { 1328e7ab3669SDouglas Gregor ActiveModule = Next; 1329e7ab3669SDouglas Gregor continue; 1330e7ab3669SDouglas Gregor } 1331e7ab3669SDouglas Gregor 1332e7ab3669SDouglas Gregor if (ActiveModule) { 1333e7ab3669SDouglas Gregor Diags.Report(Id[I].second, diag::err_mmap_missing_module_qualified) 13345b5d21eaSRichard Smith << Id[I].first 13355b5d21eaSRichard Smith << ActiveModule->getTopLevelModule()->getFullModuleName(); 1336e7ab3669SDouglas Gregor } else { 1337e7ab3669SDouglas Gregor Diags.Report(Id[I].second, diag::err_mmap_expected_module_name); 1338e7ab3669SDouglas Gregor } 1339e7ab3669SDouglas Gregor HadError = true; 1340e7ab3669SDouglas Gregor return; 1341e7ab3669SDouglas Gregor } 1342e7ab3669SDouglas Gregor } 1343e7ab3669SDouglas Gregor 1344e7ab3669SDouglas Gregor StringRef ModuleName = Id.back().first; 1345e7ab3669SDouglas Gregor SourceLocation ModuleNameLoc = Id.back().second; 1346718292f2SDouglas Gregor 1347a686e1b0SDouglas Gregor // Parse the optional attribute list. 13484442605fSBill Wendling Attributes Attrs; 13499194a91dSDouglas Gregor parseOptionalAttributes(Attrs); 1350a686e1b0SDouglas Gregor 1351718292f2SDouglas Gregor // Parse the opening brace. 1352718292f2SDouglas Gregor if (!Tok.is(MMToken::LBrace)) { 1353718292f2SDouglas Gregor Diags.Report(Tok.getLocation(), diag::err_mmap_expected_lbrace) 1354718292f2SDouglas Gregor << ModuleName; 1355718292f2SDouglas Gregor HadError = true; 1356718292f2SDouglas Gregor return; 1357718292f2SDouglas Gregor } 1358718292f2SDouglas Gregor SourceLocation LBraceLoc = consumeToken(); 1359718292f2SDouglas Gregor 1360718292f2SDouglas Gregor // Determine whether this (sub)module has already been defined. 1361eb90e830SDouglas Gregor if (Module *Existing = Map.lookupModuleQualified(ModuleName, ActiveModule)) { 1362fcc54a3bSDouglas Gregor if (Existing->DefinitionLoc.isInvalid() && !ActiveModule) { 1363fcc54a3bSDouglas Gregor // Skip the module definition. 1364fcc54a3bSDouglas Gregor skipUntil(MMToken::RBrace); 1365fcc54a3bSDouglas Gregor if (Tok.is(MMToken::RBrace)) 1366fcc54a3bSDouglas Gregor consumeToken(); 1367fcc54a3bSDouglas Gregor else { 1368fcc54a3bSDouglas Gregor Diags.Report(Tok.getLocation(), diag::err_mmap_expected_rbrace); 1369fcc54a3bSDouglas Gregor Diags.Report(LBraceLoc, diag::note_mmap_lbrace_match); 1370fcc54a3bSDouglas Gregor HadError = true; 1371fcc54a3bSDouglas Gregor } 1372fcc54a3bSDouglas Gregor return; 1373fcc54a3bSDouglas Gregor } 1374fcc54a3bSDouglas Gregor 1375718292f2SDouglas Gregor Diags.Report(ModuleNameLoc, diag::err_mmap_module_redefinition) 1376718292f2SDouglas Gregor << ModuleName; 1377eb90e830SDouglas Gregor Diags.Report(Existing->DefinitionLoc, diag::note_mmap_prev_definition); 1378718292f2SDouglas Gregor 1379718292f2SDouglas Gregor // Skip the module definition. 1380718292f2SDouglas Gregor skipUntil(MMToken::RBrace); 1381718292f2SDouglas Gregor if (Tok.is(MMToken::RBrace)) 1382718292f2SDouglas Gregor consumeToken(); 1383718292f2SDouglas Gregor 1384718292f2SDouglas Gregor HadError = true; 1385718292f2SDouglas Gregor return; 1386718292f2SDouglas Gregor } 1387718292f2SDouglas Gregor 1388beee15e7SBen Langmuir // If this is a submodule, use the parent's module map, since we don't want 1389beee15e7SBen Langmuir // the private module map file. 1390beee15e7SBen Langmuir const FileEntry *ModuleMap = ActiveModule ? ActiveModule->ModuleMap 1391beee15e7SBen Langmuir : ModuleMapFile; 1392beee15e7SBen Langmuir 1393718292f2SDouglas Gregor // Start defining this module. 1394beee15e7SBen Langmuir ActiveModule = Map.findOrCreateModule(ModuleName, ActiveModule, ModuleMap, 1395beee15e7SBen Langmuir Framework, Explicit).first; 1396eb90e830SDouglas Gregor ActiveModule->DefinitionLoc = ModuleNameLoc; 1397963c5535SDouglas Gregor if (Attrs.IsSystem || IsSystem) 1398a686e1b0SDouglas Gregor ActiveModule->IsSystem = true; 139977944868SRichard Smith if (Attrs.IsExternC) 140077944868SRichard Smith ActiveModule->IsExternC = true; 1401718292f2SDouglas Gregor 1402718292f2SDouglas Gregor bool Done = false; 1403718292f2SDouglas Gregor do { 1404718292f2SDouglas Gregor switch (Tok.Kind) { 1405718292f2SDouglas Gregor case MMToken::EndOfFile: 1406718292f2SDouglas Gregor case MMToken::RBrace: 1407718292f2SDouglas Gregor Done = true; 1408718292f2SDouglas Gregor break; 1409718292f2SDouglas Gregor 141035b13eceSDouglas Gregor case MMToken::ConfigMacros: 141135b13eceSDouglas Gregor parseConfigMacros(); 141235b13eceSDouglas Gregor break; 141335b13eceSDouglas Gregor 1414fb912657SDouglas Gregor case MMToken::Conflict: 1415fb912657SDouglas Gregor parseConflict(); 1416fb912657SDouglas Gregor break; 1417fb912657SDouglas Gregor 1418718292f2SDouglas Gregor case MMToken::ExplicitKeyword: 141997292843SDaniel Jasper case MMToken::ExternKeyword: 1420f2161a70SDouglas Gregor case MMToken::FrameworkKeyword: 1421718292f2SDouglas Gregor case MMToken::ModuleKeyword: 1422718292f2SDouglas Gregor parseModuleDecl(); 1423718292f2SDouglas Gregor break; 1424718292f2SDouglas Gregor 14252b82c2a5SDouglas Gregor case MMToken::ExportKeyword: 14262b82c2a5SDouglas Gregor parseExportDecl(); 14272b82c2a5SDouglas Gregor break; 14282b82c2a5SDouglas Gregor 1429ba7f2f71SDaniel Jasper case MMToken::UseKeyword: 1430ba7f2f71SDaniel Jasper parseUseDecl(); 1431ba7f2f71SDaniel Jasper break; 1432ba7f2f71SDaniel Jasper 14331fb5c3a6SDouglas Gregor case MMToken::RequiresKeyword: 14341fb5c3a6SDouglas Gregor parseRequiresDecl(); 14351fb5c3a6SDouglas Gregor break; 14361fb5c3a6SDouglas Gregor 1437524e33e1SDouglas Gregor case MMToken::UmbrellaKeyword: { 1438524e33e1SDouglas Gregor SourceLocation UmbrellaLoc = consumeToken(); 1439524e33e1SDouglas Gregor if (Tok.is(MMToken::HeaderKeyword)) 1440b53e5483SLawrence Crowl parseHeaderDecl(MMToken::UmbrellaKeyword, UmbrellaLoc); 1441524e33e1SDouglas Gregor else 1442524e33e1SDouglas Gregor parseUmbrellaDirDecl(UmbrellaLoc); 1443718292f2SDouglas Gregor break; 1444524e33e1SDouglas Gregor } 1445718292f2SDouglas Gregor 144659527666SDouglas Gregor case MMToken::ExcludeKeyword: { 144759527666SDouglas Gregor SourceLocation ExcludeLoc = consumeToken(); 144859527666SDouglas Gregor if (Tok.is(MMToken::HeaderKeyword)) { 1449b53e5483SLawrence Crowl parseHeaderDecl(MMToken::ExcludeKeyword, ExcludeLoc); 145059527666SDouglas Gregor } else { 145159527666SDouglas Gregor Diags.Report(Tok.getLocation(), diag::err_mmap_expected_header) 145259527666SDouglas Gregor << "exclude"; 145359527666SDouglas Gregor } 145459527666SDouglas Gregor break; 145559527666SDouglas Gregor } 145659527666SDouglas Gregor 1457b53e5483SLawrence Crowl case MMToken::PrivateKeyword: { 1458b53e5483SLawrence Crowl SourceLocation PrivateLoc = consumeToken(); 1459b53e5483SLawrence Crowl if (Tok.is(MMToken::HeaderKeyword)) { 1460b53e5483SLawrence Crowl parseHeaderDecl(MMToken::PrivateKeyword, PrivateLoc); 1461b53e5483SLawrence Crowl } else { 1462b53e5483SLawrence Crowl Diags.Report(Tok.getLocation(), diag::err_mmap_expected_header) 1463b53e5483SLawrence Crowl << "private"; 1464b53e5483SLawrence Crowl } 1465b53e5483SLawrence Crowl break; 1466b53e5483SLawrence Crowl } 1467b53e5483SLawrence Crowl 1468322f633cSDouglas Gregor case MMToken::HeaderKeyword: 1469b53e5483SLawrence Crowl parseHeaderDecl(MMToken::HeaderKeyword, SourceLocation()); 1470718292f2SDouglas Gregor break; 1471718292f2SDouglas Gregor 14726ddfca91SDouglas Gregor case MMToken::LinkKeyword: 14736ddfca91SDouglas Gregor parseLinkDecl(); 14746ddfca91SDouglas Gregor break; 14756ddfca91SDouglas Gregor 1476718292f2SDouglas Gregor default: 1477718292f2SDouglas Gregor Diags.Report(Tok.getLocation(), diag::err_mmap_expected_member); 1478718292f2SDouglas Gregor consumeToken(); 1479718292f2SDouglas Gregor break; 1480718292f2SDouglas Gregor } 1481718292f2SDouglas Gregor } while (!Done); 1482718292f2SDouglas Gregor 1483718292f2SDouglas Gregor if (Tok.is(MMToken::RBrace)) 1484718292f2SDouglas Gregor consumeToken(); 1485718292f2SDouglas Gregor else { 1486718292f2SDouglas Gregor Diags.Report(Tok.getLocation(), diag::err_mmap_expected_rbrace); 1487718292f2SDouglas Gregor Diags.Report(LBraceLoc, diag::note_mmap_lbrace_match); 1488718292f2SDouglas Gregor HadError = true; 1489718292f2SDouglas Gregor } 1490718292f2SDouglas Gregor 149111dfe6feSDouglas Gregor // If the active module is a top-level framework, and there are no link 149211dfe6feSDouglas Gregor // libraries, automatically link against the framework. 149311dfe6feSDouglas Gregor if (ActiveModule->IsFramework && !ActiveModule->isSubFramework() && 149411dfe6feSDouglas Gregor ActiveModule->LinkLibraries.empty()) { 149511dfe6feSDouglas Gregor inferFrameworkLink(ActiveModule, Directory, SourceMgr.getFileManager()); 149611dfe6feSDouglas Gregor } 149711dfe6feSDouglas Gregor 1498ec8c9752SBen Langmuir // If the module meets all requirements but is still unavailable, mark the 1499ec8c9752SBen Langmuir // whole tree as unavailable to prevent it from building. 1500ec8c9752SBen Langmuir if (!ActiveModule->IsAvailable && !ActiveModule->IsMissingRequirement && 1501ec8c9752SBen Langmuir ActiveModule->Parent) { 1502ec8c9752SBen Langmuir ActiveModule->getTopLevelModule()->markUnavailable(); 1503ec8c9752SBen Langmuir ActiveModule->getTopLevelModule()->MissingHeaders.append( 1504ec8c9752SBen Langmuir ActiveModule->MissingHeaders.begin(), ActiveModule->MissingHeaders.end()); 1505ec8c9752SBen Langmuir } 1506ec8c9752SBen Langmuir 1507e7ab3669SDouglas Gregor // We're done parsing this module. Pop back to the previous module. 1508e7ab3669SDouglas Gregor ActiveModule = PreviousActiveModule; 1509718292f2SDouglas Gregor } 1510718292f2SDouglas Gregor 151197292843SDaniel Jasper /// \brief Parse an extern module declaration. 151297292843SDaniel Jasper /// 151397292843SDaniel Jasper /// extern module-declaration: 151497292843SDaniel Jasper /// 'extern' 'module' module-id string-literal 151597292843SDaniel Jasper void ModuleMapParser::parseExternModuleDecl() { 151697292843SDaniel Jasper assert(Tok.is(MMToken::ExternKeyword)); 151797292843SDaniel Jasper consumeToken(); // 'extern' keyword 151897292843SDaniel Jasper 151997292843SDaniel Jasper // Parse 'module' keyword. 152097292843SDaniel Jasper if (!Tok.is(MMToken::ModuleKeyword)) { 152197292843SDaniel Jasper Diags.Report(Tok.getLocation(), diag::err_mmap_expected_module); 152297292843SDaniel Jasper consumeToken(); 152397292843SDaniel Jasper HadError = true; 152497292843SDaniel Jasper return; 152597292843SDaniel Jasper } 152697292843SDaniel Jasper consumeToken(); // 'module' keyword 152797292843SDaniel Jasper 152897292843SDaniel Jasper // Parse the module name. 152997292843SDaniel Jasper ModuleId Id; 153097292843SDaniel Jasper if (parseModuleId(Id)) { 153197292843SDaniel Jasper HadError = true; 153297292843SDaniel Jasper return; 153397292843SDaniel Jasper } 153497292843SDaniel Jasper 153597292843SDaniel Jasper // Parse the referenced module map file name. 153697292843SDaniel Jasper if (!Tok.is(MMToken::StringLiteral)) { 153797292843SDaniel Jasper Diags.Report(Tok.getLocation(), diag::err_mmap_expected_mmap_file); 153897292843SDaniel Jasper HadError = true; 153997292843SDaniel Jasper return; 154097292843SDaniel Jasper } 154197292843SDaniel Jasper std::string FileName = Tok.getString(); 154297292843SDaniel Jasper consumeToken(); // filename 154397292843SDaniel Jasper 154497292843SDaniel Jasper StringRef FileNameRef = FileName; 154597292843SDaniel Jasper SmallString<128> ModuleMapFileName; 154697292843SDaniel Jasper if (llvm::sys::path::is_relative(FileNameRef)) { 154797292843SDaniel Jasper ModuleMapFileName += Directory->getName(); 154897292843SDaniel Jasper llvm::sys::path::append(ModuleMapFileName, FileName); 154997292843SDaniel Jasper FileNameRef = ModuleMapFileName.str(); 155097292843SDaniel Jasper } 155197292843SDaniel Jasper if (const FileEntry *File = SourceMgr.getFileManager().getFile(FileNameRef)) 155297292843SDaniel Jasper Map.parseModuleMapFile(File, /*IsSystem=*/false); 155397292843SDaniel Jasper } 155497292843SDaniel Jasper 15551fb5c3a6SDouglas Gregor /// \brief Parse a requires declaration. 15561fb5c3a6SDouglas Gregor /// 15571fb5c3a6SDouglas Gregor /// requires-declaration: 15581fb5c3a6SDouglas Gregor /// 'requires' feature-list 15591fb5c3a6SDouglas Gregor /// 15601fb5c3a6SDouglas Gregor /// feature-list: 1561a3feee2aSRichard Smith /// feature ',' feature-list 1562a3feee2aSRichard Smith /// feature 1563a3feee2aSRichard Smith /// 1564a3feee2aSRichard Smith /// feature: 1565a3feee2aSRichard Smith /// '!'[opt] identifier 15661fb5c3a6SDouglas Gregor void ModuleMapParser::parseRequiresDecl() { 15671fb5c3a6SDouglas Gregor assert(Tok.is(MMToken::RequiresKeyword)); 15681fb5c3a6SDouglas Gregor 15691fb5c3a6SDouglas Gregor // Parse 'requires' keyword. 15701fb5c3a6SDouglas Gregor consumeToken(); 15711fb5c3a6SDouglas Gregor 15721fb5c3a6SDouglas Gregor // Parse the feature-list. 15731fb5c3a6SDouglas Gregor do { 1574a3feee2aSRichard Smith bool RequiredState = true; 1575a3feee2aSRichard Smith if (Tok.is(MMToken::Exclaim)) { 1576a3feee2aSRichard Smith RequiredState = false; 1577a3feee2aSRichard Smith consumeToken(); 1578a3feee2aSRichard Smith } 1579a3feee2aSRichard Smith 15801fb5c3a6SDouglas Gregor if (!Tok.is(MMToken::Identifier)) { 15811fb5c3a6SDouglas Gregor Diags.Report(Tok.getLocation(), diag::err_mmap_expected_feature); 15821fb5c3a6SDouglas Gregor HadError = true; 15831fb5c3a6SDouglas Gregor return; 15841fb5c3a6SDouglas Gregor } 15851fb5c3a6SDouglas Gregor 15861fb5c3a6SDouglas Gregor // Consume the feature name. 15871fb5c3a6SDouglas Gregor std::string Feature = Tok.getString(); 15881fb5c3a6SDouglas Gregor consumeToken(); 15891fb5c3a6SDouglas Gregor 15901fb5c3a6SDouglas Gregor // Add this feature. 1591a3feee2aSRichard Smith ActiveModule->addRequirement(Feature, RequiredState, 1592a3feee2aSRichard Smith Map.LangOpts, *Map.Target); 15931fb5c3a6SDouglas Gregor 15941fb5c3a6SDouglas Gregor if (!Tok.is(MMToken::Comma)) 15951fb5c3a6SDouglas Gregor break; 15961fb5c3a6SDouglas Gregor 15971fb5c3a6SDouglas Gregor // Consume the comma. 15981fb5c3a6SDouglas Gregor consumeToken(); 15991fb5c3a6SDouglas Gregor } while (true); 16001fb5c3a6SDouglas Gregor } 16011fb5c3a6SDouglas Gregor 1602f2161a70SDouglas Gregor /// \brief Append to \p Paths the set of paths needed to get to the 1603f2161a70SDouglas Gregor /// subframework in which the given module lives. 1604bf8da9d7SBenjamin Kramer static void appendSubframeworkPaths(Module *Mod, 1605f857950dSDmitri Gribenko SmallVectorImpl<char> &Path) { 1606f2161a70SDouglas Gregor // Collect the framework names from the given module to the top-level module. 1607f857950dSDmitri Gribenko SmallVector<StringRef, 2> Paths; 1608f2161a70SDouglas Gregor for (; Mod; Mod = Mod->Parent) { 1609f2161a70SDouglas Gregor if (Mod->IsFramework) 1610f2161a70SDouglas Gregor Paths.push_back(Mod->Name); 1611f2161a70SDouglas Gregor } 1612f2161a70SDouglas Gregor 1613f2161a70SDouglas Gregor if (Paths.empty()) 1614f2161a70SDouglas Gregor return; 1615f2161a70SDouglas Gregor 1616f2161a70SDouglas Gregor // Add Frameworks/Name.framework for each subframework. 161717381a06SBenjamin Kramer for (unsigned I = Paths.size() - 1; I != 0; --I) 161817381a06SBenjamin Kramer llvm::sys::path::append(Path, "Frameworks", Paths[I-1] + ".framework"); 1619f2161a70SDouglas Gregor } 1620f2161a70SDouglas Gregor 1621718292f2SDouglas Gregor /// \brief Parse a header declaration. 1622718292f2SDouglas Gregor /// 1623718292f2SDouglas Gregor /// header-declaration: 1624322f633cSDouglas Gregor /// 'umbrella'[opt] 'header' string-literal 162559527666SDouglas Gregor /// 'exclude'[opt] 'header' string-literal 1626b53e5483SLawrence Crowl void ModuleMapParser::parseHeaderDecl(MMToken::TokenKind LeadingToken, 1627b53e5483SLawrence Crowl SourceLocation LeadingLoc) { 1628718292f2SDouglas Gregor assert(Tok.is(MMToken::HeaderKeyword)); 16291871ed3dSBenjamin Kramer consumeToken(); 1630718292f2SDouglas Gregor 1631718292f2SDouglas Gregor // Parse the header name. 1632718292f2SDouglas Gregor if (!Tok.is(MMToken::StringLiteral)) { 1633718292f2SDouglas Gregor Diags.Report(Tok.getLocation(), diag::err_mmap_expected_header) 1634718292f2SDouglas Gregor << "header"; 1635718292f2SDouglas Gregor HadError = true; 1636718292f2SDouglas Gregor return; 1637718292f2SDouglas Gregor } 16380761a8a0SDaniel Jasper Module::HeaderDirective Header; 16390761a8a0SDaniel Jasper Header.FileName = Tok.getString(); 16400761a8a0SDaniel Jasper Header.FileNameLoc = consumeToken(); 1641718292f2SDouglas Gregor 1642524e33e1SDouglas Gregor // Check whether we already have an umbrella. 1643b53e5483SLawrence Crowl if (LeadingToken == MMToken::UmbrellaKeyword && ActiveModule->Umbrella) { 16440761a8a0SDaniel Jasper Diags.Report(Header.FileNameLoc, diag::err_mmap_umbrella_clash) 1645524e33e1SDouglas Gregor << ActiveModule->getFullModuleName(); 1646322f633cSDouglas Gregor HadError = true; 1647322f633cSDouglas Gregor return; 1648322f633cSDouglas Gregor } 1649322f633cSDouglas Gregor 16505257fc63SDouglas Gregor // Look for this file. 1651e7ab3669SDouglas Gregor const FileEntry *File = 0; 16523ec6663bSDouglas Gregor const FileEntry *BuiltinFile = 0; 16532c1dd271SDylan Noblesmith SmallString<128> PathName; 16540761a8a0SDaniel Jasper if (llvm::sys::path::is_absolute(Header.FileName)) { 16550761a8a0SDaniel Jasper PathName = Header.FileName; 1656e7ab3669SDouglas Gregor File = SourceMgr.getFileManager().getFile(PathName); 1657e7ab3669SDouglas Gregor } else { 1658e7ab3669SDouglas Gregor // Search for the header file within the search directory. 16597033127bSDouglas Gregor PathName = Directory->getName(); 1660e7ab3669SDouglas Gregor unsigned PathLength = PathName.size(); 1661755b2055SDouglas Gregor 1662f2161a70SDouglas Gregor if (ActiveModule->isPartOfFramework()) { 1663f2161a70SDouglas Gregor appendSubframeworkPaths(ActiveModule, PathName); 1664755b2055SDouglas Gregor 1665e7ab3669SDouglas Gregor // Check whether this file is in the public headers. 16660761a8a0SDaniel Jasper llvm::sys::path::append(PathName, "Headers", Header.FileName); 1667e7ab3669SDouglas Gregor File = SourceMgr.getFileManager().getFile(PathName); 1668e7ab3669SDouglas Gregor 1669e7ab3669SDouglas Gregor if (!File) { 1670e7ab3669SDouglas Gregor // Check whether this file is in the private headers. 1671e7ab3669SDouglas Gregor PathName.resize(PathLength); 16720761a8a0SDaniel Jasper llvm::sys::path::append(PathName, "PrivateHeaders", Header.FileName); 1673e7ab3669SDouglas Gregor File = SourceMgr.getFileManager().getFile(PathName); 1674e7ab3669SDouglas Gregor } 1675e7ab3669SDouglas Gregor } else { 1676e7ab3669SDouglas Gregor // Lookup for normal headers. 16770761a8a0SDaniel Jasper llvm::sys::path::append(PathName, Header.FileName); 1678e7ab3669SDouglas Gregor File = SourceMgr.getFileManager().getFile(PathName); 16793ec6663bSDouglas Gregor 16803ec6663bSDouglas Gregor // If this is a system module with a top-level header, this header 16813ec6663bSDouglas Gregor // may have a counterpart (or replacement) in the set of headers 16823ec6663bSDouglas Gregor // supplied by Clang. Find that builtin header. 1683b53e5483SLawrence Crowl if (ActiveModule->IsSystem && LeadingToken != MMToken::UmbrellaKeyword && 1684b53e5483SLawrence Crowl BuiltinIncludeDir && BuiltinIncludeDir != Directory && 16850761a8a0SDaniel Jasper isBuiltinHeader(Header.FileName)) { 16862c1dd271SDylan Noblesmith SmallString<128> BuiltinPathName(BuiltinIncludeDir->getName()); 16870761a8a0SDaniel Jasper llvm::sys::path::append(BuiltinPathName, Header.FileName); 16883ec6663bSDouglas Gregor BuiltinFile = SourceMgr.getFileManager().getFile(BuiltinPathName); 16893ec6663bSDouglas Gregor 16903ec6663bSDouglas Gregor // If Clang supplies this header but the underlying system does not, 16913ec6663bSDouglas Gregor // just silently swap in our builtin version. Otherwise, we'll end 16923ec6663bSDouglas Gregor // up adding both (later). 16933ec6663bSDouglas Gregor if (!File && BuiltinFile) { 16943ec6663bSDouglas Gregor File = BuiltinFile; 16953ec6663bSDouglas Gregor BuiltinFile = 0; 16963ec6663bSDouglas Gregor } 16973ec6663bSDouglas Gregor } 1698e7ab3669SDouglas Gregor } 1699e7ab3669SDouglas Gregor } 17005257fc63SDouglas Gregor 17015257fc63SDouglas Gregor // FIXME: We shouldn't be eagerly stat'ing every file named in a module map. 17025257fc63SDouglas Gregor // Come up with a lazy way to do this. 1703e7ab3669SDouglas Gregor if (File) { 170497da9178SDaniel Jasper if (LeadingToken == MMToken::UmbrellaKeyword) { 1705322f633cSDouglas Gregor const DirectoryEntry *UmbrellaDir = File->getDir(); 170659527666SDouglas Gregor if (Module *UmbrellaModule = Map.UmbrellaDirs[UmbrellaDir]) { 1707b53e5483SLawrence Crowl Diags.Report(LeadingLoc, diag::err_mmap_umbrella_clash) 170859527666SDouglas Gregor << UmbrellaModule->getFullModuleName(); 1709322f633cSDouglas Gregor HadError = true; 17105257fc63SDouglas Gregor } else { 1711322f633cSDouglas Gregor // Record this umbrella header. 1712322f633cSDouglas Gregor Map.setUmbrellaHeader(ActiveModule, File); 1713322f633cSDouglas Gregor } 1714322f633cSDouglas Gregor } else { 1715322f633cSDouglas Gregor // Record this header. 1716b53e5483SLawrence Crowl ModuleMap::ModuleHeaderRole Role = ModuleMap::NormalHeader; 1717b53e5483SLawrence Crowl if (LeadingToken == MMToken::ExcludeKeyword) 1718b53e5483SLawrence Crowl Role = ModuleMap::ExcludedHeader; 1719b53e5483SLawrence Crowl else if (LeadingToken == MMToken::PrivateKeyword) 1720b53e5483SLawrence Crowl Role = ModuleMap::PrivateHeader; 1721b53e5483SLawrence Crowl else 1722b53e5483SLawrence Crowl assert(LeadingToken == MMToken::HeaderKeyword); 1723b53e5483SLawrence Crowl 1724b53e5483SLawrence Crowl Map.addHeader(ActiveModule, File, Role); 17253ec6663bSDouglas Gregor 17263ec6663bSDouglas Gregor // If there is a builtin counterpart to this file, add it now. 17273ec6663bSDouglas Gregor if (BuiltinFile) 1728b53e5483SLawrence Crowl Map.addHeader(ActiveModule, BuiltinFile, Role); 17295257fc63SDouglas Gregor } 1730b53e5483SLawrence Crowl } else if (LeadingToken != MMToken::ExcludeKeyword) { 17314b27a64bSDouglas Gregor // Ignore excluded header files. They're optional anyway. 17324b27a64bSDouglas Gregor 17330761a8a0SDaniel Jasper // If we find a module that has a missing header, we mark this module as 17340761a8a0SDaniel Jasper // unavailable and store the header directive for displaying diagnostics. 17350761a8a0SDaniel Jasper Header.IsUmbrella = LeadingToken == MMToken::UmbrellaKeyword; 1736ec8c9752SBen Langmuir ActiveModule->markUnavailable(); 17370761a8a0SDaniel Jasper ActiveModule->MissingHeaders.push_back(Header); 17385257fc63SDouglas Gregor } 1739718292f2SDouglas Gregor } 1740718292f2SDouglas Gregor 1741524e33e1SDouglas Gregor /// \brief Parse an umbrella directory declaration. 1742524e33e1SDouglas Gregor /// 1743524e33e1SDouglas Gregor /// umbrella-dir-declaration: 1744524e33e1SDouglas Gregor /// umbrella string-literal 1745524e33e1SDouglas Gregor void ModuleMapParser::parseUmbrellaDirDecl(SourceLocation UmbrellaLoc) { 1746524e33e1SDouglas Gregor // Parse the directory name. 1747524e33e1SDouglas Gregor if (!Tok.is(MMToken::StringLiteral)) { 1748524e33e1SDouglas Gregor Diags.Report(Tok.getLocation(), diag::err_mmap_expected_header) 1749524e33e1SDouglas Gregor << "umbrella"; 1750524e33e1SDouglas Gregor HadError = true; 1751524e33e1SDouglas Gregor return; 1752524e33e1SDouglas Gregor } 1753524e33e1SDouglas Gregor 1754524e33e1SDouglas Gregor std::string DirName = Tok.getString(); 1755524e33e1SDouglas Gregor SourceLocation DirNameLoc = consumeToken(); 1756524e33e1SDouglas Gregor 1757524e33e1SDouglas Gregor // Check whether we already have an umbrella. 1758524e33e1SDouglas Gregor if (ActiveModule->Umbrella) { 1759524e33e1SDouglas Gregor Diags.Report(DirNameLoc, diag::err_mmap_umbrella_clash) 1760524e33e1SDouglas Gregor << ActiveModule->getFullModuleName(); 1761524e33e1SDouglas Gregor HadError = true; 1762524e33e1SDouglas Gregor return; 1763524e33e1SDouglas Gregor } 1764524e33e1SDouglas Gregor 1765524e33e1SDouglas Gregor // Look for this file. 1766524e33e1SDouglas Gregor const DirectoryEntry *Dir = 0; 1767524e33e1SDouglas Gregor if (llvm::sys::path::is_absolute(DirName)) 1768524e33e1SDouglas Gregor Dir = SourceMgr.getFileManager().getDirectory(DirName); 1769524e33e1SDouglas Gregor else { 17702c1dd271SDylan Noblesmith SmallString<128> PathName; 1771524e33e1SDouglas Gregor PathName = Directory->getName(); 1772524e33e1SDouglas Gregor llvm::sys::path::append(PathName, DirName); 1773524e33e1SDouglas Gregor Dir = SourceMgr.getFileManager().getDirectory(PathName); 1774524e33e1SDouglas Gregor } 1775524e33e1SDouglas Gregor 1776524e33e1SDouglas Gregor if (!Dir) { 1777524e33e1SDouglas Gregor Diags.Report(DirNameLoc, diag::err_mmap_umbrella_dir_not_found) 1778524e33e1SDouglas Gregor << DirName; 1779524e33e1SDouglas Gregor HadError = true; 1780524e33e1SDouglas Gregor return; 1781524e33e1SDouglas Gregor } 1782524e33e1SDouglas Gregor 1783524e33e1SDouglas Gregor if (Module *OwningModule = Map.UmbrellaDirs[Dir]) { 1784524e33e1SDouglas Gregor Diags.Report(UmbrellaLoc, diag::err_mmap_umbrella_clash) 1785524e33e1SDouglas Gregor << OwningModule->getFullModuleName(); 1786524e33e1SDouglas Gregor HadError = true; 1787524e33e1SDouglas Gregor return; 1788524e33e1SDouglas Gregor } 1789524e33e1SDouglas Gregor 1790524e33e1SDouglas Gregor // Record this umbrella directory. 1791524e33e1SDouglas Gregor Map.setUmbrellaDir(ActiveModule, Dir); 1792524e33e1SDouglas Gregor } 1793524e33e1SDouglas Gregor 17942b82c2a5SDouglas Gregor /// \brief Parse a module export declaration. 17952b82c2a5SDouglas Gregor /// 17962b82c2a5SDouglas Gregor /// export-declaration: 17972b82c2a5SDouglas Gregor /// 'export' wildcard-module-id 17982b82c2a5SDouglas Gregor /// 17992b82c2a5SDouglas Gregor /// wildcard-module-id: 18002b82c2a5SDouglas Gregor /// identifier 18012b82c2a5SDouglas Gregor /// '*' 18022b82c2a5SDouglas Gregor /// identifier '.' wildcard-module-id 18032b82c2a5SDouglas Gregor void ModuleMapParser::parseExportDecl() { 18042b82c2a5SDouglas Gregor assert(Tok.is(MMToken::ExportKeyword)); 18052b82c2a5SDouglas Gregor SourceLocation ExportLoc = consumeToken(); 18062b82c2a5SDouglas Gregor 18072b82c2a5SDouglas Gregor // Parse the module-id with an optional wildcard at the end. 18082b82c2a5SDouglas Gregor ModuleId ParsedModuleId; 18092b82c2a5SDouglas Gregor bool Wildcard = false; 18102b82c2a5SDouglas Gregor do { 18112b82c2a5SDouglas Gregor if (Tok.is(MMToken::Identifier)) { 18122b82c2a5SDouglas Gregor ParsedModuleId.push_back(std::make_pair(Tok.getString(), 18132b82c2a5SDouglas Gregor Tok.getLocation())); 18142b82c2a5SDouglas Gregor consumeToken(); 18152b82c2a5SDouglas Gregor 18162b82c2a5SDouglas Gregor if (Tok.is(MMToken::Period)) { 18172b82c2a5SDouglas Gregor consumeToken(); 18182b82c2a5SDouglas Gregor continue; 18192b82c2a5SDouglas Gregor } 18202b82c2a5SDouglas Gregor 18212b82c2a5SDouglas Gregor break; 18222b82c2a5SDouglas Gregor } 18232b82c2a5SDouglas Gregor 18242b82c2a5SDouglas Gregor if(Tok.is(MMToken::Star)) { 18252b82c2a5SDouglas Gregor Wildcard = true; 1826f5eedd05SDouglas Gregor consumeToken(); 18272b82c2a5SDouglas Gregor break; 18282b82c2a5SDouglas Gregor } 18292b82c2a5SDouglas Gregor 1830ba7f2f71SDaniel Jasper Diags.Report(Tok.getLocation(), diag::err_mmap_module_id); 18312b82c2a5SDouglas Gregor HadError = true; 18322b82c2a5SDouglas Gregor return; 18332b82c2a5SDouglas Gregor } while (true); 18342b82c2a5SDouglas Gregor 18352b82c2a5SDouglas Gregor Module::UnresolvedExportDecl Unresolved = { 18362b82c2a5SDouglas Gregor ExportLoc, ParsedModuleId, Wildcard 18372b82c2a5SDouglas Gregor }; 18382b82c2a5SDouglas Gregor ActiveModule->UnresolvedExports.push_back(Unresolved); 18392b82c2a5SDouglas Gregor } 18402b82c2a5SDouglas Gregor 1841ba7f2f71SDaniel Jasper /// \brief Parse a module uses declaration. 1842ba7f2f71SDaniel Jasper /// 1843ba7f2f71SDaniel Jasper /// uses-declaration: 1844ba7f2f71SDaniel Jasper /// 'uses' wildcard-module-id 1845ba7f2f71SDaniel Jasper void ModuleMapParser::parseUseDecl() { 1846ba7f2f71SDaniel Jasper assert(Tok.is(MMToken::UseKeyword)); 1847ba7f2f71SDaniel Jasper consumeToken(); 1848ba7f2f71SDaniel Jasper // Parse the module-id. 1849ba7f2f71SDaniel Jasper ModuleId ParsedModuleId; 18503cd34c76SDaniel Jasper parseModuleId(ParsedModuleId); 1851ba7f2f71SDaniel Jasper 1852ba7f2f71SDaniel Jasper ActiveModule->UnresolvedDirectUses.push_back(ParsedModuleId); 1853ba7f2f71SDaniel Jasper } 1854ba7f2f71SDaniel Jasper 18556ddfca91SDouglas Gregor /// \brief Parse a link declaration. 18566ddfca91SDouglas Gregor /// 18576ddfca91SDouglas Gregor /// module-declaration: 18586ddfca91SDouglas Gregor /// 'link' 'framework'[opt] string-literal 18596ddfca91SDouglas Gregor void ModuleMapParser::parseLinkDecl() { 18606ddfca91SDouglas Gregor assert(Tok.is(MMToken::LinkKeyword)); 18616ddfca91SDouglas Gregor SourceLocation LinkLoc = consumeToken(); 18626ddfca91SDouglas Gregor 18636ddfca91SDouglas Gregor // Parse the optional 'framework' keyword. 18646ddfca91SDouglas Gregor bool IsFramework = false; 18656ddfca91SDouglas Gregor if (Tok.is(MMToken::FrameworkKeyword)) { 18666ddfca91SDouglas Gregor consumeToken(); 18676ddfca91SDouglas Gregor IsFramework = true; 18686ddfca91SDouglas Gregor } 18696ddfca91SDouglas Gregor 18706ddfca91SDouglas Gregor // Parse the library name 18716ddfca91SDouglas Gregor if (!Tok.is(MMToken::StringLiteral)) { 18726ddfca91SDouglas Gregor Diags.Report(Tok.getLocation(), diag::err_mmap_expected_library_name) 18736ddfca91SDouglas Gregor << IsFramework << SourceRange(LinkLoc); 18746ddfca91SDouglas Gregor HadError = true; 18756ddfca91SDouglas Gregor return; 18766ddfca91SDouglas Gregor } 18776ddfca91SDouglas Gregor 18786ddfca91SDouglas Gregor std::string LibraryName = Tok.getString(); 18796ddfca91SDouglas Gregor consumeToken(); 18806ddfca91SDouglas Gregor ActiveModule->LinkLibraries.push_back(Module::LinkLibrary(LibraryName, 18816ddfca91SDouglas Gregor IsFramework)); 18826ddfca91SDouglas Gregor } 18836ddfca91SDouglas Gregor 188435b13eceSDouglas Gregor /// \brief Parse a configuration macro declaration. 188535b13eceSDouglas Gregor /// 188635b13eceSDouglas Gregor /// module-declaration: 188735b13eceSDouglas Gregor /// 'config_macros' attributes[opt] config-macro-list? 188835b13eceSDouglas Gregor /// 188935b13eceSDouglas Gregor /// config-macro-list: 189035b13eceSDouglas Gregor /// identifier (',' identifier)? 189135b13eceSDouglas Gregor void ModuleMapParser::parseConfigMacros() { 189235b13eceSDouglas Gregor assert(Tok.is(MMToken::ConfigMacros)); 189335b13eceSDouglas Gregor SourceLocation ConfigMacrosLoc = consumeToken(); 189435b13eceSDouglas Gregor 189535b13eceSDouglas Gregor // Only top-level modules can have configuration macros. 189635b13eceSDouglas Gregor if (ActiveModule->Parent) { 189735b13eceSDouglas Gregor Diags.Report(ConfigMacrosLoc, diag::err_mmap_config_macro_submodule); 189835b13eceSDouglas Gregor } 189935b13eceSDouglas Gregor 190035b13eceSDouglas Gregor // Parse the optional attributes. 190135b13eceSDouglas Gregor Attributes Attrs; 190235b13eceSDouglas Gregor parseOptionalAttributes(Attrs); 190335b13eceSDouglas Gregor if (Attrs.IsExhaustive && !ActiveModule->Parent) { 190435b13eceSDouglas Gregor ActiveModule->ConfigMacrosExhaustive = true; 190535b13eceSDouglas Gregor } 190635b13eceSDouglas Gregor 190735b13eceSDouglas Gregor // If we don't have an identifier, we're done. 190835b13eceSDouglas Gregor if (!Tok.is(MMToken::Identifier)) 190935b13eceSDouglas Gregor return; 191035b13eceSDouglas Gregor 191135b13eceSDouglas Gregor // Consume the first identifier. 191235b13eceSDouglas Gregor if (!ActiveModule->Parent) { 191335b13eceSDouglas Gregor ActiveModule->ConfigMacros.push_back(Tok.getString().str()); 191435b13eceSDouglas Gregor } 191535b13eceSDouglas Gregor consumeToken(); 191635b13eceSDouglas Gregor 191735b13eceSDouglas Gregor do { 191835b13eceSDouglas Gregor // If there's a comma, consume it. 191935b13eceSDouglas Gregor if (!Tok.is(MMToken::Comma)) 192035b13eceSDouglas Gregor break; 192135b13eceSDouglas Gregor consumeToken(); 192235b13eceSDouglas Gregor 192335b13eceSDouglas Gregor // We expect to see a macro name here. 192435b13eceSDouglas Gregor if (!Tok.is(MMToken::Identifier)) { 192535b13eceSDouglas Gregor Diags.Report(Tok.getLocation(), diag::err_mmap_expected_config_macro); 192635b13eceSDouglas Gregor break; 192735b13eceSDouglas Gregor } 192835b13eceSDouglas Gregor 192935b13eceSDouglas Gregor // Consume the macro name. 193035b13eceSDouglas Gregor if (!ActiveModule->Parent) { 193135b13eceSDouglas Gregor ActiveModule->ConfigMacros.push_back(Tok.getString().str()); 193235b13eceSDouglas Gregor } 193335b13eceSDouglas Gregor consumeToken(); 193435b13eceSDouglas Gregor } while (true); 193535b13eceSDouglas Gregor } 193635b13eceSDouglas Gregor 1937fb912657SDouglas Gregor /// \brief Format a module-id into a string. 1938fb912657SDouglas Gregor static std::string formatModuleId(const ModuleId &Id) { 1939fb912657SDouglas Gregor std::string result; 1940fb912657SDouglas Gregor { 1941fb912657SDouglas Gregor llvm::raw_string_ostream OS(result); 1942fb912657SDouglas Gregor 1943fb912657SDouglas Gregor for (unsigned I = 0, N = Id.size(); I != N; ++I) { 1944fb912657SDouglas Gregor if (I) 1945fb912657SDouglas Gregor OS << "."; 1946fb912657SDouglas Gregor OS << Id[I].first; 1947fb912657SDouglas Gregor } 1948fb912657SDouglas Gregor } 1949fb912657SDouglas Gregor 1950fb912657SDouglas Gregor return result; 1951fb912657SDouglas Gregor } 1952fb912657SDouglas Gregor 1953fb912657SDouglas Gregor /// \brief Parse a conflict declaration. 1954fb912657SDouglas Gregor /// 1955fb912657SDouglas Gregor /// module-declaration: 1956fb912657SDouglas Gregor /// 'conflict' module-id ',' string-literal 1957fb912657SDouglas Gregor void ModuleMapParser::parseConflict() { 1958fb912657SDouglas Gregor assert(Tok.is(MMToken::Conflict)); 1959fb912657SDouglas Gregor SourceLocation ConflictLoc = consumeToken(); 1960fb912657SDouglas Gregor Module::UnresolvedConflict Conflict; 1961fb912657SDouglas Gregor 1962fb912657SDouglas Gregor // Parse the module-id. 1963fb912657SDouglas Gregor if (parseModuleId(Conflict.Id)) 1964fb912657SDouglas Gregor return; 1965fb912657SDouglas Gregor 1966fb912657SDouglas Gregor // Parse the ','. 1967fb912657SDouglas Gregor if (!Tok.is(MMToken::Comma)) { 1968fb912657SDouglas Gregor Diags.Report(Tok.getLocation(), diag::err_mmap_expected_conflicts_comma) 1969fb912657SDouglas Gregor << SourceRange(ConflictLoc); 1970fb912657SDouglas Gregor return; 1971fb912657SDouglas Gregor } 1972fb912657SDouglas Gregor consumeToken(); 1973fb912657SDouglas Gregor 1974fb912657SDouglas Gregor // Parse the message. 1975fb912657SDouglas Gregor if (!Tok.is(MMToken::StringLiteral)) { 1976fb912657SDouglas Gregor Diags.Report(Tok.getLocation(), diag::err_mmap_expected_conflicts_message) 1977fb912657SDouglas Gregor << formatModuleId(Conflict.Id); 1978fb912657SDouglas Gregor return; 1979fb912657SDouglas Gregor } 1980fb912657SDouglas Gregor Conflict.Message = Tok.getString().str(); 1981fb912657SDouglas Gregor consumeToken(); 1982fb912657SDouglas Gregor 1983fb912657SDouglas Gregor // Add this unresolved conflict. 1984fb912657SDouglas Gregor ActiveModule->UnresolvedConflicts.push_back(Conflict); 1985fb912657SDouglas Gregor } 1986fb912657SDouglas Gregor 19876ddfca91SDouglas Gregor /// \brief Parse an inferred module declaration (wildcard modules). 19889194a91dSDouglas Gregor /// 19899194a91dSDouglas Gregor /// module-declaration: 19909194a91dSDouglas Gregor /// 'explicit'[opt] 'framework'[opt] 'module' * attributes[opt] 19919194a91dSDouglas Gregor /// { inferred-module-member* } 19929194a91dSDouglas Gregor /// 19939194a91dSDouglas Gregor /// inferred-module-member: 19949194a91dSDouglas Gregor /// 'export' '*' 19959194a91dSDouglas Gregor /// 'exclude' identifier 19969194a91dSDouglas Gregor void ModuleMapParser::parseInferredModuleDecl(bool Framework, bool Explicit) { 199773441091SDouglas Gregor assert(Tok.is(MMToken::Star)); 199873441091SDouglas Gregor SourceLocation StarLoc = consumeToken(); 199973441091SDouglas Gregor bool Failed = false; 200073441091SDouglas Gregor 200173441091SDouglas Gregor // Inferred modules must be submodules. 20029194a91dSDouglas Gregor if (!ActiveModule && !Framework) { 200373441091SDouglas Gregor Diags.Report(StarLoc, diag::err_mmap_top_level_inferred_submodule); 200473441091SDouglas Gregor Failed = true; 200573441091SDouglas Gregor } 200673441091SDouglas Gregor 20079194a91dSDouglas Gregor if (ActiveModule) { 2008524e33e1SDouglas Gregor // Inferred modules must have umbrella directories. 20094898cde4SBen Langmuir if (!Failed && ActiveModule->IsAvailable && 20104898cde4SBen Langmuir !ActiveModule->getUmbrellaDir()) { 201173441091SDouglas Gregor Diags.Report(StarLoc, diag::err_mmap_inferred_no_umbrella); 201273441091SDouglas Gregor Failed = true; 201373441091SDouglas Gregor } 201473441091SDouglas Gregor 201573441091SDouglas Gregor // Check for redefinition of an inferred module. 2016dd005f69SDouglas Gregor if (!Failed && ActiveModule->InferSubmodules) { 201773441091SDouglas Gregor Diags.Report(StarLoc, diag::err_mmap_inferred_redef); 2018dd005f69SDouglas Gregor if (ActiveModule->InferredSubmoduleLoc.isValid()) 2019dd005f69SDouglas Gregor Diags.Report(ActiveModule->InferredSubmoduleLoc, 202073441091SDouglas Gregor diag::note_mmap_prev_definition); 202173441091SDouglas Gregor Failed = true; 202273441091SDouglas Gregor } 202373441091SDouglas Gregor 20249194a91dSDouglas Gregor // Check for the 'framework' keyword, which is not permitted here. 20259194a91dSDouglas Gregor if (Framework) { 20269194a91dSDouglas Gregor Diags.Report(StarLoc, diag::err_mmap_inferred_framework_submodule); 20279194a91dSDouglas Gregor Framework = false; 20289194a91dSDouglas Gregor } 20299194a91dSDouglas Gregor } else if (Explicit) { 20309194a91dSDouglas Gregor Diags.Report(StarLoc, diag::err_mmap_explicit_inferred_framework); 20319194a91dSDouglas Gregor Explicit = false; 20329194a91dSDouglas Gregor } 20339194a91dSDouglas Gregor 203473441091SDouglas Gregor // If there were any problems with this inferred submodule, skip its body. 203573441091SDouglas Gregor if (Failed) { 203673441091SDouglas Gregor if (Tok.is(MMToken::LBrace)) { 203773441091SDouglas Gregor consumeToken(); 203873441091SDouglas Gregor skipUntil(MMToken::RBrace); 203973441091SDouglas Gregor if (Tok.is(MMToken::RBrace)) 204073441091SDouglas Gregor consumeToken(); 204173441091SDouglas Gregor } 204273441091SDouglas Gregor HadError = true; 204373441091SDouglas Gregor return; 204473441091SDouglas Gregor } 204573441091SDouglas Gregor 20469194a91dSDouglas Gregor // Parse optional attributes. 20474442605fSBill Wendling Attributes Attrs; 20489194a91dSDouglas Gregor parseOptionalAttributes(Attrs); 20499194a91dSDouglas Gregor 20509194a91dSDouglas Gregor if (ActiveModule) { 205173441091SDouglas Gregor // Note that we have an inferred submodule. 2052dd005f69SDouglas Gregor ActiveModule->InferSubmodules = true; 2053dd005f69SDouglas Gregor ActiveModule->InferredSubmoduleLoc = StarLoc; 2054dd005f69SDouglas Gregor ActiveModule->InferExplicitSubmodules = Explicit; 20559194a91dSDouglas Gregor } else { 20569194a91dSDouglas Gregor // We'll be inferring framework modules for this directory. 20579194a91dSDouglas Gregor Map.InferredDirectories[Directory].InferModules = true; 20589194a91dSDouglas Gregor Map.InferredDirectories[Directory].InferSystemModules = Attrs.IsSystem; 2059beee15e7SBen Langmuir Map.InferredDirectories[Directory].ModuleMapFile = ModuleMapFile; 2060131daca0SRichard Smith // FIXME: Handle the 'framework' keyword. 20619194a91dSDouglas Gregor } 206273441091SDouglas Gregor 206373441091SDouglas Gregor // Parse the opening brace. 206473441091SDouglas Gregor if (!Tok.is(MMToken::LBrace)) { 206573441091SDouglas Gregor Diags.Report(Tok.getLocation(), diag::err_mmap_expected_lbrace_wildcard); 206673441091SDouglas Gregor HadError = true; 206773441091SDouglas Gregor return; 206873441091SDouglas Gregor } 206973441091SDouglas Gregor SourceLocation LBraceLoc = consumeToken(); 207073441091SDouglas Gregor 207173441091SDouglas Gregor // Parse the body of the inferred submodule. 207273441091SDouglas Gregor bool Done = false; 207373441091SDouglas Gregor do { 207473441091SDouglas Gregor switch (Tok.Kind) { 207573441091SDouglas Gregor case MMToken::EndOfFile: 207673441091SDouglas Gregor case MMToken::RBrace: 207773441091SDouglas Gregor Done = true; 207873441091SDouglas Gregor break; 207973441091SDouglas Gregor 20809194a91dSDouglas Gregor case MMToken::ExcludeKeyword: { 20819194a91dSDouglas Gregor if (ActiveModule) { 20829194a91dSDouglas Gregor Diags.Report(Tok.getLocation(), diag::err_mmap_expected_inferred_member) 2083162405daSDouglas Gregor << (ActiveModule != 0); 20849194a91dSDouglas Gregor consumeToken(); 20859194a91dSDouglas Gregor break; 20869194a91dSDouglas Gregor } 20879194a91dSDouglas Gregor 20889194a91dSDouglas Gregor consumeToken(); 20899194a91dSDouglas Gregor if (!Tok.is(MMToken::Identifier)) { 20909194a91dSDouglas Gregor Diags.Report(Tok.getLocation(), diag::err_mmap_missing_exclude_name); 20919194a91dSDouglas Gregor break; 20929194a91dSDouglas Gregor } 20939194a91dSDouglas Gregor 20949194a91dSDouglas Gregor Map.InferredDirectories[Directory].ExcludedModules 20959194a91dSDouglas Gregor .push_back(Tok.getString()); 20969194a91dSDouglas Gregor consumeToken(); 20979194a91dSDouglas Gregor break; 20989194a91dSDouglas Gregor } 20999194a91dSDouglas Gregor 21009194a91dSDouglas Gregor case MMToken::ExportKeyword: 21019194a91dSDouglas Gregor if (!ActiveModule) { 21029194a91dSDouglas Gregor Diags.Report(Tok.getLocation(), diag::err_mmap_expected_inferred_member) 2103162405daSDouglas Gregor << (ActiveModule != 0); 21049194a91dSDouglas Gregor consumeToken(); 21059194a91dSDouglas Gregor break; 21069194a91dSDouglas Gregor } 21079194a91dSDouglas Gregor 210873441091SDouglas Gregor consumeToken(); 210973441091SDouglas Gregor if (Tok.is(MMToken::Star)) 2110dd005f69SDouglas Gregor ActiveModule->InferExportWildcard = true; 211173441091SDouglas Gregor else 211273441091SDouglas Gregor Diags.Report(Tok.getLocation(), 211373441091SDouglas Gregor diag::err_mmap_expected_export_wildcard); 211473441091SDouglas Gregor consumeToken(); 211573441091SDouglas Gregor break; 211673441091SDouglas Gregor 211773441091SDouglas Gregor case MMToken::ExplicitKeyword: 211873441091SDouglas Gregor case MMToken::ModuleKeyword: 211973441091SDouglas Gregor case MMToken::HeaderKeyword: 2120b53e5483SLawrence Crowl case MMToken::PrivateKeyword: 212173441091SDouglas Gregor case MMToken::UmbrellaKeyword: 212273441091SDouglas Gregor default: 21239194a91dSDouglas Gregor Diags.Report(Tok.getLocation(), diag::err_mmap_expected_inferred_member) 2124162405daSDouglas Gregor << (ActiveModule != 0); 212573441091SDouglas Gregor consumeToken(); 212673441091SDouglas Gregor break; 212773441091SDouglas Gregor } 212873441091SDouglas Gregor } while (!Done); 212973441091SDouglas Gregor 213073441091SDouglas Gregor if (Tok.is(MMToken::RBrace)) 213173441091SDouglas Gregor consumeToken(); 213273441091SDouglas Gregor else { 213373441091SDouglas Gregor Diags.Report(Tok.getLocation(), diag::err_mmap_expected_rbrace); 213473441091SDouglas Gregor Diags.Report(LBraceLoc, diag::note_mmap_lbrace_match); 213573441091SDouglas Gregor HadError = true; 213673441091SDouglas Gregor } 213773441091SDouglas Gregor } 213873441091SDouglas Gregor 21399194a91dSDouglas Gregor /// \brief Parse optional attributes. 21409194a91dSDouglas Gregor /// 21419194a91dSDouglas Gregor /// attributes: 21429194a91dSDouglas Gregor /// attribute attributes 21439194a91dSDouglas Gregor /// attribute 21449194a91dSDouglas Gregor /// 21459194a91dSDouglas Gregor /// attribute: 21469194a91dSDouglas Gregor /// [ identifier ] 21479194a91dSDouglas Gregor /// 21489194a91dSDouglas Gregor /// \param Attrs Will be filled in with the parsed attributes. 21499194a91dSDouglas Gregor /// 21509194a91dSDouglas Gregor /// \returns true if an error occurred, false otherwise. 21514442605fSBill Wendling bool ModuleMapParser::parseOptionalAttributes(Attributes &Attrs) { 21529194a91dSDouglas Gregor bool HadError = false; 21539194a91dSDouglas Gregor 21549194a91dSDouglas Gregor while (Tok.is(MMToken::LSquare)) { 21559194a91dSDouglas Gregor // Consume the '['. 21569194a91dSDouglas Gregor SourceLocation LSquareLoc = consumeToken(); 21579194a91dSDouglas Gregor 21589194a91dSDouglas Gregor // Check whether we have an attribute name here. 21599194a91dSDouglas Gregor if (!Tok.is(MMToken::Identifier)) { 21609194a91dSDouglas Gregor Diags.Report(Tok.getLocation(), diag::err_mmap_expected_attribute); 21619194a91dSDouglas Gregor skipUntil(MMToken::RSquare); 21629194a91dSDouglas Gregor if (Tok.is(MMToken::RSquare)) 21639194a91dSDouglas Gregor consumeToken(); 21649194a91dSDouglas Gregor HadError = true; 21659194a91dSDouglas Gregor } 21669194a91dSDouglas Gregor 21679194a91dSDouglas Gregor // Decode the attribute name. 21689194a91dSDouglas Gregor AttributeKind Attribute 21699194a91dSDouglas Gregor = llvm::StringSwitch<AttributeKind>(Tok.getString()) 217035b13eceSDouglas Gregor .Case("exhaustive", AT_exhaustive) 217177944868SRichard Smith .Case("extern_c", AT_extern_c) 21729194a91dSDouglas Gregor .Case("system", AT_system) 21739194a91dSDouglas Gregor .Default(AT_unknown); 21749194a91dSDouglas Gregor switch (Attribute) { 21759194a91dSDouglas Gregor case AT_unknown: 21769194a91dSDouglas Gregor Diags.Report(Tok.getLocation(), diag::warn_mmap_unknown_attribute) 21779194a91dSDouglas Gregor << Tok.getString(); 21789194a91dSDouglas Gregor break; 21799194a91dSDouglas Gregor 21809194a91dSDouglas Gregor case AT_system: 21819194a91dSDouglas Gregor Attrs.IsSystem = true; 21829194a91dSDouglas Gregor break; 218335b13eceSDouglas Gregor 218477944868SRichard Smith case AT_extern_c: 218577944868SRichard Smith Attrs.IsExternC = true; 218677944868SRichard Smith break; 218777944868SRichard Smith 218835b13eceSDouglas Gregor case AT_exhaustive: 218935b13eceSDouglas Gregor Attrs.IsExhaustive = true; 219035b13eceSDouglas Gregor break; 21919194a91dSDouglas Gregor } 21929194a91dSDouglas Gregor consumeToken(); 21939194a91dSDouglas Gregor 21949194a91dSDouglas Gregor // Consume the ']'. 21959194a91dSDouglas Gregor if (!Tok.is(MMToken::RSquare)) { 21969194a91dSDouglas Gregor Diags.Report(Tok.getLocation(), diag::err_mmap_expected_rsquare); 21979194a91dSDouglas Gregor Diags.Report(LSquareLoc, diag::note_mmap_lsquare_match); 21989194a91dSDouglas Gregor skipUntil(MMToken::RSquare); 21999194a91dSDouglas Gregor HadError = true; 22009194a91dSDouglas Gregor } 22019194a91dSDouglas Gregor 22029194a91dSDouglas Gregor if (Tok.is(MMToken::RSquare)) 22039194a91dSDouglas Gregor consumeToken(); 22049194a91dSDouglas Gregor } 22059194a91dSDouglas Gregor 22069194a91dSDouglas Gregor return HadError; 22079194a91dSDouglas Gregor } 22089194a91dSDouglas Gregor 2209718292f2SDouglas Gregor /// \brief Parse a module map file. 2210718292f2SDouglas Gregor /// 2211718292f2SDouglas Gregor /// module-map-file: 2212718292f2SDouglas Gregor /// module-declaration* 2213718292f2SDouglas Gregor bool ModuleMapParser::parseModuleMapFile() { 2214718292f2SDouglas Gregor do { 2215718292f2SDouglas Gregor switch (Tok.Kind) { 2216718292f2SDouglas Gregor case MMToken::EndOfFile: 2217718292f2SDouglas Gregor return HadError; 2218718292f2SDouglas Gregor 2219e7ab3669SDouglas Gregor case MMToken::ExplicitKeyword: 222097292843SDaniel Jasper case MMToken::ExternKeyword: 2221718292f2SDouglas Gregor case MMToken::ModuleKeyword: 2222755b2055SDouglas Gregor case MMToken::FrameworkKeyword: 2223718292f2SDouglas Gregor parseModuleDecl(); 2224718292f2SDouglas Gregor break; 2225718292f2SDouglas Gregor 22261fb5c3a6SDouglas Gregor case MMToken::Comma: 222735b13eceSDouglas Gregor case MMToken::ConfigMacros: 2228fb912657SDouglas Gregor case MMToken::Conflict: 2229a3feee2aSRichard Smith case MMToken::Exclaim: 223059527666SDouglas Gregor case MMToken::ExcludeKeyword: 22312b82c2a5SDouglas Gregor case MMToken::ExportKeyword: 2232718292f2SDouglas Gregor case MMToken::HeaderKeyword: 2233718292f2SDouglas Gregor case MMToken::Identifier: 2234718292f2SDouglas Gregor case MMToken::LBrace: 22356ddfca91SDouglas Gregor case MMToken::LinkKeyword: 2236a686e1b0SDouglas Gregor case MMToken::LSquare: 22372b82c2a5SDouglas Gregor case MMToken::Period: 2238b53e5483SLawrence Crowl case MMToken::PrivateKeyword: 2239718292f2SDouglas Gregor case MMToken::RBrace: 2240a686e1b0SDouglas Gregor case MMToken::RSquare: 22411fb5c3a6SDouglas Gregor case MMToken::RequiresKeyword: 22422b82c2a5SDouglas Gregor case MMToken::Star: 2243718292f2SDouglas Gregor case MMToken::StringLiteral: 2244718292f2SDouglas Gregor case MMToken::UmbrellaKeyword: 2245ba7f2f71SDaniel Jasper case MMToken::UseKeyword: 2246718292f2SDouglas Gregor Diags.Report(Tok.getLocation(), diag::err_mmap_expected_module); 2247718292f2SDouglas Gregor HadError = true; 2248718292f2SDouglas Gregor consumeToken(); 2249718292f2SDouglas Gregor break; 2250718292f2SDouglas Gregor } 2251718292f2SDouglas Gregor } while (true); 2252718292f2SDouglas Gregor } 2253718292f2SDouglas Gregor 2254963c5535SDouglas Gregor bool ModuleMap::parseModuleMapFile(const FileEntry *File, bool IsSystem) { 22554ddf2221SDouglas Gregor llvm::DenseMap<const FileEntry *, bool>::iterator Known 22564ddf2221SDouglas Gregor = ParsedModuleMap.find(File); 22574ddf2221SDouglas Gregor if (Known != ParsedModuleMap.end()) 22584ddf2221SDouglas Gregor return Known->second; 22594ddf2221SDouglas Gregor 226089929282SDouglas Gregor assert(Target != 0 && "Missing target information"); 2261cb69b57bSBen Langmuir auto FileCharacter = IsSystem ? SrcMgr::C_System : SrcMgr::C_User; 2262cb69b57bSBen Langmuir FileID ID = SourceMgr.createFileID(File, SourceLocation(), FileCharacter); 22631f76c4e8SManuel Klimek const llvm::MemoryBuffer *Buffer = SourceMgr.getBuffer(ID); 2264718292f2SDouglas Gregor if (!Buffer) 22654ddf2221SDouglas Gregor return ParsedModuleMap[File] = true; 2266718292f2SDouglas Gregor 2267984e1df7SBen Langmuir // Find the directory for the module. For frameworks, that may require going 2268984e1df7SBen Langmuir // up from the 'Modules' directory. 2269984e1df7SBen Langmuir const DirectoryEntry *Dir = File->getDir(); 2270984e1df7SBen Langmuir StringRef DirName(Dir->getName()); 2271984e1df7SBen Langmuir if (llvm::sys::path::filename(DirName) == "Modules") { 2272984e1df7SBen Langmuir DirName = llvm::sys::path::parent_path(DirName); 2273984e1df7SBen Langmuir if (DirName.endswith(".framework")) 2274984e1df7SBen Langmuir Dir = SourceMgr.getFileManager().getDirectory(DirName); 2275984e1df7SBen Langmuir assert(Dir && "parent must exist"); 2276984e1df7SBen Langmuir } 2277984e1df7SBen Langmuir 2278718292f2SDouglas Gregor // Parse this module map file. 22791f76c4e8SManuel Klimek Lexer L(ID, SourceMgr.getBuffer(ID), SourceMgr, MMapLangOpts); 2280beee15e7SBen Langmuir ModuleMapParser Parser(L, SourceMgr, Target, Diags, *this, File, Dir, 2281963c5535SDouglas Gregor BuiltinIncludeDir, IsSystem); 2282718292f2SDouglas Gregor bool Result = Parser.parseModuleMapFile(); 22834ddf2221SDouglas Gregor ParsedModuleMap[File] = Result; 2284718292f2SDouglas Gregor return Result; 2285718292f2SDouglas Gregor } 2286