InitHeaderSearch.cpp revision c706468b1e00bab49ca92eb500a39657f530f828
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"
21f0a2f51be576089e1a84cd677c09156dae067b6aRafael Espindola#include "llvm/ADT/StringExtras.h"
222cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar#include "llvm/ADT/Triple.h"
23d57a7ef9252964bc6c8471451d7bd395b0520cb8Chris Lattner#include "llvm/Support/raw_ostream.h"
240fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber#include "llvm/System/Path.h"
25121e3c207415fb4c105d690de384d8f3d49b0f2dArgyrios Kyrtzidis#include "llvm/Config/config.h"
263daed52a57d03765223021f5f921bdc280c8f3ccChris Lattner#include <cstdio>
27620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump#ifdef _MSC_VER
28620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump  #define WIN32_LEAN_AND_MEAN 1
29620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump  #include <windows.h>
30620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump#endif
310fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weberusing namespace clang;
322cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbarusing namespace clang::frontend;
332cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar
342cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbarnamespace {
352cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar
362cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar/// InitHeaderSearch - This class makes it easier to set the search paths of
372cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar///  a HeaderSearch object. InitHeaderSearch stores several search path lists
382cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar///  internally, which can be sent to a HeaderSearch object in one swoop.
392cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbarclass InitHeaderSearch {
402cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar  std::vector<DirectoryLookup> IncludeGroup[4];
412cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar  HeaderSearch& Headers;
422cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar  bool Verbose;
432cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar  std::string isysroot;
442cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar
452cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbarpublic:
462cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar
472cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar  InitHeaderSearch(HeaderSearch &HS,
482cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar      bool verbose = false, const std::string &iSysroot = "")
492cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar    : Headers(HS), Verbose(verbose), isysroot(iSysroot) {}
502cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar
512cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar  /// AddPath - Add the specified path to the specified group list.
522cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar  void AddPath(const llvm::StringRef &Path, IncludeDirGroup Group,
532cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar               bool isCXXAware, bool isUserSupplied,
542cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar               bool isFramework, bool IgnoreSysRoot = false);
552cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar
562cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar  /// AddGnuCPlusPlusIncludePaths - Add the necessary paths to suport a gnu
572cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar  ///  libstdc++.
582cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar  void AddGnuCPlusPlusIncludePaths(const std::string &Base, const char *Dir32,
592cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar                                   const char *Dir64,
602cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar                                   const llvm::Triple &triple);
612cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar
622cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar  /// AddMinGWCPlusPlusIncludePaths - Add the necessary paths to suport a MinGW
632cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar  ///  libstdc++.
642cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar  void AddMinGWCPlusPlusIncludePaths(const std::string &Base,
652cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar                                     const char *Arch,
662cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar                                     const char *Version);
672cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar
682cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar  /// AddDelimitedPaths - Add a list of paths delimited by the system PATH
692cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar  /// separator. The processing follows that of the CPATH variable for gcc.
702cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar  void AddDelimitedPaths(const char *String);
712cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar
722cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar  // AddDefaultCIncludePaths - Add paths that should always be searched.
732cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar  void AddDefaultCIncludePaths(const llvm::Triple &triple);
742cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar
752cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar  // AddDefaultCPlusPlusIncludePaths -  Add paths that should be searched when
762cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar  //  compiling c++.
772cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar  void AddDefaultCPlusPlusIncludePaths(const llvm::Triple &triple);
782cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar
792cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar  /// AddDefaultSystemIncludePaths - Adds the default system include paths so
802cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar  ///  that e.g. stdio.h is found.
812cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar  void AddDefaultSystemIncludePaths(const LangOptions &Lang,
822cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar                                    const llvm::Triple &triple);
832cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar
842cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar  /// Realize - Merges all search path lists into one list and send it to
852cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar  /// HeaderSearch.
862cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar  void Realize();
872cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar};
882cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar
892cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar}
900fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber
91458fb10ef5ba2d7b375c6c64095c1458af0a5be3Benjamin Kramervoid InitHeaderSearch::AddPath(const llvm::StringRef &Path,
92458fb10ef5ba2d7b375c6c64095c1458af0a5be3Benjamin Kramer                               IncludeDirGroup Group, bool isCXXAware,
93458fb10ef5ba2d7b375c6c64095c1458af0a5be3Benjamin Kramer                               bool isUserSupplied, bool isFramework,
94458fb10ef5ba2d7b375c6c64095c1458af0a5be3Benjamin Kramer                               bool IgnoreSysRoot) {
950fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  assert(!Path.empty() && "can't handle empty path here");
960fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  FileManager &FM = Headers.getFileMgr();
971eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
980fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  // Compute the actual path, taking into consideration -isysroot.
990fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  llvm::SmallString<256> MappedPath;
1001eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1010fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  // Handle isysroot.
1026858dd3fcc2d3ac6a706a0294be1d3ac51849022Chris Lattner  if (Group == System && !IgnoreSysRoot) {
1030fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber    // FIXME: Portability.  This should be a sys::Path interface, this doesn't
1040fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber    // handle things like C:\ right, nor win32 \\network\device\blah.
1050fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber    if (isysroot.size() != 1 || isysroot[0] != '/') // Add isysroot if present.
1060fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber      MappedPath.append(isysroot.begin(), isysroot.end());
1070fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  }
1081eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1090fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  MappedPath.append(Path.begin(), Path.end());
1100fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber
1110fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  // Compute the DirectoryLookup type.
1129d72851fec9e9c62570a027d42701562bbf29751Chris Lattner  SrcMgr::CharacteristicKind Type;
1130fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  if (Group == Quoted || Group == Angled)
1140b9e736308af5397f558ffc8e780c438c2fdb563Chris Lattner    Type = SrcMgr::C_User;
1150fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  else if (isCXXAware)
1160b9e736308af5397f558ffc8e780c438c2fdb563Chris Lattner    Type = SrcMgr::C_System;
1170fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  else
1180b9e736308af5397f558ffc8e780c438c2fdb563Chris Lattner    Type = SrcMgr::C_ExternCSystem;
1191eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1201eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1210fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  // If the directory exists, add it.
122458fb10ef5ba2d7b375c6c64095c1458af0a5be3Benjamin Kramer  if (const DirectoryEntry *DE = FM.getDirectory(MappedPath.str())) {
1230fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber    IncludeGroup[Group].push_back(DirectoryLookup(DE, Type, isUserSupplied,
1240fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber                                                  isFramework));
1250fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber    return;
1260fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  }
1271eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1280fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  // Check to see if this is an apple-style headermap (which are not allowed to
1290fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  // be frameworks).
1300fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  if (!isFramework) {
131458fb10ef5ba2d7b375c6c64095c1458af0a5be3Benjamin Kramer    if (const FileEntry *FE = FM.getFile(MappedPath.str())) {
1320fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber      if (const HeaderMap *HM = Headers.CreateHeaderMap(FE)) {
1330fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber        // It is a headermap, add it to the search path.
1340fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber        IncludeGroup[Group].push_back(DirectoryLookup(HM, Type,isUserSupplied));
1350fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber        return;
1360fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber      }
1370fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber    }
1380fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  }
1391eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1400fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  if (Verbose)
1417765934ad7e157b5fcf925792a38e01b1edbcf8aDaniel Dunbar    llvm::errs() << "ignoring nonexistent directory \""
1427765934ad7e157b5fcf925792a38e01b1edbcf8aDaniel Dunbar                 << MappedPath.str() << "\"\n";
1430fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber}
1440fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber
1450fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber
146e166582f8f36f4db8f4ea157538fab7fe6bf2658Daniel Dunbarvoid InitHeaderSearch::AddDelimitedPaths(const char *at) {
147e166582f8f36f4db8f4ea157538fab7fe6bf2658Daniel Dunbar  if (*at == 0) // Empty string should not add '.' path.
1480fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber    return;
1490fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber
1500fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  const char* delim = strchr(at, llvm::sys::PathSeparator);
1510fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  while (delim != 0) {
1520fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber    if (delim-at == 0)
1530fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber      AddPath(".", Angled, false, true, false);
1540fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber    else
155458fb10ef5ba2d7b375c6c64095c1458af0a5be3Benjamin Kramer      AddPath(llvm::StringRef(at, delim-at), Angled, false, true, false);
1560fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber    at = delim + 1;
1570fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber    delim = strchr(at, llvm::sys::PathSeparator);
1580fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  }
1590fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  if (*at == 0)
1600fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber    AddPath(".", Angled, false, true, false);
1610fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  else
1620fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber    AddPath(at, Angled, false, true, false);
1630fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber}
1640fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber
165ec05766fe525a8f81c7dc96ddbdc874814a14b2cMike Stumpvoid InitHeaderSearch::AddGnuCPlusPlusIncludePaths(const std::string &Base,
16631b63beefa058aa42a12a1c2b72b43a4ba21f7f1Rafael Espindola                                                   const char *Dir32,
16731b63beefa058aa42a12a1c2b72b43a4ba21f7f1Rafael Espindola                                                   const char *Dir64,
16831b63beefa058aa42a12a1c2b72b43a4ba21f7f1Rafael Espindola                                                   const llvm::Triple &triple) {
16931b63beefa058aa42a12a1c2b72b43a4ba21f7f1Rafael Espindola  llvm::Triple::ArchType arch = triple.getArch();
17031b63beefa058aa42a12a1c2b72b43a4ba21f7f1Rafael Espindola  bool is64bit = arch == llvm::Triple::ppc64 || arch == llvm::Triple::x86_64;
17131b63beefa058aa42a12a1c2b72b43a4ba21f7f1Rafael Espindola
17231b63beefa058aa42a12a1c2b72b43a4ba21f7f1Rafael Espindola  AddPath(Base, System, true, false, false);
17331b63beefa058aa42a12a1c2b72b43a4ba21f7f1Rafael Espindola  if (is64bit)
17431b63beefa058aa42a12a1c2b72b43a4ba21f7f1Rafael Espindola    AddPath(Base + "/" + Dir64, System, true, false, false);
17531b63beefa058aa42a12a1c2b72b43a4ba21f7f1Rafael Espindola  else
17631b63beefa058aa42a12a1c2b72b43a4ba21f7f1Rafael Espindola    AddPath(Base + "/" + Dir32, System, true, false, false);
17731b63beefa058aa42a12a1c2b72b43a4ba21f7f1Rafael Espindola  AddPath(Base + "/backward", System, true, false, false);
1782e9f652d53346bf7e64c8a12a9ff06b004a3e489Rafael Espindola}
1790fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber
180620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stumpvoid InitHeaderSearch::AddMinGWCPlusPlusIncludePaths(const std::string &Base,
181620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump                                                     const char *Arch,
182620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump                                                     const char *Version) {
1835c5758bb0a4ec0f572cd20249277736da5715003Daniel Dunbar  std::string localBase = Base + "/" + Arch + "/" + Version + "/include";
1845c5758bb0a4ec0f572cd20249277736da5715003Daniel Dunbar  AddPath(localBase, System, true, false, false);
1855c5758bb0a4ec0f572cd20249277736da5715003Daniel Dunbar  AddPath(localBase + "/c++", System, true, false, false);
1865c5758bb0a4ec0f572cd20249277736da5715003Daniel Dunbar  AddPath(localBase + "/c++/backward", System, true, false, false);
187620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump}
18843d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump
189620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump  // FIXME: This probably should goto to some platform utils place.
190620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump#ifdef _MSC_VER
191620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump  // Read registry string.
192620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stumpbool getSystemRegistryString(const char *keyPath, const char *valueName,
19343d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump                       char *value, size_t maxLength) {
19443d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  HKEY hRootKey = NULL;
19543d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  HKEY hKey = NULL;
19643d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  const char* subKey = NULL;
19743d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  DWORD valueType;
19843d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  DWORD valueSize = maxLength - 1;
19943d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  bool returnValue = false;
20043d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  if (strncmp(keyPath, "HKEY_CLASSES_ROOT\\", 18) == 0) {
20143d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump    hRootKey = HKEY_CLASSES_ROOT;
20243d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump    subKey = keyPath + 18;
20343d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  }
20443d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  else if (strncmp(keyPath, "HKEY_USERS\\", 11) == 0) {
20543d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump    hRootKey = HKEY_USERS;
20643d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump    subKey = keyPath + 11;
20743d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  }
20843d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  else if (strncmp(keyPath, "HKEY_LOCAL_MACHINE\\", 19) == 0) {
20943d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump    hRootKey = HKEY_LOCAL_MACHINE;
21043d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump    subKey = keyPath + 19;
211121e3c207415fb4c105d690de384d8f3d49b0f2dArgyrios Kyrtzidis  }
21243d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  else if (strncmp(keyPath, "HKEY_CURRENT_USER\\", 18) == 0) {
21343d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump    hRootKey = HKEY_CURRENT_USER;
21443d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump    subKey = keyPath + 18;
21543d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  }
21643d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  else
21743d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump    return(false);
21843d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  long lResult = RegOpenKeyEx(hRootKey, subKey, 0, KEY_READ, &hKey);
21943d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  if (lResult == ERROR_SUCCESS) {
220620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump    lResult = RegQueryValueEx(hKey, valueName, NULL, &valueType,
221620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump      (LPBYTE)value, &valueSize);
22243d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump    if (lResult == ERROR_SUCCESS)
22343d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump      returnValue = true;
224620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump    RegCloseKey(hKey);
22543d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  }
22643d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  return(returnValue);
22743d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump}
228620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump#else // _MSC_VER
229620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump  // Read registry string.
230620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stumpbool getSystemRegistryString(const char *, const char *, char *, size_t) {
231620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump  return(false);
232620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump}
233620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump#endif // _MSC_VER
234121e3c207415fb4c105d690de384d8f3d49b0f2dArgyrios Kyrtzidis
23543d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  // Get Visual Studio installation directory.
23643d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stumpbool getVisualStudioDir(std::string &path) {
237620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump  // Try the Windows registry first.
238620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump  char vs80IDEInstallDir[256];
239620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump  char vs90IDEInstallDir[256];
240620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump  const char* vsIDEInstallDir = NULL;
241620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump  bool has80 = getSystemRegistryString(
24243d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump    "HKEY_LOCAL_MACHINE\\SOFTWARE\\Microsoft\\VisualStudio\\8.0",
243620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump    "InstallDir", vs80IDEInstallDir, sizeof(vs80IDEInstallDir) - 1);
244620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump  bool has90 = getSystemRegistryString(
24543d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump    "HKEY_LOCAL_MACHINE\\SOFTWARE\\Microsoft\\VisualStudio\\9.0",
246620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump    "InstallDir", vs90IDEInstallDir, sizeof(vs90IDEInstallDir) - 1);
24743d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump    // If we have both vc80 and vc90, pick version we were compiled with.
24843d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  if (has80 && has90) {
24943d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump    #ifdef _MSC_VER
25043d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump      #if (_MSC_VER >= 1500)  // VC90
251620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump          vsIDEInstallDir = vs90IDEInstallDir;
25243d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump      #elif (_MSC_VER == 1400) // VC80
253620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump          vsIDEInstallDir = vs80IDEInstallDir;
25443d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump      #else
255620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump          vsIDEInstallDir = vs90IDEInstallDir;
25643d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump      #endif
25743d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump    #else
258620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump      vsIDEInstallDir = vs90IDEInstallDir;
25943d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump    #endif
26043d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  }
26143d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  else if (has90)
262620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump    vsIDEInstallDir = vs90IDEInstallDir;
26343d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  else if (has80)
264620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump    vsIDEInstallDir = vs80IDEInstallDir;
265620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump  if (vsIDEInstallDir && *vsIDEInstallDir) {
266620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump    char *p = (char*)strstr(vsIDEInstallDir, "\\Common7\\IDE");
267620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump    if (p)
268620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump      *p = '\0';
269620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump    path = vsIDEInstallDir;
270620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump    return(true);
27143d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  }
272620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump  else {
273620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump    // Try the environment.
274620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump    const char* vs90comntools = getenv("VS90COMNTOOLS");
275620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump    const char* vs80comntools = getenv("VS80COMNTOOLS");
276620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump    const char* vscomntools = NULL;
277620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump      // If we have both vc80 and vc90, pick version we were compiled with.
278620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump    if (vs90comntools && vs80comntools) {
279620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump      #if (_MSC_VER >= 1500)  // VC90
280620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump          vscomntools = vs90comntools;
281620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump      #elif (_MSC_VER == 1400) // VC80
282620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump          vscomntools = vs80comntools;
283620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump      #else
284620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump          vscomntools = vs90comntools;
285620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump      #endif
286620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump    }
287620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump    else if (vs90comntools)
288620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump      vscomntools = vs90comntools;
289620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump    else if (vs80comntools)
290620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump      vscomntools = vs80comntools;
291620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump    if (vscomntools && *vscomntools) {
292620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump      char *p = (char*)strstr(vscomntools, "\\Common7\\Tools");
293620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump      if (p)
294620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump        *p = '\0';
295620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump      path = vscomntools;
296620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump      return(true);
297620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump    }
298620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump    else
299620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump      return(false);
300620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump  }
301620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump  return(false);
30243d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump}
3034d374d4a4832a01d5d3b75c59cb1567e60df41e7Sebastian Redl
304e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindolavoid InitHeaderSearch::AddDefaultCIncludePaths(const llvm::Triple &triple) {
30543d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  // FIXME: temporary hack: hard-coded paths.
306c706468b1e00bab49ca92eb500a39657f530f828Daniel Dunbar  llvm::StringRef CIncludeDirs(C_INCLUDE_DIRS);
307c706468b1e00bab49ca92eb500a39657f530f828Daniel Dunbar  if (CIncludeDirs != "") {
308f0a2f51be576089e1a84cd677c09156dae067b6aRafael Espindola    std::vector<std::string> dirs;
309c706468b1e00bab49ca92eb500a39657f530f828Daniel Dunbar    llvm::SplitString(CIncludeDirs, dirs, ":");
310f0a2f51be576089e1a84cd677c09156dae067b6aRafael Espindola    for (std::vector<std::string>::iterator i = dirs.begin(); i != dirs.end(); ++i)
311f0a2f51be576089e1a84cd677c09156dae067b6aRafael Espindola      AddPath(*i, System, false, false, false);
312f0a2f51be576089e1a84cd677c09156dae067b6aRafael Espindola    return;
313f0a2f51be576089e1a84cd677c09156dae067b6aRafael Espindola  }
31443d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  llvm::Triple::OSType os = triple.getOS();
31543d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  switch (os) {
31643d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  case llvm::Triple::Win32:
31743d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump    {
318620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump      std::string VSDir;
319620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump      if (getVisualStudioDir(VSDir)) {
320620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump        AddPath(VSDir + "\\VC\\include", System, false, false, false);
321620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump        AddPath(VSDir + "\\VC\\PlatformSDK\\Include",
322620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump          System, false, false, false);
323620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump      }
324620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump      else {
32543d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump          // Default install paths.
326620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump        AddPath("C:/Program Files/Microsoft Visual Studio 9.0/VC/include",
32743d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump          System, false, false, false);
328620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump        AddPath(
329620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump        "C:/Program Files/Microsoft Visual Studio 9.0/VC/PlatformSDK/Include",
33043d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump          System, false, false, false);
331620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump        AddPath("C:/Program Files/Microsoft Visual Studio 8/VC/include",
332620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump          System, false, false, false);
333620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump        AddPath(
334620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump        "C:/Program Files/Microsoft Visual Studio 8/VC/PlatformSDK/Include",
335620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump          System, false, false, false);
336620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump          // For some clang developers.
337620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump        AddPath("G:/Program Files/Microsoft Visual Studio 9.0/VC/include",
338620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump          System, false, false, false);
339620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump        AddPath(
340620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump        "G:/Program Files/Microsoft Visual Studio 9.0/VC/PlatformSDK/Include",
341620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump          System, false, false, false);
342620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump      }
34343d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump    }
34443d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump    break;
34543d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  case llvm::Triple::MinGW64:
346620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump  case llvm::Triple::MinGW32:
34743d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump    AddPath("c:/mingw/include", System, true, false, false);
34843d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump    break;
34943d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  default:
35043d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump    break;
3510fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  }
352d3f88343af2dec3459493fafa22532317321eaf0John Thompson
353d3f88343af2dec3459493fafa22532317321eaf0John Thompson  AddPath("/usr/local/include", System, false, false, false);
354d3f88343af2dec3459493fafa22532317321eaf0John Thompson  AddPath("/usr/include", System, false, false, false);
355e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola}
356e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola
357e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindolavoid InitHeaderSearch::AddDefaultCPlusPlusIncludePaths(const llvm::Triple &triple) {
358e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola  llvm::Triple::OSType os = triple.getOS();
359e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola  // FIXME: temporary hack: hard-coded paths.
360e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola  switch (os) {
361e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola  case llvm::Triple::Cygwin:
362e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    AddPath("/lib/gcc/i686-pc-cygwin/3.4.4/include",
363e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola        System, true, false, false);
364e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    AddPath("/lib/gcc/i686-pc-cygwin/3.4.4/include/c++",
365e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola        System, true, false, false);
366e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    break;
367e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola  case llvm::Triple::MinGW64:
368e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    // Try gcc 4.4.0
369e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    AddMinGWCPlusPlusIncludePaths("c:/MinGW/lib/gcc", "mingw64", "4.4.0");
370e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    // Try gcc 4.3.0
371e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    AddMinGWCPlusPlusIncludePaths("c:/MinGW/lib/gcc", "mingw64", "4.3.0");
372e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    // Fall through.
373e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola  case llvm::Triple::MinGW32:
374e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    // Try gcc 4.4.0
375e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    AddMinGWCPlusPlusIncludePaths("c:/MinGW/lib/gcc", "mingw32", "4.4.0");
376e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    // Try gcc 4.3.0
377e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    AddMinGWCPlusPlusIncludePaths("c:/MinGW/lib/gcc", "mingw32", "4.3.0");
378e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    break;
379e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola  case llvm::Triple::Darwin:
380e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.2.1",
38140d1bb6383c14d5f8249d0640e8088a91b694c66John Thompson        "i686-apple-darwin10",
38240d1bb6383c14d5f8249d0640e8088a91b694c66John Thompson        "i686-apple-darwin10/x86_64",
38340d1bb6383c14d5f8249d0640e8088a91b694c66John Thompson        triple);
384e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.0.0",
38540d1bb6383c14d5f8249d0640e8088a91b694c66John Thompson        "i686-apple-darwin8",
38640d1bb6383c14d5f8249d0640e8088a91b694c66John Thompson        "i686-apple-darwin8",
38740d1bb6383c14d5f8249d0640e8088a91b694c66John Thompson        triple);
388e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    break;
389e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola  case llvm::Triple::Linux:
390e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    // Ubuntu 7.10 - Gutsy Gibbon
391e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.1.3",
39240d1bb6383c14d5f8249d0640e8088a91b694c66John Thompson        "i486-linux-gnu",
39340d1bb6383c14d5f8249d0640e8088a91b694c66John Thompson        "i486-linux-gnu",
39440d1bb6383c14d5f8249d0640e8088a91b694c66John Thompson        triple);
395e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    // Ubuntu 9.04
396e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.3.3",
39740d1bb6383c14d5f8249d0640e8088a91b694c66John Thompson        "x86_64-linux-gnu/32",
39840d1bb6383c14d5f8249d0640e8088a91b694c66John Thompson        "x86_64-linux-gnu",
39940d1bb6383c14d5f8249d0640e8088a91b694c66John Thompson        triple);
4005114ecab149ab91d13d3f7fd165dbe0839c1f2e8Sebastian Redl    // Ubuntu 9.10
4015114ecab149ab91d13d3f7fd165dbe0839c1f2e8Sebastian Redl    AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.4.1",
40240d1bb6383c14d5f8249d0640e8088a91b694c66John Thompson        "x86_64-linux-gnu/32",
40340d1bb6383c14d5f8249d0640e8088a91b694c66John Thompson        "x86_64-linux-gnu",
40440d1bb6383c14d5f8249d0640e8088a91b694c66John Thompson        triple);
405e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    // Fedora 8
406e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.1.2",
40740d1bb6383c14d5f8249d0640e8088a91b694c66John Thompson        "i386-redhat-linux",
40840d1bb6383c14d5f8249d0640e8088a91b694c66John Thompson        "i386-redhat-linux",
40940d1bb6383c14d5f8249d0640e8088a91b694c66John Thompson        triple);
410e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    // Fedora 9
411e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.3.0",
41240d1bb6383c14d5f8249d0640e8088a91b694c66John Thompson        "i386-redhat-linux",
41340d1bb6383c14d5f8249d0640e8088a91b694c66John Thompson        "i386-redhat-linux",
41440d1bb6383c14d5f8249d0640e8088a91b694c66John Thompson        triple);
415e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    // Fedora 10
416e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.3.2",
41740d1bb6383c14d5f8249d0640e8088a91b694c66John Thompson        "i386-redhat-linux",
41840d1bb6383c14d5f8249d0640e8088a91b694c66John Thompson        "i386-redhat-linux",
41940d1bb6383c14d5f8249d0640e8088a91b694c66John Thompson        triple);
420e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    // openSUSE 11.1 32 bit
421e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.3",
42240d1bb6383c14d5f8249d0640e8088a91b694c66John Thompson        "i586-suse-linux",
42340d1bb6383c14d5f8249d0640e8088a91b694c66John Thompson        "i586-suse-linux",
42440d1bb6383c14d5f8249d0640e8088a91b694c66John Thompson        triple);
425e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    // openSUSE 11.1 64 bit
426e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.3",
42740d1bb6383c14d5f8249d0640e8088a91b694c66John Thompson        "x86_64-suse-linux/32",
42840d1bb6383c14d5f8249d0640e8088a91b694c66John Thompson        "x86_64-suse-linux",
42940d1bb6383c14d5f8249d0640e8088a91b694c66John Thompson        triple);
430e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    // openSUSE 11.2
431e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.4",
43240d1bb6383c14d5f8249d0640e8088a91b694c66John Thompson        "i586-suse-linux",
43340d1bb6383c14d5f8249d0640e8088a91b694c66John Thompson        "i586-suse-linux",
43440d1bb6383c14d5f8249d0640e8088a91b694c66John Thompson        triple);
435e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.4",
43640d1bb6383c14d5f8249d0640e8088a91b694c66John Thompson        "x86_64-suse-linux",
43740d1bb6383c14d5f8249d0640e8088a91b694c66John Thompson        "x86_64-suse-linux",
43840d1bb6383c14d5f8249d0640e8088a91b694c66John Thompson        triple);
439e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    // Arch Linux 2008-06-24
440e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.3.1",
44140d1bb6383c14d5f8249d0640e8088a91b694c66John Thompson        "i686-pc-linux-gnu",
44240d1bb6383c14d5f8249d0640e8088a91b694c66John Thompson        "i686-pc-linux-gnu",
44340d1bb6383c14d5f8249d0640e8088a91b694c66John Thompson        triple);
444e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.3.1",
44540d1bb6383c14d5f8249d0640e8088a91b694c66John Thompson        "x86_64-unknown-linux-gnu",
44640d1bb6383c14d5f8249d0640e8088a91b694c66John Thompson        "x86_64-unknown-linux-gnu",
44740d1bb6383c14d5f8249d0640e8088a91b694c66John Thompson        triple);
448e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    // Gentoo x86 2009.1 stable
449e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    AddGnuCPlusPlusIncludePaths(
45040d1bb6383c14d5f8249d0640e8088a91b694c66John Thompson      "/usr/lib/gcc/i686-pc-linux-gnu/4.3.4/include/g++-v4",
45140d1bb6383c14d5f8249d0640e8088a91b694c66John Thompson      "i686-pc-linux-gnu",
45240d1bb6383c14d5f8249d0640e8088a91b694c66John Thompson      "i686-pc-linux-gnu",
45340d1bb6383c14d5f8249d0640e8088a91b694c66John Thompson      triple);
454e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    // Gentoo x86 2009.0 stable
455e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    AddGnuCPlusPlusIncludePaths(
45640d1bb6383c14d5f8249d0640e8088a91b694c66John Thompson      "/usr/lib/gcc/i686-pc-linux-gnu/4.3.2/include/g++-v4",
45740d1bb6383c14d5f8249d0640e8088a91b694c66John Thompson      "i686-pc-linux-gnu",
45840d1bb6383c14d5f8249d0640e8088a91b694c66John Thompson      "i686-pc-linux-gnu",
45940d1bb6383c14d5f8249d0640e8088a91b694c66John Thompson      triple);
460e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    // Gentoo x86 2008.0 stable
461e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    AddGnuCPlusPlusIncludePaths(
46240d1bb6383c14d5f8249d0640e8088a91b694c66John Thompson      "/usr/lib/gcc/i686-pc-linux-gnu/4.1.2/include/g++-v4",
46340d1bb6383c14d5f8249d0640e8088a91b694c66John Thompson      "i686-pc-linux-gnu",
46440d1bb6383c14d5f8249d0640e8088a91b694c66John Thompson      "i686-pc-linux-gnu",
46540d1bb6383c14d5f8249d0640e8088a91b694c66John Thompson      triple);
466e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    // Ubuntu 8.10
467e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.3",
46840d1bb6383c14d5f8249d0640e8088a91b694c66John Thompson        "i486-pc-linux-gnu",
46940d1bb6383c14d5f8249d0640e8088a91b694c66John Thompson        "i486-pc-linux-gnu",
47040d1bb6383c14d5f8249d0640e8088a91b694c66John Thompson        triple);
471e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    // Ubuntu 9.04
472e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.3",
47340d1bb6383c14d5f8249d0640e8088a91b694c66John Thompson        "i486-linux-gnu",
47440d1bb6383c14d5f8249d0640e8088a91b694c66John Thompson        "i486-linux-gnu",
47540d1bb6383c14d5f8249d0640e8088a91b694c66John Thompson        triple);
476e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    // Gentoo amd64 stable
477e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    AddGnuCPlusPlusIncludePaths(
478e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola        "/usr/lib/gcc/x86_64-pc-linux-gnu/4.1.2/include/g++-v4",
479e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola        "i686-pc-linux-gnu",
480e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola        "i686-pc-linux-gnu",
481e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola        triple);
4825d7a188cbc66d0af5aa3c135942a9e4aa7cb69f9Benjamin Kramer    // Exherbo (2009-10-26)
4835d7a188cbc66d0af5aa3c135942a9e4aa7cb69f9Benjamin Kramer    AddGnuCPlusPlusIncludePaths(
4845d7a188cbc66d0af5aa3c135942a9e4aa7cb69f9Benjamin Kramer        "/usr/include/c++/4.4.2",
4855d7a188cbc66d0af5aa3c135942a9e4aa7cb69f9Benjamin Kramer        "x86_64-pc-linux-gnu/32",
4865d7a188cbc66d0af5aa3c135942a9e4aa7cb69f9Benjamin Kramer        "x86_64-pc-linux-gnu",
4875d7a188cbc66d0af5aa3c135942a9e4aa7cb69f9Benjamin Kramer        triple);
4885d7a188cbc66d0af5aa3c135942a9e4aa7cb69f9Benjamin Kramer    AddGnuCPlusPlusIncludePaths(
4895d7a188cbc66d0af5aa3c135942a9e4aa7cb69f9Benjamin Kramer        "/usr/include/c++/4.4.2",
4905d7a188cbc66d0af5aa3c135942a9e4aa7cb69f9Benjamin Kramer        "i686-pc-linux-gnu",
4915d7a188cbc66d0af5aa3c135942a9e4aa7cb69f9Benjamin Kramer        "i686-pc-linux-gnu",
4925d7a188cbc66d0af5aa3c135942a9e4aa7cb69f9Benjamin Kramer        triple);
493e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    break;
494e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola  case llvm::Triple::FreeBSD:
495e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    // DragonFly
496e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    AddPath("/usr/include/c++/4.1", System, true, false, false);
497e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    // FreeBSD
498e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    AddPath("/usr/include/c++/4.2", System, true, false, false);
499e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    break;
500e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola  case llvm::Triple::Solaris:
501e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    // Solaris - Fall though..
502e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola  case llvm::Triple::AuroraUX:
503e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    // AuroraUX
504e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    AddGnuCPlusPlusIncludePaths("/opt/gcc4/include/c++/4.2.4",
505e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola                                "i386-pc-solaris2.11",
506e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola                                "i386-pc-solaris2.11",
507e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola                                triple);
508e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    break;
509e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola  default:
510e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    break;
511e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola  }
512e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola}
513e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola
514e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindolavoid InitHeaderSearch::AddDefaultSystemIncludePaths(const LangOptions &Lang,
515e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola                                                    const llvm::Triple &triple) {
516e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola  AddDefaultCIncludePaths(triple);
517e166582f8f36f4db8f4ea157538fab7fe6bf2658Daniel Dunbar
518e166582f8f36f4db8f4ea157538fab7fe6bf2658Daniel Dunbar  // Add the default framework include paths on Darwin.
519e166582f8f36f4db8f4ea157538fab7fe6bf2658Daniel Dunbar  if (triple.getOS() == llvm::Triple::Darwin) {
520e166582f8f36f4db8f4ea157538fab7fe6bf2658Daniel Dunbar    AddPath("/System/Library/Frameworks", System, true, false, true);
521e166582f8f36f4db8f4ea157538fab7fe6bf2658Daniel Dunbar    AddPath("/Library/Frameworks", System, true, false, true);
522e166582f8f36f4db8f4ea157538fab7fe6bf2658Daniel Dunbar  }
523e166582f8f36f4db8f4ea157538fab7fe6bf2658Daniel Dunbar
524e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola  if (Lang.CPlusPlus)
525e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    AddDefaultCPlusPlusIncludePaths(triple);
526e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola}
527e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola
5280fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber/// RemoveDuplicates - If there are duplicate directory entries in the specified
5290fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber/// search list, remove the later (dead) ones.
5300fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weberstatic void RemoveDuplicates(std::vector<DirectoryLookup> &SearchList,
5310fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber                             bool Verbose) {
5320fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  llvm::SmallPtrSet<const DirectoryEntry *, 8> SeenDirs;
5330fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  llvm::SmallPtrSet<const DirectoryEntry *, 8> SeenFrameworkDirs;
5340fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  llvm::SmallPtrSet<const HeaderMap *, 8> SeenHeaderMaps;
5350fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  for (unsigned i = 0; i != SearchList.size(); ++i) {
5367a739401983eb9bc25c9b4b25177e116f5e5ba92Chris Lattner    unsigned DirToRemove = i;
5371eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
53843eee07270bf1966ea7289310066aa670c4b647fChris Lattner    const DirectoryLookup &CurEntry = SearchList[i];
5391eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
54043eee07270bf1966ea7289310066aa670c4b647fChris Lattner    if (CurEntry.isNormalDir()) {
5410fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber      // If this isn't the first time we've seen this dir, remove it.
54243eee07270bf1966ea7289310066aa670c4b647fChris Lattner      if (SeenDirs.insert(CurEntry.getDir()))
5430fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber        continue;
54443eee07270bf1966ea7289310066aa670c4b647fChris Lattner    } else if (CurEntry.isFramework()) {
5450fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber      // If this isn't the first time we've seen this framework dir, remove it.
54643eee07270bf1966ea7289310066aa670c4b647fChris Lattner      if (SeenFrameworkDirs.insert(CurEntry.getFrameworkDir()))
5470fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber        continue;
5480fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber    } else {
54943eee07270bf1966ea7289310066aa670c4b647fChris Lattner      assert(CurEntry.isHeaderMap() && "Not a headermap or normal dir?");
5500fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber      // If this isn't the first time we've seen this headermap, remove it.
55143eee07270bf1966ea7289310066aa670c4b647fChris Lattner      if (SeenHeaderMaps.insert(CurEntry.getHeaderMap()))
5520fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber        continue;
55330f05b553db067c994966daca37e75324ee7b424Chris Lattner    }
5541eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
55530f05b553db067c994966daca37e75324ee7b424Chris Lattner    // If we have a normal #include dir/framework/headermap that is shadowed
55630f05b553db067c994966daca37e75324ee7b424Chris Lattner    // later in the chain by a system include location, we actually want to
55730f05b553db067c994966daca37e75324ee7b424Chris Lattner    // ignore the user's request and drop the user dir... keeping the system
55830f05b553db067c994966daca37e75324ee7b424Chris Lattner    // dir.  This is weird, but required to emulate GCC's search path correctly.
55930f05b553db067c994966daca37e75324ee7b424Chris Lattner    //
56030f05b553db067c994966daca37e75324ee7b424Chris Lattner    // Since dupes of system dirs are rare, just rescan to find the original
56130f05b553db067c994966daca37e75324ee7b424Chris Lattner    // that we're nuking instead of using a DenseMap.
56243eee07270bf1966ea7289310066aa670c4b647fChris Lattner    if (CurEntry.getDirCharacteristic() != SrcMgr::C_User) {
56330f05b553db067c994966daca37e75324ee7b424Chris Lattner      // Find the dir that this is the same of.
56430f05b553db067c994966daca37e75324ee7b424Chris Lattner      unsigned FirstDir;
56530f05b553db067c994966daca37e75324ee7b424Chris Lattner      for (FirstDir = 0; ; ++FirstDir) {
56630f05b553db067c994966daca37e75324ee7b424Chris Lattner        assert(FirstDir != i && "Didn't find dupe?");
5671eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
56843eee07270bf1966ea7289310066aa670c4b647fChris Lattner        const DirectoryLookup &SearchEntry = SearchList[FirstDir];
56943eee07270bf1966ea7289310066aa670c4b647fChris Lattner
57030f05b553db067c994966daca37e75324ee7b424Chris Lattner        // If these are different lookup types, then they can't be the dupe.
57143eee07270bf1966ea7289310066aa670c4b647fChris Lattner        if (SearchEntry.getLookupType() != CurEntry.getLookupType())
57230f05b553db067c994966daca37e75324ee7b424Chris Lattner          continue;
5731eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
57430f05b553db067c994966daca37e75324ee7b424Chris Lattner        bool isSame;
57543eee07270bf1966ea7289310066aa670c4b647fChris Lattner        if (CurEntry.isNormalDir())
57643eee07270bf1966ea7289310066aa670c4b647fChris Lattner          isSame = SearchEntry.getDir() == CurEntry.getDir();
57743eee07270bf1966ea7289310066aa670c4b647fChris Lattner        else if (CurEntry.isFramework())
57843eee07270bf1966ea7289310066aa670c4b647fChris Lattner          isSame = SearchEntry.getFrameworkDir() == CurEntry.getFrameworkDir();
57930f05b553db067c994966daca37e75324ee7b424Chris Lattner        else {
58043eee07270bf1966ea7289310066aa670c4b647fChris Lattner          assert(CurEntry.isHeaderMap() && "Not a headermap or normal dir?");
58143eee07270bf1966ea7289310066aa670c4b647fChris Lattner          isSame = SearchEntry.getHeaderMap() == CurEntry.getHeaderMap();
58230f05b553db067c994966daca37e75324ee7b424Chris Lattner        }
5831eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
58430f05b553db067c994966daca37e75324ee7b424Chris Lattner        if (isSame)
58530f05b553db067c994966daca37e75324ee7b424Chris Lattner          break;
58630f05b553db067c994966daca37e75324ee7b424Chris Lattner      }
5871eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
58830f05b553db067c994966daca37e75324ee7b424Chris Lattner      // If the first dir in the search path is a non-system dir, zap it
58930f05b553db067c994966daca37e75324ee7b424Chris Lattner      // instead of the system one.
59030f05b553db067c994966daca37e75324ee7b424Chris Lattner      if (SearchList[FirstDir].getDirCharacteristic() == SrcMgr::C_User)
59130f05b553db067c994966daca37e75324ee7b424Chris Lattner        DirToRemove = FirstDir;
59230f05b553db067c994966daca37e75324ee7b424Chris Lattner    }
59330f05b553db067c994966daca37e75324ee7b424Chris Lattner
59430f05b553db067c994966daca37e75324ee7b424Chris Lattner    if (Verbose) {
59543eee07270bf1966ea7289310066aa670c4b647fChris Lattner      fprintf(stderr, "ignoring duplicate directory \"%s\"\n",
59643eee07270bf1966ea7289310066aa670c4b647fChris Lattner              CurEntry.getName());
59730f05b553db067c994966daca37e75324ee7b424Chris Lattner      if (DirToRemove != i)
59830f05b553db067c994966daca37e75324ee7b424Chris Lattner        fprintf(stderr, "  as it is a non-system directory that duplicates"
59930f05b553db067c994966daca37e75324ee7b424Chris Lattner                " a system directory\n");
6000fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber    }
6011eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
6027a739401983eb9bc25c9b4b25177e116f5e5ba92Chris Lattner    // This is reached if the current entry is a duplicate.  Remove the
6037a739401983eb9bc25c9b4b25177e116f5e5ba92Chris Lattner    // DirToRemove (usually the current dir).
6047a739401983eb9bc25c9b4b25177e116f5e5ba92Chris Lattner    SearchList.erase(SearchList.begin()+DirToRemove);
6050fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber    --i;
6060fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  }
6070fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber}
6080fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber
6090fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber
6100fca022d77b89100e746f4d659b84ed5b1ee0158Nico Webervoid InitHeaderSearch::Realize() {
6110fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  // Concatenate ANGLE+SYSTEM+AFTER chains together into SearchList.
6120fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  std::vector<DirectoryLookup> SearchList;
6130fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  SearchList = IncludeGroup[Angled];
6140fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  SearchList.insert(SearchList.end(), IncludeGroup[System].begin(),
6150fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber                    IncludeGroup[System].end());
6160fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  SearchList.insert(SearchList.end(), IncludeGroup[After].begin(),
6170fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber                    IncludeGroup[After].end());
6180fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  RemoveDuplicates(SearchList, Verbose);
6190fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  RemoveDuplicates(IncludeGroup[Quoted], Verbose);
6201eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
6210fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  // Prepend QUOTED list on the search list.
6221eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  SearchList.insert(SearchList.begin(), IncludeGroup[Quoted].begin(),
6230fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber                    IncludeGroup[Quoted].end());
6241eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
6250fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber
6260fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  bool DontSearchCurDir = false;  // TODO: set to true if -I- is set?
6270fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  Headers.SetSearchPaths(SearchList, IncludeGroup[Quoted].size(),
6280fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber                         DontSearchCurDir);
6290fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber
6300fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  // If verbose, print the list of directories that will be searched.
6310fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  if (Verbose) {
6320fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber    fprintf(stderr, "#include \"...\" search starts here:\n");
6330fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber    unsigned QuotedIdx = IncludeGroup[Quoted].size();
6340fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber    for (unsigned i = 0, e = SearchList.size(); i != e; ++i) {
6350fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber      if (i == QuotedIdx)
6360fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber        fprintf(stderr, "#include <...> search starts here:\n");
6370fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber      const char *Name = SearchList[i].getName();
6380fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber      const char *Suffix;
6390fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber      if (SearchList[i].isNormalDir())
6400fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber        Suffix = "";
6410fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber      else if (SearchList[i].isFramework())
6420fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber        Suffix = " (framework directory)";
6430fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber      else {
6440fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber        assert(SearchList[i].isHeaderMap() && "Unknown DirectoryLookup");
6450fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber        Suffix = " (headermap)";
6460fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber      }
6470fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber      fprintf(stderr, " %s%s\n", Name, Suffix);
6480fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber    }
6490fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber    fprintf(stderr, "End of search list.\n");
6500fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  }
6510fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber}
65263c8b77334f90472260d2f48df2742ed5067261eDaniel Dunbar
6535814e657c9ad9ef6049a2a4af0d2aad248a8a15cDaniel Dunbarvoid clang::ApplyHeaderSearchOptions(HeaderSearch &HS,
6545814e657c9ad9ef6049a2a4af0d2aad248a8a15cDaniel Dunbar                                     const HeaderSearchOptions &HSOpts,
6555814e657c9ad9ef6049a2a4af0d2aad248a8a15cDaniel Dunbar                                     const LangOptions &Lang,
65663c8b77334f90472260d2f48df2742ed5067261eDaniel Dunbar                                     const llvm::Triple &Triple) {
65763c8b77334f90472260d2f48df2742ed5067261eDaniel Dunbar  InitHeaderSearch Init(HS, HSOpts.Verbose, HSOpts.Sysroot);
65863c8b77334f90472260d2f48df2742ed5067261eDaniel Dunbar
65963c8b77334f90472260d2f48df2742ed5067261eDaniel Dunbar  // Add the user defined entries.
66063c8b77334f90472260d2f48df2742ed5067261eDaniel Dunbar  for (unsigned i = 0, e = HSOpts.UserEntries.size(); i != e; ++i) {
66163c8b77334f90472260d2f48df2742ed5067261eDaniel Dunbar    const HeaderSearchOptions::Entry &E = HSOpts.UserEntries[i];
66263c8b77334f90472260d2f48df2742ed5067261eDaniel Dunbar    Init.AddPath(E.Path, E.Group, E.IsCXXAware, E.IsUserSupplied, E.IsFramework,
66363c8b77334f90472260d2f48df2742ed5067261eDaniel Dunbar                 E.IgnoreSysRoot);
66463c8b77334f90472260d2f48df2742ed5067261eDaniel Dunbar  }
66563c8b77334f90472260d2f48df2742ed5067261eDaniel Dunbar
66663c8b77334f90472260d2f48df2742ed5067261eDaniel Dunbar  // Add entries from CPATH and friends.
66763c8b77334f90472260d2f48df2742ed5067261eDaniel Dunbar  Init.AddDelimitedPaths(HSOpts.EnvIncPath.c_str());
66863c8b77334f90472260d2f48df2742ed5067261eDaniel Dunbar  Init.AddDelimitedPaths(HSOpts.LangEnvIncPath.c_str());
66963c8b77334f90472260d2f48df2742ed5067261eDaniel Dunbar
67063c8b77334f90472260d2f48df2742ed5067261eDaniel Dunbar  if (!HSOpts.BuiltinIncludePath.empty()) {
67163c8b77334f90472260d2f48df2742ed5067261eDaniel Dunbar    // Ignore the sys root, we *always* look for clang headers relative to
67263c8b77334f90472260d2f48df2742ed5067261eDaniel Dunbar    // supplied path.
6732cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar    Init.AddPath(HSOpts.BuiltinIncludePath, System,
67463c8b77334f90472260d2f48df2742ed5067261eDaniel Dunbar                 false, false, false, /*IgnoreSysRoot=*/ true);
67563c8b77334f90472260d2f48df2742ed5067261eDaniel Dunbar  }
67663c8b77334f90472260d2f48df2742ed5067261eDaniel Dunbar
677dd35ce9a783dea7e6660f9cb1d973742f3b3c73fDaniel Dunbar  if (HSOpts.UseStandardIncludes)
67863c8b77334f90472260d2f48df2742ed5067261eDaniel Dunbar    Init.AddDefaultSystemIncludePaths(Lang, Triple);
67963c8b77334f90472260d2f48df2742ed5067261eDaniel Dunbar
68063c8b77334f90472260d2f48df2742ed5067261eDaniel Dunbar  Init.Realize();
68163c8b77334f90472260d2f48df2742ed5067261eDaniel Dunbar}
682