1212904Sdim//===--- InitHeaderSearch.cpp - Initialize header search paths ------------===//
2193326Sed//
3193326Sed//                     The LLVM Compiler Infrastructure
4193326Sed//
5193326Sed// This file is distributed under the University of Illinois Open Source
6193326Sed// License. See LICENSE.TXT for details.
7193326Sed//
8193326Sed//===----------------------------------------------------------------------===//
9193326Sed//
10193326Sed// This file implements the InitHeaderSearch class.
11193326Sed//
12193326Sed//===----------------------------------------------------------------------===//
13193326Sed
14199482Srdivacky#include "clang/Frontend/Utils.h"
15193326Sed#include "clang/Basic/FileManager.h"
16193326Sed#include "clang/Basic/LangOptions.h"
17208961Srdivacky#include "clang/Basic/Version.h"
18249423Sdim#include "clang/Config/config.h" // C_INCLUDE_DIRS
19249423Sdim#include "clang/Lex/HeaderSearch.h"
20243830Sdim#include "clang/Lex/HeaderSearchOptions.h"
21249423Sdim#include "llvm/ADT/SmallPtrSet.h"
22193326Sed#include "llvm/ADT/SmallString.h"
23199482Srdivacky#include "llvm/ADT/SmallVector.h"
24199482Srdivacky#include "llvm/ADT/StringExtras.h"
25199482Srdivacky#include "llvm/ADT/Triple.h"
26200583Srdivacky#include "llvm/ADT/Twine.h"
27228379Sdim#include "llvm/Support/ErrorHandling.h"
28251662Sdim#include "llvm/Support/FileSystem.h"
29218893Sdim#include "llvm/Support/Path.h"
30249423Sdim#include "llvm/Support/raw_ostream.h"
31234353Sdim
32193326Sedusing namespace clang;
33199482Srdivackyusing namespace clang::frontend;
34193326Sed
35199482Srdivackynamespace {
36199482Srdivacky
37199482Srdivacky/// InitHeaderSearch - This class makes it easier to set the search paths of
38199482Srdivacky///  a HeaderSearch object. InitHeaderSearch stores several search path lists
39199482Srdivacky///  internally, which can be sent to a HeaderSearch object in one swoop.
40199482Srdivackyclass InitHeaderSearch {
41219077Sdim  std::vector<std::pair<IncludeDirGroup, DirectoryLookup> > IncludePath;
42219077Sdim  typedef std::vector<std::pair<IncludeDirGroup,
43219077Sdim                      DirectoryLookup> >::const_iterator path_iterator;
44239462Sdim  std::vector<std::pair<std::string, bool> > SystemHeaderPrefixes;
45224145Sdim  HeaderSearch &Headers;
46199482Srdivacky  bool Verbose;
47218893Sdim  std::string IncludeSysroot;
48249423Sdim  bool HasSysroot;
49199482Srdivacky
50199482Srdivackypublic:
51199482Srdivacky
52226633Sdim  InitHeaderSearch(HeaderSearch &HS, bool verbose, StringRef sysroot)
53218893Sdim    : Headers(HS), Verbose(verbose), IncludeSysroot(sysroot),
54249423Sdim      HasSysroot(!(sysroot.empty() || sysroot == "/")) {
55218893Sdim  }
56199482Srdivacky
57249423Sdim  /// AddPath - Add the specified path to the specified group list, prefixing
58249423Sdim  /// the sysroot if used.
59249423Sdim  void AddPath(const Twine &Path, IncludeDirGroup Group, bool isFramework);
60199482Srdivacky
61249423Sdim  /// AddUnmappedPath - Add the specified path to the specified group list,
62249423Sdim  /// without performing any sysroot remapping.
63249423Sdim  void AddUnmappedPath(const Twine &Path, IncludeDirGroup Group,
64249423Sdim                       bool isFramework);
65249423Sdim
66239462Sdim  /// AddSystemHeaderPrefix - Add the specified prefix to the system header
67239462Sdim  /// prefix list.
68239462Sdim  void AddSystemHeaderPrefix(StringRef Prefix, bool IsSystemHeader) {
69239462Sdim    SystemHeaderPrefixes.push_back(std::make_pair(Prefix, IsSystemHeader));
70239462Sdim  }
71239462Sdim
72208600Srdivacky  /// AddGnuCPlusPlusIncludePaths - Add the necessary paths to support a gnu
73199482Srdivacky  ///  libstdc++.
74226633Sdim  void AddGnuCPlusPlusIncludePaths(StringRef Base,
75226633Sdim                                   StringRef ArchDir,
76226633Sdim                                   StringRef Dir32,
77226633Sdim                                   StringRef Dir64,
78199482Srdivacky                                   const llvm::Triple &triple);
79199482Srdivacky
80218893Sdim  /// AddMinGWCPlusPlusIncludePaths - Add the necessary paths to support a MinGW
81199482Srdivacky  ///  libstdc++.
82226633Sdim  void AddMinGWCPlusPlusIncludePaths(StringRef Base,
83226633Sdim                                     StringRef Arch,
84226633Sdim                                     StringRef Version);
85199482Srdivacky
86221345Sdim  /// AddMinGW64CXXPaths - Add the necessary paths to support
87221345Sdim  /// libstdc++ of x86_64-w64-mingw32 aka mingw-w64.
88226633Sdim  void AddMinGW64CXXPaths(StringRef Base,
89226633Sdim                          StringRef Version);
90221345Sdim
91199482Srdivacky  // AddDefaultCIncludePaths - Add paths that should always be searched.
92208600Srdivacky  void AddDefaultCIncludePaths(const llvm::Triple &triple,
93208600Srdivacky                               const HeaderSearchOptions &HSOpts);
94199482Srdivacky
95199482Srdivacky  // AddDefaultCPlusPlusIncludePaths -  Add paths that should be searched when
96199482Srdivacky  //  compiling c++.
97224145Sdim  void AddDefaultCPlusPlusIncludePaths(const llvm::Triple &triple,
98224145Sdim                                       const HeaderSearchOptions &HSOpts);
99199482Srdivacky
100199482Srdivacky  /// AddDefaultSystemIncludePaths - Adds the default system include paths so
101199482Srdivacky  ///  that e.g. stdio.h is found.
102226633Sdim  void AddDefaultIncludePaths(const LangOptions &Lang,
103226633Sdim                              const llvm::Triple &triple,
104226633Sdim                              const HeaderSearchOptions &HSOpts);
105199482Srdivacky
106199482Srdivacky  /// Realize - Merges all search path lists into one list and send it to
107199482Srdivacky  /// HeaderSearch.
108219077Sdim  void Realize(const LangOptions &Lang);
109199482Srdivacky};
110199482Srdivacky
111224145Sdim}  // end anonymous namespace.
112199482Srdivacky
113249423Sdimstatic bool CanPrefixSysroot(StringRef Path) {
114249423Sdim#if defined(_WIN32)
115249423Sdim  return !Path.empty() && llvm::sys::path::is_separator(Path[0]);
116249423Sdim#else
117249423Sdim  return llvm::sys::path::is_absolute(Path);
118249423Sdim#endif
119249423Sdim}
120249423Sdim
121249423Sdimvoid InitHeaderSearch::AddPath(const Twine &Path, IncludeDirGroup Group,
122249423Sdim                               bool isFramework) {
123249423Sdim  // Add the path with sysroot prepended, if desired and this is a system header
124249423Sdim  // group.
125249423Sdim  if (HasSysroot) {
126249423Sdim    SmallString<256> MappedPathStorage;
127249423Sdim    StringRef MappedPathStr = Path.toStringRef(MappedPathStorage);
128249423Sdim    if (CanPrefixSysroot(MappedPathStr)) {
129249423Sdim      AddUnmappedPath(IncludeSysroot + Path, Group, isFramework);
130249423Sdim      return;
131249423Sdim    }
132249423Sdim  }
133249423Sdim
134249423Sdim  AddUnmappedPath(Path, Group, isFramework);
135249423Sdim}
136249423Sdim
137249423Sdimvoid InitHeaderSearch::AddUnmappedPath(const Twine &Path, IncludeDirGroup Group,
138249423Sdim                                       bool isFramework) {
139200583Srdivacky  assert(!Path.isTriviallyEmpty() && "can't handle empty path here");
140249423Sdim
141193326Sed  FileManager &FM = Headers.getFileMgr();
142234353Sdim  SmallString<256> MappedPathStorage;
143226633Sdim  StringRef MappedPathStr = Path.toStringRef(MappedPathStorage);
144198092Srdivacky
145193326Sed  // Compute the DirectoryLookup type.
146193326Sed  SrcMgr::CharacteristicKind Type;
147249423Sdim  if (Group == Quoted || Group == Angled || Group == IndexHeaderMap) {
148193326Sed    Type = SrcMgr::C_User;
149249423Sdim  } else if (Group == ExternCSystem) {
150249423Sdim    Type = SrcMgr::C_ExternCSystem;
151249423Sdim  } else {
152193326Sed    Type = SrcMgr::C_System;
153249423Sdim  }
154198092Srdivacky
155193326Sed  // If the directory exists, add it.
156218893Sdim  if (const DirectoryEntry *DE = FM.getDirectory(MappedPathStr)) {
157249423Sdim    IncludePath.push_back(
158249423Sdim      std::make_pair(Group, DirectoryLookup(DE, Type, isFramework)));
159193326Sed    return;
160193326Sed  }
161198092Srdivacky
162193326Sed  // Check to see if this is an apple-style headermap (which are not allowed to
163193326Sed  // be frameworks).
164193326Sed  if (!isFramework) {
165218893Sdim    if (const FileEntry *FE = FM.getFile(MappedPathStr)) {
166193326Sed      if (const HeaderMap *HM = Headers.CreateHeaderMap(FE)) {
167193326Sed        // It is a headermap, add it to the search path.
168249423Sdim        IncludePath.push_back(
169249423Sdim          std::make_pair(Group,
170249423Sdim                         DirectoryLookup(HM, Type, Group == IndexHeaderMap)));
171193326Sed        return;
172193326Sed      }
173193326Sed    }
174193326Sed  }
175198092Srdivacky
176193326Sed  if (Verbose)
177198092Srdivacky    llvm::errs() << "ignoring nonexistent directory \""
178218893Sdim                 << MappedPathStr << "\"\n";
179193326Sed}
180193326Sed
181226633Sdimvoid InitHeaderSearch::AddGnuCPlusPlusIncludePaths(StringRef Base,
182226633Sdim                                                   StringRef ArchDir,
183226633Sdim                                                   StringRef Dir32,
184226633Sdim                                                   StringRef Dir64,
185198092Srdivacky                                                   const llvm::Triple &triple) {
186199990Srdivacky  // Add the base dir
187249423Sdim  AddPath(Base, CXXSystem, false);
188199482Srdivacky
189199482Srdivacky  // Add the multilib dirs
190198092Srdivacky  llvm::Triple::ArchType arch = triple.getArch();
191198092Srdivacky  bool is64bit = arch == llvm::Triple::ppc64 || arch == llvm::Triple::x86_64;
192198092Srdivacky  if (is64bit)
193249423Sdim    AddPath(Base + "/" + ArchDir + "/" + Dir64, CXXSystem, false);
194198092Srdivacky  else
195249423Sdim    AddPath(Base + "/" + ArchDir + "/" + Dir32, CXXSystem, false);
196199990Srdivacky
197199990Srdivacky  // Add the backward dir
198249423Sdim  AddPath(Base + "/backward", CXXSystem, false);
199198092Srdivacky}
200193326Sed
201226633Sdimvoid InitHeaderSearch::AddMinGWCPlusPlusIncludePaths(StringRef Base,
202226633Sdim                                                     StringRef Arch,
203226633Sdim                                                     StringRef Version) {
204202879Srdivacky  AddPath(Base + "/" + Arch + "/" + Version + "/include/c++",
205249423Sdim          CXXSystem, false);
206212904Sdim  AddPath(Base + "/" + Arch + "/" + Version + "/include/c++/" + Arch,
207249423Sdim          CXXSystem, false);
208202879Srdivacky  AddPath(Base + "/" + Arch + "/" + Version + "/include/c++/backward",
209249423Sdim          CXXSystem, false);
210198092Srdivacky}
211198092Srdivacky
212226633Sdimvoid InitHeaderSearch::AddMinGW64CXXPaths(StringRef Base,
213226633Sdim                                          StringRef Version) {
214224145Sdim  // Assumes Base is HeaderSearchOpts' ResourceDir
215224145Sdim  AddPath(Base + "/../../../include/c++/" + Version,
216249423Sdim          CXXSystem, false);
217224145Sdim  AddPath(Base + "/../../../include/c++/" + Version + "/x86_64-w64-mingw32",
218249423Sdim          CXXSystem, false);
219224145Sdim  AddPath(Base + "/../../../include/c++/" + Version + "/i686-w64-mingw32",
220249423Sdim          CXXSystem, false);
221224145Sdim  AddPath(Base + "/../../../include/c++/" + Version + "/backward",
222249423Sdim          CXXSystem, false);
223221345Sdim}
224221345Sdim
225208600Srdivackyvoid InitHeaderSearch::AddDefaultCIncludePaths(const llvm::Triple &triple,
226208600Srdivacky                                            const HeaderSearchOptions &HSOpts) {
227218893Sdim  llvm::Triple::OSType os = triple.getOS();
228208600Srdivacky
229226633Sdim  if (HSOpts.UseStandardSystemIncludes) {
230226633Sdim    switch (os) {
231226633Sdim    case llvm::Triple::FreeBSD:
232226633Sdim    case llvm::Triple::NetBSD:
233239462Sdim    case llvm::Triple::OpenBSD:
234239462Sdim    case llvm::Triple::Bitrig:
235226633Sdim      break;
236226633Sdim    default:
237226633Sdim      // FIXME: temporary hack: hard-coded paths.
238249423Sdim      AddPath("/usr/local/include", System, false);
239226633Sdim      break;
240226633Sdim    }
241218893Sdim  }
242218893Sdim
243208600Srdivacky  // Builtin includes use #include_next directives and should be positioned
244208600Srdivacky  // just prior C include dirs.
245208600Srdivacky  if (HSOpts.UseBuiltinIncludes) {
246208600Srdivacky    // Ignore the sys root, we *always* look for clang headers relative to
247208600Srdivacky    // supplied path.
248263508Sdim    SmallString<128> P = StringRef(HSOpts.ResourceDir);
249263508Sdim    llvm::sys::path::append(P, "include");
250249423Sdim    AddUnmappedPath(P.str(), ExternCSystem, false);
251208600Srdivacky  }
252208600Srdivacky
253226633Sdim  // All remaining additions are for system include directories, early exit if
254226633Sdim  // we aren't using them.
255226633Sdim  if (!HSOpts.UseStandardSystemIncludes)
256226633Sdim    return;
257226633Sdim
258208600Srdivacky  // Add dirs specified via 'configure --with-c-include-dirs'.
259226633Sdim  StringRef CIncludeDirs(C_INCLUDE_DIRS);
260199482Srdivacky  if (CIncludeDirs != "") {
261226633Sdim    SmallVector<StringRef, 5> dirs;
262199482Srdivacky    CIncludeDirs.split(dirs, ":");
263226633Sdim    for (SmallVectorImpl<StringRef>::iterator i = dirs.begin();
264199482Srdivacky         i != dirs.end();
265218893Sdim         ++i)
266249423Sdim      AddPath(*i, ExternCSystem, false);
267199482Srdivacky    return;
268199482Srdivacky  }
269218893Sdim
270198092Srdivacky  switch (os) {
271228379Sdim  case llvm::Triple::Linux:
272228379Sdim  case llvm::Triple::Win32:
273228379Sdim    llvm_unreachable("Include management is handled in the driver.");
274228379Sdim
275207619Srdivacky  case llvm::Triple::Haiku:
276249423Sdim    AddPath("/boot/common/include", System, false);
277249423Sdim    AddPath("/boot/develop/headers/os", System, false);
278249423Sdim    AddPath("/boot/develop/headers/os/app", System, false);
279249423Sdim    AddPath("/boot/develop/headers/os/arch", System, false);
280249423Sdim    AddPath("/boot/develop/headers/os/device", System, false);
281249423Sdim    AddPath("/boot/develop/headers/os/drivers", System, false);
282249423Sdim    AddPath("/boot/develop/headers/os/game", System, false);
283249423Sdim    AddPath("/boot/develop/headers/os/interface", System, false);
284249423Sdim    AddPath("/boot/develop/headers/os/kernel", System, false);
285249423Sdim    AddPath("/boot/develop/headers/os/locale", System, false);
286249423Sdim    AddPath("/boot/develop/headers/os/mail", System, false);
287249423Sdim    AddPath("/boot/develop/headers/os/media", System, false);
288249423Sdim    AddPath("/boot/develop/headers/os/midi", System, false);
289249423Sdim    AddPath("/boot/develop/headers/os/midi2", System, false);
290249423Sdim    AddPath("/boot/develop/headers/os/net", System, false);
291249423Sdim    AddPath("/boot/develop/headers/os/storage", System, false);
292249423Sdim    AddPath("/boot/develop/headers/os/support", System, false);
293249423Sdim    AddPath("/boot/develop/headers/os/translation", System, false);
294249423Sdim    AddPath("/boot/develop/headers/os/add-ons/graphics", System, false);
295249423Sdim    AddPath("/boot/develop/headers/os/add-ons/input_server", System, false);
296249423Sdim    AddPath("/boot/develop/headers/os/add-ons/screen_saver", System, false);
297249423Sdim    AddPath("/boot/develop/headers/os/add-ons/tracker", System, false);
298249423Sdim    AddPath("/boot/develop/headers/os/be_apps/Deskbar", System, false);
299249423Sdim    AddPath("/boot/develop/headers/os/be_apps/NetPositive", System, false);
300249423Sdim    AddPath("/boot/develop/headers/os/be_apps/Tracker", System, false);
301249423Sdim    AddPath("/boot/develop/headers/cpp", System, false);
302249423Sdim    AddPath("/boot/develop/headers/cpp/i586-pc-haiku", System, false);
303249423Sdim    AddPath("/boot/develop/headers/3rdparty", System, false);
304249423Sdim    AddPath("/boot/develop/headers/bsd", System, false);
305249423Sdim    AddPath("/boot/develop/headers/glibc", System, false);
306249423Sdim    AddPath("/boot/develop/headers/posix", System, false);
307249423Sdim    AddPath("/boot/develop/headers",  System, false);
308212904Sdim    break;
309224145Sdim  case llvm::Triple::RTEMS:
310224145Sdim    break;
311218893Sdim  case llvm::Triple::Cygwin:
312249423Sdim    AddPath("/usr/include/w32api", System, false);
313218893Sdim    break;
314224145Sdim  case llvm::Triple::MinGW32: {
315224145Sdim      // mingw-w64 crt include paths
316263508Sdim      // <sysroot>/i686-w64-mingw32/include
317263508Sdim      SmallString<128> P = StringRef(HSOpts.ResourceDir);
318263508Sdim      llvm::sys::path::append(P, "../../../i686-w64-mingw32/include");
319249423Sdim      AddPath(P.str(), System, false);
320263508Sdim
321263508Sdim      // <sysroot>/x86_64-w64-mingw32/include
322263508Sdim      P.resize(HSOpts.ResourceDir.size());
323263508Sdim      llvm::sys::path::append(P, "../../../x86_64-w64-mingw32/include");
324249423Sdim      AddPath(P.str(), System, false);
325263508Sdim
326224145Sdim      // mingw.org crt include paths
327263508Sdim      // <sysroot>/include
328263508Sdim      P.resize(HSOpts.ResourceDir.size());
329263508Sdim      llvm::sys::path::append(P, "../../../include");
330249423Sdim      AddPath(P.str(), System, false);
331249423Sdim      AddPath("/mingw/include", System, false);
332243830Sdim#if defined(_WIN32)
333249423Sdim      AddPath("c:/mingw/include", System, false);
334243830Sdim#endif
335224145Sdim    }
336198893Srdivacky    break;
337211573Srpaulo  case llvm::Triple::FreeBSD:
338249423Sdim    AddPath("/usr/include/clang/" CLANG_VERSION_STRING, System, false);
339211573Srpaulo    break;
340224145Sdim
341198893Srdivacky  default:
342198893Srdivacky    break;
343198893Srdivacky  }
344198893Srdivacky
345224145Sdim  if ( os != llvm::Triple::RTEMS )
346249423Sdim    AddPath("/usr/include", ExternCSystem, false);
347198893Srdivacky}
348198893Srdivacky
349208600Srdivackyvoid InitHeaderSearch::
350224145SdimAddDefaultCPlusPlusIncludePaths(const llvm::Triple &triple, const HeaderSearchOptions &HSOpts) {
351198893Srdivacky  llvm::Triple::OSType os = triple.getOS();
352198893Srdivacky  // FIXME: temporary hack: hard-coded paths.
353221345Sdim
354221345Sdim  if (triple.isOSDarwin()) {
355210299Sed    switch (triple.getArch()) {
356210299Sed    default: break;
357210299Sed
358218893Sdim    case llvm::Triple::ppc:
359210299Sed    case llvm::Triple::ppc64:
360210299Sed      AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.2.1",
361218893Sdim                                  "powerpc-apple-darwin10", "", "ppc64",
362210299Sed                                  triple);
363210299Sed      AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.0.0",
364218893Sdim                                  "powerpc-apple-darwin10", "", "ppc64",
365210299Sed                                  triple);
366210299Sed      break;
367210299Sed
368210299Sed    case llvm::Triple::x86:
369210299Sed    case llvm::Triple::x86_64:
370210299Sed      AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.2.1",
371210299Sed                                  "i686-apple-darwin10", "", "x86_64", triple);
372210299Sed      AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.0.0",
373210299Sed                                  "i686-apple-darwin8", "", "", triple);
374210299Sed      break;
375210299Sed
376210299Sed    case llvm::Triple::arm:
377210299Sed    case llvm::Triple::thumb:
378210299Sed      AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.2.1",
379210299Sed                                  "arm-apple-darwin10", "v7", "", triple);
380210299Sed      AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.2.1",
381210299Sed                                  "arm-apple-darwin10", "v6", "", triple);
382210299Sed      break;
383210299Sed    }
384221345Sdim    return;
385221345Sdim  }
386221345Sdim
387221345Sdim  switch (os) {
388228379Sdim  case llvm::Triple::Linux:
389228379Sdim  case llvm::Triple::Win32:
390228379Sdim    llvm_unreachable("Include management is handled in the driver.");
391228379Sdim
392221345Sdim  case llvm::Triple::Cygwin:
393221345Sdim    // Cygwin-1.7
394263508Sdim    AddMinGWCPlusPlusIncludePaths("/usr/lib/gcc", "i686-pc-cygwin", "4.7.3");
395234353Sdim    AddMinGWCPlusPlusIncludePaths("/usr/lib/gcc", "i686-pc-cygwin", "4.5.3");
396221345Sdim    AddMinGWCPlusPlusIncludePaths("/usr/lib/gcc", "i686-pc-cygwin", "4.3.4");
397221345Sdim    // g++-4 / Cygwin-1.5
398221345Sdim    AddMinGWCPlusPlusIncludePaths("/usr/lib/gcc", "i686-pc-cygwin", "4.3.2");
399198893Srdivacky    break;
400221345Sdim  case llvm::Triple::MinGW32:
401224145Sdim    // mingw-w64 C++ include paths (i686-w64-mingw32 and x86_64-w64-mingw32)
402224145Sdim    AddMinGW64CXXPaths(HSOpts.ResourceDir, "4.5.0");
403224145Sdim    AddMinGW64CXXPaths(HSOpts.ResourceDir, "4.5.1");
404224145Sdim    AddMinGW64CXXPaths(HSOpts.ResourceDir, "4.5.2");
405224145Sdim    AddMinGW64CXXPaths(HSOpts.ResourceDir, "4.5.3");
406234353Sdim    AddMinGW64CXXPaths(HSOpts.ResourceDir, "4.5.4");
407224145Sdim    AddMinGW64CXXPaths(HSOpts.ResourceDir, "4.6.0");
408224145Sdim    AddMinGW64CXXPaths(HSOpts.ResourceDir, "4.6.1");
409224145Sdim    AddMinGW64CXXPaths(HSOpts.ResourceDir, "4.6.2");
410234353Sdim    AddMinGW64CXXPaths(HSOpts.ResourceDir, "4.6.3");
411224145Sdim    AddMinGW64CXXPaths(HSOpts.ResourceDir, "4.7.0");
412224145Sdim    // mingw.org C++ include paths
413224145Sdim    AddMinGWCPlusPlusIncludePaths("/mingw/lib/gcc", "mingw32", "4.5.2"); //MSYS
414243830Sdim#if defined(_WIN32)
415263508Sdim    AddMinGWCPlusPlusIncludePaths("c:/MinGW/lib/gcc", "mingw32", "4.8.1");
416234353Sdim    AddMinGWCPlusPlusIncludePaths("c:/MinGW/lib/gcc", "mingw32", "4.6.2");
417234353Sdim    AddMinGWCPlusPlusIncludePaths("c:/MinGW/lib/gcc", "mingw32", "4.6.1");
418234353Sdim    AddMinGWCPlusPlusIncludePaths("c:/MinGW/lib/gcc", "mingw32", "4.5.2");
419221345Sdim    AddMinGWCPlusPlusIncludePaths("c:/MinGW/lib/gcc", "mingw32", "4.5.0");
420221345Sdim    AddMinGWCPlusPlusIncludePaths("c:/MinGW/lib/gcc", "mingw32", "4.4.0");
421221345Sdim    AddMinGWCPlusPlusIncludePaths("c:/MinGW/lib/gcc", "mingw32", "4.3.0");
422243830Sdim#endif
423221345Sdim    break;
424202379Srdivacky  case llvm::Triple::DragonFly:
425251662Sdim    if (llvm::sys::fs::exists("/usr/lib/gcc47"))
426251662Sdim      AddPath("/usr/include/c++/4.7", CXXSystem, false);
427251662Sdim    else
428251662Sdim      AddPath("/usr/include/c++/4.4", CXXSystem, false);
429202379Srdivacky    break;
430212904Sdim  case llvm::Triple::OpenBSD: {
431212904Sdim    std::string t = triple.getTriple();
432212904Sdim    if (t.substr(0, 6) == "x86_64")
433212904Sdim      t.replace(0, 6, "amd64");
434212904Sdim    AddGnuCPlusPlusIncludePaths("/usr/include/g++",
435212904Sdim                                t, "", "", triple);
436212904Sdim    break;
437212904Sdim  }
438210299Sed  case llvm::Triple::Minix:
439210299Sed    AddGnuCPlusPlusIncludePaths("/usr/gnu/include/c++/4.4.3",
440210299Sed                                "", "", "", triple);
441210299Sed    break;
442198893Srdivacky  case llvm::Triple::Solaris:
443234353Sdim    AddGnuCPlusPlusIncludePaths("/usr/gcc/4.5/include/c++/4.5.2/",
444234353Sdim                                "i386-pc-solaris2.11", "", "", triple);
445198893Srdivacky    // Solaris - Fall though..
446198893Srdivacky  case llvm::Triple::AuroraUX:
447198893Srdivacky    // AuroraUX
448198893Srdivacky    AddGnuCPlusPlusIncludePaths("/opt/gcc4/include/c++/4.2.4",
449199482Srdivacky                                "i386-pc-solaris2.11", "", "", triple);
450198893Srdivacky    break;
451198092Srdivacky  default:
452198092Srdivacky    break;
453193326Sed  }
454198893Srdivacky}
455193326Sed
456226633Sdimvoid InitHeaderSearch::AddDefaultIncludePaths(const LangOptions &Lang,
457226633Sdim                                              const llvm::Triple &triple,
458208600Srdivacky                                            const HeaderSearchOptions &HSOpts) {
459228379Sdim  // NB: This code path is going away. All of the logic is moving into the
460228379Sdim  // driver which has the information necessary to do target-specific
461228379Sdim  // selections of default include paths. Each target which moves there will be
462228379Sdim  // exempted from this logic here until we can delete the entire pile of code.
463228379Sdim  switch (triple.getOS()) {
464228379Sdim  default:
465228379Sdim    break; // Everything else continues to use this routine's logic.
466228379Sdim
467228379Sdim  case llvm::Triple::Linux:
468228379Sdim  case llvm::Triple::Win32:
469228379Sdim    return;
470228379Sdim  }
471228379Sdim
472226633Sdim  if (Lang.CPlusPlus && HSOpts.UseStandardCXXIncludes &&
473226633Sdim      HSOpts.UseStandardSystemIncludes) {
474226633Sdim    if (HSOpts.UseLibcxx) {
475226633Sdim      if (triple.isOSDarwin()) {
476226633Sdim        // On Darwin, libc++ may be installed alongside the compiler in
477263508Sdim        // include/c++/v1.
478263508Sdim        if (!HSOpts.ResourceDir.empty()) {
479263508Sdim          // Remove version from foo/lib/clang/version
480263508Sdim          StringRef NoVer = llvm::sys::path::parent_path(HSOpts.ResourceDir);
481263508Sdim          // Remove clang from foo/lib/clang
482263508Sdim          StringRef Lib = llvm::sys::path::parent_path(NoVer);
483263508Sdim          // Remove lib from foo/lib
484263508Sdim          SmallString<128> P = llvm::sys::path::parent_path(Lib);
485263508Sdim
486263508Sdim          // Get foo/include/c++/v1
487263508Sdim          llvm::sys::path::append(P, "include", "c++", "v1");
488249423Sdim          AddUnmappedPath(P.str(), CXXSystem, false);
489226633Sdim        }
490226633Sdim      }
491234353Sdim      // On Solaris, include the support directory for things like xlocale and
492234353Sdim      // fudged system headers.
493234353Sdim      if (triple.getOS() == llvm::Triple::Solaris)
494249423Sdim        AddPath("/usr/include/c++/v1/support/solaris", CXXSystem, false);
495226633Sdim
496249423Sdim      AddPath("/usr/include/c++/v1", CXXSystem, false);
497226633Sdim    } else {
498224145Sdim      AddDefaultCPlusPlusIncludePaths(triple, HSOpts);
499226633Sdim    }
500224145Sdim  }
501199990Srdivacky
502208600Srdivacky  AddDefaultCIncludePaths(triple, HSOpts);
503199482Srdivacky
504199482Srdivacky  // Add the default framework include paths on Darwin.
505226633Sdim  if (HSOpts.UseStandardSystemIncludes) {
506226633Sdim    if (triple.isOSDarwin()) {
507249423Sdim      AddPath("/System/Library/Frameworks", System, true);
508249423Sdim      AddPath("/Library/Frameworks", System, true);
509226633Sdim    }
510199482Srdivacky  }
511198893Srdivacky}
512198893Srdivacky
513193326Sed/// RemoveDuplicates - If there are duplicate directory entries in the specified
514226633Sdim/// search list, remove the later (dead) ones.  Returns the number of non-system
515226633Sdim/// headers removed, which is used to update NumAngled.
516226633Sdimstatic unsigned RemoveDuplicates(std::vector<DirectoryLookup> &SearchList,
517226633Sdim                                 unsigned First, bool Verbose) {
518193326Sed  llvm::SmallPtrSet<const DirectoryEntry *, 8> SeenDirs;
519193326Sed  llvm::SmallPtrSet<const DirectoryEntry *, 8> SeenFrameworkDirs;
520193326Sed  llvm::SmallPtrSet<const HeaderMap *, 8> SeenHeaderMaps;
521226633Sdim  unsigned NonSystemRemoved = 0;
522219077Sdim  for (unsigned i = First; i != SearchList.size(); ++i) {
523193326Sed    unsigned DirToRemove = i;
524198092Srdivacky
525193326Sed    const DirectoryLookup &CurEntry = SearchList[i];
526198092Srdivacky
527193326Sed    if (CurEntry.isNormalDir()) {
528193326Sed      // If this isn't the first time we've seen this dir, remove it.
529193326Sed      if (SeenDirs.insert(CurEntry.getDir()))
530193326Sed        continue;
531193326Sed    } else if (CurEntry.isFramework()) {
532193326Sed      // If this isn't the first time we've seen this framework dir, remove it.
533193326Sed      if (SeenFrameworkDirs.insert(CurEntry.getFrameworkDir()))
534193326Sed        continue;
535193326Sed    } else {
536193326Sed      assert(CurEntry.isHeaderMap() && "Not a headermap or normal dir?");
537193326Sed      // If this isn't the first time we've seen this headermap, remove it.
538193326Sed      if (SeenHeaderMaps.insert(CurEntry.getHeaderMap()))
539193326Sed        continue;
540193326Sed    }
541198092Srdivacky
542193326Sed    // If we have a normal #include dir/framework/headermap that is shadowed
543193326Sed    // later in the chain by a system include location, we actually want to
544193326Sed    // ignore the user's request and drop the user dir... keeping the system
545193326Sed    // dir.  This is weird, but required to emulate GCC's search path correctly.
546193326Sed    //
547193326Sed    // Since dupes of system dirs are rare, just rescan to find the original
548193326Sed    // that we're nuking instead of using a DenseMap.
549193326Sed    if (CurEntry.getDirCharacteristic() != SrcMgr::C_User) {
550193326Sed      // Find the dir that this is the same of.
551193326Sed      unsigned FirstDir;
552193326Sed      for (FirstDir = 0; ; ++FirstDir) {
553193326Sed        assert(FirstDir != i && "Didn't find dupe?");
554198092Srdivacky
555193326Sed        const DirectoryLookup &SearchEntry = SearchList[FirstDir];
556193326Sed
557193326Sed        // If these are different lookup types, then they can't be the dupe.
558193326Sed        if (SearchEntry.getLookupType() != CurEntry.getLookupType())
559193326Sed          continue;
560198092Srdivacky
561193326Sed        bool isSame;
562193326Sed        if (CurEntry.isNormalDir())
563193326Sed          isSame = SearchEntry.getDir() == CurEntry.getDir();
564193326Sed        else if (CurEntry.isFramework())
565193326Sed          isSame = SearchEntry.getFrameworkDir() == CurEntry.getFrameworkDir();
566193326Sed        else {
567193326Sed          assert(CurEntry.isHeaderMap() && "Not a headermap or normal dir?");
568193326Sed          isSame = SearchEntry.getHeaderMap() == CurEntry.getHeaderMap();
569193326Sed        }
570198092Srdivacky
571193326Sed        if (isSame)
572193326Sed          break;
573193326Sed      }
574198092Srdivacky
575193326Sed      // If the first dir in the search path is a non-system dir, zap it
576193326Sed      // instead of the system one.
577193326Sed      if (SearchList[FirstDir].getDirCharacteristic() == SrcMgr::C_User)
578193326Sed        DirToRemove = FirstDir;
579193326Sed    }
580193326Sed
581193326Sed    if (Verbose) {
582200583Srdivacky      llvm::errs() << "ignoring duplicate directory \""
583200583Srdivacky                   << CurEntry.getName() << "\"\n";
584193326Sed      if (DirToRemove != i)
585200583Srdivacky        llvm::errs() << "  as it is a non-system directory that duplicates "
586200583Srdivacky                     << "a system directory\n";
587193326Sed    }
588226633Sdim    if (DirToRemove != i)
589226633Sdim      ++NonSystemRemoved;
590198092Srdivacky
591193326Sed    // This is reached if the current entry is a duplicate.  Remove the
592193326Sed    // DirToRemove (usually the current dir).
593193326Sed    SearchList.erase(SearchList.begin()+DirToRemove);
594193326Sed    --i;
595193326Sed  }
596226633Sdim  return NonSystemRemoved;
597193326Sed}
598193326Sed
599193326Sed
600219077Sdimvoid InitHeaderSearch::Realize(const LangOptions &Lang) {
601193326Sed  // Concatenate ANGLE+SYSTEM+AFTER chains together into SearchList.
602193326Sed  std::vector<DirectoryLookup> SearchList;
603219077Sdim  SearchList.reserve(IncludePath.size());
604198092Srdivacky
605224145Sdim  // Quoted arguments go first.
606219077Sdim  for (path_iterator it = IncludePath.begin(), ie = IncludePath.end();
607219077Sdim       it != ie; ++it) {
608219077Sdim    if (it->first == Quoted)
609219077Sdim      SearchList.push_back(it->second);
610219077Sdim  }
611224145Sdim  // Deduplicate and remember index.
612219077Sdim  RemoveDuplicates(SearchList, 0, Verbose);
613224145Sdim  unsigned NumQuoted = SearchList.size();
614193326Sed
615219077Sdim  for (path_iterator it = IncludePath.begin(), ie = IncludePath.end();
616219077Sdim       it != ie; ++it) {
617226633Sdim    if (it->first == Angled || it->first == IndexHeaderMap)
618219077Sdim      SearchList.push_back(it->second);
619219077Sdim  }
620198092Srdivacky
621224145Sdim  RemoveDuplicates(SearchList, NumQuoted, Verbose);
622224145Sdim  unsigned NumAngled = SearchList.size();
623224145Sdim
624219077Sdim  for (path_iterator it = IncludePath.begin(), ie = IncludePath.end();
625219077Sdim       it != ie; ++it) {
626249423Sdim    if (it->first == System || it->first == ExternCSystem ||
627226633Sdim        (!Lang.ObjC1 && !Lang.CPlusPlus && it->first == CSystem)    ||
628226633Sdim        (/*FIXME !Lang.ObjC1 && */Lang.CPlusPlus  && it->first == CXXSystem)  ||
629226633Sdim        (Lang.ObjC1  && !Lang.CPlusPlus && it->first == ObjCSystem) ||
630226633Sdim        (Lang.ObjC1  && Lang.CPlusPlus  && it->first == ObjCXXSystem))
631219077Sdim      SearchList.push_back(it->second);
632219077Sdim  }
633219077Sdim
634219077Sdim  for (path_iterator it = IncludePath.begin(), ie = IncludePath.end();
635219077Sdim       it != ie; ++it) {
636219077Sdim    if (it->first == After)
637219077Sdim      SearchList.push_back(it->second);
638219077Sdim  }
639219077Sdim
640224145Sdim  // Remove duplicates across both the Angled and System directories.  GCC does
641224145Sdim  // this and failing to remove duplicates across these two groups breaks
642224145Sdim  // #include_next.
643226633Sdim  unsigned NonSystemRemoved = RemoveDuplicates(SearchList, NumQuoted, Verbose);
644226633Sdim  NumAngled -= NonSystemRemoved;
645219077Sdim
646193326Sed  bool DontSearchCurDir = false;  // TODO: set to true if -I- is set?
647224145Sdim  Headers.SetSearchPaths(SearchList, NumQuoted, NumAngled, DontSearchCurDir);
648193326Sed
649239462Sdim  Headers.SetSystemHeaderPrefixes(SystemHeaderPrefixes);
650239462Sdim
651193326Sed  // If verbose, print the list of directories that will be searched.
652193326Sed  if (Verbose) {
653200583Srdivacky    llvm::errs() << "#include \"...\" search starts here:\n";
654193326Sed    for (unsigned i = 0, e = SearchList.size(); i != e; ++i) {
655224145Sdim      if (i == NumQuoted)
656200583Srdivacky        llvm::errs() << "#include <...> search starts here:\n";
657193326Sed      const char *Name = SearchList[i].getName();
658193326Sed      const char *Suffix;
659193326Sed      if (SearchList[i].isNormalDir())
660193326Sed        Suffix = "";
661193326Sed      else if (SearchList[i].isFramework())
662193326Sed        Suffix = " (framework directory)";
663193326Sed      else {
664193326Sed        assert(SearchList[i].isHeaderMap() && "Unknown DirectoryLookup");
665193326Sed        Suffix = " (headermap)";
666193326Sed      }
667200583Srdivacky      llvm::errs() << " " << Name << Suffix << "\n";
668193326Sed    }
669200583Srdivacky    llvm::errs() << "End of search list.\n";
670193326Sed  }
671193326Sed}
672199482Srdivacky
673199482Srdivackyvoid clang::ApplyHeaderSearchOptions(HeaderSearch &HS,
674199482Srdivacky                                     const HeaderSearchOptions &HSOpts,
675199482Srdivacky                                     const LangOptions &Lang,
676199482Srdivacky                                     const llvm::Triple &Triple) {
677199482Srdivacky  InitHeaderSearch Init(HS, HSOpts.Verbose, HSOpts.Sysroot);
678199482Srdivacky
679199482Srdivacky  // Add the user defined entries.
680199482Srdivacky  for (unsigned i = 0, e = HSOpts.UserEntries.size(); i != e; ++i) {
681199482Srdivacky    const HeaderSearchOptions::Entry &E = HSOpts.UserEntries[i];
682249423Sdim    if (E.IgnoreSysRoot) {
683249423Sdim      Init.AddUnmappedPath(E.Path, E.Group, E.IsFramework);
684249423Sdim    } else {
685249423Sdim      Init.AddPath(E.Path, E.Group, E.IsFramework);
686249423Sdim    }
687199482Srdivacky  }
688199482Srdivacky
689226633Sdim  Init.AddDefaultIncludePaths(Lang, Triple, HSOpts);
690199482Srdivacky
691239462Sdim  for (unsigned i = 0, e = HSOpts.SystemHeaderPrefixes.size(); i != e; ++i)
692239462Sdim    Init.AddSystemHeaderPrefix(HSOpts.SystemHeaderPrefixes[i].Prefix,
693239462Sdim                               HSOpts.SystemHeaderPrefixes[i].IsSystemHeader);
694239462Sdim
695234353Sdim  if (HSOpts.UseBuiltinIncludes) {
696234353Sdim    // Set up the builtin include directory in the module map.
697263508Sdim    SmallString<128> P = StringRef(HSOpts.ResourceDir);
698263508Sdim    llvm::sys::path::append(P, "include");
699234353Sdim    if (const DirectoryEntry *Dir = HS.getFileMgr().getDirectory(P.str()))
700234353Sdim      HS.getModuleMap().setBuiltinIncludeDir(Dir);
701234353Sdim  }
702234353Sdim
703219077Sdim  Init.Realize(Lang);
704199482Srdivacky}
705