commit 27705f487d897935b2735e5b50278a5503912616
author Dominic Coyne <dominic.coyne@jhc.financial> Tue Sep 11 13:03:17 2018 +0100
committer James E. King III <jking@apache.org> Sun Sep 16 06:51:38 2018 -0400
tree b01681d7566b604ef8eb7854511ab3b648fbb1e7
parent 180888820aa7d48485364718eeabbcfe8cba4edf

THRIFT-4620: Ensure enough space for for zlib flush marker
Client: lib/cpp

lib/cpp/src/thrift/transport/TZlibTransport.cpp

diff --git a/lib/cpp/src/thrift/transport/TZlibTransport.cpp b/lib/cpp/src/thrift/transport/TZlibTransport.cpp
index fb5cc5d..e426dc3 100644
--- a/lib/cpp/src/thrift/transport/TZlibTransport.cpp
+++ b/lib/cpp/src/thrift/transport/TZlibTransport.cpp
@@ -255,6 +255,15 @@
     throw TTransportException(TTransportException::BAD_ARGS, "flush() called after finish()");
   }
 
+  flushToZlib(uwbuf_, uwpos_, Z_BLOCK);
+  uwpos_ = 0;
+
+  if(wstream_->avail_out < 6){
+    transport_->write(cwbuf_, cwbuf_size_ - wstream_->avail_out);
+    wstream_->next_out = cwbuf_;
+    wstream_->avail_out = cwbuf_size_;
+  }
+
   flushToTransport(Z_FULL_FLUSH);
 }
 
@@ -285,7 +294,7 @@
   wstream_->avail_in = len;
 
   while (true) {
-    if (flush == Z_NO_FLUSH && wstream_->avail_in == 0) {
+    if ((flush == Z_NO_FLUSH || flush == Z_BLOCK) && wstream_->avail_in == 0) {
       break;
     }