InitHeaderSearch.cpp revision ab7ae95c46f288ac52974aa60334a9575f9a850c
10fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber//===--- InitHeaderSearch.cpp - Initialize header search paths ----------*-===//
20fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber//
30fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber//                     The LLVM Compiler Infrastructure
40fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber//
50fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber// This file is distributed under the University of Illinois Open Source
60fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber// License. See LICENSE.TXT for details.
70fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber//
80fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber//===----------------------------------------------------------------------===//
90fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber//
100fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber// This file implements the InitHeaderSearch class.
110fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber//
120fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber//===----------------------------------------------------------------------===//
130fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber
142cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar#include "clang/Frontend/Utils.h"
150fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber#include "clang/Basic/FileManager.h"
160fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber#include "clang/Basic/LangOptions.h"
1763c8b77334f90472260d2f48df2742ed5067261eDaniel Dunbar#include "clang/Frontend/HeaderSearchOptions.h"
1863c8b77334f90472260d2f48df2742ed5067261eDaniel Dunbar#include "clang/Lex/HeaderSearch.h"
190fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber#include "llvm/ADT/SmallString.h"
200fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber#include "llvm/ADT/SmallPtrSet.h"
21aadd7a48453b4f58bd8e1e9eb670918ee7d6a711Rafael Espindola#include "llvm/ADT/SmallVector.h"
22f0a2f51be576089e1a84cd677c09156dae067b6aRafael Espindola#include "llvm/ADT/StringExtras.h"
232cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar#include "llvm/ADT/Triple.h"
24d57a7ef9252964bc6c8471451d7bd395b0520cb8Chris Lattner#include "llvm/Support/raw_ostream.h"
250fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber#include "llvm/System/Path.h"
26121e3c207415fb4c105d690de384d8f3d49b0f2dArgyrios Kyrtzidis#include "llvm/Config/config.h"
273daed52a57d03765223021f5f921bdc280c8f3ccChris Lattner#include <cstdio>
28620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump#ifdef _MSC_VER
29620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump  #define WIN32_LEAN_AND_MEAN 1
30620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump  #include <windows.h>
31620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump#endif
320fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weberusing namespace clang;
332cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbarusing namespace clang::frontend;
342cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar
352cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbarnamespace {
362cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar
372cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar/// InitHeaderSearch - This class makes it easier to set the search paths of
382cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar///  a HeaderSearch object. InitHeaderSearch stores several search path lists
392cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar///  internally, which can be sent to a HeaderSearch object in one swoop.
402cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbarclass InitHeaderSearch {
412cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar  std::vector<DirectoryLookup> IncludeGroup[4];
422cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar  HeaderSearch& Headers;
432cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar  bool Verbose;
442cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar  std::string isysroot;
452cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar
462cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbarpublic:
472cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar
482cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar  InitHeaderSearch(HeaderSearch &HS,
492cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar      bool verbose = false, const std::string &iSysroot = "")
502cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar    : Headers(HS), Verbose(verbose), isysroot(iSysroot) {}
512cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar
522cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar  /// AddPath - Add the specified path to the specified group list.
532cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar  void AddPath(const llvm::StringRef &Path, IncludeDirGroup Group,
542cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar               bool isCXXAware, bool isUserSupplied,
552cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar               bool isFramework, bool IgnoreSysRoot = false);
562cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar
572cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar  /// AddGnuCPlusPlusIncludePaths - Add the necessary paths to suport a gnu
582cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar  ///  libstdc++.
59ab7ae95c46f288ac52974aa60334a9575f9a850cRafael Espindola  void AddGnuCPlusPlusIncludePaths(const std::string &Base,
60ab7ae95c46f288ac52974aa60334a9575f9a850cRafael Espindola                                   const char *ArchDir,
61ab7ae95c46f288ac52974aa60334a9575f9a850cRafael Espindola                                   const char *Dir32,
622cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar                                   const char *Dir64,
632cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar                                   const llvm::Triple &triple);
642cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar
652cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar  /// AddMinGWCPlusPlusIncludePaths - Add the necessary paths to suport a MinGW
662cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar  ///  libstdc++.
672cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar  void AddMinGWCPlusPlusIncludePaths(const std::string &Base,
682cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar                                     const char *Arch,
692cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar                                     const char *Version);
702cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar
712cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar  /// AddDelimitedPaths - Add a list of paths delimited by the system PATH
722cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar  /// separator. The processing follows that of the CPATH variable for gcc.
732cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar  void AddDelimitedPaths(const char *String);
742cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar
752cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar  // AddDefaultCIncludePaths - Add paths that should always be searched.
762cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar  void AddDefaultCIncludePaths(const llvm::Triple &triple);
772cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar
782cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar  // AddDefaultCPlusPlusIncludePaths -  Add paths that should be searched when
792cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar  //  compiling c++.
802cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar  void AddDefaultCPlusPlusIncludePaths(const llvm::Triple &triple);
812cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar
822cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar  /// AddDefaultSystemIncludePaths - Adds the default system include paths so
832cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar  ///  that e.g. stdio.h is found.
842cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar  void AddDefaultSystemIncludePaths(const LangOptions &Lang,
852cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar                                    const llvm::Triple &triple);
862cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar
872cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar  /// Realize - Merges all search path lists into one list and send it to
882cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar  /// HeaderSearch.
892cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar  void Realize();
902cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar};
912cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar
922cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar}
930fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber
94458fb10ef5ba2d7b375c6c64095c1458af0a5be3Benjamin Kramervoid InitHeaderSearch::AddPath(const llvm::StringRef &Path,
95458fb10ef5ba2d7b375c6c64095c1458af0a5be3Benjamin Kramer                               IncludeDirGroup Group, bool isCXXAware,
96458fb10ef5ba2d7b375c6c64095c1458af0a5be3Benjamin Kramer                               bool isUserSupplied, bool isFramework,
97458fb10ef5ba2d7b375c6c64095c1458af0a5be3Benjamin Kramer                               bool IgnoreSysRoot) {
980fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  assert(!Path.empty() && "can't handle empty path here");
990fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  FileManager &FM = Headers.getFileMgr();
1001eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1010fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  // Compute the actual path, taking into consideration -isysroot.
1020fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  llvm::SmallString<256> MappedPath;
1031eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1040fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  // Handle isysroot.
1056858dd3fcc2d3ac6a706a0294be1d3ac51849022Chris Lattner  if (Group == System && !IgnoreSysRoot) {
1060fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber    // FIXME: Portability.  This should be a sys::Path interface, this doesn't
1070fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber    // handle things like C:\ right, nor win32 \\network\device\blah.
1080fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber    if (isysroot.size() != 1 || isysroot[0] != '/') // Add isysroot if present.
1090fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber      MappedPath.append(isysroot.begin(), isysroot.end());
1100fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  }
1111eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1120fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  MappedPath.append(Path.begin(), Path.end());
1130fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber
1140fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  // Compute the DirectoryLookup type.
1159d72851fec9e9c62570a027d42701562bbf29751Chris Lattner  SrcMgr::CharacteristicKind Type;
1160fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  if (Group == Quoted || Group == Angled)
1170b9e736308af5397f558ffc8e780c438c2fdb563Chris Lattner    Type = SrcMgr::C_User;
1180fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  else if (isCXXAware)
1190b9e736308af5397f558ffc8e780c438c2fdb563Chris Lattner    Type = SrcMgr::C_System;
1200fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  else
1210b9e736308af5397f558ffc8e780c438c2fdb563Chris Lattner    Type = SrcMgr::C_ExternCSystem;
1221eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1231eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1240fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  // If the directory exists, add it.
125458fb10ef5ba2d7b375c6c64095c1458af0a5be3Benjamin Kramer  if (const DirectoryEntry *DE = FM.getDirectory(MappedPath.str())) {
1260fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber    IncludeGroup[Group].push_back(DirectoryLookup(DE, Type, isUserSupplied,
1270fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber                                                  isFramework));
1280fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber    return;
1290fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  }
1301eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1310fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  // Check to see if this is an apple-style headermap (which are not allowed to
1320fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  // be frameworks).
1330fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  if (!isFramework) {
134458fb10ef5ba2d7b375c6c64095c1458af0a5be3Benjamin Kramer    if (const FileEntry *FE = FM.getFile(MappedPath.str())) {
1350fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber      if (const HeaderMap *HM = Headers.CreateHeaderMap(FE)) {
1360fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber        // It is a headermap, add it to the search path.
1370fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber        IncludeGroup[Group].push_back(DirectoryLookup(HM, Type,isUserSupplied));
1380fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber        return;
1390fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber      }
1400fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber    }
1410fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  }
1421eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1430fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  if (Verbose)
1447765934ad7e157b5fcf925792a38e01b1edbcf8aDaniel Dunbar    llvm::errs() << "ignoring nonexistent directory \""
1457765934ad7e157b5fcf925792a38e01b1edbcf8aDaniel Dunbar                 << MappedPath.str() << "\"\n";
1460fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber}
1470fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber
1480fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber
149e166582f8f36f4db8f4ea157538fab7fe6bf2658Daniel Dunbarvoid InitHeaderSearch::AddDelimitedPaths(const char *at) {
150e166582f8f36f4db8f4ea157538fab7fe6bf2658Daniel Dunbar  if (*at == 0) // Empty string should not add '.' path.
1510fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber    return;
1520fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber
1530fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  const char* delim = strchr(at, llvm::sys::PathSeparator);
1540fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  while (delim != 0) {
1550fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber    if (delim-at == 0)
1560fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber      AddPath(".", Angled, false, true, false);
1570fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber    else
158458fb10ef5ba2d7b375c6c64095c1458af0a5be3Benjamin Kramer      AddPath(llvm::StringRef(at, delim-at), Angled, false, true, false);
1590fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber    at = delim + 1;
1600fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber    delim = strchr(at, llvm::sys::PathSeparator);
1610fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  }
1620fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  if (*at == 0)
1630fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber    AddPath(".", Angled, false, true, false);
1640fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  else
1650fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber    AddPath(at, Angled, false, true, false);
1660fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber}
1670fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber
168ec05766fe525a8f81c7dc96ddbdc874814a14b2cMike Stumpvoid InitHeaderSearch::AddGnuCPlusPlusIncludePaths(const std::string &Base,
169ab7ae95c46f288ac52974aa60334a9575f9a850cRafael Espindola                                                   const char *ArchDir,
17031b63beefa058aa42a12a1c2b72b43a4ba21f7f1Rafael Espindola                                                   const char *Dir32,
17131b63beefa058aa42a12a1c2b72b43a4ba21f7f1Rafael Espindola                                                   const char *Dir64,
17231b63beefa058aa42a12a1c2b72b43a4ba21f7f1Rafael Espindola                                                   const llvm::Triple &triple) {
173ab7ae95c46f288ac52974aa60334a9575f9a850cRafael Espindola  // Add the common dirs
174ab7ae95c46f288ac52974aa60334a9575f9a850cRafael Espindola  AddPath(Base, System, true, false, false);
175ab7ae95c46f288ac52974aa60334a9575f9a850cRafael Espindola  AddPath(Base + "/backward", System, true, false, false);
176ab7ae95c46f288ac52974aa60334a9575f9a850cRafael Espindola
177ab7ae95c46f288ac52974aa60334a9575f9a850cRafael Espindola  // Add the multilib dirs
17831b63beefa058aa42a12a1c2b72b43a4ba21f7f1Rafael Espindola  llvm::Triple::ArchType arch = triple.getArch();
17931b63beefa058aa42a12a1c2b72b43a4ba21f7f1Rafael Espindola  bool is64bit = arch == llvm::Triple::ppc64 || arch == llvm::Triple::x86_64;
18031b63beefa058aa42a12a1c2b72b43a4ba21f7f1Rafael Espindola  if (is64bit)
181ab7ae95c46f288ac52974aa60334a9575f9a850cRafael Espindola    AddPath(Base + "/" + ArchDir + "/" + Dir64, System, true, false, false);
18231b63beefa058aa42a12a1c2b72b43a4ba21f7f1Rafael Espindola  else
183ab7ae95c46f288ac52974aa60334a9575f9a850cRafael Espindola    AddPath(Base + "/" + ArchDir + "/" + Dir32, System, true, false, false);
1842e9f652d53346bf7e64c8a12a9ff06b004a3e489Rafael Espindola}
1850fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber
186620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stumpvoid InitHeaderSearch::AddMinGWCPlusPlusIncludePaths(const std::string &Base,
187620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump                                                     const char *Arch,
188620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump                                                     const char *Version) {
1895c5758bb0a4ec0f572cd20249277736da5715003Daniel Dunbar  std::string localBase = Base + "/" + Arch + "/" + Version + "/include";
1905c5758bb0a4ec0f572cd20249277736da5715003Daniel Dunbar  AddPath(localBase, System, true, false, false);
1915c5758bb0a4ec0f572cd20249277736da5715003Daniel Dunbar  AddPath(localBase + "/c++", System, true, false, false);
1925c5758bb0a4ec0f572cd20249277736da5715003Daniel Dunbar  AddPath(localBase + "/c++/backward", System, true, false, false);
193620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump}
19443d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump
195620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump  // FIXME: This probably should goto to some platform utils place.
196620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump#ifdef _MSC_VER
197620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump  // Read registry string.
198620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stumpbool getSystemRegistryString(const char *keyPath, const char *valueName,
19943d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump                       char *value, size_t maxLength) {
20043d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  HKEY hRootKey = NULL;
20143d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  HKEY hKey = NULL;
20243d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  const char* subKey = NULL;
20343d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  DWORD valueType;
20443d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  DWORD valueSize = maxLength - 1;
20543d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  bool returnValue = false;
20643d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  if (strncmp(keyPath, "HKEY_CLASSES_ROOT\\", 18) == 0) {
20743d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump    hRootKey = HKEY_CLASSES_ROOT;
20843d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump    subKey = keyPath + 18;
20943d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  }
21043d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  else if (strncmp(keyPath, "HKEY_USERS\\", 11) == 0) {
21143d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump    hRootKey = HKEY_USERS;
21243d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump    subKey = keyPath + 11;
21343d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  }
21443d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  else if (strncmp(keyPath, "HKEY_LOCAL_MACHINE\\", 19) == 0) {
21543d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump    hRootKey = HKEY_LOCAL_MACHINE;
21643d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump    subKey = keyPath + 19;
217121e3c207415fb4c105d690de384d8f3d49b0f2dArgyrios Kyrtzidis  }
21843d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  else if (strncmp(keyPath, "HKEY_CURRENT_USER\\", 18) == 0) {
21943d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump    hRootKey = HKEY_CURRENT_USER;
22043d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump    subKey = keyPath + 18;
22143d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  }
22243d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  else
22343d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump    return(false);
22443d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  long lResult = RegOpenKeyEx(hRootKey, subKey, 0, KEY_READ, &hKey);
22543d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  if (lResult == ERROR_SUCCESS) {
226620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump    lResult = RegQueryValueEx(hKey, valueName, NULL, &valueType,
227620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump      (LPBYTE)value, &valueSize);
22843d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump    if (lResult == ERROR_SUCCESS)
22943d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump      returnValue = true;
230620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump    RegCloseKey(hKey);
23143d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  }
23243d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  return(returnValue);
23343d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump}
234620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump#else // _MSC_VER
235620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump  // Read registry string.
236620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stumpbool getSystemRegistryString(const char *, const char *, char *, size_t) {
237620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump  return(false);
238620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump}
239620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump#endif // _MSC_VER
240121e3c207415fb4c105d690de384d8f3d49b0f2dArgyrios Kyrtzidis
24143d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  // Get Visual Studio installation directory.
24243d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stumpbool getVisualStudioDir(std::string &path) {
243620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump  // Try the Windows registry first.
244620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump  char vs80IDEInstallDir[256];
245620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump  char vs90IDEInstallDir[256];
246620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump  const char* vsIDEInstallDir = NULL;
247620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump  bool has80 = getSystemRegistryString(
24843d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump    "HKEY_LOCAL_MACHINE\\SOFTWARE\\Microsoft\\VisualStudio\\8.0",
249620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump    "InstallDir", vs80IDEInstallDir, sizeof(vs80IDEInstallDir) - 1);
250620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump  bool has90 = getSystemRegistryString(
25143d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump    "HKEY_LOCAL_MACHINE\\SOFTWARE\\Microsoft\\VisualStudio\\9.0",
252620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump    "InstallDir", vs90IDEInstallDir, sizeof(vs90IDEInstallDir) - 1);
25343d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump    // If we have both vc80 and vc90, pick version we were compiled with.
25443d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  if (has80 && has90) {
25543d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump    #ifdef _MSC_VER
25643d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump      #if (_MSC_VER >= 1500)  // VC90
257620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump          vsIDEInstallDir = vs90IDEInstallDir;
25843d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump      #elif (_MSC_VER == 1400) // VC80
259620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump          vsIDEInstallDir = vs80IDEInstallDir;
26043d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump      #else
261620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump          vsIDEInstallDir = vs90IDEInstallDir;
26243d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump      #endif
26343d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump    #else
264620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump      vsIDEInstallDir = vs90IDEInstallDir;
26543d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump    #endif
26643d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  }
26743d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  else if (has90)
268620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump    vsIDEInstallDir = vs90IDEInstallDir;
26943d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  else if (has80)
270620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump    vsIDEInstallDir = vs80IDEInstallDir;
271620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump  if (vsIDEInstallDir && *vsIDEInstallDir) {
272620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump    char *p = (char*)strstr(vsIDEInstallDir, "\\Common7\\IDE");
273620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump    if (p)
274620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump      *p = '\0';
275620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump    path = vsIDEInstallDir;
276620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump    return(true);
27743d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  }
278620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump  else {
279620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump    // Try the environment.
280620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump    const char* vs90comntools = getenv("VS90COMNTOOLS");
281620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump    const char* vs80comntools = getenv("VS80COMNTOOLS");
282620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump    const char* vscomntools = NULL;
283620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump      // If we have both vc80 and vc90, pick version we were compiled with.
284620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump    if (vs90comntools && vs80comntools) {
285620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump      #if (_MSC_VER >= 1500)  // VC90
286620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump          vscomntools = vs90comntools;
287620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump      #elif (_MSC_VER == 1400) // VC80
288620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump          vscomntools = vs80comntools;
289620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump      #else
290620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump          vscomntools = vs90comntools;
291620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump      #endif
292620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump    }
293620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump    else if (vs90comntools)
294620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump      vscomntools = vs90comntools;
295620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump    else if (vs80comntools)
296620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump      vscomntools = vs80comntools;
297620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump    if (vscomntools && *vscomntools) {
298620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump      char *p = (char*)strstr(vscomntools, "\\Common7\\Tools");
299620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump      if (p)
300620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump        *p = '\0';
301620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump      path = vscomntools;
302620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump      return(true);
303620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump    }
304620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump    else
305620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump      return(false);
306620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump  }
307620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump  return(false);
30843d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump}
3094d374d4a4832a01d5d3b75c59cb1567e60df41e7Sebastian Redl
310e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindolavoid InitHeaderSearch::AddDefaultCIncludePaths(const llvm::Triple &triple) {
31143d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  // FIXME: temporary hack: hard-coded paths.
312c706468b1e00bab49ca92eb500a39657f530f828Daniel Dunbar  llvm::StringRef CIncludeDirs(C_INCLUDE_DIRS);
313c706468b1e00bab49ca92eb500a39657f530f828Daniel Dunbar  if (CIncludeDirs != "") {
314aadd7a48453b4f58bd8e1e9eb670918ee7d6a711Rafael Espindola    llvm::SmallVector<llvm::StringRef, 5> dirs;
315aadd7a48453b4f58bd8e1e9eb670918ee7d6a711Rafael Espindola    CIncludeDirs.split(dirs, ":");
316aadd7a48453b4f58bd8e1e9eb670918ee7d6a711Rafael Espindola    for (llvm::SmallVectorImpl<llvm::StringRef>::iterator i = dirs.begin();
317aadd7a48453b4f58bd8e1e9eb670918ee7d6a711Rafael Espindola         i != dirs.end();
318aadd7a48453b4f58bd8e1e9eb670918ee7d6a711Rafael Espindola         ++i)
319f0a2f51be576089e1a84cd677c09156dae067b6aRafael Espindola      AddPath(*i, System, false, false, false);
320f0a2f51be576089e1a84cd677c09156dae067b6aRafael Espindola    return;
321f0a2f51be576089e1a84cd677c09156dae067b6aRafael Espindola  }
32243d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  llvm::Triple::OSType os = triple.getOS();
32343d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  switch (os) {
32443d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  case llvm::Triple::Win32:
32543d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump    {
326620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump      std::string VSDir;
327620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump      if (getVisualStudioDir(VSDir)) {
328620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump        AddPath(VSDir + "\\VC\\include", System, false, false, false);
329620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump        AddPath(VSDir + "\\VC\\PlatformSDK\\Include",
330620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump          System, false, false, false);
331620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump      }
332620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump      else {
33343d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump          // Default install paths.
334620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump        AddPath("C:/Program Files/Microsoft Visual Studio 9.0/VC/include",
33543d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump          System, false, false, false);
336620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump        AddPath(
337620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump        "C:/Program Files/Microsoft Visual Studio 9.0/VC/PlatformSDK/Include",
33843d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump          System, false, false, false);
339620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump        AddPath("C:/Program Files/Microsoft Visual Studio 8/VC/include",
340620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump          System, false, false, false);
341620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump        AddPath(
342620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump        "C:/Program Files/Microsoft Visual Studio 8/VC/PlatformSDK/Include",
343620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump          System, false, false, false);
344620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump          // For some clang developers.
345620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump        AddPath("G:/Program Files/Microsoft Visual Studio 9.0/VC/include",
346620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump          System, false, false, false);
347620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump        AddPath(
348620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump        "G:/Program Files/Microsoft Visual Studio 9.0/VC/PlatformSDK/Include",
349620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump          System, false, false, false);
350620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump      }
35143d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump    }
35243d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump    break;
35343d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  case llvm::Triple::MinGW64:
354620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump  case llvm::Triple::MinGW32:
35543d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump    AddPath("c:/mingw/include", System, true, false, false);
35643d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump    break;
35743d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  default:
35843d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump    break;
3590fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  }
360d3f88343af2dec3459493fafa22532317321eaf0John Thompson
361d3f88343af2dec3459493fafa22532317321eaf0John Thompson  AddPath("/usr/local/include", System, false, false, false);
362d3f88343af2dec3459493fafa22532317321eaf0John Thompson  AddPath("/usr/include", System, false, false, false);
363e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola}
364e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola
365e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindolavoid InitHeaderSearch::AddDefaultCPlusPlusIncludePaths(const llvm::Triple &triple) {
366e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola  llvm::Triple::OSType os = triple.getOS();
367ab7ae95c46f288ac52974aa60334a9575f9a850cRafael Espindola  llvm::StringRef CxxIncludeRoot(CXX_INCLUDE_ROOT);
368ab7ae95c46f288ac52974aa60334a9575f9a850cRafael Espindola  if (CxxIncludeRoot != "") {
369ab7ae95c46f288ac52974aa60334a9575f9a850cRafael Espindola    llvm::StringRef CxxIncludeArch(CXX_INCLUDE_ARCH);
370ab7ae95c46f288ac52974aa60334a9575f9a850cRafael Espindola    if (CxxIncludeArch == "")
371ab7ae95c46f288ac52974aa60334a9575f9a850cRafael Espindola      AddGnuCPlusPlusIncludePaths(CxxIncludeRoot, triple.str().c_str(),
372ab7ae95c46f288ac52974aa60334a9575f9a850cRafael Espindola                                  CXX_INCLUDE_32BIT_DIR, CXX_INCLUDE_64BIT_DIR, triple);
373ab7ae95c46f288ac52974aa60334a9575f9a850cRafael Espindola    else
374ab7ae95c46f288ac52974aa60334a9575f9a850cRafael Espindola      AddGnuCPlusPlusIncludePaths(CxxIncludeRoot, CXX_INCLUDE_ARCH,
375ab7ae95c46f288ac52974aa60334a9575f9a850cRafael Espindola                                  CXX_INCLUDE_32BIT_DIR, CXX_INCLUDE_64BIT_DIR, triple);
376ab7ae95c46f288ac52974aa60334a9575f9a850cRafael Espindola    return;
377ab7ae95c46f288ac52974aa60334a9575f9a850cRafael Espindola  }
378e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola  // FIXME: temporary hack: hard-coded paths.
379e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola  switch (os) {
380e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola  case llvm::Triple::Cygwin:
381e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    AddPath("/lib/gcc/i686-pc-cygwin/3.4.4/include",
382e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola        System, true, false, false);
383e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    AddPath("/lib/gcc/i686-pc-cygwin/3.4.4/include/c++",
384e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola        System, true, false, false);
385e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    break;
386e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola  case llvm::Triple::MinGW64:
387e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    // Try gcc 4.4.0
388e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    AddMinGWCPlusPlusIncludePaths("c:/MinGW/lib/gcc", "mingw64", "4.4.0");
389e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    // Try gcc 4.3.0
390e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    AddMinGWCPlusPlusIncludePaths("c:/MinGW/lib/gcc", "mingw64", "4.3.0");
391e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    // Fall through.
392e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola  case llvm::Triple::MinGW32:
393e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    // Try gcc 4.4.0
394e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    AddMinGWCPlusPlusIncludePaths("c:/MinGW/lib/gcc", "mingw32", "4.4.0");
395e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    // Try gcc 4.3.0
396e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    AddMinGWCPlusPlusIncludePaths("c:/MinGW/lib/gcc", "mingw32", "4.3.0");
397e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    break;
398e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola  case llvm::Triple::Darwin:
399e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.2.1",
400ab7ae95c46f288ac52974aa60334a9575f9a850cRafael Espindola                                "i686-apple-darwin10", "", "x86_64", triple);
401e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.0.0",
402ab7ae95c46f288ac52974aa60334a9575f9a850cRafael Espindola                                "i686-apple-darwin8", "", "", triple);
403e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    break;
404e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola  case llvm::Triple::Linux:
405e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    // Ubuntu 7.10 - Gutsy Gibbon
406e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.1.3",
407ab7ae95c46f288ac52974aa60334a9575f9a850cRafael Espindola                                "i486-linux-gnu", "", "", triple);
408e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    // Ubuntu 9.04
409e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.3.3",
410ab7ae95c46f288ac52974aa60334a9575f9a850cRafael Espindola                                "x86_64-linux-gnu","32", "", triple);
4115114ecab149ab91d13d3f7fd165dbe0839c1f2e8Sebastian Redl    // Ubuntu 9.10
4125114ecab149ab91d13d3f7fd165dbe0839c1f2e8Sebastian Redl    AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.4.1",
413ab7ae95c46f288ac52974aa60334a9575f9a850cRafael Espindola                                "x86_64-linux-gnu", "32", "", triple);
414e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    // Fedora 8
415e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.1.2",
416ab7ae95c46f288ac52974aa60334a9575f9a850cRafael Espindola                                "i386-redhat-linux", "", "", triple);
417e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    // Fedora 9
418e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.3.0",
419ab7ae95c46f288ac52974aa60334a9575f9a850cRafael Espindola                                "i386-redhat-linux", "", "", triple);
420e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    // Fedora 10
421e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.3.2",
422ab7ae95c46f288ac52974aa60334a9575f9a850cRafael Espindola                                "i386-redhat-linux","", "", triple);
423e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    // openSUSE 11.1 32 bit
424e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.3",
425ab7ae95c46f288ac52974aa60334a9575f9a850cRafael Espindola                                "i586-suse-linux", "", "", triple);
426e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    // openSUSE 11.1 64 bit
427e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.3",
428ab7ae95c46f288ac52974aa60334a9575f9a850cRafael Espindola                                "x86_64-suse-linux", "32", "", triple);
429e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    // openSUSE 11.2
430e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.4",
431ab7ae95c46f288ac52974aa60334a9575f9a850cRafael Espindola                                "i586-suse-linux", "", "", triple);
432e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.4",
433ab7ae95c46f288ac52974aa60334a9575f9a850cRafael Espindola                                "x86_64-suse-linux", "", "", triple);
434e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    // Arch Linux 2008-06-24
435e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.3.1",
436ab7ae95c46f288ac52974aa60334a9575f9a850cRafael Espindola                                "i686-pc-linux-gnu", "", "", triple);
437e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.3.1",
438ab7ae95c46f288ac52974aa60334a9575f9a850cRafael Espindola                                "x86_64-unknown-linux-gnu", "", "", triple);
439e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    // Gentoo x86 2009.1 stable
440e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    AddGnuCPlusPlusIncludePaths(
44140d1bb6383c14d5f8249d0640e8088a91b694c66John Thompson      "/usr/lib/gcc/i686-pc-linux-gnu/4.3.4/include/g++-v4",
442ab7ae95c46f288ac52974aa60334a9575f9a850cRafael Espindola      "i686-pc-linux-gnu", "", "", triple);
443e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    // Gentoo x86 2009.0 stable
444e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    AddGnuCPlusPlusIncludePaths(
44540d1bb6383c14d5f8249d0640e8088a91b694c66John Thompson      "/usr/lib/gcc/i686-pc-linux-gnu/4.3.2/include/g++-v4",
446ab7ae95c46f288ac52974aa60334a9575f9a850cRafael Espindola      "i686-pc-linux-gnu", "", "", triple);
447e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    // Gentoo x86 2008.0 stable
448e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    AddGnuCPlusPlusIncludePaths(
44940d1bb6383c14d5f8249d0640e8088a91b694c66John Thompson      "/usr/lib/gcc/i686-pc-linux-gnu/4.1.2/include/g++-v4",
450ab7ae95c46f288ac52974aa60334a9575f9a850cRafael Espindola      "i686-pc-linux-gnu", "", "", triple);
451e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    // Ubuntu 8.10
452e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.3",
453ab7ae95c46f288ac52974aa60334a9575f9a850cRafael Espindola                                "i486-pc-linux-gnu", "", "", triple);
454e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    // Ubuntu 9.04
455e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.3",
456ab7ae95c46f288ac52974aa60334a9575f9a850cRafael Espindola                                "i486-linux-gnu","", "", triple);
457e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    // Gentoo amd64 stable
458e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    AddGnuCPlusPlusIncludePaths(
459e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola        "/usr/lib/gcc/x86_64-pc-linux-gnu/4.1.2/include/g++-v4",
460ab7ae95c46f288ac52974aa60334a9575f9a850cRafael Espindola        "i686-pc-linux-gnu", "", "", triple);
4615d7a188cbc66d0af5aa3c135942a9e4aa7cb69f9Benjamin Kramer    // Exherbo (2009-10-26)
462ab7ae95c46f288ac52974aa60334a9575f9a850cRafael Espindola    AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.4.2",
463ab7ae95c46f288ac52974aa60334a9575f9a850cRafael Espindola                                "x86_64-pc-linux-gnu", "32", "", triple);
464ab7ae95c46f288ac52974aa60334a9575f9a850cRafael Espindola    AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.4.2",
465ab7ae95c46f288ac52974aa60334a9575f9a850cRafael Espindola                                "i686-pc-linux-gnu", "", "", triple);
466e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    break;
467e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola  case llvm::Triple::FreeBSD:
468e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    // DragonFly
469e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    AddPath("/usr/include/c++/4.1", System, true, false, false);
470e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    // FreeBSD
471e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    AddPath("/usr/include/c++/4.2", System, true, false, false);
472e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    break;
473e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola  case llvm::Triple::Solaris:
474e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    // Solaris - Fall though..
475e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola  case llvm::Triple::AuroraUX:
476e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    // AuroraUX
477e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    AddGnuCPlusPlusIncludePaths("/opt/gcc4/include/c++/4.2.4",
478ab7ae95c46f288ac52974aa60334a9575f9a850cRafael Espindola                                "i386-pc-solaris2.11", "", "", triple);
479e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    break;
480e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola  default:
481e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    break;
482e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola  }
483e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola}
484e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola
485e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindolavoid InitHeaderSearch::AddDefaultSystemIncludePaths(const LangOptions &Lang,
486e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola                                                    const llvm::Triple &triple) {
487e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola  AddDefaultCIncludePaths(triple);
488e166582f8f36f4db8f4ea157538fab7fe6bf2658Daniel Dunbar
489e166582f8f36f4db8f4ea157538fab7fe6bf2658Daniel Dunbar  // Add the default framework include paths on Darwin.
490e166582f8f36f4db8f4ea157538fab7fe6bf2658Daniel Dunbar  if (triple.getOS() == llvm::Triple::Darwin) {
491e166582f8f36f4db8f4ea157538fab7fe6bf2658Daniel Dunbar    AddPath("/System/Library/Frameworks", System, true, false, true);
492e166582f8f36f4db8f4ea157538fab7fe6bf2658Daniel Dunbar    AddPath("/Library/Frameworks", System, true, false, true);
493e166582f8f36f4db8f4ea157538fab7fe6bf2658Daniel Dunbar  }
494e166582f8f36f4db8f4ea157538fab7fe6bf2658Daniel Dunbar
495e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola  if (Lang.CPlusPlus)
496e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    AddDefaultCPlusPlusIncludePaths(triple);
497e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola}
498e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola
4990fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber/// RemoveDuplicates - If there are duplicate directory entries in the specified
5000fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber/// search list, remove the later (dead) ones.
5010fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weberstatic void RemoveDuplicates(std::vector<DirectoryLookup> &SearchList,
5020fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber                             bool Verbose) {
5030fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  llvm::SmallPtrSet<const DirectoryEntry *, 8> SeenDirs;
5040fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  llvm::SmallPtrSet<const DirectoryEntry *, 8> SeenFrameworkDirs;
5050fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  llvm::SmallPtrSet<const HeaderMap *, 8> SeenHeaderMaps;
5060fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  for (unsigned i = 0; i != SearchList.size(); ++i) {
5077a739401983eb9bc25c9b4b25177e116f5e5ba92Chris Lattner    unsigned DirToRemove = i;
5081eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
50943eee07270bf1966ea7289310066aa670c4b647fChris Lattner    const DirectoryLookup &CurEntry = SearchList[i];
5101eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
51143eee07270bf1966ea7289310066aa670c4b647fChris Lattner    if (CurEntry.isNormalDir()) {
5120fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber      // If this isn't the first time we've seen this dir, remove it.
51343eee07270bf1966ea7289310066aa670c4b647fChris Lattner      if (SeenDirs.insert(CurEntry.getDir()))
5140fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber        continue;
51543eee07270bf1966ea7289310066aa670c4b647fChris Lattner    } else if (CurEntry.isFramework()) {
5160fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber      // If this isn't the first time we've seen this framework dir, remove it.
51743eee07270bf1966ea7289310066aa670c4b647fChris Lattner      if (SeenFrameworkDirs.insert(CurEntry.getFrameworkDir()))
5180fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber        continue;
5190fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber    } else {
52043eee07270bf1966ea7289310066aa670c4b647fChris Lattner      assert(CurEntry.isHeaderMap() && "Not a headermap or normal dir?");
5210fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber      // If this isn't the first time we've seen this headermap, remove it.
52243eee07270bf1966ea7289310066aa670c4b647fChris Lattner      if (SeenHeaderMaps.insert(CurEntry.getHeaderMap()))
5230fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber        continue;
52430f05b553db067c994966daca37e75324ee7b424Chris Lattner    }
5251eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
52630f05b553db067c994966daca37e75324ee7b424Chris Lattner    // If we have a normal #include dir/framework/headermap that is shadowed
52730f05b553db067c994966daca37e75324ee7b424Chris Lattner    // later in the chain by a system include location, we actually want to
52830f05b553db067c994966daca37e75324ee7b424Chris Lattner    // ignore the user's request and drop the user dir... keeping the system
52930f05b553db067c994966daca37e75324ee7b424Chris Lattner    // dir.  This is weird, but required to emulate GCC's search path correctly.
53030f05b553db067c994966daca37e75324ee7b424Chris Lattner    //
53130f05b553db067c994966daca37e75324ee7b424Chris Lattner    // Since dupes of system dirs are rare, just rescan to find the original
53230f05b553db067c994966daca37e75324ee7b424Chris Lattner    // that we're nuking instead of using a DenseMap.
53343eee07270bf1966ea7289310066aa670c4b647fChris Lattner    if (CurEntry.getDirCharacteristic() != SrcMgr::C_User) {
53430f05b553db067c994966daca37e75324ee7b424Chris Lattner      // Find the dir that this is the same of.
53530f05b553db067c994966daca37e75324ee7b424Chris Lattner      unsigned FirstDir;
53630f05b553db067c994966daca37e75324ee7b424Chris Lattner      for (FirstDir = 0; ; ++FirstDir) {
53730f05b553db067c994966daca37e75324ee7b424Chris Lattner        assert(FirstDir != i && "Didn't find dupe?");
5381eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
53943eee07270bf1966ea7289310066aa670c4b647fChris Lattner        const DirectoryLookup &SearchEntry = SearchList[FirstDir];
54043eee07270bf1966ea7289310066aa670c4b647fChris Lattner
54130f05b553db067c994966daca37e75324ee7b424Chris Lattner        // If these are different lookup types, then they can't be the dupe.
54243eee07270bf1966ea7289310066aa670c4b647fChris Lattner        if (SearchEntry.getLookupType() != CurEntry.getLookupType())
54330f05b553db067c994966daca37e75324ee7b424Chris Lattner          continue;
5441eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
54530f05b553db067c994966daca37e75324ee7b424Chris Lattner        bool isSame;
54643eee07270bf1966ea7289310066aa670c4b647fChris Lattner        if (CurEntry.isNormalDir())
54743eee07270bf1966ea7289310066aa670c4b647fChris Lattner          isSame = SearchEntry.getDir() == CurEntry.getDir();
54843eee07270bf1966ea7289310066aa670c4b647fChris Lattner        else if (CurEntry.isFramework())
54943eee07270bf1966ea7289310066aa670c4b647fChris Lattner          isSame = SearchEntry.getFrameworkDir() == CurEntry.getFrameworkDir();
55030f05b553db067c994966daca37e75324ee7b424Chris Lattner        else {
55143eee07270bf1966ea7289310066aa670c4b647fChris Lattner          assert(CurEntry.isHeaderMap() && "Not a headermap or normal dir?");
55243eee07270bf1966ea7289310066aa670c4b647fChris Lattner          isSame = SearchEntry.getHeaderMap() == CurEntry.getHeaderMap();
55330f05b553db067c994966daca37e75324ee7b424Chris Lattner        }
5541eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
55530f05b553db067c994966daca37e75324ee7b424Chris Lattner        if (isSame)
55630f05b553db067c994966daca37e75324ee7b424Chris Lattner          break;
55730f05b553db067c994966daca37e75324ee7b424Chris Lattner      }
5581eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
55930f05b553db067c994966daca37e75324ee7b424Chris Lattner      // If the first dir in the search path is a non-system dir, zap it
56030f05b553db067c994966daca37e75324ee7b424Chris Lattner      // instead of the system one.
56130f05b553db067c994966daca37e75324ee7b424Chris Lattner      if (SearchList[FirstDir].getDirCharacteristic() == SrcMgr::C_User)
56230f05b553db067c994966daca37e75324ee7b424Chris Lattner        DirToRemove = FirstDir;
56330f05b553db067c994966daca37e75324ee7b424Chris Lattner    }
56430f05b553db067c994966daca37e75324ee7b424Chris Lattner
56530f05b553db067c994966daca37e75324ee7b424Chris Lattner    if (Verbose) {
56643eee07270bf1966ea7289310066aa670c4b647fChris Lattner      fprintf(stderr, "ignoring duplicate directory \"%s\"\n",
56743eee07270bf1966ea7289310066aa670c4b647fChris Lattner              CurEntry.getName());
56830f05b553db067c994966daca37e75324ee7b424Chris Lattner      if (DirToRemove != i)
56930f05b553db067c994966daca37e75324ee7b424Chris Lattner        fprintf(stderr, "  as it is a non-system directory that duplicates"
57030f05b553db067c994966daca37e75324ee7b424Chris Lattner                " a system directory\n");
5710fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber    }
5721eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
5737a739401983eb9bc25c9b4b25177e116f5e5ba92Chris Lattner    // This is reached if the current entry is a duplicate.  Remove the
5747a739401983eb9bc25c9b4b25177e116f5e5ba92Chris Lattner    // DirToRemove (usually the current dir).
5757a739401983eb9bc25c9b4b25177e116f5e5ba92Chris Lattner    SearchList.erase(SearchList.begin()+DirToRemove);
5760fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber    --i;
5770fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  }
5780fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber}
5790fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber
5800fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber
5810fca022d77b89100e746f4d659b84ed5b1ee0158Nico Webervoid InitHeaderSearch::Realize() {
5820fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  // Concatenate ANGLE+SYSTEM+AFTER chains together into SearchList.
5830fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  std::vector<DirectoryLookup> SearchList;
5840fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  SearchList = IncludeGroup[Angled];
5850fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  SearchList.insert(SearchList.end(), IncludeGroup[System].begin(),
5860fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber                    IncludeGroup[System].end());
5870fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  SearchList.insert(SearchList.end(), IncludeGroup[After].begin(),
5880fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber                    IncludeGroup[After].end());
5890fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  RemoveDuplicates(SearchList, Verbose);
5900fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  RemoveDuplicates(IncludeGroup[Quoted], Verbose);
5911eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
5920fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  // Prepend QUOTED list on the search list.
5931eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  SearchList.insert(SearchList.begin(), IncludeGroup[Quoted].begin(),
5940fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber                    IncludeGroup[Quoted].end());
5951eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
5960fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber
5970fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  bool DontSearchCurDir = false;  // TODO: set to true if -I- is set?
5980fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  Headers.SetSearchPaths(SearchList, IncludeGroup[Quoted].size(),
5990fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber                         DontSearchCurDir);
6000fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber
6010fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  // If verbose, print the list of directories that will be searched.
6020fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  if (Verbose) {
6030fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber    fprintf(stderr, "#include \"...\" search starts here:\n");
6040fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber    unsigned QuotedIdx = IncludeGroup[Quoted].size();
6050fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber    for (unsigned i = 0, e = SearchList.size(); i != e; ++i) {
6060fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber      if (i == QuotedIdx)
6070fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber        fprintf(stderr, "#include <...> search starts here:\n");
6080fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber      const char *Name = SearchList[i].getName();
6090fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber      const char *Suffix;
6100fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber      if (SearchList[i].isNormalDir())
6110fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber        Suffix = "";
6120fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber      else if (SearchList[i].isFramework())
6130fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber        Suffix = " (framework directory)";
6140fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber      else {
6150fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber        assert(SearchList[i].isHeaderMap() && "Unknown DirectoryLookup");
6160fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber        Suffix = " (headermap)";
6170fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber      }
6180fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber      fprintf(stderr, " %s%s\n", Name, Suffix);
6190fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber    }
6200fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber    fprintf(stderr, "End of search list.\n");
6210fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  }
6220fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber}
62363c8b77334f90472260d2f48df2742ed5067261eDaniel Dunbar
6245814e657c9ad9ef6049a2a4af0d2aad248a8a15cDaniel Dunbarvoid clang::ApplyHeaderSearchOptions(HeaderSearch &HS,
6255814e657c9ad9ef6049a2a4af0d2aad248a8a15cDaniel Dunbar                                     const HeaderSearchOptions &HSOpts,
6265814e657c9ad9ef6049a2a4af0d2aad248a8a15cDaniel Dunbar                                     const LangOptions &Lang,
62763c8b77334f90472260d2f48df2742ed5067261eDaniel Dunbar                                     const llvm::Triple &Triple) {
62863c8b77334f90472260d2f48df2742ed5067261eDaniel Dunbar  InitHeaderSearch Init(HS, HSOpts.Verbose, HSOpts.Sysroot);
62963c8b77334f90472260d2f48df2742ed5067261eDaniel Dunbar
63063c8b77334f90472260d2f48df2742ed5067261eDaniel Dunbar  // Add the user defined entries.
63163c8b77334f90472260d2f48df2742ed5067261eDaniel Dunbar  for (unsigned i = 0, e = HSOpts.UserEntries.size(); i != e; ++i) {
63263c8b77334f90472260d2f48df2742ed5067261eDaniel Dunbar    const HeaderSearchOptions::Entry &E = HSOpts.UserEntries[i];
63363c8b77334f90472260d2f48df2742ed5067261eDaniel Dunbar    Init.AddPath(E.Path, E.Group, E.IsCXXAware, E.IsUserSupplied, E.IsFramework,
63463c8b77334f90472260d2f48df2742ed5067261eDaniel Dunbar                 E.IgnoreSysRoot);
63563c8b77334f90472260d2f48df2742ed5067261eDaniel Dunbar  }
63663c8b77334f90472260d2f48df2742ed5067261eDaniel Dunbar
63763c8b77334f90472260d2f48df2742ed5067261eDaniel Dunbar  // Add entries from CPATH and friends.
63863c8b77334f90472260d2f48df2742ed5067261eDaniel Dunbar  Init.AddDelimitedPaths(HSOpts.EnvIncPath.c_str());
63963c8b77334f90472260d2f48df2742ed5067261eDaniel Dunbar  Init.AddDelimitedPaths(HSOpts.LangEnvIncPath.c_str());
64063c8b77334f90472260d2f48df2742ed5067261eDaniel Dunbar
64163c8b77334f90472260d2f48df2742ed5067261eDaniel Dunbar  if (!HSOpts.BuiltinIncludePath.empty()) {
64263c8b77334f90472260d2f48df2742ed5067261eDaniel Dunbar    // Ignore the sys root, we *always* look for clang headers relative to
64363c8b77334f90472260d2f48df2742ed5067261eDaniel Dunbar    // supplied path.
6442cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar    Init.AddPath(HSOpts.BuiltinIncludePath, System,
64563c8b77334f90472260d2f48df2742ed5067261eDaniel Dunbar                 false, false, false, /*IgnoreSysRoot=*/ true);
64663c8b77334f90472260d2f48df2742ed5067261eDaniel Dunbar  }
64763c8b77334f90472260d2f48df2742ed5067261eDaniel Dunbar
648dd35ce9a783dea7e6660f9cb1d973742f3b3c73fDaniel Dunbar  if (HSOpts.UseStandardIncludes)
64963c8b77334f90472260d2f48df2742ed5067261eDaniel Dunbar    Init.AddDefaultSystemIncludePaths(Lang, Triple);
65063c8b77334f90472260d2f48df2742ed5067261eDaniel Dunbar
65163c8b77334f90472260d2f48df2742ed5067261eDaniel Dunbar  Init.Realize();
65263c8b77334f90472260d2f48df2742ed5067261eDaniel Dunbar}
653