This commit is contained in:
blueloveTH 2024-12-14 00:23:07 +08:00
parent 24d6b96f91
commit 98c38f2b1f

View File

@ -16,7 +16,7 @@ static bool lz4_compress(int argc, py_Ref argv) {
memcpy(p, &src_size, sizeof(int)); memcpy(p, &src_size, sizeof(int));
char* dst = p + sizeof(int); char* dst = p + sizeof(int);
int dst_size = LZ4_compress_default(src, dst, src_size, dst_capacity); int dst_size = LZ4_compress_default(src, dst, src_size, dst_capacity);
if(dst_size <= 0) return IOError("LZ4 compression failed"); if(dst_size <= 0) return ValueError("LZ4 compression failed");
py_bytes_resize(py_retval(), sizeof(int) + dst_size); py_bytes_resize(py_retval(), sizeof(int) + dst_size);
return true; return true;
} }
@ -32,7 +32,7 @@ static bool lz4_decompress(int argc, py_Ref argv) {
if(uncompressed_size < 0) return ValueError("invalid LZ4 data"); if(uncompressed_size < 0) return ValueError("invalid LZ4 data");
char* dst = (char*)py_newbytes(py_retval(), uncompressed_size); char* dst = (char*)py_newbytes(py_retval(), uncompressed_size);
int dst_size = LZ4_decompress_safe(src, dst, total_size - sizeof(int), uncompressed_size); int dst_size = LZ4_decompress_safe(src, dst, total_size - sizeof(int), uncompressed_size);
if(dst_size < 0) return IOError("LZ4 decompression failed"); if(dst_size < 0) return ValueError("LZ4 decompression failed");
assert(dst_size == uncompressed_size); assert(dst_size == uncompressed_size);
return true; return true;
} }