blob: 3bf1fab1c27dab40dd37ef2133025104ea42d1bb [file] [log] [blame]
Argyrios Kyrtzidis4b562cf2009-06-20 08:27:14 +00001//===--- ASTUnit.cpp - ASTUnit utility ------------------------------------===//
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// ASTUnit Implementation.
11//
12//===----------------------------------------------------------------------===//
13
Argyrios Kyrtzidis0853a022009-06-20 08:08:23 +000014#include "clang/Frontend/ASTUnit.h"
15#include "clang/Frontend/PCHReader.h"
Argyrios Kyrtzidis0853a022009-06-20 08:08:23 +000016#include "clang/AST/ASTContext.h"
Daniel Dunbar521bf9c2009-12-01 09:51:01 +000017#include "clang/AST/ASTConsumer.h"
Argyrios Kyrtzidis0853a022009-06-20 08:08:23 +000018#include "clang/AST/DeclVisitor.h"
19#include "clang/AST/StmtVisitor.h"
Daniel Dunbar7b556682009-12-02 03:23:45 +000020#include "clang/Driver/Compilation.h"
21#include "clang/Driver/Driver.h"
22#include "clang/Driver/Job.h"
23#include "clang/Driver/Tool.h"
Daniel Dunbar521bf9c2009-12-01 09:51:01 +000024#include "clang/Frontend/CompilerInstance.h"
25#include "clang/Frontend/FrontendActions.h"
Daniel Dunbar7b556682009-12-02 03:23:45 +000026#include "clang/Frontend/FrontendDiagnostic.h"
Daniel Dunbar521bf9c2009-12-01 09:51:01 +000027#include "clang/Frontend/FrontendOptions.h"
Argyrios Kyrtzidis0853a022009-06-20 08:08:23 +000028#include "clang/Lex/HeaderSearch.h"
29#include "clang/Lex/Preprocessor.h"
Daniel Dunbard58c03f2009-11-15 06:48:46 +000030#include "clang/Basic/TargetOptions.h"
Argyrios Kyrtzidis0853a022009-06-20 08:08:23 +000031#include "clang/Basic/TargetInfo.h"
32#include "clang/Basic/Diagnostic.h"
Douglas Gregor4db64a42010-01-23 00:14:00 +000033#include "llvm/Support/MemoryBuffer.h"
Daniel Dunbar7b556682009-12-02 03:23:45 +000034#include "llvm/System/Host.h"
Benjamin Kramer4a630d32009-10-18 11:34:14 +000035#include "llvm/System/Path.h"
Argyrios Kyrtzidis0853a022009-06-20 08:08:23 +000036using namespace clang;
37
Douglas Gregorf715ca12010-03-16 00:06:06 +000038ASTUnit::ASTUnit(Diagnostic &Diag, bool _MainFileIsAST)
39 : SourceMgr(Diag), MainFileIsAST(_MainFileIsAST),
40 ConcurrencyCheckValue(CheckUnlocked) {
Steve Naroff36c44642009-10-19 14:34:22 +000041}
Daniel Dunbar521bf9c2009-12-01 09:51:01 +000042ASTUnit::~ASTUnit() {
Douglas Gregorbdf60622010-03-05 21:16:25 +000043 ConcurrencyCheckValue = CheckLocked;
Douglas Gregor313e26c2010-02-18 23:35:40 +000044 for (unsigned I = 0, N = TemporaryFiles.size(); I != N; ++I)
45 TemporaryFiles[I].eraseFromDisk();
Steve Naroffe19944c2009-10-15 22:23:48 +000046}
Argyrios Kyrtzidis0853a022009-06-20 08:08:23 +000047
48namespace {
49
50/// \brief Gathers information from PCHReader that will be used to initialize
51/// a Preprocessor.
Benjamin Kramerbd218282009-11-28 10:07:24 +000052class PCHInfoCollector : public PCHReaderListener {
Argyrios Kyrtzidis0853a022009-06-20 08:08:23 +000053 LangOptions &LangOpt;
54 HeaderSearch &HSI;
55 std::string &TargetTriple;
56 std::string &Predefines;
57 unsigned &Counter;
Mike Stump1eb44332009-09-09 15:08:12 +000058
Argyrios Kyrtzidis0853a022009-06-20 08:08:23 +000059 unsigned NumHeaderInfos;
Mike Stump1eb44332009-09-09 15:08:12 +000060
Argyrios Kyrtzidis0853a022009-06-20 08:08:23 +000061public:
62 PCHInfoCollector(LangOptions &LangOpt, HeaderSearch &HSI,
63 std::string &TargetTriple, std::string &Predefines,
64 unsigned &Counter)
65 : LangOpt(LangOpt), HSI(HSI), TargetTriple(TargetTriple),
66 Predefines(Predefines), Counter(Counter), NumHeaderInfos(0) {}
Mike Stump1eb44332009-09-09 15:08:12 +000067
Argyrios Kyrtzidis0853a022009-06-20 08:08:23 +000068 virtual bool ReadLanguageOptions(const LangOptions &LangOpts) {
69 LangOpt = LangOpts;
70 return false;
71 }
Mike Stump1eb44332009-09-09 15:08:12 +000072
Daniel Dunbardc3c0d22009-11-11 00:52:11 +000073 virtual bool ReadTargetTriple(llvm::StringRef Triple) {
Argyrios Kyrtzidis0853a022009-06-20 08:08:23 +000074 TargetTriple = Triple;
75 return false;
76 }
Mike Stump1eb44332009-09-09 15:08:12 +000077
Daniel Dunbardc3c0d22009-11-11 00:52:11 +000078 virtual bool ReadPredefinesBuffer(llvm::StringRef PCHPredef,
Argyrios Kyrtzidis0853a022009-06-20 08:08:23 +000079 FileID PCHBufferID,
Daniel Dunbar7b5a1212009-11-11 05:29:04 +000080 llvm::StringRef OriginalFileName,
Argyrios Kyrtzidis0853a022009-06-20 08:08:23 +000081 std::string &SuggestedPredefines) {
82 Predefines = PCHPredef;
83 return false;
84 }
Mike Stump1eb44332009-09-09 15:08:12 +000085
Argyrios Kyrtzidis0853a022009-06-20 08:08:23 +000086 virtual void ReadHeaderFileInfo(const HeaderFileInfo &HFI) {
87 HSI.setHeaderFileInfoForUID(HFI, NumHeaderInfos++);
88 }
Mike Stump1eb44332009-09-09 15:08:12 +000089
Argyrios Kyrtzidis0853a022009-06-20 08:08:23 +000090 virtual void ReadCounter(unsigned Value) {
91 Counter = Value;
92 }
93};
94
Douglas Gregora88084b2010-02-18 18:08:43 +000095class StoredDiagnosticClient : public DiagnosticClient {
96 llvm::SmallVectorImpl<StoredDiagnostic> &StoredDiags;
97
98public:
99 explicit StoredDiagnosticClient(
100 llvm::SmallVectorImpl<StoredDiagnostic> &StoredDiags)
101 : StoredDiags(StoredDiags) { }
102
103 virtual void HandleDiagnostic(Diagnostic::Level Level,
104 const DiagnosticInfo &Info);
105};
106
107/// \brief RAII object that optionally captures diagnostics, if
108/// there is no diagnostic client to capture them already.
109class CaptureDroppedDiagnostics {
110 Diagnostic &Diags;
111 StoredDiagnosticClient Client;
112 DiagnosticClient *PreviousClient;
113
114public:
115 CaptureDroppedDiagnostics(bool RequestCapture, Diagnostic &Diags,
116 llvm::SmallVectorImpl<StoredDiagnostic> &StoredDiags)
117 : Diags(Diags), Client(StoredDiags), PreviousClient(Diags.getClient())
118 {
119 if (RequestCapture || Diags.getClient() == 0)
120 Diags.setClient(&Client);
121 }
122
123 ~CaptureDroppedDiagnostics() {
124 Diags.setClient(PreviousClient);
125 }
126};
127
Argyrios Kyrtzidis0853a022009-06-20 08:08:23 +0000128} // anonymous namespace
129
Douglas Gregora88084b2010-02-18 18:08:43 +0000130void StoredDiagnosticClient::HandleDiagnostic(Diagnostic::Level Level,
131 const DiagnosticInfo &Info) {
132 StoredDiags.push_back(StoredDiagnostic(Level, Info));
133}
134
Steve Naroff77accc12009-09-03 18:19:54 +0000135const std::string &ASTUnit::getOriginalSourceFileName() {
Daniel Dunbar68d40e22009-12-02 08:44:16 +0000136 return OriginalSourceFile;
Steve Naroff77accc12009-09-03 18:19:54 +0000137}
Argyrios Kyrtzidis0853a022009-06-20 08:08:23 +0000138
Steve Naroffe19944c2009-10-15 22:23:48 +0000139const std::string &ASTUnit::getPCHFileName() {
Daniel Dunbarc7822db2009-12-02 21:47:43 +0000140 assert(isMainFileAST() && "Not an ASTUnit from a PCH file!");
Benjamin Kramer7297c182010-01-30 16:23:25 +0000141 return static_cast<PCHReader *>(Ctx->getExternalSource())->getFileName();
Steve Naroffe19944c2009-10-15 22:23:48 +0000142}
143
Argyrios Kyrtzidis0853a022009-06-20 08:08:23 +0000144ASTUnit *ASTUnit::LoadFromPCHFile(const std::string &Filename,
Daniel Dunbar5262fda2009-12-03 01:45:44 +0000145 Diagnostic &Diags,
Ted Kremenek5cf48762009-10-17 00:34:24 +0000146 bool OnlyLocalDecls,
Douglas Gregor4db64a42010-01-23 00:14:00 +0000147 RemappedFile *RemappedFiles,
Douglas Gregora88084b2010-02-18 18:08:43 +0000148 unsigned NumRemappedFiles,
149 bool CaptureDiagnostics) {
Douglas Gregorf715ca12010-03-16 00:06:06 +0000150 llvm::OwningPtr<ASTUnit> AST(new ASTUnit(Diags, true));
Douglas Gregor7d1d49d2009-10-16 20:01:17 +0000151 AST->OnlyLocalDecls = OnlyLocalDecls;
Steve Naroff36c44642009-10-19 14:34:22 +0000152 AST->HeaderInfo.reset(new HeaderSearch(AST->getFileManager()));
Argyrios Kyrtzidis0853a022009-06-20 08:08:23 +0000153
Douglas Gregora88084b2010-02-18 18:08:43 +0000154 // If requested, capture diagnostics in the ASTUnit.
155 CaptureDroppedDiagnostics Capture(CaptureDiagnostics, Diags,
156 AST->Diagnostics);
157
Douglas Gregor4db64a42010-01-23 00:14:00 +0000158 for (unsigned I = 0; I != NumRemappedFiles; ++I) {
159 // Create the file entry for the file that we're mapping from.
160 const FileEntry *FromFile
161 = AST->getFileManager().getVirtualFile(RemappedFiles[I].first,
162 RemappedFiles[I].second->getBufferSize(),
163 0);
164 if (!FromFile) {
165 Diags.Report(diag::err_fe_remap_missing_from_file)
166 << RemappedFiles[I].first;
Douglas Gregorc8dfe5e2010-02-27 01:32:48 +0000167 delete RemappedFiles[I].second;
Douglas Gregor4db64a42010-01-23 00:14:00 +0000168 continue;
169 }
170
171 // Override the contents of the "from" file with the contents of
172 // the "to" file.
173 AST->getSourceManager().overrideFileContents(FromFile,
174 RemappedFiles[I].second);
175 }
176
Argyrios Kyrtzidis0853a022009-06-20 08:08:23 +0000177 // Gather Info for preprocessor construction later on.
Mike Stump1eb44332009-09-09 15:08:12 +0000178
Argyrios Kyrtzidis0853a022009-06-20 08:08:23 +0000179 LangOptions LangInfo;
180 HeaderSearch &HeaderInfo = *AST->HeaderInfo.get();
181 std::string TargetTriple;
182 std::string Predefines;
183 unsigned Counter;
184
Daniel Dunbarbce6f622009-09-03 05:59:50 +0000185 llvm::OwningPtr<PCHReader> Reader;
Argyrios Kyrtzidis0853a022009-06-20 08:08:23 +0000186 llvm::OwningPtr<ExternalASTSource> Source;
187
Ted Kremenekfc062212009-10-19 21:44:57 +0000188 Reader.reset(new PCHReader(AST->getSourceManager(), AST->getFileManager(),
Daniel Dunbar5262fda2009-12-03 01:45:44 +0000189 Diags));
Daniel Dunbarcc318932009-09-03 05:59:35 +0000190 Reader->setListener(new PCHInfoCollector(LangInfo, HeaderInfo, TargetTriple,
191 Predefines, Counter));
192
193 switch (Reader->ReadPCH(Filename)) {
Argyrios Kyrtzidis0853a022009-06-20 08:08:23 +0000194 case PCHReader::Success:
195 break;
Mike Stump1eb44332009-09-09 15:08:12 +0000196
Argyrios Kyrtzidis0853a022009-06-20 08:08:23 +0000197 case PCHReader::Failure:
Argyrios Kyrtzidis106c9982009-06-25 18:22:30 +0000198 case PCHReader::IgnorePCH:
Daniel Dunbar5262fda2009-12-03 01:45:44 +0000199 Diags.Report(diag::err_fe_unable_to_load_pch);
Argyrios Kyrtzidis0853a022009-06-20 08:08:23 +0000200 return NULL;
Argyrios Kyrtzidis0853a022009-06-20 08:08:23 +0000201 }
Mike Stump1eb44332009-09-09 15:08:12 +0000202
Daniel Dunbar68d40e22009-12-02 08:44:16 +0000203 AST->OriginalSourceFile = Reader->getOriginalSourceFile();
204
Argyrios Kyrtzidis0853a022009-06-20 08:08:23 +0000205 // PCH loaded successfully. Now create the preprocessor.
Mike Stump1eb44332009-09-09 15:08:12 +0000206
Argyrios Kyrtzidis0853a022009-06-20 08:08:23 +0000207 // Get information about the target being compiled for.
Daniel Dunbard58c03f2009-11-15 06:48:46 +0000208 //
209 // FIXME: This is broken, we should store the TargetOptions in the PCH.
210 TargetOptions TargetOpts;
211 TargetOpts.ABI = "";
212 TargetOpts.CPU = "";
213 TargetOpts.Features.clear();
214 TargetOpts.Triple = TargetTriple;
Daniel Dunbar5262fda2009-12-03 01:45:44 +0000215 AST->Target.reset(TargetInfo::CreateTargetInfo(Diags, TargetOpts));
216 AST->PP.reset(new Preprocessor(Diags, LangInfo, *AST->Target.get(),
Daniel Dunbar31b87d82009-09-21 03:03:39 +0000217 AST->getSourceManager(), HeaderInfo));
Argyrios Kyrtzidis0853a022009-06-20 08:08:23 +0000218 Preprocessor &PP = *AST->PP.get();
219
Daniel Dunbard5b61262009-09-21 03:03:47 +0000220 PP.setPredefines(Reader->getSuggestedPredefines());
Argyrios Kyrtzidis0853a022009-06-20 08:08:23 +0000221 PP.setCounterValue(Counter);
Daniel Dunbarcc318932009-09-03 05:59:35 +0000222 Reader->setPreprocessor(PP);
Mike Stump1eb44332009-09-09 15:08:12 +0000223
Argyrios Kyrtzidis0853a022009-06-20 08:08:23 +0000224 // Create and initialize the ASTContext.
225
226 AST->Ctx.reset(new ASTContext(LangInfo,
Daniel Dunbar31b87d82009-09-21 03:03:39 +0000227 AST->getSourceManager(),
Argyrios Kyrtzidis0853a022009-06-20 08:08:23 +0000228 *AST->Target.get(),
229 PP.getIdentifierTable(),
230 PP.getSelectorTable(),
231 PP.getBuiltinInfo(),
Daniel Dunbarb26d4832010-02-16 01:55:04 +0000232 /* FreeMemory = */ false,
Argyrios Kyrtzidis0853a022009-06-20 08:08:23 +0000233 /* size_reserve = */0));
234 ASTContext &Context = *AST->Ctx.get();
Mike Stump1eb44332009-09-09 15:08:12 +0000235
Daniel Dunbarcc318932009-09-03 05:59:35 +0000236 Reader->InitializeContext(Context);
Mike Stump1eb44332009-09-09 15:08:12 +0000237
Argyrios Kyrtzidis0853a022009-06-20 08:08:23 +0000238 // Attach the PCH reader to the AST context as an external AST
239 // source, so that declarations will be deserialized from the
240 // PCH file as needed.
Daniel Dunbarcc318932009-09-03 05:59:35 +0000241 Source.reset(Reader.take());
Argyrios Kyrtzidis0853a022009-06-20 08:08:23 +0000242 Context.setExternalSource(Source);
243
Mike Stump1eb44332009-09-09 15:08:12 +0000244 return AST.take();
Argyrios Kyrtzidis0853a022009-06-20 08:08:23 +0000245}
Daniel Dunbar521bf9c2009-12-01 09:51:01 +0000246
247namespace {
248
Daniel Dunbarf772d1e2009-12-04 08:17:33 +0000249class TopLevelDeclTrackerConsumer : public ASTConsumer {
250 ASTUnit &Unit;
251
252public:
253 TopLevelDeclTrackerConsumer(ASTUnit &_Unit) : Unit(_Unit) {}
254
255 void HandleTopLevelDecl(DeclGroupRef D) {
256 for (DeclGroupRef::iterator it = D.begin(), ie = D.end(); it != ie; ++it)
257 Unit.getTopLevelDecls().push_back(*it);
258 }
259};
260
261class TopLevelDeclTrackerAction : public ASTFrontendAction {
262public:
263 ASTUnit &Unit;
264
Daniel Dunbar521bf9c2009-12-01 09:51:01 +0000265 virtual ASTConsumer *CreateASTConsumer(CompilerInstance &CI,
266 llvm::StringRef InFile) {
Daniel Dunbarf772d1e2009-12-04 08:17:33 +0000267 return new TopLevelDeclTrackerConsumer(Unit);
Daniel Dunbar521bf9c2009-12-01 09:51:01 +0000268 }
269
270public:
Daniel Dunbarf772d1e2009-12-04 08:17:33 +0000271 TopLevelDeclTrackerAction(ASTUnit &_Unit) : Unit(_Unit) {}
272
Daniel Dunbar521bf9c2009-12-01 09:51:01 +0000273 virtual bool hasCodeCompletionSupport() const { return false; }
274};
275
276}
277
Daniel Dunbarf7acc372010-02-16 01:54:54 +0000278ASTUnit *ASTUnit::LoadFromCompilerInvocation(CompilerInvocation *CI,
Daniel Dunbar521bf9c2009-12-01 09:51:01 +0000279 Diagnostic &Diags,
Douglas Gregora88084b2010-02-18 18:08:43 +0000280 bool OnlyLocalDecls,
281 bool CaptureDiagnostics) {
Daniel Dunbar521bf9c2009-12-01 09:51:01 +0000282 // Create the compiler instance to use for building the AST.
Daniel Dunbarcb6dda12009-12-02 08:43:56 +0000283 CompilerInstance Clang;
Daniel Dunbar521bf9c2009-12-01 09:51:01 +0000284 llvm::OwningPtr<ASTUnit> AST;
Daniel Dunbarf772d1e2009-12-04 08:17:33 +0000285 llvm::OwningPtr<TopLevelDeclTrackerAction> Act;
Daniel Dunbar521bf9c2009-12-01 09:51:01 +0000286
Daniel Dunbarf7acc372010-02-16 01:54:54 +0000287 Clang.setInvocation(CI);
Daniel Dunbar521bf9c2009-12-01 09:51:01 +0000288
289 Clang.setDiagnostics(&Diags);
290 Clang.setDiagnosticClient(Diags.getClient());
291
292 // Create the target instance.
293 Clang.setTarget(TargetInfo::CreateTargetInfo(Clang.getDiagnostics(),
294 Clang.getTargetOpts()));
Douglas Gregora88084b2010-02-18 18:08:43 +0000295 if (!Clang.hasTarget()) {
296 Clang.takeSourceManager();
297 Clang.takeFileManager();
298 Clang.takeDiagnosticClient();
299 Clang.takeDiagnostics();
300 return 0;
301 }
Daniel Dunbar521bf9c2009-12-01 09:51:01 +0000302
303 // Inform the target of the language options.
304 //
305 // FIXME: We shouldn't need to do this, the target should be immutable once
306 // created. This complexity should be lifted elsewhere.
307 Clang.getTarget().setForcedLangOptions(Clang.getLangOpts());
308
309 assert(Clang.getFrontendOpts().Inputs.size() == 1 &&
310 "Invocation must have exactly one source file!");
311 assert(Clang.getFrontendOpts().Inputs[0].first != FrontendOptions::IK_AST &&
312 "FIXME: AST inputs not yet supported here!");
313
314 // Create the AST unit.
Douglas Gregorf715ca12010-03-16 00:06:06 +0000315 AST.reset(new ASTUnit(Diags, false));
Daniel Dunbar68ea2ac2009-12-02 21:47:32 +0000316 AST->OnlyLocalDecls = OnlyLocalDecls;
Daniel Dunbar68d40e22009-12-02 08:44:16 +0000317 AST->OriginalSourceFile = Clang.getFrontendOpts().Inputs[0].second;
318
Douglas Gregora88084b2010-02-18 18:08:43 +0000319 // Capture any diagnostics that would otherwise be dropped.
320 CaptureDroppedDiagnostics Capture(CaptureDiagnostics,
321 Clang.getDiagnostics(),
322 AST->Diagnostics);
323
Daniel Dunbar521bf9c2009-12-01 09:51:01 +0000324 // Create a file manager object to provide access to and cache the filesystem.
325 Clang.setFileManager(&AST->getFileManager());
326
327 // Create the source manager.
328 Clang.setSourceManager(&AST->getSourceManager());
329
330 // Create the preprocessor.
331 Clang.createPreprocessor();
332
Daniel Dunbarf772d1e2009-12-04 08:17:33 +0000333 Act.reset(new TopLevelDeclTrackerAction(*AST));
334 if (!Act->BeginSourceFile(Clang, Clang.getFrontendOpts().Inputs[0].second,
Daniel Dunbar521bf9c2009-12-01 09:51:01 +0000335 /*IsAST=*/false))
336 goto error;
337
Daniel Dunbarf772d1e2009-12-04 08:17:33 +0000338 Act->Execute();
Daniel Dunbar521bf9c2009-12-01 09:51:01 +0000339
Daniel Dunbar64a32ba2009-12-01 21:57:33 +0000340 // Steal the created target, context, and preprocessor, and take back the
341 // source and file managers.
Daniel Dunbar521bf9c2009-12-01 09:51:01 +0000342 AST->Ctx.reset(Clang.takeASTContext());
343 AST->PP.reset(Clang.takePreprocessor());
344 Clang.takeSourceManager();
345 Clang.takeFileManager();
Daniel Dunbar64a32ba2009-12-01 21:57:33 +0000346 AST->Target.reset(Clang.takeTarget());
Daniel Dunbar521bf9c2009-12-01 09:51:01 +0000347
Daniel Dunbarf772d1e2009-12-04 08:17:33 +0000348 Act->EndSourceFile();
Daniel Dunbar521bf9c2009-12-01 09:51:01 +0000349
350 Clang.takeDiagnosticClient();
351 Clang.takeDiagnostics();
Daniel Dunbar807b0612010-01-30 21:47:16 +0000352 Clang.takeInvocation();
Daniel Dunbar521bf9c2009-12-01 09:51:01 +0000353
Daniel Dunbarf7acc372010-02-16 01:54:54 +0000354 AST->Invocation.reset(Clang.takeInvocation());
Daniel Dunbar521bf9c2009-12-01 09:51:01 +0000355 return AST.take();
356
357error:
358 Clang.takeSourceManager();
359 Clang.takeFileManager();
360 Clang.takeDiagnosticClient();
361 Clang.takeDiagnostics();
362 return 0;
363}
Daniel Dunbar7b556682009-12-02 03:23:45 +0000364
365ASTUnit *ASTUnit::LoadFromCommandLine(const char **ArgBegin,
366 const char **ArgEnd,
367 Diagnostic &Diags,
Daniel Dunbar869824e2009-12-13 03:46:13 +0000368 llvm::StringRef ResourceFilesPath,
Daniel Dunbar7b556682009-12-02 03:23:45 +0000369 bool OnlyLocalDecls,
Douglas Gregor4db64a42010-01-23 00:14:00 +0000370 RemappedFile *RemappedFiles,
Douglas Gregora88084b2010-02-18 18:08:43 +0000371 unsigned NumRemappedFiles,
372 bool CaptureDiagnostics) {
Daniel Dunbar7b556682009-12-02 03:23:45 +0000373 llvm::SmallVector<const char *, 16> Args;
374 Args.push_back("<clang>"); // FIXME: Remove dummy argument.
375 Args.insert(Args.end(), ArgBegin, ArgEnd);
376
377 // FIXME: Find a cleaner way to force the driver into restricted modes. We
378 // also want to force it to use clang.
379 Args.push_back("-fsyntax-only");
380
Daniel Dunbar869824e2009-12-13 03:46:13 +0000381 // FIXME: We shouldn't have to pass in the path info.
382 driver::Driver TheDriver("clang", "/", llvm::sys::getHostTriple(),
Daniel Dunbar7b556682009-12-02 03:23:45 +0000383 "a.out", false, Diags);
Daniel Dunbar3bd54cc2010-01-25 00:44:02 +0000384
385 // Don't check that inputs exist, they have been remapped.
386 TheDriver.setCheckInputsExist(false);
387
Daniel Dunbar7b556682009-12-02 03:23:45 +0000388 llvm::OwningPtr<driver::Compilation> C(
389 TheDriver.BuildCompilation(Args.size(), Args.data()));
390
391 // We expect to get back exactly one command job, if we didn't something
392 // failed.
393 const driver::JobList &Jobs = C->getJobs();
394 if (Jobs.size() != 1 || !isa<driver::Command>(Jobs.begin())) {
395 llvm::SmallString<256> Msg;
396 llvm::raw_svector_ostream OS(Msg);
397 C->PrintJob(OS, C->getJobs(), "; ", true);
398 Diags.Report(diag::err_fe_expected_compiler_job) << OS.str();
399 return 0;
400 }
401
402 const driver::Command *Cmd = cast<driver::Command>(*Jobs.begin());
403 if (llvm::StringRef(Cmd->getCreator().getName()) != "clang") {
404 Diags.Report(diag::err_fe_expected_clang_command);
405 return 0;
406 }
407
408 const driver::ArgStringList &CCArgs = Cmd->getArguments();
Daniel Dunbar807b0612010-01-30 21:47:16 +0000409 llvm::OwningPtr<CompilerInvocation> CI(new CompilerInvocation);
410 CompilerInvocation::CreateFromArgs(*CI, (const char**) CCArgs.data(),
Daniel Dunbar7b556682009-12-02 03:23:45 +0000411 (const char**) CCArgs.data()+CCArgs.size(),
Daniel Dunbar1e69fe32009-12-13 03:45:58 +0000412 Diags);
413
Douglas Gregor4db64a42010-01-23 00:14:00 +0000414 // Override any files that need remapping
415 for (unsigned I = 0; I != NumRemappedFiles; ++I)
Daniel Dunbar807b0612010-01-30 21:47:16 +0000416 CI->getPreprocessorOpts().addRemappedFile(RemappedFiles[I].first,
Daniel Dunbarb26d4832010-02-16 01:55:04 +0000417 RemappedFiles[I].second);
Douglas Gregor4db64a42010-01-23 00:14:00 +0000418
Daniel Dunbar8b9adfe2009-12-15 00:06:45 +0000419 // Override the resources path.
Daniel Dunbar807b0612010-01-30 21:47:16 +0000420 CI->getHeaderSearchOpts().ResourceDir = ResourceFilesPath;
Daniel Dunbar7b556682009-12-02 03:23:45 +0000421
Daniel Dunbarb26d4832010-02-16 01:55:04 +0000422 CI->getFrontendOpts().DisableFree = true;
Douglas Gregora88084b2010-02-18 18:08:43 +0000423 return LoadFromCompilerInvocation(CI.take(), Diags, OnlyLocalDecls,
424 CaptureDiagnostics);
Daniel Dunbar7b556682009-12-02 03:23:45 +0000425}