1 files changed, 143 insertions, 0 deletions
diff --git a/neozip/test/test_deflate_params.cc b/neozip/test/test_deflate_params.cc
new file mode 100644
index 0000000000..9fadea85f9
--- /dev/null
+++ b/neozip/test/test_deflate_params.cc
@@ -0,0 +1,143 @@
+/* test_deflate_params.cc - Test deflate() with dynamic change of compression level */
+
+#include "zbuild.h"
+#ifdef ZLIB_COMPAT
+#  include "zlib.h"
+#else
+#  include "zlib-ng.h"
+#endif
+
+#include <stdio.h>
+#include <stdint.h>
+#include <stdlib.h>
+#include <string.h>
+#include <inttypes.h>
+#include <time.h>
+
+#include "deflate.h"
+
+#include <gtest/gtest.h>
+
+#include "test_shared.h"
+
+#define COMPR_BUFFER_SIZE (48 * 1024)
+#define UNCOMPR_BUFFER_SIZE (64 * 1024)
+#define UNCOMPR_RAND_SIZE (8 * 1024)
+
+TEST(deflate, params) {
+    PREFIX3(stream) c_stream, d_stream;
+    uint8_t *compr, *uncompr;
+    uint32_t compr_len, uncompr_len;
+    uint32_t diff;
+    int32_t i;
+    time_t now;
+    int err;
+#ifndef ZLIB_COMPAT
+    int level = -1;
+    int strategy = -1;
+    zng_deflate_param_value params[2];
+
+    params[0].param = Z_DEFLATE_LEVEL;
+    params[0].buf = &level;
+    params[0].size = sizeof(level);
+
+    params[1].param = Z_DEFLATE_STRATEGY;
+    params[1].buf = &strategy;
+    params[1].size = sizeof(strategy);
+#endif
+
+    memset(&c_stream, 0, sizeof(c_stream));
+    memset(&d_stream, 0, sizeof(d_stream));
+
+    compr = (uint8_t *)calloc(1, COMPR_BUFFER_SIZE);
+    ASSERT_TRUE(compr != NULL);
+    uncompr = (uint8_t *)calloc(1, UNCOMPR_BUFFER_SIZE);
+    ASSERT_TRUE(uncompr != NULL);
+
+    compr_len = COMPR_BUFFER_SIZE;
+    uncompr_len = UNCOMPR_BUFFER_SIZE;
+
+    srand((unsigned)time(&now));
+    for (i = 0; i < UNCOMPR_RAND_SIZE; i++)
+        uncompr[i] = (uint8_t)(rand() % 256);
+
+    err = PREFIX(deflateInit)(&c_stream, Z_BEST_SPEED);
+    EXPECT_EQ(err, Z_OK);
+
+    c_stream.next_out = compr;
+    c_stream.avail_out = compr_len;
+    c_stream.next_in = uncompr;
+    c_stream.avail_in = uncompr_len;
+
+    err = PREFIX(deflate)(&c_stream, Z_NO_FLUSH);
+    EXPECT_EQ(err, Z_OK);
+    EXPECT_EQ(c_stream.avail_in, 0);
+
+    /* Feed in already compressed data and switch to no compression: */
+#ifndef ZLIB_COMPAT
+    zng_deflateGetParams(&c_stream, params, sizeof(params) / sizeof(params[0]));
+    EXPECT_EQ(level, Z_BEST_SPEED);
+    EXPECT_EQ(strategy, Z_DEFAULT_STRATEGY);
+
+    level = Z_NO_COMPRESSION;
+    strategy = Z_DEFAULT_STRATEGY;
+    zng_deflateSetParams(&c_stream, params, sizeof(params) / sizeof(params[0]));
+#else
+    PREFIX(deflateParams)(&c_stream, Z_NO_COMPRESSION, Z_DEFAULT_STRATEGY);
+#endif
+
+    c_stream.next_in = compr;
+    diff = (unsigned int)(c_stream.next_out - compr);
+    c_stream.avail_in = diff;
+    err = PREFIX(deflate)(&c_stream, Z_NO_FLUSH);
+    EXPECT_EQ(err, Z_OK);
+
+    /* Switch back to compressing mode: */
+#ifndef ZLIB_COMPAT
+    level = -1;
+    strategy = -1;
+    zng_deflateGetParams(&c_stream, params, sizeof(params) / sizeof(params[0]));
+    EXPECT_EQ(level, Z_NO_COMPRESSION);
+    EXPECT_EQ(strategy, Z_DEFAULT_STRATEGY);
+
+    level = Z_BEST_COMPRESSION;
+    strategy = Z_FILTERED;
+    zng_deflateSetParams(&c_stream, params, sizeof(params) / sizeof(params[0]));
+#else
+    PREFIX(deflateParams)(&c_stream, Z_BEST_COMPRESSION, Z_FILTERED);
+#endif
+
+    c_stream.next_in = uncompr;
+    c_stream.avail_in = (unsigned int)uncompr_len;
+    err = PREFIX(deflate)(&c_stream, Z_NO_FLUSH);
+    EXPECT_EQ(err, Z_OK);
+
+    err = PREFIX(deflate)(&c_stream, Z_FINISH);
+    EXPECT_EQ(err, Z_STREAM_END);
+
+    err = PREFIX(deflateEnd)(&c_stream);
+    EXPECT_EQ(err, Z_OK);
+
+    d_stream.next_in  = compr;
+    d_stream.avail_in = (unsigned int)compr_len;
+
+    err = PREFIX(inflateInit)(&d_stream);
+    EXPECT_EQ(err, Z_OK);
+
+    do {
+        d_stream.next_out = uncompr;            /* discard the output */
+        d_stream.avail_out = uncompr_len;
+        err = PREFIX(inflate)(&d_stream, Z_NO_FLUSH);
+        if (err == Z_STREAM_END)
+            break;
+        EXPECT_EQ(err, Z_OK);
+    } while (err == Z_OK);
+
+    err = PREFIX(inflateEnd)(&d_stream);
+    EXPECT_EQ(err, Z_OK);
+
+    EXPECT_EQ(d_stream.total_out, (2 * uncompr_len) + diff);
+
+    free(compr);
+    free(uncompr);
+}