InitHeaderSearch.cpp revision ff58e3610f4e12094def69eb2d6dcb4330378d8f
1df22c2cc7703c7df04eadd2e6e59c61f545a5741Nick Lewycky//===--- 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"
24e89ba59005479529f5567f12b436617a2ca73ec2Benjamin Kramer#include "llvm/ADT/Twine.h"
25d57a7ef9252964bc6c8471451d7bd395b0520cb8Chris Lattner#include "llvm/Support/raw_ostream.h"
260fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber#include "llvm/System/Path.h"
27121e3c207415fb4c105d690de384d8f3d49b0f2dArgyrios Kyrtzidis#include "llvm/Config/config.h"
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.
53e89ba59005479529f5567f12b436617a2ca73ec2Benjamin Kramer  void AddPath(const llvm::Twine &Path, IncludeDirGroup Group,
542cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar               bool isCXXAware, bool isUserSupplied,
552cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar               bool isFramework, bool IgnoreSysRoot = false);
562cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar
57a608737659d4330d9ae7622bf1a87c33b9d97072mike-m  /// AddGnuCPlusPlusIncludePaths - Add the necessary paths to support a gnu
582cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar  ///  libstdc++.
59e89ba59005479529f5567f12b436617a2ca73ec2Benjamin Kramer  void AddGnuCPlusPlusIncludePaths(llvm::StringRef Base,
60e89ba59005479529f5567f12b436617a2ca73ec2Benjamin Kramer                                   llvm::StringRef ArchDir,
61e89ba59005479529f5567f12b436617a2ca73ec2Benjamin Kramer                                   llvm::StringRef Dir32,
62e89ba59005479529f5567f12b436617a2ca73ec2Benjamin Kramer                                   llvm::StringRef Dir64,
632cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar                                   const llvm::Triple &triple);
642cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar
652cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar  /// AddMinGWCPlusPlusIncludePaths - Add the necessary paths to suport a MinGW
662cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar  ///  libstdc++.
67e89ba59005479529f5567f12b436617a2ca73ec2Benjamin Kramer  void AddMinGWCPlusPlusIncludePaths(llvm::StringRef Base,
68e89ba59005479529f5567f12b436617a2ca73ec2Benjamin Kramer                                     llvm::StringRef Arch,
69e89ba59005479529f5567f12b436617a2ca73ec2Benjamin Kramer                                     llvm::StringRef 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.
739e9ddf6100fd50ca2f9f9d0106b856e20b43300fBenjamin Kramer  void AddDelimitedPaths(llvm::StringRef String);
742cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar
752cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar  // AddDefaultCIncludePaths - Add paths that should always be searched.
7679bc57c074cb91f574fba19215f95b63dcf6b04bmike-m  void AddDefaultCIncludePaths(const llvm::Triple &triple,
7779bc57c074cb91f574fba19215f95b63dcf6b04bmike-m                               const HeaderSearchOptions &HSOpts);
782cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar
792cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar  // AddDefaultCPlusPlusIncludePaths -  Add paths that should be searched when
802cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar  //  compiling c++.
812cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar  void AddDefaultCPlusPlusIncludePaths(const llvm::Triple &triple);
822cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar
832cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar  /// AddDefaultSystemIncludePaths - Adds the default system include paths so
842cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar  ///  that e.g. stdio.h is found.
852cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar  void AddDefaultSystemIncludePaths(const LangOptions &Lang,
864c2bcad7b843c10fd4a2ffd43da40bfefb4dc8baDouglas Gregor                                    const llvm::Triple &triple,
8779bc57c074cb91f574fba19215f95b63dcf6b04bmike-m                                    const HeaderSearchOptions &HSOpts);
882cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar
892cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar  /// Realize - Merges all search path lists into one list and send it to
902cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar  /// HeaderSearch.
912cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar  void Realize();
922cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar};
932cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar
942cdafa8001ee69b75d2906cbb36f16cf8e1dc60aDaniel Dunbar}
950fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber
96e89ba59005479529f5567f12b436617a2ca73ec2Benjamin Kramervoid InitHeaderSearch::AddPath(const llvm::Twine &Path,
97458fb10ef5ba2d7b375c6c64095c1458af0a5be3Benjamin Kramer                               IncludeDirGroup Group, bool isCXXAware,
98458fb10ef5ba2d7b375c6c64095c1458af0a5be3Benjamin Kramer                               bool isUserSupplied, bool isFramework,
99458fb10ef5ba2d7b375c6c64095c1458af0a5be3Benjamin Kramer                               bool IgnoreSysRoot) {
100e89ba59005479529f5567f12b436617a2ca73ec2Benjamin Kramer  assert(!Path.isTriviallyEmpty() && "can't handle empty path here");
1010fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  FileManager &FM = Headers.getFileMgr();
1021eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1030fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  // Compute the actual path, taking into consideration -isysroot.
104e89ba59005479529f5567f12b436617a2ca73ec2Benjamin Kramer  llvm::SmallString<256> MappedPathStr;
105e89ba59005479529f5567f12b436617a2ca73ec2Benjamin Kramer  llvm::raw_svector_ostream MappedPath(MappedPathStr);
1061eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1070fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  // Handle isysroot.
1086858dd3fcc2d3ac6a706a0294be1d3ac51849022Chris Lattner  if (Group == System && !IgnoreSysRoot) {
1090fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber    // FIXME: Portability.  This should be a sys::Path interface, this doesn't
1100fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber    // handle things like C:\ right, nor win32 \\network\device\blah.
1110fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber    if (isysroot.size() != 1 || isysroot[0] != '/') // Add isysroot if present.
112e89ba59005479529f5567f12b436617a2ca73ec2Benjamin Kramer      MappedPath << isysroot;
1130fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  }
1141eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
115e89ba59005479529f5567f12b436617a2ca73ec2Benjamin Kramer  Path.print(MappedPath);
1160fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber
1170fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  // Compute the DirectoryLookup type.
1189d72851fec9e9c62570a027d42701562bbf29751Chris Lattner  SrcMgr::CharacteristicKind Type;
1190fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  if (Group == Quoted || Group == Angled)
1200b9e736308af5397f558ffc8e780c438c2fdb563Chris Lattner    Type = SrcMgr::C_User;
1210fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  else if (isCXXAware)
1220b9e736308af5397f558ffc8e780c438c2fdb563Chris Lattner    Type = SrcMgr::C_System;
1230fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  else
1240b9e736308af5397f558ffc8e780c438c2fdb563Chris Lattner    Type = SrcMgr::C_ExternCSystem;
1251eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1261eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1270fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  // If the directory exists, add it.
128458fb10ef5ba2d7b375c6c64095c1458af0a5be3Benjamin Kramer  if (const DirectoryEntry *DE = FM.getDirectory(MappedPath.str())) {
1290fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber    IncludeGroup[Group].push_back(DirectoryLookup(DE, Type, isUserSupplied,
1300fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber                                                  isFramework));
1310fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber    return;
1320fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  }
1331eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1340fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  // Check to see if this is an apple-style headermap (which are not allowed to
1350fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  // be frameworks).
1360fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  if (!isFramework) {
137458fb10ef5ba2d7b375c6c64095c1458af0a5be3Benjamin Kramer    if (const FileEntry *FE = FM.getFile(MappedPath.str())) {
1380fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber      if (const HeaderMap *HM = Headers.CreateHeaderMap(FE)) {
1390fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber        // It is a headermap, add it to the search path.
1400fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber        IncludeGroup[Group].push_back(DirectoryLookup(HM, Type,isUserSupplied));
1410fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber        return;
1420fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber      }
1430fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber    }
1440fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  }
1451eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1460fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  if (Verbose)
1477765934ad7e157b5fcf925792a38e01b1edbcf8aDaniel Dunbar    llvm::errs() << "ignoring nonexistent directory \""
1487765934ad7e157b5fcf925792a38e01b1edbcf8aDaniel Dunbar                 << MappedPath.str() << "\"\n";
1490fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber}
1500fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber
1510fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber
1529e9ddf6100fd50ca2f9f9d0106b856e20b43300fBenjamin Kramervoid InitHeaderSearch::AddDelimitedPaths(llvm::StringRef at) {
1539e9ddf6100fd50ca2f9f9d0106b856e20b43300fBenjamin Kramer  if (at.empty()) // Empty string should not add '.' path.
1540fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber    return;
1550fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber
1569e9ddf6100fd50ca2f9f9d0106b856e20b43300fBenjamin Kramer  llvm::StringRef::size_type delim;
1579e9ddf6100fd50ca2f9f9d0106b856e20b43300fBenjamin Kramer  while ((delim = at.find(llvm::sys::PathSeparator)) != llvm::StringRef::npos) {
1589e9ddf6100fd50ca2f9f9d0106b856e20b43300fBenjamin Kramer    if (delim == 0)
1590fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber      AddPath(".", Angled, false, true, false);
1600fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber    else
1619e9ddf6100fd50ca2f9f9d0106b856e20b43300fBenjamin Kramer      AddPath(at.substr(0, delim), Angled, false, true, false);
1629e9ddf6100fd50ca2f9f9d0106b856e20b43300fBenjamin Kramer    at = at.substr(delim + 1);
1630fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  }
1649e9ddf6100fd50ca2f9f9d0106b856e20b43300fBenjamin Kramer
1659e9ddf6100fd50ca2f9f9d0106b856e20b43300fBenjamin Kramer  if (at.empty())
1660fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber    AddPath(".", Angled, false, true, false);
1670fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  else
1680fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber    AddPath(at, Angled, false, true, false);
1690fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber}
1700fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber
171e89ba59005479529f5567f12b436617a2ca73ec2Benjamin Kramervoid InitHeaderSearch::AddGnuCPlusPlusIncludePaths(llvm::StringRef Base,
172e89ba59005479529f5567f12b436617a2ca73ec2Benjamin Kramer                                                   llvm::StringRef ArchDir,
173e89ba59005479529f5567f12b436617a2ca73ec2Benjamin Kramer                                                   llvm::StringRef Dir32,
174e89ba59005479529f5567f12b436617a2ca73ec2Benjamin Kramer                                                   llvm::StringRef Dir64,
17531b63beefa058aa42a12a1c2b72b43a4ba21f7f1Rafael Espindola                                                   const llvm::Triple &triple) {
1766ec18a3f86c702004af66cce6ff979999a736898Rafael Espindola  // Add the base dir
177ab7ae95c46f288ac52974aa60334a9575f9a850cRafael Espindola  AddPath(Base, System, true, false, false);
178ab7ae95c46f288ac52974aa60334a9575f9a850cRafael Espindola
179ab7ae95c46f288ac52974aa60334a9575f9a850cRafael Espindola  // Add the multilib dirs
18031b63beefa058aa42a12a1c2b72b43a4ba21f7f1Rafael Espindola  llvm::Triple::ArchType arch = triple.getArch();
18131b63beefa058aa42a12a1c2b72b43a4ba21f7f1Rafael Espindola  bool is64bit = arch == llvm::Triple::ppc64 || arch == llvm::Triple::x86_64;
18231b63beefa058aa42a12a1c2b72b43a4ba21f7f1Rafael Espindola  if (is64bit)
183ab7ae95c46f288ac52974aa60334a9575f9a850cRafael Espindola    AddPath(Base + "/" + ArchDir + "/" + Dir64, System, true, false, false);
18431b63beefa058aa42a12a1c2b72b43a4ba21f7f1Rafael Espindola  else
185ab7ae95c46f288ac52974aa60334a9575f9a850cRafael Espindola    AddPath(Base + "/" + ArchDir + "/" + Dir32, System, true, false, false);
1866ec18a3f86c702004af66cce6ff979999a736898Rafael Espindola
1876ec18a3f86c702004af66cce6ff979999a736898Rafael Espindola  // Add the backward dir
1886ec18a3f86c702004af66cce6ff979999a736898Rafael Espindola  AddPath(Base + "/backward", System, true, false, false);
1892e9f652d53346bf7e64c8a12a9ff06b004a3e489Rafael Espindola}
1900fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber
191e89ba59005479529f5567f12b436617a2ca73ec2Benjamin Kramervoid InitHeaderSearch::AddMinGWCPlusPlusIncludePaths(llvm::StringRef Base,
192e89ba59005479529f5567f12b436617a2ca73ec2Benjamin Kramer                                                     llvm::StringRef Arch,
193e89ba59005479529f5567f12b436617a2ca73ec2Benjamin Kramer                                                     llvm::StringRef Version) {
194ab8ae19ea4285a15ca43123e8e218e401dc5cb5aBenjamin Kramer  AddPath(Base + "/" + Arch + "/" + Version + "/include",
195ab8ae19ea4285a15ca43123e8e218e401dc5cb5aBenjamin Kramer          System, true, false, false);
196ab8ae19ea4285a15ca43123e8e218e401dc5cb5aBenjamin Kramer  AddPath(Base + "/" + Arch + "/" + Version + "/include/c++",
197ab8ae19ea4285a15ca43123e8e218e401dc5cb5aBenjamin Kramer          System, true, false, false);
198ab8ae19ea4285a15ca43123e8e218e401dc5cb5aBenjamin Kramer  AddPath(Base + "/" + Arch + "/" + Version + "/include/c++/backward",
199ab8ae19ea4285a15ca43123e8e218e401dc5cb5aBenjamin Kramer          System, true, false, false);
200620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump}
20143d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump
202620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump  // FIXME: This probably should goto to some platform utils place.
203620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump#ifdef _MSC_VER
20475ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson
205620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump  // Read registry string.
20675ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson  // This also supports a means to look for high-versioned keys by use
20775ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson  // of a $VERSION placeholder in the key path.
20875ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson  // $VERSION in the key path is a placeholder for the version number,
20975ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson  // causing the highest value path to be searched for and used.
21075ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson  // I.e. "HKEY_LOCAL_MACHINE\\SOFTWARE\\Microsoft\\VisualStudio\\$VERSION".
21175ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson  // There can be additional characters in the component.  Only the numberic
21275ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson  // characters are compared.
2136cd5216bb61e6165d8f6e8fb781d45c035bd4e00Benjamin Kramerstatic bool getSystemRegistryString(const char *keyPath, const char *valueName,
2146cd5216bb61e6165d8f6e8fb781d45c035bd4e00Benjamin Kramer                                    char *value, size_t maxLength) {
21543d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  HKEY hRootKey = NULL;
21643d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  HKEY hKey = NULL;
21743d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  const char* subKey = NULL;
21843d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  DWORD valueType;
21943d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  DWORD valueSize = maxLength - 1;
22075ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson  long lResult;
22143d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  bool returnValue = false;
22243d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  if (strncmp(keyPath, "HKEY_CLASSES_ROOT\\", 18) == 0) {
22343d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump    hRootKey = HKEY_CLASSES_ROOT;
22443d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump    subKey = keyPath + 18;
22543d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  }
22643d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  else if (strncmp(keyPath, "HKEY_USERS\\", 11) == 0) {
22743d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump    hRootKey = HKEY_USERS;
22843d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump    subKey = keyPath + 11;
22943d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  }
23043d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  else if (strncmp(keyPath, "HKEY_LOCAL_MACHINE\\", 19) == 0) {
23143d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump    hRootKey = HKEY_LOCAL_MACHINE;
23243d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump    subKey = keyPath + 19;
233121e3c207415fb4c105d690de384d8f3d49b0f2dArgyrios Kyrtzidis  }
23443d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  else if (strncmp(keyPath, "HKEY_CURRENT_USER\\", 18) == 0) {
23543d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump    hRootKey = HKEY_CURRENT_USER;
23643d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump    subKey = keyPath + 18;
23743d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  }
23843d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  else
23943d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump    return(false);
24075ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson  const char *placeHolder = strstr(subKey, "$VERSION");
24175ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson  char bestName[256];
24275ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson  bestName[0] = '\0';
24375ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson  // If we have a $VERSION placeholder, do the highest-version search.
24475ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson  if (placeHolder) {
24575ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson    const char *keyEnd = placeHolder - 1;
24675ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson    const char *nextKey = placeHolder;
24775ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson    // Find end of previous key.
24875ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson    while ((keyEnd > subKey) && (*keyEnd != '\\'))
24975ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson      keyEnd--;
25075ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson    // Find end of key containing $VERSION.
25175ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson    while (*nextKey && (*nextKey != '\\'))
25275ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson      nextKey++;
25375ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson    size_t partialKeyLength = keyEnd - subKey;
25475ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson    char partialKey[256];
25575ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson    if (partialKeyLength > sizeof(partialKey))
25675ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson      partialKeyLength = sizeof(partialKey);
25775ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson    strncpy(partialKey, subKey, partialKeyLength);
25875ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson    partialKey[partialKeyLength] = '\0';
25975ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson    HKEY hTopKey = NULL;
26075ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson    lResult = RegOpenKeyEx(hRootKey, partialKey, 0, KEY_READ, &hTopKey);
26175ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson    if (lResult == ERROR_SUCCESS) {
26275ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson      char keyName[256];
26375ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson      int bestIndex = -1;
26475ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson      double bestValue = 0.0;
26575ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson      DWORD index, size = sizeof(keyName) - 1;
26633cc2437ae3a609cdc44179931a2909eb48a8200Nuno Lopes      for (index = 0; RegEnumKeyEx(hTopKey, index, keyName, &size, NULL,
26733cc2437ae3a609cdc44179931a2909eb48a8200Nuno Lopes          NULL, NULL, NULL) == ERROR_SUCCESS; index++) {
26833cc2437ae3a609cdc44179931a2909eb48a8200Nuno Lopes        const char *sp = keyName;
26933cc2437ae3a609cdc44179931a2909eb48a8200Nuno Lopes        while (*sp && !isdigit(*sp))
27033cc2437ae3a609cdc44179931a2909eb48a8200Nuno Lopes          sp++;
27133cc2437ae3a609cdc44179931a2909eb48a8200Nuno Lopes        if (!*sp)
27233cc2437ae3a609cdc44179931a2909eb48a8200Nuno Lopes          continue;
27333cc2437ae3a609cdc44179931a2909eb48a8200Nuno Lopes        const char *ep = sp + 1;
27433cc2437ae3a609cdc44179931a2909eb48a8200Nuno Lopes        while (*ep && (isdigit(*ep) || (*ep == '.')))
27533cc2437ae3a609cdc44179931a2909eb48a8200Nuno Lopes          ep++;
27633cc2437ae3a609cdc44179931a2909eb48a8200Nuno Lopes        char numBuf[32];
27733cc2437ae3a609cdc44179931a2909eb48a8200Nuno Lopes        strncpy(numBuf, sp, sizeof(numBuf) - 1);
27833cc2437ae3a609cdc44179931a2909eb48a8200Nuno Lopes        numBuf[sizeof(numBuf) - 1] = '\0';
27933cc2437ae3a609cdc44179931a2909eb48a8200Nuno Lopes        double value = strtod(numBuf, NULL);
28033cc2437ae3a609cdc44179931a2909eb48a8200Nuno Lopes        if (value > bestValue) {
28133cc2437ae3a609cdc44179931a2909eb48a8200Nuno Lopes          bestIndex = (int)index;
28233cc2437ae3a609cdc44179931a2909eb48a8200Nuno Lopes          bestValue = value;
28333cc2437ae3a609cdc44179931a2909eb48a8200Nuno Lopes          strcpy(bestName, keyName);
28433cc2437ae3a609cdc44179931a2909eb48a8200Nuno Lopes        }
28575ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson        size = sizeof(keyName) - 1;
28675ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson      }
28775ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson      // If we found the highest versioned key, open the key and get the value.
28875ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson      if (bestIndex != -1) {
28975ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson        // Append rest of key.
29075ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson        strncat(bestName, nextKey, sizeof(bestName) - 1);
29175ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson        bestName[sizeof(bestName) - 1] = '\0';
29275ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson        // Open the chosen key path remainder.
29375ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson        lResult = RegOpenKeyEx(hTopKey, bestName, 0, KEY_READ, &hKey);
29475ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson        if (lResult == ERROR_SUCCESS) {
29575ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson          lResult = RegQueryValueEx(hKey, valueName, NULL, &valueType,
29675ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson            (LPBYTE)value, &valueSize);
29775ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson          if (lResult == ERROR_SUCCESS)
29875ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson            returnValue = true;
29975ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson          RegCloseKey(hKey);
30075ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson        }
30175ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson      }
30275ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson      RegCloseKey(hTopKey);
30375ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson    }
30475ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson  }
30575ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson  else {
30675ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson    lResult = RegOpenKeyEx(hRootKey, subKey, 0, KEY_READ, &hKey);
30775ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson    if (lResult == ERROR_SUCCESS) {
30875ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson      lResult = RegQueryValueEx(hKey, valueName, NULL, &valueType,
30975ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson        (LPBYTE)value, &valueSize);
31075ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson      if (lResult == ERROR_SUCCESS)
31175ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson        returnValue = true;
31275ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson      RegCloseKey(hKey);
31375ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson    }
31443d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  }
31543d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  return(returnValue);
31643d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump}
317620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump#else // _MSC_VER
318620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump  // Read registry string.
3196cd5216bb61e6165d8f6e8fb781d45c035bd4e00Benjamin Kramerstatic bool getSystemRegistryString(const char*, const char*, char*, size_t) {
320620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump  return(false);
321620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump}
322620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump#endif // _MSC_VER
323121e3c207415fb4c105d690de384d8f3d49b0f2dArgyrios Kyrtzidis
32443d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  // Get Visual Studio installation directory.
3256cd5216bb61e6165d8f6e8fb781d45c035bd4e00Benjamin Kramerstatic bool getVisualStudioDir(std::string &path) {
326ff58e3610f4e12094def69eb2d6dcb4330378d8fMichael J. Spencer  // First check the environment variables that vsvars32.bat sets.
327ff58e3610f4e12094def69eb2d6dcb4330378d8fMichael J. Spencer  const char* vcinstalldir = getenv("VCINSTALLDIR");
328ff58e3610f4e12094def69eb2d6dcb4330378d8fMichael J. Spencer  if(vcinstalldir) {
329ff58e3610f4e12094def69eb2d6dcb4330378d8fMichael J. Spencer    char *p = const_cast<char *>(strstr(vcinstalldir, "\\VC"));
330ff58e3610f4e12094def69eb2d6dcb4330378d8fMichael J. Spencer    if (p)
331ff58e3610f4e12094def69eb2d6dcb4330378d8fMichael J. Spencer      *p = '\0';
332ff58e3610f4e12094def69eb2d6dcb4330378d8fMichael J. Spencer    path = vcinstalldir;
333ff58e3610f4e12094def69eb2d6dcb4330378d8fMichael J. Spencer    return(true);
334ff58e3610f4e12094def69eb2d6dcb4330378d8fMichael J. Spencer  }
335ff58e3610f4e12094def69eb2d6dcb4330378d8fMichael J. Spencer
33675ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson  char vsIDEInstallDir[256];
33780f93d9fe82ebe7e58913646248240de838283beDouglas Gregor  char vsExpressIDEInstallDir[256];
338ff58e3610f4e12094def69eb2d6dcb4330378d8fMichael J. Spencer  // Then try the windows registry.
33975ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson  bool hasVCDir = getSystemRegistryString(
34075ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson    "HKEY_LOCAL_MACHINE\\SOFTWARE\\Microsoft\\VisualStudio\\$VERSION",
34175ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson    "InstallDir", vsIDEInstallDir, sizeof(vsIDEInstallDir) - 1);
34280f93d9fe82ebe7e58913646248240de838283beDouglas Gregor  bool hasVCExpressDir = getSystemRegistryString(
34380f93d9fe82ebe7e58913646248240de838283beDouglas Gregor    "HKEY_LOCAL_MACHINE\\SOFTWARE\\Microsoft\\VCExpress\\$VERSION",
34480f93d9fe82ebe7e58913646248240de838283beDouglas Gregor    "InstallDir", vsExpressIDEInstallDir, sizeof(vsExpressIDEInstallDir) - 1);
34543d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump    // If we have both vc80 and vc90, pick version we were compiled with.
34675ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson  if (hasVCDir && vsIDEInstallDir[0]) {
347620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump    char *p = (char*)strstr(vsIDEInstallDir, "\\Common7\\IDE");
348620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump    if (p)
349620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump      *p = '\0';
350620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump    path = vsIDEInstallDir;
351620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump    return(true);
35243d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  }
35380f93d9fe82ebe7e58913646248240de838283beDouglas Gregor  else if (hasVCExpressDir && vsExpressIDEInstallDir[0]) {
35480f93d9fe82ebe7e58913646248240de838283beDouglas Gregor    char *p = (char*)strstr(vsExpressIDEInstallDir, "\\Common7\\IDE");
35580f93d9fe82ebe7e58913646248240de838283beDouglas Gregor    if (p)
35680f93d9fe82ebe7e58913646248240de838283beDouglas Gregor      *p = '\0';
35780f93d9fe82ebe7e58913646248240de838283beDouglas Gregor    path = vsExpressIDEInstallDir;
35880f93d9fe82ebe7e58913646248240de838283beDouglas Gregor    return(true);
35980f93d9fe82ebe7e58913646248240de838283beDouglas Gregor  }
360620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump  else {
361620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump    // Try the environment.
36280f93d9fe82ebe7e58913646248240de838283beDouglas Gregor    const char* vs100comntools = getenv("VS100COMNTOOLS");
363620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump    const char* vs90comntools = getenv("VS90COMNTOOLS");
364620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump    const char* vs80comntools = getenv("VS80COMNTOOLS");
365620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump    const char* vscomntools = NULL;
36680f93d9fe82ebe7e58913646248240de838283beDouglas Gregor
36780f93d9fe82ebe7e58913646248240de838283beDouglas Gregor    // Try to find the version that we were compiled with
36880f93d9fe82ebe7e58913646248240de838283beDouglas Gregor    if(false) {}
36980f93d9fe82ebe7e58913646248240de838283beDouglas Gregor    #if (_MSC_VER >= 1600)  // VC100
37080f93d9fe82ebe7e58913646248240de838283beDouglas Gregor    else if(vs100comntools) {
37180f93d9fe82ebe7e58913646248240de838283beDouglas Gregor      vscomntools = vs100comntools;
372620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump    }
37380f93d9fe82ebe7e58913646248240de838283beDouglas Gregor    #elif (_MSC_VER == 1500) // VC80
37480f93d9fe82ebe7e58913646248240de838283beDouglas Gregor    else if(vs90comntools) {
37580f93d9fe82ebe7e58913646248240de838283beDouglas Gregor      vscomntools = vs90comntools;
37680f93d9fe82ebe7e58913646248240de838283beDouglas Gregor    }
37780f93d9fe82ebe7e58913646248240de838283beDouglas Gregor    #elif (_MSC_VER == 1400) // VC80
37880f93d9fe82ebe7e58913646248240de838283beDouglas Gregor    else if(vs80comntools) {
37980f93d9fe82ebe7e58913646248240de838283beDouglas Gregor      vscomntools = vs80comntools;
38080f93d9fe82ebe7e58913646248240de838283beDouglas Gregor    }
38180f93d9fe82ebe7e58913646248240de838283beDouglas Gregor    #endif
38280f93d9fe82ebe7e58913646248240de838283beDouglas Gregor    // Otherwise find any version we can
38380f93d9fe82ebe7e58913646248240de838283beDouglas Gregor    else if (vs100comntools)
38480f93d9fe82ebe7e58913646248240de838283beDouglas Gregor      vscomntools = vs100comntools;
385620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump    else if (vs90comntools)
386620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump      vscomntools = vs90comntools;
387620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump    else if (vs80comntools)
388620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump      vscomntools = vs80comntools;
38980f93d9fe82ebe7e58913646248240de838283beDouglas Gregor
390620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump    if (vscomntools && *vscomntools) {
391cb421fa690da545b58a720abe5f1c49b166dbde7Dan Gohman      char *p = const_cast<char *>(strstr(vscomntools, "\\Common7\\Tools"));
392620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump      if (p)
393620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump        *p = '\0';
394620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump      path = vscomntools;
395620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump      return(true);
396620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump    }
397620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump    else
398620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump      return(false);
399620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump  }
400620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump  return(false);
40143d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump}
4024d374d4a4832a01d5d3b75c59cb1567e60df41e7Sebastian Redl
40375ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson  // Get Windows SDK installation directory.
4046cd5216bb61e6165d8f6e8fb781d45c035bd4e00Benjamin Kramerstatic bool getWindowsSDKDir(std::string &path) {
40575ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson  char windowsSDKInstallDir[256];
40675ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson  // Try the Windows registry.
40775ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson  bool hasSDKDir = getSystemRegistryString(
40875ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson   "HKEY_LOCAL_MACHINE\\SOFTWARE\\Microsoft\\Microsoft SDKs\\Windows\\$VERSION",
40975ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson    "InstallationFolder", windowsSDKInstallDir, sizeof(windowsSDKInstallDir) - 1);
41075ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson    // If we have both vc80 and vc90, pick version we were compiled with.
41175ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson  if (hasSDKDir && windowsSDKInstallDir[0]) {
41275ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson    path = windowsSDKInstallDir;
41375ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson    return(true);
41475ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson  }
41575ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson  return(false);
41675ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson}
41775ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson
41879bc57c074cb91f574fba19215f95b63dcf6b04bmike-mvoid InitHeaderSearch::AddDefaultCIncludePaths(const llvm::Triple &triple,
41979bc57c074cb91f574fba19215f95b63dcf6b04bmike-m                                            const HeaderSearchOptions &HSOpts) {
42043d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  // FIXME: temporary hack: hard-coded paths.
42179bc57c074cb91f574fba19215f95b63dcf6b04bmike-m  AddPath("/usr/local/include", System, true, false, false);
42279bc57c074cb91f574fba19215f95b63dcf6b04bmike-m
42379bc57c074cb91f574fba19215f95b63dcf6b04bmike-m  // Builtin includes use #include_next directives and should be positioned
42479bc57c074cb91f574fba19215f95b63dcf6b04bmike-m  // just prior C include dirs.
42579bc57c074cb91f574fba19215f95b63dcf6b04bmike-m  if (HSOpts.UseBuiltinIncludes) {
42679bc57c074cb91f574fba19215f95b63dcf6b04bmike-m    // Ignore the sys root, we *always* look for clang headers relative to
42779bc57c074cb91f574fba19215f95b63dcf6b04bmike-m    // supplied path.
42879bc57c074cb91f574fba19215f95b63dcf6b04bmike-m    llvm::sys::Path P(HSOpts.ResourceDir);
42979bc57c074cb91f574fba19215f95b63dcf6b04bmike-m    P.appendComponent("include");
43079bc57c074cb91f574fba19215f95b63dcf6b04bmike-m    AddPath(P.str(), System, false, false, false, /*IgnoreSysRoot=*/ true);
43179bc57c074cb91f574fba19215f95b63dcf6b04bmike-m  }
43279bc57c074cb91f574fba19215f95b63dcf6b04bmike-m
43379bc57c074cb91f574fba19215f95b63dcf6b04bmike-m  // Add dirs specified via 'configure --with-c-include-dirs'.
434c706468b1e00bab49ca92eb500a39657f530f828Daniel Dunbar  llvm::StringRef CIncludeDirs(C_INCLUDE_DIRS);
435c706468b1e00bab49ca92eb500a39657f530f828Daniel Dunbar  if (CIncludeDirs != "") {
436aadd7a48453b4f58bd8e1e9eb670918ee7d6a711Rafael Espindola    llvm::SmallVector<llvm::StringRef, 5> dirs;
437aadd7a48453b4f58bd8e1e9eb670918ee7d6a711Rafael Espindola    CIncludeDirs.split(dirs, ":");
438aadd7a48453b4f58bd8e1e9eb670918ee7d6a711Rafael Espindola    for (llvm::SmallVectorImpl<llvm::StringRef>::iterator i = dirs.begin();
439aadd7a48453b4f58bd8e1e9eb670918ee7d6a711Rafael Espindola         i != dirs.end();
440aadd7a48453b4f58bd8e1e9eb670918ee7d6a711Rafael Espindola         ++i)
441f0a2f51be576089e1a84cd677c09156dae067b6aRafael Espindola      AddPath(*i, System, false, false, false);
442f0a2f51be576089e1a84cd677c09156dae067b6aRafael Espindola    return;
443f0a2f51be576089e1a84cd677c09156dae067b6aRafael Espindola  }
44443d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  llvm::Triple::OSType os = triple.getOS();
44543d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  switch (os) {
44643d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  case llvm::Triple::Win32:
44743d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump    {
448620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump      std::string VSDir;
44975ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson      std::string WindowsSDKDir;
450620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump      if (getVisualStudioDir(VSDir)) {
451620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump        AddPath(VSDir + "\\VC\\include", System, false, false, false);
45275ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson        if (getWindowsSDKDir(WindowsSDKDir))
453ff58e3610f4e12094def69eb2d6dcb4330378d8fMichael J. Spencer          AddPath(WindowsSDKDir + "\\include", System, false, false, false);
45475ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson        else
45575ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson          AddPath(VSDir + "\\VC\\PlatformSDK\\Include",
45675ee3bd6e7ed2b4286d5e717ae7b94411f90b3b9John Thompson            System, false, false, false);
457620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump      }
4589319f02dea2449caf5c7a0d07c596ca324e43574John Thompson      else {
4599319f02dea2449caf5c7a0d07c596ca324e43574John Thompson          // Default install paths.
46080f93d9fe82ebe7e58913646248240de838283beDouglas Gregor        AddPath("C:/Program Files/Microsoft Visual Studio 10.0/VC/include",
46180f93d9fe82ebe7e58913646248240de838283beDouglas Gregor          System, false, false, false);
4629319f02dea2449caf5c7a0d07c596ca324e43574John Thompson        AddPath("C:/Program Files/Microsoft Visual Studio 9.0/VC/include",
4639319f02dea2449caf5c7a0d07c596ca324e43574John Thompson          System, false, false, false);
4649319f02dea2449caf5c7a0d07c596ca324e43574John Thompson        AddPath(
4659319f02dea2449caf5c7a0d07c596ca324e43574John Thompson        "C:/Program Files/Microsoft Visual Studio 9.0/VC/PlatformSDK/Include",
4669319f02dea2449caf5c7a0d07c596ca324e43574John Thompson          System, false, false, false);
4679319f02dea2449caf5c7a0d07c596ca324e43574John Thompson        AddPath("C:/Program Files/Microsoft Visual Studio 8/VC/include",
4689319f02dea2449caf5c7a0d07c596ca324e43574John Thompson          System, false, false, false);
4699319f02dea2449caf5c7a0d07c596ca324e43574John Thompson        AddPath(
4709319f02dea2449caf5c7a0d07c596ca324e43574John Thompson        "C:/Program Files/Microsoft Visual Studio 8/VC/PlatformSDK/Include",
4719319f02dea2449caf5c7a0d07c596ca324e43574John Thompson          System, false, false, false);
4729319f02dea2449caf5c7a0d07c596ca324e43574John Thompson          // For some clang developers.
4739319f02dea2449caf5c7a0d07c596ca324e43574John Thompson        AddPath("G:/Program Files/Microsoft Visual Studio 9.0/VC/include",
4749319f02dea2449caf5c7a0d07c596ca324e43574John Thompson          System, false, false, false);
4759319f02dea2449caf5c7a0d07c596ca324e43574John Thompson        AddPath(
4769319f02dea2449caf5c7a0d07c596ca324e43574John Thompson        "G:/Program Files/Microsoft Visual Studio 9.0/VC/PlatformSDK/Include",
4779319f02dea2449caf5c7a0d07c596ca324e43574John Thompson          System, false, false, false);
4789319f02dea2449caf5c7a0d07c596ca324e43574John Thompson      }
47943d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump    }
48043d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump    break;
48186ed3a326342fa2a8c9d9b5d1d7280c9bad2b007Chris Lattner  case llvm::Triple::Haiku:
48286ed3a326342fa2a8c9d9b5d1d7280c9bad2b007Chris Lattner    AddPath("/boot/common/include", System, true, false, false);
48386ed3a326342fa2a8c9d9b5d1d7280c9bad2b007Chris Lattner    AddPath("/boot/develop/headers/os", System, true, false, false);
48486ed3a326342fa2a8c9d9b5d1d7280c9bad2b007Chris Lattner    AddPath("/boot/develop/headers/os/app", System, true, false, false);
48586ed3a326342fa2a8c9d9b5d1d7280c9bad2b007Chris Lattner    AddPath("/boot/develop/headers/os/arch", System, true, false, false);
48686ed3a326342fa2a8c9d9b5d1d7280c9bad2b007Chris Lattner    AddPath("/boot/develop/headers/os/device", System, true, false, false);
48786ed3a326342fa2a8c9d9b5d1d7280c9bad2b007Chris Lattner    AddPath("/boot/develop/headers/os/drivers", System, true, false, false);
48886ed3a326342fa2a8c9d9b5d1d7280c9bad2b007Chris Lattner    AddPath("/boot/develop/headers/os/game", System, true, false, false);
48986ed3a326342fa2a8c9d9b5d1d7280c9bad2b007Chris Lattner    AddPath("/boot/develop/headers/os/interface", System, true, false, false);
49086ed3a326342fa2a8c9d9b5d1d7280c9bad2b007Chris Lattner    AddPath("/boot/develop/headers/os/kernel", System, true, false, false);
49186ed3a326342fa2a8c9d9b5d1d7280c9bad2b007Chris Lattner    AddPath("/boot/develop/headers/os/locale", System, true, false, false);
49286ed3a326342fa2a8c9d9b5d1d7280c9bad2b007Chris Lattner    AddPath("/boot/develop/headers/os/mail", System, true, false, false);
49386ed3a326342fa2a8c9d9b5d1d7280c9bad2b007Chris Lattner    AddPath("/boot/develop/headers/os/media", System, true, false, false);
49486ed3a326342fa2a8c9d9b5d1d7280c9bad2b007Chris Lattner    AddPath("/boot/develop/headers/os/midi", System, true, false, false);
49586ed3a326342fa2a8c9d9b5d1d7280c9bad2b007Chris Lattner    AddPath("/boot/develop/headers/os/midi2", System, true, false, false);
49686ed3a326342fa2a8c9d9b5d1d7280c9bad2b007Chris Lattner    AddPath("/boot/develop/headers/os/net", System, true, false, false);
49786ed3a326342fa2a8c9d9b5d1d7280c9bad2b007Chris Lattner    AddPath("/boot/develop/headers/os/storage", System, true, false, false);
49886ed3a326342fa2a8c9d9b5d1d7280c9bad2b007Chris Lattner    AddPath("/boot/develop/headers/os/support", System, true, false, false);
49986ed3a326342fa2a8c9d9b5d1d7280c9bad2b007Chris Lattner    AddPath("/boot/develop/headers/os/translation",
50086ed3a326342fa2a8c9d9b5d1d7280c9bad2b007Chris Lattner      System, true, false, false);
50186ed3a326342fa2a8c9d9b5d1d7280c9bad2b007Chris Lattner    AddPath("/boot/develop/headers/os/add-ons/graphics",
50286ed3a326342fa2a8c9d9b5d1d7280c9bad2b007Chris Lattner      System, true, false, false);
50386ed3a326342fa2a8c9d9b5d1d7280c9bad2b007Chris Lattner    AddPath("/boot/develop/headers/os/add-ons/input_server",
50486ed3a326342fa2a8c9d9b5d1d7280c9bad2b007Chris Lattner      System, true, false, false);
50586ed3a326342fa2a8c9d9b5d1d7280c9bad2b007Chris Lattner    AddPath("/boot/develop/headers/os/add-ons/screen_saver",
50686ed3a326342fa2a8c9d9b5d1d7280c9bad2b007Chris Lattner      System, true, false, false);
50786ed3a326342fa2a8c9d9b5d1d7280c9bad2b007Chris Lattner    AddPath("/boot/develop/headers/os/add-ons/tracker",
50886ed3a326342fa2a8c9d9b5d1d7280c9bad2b007Chris Lattner      System, true, false, false);
50986ed3a326342fa2a8c9d9b5d1d7280c9bad2b007Chris Lattner    AddPath("/boot/develop/headers/os/be_apps/Deskbar",
51086ed3a326342fa2a8c9d9b5d1d7280c9bad2b007Chris Lattner      System, true, false, false);
51186ed3a326342fa2a8c9d9b5d1d7280c9bad2b007Chris Lattner    AddPath("/boot/develop/headers/os/be_apps/NetPositive",
51286ed3a326342fa2a8c9d9b5d1d7280c9bad2b007Chris Lattner      System, true, false, false);
51386ed3a326342fa2a8c9d9b5d1d7280c9bad2b007Chris Lattner    AddPath("/boot/develop/headers/os/be_apps/Tracker",
51486ed3a326342fa2a8c9d9b5d1d7280c9bad2b007Chris Lattner      System, true, false, false);
51586ed3a326342fa2a8c9d9b5d1d7280c9bad2b007Chris Lattner    AddPath("/boot/develop/headers/cpp", System, true, false, false);
51686ed3a326342fa2a8c9d9b5d1d7280c9bad2b007Chris Lattner    AddPath("/boot/develop/headers/cpp/i586-pc-haiku",
51786ed3a326342fa2a8c9d9b5d1d7280c9bad2b007Chris Lattner      System, true, false, false);
51886ed3a326342fa2a8c9d9b5d1d7280c9bad2b007Chris Lattner    AddPath("/boot/develop/headers/3rdparty", System, true, false, false);
51986ed3a326342fa2a8c9d9b5d1d7280c9bad2b007Chris Lattner    AddPath("/boot/develop/headers/bsd", System, true, false, false);
52086ed3a326342fa2a8c9d9b5d1d7280c9bad2b007Chris Lattner    AddPath("/boot/develop/headers/glibc", System, true, false, false);
52186ed3a326342fa2a8c9d9b5d1d7280c9bad2b007Chris Lattner    AddPath("/boot/develop/headers/posix", System, true, false, false);
52286ed3a326342fa2a8c9d9b5d1d7280c9bad2b007Chris Lattner    AddPath("/boot/develop/headers",  System, true, false, false);
52386ed3a326342fa2a8c9d9b5d1d7280c9bad2b007Chris Lattner  	break;
52443d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  case llvm::Triple::MinGW64:
525620d57a293143e3f07d6e4f5ba50020a80f45564Mike Stump  case llvm::Triple::MinGW32:
52643d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump    AddPath("c:/mingw/include", System, true, false, false);
52743d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump    break;
52843d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump  default:
52943d8176d2e8e304b2d419fb0fe139cc07af80deaMike Stump    break;
5300fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  }
531d3f88343af2dec3459493fafa22532317321eaf0John Thompson
532d3f88343af2dec3459493fafa22532317321eaf0John Thompson  AddPath("/usr/include", System, false, false, false);
533e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola}
534e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola
5350e3cc05542e62fe89345ce94a73494525ee463e0Chris Lattnervoid InitHeaderSearch::
5360e3cc05542e62fe89345ce94a73494525ee463e0Chris LattnerAddDefaultCPlusPlusIncludePaths(const llvm::Triple &triple) {
537e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola  llvm::Triple::OSType os = triple.getOS();
538ab7ae95c46f288ac52974aa60334a9575f9a850cRafael Espindola  llvm::StringRef CxxIncludeRoot(CXX_INCLUDE_ROOT);
539ab7ae95c46f288ac52974aa60334a9575f9a850cRafael Espindola  if (CxxIncludeRoot != "") {
540ab7ae95c46f288ac52974aa60334a9575f9a850cRafael Espindola    llvm::StringRef CxxIncludeArch(CXX_INCLUDE_ARCH);
541ab7ae95c46f288ac52974aa60334a9575f9a850cRafael Espindola    if (CxxIncludeArch == "")
542ab7ae95c46f288ac52974aa60334a9575f9a850cRafael Espindola      AddGnuCPlusPlusIncludePaths(CxxIncludeRoot, triple.str().c_str(),
5430e3cc05542e62fe89345ce94a73494525ee463e0Chris Lattner                                  CXX_INCLUDE_32BIT_DIR, CXX_INCLUDE_64BIT_DIR,
5440e3cc05542e62fe89345ce94a73494525ee463e0Chris Lattner                                  triple);
545ab7ae95c46f288ac52974aa60334a9575f9a850cRafael Espindola    else
546ab7ae95c46f288ac52974aa60334a9575f9a850cRafael Espindola      AddGnuCPlusPlusIncludePaths(CxxIncludeRoot, CXX_INCLUDE_ARCH,
5470e3cc05542e62fe89345ce94a73494525ee463e0Chris Lattner                                  CXX_INCLUDE_32BIT_DIR, CXX_INCLUDE_64BIT_DIR,
5480e3cc05542e62fe89345ce94a73494525ee463e0Chris Lattner                                  triple);
549ab7ae95c46f288ac52974aa60334a9575f9a850cRafael Espindola    return;
550ab7ae95c46f288ac52974aa60334a9575f9a850cRafael Espindola  }
551e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola  // FIXME: temporary hack: hard-coded paths.
552e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola  switch (os) {
553e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola  case llvm::Triple::Cygwin:
554e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    AddPath("/lib/gcc/i686-pc-cygwin/3.4.4/include",
555e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola        System, true, false, false);
556e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    AddPath("/lib/gcc/i686-pc-cygwin/3.4.4/include/c++",
557e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola        System, true, false, false);
558ea9e56d38485921d596f2366d625c471a6d3c15aDouglas Gregor    AddPath("/lib/gcc/i686-pc-cygwin/3.4.4/include/c++/i686-pc-cygwin",
559ea9e56d38485921d596f2366d625c471a6d3c15aDouglas Gregor        System, true, false, false);
560e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    break;
561e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola  case llvm::Triple::MinGW64:
562e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    // Try gcc 4.4.0
563e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    AddMinGWCPlusPlusIncludePaths("c:/MinGW/lib/gcc", "mingw64", "4.4.0");
564e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    // Try gcc 4.3.0
565e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    AddMinGWCPlusPlusIncludePaths("c:/MinGW/lib/gcc", "mingw64", "4.3.0");
566e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    // Fall through.
567e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola  case llvm::Triple::MinGW32:
568e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    // Try gcc 4.4.0
569e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    AddMinGWCPlusPlusIncludePaths("c:/MinGW/lib/gcc", "mingw32", "4.4.0");
570e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    // Try gcc 4.3.0
571e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    AddMinGWCPlusPlusIncludePaths("c:/MinGW/lib/gcc", "mingw32", "4.3.0");
572e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    break;
573e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola  case llvm::Triple::Darwin:
574f2070b374c1d6fc72aa6bdb2f49c1efdc4f6d963Daniel Dunbar    switch (triple.getArch()) {
575f2070b374c1d6fc72aa6bdb2f49c1efdc4f6d963Daniel Dunbar    default: break;
576f2070b374c1d6fc72aa6bdb2f49c1efdc4f6d963Daniel Dunbar
577582c30135dce1857db62e58b916ab0fa5edfb1d6Douglas Gregor    case llvm::Triple::ppc:
578582c30135dce1857db62e58b916ab0fa5edfb1d6Douglas Gregor    case llvm::Triple::ppc64:
579616d4362f01a8f819f9351372c3c4368777d4480Douglas Gregor      AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.2.1",
580616d4362f01a8f819f9351372c3c4368777d4480Douglas Gregor                                  "powerpc-apple-darwin10", "", "ppc64",
581616d4362f01a8f819f9351372c3c4368777d4480Douglas Gregor                                  triple);
582582c30135dce1857db62e58b916ab0fa5edfb1d6Douglas Gregor      AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.0.0",
583582c30135dce1857db62e58b916ab0fa5edfb1d6Douglas Gregor                                  "powerpc-apple-darwin10", "", "ppc64",
584582c30135dce1857db62e58b916ab0fa5edfb1d6Douglas Gregor                                  triple);
585582c30135dce1857db62e58b916ab0fa5edfb1d6Douglas Gregor      break;
586582c30135dce1857db62e58b916ab0fa5edfb1d6Douglas Gregor
587f2070b374c1d6fc72aa6bdb2f49c1efdc4f6d963Daniel Dunbar    case llvm::Triple::x86:
588f2070b374c1d6fc72aa6bdb2f49c1efdc4f6d963Daniel Dunbar    case llvm::Triple::x86_64:
589f2070b374c1d6fc72aa6bdb2f49c1efdc4f6d963Daniel Dunbar      AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.2.1",
590f2070b374c1d6fc72aa6bdb2f49c1efdc4f6d963Daniel Dunbar                                  "i686-apple-darwin10", "", "x86_64", triple);
591f2070b374c1d6fc72aa6bdb2f49c1efdc4f6d963Daniel Dunbar      AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.0.0",
592f2070b374c1d6fc72aa6bdb2f49c1efdc4f6d963Daniel Dunbar                                  "i686-apple-darwin8", "", "", triple);
593f2070b374c1d6fc72aa6bdb2f49c1efdc4f6d963Daniel Dunbar      break;
594f2070b374c1d6fc72aa6bdb2f49c1efdc4f6d963Daniel Dunbar
595f2070b374c1d6fc72aa6bdb2f49c1efdc4f6d963Daniel Dunbar    case llvm::Triple::arm:
596f2070b374c1d6fc72aa6bdb2f49c1efdc4f6d963Daniel Dunbar    case llvm::Triple::thumb:
597f2070b374c1d6fc72aa6bdb2f49c1efdc4f6d963Daniel Dunbar      AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.2.1",
598f2070b374c1d6fc72aa6bdb2f49c1efdc4f6d963Daniel Dunbar                                  "arm-apple-darwin10", "v7", "", triple);
599f2070b374c1d6fc72aa6bdb2f49c1efdc4f6d963Daniel Dunbar      AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.2.1",
600f2070b374c1d6fc72aa6bdb2f49c1efdc4f6d963Daniel Dunbar                                  "arm-apple-darwin10", "v6", "", triple);
601f2070b374c1d6fc72aa6bdb2f49c1efdc4f6d963Daniel Dunbar      break;
602f2070b374c1d6fc72aa6bdb2f49c1efdc4f6d963Daniel Dunbar    }
603e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    break;
6047a7ca281bcf50b34de1ce067d3c55879b81722aeChris Lattner  case llvm::Triple::DragonFly:
6057a7ca281bcf50b34de1ce067d3c55879b81722aeChris Lattner    AddPath("/usr/include/c++/4.1", System, true, false, false);
6067a7ca281bcf50b34de1ce067d3c55879b81722aeChris Lattner    break;
607e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola  case llvm::Triple::Linux:
608ac78b7a7af0e67f2f4f45d9d94393c281ff4417bmike-m    //===------------------------------------------------------------------===//
609ac78b7a7af0e67f2f4f45d9d94393c281ff4417bmike-m    // Debian based distros.
610ac78b7a7af0e67f2f4f45d9d94393c281ff4417bmike-m    // Note: these distros symlink /usr/include/c++/X.Y.Z -> X.Y
611ac78b7a7af0e67f2f4f45d9d94393c281ff4417bmike-m    //===------------------------------------------------------------------===//
612ac78b7a7af0e67f2f4f45d9d94393c281ff4417bmike-m    // Ubuntu 10.04 LTS "Lucid Lynx" -- gcc-4.4.3
613ac78b7a7af0e67f2f4f45d9d94393c281ff4417bmike-m    // Ubuntu 9.10 "Karmic Koala"    -- gcc-4.4.1
614ac78b7a7af0e67f2f4f45d9d94393c281ff4417bmike-m    // Debian 6.0 "squeeze"          -- gcc-4.4.2
615e69f047a6c882ce36bd260a8fed0ae31114a5a7eNick Lewycky    AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.4",
616331e8010aed1c2175328ed470aea75e93ff70b32Torok Edwin                                "x86_64-linux-gnu", "32", "", triple);
617e69f047a6c882ce36bd260a8fed0ae31114a5a7eNick Lewycky    AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.4",
618ac78b7a7af0e67f2f4f45d9d94393c281ff4417bmike-m                                "i486-linux-gnu", "", "64", triple);
619ac78b7a7af0e67f2f4f45d9d94393c281ff4417bmike-m    // Ubuntu 9.04 "Jaunty Jackalope" -- gcc-4.3.3
620ac78b7a7af0e67f2f4f45d9d94393c281ff4417bmike-m    // Ubuntu 8.10 "Intrepid Ibex"    -- gcc-4.3.2
621ac78b7a7af0e67f2f4f45d9d94393c281ff4417bmike-m    // Debian 5.0 "lenny"             -- gcc-4.3.2
622ac78b7a7af0e67f2f4f45d9d94393c281ff4417bmike-m    AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.3",
623ab7ae95c46f288ac52974aa60334a9575f9a850cRafael Espindola                                "x86_64-linux-gnu", "32", "", triple);
624ac78b7a7af0e67f2f4f45d9d94393c281ff4417bmike-m    AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.3",
625ac78b7a7af0e67f2f4f45d9d94393c281ff4417bmike-m                                "i486-linux-gnu", "", "64", triple);
626e7d6c2cae6a9b9985db02cfc8bb4e7b714630ed5Rafael Espindola    AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.3",
627e7d6c2cae6a9b9985db02cfc8bb4e7b714630ed5Rafael Espindola                                "arm-linux-gnueabi", "", "", triple);
628ac78b7a7af0e67f2f4f45d9d94393c281ff4417bmike-m    // Ubuntu 8.04.4 LTS "Hardy Heron"     -- gcc-4.2.4
629ac78b7a7af0e67f2f4f45d9d94393c281ff4417bmike-m    // Ubuntu 8.04.[0-3] LTS "Hardy Heron" -- gcc-4.2.3
630ac78b7a7af0e67f2f4f45d9d94393c281ff4417bmike-m    AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.2",
631ac78b7a7af0e67f2f4f45d9d94393c281ff4417bmike-m                                "x86_64-linux-gnu", "32", "", triple);
632ac78b7a7af0e67f2f4f45d9d94393c281ff4417bmike-m    AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.2",
633ac78b7a7af0e67f2f4f45d9d94393c281ff4417bmike-m                                "i486-linux-gnu", "", "64", triple);
634ac78b7a7af0e67f2f4f45d9d94393c281ff4417bmike-m    // Ubuntu 7.10 "Gutsy Gibbon" -- gcc-4.1.3
635ac78b7a7af0e67f2f4f45d9d94393c281ff4417bmike-m    AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.1",
636ac78b7a7af0e67f2f4f45d9d94393c281ff4417bmike-m                                "x86_64-linux-gnu", "32", "", triple);
637ac78b7a7af0e67f2f4f45d9d94393c281ff4417bmike-m    AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.1",
638ac78b7a7af0e67f2f4f45d9d94393c281ff4417bmike-m                                "i486-linux-gnu", "", "64", triple);
639189a14869a36ecf8bd5edc94255a34061044efe2Nuno Lopes
640ac78b7a7af0e67f2f4f45d9d94393c281ff4417bmike-m    //===------------------------------------------------------------------===//
641ac78b7a7af0e67f2f4f45d9d94393c281ff4417bmike-m    // Redhat based distros.
642ac78b7a7af0e67f2f4f45d9d94393c281ff4417bmike-m    //===------------------------------------------------------------------===//
643ac78b7a7af0e67f2f4f45d9d94393c281ff4417bmike-m    // Fedora 13
6444336e191cc52aa1ef72f9a542f5a6d5f9f02a825Chris Lattner    AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.4.4",
6454336e191cc52aa1ef72f9a542f5a6d5f9f02a825Chris Lattner                                "x86_64-redhat-linux", "32", "", triple);
6464336e191cc52aa1ef72f9a542f5a6d5f9f02a825Chris Lattner    AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.4.4",
6474336e191cc52aa1ef72f9a542f5a6d5f9f02a825Chris Lattner                                "i686-redhat-linux","", "", triple);
648ac78b7a7af0e67f2f4f45d9d94393c281ff4417bmike-m    // Fedora 12
649ac78b7a7af0e67f2f4f45d9d94393c281ff4417bmike-m    AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.4.3",
6500720b5159010479c08f8264a8a209037ecf55922Chris Lattner                                "x86_64-redhat-linux", "32", "", triple);
651ac78b7a7af0e67f2f4f45d9d94393c281ff4417bmike-m    AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.4.3",
652ac78b7a7af0e67f2f4f45d9d94393c281ff4417bmike-m                                "i686-redhat-linux","", "", triple);
653ac78b7a7af0e67f2f4f45d9d94393c281ff4417bmike-m    // Fedora 12 (pre-FEB-2010)
654ac78b7a7af0e67f2f4f45d9d94393c281ff4417bmike-m    AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.4.2",
655ac78b7a7af0e67f2f4f45d9d94393c281ff4417bmike-m                                "x86_64-redhat-linux", "32", "", triple);
656ac78b7a7af0e67f2f4f45d9d94393c281ff4417bmike-m    AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.4.2",
657ac78b7a7af0e67f2f4f45d9d94393c281ff4417bmike-m                                "i686-redhat-linux","", "", triple);
658189a14869a36ecf8bd5edc94255a34061044efe2Nuno Lopes    // Fedora 11
659189a14869a36ecf8bd5edc94255a34061044efe2Nuno Lopes    AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.4.1",
660ac78b7a7af0e67f2f4f45d9d94393c281ff4417bmike-m                                "x86_64-redhat-linux", "32", "", triple);
661ea00f847a0e1771b1fcfe683d3506940dc2de1ecChris Lattner    AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.4.1",
662ac78b7a7af0e67f2f4f45d9d94393c281ff4417bmike-m                                "i586-redhat-linux","", "", triple);
663ac78b7a7af0e67f2f4f45d9d94393c281ff4417bmike-m    // Fedora 10
664ac78b7a7af0e67f2f4f45d9d94393c281ff4417bmike-m    AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.3.2",
665ea00f847a0e1771b1fcfe683d3506940dc2de1ecChris Lattner                                "x86_64-redhat-linux", "32", "", triple);
666ac78b7a7af0e67f2f4f45d9d94393c281ff4417bmike-m    AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.3.2",
667ac78b7a7af0e67f2f4f45d9d94393c281ff4417bmike-m                                "i386-redhat-linux","", "", triple);
668ac78b7a7af0e67f2f4f45d9d94393c281ff4417bmike-m    // Fedora 9
669ac78b7a7af0e67f2f4f45d9d94393c281ff4417bmike-m    AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.3.0",
670ea00f847a0e1771b1fcfe683d3506940dc2de1ecChris Lattner                                "x86_64-redhat-linux", "32", "", triple);
671ac78b7a7af0e67f2f4f45d9d94393c281ff4417bmike-m    AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.3.0",
672ac78b7a7af0e67f2f4f45d9d94393c281ff4417bmike-m                                "i386-redhat-linux", "", "", triple);
673ac78b7a7af0e67f2f4f45d9d94393c281ff4417bmike-m    // Fedora 8
674ac78b7a7af0e67f2f4f45d9d94393c281ff4417bmike-m    AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.1.2",
675ac78b7a7af0e67f2f4f45d9d94393c281ff4417bmike-m                                "x86_64-redhat-linux", "", "", triple);
676ac78b7a7af0e67f2f4f45d9d94393c281ff4417bmike-m    AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.1.2",
677ac78b7a7af0e67f2f4f45d9d94393c281ff4417bmike-m                                "i386-redhat-linux", "", "", triple);
678ea00f847a0e1771b1fcfe683d3506940dc2de1ecChris Lattner
679ac78b7a7af0e67f2f4f45d9d94393c281ff4417bmike-m    //===------------------------------------------------------------------===//
680ea00f847a0e1771b1fcfe683d3506940dc2de1ecChris Lattner
681ac78b7a7af0e67f2f4f45d9d94393c281ff4417bmike-m    // Exherbo (2010-01-25)
682ea00f847a0e1771b1fcfe683d3506940dc2de1ecChris Lattner    AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.4.3",
683ac78b7a7af0e67f2f4f45d9d94393c281ff4417bmike-m                                "x86_64-pc-linux-gnu", "32", "", triple);
684ac78b7a7af0e67f2f4f45d9d94393c281ff4417bmike-m    AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.4.3",
685ac78b7a7af0e67f2f4f45d9d94393c281ff4417bmike-m                                "i686-pc-linux-gnu", "", "", triple);
686ea00f847a0e1771b1fcfe683d3506940dc2de1ecChris Lattner
687e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    // openSUSE 11.1 32 bit
688e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.3",
689ab7ae95c46f288ac52974aa60334a9575f9a850cRafael Espindola                                "i586-suse-linux", "", "", triple);
690e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    // openSUSE 11.1 64 bit
691e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.3",
692ab7ae95c46f288ac52974aa60334a9575f9a850cRafael Espindola                                "x86_64-suse-linux", "32", "", triple);
693e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    // openSUSE 11.2
694e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.4",
695ab7ae95c46f288ac52974aa60334a9575f9a850cRafael Espindola                                "i586-suse-linux", "", "", triple);
696e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.4",
697ab7ae95c46f288ac52974aa60334a9575f9a850cRafael Espindola                                "x86_64-suse-linux", "", "", triple);
698e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    // Arch Linux 2008-06-24
699e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.3.1",
700ab7ae95c46f288ac52974aa60334a9575f9a850cRafael Espindola                                "i686-pc-linux-gnu", "", "", triple);
701e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.3.1",
702ab7ae95c46f288ac52974aa60334a9575f9a850cRafael Espindola                                "x86_64-unknown-linux-gnu", "", "", triple);
703e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    // Gentoo x86 2009.1 stable
704e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    AddGnuCPlusPlusIncludePaths(
70540d1bb6383c14d5f8249d0640e8088a91b694c66John Thompson      "/usr/lib/gcc/i686-pc-linux-gnu/4.3.4/include/g++-v4",
706ab7ae95c46f288ac52974aa60334a9575f9a850cRafael Espindola      "i686-pc-linux-gnu", "", "", triple);
707e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    // Gentoo x86 2009.0 stable
708e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    AddGnuCPlusPlusIncludePaths(
70940d1bb6383c14d5f8249d0640e8088a91b694c66John Thompson      "/usr/lib/gcc/i686-pc-linux-gnu/4.3.2/include/g++-v4",
710ab7ae95c46f288ac52974aa60334a9575f9a850cRafael Espindola      "i686-pc-linux-gnu", "", "", triple);
711e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    // Gentoo x86 2008.0 stable
712e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    AddGnuCPlusPlusIncludePaths(
71340d1bb6383c14d5f8249d0640e8088a91b694c66John Thompson      "/usr/lib/gcc/i686-pc-linux-gnu/4.1.2/include/g++-v4",
714ab7ae95c46f288ac52974aa60334a9575f9a850cRafael Espindola      "i686-pc-linux-gnu", "", "", triple);
715e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    // Gentoo amd64 stable
716e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    AddGnuCPlusPlusIncludePaths(
717e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola        "/usr/lib/gcc/x86_64-pc-linux-gnu/4.1.2/include/g++-v4",
718ab7ae95c46f288ac52974aa60334a9575f9a850cRafael Espindola        "i686-pc-linux-gnu", "", "", triple);
71970d9d4196e6531926ca18b2ca850eb218041488dEric Christopher
72070d9d4196e6531926ca18b2ca850eb218041488dEric Christopher    // Gentoo amd64 gcc 4.3.2
72170d9d4196e6531926ca18b2ca850eb218041488dEric Christopher    AddGnuCPlusPlusIncludePaths(
72270d9d4196e6531926ca18b2ca850eb218041488dEric Christopher        "/usr/lib/gcc/x86_64-pc-linux-gnu/4.3.2/include/g++-v4",
72370d9d4196e6531926ca18b2ca850eb218041488dEric Christopher        "x86_64-pc-linux-gnu", "", "", triple);
72470d9d4196e6531926ca18b2ca850eb218041488dEric Christopher
72570d9d4196e6531926ca18b2ca850eb218041488dEric Christopher    // Gentoo amd64 gcc 4.4.3
72670d9d4196e6531926ca18b2ca850eb218041488dEric Christopher    AddGnuCPlusPlusIncludePaths(
72770d9d4196e6531926ca18b2ca850eb218041488dEric Christopher        "/usr/lib/gcc/x86_64-pc-linux-gnu/4.4.3/include/g++-v4",
72870d9d4196e6531926ca18b2ca850eb218041488dEric Christopher        "x86_64-pc-linux-gnu", "32", "", triple);
7296693534adb31fdd13b1ee873df3e90c99f916ad5Nick Lewycky
7306693534adb31fdd13b1ee873df3e90c99f916ad5Nick Lewycky    // Gentoo amd64 llvm-gcc trunk
7316693534adb31fdd13b1ee873df3e90c99f916ad5Nick Lewycky    AddGnuCPlusPlusIncludePaths(
7326693534adb31fdd13b1ee873df3e90c99f916ad5Nick Lewycky        "/usr/lib/llvm-gcc-4.2-9999/include/c++/4.2.1",
7336693534adb31fdd13b1ee873df3e90c99f916ad5Nick Lewycky        "x86_64-pc-linux-gnu", "", "", triple);
73470d9d4196e6531926ca18b2ca850eb218041488dEric Christopher
735e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    break;
736e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola  case llvm::Triple::FreeBSD:
737ac78b7a7af0e67f2f4f45d9d94393c281ff4417bmike-m    // FreeBSD 8.0
738ac78b7a7af0e67f2f4f45d9d94393c281ff4417bmike-m    // FreeBSD 7.3
739afe859a9a6217f177daf42ebab294a90bfe8a6abNuno Lopes    AddGnuCPlusPlusIncludePaths("/usr/include/c++/4.2", "", "", "", triple);
740e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    break;
74195c0457054518102af6250e718a13a5b2ee5dbeaDaniel Dunbar  case llvm::Triple::OpenBSD: {
74295c0457054518102af6250e718a13a5b2ee5dbeaDaniel Dunbar    std::string t = triple.getTriple();
74395c0457054518102af6250e718a13a5b2ee5dbeaDaniel Dunbar    if (t.substr(0, 6) == "x86_64")
74495c0457054518102af6250e718a13a5b2ee5dbeaDaniel Dunbar      t.replace(0, 6, "amd64");
74595c0457054518102af6250e718a13a5b2ee5dbeaDaniel Dunbar    AddGnuCPlusPlusIncludePaths("/usr/include/g++",
74695c0457054518102af6250e718a13a5b2ee5dbeaDaniel Dunbar                                t, "", "", triple);
74795c0457054518102af6250e718a13a5b2ee5dbeaDaniel Dunbar    break;
74895c0457054518102af6250e718a13a5b2ee5dbeaDaniel Dunbar  }
74938e317d6dce161b249508686cc67eb7176958762Chris Lattner  case llvm::Triple::Minix:
75038e317d6dce161b249508686cc67eb7176958762Chris Lattner    AddGnuCPlusPlusIncludePaths("/usr/gnu/include/c++/4.4.3",
75138e317d6dce161b249508686cc67eb7176958762Chris Lattner                                "", "", "", triple);
75238e317d6dce161b249508686cc67eb7176958762Chris Lattner    break;
753e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola  case llvm::Triple::Solaris:
754e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    // Solaris - Fall though..
755e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola  case llvm::Triple::AuroraUX:
756e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    // AuroraUX
757e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    AddGnuCPlusPlusIncludePaths("/opt/gcc4/include/c++/4.2.4",
758ab7ae95c46f288ac52974aa60334a9575f9a850cRafael Espindola                                "i386-pc-solaris2.11", "", "", triple);
759e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    break;
760e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola  default:
761e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola    break;
762e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola  }
763e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola}
764e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola
765e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindolavoid InitHeaderSearch::AddDefaultSystemIncludePaths(const LangOptions &Lang,
7664c2bcad7b843c10fd4a2ffd43da40bfefb4dc8baDouglas Gregor                                                    const llvm::Triple &triple,
76779bc57c074cb91f574fba19215f95b63dcf6b04bmike-m                                            const HeaderSearchOptions &HSOpts) {
76879bc57c074cb91f574fba19215f95b63dcf6b04bmike-m  if (Lang.CPlusPlus && HSOpts.UseStandardCXXIncludes)
7696ec18a3f86c702004af66cce6ff979999a736898Rafael Espindola    AddDefaultCPlusPlusIncludePaths(triple);
7706ec18a3f86c702004af66cce6ff979999a736898Rafael Espindola
77179bc57c074cb91f574fba19215f95b63dcf6b04bmike-m  AddDefaultCIncludePaths(triple, HSOpts);
772e166582f8f36f4db8f4ea157538fab7fe6bf2658Daniel Dunbar
773e166582f8f36f4db8f4ea157538fab7fe6bf2658Daniel Dunbar  // Add the default framework include paths on Darwin.
774e166582f8f36f4db8f4ea157538fab7fe6bf2658Daniel Dunbar  if (triple.getOS() == llvm::Triple::Darwin) {
775e166582f8f36f4db8f4ea157538fab7fe6bf2658Daniel Dunbar    AddPath("/System/Library/Frameworks", System, true, false, true);
776e166582f8f36f4db8f4ea157538fab7fe6bf2658Daniel Dunbar    AddPath("/Library/Frameworks", System, true, false, true);
777e166582f8f36f4db8f4ea157538fab7fe6bf2658Daniel Dunbar  }
778e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola}
779e4b255c80a786c6f16d3f0362ecb7e1b1e959f63Rafael Espindola
7800fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber/// RemoveDuplicates - If there are duplicate directory entries in the specified
7810fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber/// search list, remove the later (dead) ones.
7820fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weberstatic void RemoveDuplicates(std::vector<DirectoryLookup> &SearchList,
7830fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber                             bool Verbose) {
7840fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  llvm::SmallPtrSet<const DirectoryEntry *, 8> SeenDirs;
7850fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  llvm::SmallPtrSet<const DirectoryEntry *, 8> SeenFrameworkDirs;
7860fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  llvm::SmallPtrSet<const HeaderMap *, 8> SeenHeaderMaps;
7870fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  for (unsigned i = 0; i != SearchList.size(); ++i) {
7887a739401983eb9bc25c9b4b25177e116f5e5ba92Chris Lattner    unsigned DirToRemove = i;
7891eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
79043eee07270bf1966ea7289310066aa670c4b647fChris Lattner    const DirectoryLookup &CurEntry = SearchList[i];
7911eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
79243eee07270bf1966ea7289310066aa670c4b647fChris Lattner    if (CurEntry.isNormalDir()) {
7930fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber      // If this isn't the first time we've seen this dir, remove it.
79443eee07270bf1966ea7289310066aa670c4b647fChris Lattner      if (SeenDirs.insert(CurEntry.getDir()))
7950fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber        continue;
79643eee07270bf1966ea7289310066aa670c4b647fChris Lattner    } else if (CurEntry.isFramework()) {
7970fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber      // If this isn't the first time we've seen this framework dir, remove it.
79843eee07270bf1966ea7289310066aa670c4b647fChris Lattner      if (SeenFrameworkDirs.insert(CurEntry.getFrameworkDir()))
7990fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber        continue;
8000fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber    } else {
80143eee07270bf1966ea7289310066aa670c4b647fChris Lattner      assert(CurEntry.isHeaderMap() && "Not a headermap or normal dir?");
8020fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber      // If this isn't the first time we've seen this headermap, remove it.
80343eee07270bf1966ea7289310066aa670c4b647fChris Lattner      if (SeenHeaderMaps.insert(CurEntry.getHeaderMap()))
8040fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber        continue;
80530f05b553db067c994966daca37e75324ee7b424Chris Lattner    }
8061eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
80730f05b553db067c994966daca37e75324ee7b424Chris Lattner    // If we have a normal #include dir/framework/headermap that is shadowed
80830f05b553db067c994966daca37e75324ee7b424Chris Lattner    // later in the chain by a system include location, we actually want to
80930f05b553db067c994966daca37e75324ee7b424Chris Lattner    // ignore the user's request and drop the user dir... keeping the system
81030f05b553db067c994966daca37e75324ee7b424Chris Lattner    // dir.  This is weird, but required to emulate GCC's search path correctly.
81130f05b553db067c994966daca37e75324ee7b424Chris Lattner    //
81230f05b553db067c994966daca37e75324ee7b424Chris Lattner    // Since dupes of system dirs are rare, just rescan to find the original
81330f05b553db067c994966daca37e75324ee7b424Chris Lattner    // that we're nuking instead of using a DenseMap.
81443eee07270bf1966ea7289310066aa670c4b647fChris Lattner    if (CurEntry.getDirCharacteristic() != SrcMgr::C_User) {
81530f05b553db067c994966daca37e75324ee7b424Chris Lattner      // Find the dir that this is the same of.
81630f05b553db067c994966daca37e75324ee7b424Chris Lattner      unsigned FirstDir;
81730f05b553db067c994966daca37e75324ee7b424Chris Lattner      for (FirstDir = 0; ; ++FirstDir) {
81830f05b553db067c994966daca37e75324ee7b424Chris Lattner        assert(FirstDir != i && "Didn't find dupe?");
8191eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
82043eee07270bf1966ea7289310066aa670c4b647fChris Lattner        const DirectoryLookup &SearchEntry = SearchList[FirstDir];
82143eee07270bf1966ea7289310066aa670c4b647fChris Lattner
82230f05b553db067c994966daca37e75324ee7b424Chris Lattner        // If these are different lookup types, then they can't be the dupe.
82343eee07270bf1966ea7289310066aa670c4b647fChris Lattner        if (SearchEntry.getLookupType() != CurEntry.getLookupType())
82430f05b553db067c994966daca37e75324ee7b424Chris Lattner          continue;
8251eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
82630f05b553db067c994966daca37e75324ee7b424Chris Lattner        bool isSame;
82743eee07270bf1966ea7289310066aa670c4b647fChris Lattner        if (CurEntry.isNormalDir())
82843eee07270bf1966ea7289310066aa670c4b647fChris Lattner          isSame = SearchEntry.getDir() == CurEntry.getDir();
82943eee07270bf1966ea7289310066aa670c4b647fChris Lattner        else if (CurEntry.isFramework())
83043eee07270bf1966ea7289310066aa670c4b647fChris Lattner          isSame = SearchEntry.getFrameworkDir() == CurEntry.getFrameworkDir();
83130f05b553db067c994966daca37e75324ee7b424Chris Lattner        else {
83243eee07270bf1966ea7289310066aa670c4b647fChris Lattner          assert(CurEntry.isHeaderMap() && "Not a headermap or normal dir?");
83343eee07270bf1966ea7289310066aa670c4b647fChris Lattner          isSame = SearchEntry.getHeaderMap() == CurEntry.getHeaderMap();
83430f05b553db067c994966daca37e75324ee7b424Chris Lattner        }
8351eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
83630f05b553db067c994966daca37e75324ee7b424Chris Lattner        if (isSame)
83730f05b553db067c994966daca37e75324ee7b424Chris Lattner          break;
83830f05b553db067c994966daca37e75324ee7b424Chris Lattner      }
8391eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
84030f05b553db067c994966daca37e75324ee7b424Chris Lattner      // If the first dir in the search path is a non-system dir, zap it
84130f05b553db067c994966daca37e75324ee7b424Chris Lattner      // instead of the system one.
84230f05b553db067c994966daca37e75324ee7b424Chris Lattner      if (SearchList[FirstDir].getDirCharacteristic() == SrcMgr::C_User)
84330f05b553db067c994966daca37e75324ee7b424Chris Lattner        DirToRemove = FirstDir;
84430f05b553db067c994966daca37e75324ee7b424Chris Lattner    }
84530f05b553db067c994966daca37e75324ee7b424Chris Lattner
84630f05b553db067c994966daca37e75324ee7b424Chris Lattner    if (Verbose) {
847e7cb7e4570842297f698bd7fd8d85520fc008acdDaniel Dunbar      llvm::errs() << "ignoring duplicate directory \""
848e7cb7e4570842297f698bd7fd8d85520fc008acdDaniel Dunbar                   << CurEntry.getName() << "\"\n";
84930f05b553db067c994966daca37e75324ee7b424Chris Lattner      if (DirToRemove != i)
850e7cb7e4570842297f698bd7fd8d85520fc008acdDaniel Dunbar        llvm::errs() << "  as it is a non-system directory that duplicates "
851e7cb7e4570842297f698bd7fd8d85520fc008acdDaniel Dunbar                     << "a system directory\n";
8520fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber    }
8531eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
8547a739401983eb9bc25c9b4b25177e116f5e5ba92Chris Lattner    // This is reached if the current entry is a duplicate.  Remove the
8557a739401983eb9bc25c9b4b25177e116f5e5ba92Chris Lattner    // DirToRemove (usually the current dir).
8567a739401983eb9bc25c9b4b25177e116f5e5ba92Chris Lattner    SearchList.erase(SearchList.begin()+DirToRemove);
8570fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber    --i;
8580fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  }
8590fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber}
8600fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber
8610fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber
8620fca022d77b89100e746f4d659b84ed5b1ee0158Nico Webervoid InitHeaderSearch::Realize() {
8630fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  // Concatenate ANGLE+SYSTEM+AFTER chains together into SearchList.
8640fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  std::vector<DirectoryLookup> SearchList;
8650fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  SearchList = IncludeGroup[Angled];
8660fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  SearchList.insert(SearchList.end(), IncludeGroup[System].begin(),
8670fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber                    IncludeGroup[System].end());
8680fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  SearchList.insert(SearchList.end(), IncludeGroup[After].begin(),
8690fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber                    IncludeGroup[After].end());
8700fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  RemoveDuplicates(SearchList, Verbose);
8710fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  RemoveDuplicates(IncludeGroup[Quoted], Verbose);
8721eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
8730fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  // Prepend QUOTED list on the search list.
8741eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  SearchList.insert(SearchList.begin(), IncludeGroup[Quoted].begin(),
8750fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber                    IncludeGroup[Quoted].end());
8761eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
8770fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber
8780fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  bool DontSearchCurDir = false;  // TODO: set to true if -I- is set?
8790fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  Headers.SetSearchPaths(SearchList, IncludeGroup[Quoted].size(),
8800fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber                         DontSearchCurDir);
8810fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber
8820fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  // If verbose, print the list of directories that will be searched.
8830fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  if (Verbose) {
884e7cb7e4570842297f698bd7fd8d85520fc008acdDaniel Dunbar    llvm::errs() << "#include \"...\" search starts here:\n";
8850fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber    unsigned QuotedIdx = IncludeGroup[Quoted].size();
8860fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber    for (unsigned i = 0, e = SearchList.size(); i != e; ++i) {
8870fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber      if (i == QuotedIdx)
888e7cb7e4570842297f698bd7fd8d85520fc008acdDaniel Dunbar        llvm::errs() << "#include <...> search starts here:\n";
8890fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber      const char *Name = SearchList[i].getName();
8900fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber      const char *Suffix;
8910fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber      if (SearchList[i].isNormalDir())
8920fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber        Suffix = "";
8930fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber      else if (SearchList[i].isFramework())
8940fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber        Suffix = " (framework directory)";
8950fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber      else {
8960fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber        assert(SearchList[i].isHeaderMap() && "Unknown DirectoryLookup");
8970fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber        Suffix = " (headermap)";
8980fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber      }
899e7cb7e4570842297f698bd7fd8d85520fc008acdDaniel Dunbar      llvm::errs() << " " << Name << Suffix << "\n";
9000fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber    }
901e7cb7e4570842297f698bd7fd8d85520fc008acdDaniel Dunbar    llvm::errs() << "End of search list.\n";
9020fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber  }
9030fca022d77b89100e746f4d659b84ed5b1ee0158Nico Weber}
90463c8b77334f90472260d2f48df2742ed5067261eDaniel Dunbar
9055814e657c9ad9ef6049a2a4af0d2aad248a8a15cDaniel Dunbarvoid clang::ApplyHeaderSearchOptions(HeaderSearch &HS,
9065814e657c9ad9ef6049a2a4af0d2aad248a8a15cDaniel Dunbar                                     const HeaderSearchOptions &HSOpts,
9075814e657c9ad9ef6049a2a4af0d2aad248a8a15cDaniel Dunbar                                     const LangOptions &Lang,
90863c8b77334f90472260d2f48df2742ed5067261eDaniel Dunbar                                     const llvm::Triple &Triple) {
90963c8b77334f90472260d2f48df2742ed5067261eDaniel Dunbar  InitHeaderSearch Init(HS, HSOpts.Verbose, HSOpts.Sysroot);
91063c8b77334f90472260d2f48df2742ed5067261eDaniel Dunbar
91163c8b77334f90472260d2f48df2742ed5067261eDaniel Dunbar  // Add the user defined entries.
91263c8b77334f90472260d2f48df2742ed5067261eDaniel Dunbar  for (unsigned i = 0, e = HSOpts.UserEntries.size(); i != e; ++i) {
91363c8b77334f90472260d2f48df2742ed5067261eDaniel Dunbar    const HeaderSearchOptions::Entry &E = HSOpts.UserEntries[i];
9141b483e70ad3f670f6dd0bc62dbe0921a28195f24Daniel Dunbar    Init.AddPath(E.Path, E.Group, false, E.IsUserSupplied, E.IsFramework,
9151b483e70ad3f670f6dd0bc62dbe0921a28195f24Daniel Dunbar                 false);
91663c8b77334f90472260d2f48df2742ed5067261eDaniel Dunbar  }
91763c8b77334f90472260d2f48df2742ed5067261eDaniel Dunbar
91863c8b77334f90472260d2f48df2742ed5067261eDaniel Dunbar  // Add entries from CPATH and friends.
9199e9ddf6100fd50ca2f9f9d0106b856e20b43300fBenjamin Kramer  Init.AddDelimitedPaths(HSOpts.EnvIncPath);
920c363cb1929c7908c8c8c4b77d025c2ccb09284aeDaniel Dunbar  if (Lang.CPlusPlus && Lang.ObjC1)
9219e9ddf6100fd50ca2f9f9d0106b856e20b43300fBenjamin Kramer    Init.AddDelimitedPaths(HSOpts.ObjCXXEnvIncPath);
922c363cb1929c7908c8c8c4b77d025c2ccb09284aeDaniel Dunbar  else if (Lang.CPlusPlus)
9239e9ddf6100fd50ca2f9f9d0106b856e20b43300fBenjamin Kramer    Init.AddDelimitedPaths(HSOpts.CXXEnvIncPath);
924c363cb1929c7908c8c8c4b77d025c2ccb09284aeDaniel Dunbar  else if (Lang.ObjC1)
9259e9ddf6100fd50ca2f9f9d0106b856e20b43300fBenjamin Kramer    Init.AddDelimitedPaths(HSOpts.ObjCEnvIncPath);
926c363cb1929c7908c8c8c4b77d025c2ccb09284aeDaniel Dunbar  else
9279e9ddf6100fd50ca2f9f9d0106b856e20b43300fBenjamin Kramer    Init.AddDelimitedPaths(HSOpts.CEnvIncPath);
92863c8b77334f90472260d2f48df2742ed5067261eDaniel Dunbar
929dd35ce9a783dea7e6660f9cb1d973742f3b3c73fDaniel Dunbar  if (HSOpts.UseStandardIncludes)
93079bc57c074cb91f574fba19215f95b63dcf6b04bmike-m    Init.AddDefaultSystemIncludePaths(Lang, Triple, HSOpts);
93163c8b77334f90472260d2f48df2742ed5067261eDaniel Dunbar
93263c8b77334f90472260d2f48df2742ed5067261eDaniel Dunbar  Init.Realize();
93363c8b77334f90472260d2f48df2742ed5067261eDaniel Dunbar}
934