More cython fixes for Lz4 error handling

Author: bmoscon

src/_compress.pyx

@@ -106,6 +106,7 @@ def decompress(pString):
     cdef char *cString    # *char pStr
     cdef char *result     # destination buffer
     cdef bytes pyResult   # python wrapped result
+    cdef int ret
 
     # convert to char*
     cString = pString
@@ -114,7 +115,8 @@ def decompress(pString):
     # malloc 
     result = <char*>malloc(original_size)
     # decompress
-    if LZ4_decompress_safe(cString + hdr_size, result, compressed_size - hdr_size, original_size) < 0:
+    ret = LZ4_decompress_safe(cString + hdr_size, result, compressed_size - hdr_size, original_size)
+    if ret <= 0 or ret != original_size:
         free(result)
         raise Exception("Error decompressing")
     # cast back into python string
@@ -211,7 +213,8 @@ def decompressarr(pStrList):
     cdef uint32_t compressed_size
     cdef char *result
     cdef Py_ssize_t i
-    cdef int ret = 0
+    cdef int ret
+    cdef int error = 0
 
     # output parameters
     cdef char **cResult = <char **>malloc(n * sizeof(char *))
@@ -232,8 +235,9 @@ def decompressarr(pStrList):
             # malloc 
             result = <char*>malloc(original_size)
             # decompress
-            if LZ4_decompress_safe(cString + hdr_size, result, compressed_size - hdr_size, original_size) < 0:
-                ret = -1
+            ret = LZ4_decompress_safe(cString + hdr_size, result, compressed_size - hdr_size, original_size)
+            if ret <= 0 or ret != original_size:
+                error = -1
             # assign to result
             cResult[i] = result
             lengths[i] = original_size
@@ -248,7 +252,7 @@ def decompressarr(pStrList):
     free(cResult)
     free(cStrList)
 
-    if ret == -1:
+    if error == -1:
         raise Exception("Error decompressing array")
 
     return result_list