Make create_decompressor() checks of max_block_size consistent

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;

diff --git a/src/lzms-decompress.c b/src/lzms-decompress.c
index f8bfcdb17344b3055d343e7be7615897484d7bfc..7254091449c052442bae510444f93d7f92b9a9d0 100644 (file)
--- a/src/lzms-decompress.c
+++ b/src/lzms-decompress.c
@@ -1012,16 +1012,6 @@ lzms_decompress(const void *compressed_data, size_t compressed_size,
        if (uncompressed_size == 0)
                return 0;
 
-       /* The x86 post-processor requires that the uncompressed length fit into
-        * a signed 32-bit integer.  Also, the position slot table cannot be
-        * searched for a position of INT32_MAX or greater.  */
-       if (uncompressed_size >= INT32_MAX) {
-               LZMS_DEBUG("Uncompressed length too large "
-                          "(got %zu, expected < INT32_MAX)",
-                          uncompressed_size);
-               return -1;
-       }
-
        /* Decode the literals and matches.  */
        if (lzms_decode_items(compressed_data, compressed_size,
                              uncompressed_data, uncompressed_size, ctx))
@@ -1048,6 +1038,12 @@ lzms_create_decompressor(size_t max_block_size, void **ctx_ret)
 {
        struct lzms_decompressor *ctx;
 
+       /* The x86 post-processor requires that the uncompressed length fit into
+        * a signed 32-bit integer.  Also, the position slot table cannot be
+        * searched for a position of INT32_MAX or greater.  */
+       if (max_block_size >= INT32_MAX)
+               return WIMLIB_ERR_INVALID_PARAM;
+
        ctx = ALIGNED_MALLOC(sizeof(struct lzms_decompressor),
                             DECODE_TABLE_ALIGNMENT);
        if (ctx == NULL)

diff --git a/src/xpress-compress.c b/src/xpress-compress.c
index e1884978377f0ac6d1555282600e775aa8141ab2..5ff23e028ddd261a4e56c6e1fd4d464f055a3ae9 100644 (file)
--- a/src/xpress-compress.c
+++ b/src/xpress-compress.c
@@ -1068,12 +1068,6 @@ xpress_build_mf_params(const struct xpress_compressor_params *xpress_params,
        mf_params->nice_match_len = xpress_params->nice_match_length;
 }
 
-static inline bool
-xpress_window_size_valid(size_t window_size)
-{
-       return (window_size > 0 && window_size <= XPRESS_MAX_OFFSET + 1);
-}
-
 static void
 xpress_free_compressor(void *_c);
 
@@ -1083,7 +1077,7 @@ xpress_get_needed_memory(size_t max_window_size, unsigned int compression_level)
        u64 size = 0;
        struct xpress_compressor_params params;
 
-       if (!xpress_window_size_valid(max_window_size))
+       if (max_window_size > XPRESS_MAX_OFFSET + 1)
                return 0;
 
        xpress_build_params(compression_level, max_window_size, &params);
@@ -1117,7 +1111,7 @@ xpress_create_compressor(size_t max_window_size, unsigned int compression_level,
        struct xpress_compressor_params params;
        struct lz_mf_params mf_params;
 
-       if (!xpress_window_size_valid(max_window_size))
+       if (max_window_size > XPRESS_MAX_OFFSET + 1)
                return WIMLIB_ERR_INVALID_PARAM;
 
        xpress_build_params(compression_level, max_window_size, &params);

diff --git a/src/xpress-decompress.c b/src/xpress-decompress.c
index f060d0b0df2afeccfec25fbae91d41321f6beda6..8d7741090dbae4519f9c97ccfb43c77836110b10 100644 (file)
--- a/src/xpress-decompress.c
+++ b/src/xpress-decompress.c
@@ -71,6 +71,7 @@
 
 #include "wimlib/decompressor_ops.h"
 #include "wimlib/decompress_common.h"
+#include "wimlib/error.h"
 #include "wimlib/xpress.h"
 
 /*
@@ -198,6 +199,16 @@ xpress_decompress(const void *compressed_data, size_t compressed_size,
                                uncompressed_size, decode_table);
 }
 
+static int
+xpress_create_decompressor(size_t max_block_size, void **dec_ret)
+{
+       if (max_block_size > XPRESS_MAX_OFFSET + 1)
+               return WIMLIB_ERR_INVALID_PARAM;
+
+       return 0;
+}
+
 const struct decompressor_ops xpress_decompressor_ops = {
-       .decompress = xpress_decompress,
+       .create_decompressor = xpress_create_decompressor,
+       .decompress          = xpress_decompress,
 };