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