From 1cf4460412fdf79422e67546d578d0d289599bc9 Mon Sep 17 00:00:00 2001
From: Con Kolivas <kernel@kolivas.org>
Date: Wed, 16 May 2018 15:13:55 +1000
Subject: [PATCH] Missed one decomp clean up.

---
 stream.c | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/stream.c b/stream.c
index c5628ce..cc2b4a4 100644
--- a/stream.c
+++ b/stream.c
@@ -588,7 +588,6 @@ static int lzo_decompress_buf(rzip_control *control __UNUSED__, struct uncomp_th
 	lzerr = lzo1x_decompress((uchar*)c_buf, ucthread->c_len, (uchar*)ucthread->s_buf, &dlen, NULL);
 	if (unlikely(lzerr != LZO_E_OK)) {
 		print_err("Failed to decompress buffer - lzerr=%d\n", lzerr);
-		dealloc(ucthread->s_buf);
 		ret = -1;
 		goto out;
 	}
@@ -599,8 +598,10 @@ static int lzo_decompress_buf(rzip_control *control __UNUSED__, struct uncomp_th
 	} else
 		dealloc(c_buf);
 out:
-	if (ret == -1)
+	if (ret == -1) {
+		dealloc(ucthread->s_buf);
 		ucthread->s_buf = c_buf;
+	}
 	return ret;
 }