]> CyberLeo.Net >> Repos - FreeBSD/FreeBSD.git/blob - contrib/llvm/tools/clang/lib/Frontend/InitHeaderSearch.cpp
MFV r319742: 8056 zfs send size estimate is inaccurate for some zvols
[FreeBSD/FreeBSD.git] / contrib / llvm / tools / clang / lib / Frontend / InitHeaderSearch.cpp
1 //===--- InitHeaderSearch.cpp - Initialize header search paths ------------===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 //
10 // This file implements the InitHeaderSearch class.
11 //
12 //===----------------------------------------------------------------------===//
13
14 #include "clang/Basic/FileManager.h"
15 #include "clang/Basic/LangOptions.h"
16 #include "clang/Config/config.h" // C_INCLUDE_DIRS
17 #include "clang/Frontend/Utils.h"
18 #include "clang/Lex/HeaderMap.h"
19 #include "clang/Lex/HeaderSearch.h"
20 #include "clang/Lex/HeaderSearchOptions.h"
21 #include "llvm/ADT/SmallPtrSet.h"
22 #include "llvm/ADT/SmallString.h"
23 #include "llvm/ADT/SmallVector.h"
24 #include "llvm/ADT/StringExtras.h"
25 #include "llvm/ADT/Triple.h"
26 #include "llvm/ADT/Twine.h"
27 #include "llvm/Support/ErrorHandling.h"
28 #include "llvm/Support/Path.h"
29 #include "llvm/Support/raw_ostream.h"
30
31 using namespace clang;
32 using namespace clang::frontend;
33
34 namespace {
35
36 /// InitHeaderSearch - This class makes it easier to set the search paths of
37 ///  a HeaderSearch object. InitHeaderSearch stores several search path lists
38 ///  internally, which can be sent to a HeaderSearch object in one swoop.
39 class InitHeaderSearch {
40   std::vector<std::pair<IncludeDirGroup, DirectoryLookup> > IncludePath;
41   typedef std::vector<std::pair<IncludeDirGroup,
42                       DirectoryLookup> >::const_iterator path_iterator;
43   std::vector<std::pair<std::string, bool> > SystemHeaderPrefixes;
44   HeaderSearch &Headers;
45   bool Verbose;
46   std::string IncludeSysroot;
47   bool HasSysroot;
48
49 public:
50
51   InitHeaderSearch(HeaderSearch &HS, bool verbose, StringRef sysroot)
52     : Headers(HS), Verbose(verbose), IncludeSysroot(sysroot),
53       HasSysroot(!(sysroot.empty() || sysroot == "/")) {
54   }
55
56   /// AddPath - Add the specified path to the specified group list, prefixing
57   /// the sysroot if used.
58   void AddPath(const Twine &Path, IncludeDirGroup Group, bool isFramework);
59
60   /// AddUnmappedPath - Add the specified path to the specified group list,
61   /// without performing any sysroot remapping.
62   void AddUnmappedPath(const Twine &Path, IncludeDirGroup Group,
63                        bool isFramework);
64
65   /// AddSystemHeaderPrefix - Add the specified prefix to the system header
66   /// prefix list.
67   void AddSystemHeaderPrefix(StringRef Prefix, bool IsSystemHeader) {
68     SystemHeaderPrefixes.emplace_back(Prefix, IsSystemHeader);
69   }
70
71   /// AddGnuCPlusPlusIncludePaths - Add the necessary paths to support a gnu
72   ///  libstdc++.
73   void AddGnuCPlusPlusIncludePaths(StringRef Base,
74                                    StringRef ArchDir,
75                                    StringRef Dir32,
76                                    StringRef Dir64,
77                                    const llvm::Triple &triple);
78
79   /// AddMinGWCPlusPlusIncludePaths - Add the necessary paths to support a MinGW
80   ///  libstdc++.
81   void AddMinGWCPlusPlusIncludePaths(StringRef Base,
82                                      StringRef Arch,
83                                      StringRef Version);
84
85   // AddDefaultCIncludePaths - Add paths that should always be searched.
86   void AddDefaultCIncludePaths(const llvm::Triple &triple,
87                                const HeaderSearchOptions &HSOpts);
88
89   // AddDefaultCPlusPlusIncludePaths -  Add paths that should be searched when
90   //  compiling c++.
91   void AddDefaultCPlusPlusIncludePaths(const llvm::Triple &triple,
92                                        const HeaderSearchOptions &HSOpts);
93
94   /// AddDefaultSystemIncludePaths - Adds the default system include paths so
95   ///  that e.g. stdio.h is found.
96   void AddDefaultIncludePaths(const LangOptions &Lang,
97                               const llvm::Triple &triple,
98                               const HeaderSearchOptions &HSOpts);
99
100   /// Realize - Merges all search path lists into one list and send it to
101   /// HeaderSearch.
102   void Realize(const LangOptions &Lang);
103 };
104
105 }  // end anonymous namespace.
106
107 static bool CanPrefixSysroot(StringRef Path) {
108 #if defined(LLVM_ON_WIN32)
109   return !Path.empty() && llvm::sys::path::is_separator(Path[0]);
110 #else
111   return llvm::sys::path::is_absolute(Path);
112 #endif
113 }
114
115 void InitHeaderSearch::AddPath(const Twine &Path, IncludeDirGroup Group,
116                                bool isFramework) {
117   // Add the path with sysroot prepended, if desired and this is a system header
118   // group.
119   if (HasSysroot) {
120     SmallString<256> MappedPathStorage;
121     StringRef MappedPathStr = Path.toStringRef(MappedPathStorage);
122     if (CanPrefixSysroot(MappedPathStr)) {
123       AddUnmappedPath(IncludeSysroot + Path, Group, isFramework);
124       return;
125     }
126   }
127
128   AddUnmappedPath(Path, Group, isFramework);
129 }
130
131 void InitHeaderSearch::AddUnmappedPath(const Twine &Path, IncludeDirGroup Group,
132                                        bool isFramework) {
133   assert(!Path.isTriviallyEmpty() && "can't handle empty path here");
134
135   FileManager &FM = Headers.getFileMgr();
136   SmallString<256> MappedPathStorage;
137   StringRef MappedPathStr = Path.toStringRef(MappedPathStorage);
138
139   // Compute the DirectoryLookup type.
140   SrcMgr::CharacteristicKind Type;
141   if (Group == Quoted || Group == Angled || Group == IndexHeaderMap) {
142     Type = SrcMgr::C_User;
143   } else if (Group == ExternCSystem) {
144     Type = SrcMgr::C_ExternCSystem;
145   } else {
146     Type = SrcMgr::C_System;
147   }
148
149   // If the directory exists, add it.
150   if (const DirectoryEntry *DE = FM.getDirectory(MappedPathStr)) {
151     IncludePath.push_back(
152       std::make_pair(Group, DirectoryLookup(DE, Type, isFramework)));
153     return;
154   }
155
156   // Check to see if this is an apple-style headermap (which are not allowed to
157   // be frameworks).
158   if (!isFramework) {
159     if (const FileEntry *FE = FM.getFile(MappedPathStr)) {
160       if (const HeaderMap *HM = Headers.CreateHeaderMap(FE)) {
161         // It is a headermap, add it to the search path.
162         IncludePath.push_back(
163           std::make_pair(Group,
164                          DirectoryLookup(HM, Type, Group == IndexHeaderMap)));
165         return;
166       }
167     }
168   }
169
170   if (Verbose)
171     llvm::errs() << "ignoring nonexistent directory \""
172                  << MappedPathStr << "\"\n";
173 }
174
175 void InitHeaderSearch::AddGnuCPlusPlusIncludePaths(StringRef Base,
176                                                    StringRef ArchDir,
177                                                    StringRef Dir32,
178                                                    StringRef Dir64,
179                                                    const llvm::Triple &triple) {
180   // Add the base dir
181   AddPath(Base, CXXSystem, false);
182
183   // Add the multilib dirs
184   llvm::Triple::ArchType arch = triple.getArch();
185   bool is64bit = arch == llvm::Triple::ppc64 || arch == llvm::Triple::x86_64;
186   if (is64bit)
187     AddPath(Base + "/" + ArchDir + "/" + Dir64, CXXSystem, false);
188   else
189     AddPath(Base + "/" + ArchDir + "/" + Dir32, CXXSystem, false);
190
191   // Add the backward dir
192   AddPath(Base + "/backward", CXXSystem, false);
193 }
194
195 void InitHeaderSearch::AddMinGWCPlusPlusIncludePaths(StringRef Base,
196                                                      StringRef Arch,
197                                                      StringRef Version) {
198   AddPath(Base + "/" + Arch + "/" + Version + "/include/c++",
199           CXXSystem, false);
200   AddPath(Base + "/" + Arch + "/" + Version + "/include/c++/" + Arch,
201           CXXSystem, false);
202   AddPath(Base + "/" + Arch + "/" + Version + "/include/c++/backward",
203           CXXSystem, false);
204 }
205
206 void InitHeaderSearch::AddDefaultCIncludePaths(const llvm::Triple &triple,
207                                             const HeaderSearchOptions &HSOpts) {
208   llvm::Triple::OSType os = triple.getOS();
209
210   if (HSOpts.UseStandardSystemIncludes) {
211     switch (os) {
212     case llvm::Triple::CloudABI:
213     case llvm::Triple::FreeBSD:
214     case llvm::Triple::NetBSD:
215     case llvm::Triple::OpenBSD:
216     case llvm::Triple::Bitrig:
217     case llvm::Triple::NaCl:
218     case llvm::Triple::PS4:
219     case llvm::Triple::ELFIAMCU:
220       break;
221     case llvm::Triple::Win32:
222       if (triple.getEnvironment() != llvm::Triple::Cygnus)
223         break;
224     default:
225       // FIXME: temporary hack: hard-coded paths.
226       AddPath("/usr/local/include", System, false);
227       break;
228     }
229   }
230
231   // Builtin includes use #include_next directives and should be positioned
232   // just prior C include dirs.
233   if (HSOpts.UseBuiltinIncludes) {
234     // Ignore the sys root, we *always* look for clang headers relative to
235     // supplied path.
236     SmallString<128> P = StringRef(HSOpts.ResourceDir);
237     llvm::sys::path::append(P, "include");
238     AddUnmappedPath(P, ExternCSystem, false);
239   }
240
241   // All remaining additions are for system include directories, early exit if
242   // we aren't using them.
243   if (!HSOpts.UseStandardSystemIncludes)
244     return;
245
246   // Add dirs specified via 'configure --with-c-include-dirs'.
247   StringRef CIncludeDirs(C_INCLUDE_DIRS);
248   if (CIncludeDirs != "") {
249     SmallVector<StringRef, 5> dirs;
250     CIncludeDirs.split(dirs, ":");
251     for (StringRef dir : dirs)
252       AddPath(dir, ExternCSystem, false);
253     return;
254   }
255
256   switch (os) {
257   case llvm::Triple::Linux:
258     llvm_unreachable("Include management is handled in the driver.");
259
260   case llvm::Triple::CloudABI: {
261     // <sysroot>/<triple>/include
262     SmallString<128> P = StringRef(HSOpts.ResourceDir);
263     llvm::sys::path::append(P, "../../..", triple.str(), "include");
264     AddPath(P, System, false);
265     break;
266   }
267
268   case llvm::Triple::Haiku:
269     AddPath("/boot/system/non-packaged/develop/headers", System, false);
270     AddPath("/boot/system/develop/headers/os", System, false);
271     AddPath("/boot/system/develop/headers/os/app", System, false);
272     AddPath("/boot/system/develop/headers/os/arch", System, false);
273     AddPath("/boot/system/develop/headers/os/device", System, false);
274     AddPath("/boot/system/develop/headers/os/drivers", System, false);
275     AddPath("/boot/system/develop/headers/os/game", System, false);
276     AddPath("/boot/system/develop/headers/os/interface", System, false);
277     AddPath("/boot/system/develop/headers/os/kernel", System, false);
278     AddPath("/boot/system/develop/headers/os/locale", System, false);
279     AddPath("/boot/system/develop/headers/os/mail", System, false);
280     AddPath("/boot/system/develop/headers/os/media", System, false);
281     AddPath("/boot/system/develop/headers/os/midi", System, false);
282     AddPath("/boot/system/develop/headers/os/midi2", System, false);
283     AddPath("/boot/system/develop/headers/os/net", System, false);
284     AddPath("/boot/system/develop/headers/os/opengl", System, false);
285     AddPath("/boot/system/develop/headers/os/storage", System, false);
286     AddPath("/boot/system/develop/headers/os/support", System, false);
287     AddPath("/boot/system/develop/headers/os/translation", System, false);
288     AddPath("/boot/system/develop/headers/os/add-ons/graphics", System, false);
289     AddPath("/boot/system/develop/headers/os/add-ons/input_server", System, false);
290     AddPath("/boot/system/develop/headers/os/add-ons/mail_daemon", System, false);
291     AddPath("/boot/system/develop/headers/os/add-ons/registrar", System, false);
292     AddPath("/boot/system/develop/headers/os/add-ons/screen_saver", System, false);
293     AddPath("/boot/system/develop/headers/os/add-ons/tracker", System, false);
294     AddPath("/boot/system/develop/headers/os/be_apps/Deskbar", System, false);
295     AddPath("/boot/system/develop/headers/os/be_apps/NetPositive", System, false);
296     AddPath("/boot/system/develop/headers/os/be_apps/Tracker", System, false);
297     AddPath("/boot/system/develop/headers/3rdparty", System, false);
298     AddPath("/boot/system/develop/headers/bsd", System, false);
299     AddPath("/boot/system/develop/headers/glibc", System, false);
300     AddPath("/boot/system/develop/headers/posix", System, false);
301     AddPath("/boot/system/develop/headers",  System, false);
302     break;
303   case llvm::Triple::RTEMS:
304     break;
305   case llvm::Triple::Win32:
306     switch (triple.getEnvironment()) {
307     default: llvm_unreachable("Include management is handled in the driver.");
308     case llvm::Triple::Cygnus:
309       AddPath("/usr/include/w32api", System, false);
310       break;
311     case llvm::Triple::GNU:
312       break;
313     }
314     break;
315   default:
316     break;
317   }
318
319   switch (os) {
320   case llvm::Triple::CloudABI:
321   case llvm::Triple::RTEMS:
322   case llvm::Triple::NaCl:
323   case llvm::Triple::ELFIAMCU:
324     break;
325   case llvm::Triple::PS4: {
326     // <isysroot> gets prepended later in AddPath().
327     std::string BaseSDKPath = "";
328     if (!HasSysroot) {
329       const char *envValue = getenv("SCE_ORBIS_SDK_DIR");
330       if (envValue)
331         BaseSDKPath = envValue;
332       else {
333         // HSOpts.ResourceDir variable contains the location of Clang's
334         // resource files.
335         // Assuming that Clang is configured for PS4 without
336         // --with-clang-resource-dir option, the location of Clang's resource
337         // files is <SDK_DIR>/host_tools/lib/clang
338         SmallString<128> P = StringRef(HSOpts.ResourceDir);
339         llvm::sys::path::append(P, "../../..");
340         BaseSDKPath = P.str();
341       }
342     }
343     AddPath(BaseSDKPath + "/target/include", System, false);
344     if (triple.isPS4CPU())
345       AddPath(BaseSDKPath + "/target/include_common", System, false);
346   }
347   default:
348     AddPath("/usr/include", ExternCSystem, false);
349     break;
350   }
351 }
352
353 void InitHeaderSearch::
354 AddDefaultCPlusPlusIncludePaths(const llvm::Triple &triple, const HeaderSearchOptions &HSOpts) {
355   llvm::Triple::OSType os = triple.getOS();
356   // FIXME: temporary hack: hard-coded paths.
357
358   if (triple.isOSDarwin()) {
359     switch (triple.getArch()) {
360     default: break;
361
362     case llvm::Triple::ppc:
363     case llvm::Triple::ppc64:
364       AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.2.1",
365                                   "powerpc-apple-darwin10", "", "ppc64",
366                                   triple);
367       AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.0.0",
368                                   "powerpc-apple-darwin10", "", "ppc64",
369                                   triple);
370       break;
371
372     case llvm::Triple::x86:
373     case llvm::Triple::x86_64:
374       AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.2.1",
375                                   "i686-apple-darwin10", "", "x86_64", triple);
376       AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.0.0",
377                                   "i686-apple-darwin8", "", "", triple);
378       break;
379
380     case llvm::Triple::arm:
381     case llvm::Triple::thumb:
382       AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.2.1",
383                                   "arm-apple-darwin10", "v7", "", triple);
384       AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.2.1",
385                                   "arm-apple-darwin10", "v6", "", triple);
386       break;
387
388     case llvm::Triple::aarch64:
389       AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.2.1",
390                                   "arm64-apple-darwin10", "", "", triple);
391       break;
392     }
393     return;
394   }
395
396   switch (os) {
397   case llvm::Triple::Linux:
398     llvm_unreachable("Include management is handled in the driver.");
399     break;
400   case llvm::Triple::Win32:
401     switch (triple.getEnvironment()) {
402     default: llvm_unreachable("Include management is handled in the driver.");
403     case llvm::Triple::Cygnus:
404       // Cygwin-1.7
405       AddMinGWCPlusPlusIncludePaths("/usr/lib/gcc", "i686-pc-cygwin", "4.7.3");
406       AddMinGWCPlusPlusIncludePaths("/usr/lib/gcc", "i686-pc-cygwin", "4.5.3");
407       AddMinGWCPlusPlusIncludePaths("/usr/lib/gcc", "i686-pc-cygwin", "4.3.4");
408       // g++-4 / Cygwin-1.5
409       AddMinGWCPlusPlusIncludePaths("/usr/lib/gcc", "i686-pc-cygwin", "4.3.2");
410       break;
411     }
412     break;
413   case llvm::Triple::DragonFly:
414     AddPath("/usr/include/c++/5.0", CXXSystem, false);
415     break;
416   case llvm::Triple::OpenBSD: {
417     std::string t = triple.getTriple();
418     if (t.substr(0, 6) == "x86_64")
419       t.replace(0, 6, "amd64");
420     AddGnuCPlusPlusIncludePaths("/usr/include/g++",
421                                 t, "", "", triple);
422     break;
423   }
424   case llvm::Triple::Minix:
425     AddGnuCPlusPlusIncludePaths("/usr/gnu/include/c++/4.4.3",
426                                 "", "", "", triple);
427     break;
428   default:
429     break;
430   }
431 }
432
433 void InitHeaderSearch::AddDefaultIncludePaths(const LangOptions &Lang,
434                                               const llvm::Triple &triple,
435                                             const HeaderSearchOptions &HSOpts) {
436   // NB: This code path is going away. All of the logic is moving into the
437   // driver which has the information necessary to do target-specific
438   // selections of default include paths. Each target which moves there will be
439   // exempted from this logic here until we can delete the entire pile of code.
440   switch (triple.getOS()) {
441   default:
442     break; // Everything else continues to use this routine's logic.
443
444   case llvm::Triple::Linux:
445     return;
446
447   case llvm::Triple::Win32:
448     if (triple.getEnvironment() != llvm::Triple::Cygnus ||
449         triple.isOSBinFormatMachO())
450       return;
451     break;
452   }
453
454   if (Lang.CPlusPlus && HSOpts.UseStandardCXXIncludes &&
455       HSOpts.UseStandardSystemIncludes) {
456     if (HSOpts.UseLibcxx) {
457       if (triple.isOSDarwin()) {
458         // On Darwin, libc++ may be installed alongside the compiler in
459         // include/c++/v1.
460         if (!HSOpts.ResourceDir.empty()) {
461           // Remove version from foo/lib/clang/version
462           StringRef NoVer = llvm::sys::path::parent_path(HSOpts.ResourceDir);
463           // Remove clang from foo/lib/clang
464           StringRef Lib = llvm::sys::path::parent_path(NoVer);
465           // Remove lib from foo/lib
466           SmallString<128> P = llvm::sys::path::parent_path(Lib);
467
468           // Get foo/include/c++/v1
469           llvm::sys::path::append(P, "include", "c++", "v1");
470           AddUnmappedPath(P, CXXSystem, false);
471         }
472       }
473       AddPath("/usr/include/c++/v1", CXXSystem, false);
474     } else {
475       AddDefaultCPlusPlusIncludePaths(triple, HSOpts);
476     }
477   }
478
479   AddDefaultCIncludePaths(triple, HSOpts);
480
481   // Add the default framework include paths on Darwin.
482   if (HSOpts.UseStandardSystemIncludes) {
483     if (triple.isOSDarwin()) {
484       AddPath("/System/Library/Frameworks", System, true);
485       AddPath("/Library/Frameworks", System, true);
486     }
487   }
488 }
489
490 /// RemoveDuplicates - If there are duplicate directory entries in the specified
491 /// search list, remove the later (dead) ones.  Returns the number of non-system
492 /// headers removed, which is used to update NumAngled.
493 static unsigned RemoveDuplicates(std::vector<DirectoryLookup> &SearchList,
494                                  unsigned First, bool Verbose) {
495   llvm::SmallPtrSet<const DirectoryEntry *, 8> SeenDirs;
496   llvm::SmallPtrSet<const DirectoryEntry *, 8> SeenFrameworkDirs;
497   llvm::SmallPtrSet<const HeaderMap *, 8> SeenHeaderMaps;
498   unsigned NonSystemRemoved = 0;
499   for (unsigned i = First; i != SearchList.size(); ++i) {
500     unsigned DirToRemove = i;
501
502     const DirectoryLookup &CurEntry = SearchList[i];
503
504     if (CurEntry.isNormalDir()) {
505       // If this isn't the first time we've seen this dir, remove it.
506       if (SeenDirs.insert(CurEntry.getDir()).second)
507         continue;
508     } else if (CurEntry.isFramework()) {
509       // If this isn't the first time we've seen this framework dir, remove it.
510       if (SeenFrameworkDirs.insert(CurEntry.getFrameworkDir()).second)
511         continue;
512     } else {
513       assert(CurEntry.isHeaderMap() && "Not a headermap or normal dir?");
514       // If this isn't the first time we've seen this headermap, remove it.
515       if (SeenHeaderMaps.insert(CurEntry.getHeaderMap()).second)
516         continue;
517     }
518
519     // If we have a normal #include dir/framework/headermap that is shadowed
520     // later in the chain by a system include location, we actually want to
521     // ignore the user's request and drop the user dir... keeping the system
522     // dir.  This is weird, but required to emulate GCC's search path correctly.
523     //
524     // Since dupes of system dirs are rare, just rescan to find the original
525     // that we're nuking instead of using a DenseMap.
526     if (CurEntry.getDirCharacteristic() != SrcMgr::C_User) {
527       // Find the dir that this is the same of.
528       unsigned FirstDir;
529       for (FirstDir = First;; ++FirstDir) {
530         assert(FirstDir != i && "Didn't find dupe?");
531
532         const DirectoryLookup &SearchEntry = SearchList[FirstDir];
533
534         // If these are different lookup types, then they can't be the dupe.
535         if (SearchEntry.getLookupType() != CurEntry.getLookupType())
536           continue;
537
538         bool isSame;
539         if (CurEntry.isNormalDir())
540           isSame = SearchEntry.getDir() == CurEntry.getDir();
541         else if (CurEntry.isFramework())
542           isSame = SearchEntry.getFrameworkDir() == CurEntry.getFrameworkDir();
543         else {
544           assert(CurEntry.isHeaderMap() && "Not a headermap or normal dir?");
545           isSame = SearchEntry.getHeaderMap() == CurEntry.getHeaderMap();
546         }
547
548         if (isSame)
549           break;
550       }
551
552       // If the first dir in the search path is a non-system dir, zap it
553       // instead of the system one.
554       if (SearchList[FirstDir].getDirCharacteristic() == SrcMgr::C_User)
555         DirToRemove = FirstDir;
556     }
557
558     if (Verbose) {
559       llvm::errs() << "ignoring duplicate directory \""
560                    << CurEntry.getName() << "\"\n";
561       if (DirToRemove != i)
562         llvm::errs() << "  as it is a non-system directory that duplicates "
563                      << "a system directory\n";
564     }
565     if (DirToRemove != i)
566       ++NonSystemRemoved;
567
568     // This is reached if the current entry is a duplicate.  Remove the
569     // DirToRemove (usually the current dir).
570     SearchList.erase(SearchList.begin()+DirToRemove);
571     --i;
572   }
573   return NonSystemRemoved;
574 }
575
576
577 void InitHeaderSearch::Realize(const LangOptions &Lang) {
578   // Concatenate ANGLE+SYSTEM+AFTER chains together into SearchList.
579   std::vector<DirectoryLookup> SearchList;
580   SearchList.reserve(IncludePath.size());
581
582   // Quoted arguments go first.
583   for (auto &Include : IncludePath)
584     if (Include.first == Quoted)
585       SearchList.push_back(Include.second);
586
587   // Deduplicate and remember index.
588   RemoveDuplicates(SearchList, 0, Verbose);
589   unsigned NumQuoted = SearchList.size();
590
591   for (auto &Include : IncludePath)
592     if (Include.first == Angled || Include.first == IndexHeaderMap)
593       SearchList.push_back(Include.second);
594
595   RemoveDuplicates(SearchList, NumQuoted, Verbose);
596   unsigned NumAngled = SearchList.size();
597
598   for (auto &Include : IncludePath)
599     if (Include.first == System || Include.first == ExternCSystem ||
600         (!Lang.ObjC1 && !Lang.CPlusPlus && Include.first == CSystem) ||
601         (/*FIXME !Lang.ObjC1 && */ Lang.CPlusPlus &&
602          Include.first == CXXSystem) ||
603         (Lang.ObjC1 && !Lang.CPlusPlus && Include.first == ObjCSystem) ||
604         (Lang.ObjC1 && Lang.CPlusPlus && Include.first == ObjCXXSystem))
605       SearchList.push_back(Include.second);
606
607   for (auto &Include : IncludePath)
608     if (Include.first == After)
609       SearchList.push_back(Include.second);
610
611   // Remove duplicates across both the Angled and System directories.  GCC does
612   // this and failing to remove duplicates across these two groups breaks
613   // #include_next.
614   unsigned NonSystemRemoved = RemoveDuplicates(SearchList, NumQuoted, Verbose);
615   NumAngled -= NonSystemRemoved;
616
617   bool DontSearchCurDir = false;  // TODO: set to true if -I- is set?
618   Headers.SetSearchPaths(SearchList, NumQuoted, NumAngled, DontSearchCurDir);
619
620   Headers.SetSystemHeaderPrefixes(SystemHeaderPrefixes);
621
622   // If verbose, print the list of directories that will be searched.
623   if (Verbose) {
624     llvm::errs() << "#include \"...\" search starts here:\n";
625     for (unsigned i = 0, e = SearchList.size(); i != e; ++i) {
626       if (i == NumQuoted)
627         llvm::errs() << "#include <...> search starts here:\n";
628       StringRef Name = SearchList[i].getName();
629       const char *Suffix;
630       if (SearchList[i].isNormalDir())
631         Suffix = "";
632       else if (SearchList[i].isFramework())
633         Suffix = " (framework directory)";
634       else {
635         assert(SearchList[i].isHeaderMap() && "Unknown DirectoryLookup");
636         Suffix = " (headermap)";
637       }
638       llvm::errs() << " " << Name << Suffix << "\n";
639     }
640     llvm::errs() << "End of search list.\n";
641   }
642 }
643
644 void clang::ApplyHeaderSearchOptions(HeaderSearch &HS,
645                                      const HeaderSearchOptions &HSOpts,
646                                      const LangOptions &Lang,
647                                      const llvm::Triple &Triple) {
648   InitHeaderSearch Init(HS, HSOpts.Verbose, HSOpts.Sysroot);
649
650   // Add the user defined entries.
651   for (unsigned i = 0, e = HSOpts.UserEntries.size(); i != e; ++i) {
652     const HeaderSearchOptions::Entry &E = HSOpts.UserEntries[i];
653     if (E.IgnoreSysRoot) {
654       Init.AddUnmappedPath(E.Path, E.Group, E.IsFramework);
655     } else {
656       Init.AddPath(E.Path, E.Group, E.IsFramework);
657     }
658   }
659
660   Init.AddDefaultIncludePaths(Lang, Triple, HSOpts);
661
662   for (unsigned i = 0, e = HSOpts.SystemHeaderPrefixes.size(); i != e; ++i)
663     Init.AddSystemHeaderPrefix(HSOpts.SystemHeaderPrefixes[i].Prefix,
664                                HSOpts.SystemHeaderPrefixes[i].IsSystemHeader);
665
666   if (HSOpts.UseBuiltinIncludes) {
667     // Set up the builtin include directory in the module map.
668     SmallString<128> P = StringRef(HSOpts.ResourceDir);
669     llvm::sys::path::append(P, "include");
670     if (const DirectoryEntry *Dir = HS.getFileMgr().getDirectory(P))
671       HS.getModuleMap().setBuiltinIncludeDir(Dir);
672   }
673
674   Init.Realize(Lang);
675 }