X-Git-Url: http://plrg.eecs.uci.edu/git/?a=blobdiff_plain;f=fs%2Fsquashfs%2Fzlib_wrapper.c;fp=fs%2Fsquashfs%2Fzlib_wrapper.c;h=ab5801f66e2670931bc3e6e8fff8704df107beca;hb=e7ee11f0ecd587caed0063c5f68ca20fef699f32;hp=1f4833b87ea30677f070ac45d2cc57a5765d57fc;hpb=170cf02165272dfe026eba183563bad973ca4f05;p=firefly-linux-kernel-4.4.55.git

diff --git a/fs/squashfs/zlib_wrapper.c b/fs/squashfs/zlib_wrapper.c
index 1f4833b87ea3..ab5801f66e26 100644
--- a/fs/squashfs/zlib_wrapper.c
+++ b/fs/squashfs/zlib_wrapper.c
@@ -127,6 +127,11 @@ static int zlib_uncompress(struct squashfs_sb_info *msblk, void **buffer,
 		goto release_mutex;
 	}
 
+	if (k < b) {
+		ERROR("zlib_uncompress error, data remaining\n");
+		goto release_mutex;
+	}
+
 	length = stream->total_out;
 	mutex_unlock(&msblk->read_data_mutex);
 	return length;