tools/libclang/CIndexer.cpp

   1 //===- CIndex.cpp - Clang-C Source Indexing Library -----------------------===//
   2 //
   3 //                     The LLVM Compiler Infrastructure
   4 //
   5 // This file is distributed under the University of Illinois Open Source
   6 // License. See LICENSE.TXT for details.
   7 //
   8 //===----------------------------------------------------------------------===//
   9 //
  10 // This file implements the Clang-C Source Indexing library.
  11 //
  12 //===----------------------------------------------------------------------===//
  13
  14 #include "CIndexer.h"
  15
  16 #include "clang/AST/Decl.h"
  17 #include "clang/AST/DeclVisitor.h"
  18 #include "clang/AST/StmtVisitor.h"
  19 #include "clang/Basic/FileManager.h"
  20 #include "clang/Basic/SourceManager.h"
  21 #include "clang/Basic/Version.h"
  22 #include "clang/Sema/CodeCompleteConsumer.h"
  23 #include "llvm/ADT/StringExtras.h"
  24 #include "llvm/Config/config.h"
  25 #include "llvm/Support/Compiler.h"
  26 #include "llvm/Support/MemoryBuffer.h"
  27 #include "llvm/Support/raw_ostream.h"
  28 #include "llvm/Support/Program.h"
  29
  30 #include <cstdio>
  31 #include <vector>
  32 #include <sstream>
  33
  34 #ifdef __CYGWIN__
  35 #include <sys/cygwin.h>
  36 #define LLVM_ON_WIN32 1
  37 #endif
  38
  39 #ifdef LLVM_ON_WIN32
  40 #include <windows.h>
  41 #else
  42 #include <dlfcn.h>
  43 #endif
  44
  45 using namespace clang;
  46
  47 std::string CIndexer::getClangResourcesPath() {
  48   // Did we already compute the path?
  49   if (!ResourcesPath.empty())
  50     return ResourcesPath.str();
  51
  52   // Find the location where this library lives (libclang.dylib).
  53 #ifdef LLVM_ON_WIN32
  54   MEMORY_BASIC_INFORMATION mbi;
  55   char path[MAX_PATH];
  56   VirtualQuery((void *)(uintptr_t)clang_createTranslationUnit, &mbi,
  57                sizeof(mbi));
  58   GetModuleFileNameA((HINSTANCE)mbi.AllocationBase, path, MAX_PATH);
  59
  60 #ifdef __CYGWIN__
  61   char w32path[MAX_PATH];
  62   strcpy(w32path, path);
  63   cygwin_conv_to_full_posix_path(w32path, path);
  64 #endif
  65
  66   llvm::sys::Path LibClangPath(path);
  67   LibClangPath.eraseComponent();
  68 #else
  69   // This silly cast below avoids a C++ warning.
  70   Dl_info info;
  71   if (dladdr((void *)(uintptr_t)clang_createTranslationUnit, &info) == 0)
  72     assert(0 && "Call to dladdr() failed");
  73
  74   llvm::sys::Path LibClangPath(info.dli_fname);
  75
  76   // We now have the CIndex directory, locate clang relative to it.
  77   LibClangPath.eraseComponent();
  78 #endif
  79
  80   LibClangPath.appendComponent("clang");
  81   LibClangPath.appendComponent(CLANG_VERSION_STRING);
  82
  83   // Cache our result.
  84   ResourcesPath = LibClangPath;
  85   return LibClangPath.str();
  86 }
  87
  88 static llvm::sys::Path GetTemporaryPath() {
  89   // FIXME: This is lame; sys::Path should provide this function (in particular,
  90   // it should know how to find the temporary files dir).
  91   std::string Error;
  92   const char *TmpDir = ::getenv("TMPDIR");
  93   if (!TmpDir)
  94     TmpDir = ::getenv("TEMP");
  95   if (!TmpDir)
  96     TmpDir = ::getenv("TMP");
  97   if (!TmpDir)
  98     TmpDir = "/tmp";
  99   llvm::sys::Path P(TmpDir);
 100   P.appendComponent("remap");
 101   if (P.makeUnique(false, &Error))
 102     return llvm::sys::Path("");
 103
 104   // FIXME: Grumble, makeUnique sometimes leaves the file around!?  PR3837.
 105   P.eraseFromDisk(false, 0);
 106
 107   return P;
 108 }
 109
 110 bool clang::RemapFiles(unsigned num_unsaved_files,
 111                        struct CXUnsavedFile *unsaved_files,
 112                        std::vector<std::string> &RemapArgs,
 113                        std::vector<llvm::sys::Path> &TemporaryFiles) {
 114   for (unsigned i = 0; i != num_unsaved_files; ++i) {
 115     // Write the contents of this unsaved file into the temporary file.
 116     llvm::sys::Path SavedFile(GetTemporaryPath());
 117     if (SavedFile.empty())
 118       return true;
 119
 120     std::string ErrorInfo;
 121     llvm::raw_fd_ostream OS(SavedFile.c_str(), ErrorInfo,
 122                             llvm::raw_fd_ostream::F_Binary);
 123     if (!ErrorInfo.empty())
 124       return true;
 125
 126     OS.write(unsaved_files[i].Contents, unsaved_files[i].Length);
 127     OS.close();
 128     if (OS.has_error()) {
 129       SavedFile.eraseFromDisk();
 130       OS.clear_error();
 131       return true;
 132     }
 133
 134     // Remap the file.
 135     std::string RemapArg = unsaved_files[i].Filename;
 136     RemapArg += ';';
 137     RemapArg += SavedFile.str();
 138     RemapArgs.push_back("-Xclang");
 139     RemapArgs.push_back("-remap-file");
 140     RemapArgs.push_back("-Xclang");
 141     RemapArgs.push_back(RemapArg);
 142     TemporaryFiles.push_back(SavedFile);
 143   }
 144
 145   return false;
 146 }
 147