From 159346a82bfa5063ead81af903a2bc5e5a9e5b36 Mon Sep 17 00:00:00 2001 From: Ted Kremenek Date: Thu, 7 Aug 2008 19:47:41 +0000 Subject: ParseAST now conditionally deletes the passed ASTConsumer. ModuleBuilder now performs llvmgen in HandleTranslationUnit. This patch follows from the discussion on the following thread on cfe-commits: http://lists.cs.uiuc.edu/pipermail/cfe-commits/Week-of-Mon-20080804/006849.html git-svn-id: https://llvm.org/svn/llvm-project/cfe/trunk@54486 91177308-0d34-0410-b5e6-96231b3b80d8 --- lib/CodeGen/ModuleBuilder.cpp | 19 +++++++++++-------- 1 file changed, 11 insertions(+), 8 deletions(-) (limited to 'lib/CodeGen/ModuleBuilder.cpp') diff --git a/lib/CodeGen/ModuleBuilder.cpp b/lib/CodeGen/ModuleBuilder.cpp index 7329ed1d90..6ce4e6894d 100644 --- a/lib/CodeGen/ModuleBuilder.cpp +++ b/lib/CodeGen/ModuleBuilder.cpp @@ -49,13 +49,7 @@ namespace { virtual ~CodeGeneratorImpl() {} - virtual llvm::Module* ReleaseModule() { - if (Diags.hasErrorOccurred()) - return 0; - - if (Builder) - Builder->Release(); - + virtual llvm::Module* ReleaseModule() { return M.take(); } @@ -134,7 +128,16 @@ namespace { virtual void HandleTagDeclDefinition(TagDecl *D) { Builder->UpdateCompletedType(D); } - + + virtual void HandleTranslationUnit(TranslationUnit& TU) { + if (Diags.hasErrorOccurred()) { + M.reset(); + return; + } + + if (Builder) + Builder->Release(); + }; }; } -- cgit v1.2.3-18-g5258