pass diagnostics into the rewrite test client.
git-svn-id: https://llvm.org/svn/llvm-project/cfe/trunk@44468 91177308-0d34-0410-b5e6-96231b3b80d8
diff --git a/Driver/RewriteTest.cpp b/Driver/RewriteTest.cpp
index 043b260..5d14f03 100644
--- a/Driver/RewriteTest.cpp
+++ b/Driver/RewriteTest.cpp
@@ -27,6 +27,7 @@
namespace {
class RewriteTest : public ASTConsumer {
Rewriter Rewrite;
+ Diagnostic &Diags;
ASTContext *Context;
SourceManager *SM;
unsigned MainFileID;
@@ -92,6 +93,7 @@
// Top Level Driver code.
virtual void HandleTopLevelDecl(Decl *D);
void HandleDeclInMainFile(Decl *D);
+ RewriteTest(Diagnostic &D) : Diags(D) {}
~RewriteTest();
// Syntactic Rewriting.
@@ -160,7 +162,9 @@
};
}
-ASTConsumer *clang::CreateCodeRewriterTest() { return new RewriteTest(); }
+ASTConsumer *clang::CreateCodeRewriterTest(Diagnostic &Diags) {
+ return new RewriteTest(Diags);
+}
//===----------------------------------------------------------------------===//
// Top Level Driver Code
@@ -860,7 +864,11 @@
Expr *Replacement = new StringLiteral(StrEncoding.c_str(),
StrEncoding.length(), false, StrType,
SourceLocation(), SourceLocation());
- Rewrite.ReplaceStmt(Exp, Replacement);
+ if (Rewrite.ReplaceStmt(Exp, Replacement)) {
+ // replacement failed.
+ return Exp;
+ }
+
delete Exp;
return Replacement;
}