Be sure to flush raw_string_ostream objects in the Objective-C

rewriter, from Eric Niebler!

llvm-svn: 129849

1 files changed, 2 insertions, 0 deletions

diff --git a/clang/lib/Rewrite/Rewriter.cpp b/clang/lib/Rewrite/Rewriter.cpp
index 2ff61363ffb..10ed38b86b7 100644
--- a/clang/lib/Rewrite/Rewriter.cpp
+++ b/clang/lib/Rewrite/Rewriter.cpp
@@ -300,6 +300,7 @@ bool Rewriter::ReplaceStmt(Stmt *From, Stmt *To) {
   std::string SStr;
   llvm::raw_string_ostream S(SStr);
   To->printPretty(S, 0, PrintingPolicy(*LangOpts));
+  S.flush();
   const std::string &Str = S.str();
 
   ReplaceText(From->getLocStart(), Size, Str);
@@ -310,6 +311,7 @@ std::string Rewriter::ConvertToString(Stmt *From) {
   std::string SStr;
   llvm::raw_string_ostream S(SStr);
   From->printPretty(S, 0, PrintingPolicy(*LangOpts));
+  S.flush();
   return SStr;
 }