compression: Add bounds check for first output buffer write

Signed-off-by: Joseph Sutton <josephsutton@catalyst.net.nz>
Reviewed-by: Douglas Bagnall <douglas.bagnall@catalyst.net.nz>

diff --git a/lib/compression/lzxpress.c b/lib/compression/lzxpress.c
index 65feed0aa730657e1db4a118f6417c929eaee736..647545bbac3c18aaab883964beed11dce3049875 100644 (file)
--- a/lib/compression/lzxpress.c
+++ b/lib/compression/lzxpress.c
@@ -73,9 +73,11 @@ ssize_t lzxpress_compress(const uint8_t *uncompressed,
        }
 
        uncompressed_pos = 0;
+       compressed_pos = 0;
        indic = 0;
+       CHECK_OUTPUT_BYTES(sizeof(uint32_t));
        *(uint32_t *)compressed = 0;
-       compressed_pos = sizeof(uint32_t);
+       compressed_pos += sizeof(uint32_t);
        indic_pos = 0;
 
        indic_bit = 0;