1 files changed, 12 insertions, 1 deletions
diff --git a/Swiften/Compress/ZLibCompressor.cpp b/Swiften/Compress/ZLibCompressor.cpp
index 5d98e38..6af8d83 100644
--- a/Swiften/Compress/ZLibCompressor.cpp
+++ b/Swiften/Compress/ZLibCompressor.cpp
@@ -7,15 +7,26 @@
 #include <Swiften/Compress/ZLibCompressor.h>
 
 #include <cassert>
+#include <zlib.h>
+
+#include <Swiften/Compress/ZLibCodecompressor_Private.h>
 
 #pragma GCC diagnostic ignored "-Wold-style-cast"
 
 namespace Swift {
 
 ZLibCompressor::ZLibCompressor() {
-	int result = deflateInit(&stream_, COMPRESSION_LEVEL);
+	int result = deflateInit(&p->stream, COMPRESSION_LEVEL);
 	assert(result == Z_OK);
 	(void) result;
 }
 
+ZLibCompressor::~ZLibCompressor() {
+	deflateEnd(&p->stream);
+}
+
+int ZLibCompressor::processZStream() {
+	return deflate(&p->stream, Z_SYNC_FLUSH);
+}
+
 }