From 928e1fdeccff04dfdb588773262c03bb64031431 Mon Sep 17 00:00:00 2001 From: Argyrios Kyrtzidis Date: Fri, 11 Jan 2013 22:11:14 +0000 Subject: [libclang] In ASTUnit::getMainFileName() Invocation may be null because the ASTUnit came from loading a PCH/module. llvm-svn: 172259 --- clang/lib/Frontend/ASTUnit.cpp | 16 +++++++++++++++- 1 file changed, 15 insertions(+), 1 deletion(-) (limited to 'clang/lib/Frontend/ASTUnit.cpp') diff --git a/clang/lib/Frontend/ASTUnit.cpp b/clang/lib/Frontend/ASTUnit.cpp index 242586a9be8..35a9aa4ff00 100644 --- a/clang/lib/Frontend/ASTUnit.cpp +++ b/clang/lib/Frontend/ASTUnit.cpp @@ -1689,7 +1689,21 @@ void ASTUnit::transferASTDataFromCompilerInstance(CompilerInstance &CI) { } StringRef ASTUnit::getMainFileName() const { - return Invocation->getFrontendOpts().Inputs[0].getFile(); + if (Invocation && !Invocation->getFrontendOpts().Inputs.empty()) { + const FrontendInputFile &Input = Invocation->getFrontendOpts().Inputs[0]; + if (Input.isFile()) + return Input.getFile(); + else + return Input.getBuffer()->getBufferIdentifier(); + } + + if (SourceMgr) { + if (const FileEntry * + FE = SourceMgr->getFileEntryForID(SourceMgr->getMainFileID())) + return FE->getName(); + } + + return StringRef(); } ASTUnit *ASTUnit::create(CompilerInvocation *CI, -- cgit v1.2.3