Make create_decompressor() checks of max_block_size consistent

[wimlib] / src / lzms-compress.c

diff --git a/src/lzms-compress.c b/src/lzms-compress.c
index b4391c2be9c95b3dda653588f801c4ef6baac60f..21c82ea87bf9f93e2005406018960f73bb6488f6 100644 (file)
--- a/src/lzms-compress.c
+++ b/src/lzms-compress.c
@@ -1335,10 +1335,12 @@ static u64
 lzms_get_needed_memory(size_t max_block_size, unsigned int compression_level)
 {
        struct lzms_compressor_params params;
+       u64 size = 0;
 
-       lzms_build_params(compression_level, &params);
+       if (max_block_size >= INT32_MAX)
+               return 0;
 
-       u64 size = 0;
+       lzms_build_params(compression_level, &params);
 
        size += sizeof(struct lzms_compressor);
        size += max_block_size;