fix missing flush in DataSink_Stream::end_msg

2 files changed, 12 insertions, 0 deletions

diff --git a/src/lib/filters/data_snk.cpp b/src/lib/filters/data_snk.cpp
index 7a213cfeb..38734f0e3 100644
--- a/src/lib/filters/data_snk.cpp
+++ b/src/lib/filters/data_snk.cpp
@@ -2,6 +2,7 @@
 * DataSink
 * (C) 1999-2007 Jack Lloyd
 *     2005 Matthew Gregan
+*     2017 Philippe Lieser
 *
 * Botan is released under the Simplified BSD License (see license.txt)
 */
@@ -27,6 +28,14 @@ void DataSink_Stream::write(const uint8_t out[], size_t length)
    }
 
 /*
+* Flush the stream
+*/
+void DataSink_Stream::end_msg()
+   {
+   m_sink.flush();
+   }
+
+/*
 * DataSink_Stream Constructor
 */
 DataSink_Stream::DataSink_Stream(std::ostream& out,
diff --git a/src/lib/filters/data_snk.h b/src/lib/filters/data_snk.h
index 7accef358..b3e5e06cf 100644
--- a/src/lib/filters/data_snk.h
+++ b/src/lib/filters/data_snk.h
@@ -1,6 +1,7 @@
 /*
 * DataSink
 * (C) 1999-2007 Jack Lloyd
+*     2017 Philippe Lieser
 *
 * Botan is released under the Simplified BSD License (see license.txt)
 */
@@ -58,6 +59,8 @@ class BOTAN_DLL DataSink_Stream : public DataSink
 
       void write(const uint8_t[], size_t) override;
 
+      void end_msg() override;
+
       ~DataSink_Stream();
 
    private: