From 092fe0b727399cd971709c16c030bc093ca54d87 Mon Sep 17 00:00:00 2001
From: Jon Simons <jon@jonsimons.org>
Date: Fri, 26 Sep 2014 19:43:47 -0700
Subject: [PATCH] wrapper: fix z_stream leak

Ensure to free the z_stream structures as allocated from
the gzip.c initcompress, initdecompress functions.

Signed-off-by: Jon Simons <jon@jonsimons.org>
Reviewed-by: Andreas Schneider <asn@cryptomilk.org>
---
 src/wrapper.c | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/wrapper.c b/src/wrapper.c
index f2f0b6ad..bcd941b3 100644
--- a/src/wrapper.c
+++ b/src/wrapper.c
@@ -170,10 +170,13 @@ void crypto_free(struct ssh_crypto_struct *crypto){
       (deflateEnd(crypto->compress_out_ctx) != 0)) {
     inflateEnd(crypto->compress_out_ctx);
   }
+  SAFE_FREE(crypto->compress_out_ctx);
+
   if (crypto->compress_in_ctx &&
       (deflateEnd(crypto->compress_in_ctx) != 0)) {
     inflateEnd(crypto->compress_in_ctx);
   }
+  SAFE_FREE(crypto->compress_in_ctx);
 #endif /* WITH_ZLIB */
   if(crypto->encryptIV)
     SAFE_FREE(crypto->encryptIV);