summaryrefslogtreecommitdiffstats
path: root/src/isa-l/igzip/igzip_rand_test.c
diff options
context:
space:
mode:
authorDaniel Baumann <daniel.baumann@progress-linux.org>2024-04-27 18:24:20 +0000
committerDaniel Baumann <daniel.baumann@progress-linux.org>2024-04-27 18:24:20 +0000
commit483eb2f56657e8e7f419ab1a4fab8dce9ade8609 (patch)
treee5d88d25d870d5dedacb6bbdbe2a966086a0a5cf /src/isa-l/igzip/igzip_rand_test.c
parentInitial commit. (diff)
downloadceph-upstream.tar.xz
ceph-upstream.zip
Adding upstream version 14.2.21.upstream/14.2.21upstream
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'src/isa-l/igzip/igzip_rand_test.c')
-rw-r--r--src/isa-l/igzip/igzip_rand_test.c2161
1 files changed, 2161 insertions, 0 deletions
diff --git a/src/isa-l/igzip/igzip_rand_test.c b/src/isa-l/igzip/igzip_rand_test.c
new file mode 100644
index 00000000..5ceb5333
--- /dev/null
+++ b/src/isa-l/igzip/igzip_rand_test.c
@@ -0,0 +1,2161 @@
+/**********************************************************************
+ Copyright(c) 2011-2016 Intel Corporation All rights reserved.
+
+ Redistribution and use in source and binary forms, with or without
+ modification, are permitted provided that the following conditions
+ are met:
+ * Redistributions of source code must retain the above copyright
+ notice, this list of conditions and the following disclaimer.
+ * Redistributions in binary form must reproduce the above copyright
+ notice, this list of conditions and the following disclaimer in
+ the documentation and/or other materials provided with the
+ distribution.
+ * Neither the name of Intel Corporation nor the names of its
+ contributors may be used to endorse or promote products derived
+ from this software without specific prior written permission.
+
+ THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
+ "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
+ LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
+ A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
+ OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
+ SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
+ LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
+ DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
+ THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+ (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+ OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+**********************************************************************/
+
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <assert.h>
+#include "igzip_lib.h"
+#include "crc_inflate.h"
+#include "inflate_std_vects.h"
+#include <math.h>
+
+#ifndef RANDOMS
+# define RANDOMS 0x40
+#endif
+#ifndef TEST_SEED
+# define TEST_SEED 0x1234
+#endif
+
+#define IBUF_SIZE (1024*1024)
+
+#define PAGE_SIZE 4*1024
+
+#define str1 "Short test string"
+#define str2 "one two three four five six seven eight nine ten eleven twelve " \
+ "thirteen fourteen fifteen sixteen"
+
+#define TYPE0_HDR_SIZE 5 /* Size of a type 0 blocks header in bytes */
+#define TYPE0_MAX_SIZE 65535 /* Max length of a type 0 block in bytes (excludes the header) */
+
+#define MAX_LOOPS 20
+/* Defines for the possible error conditions */
+enum IGZIP_TEST_ERROR_CODES {
+ IGZIP_COMP_OK,
+
+ MALLOC_FAILED,
+ FILE_READ_FAILED,
+
+ COMPRESS_INCORRECT_STATE,
+ COMPRESS_INPUT_STREAM_INTEGRITY_ERROR,
+ COMPRESS_OUTPUT_STREAM_INTEGRITY_ERROR,
+ COMPRESS_END_OF_STREAM_NOT_SET,
+ COMPRESS_ALL_INPUT_FAIL,
+ COMPRESS_OUT_BUFFER_OVERFLOW,
+ COMPRESS_LOOP_COUNT_OVERFLOW,
+ COMPRESS_GENERAL_ERROR,
+
+ INFLATE_END_OF_INPUT,
+ INFLATE_INVALID_BLOCK_HEADER,
+ INFLATE_INVALID_SYMBOL,
+ INFLATE_OUT_BUFFER_OVERFLOW,
+ INFLATE_LEFTOVER_INPUT,
+ INFLATE_INCORRECT_OUTPUT_SIZE,
+ INFLATE_INVALID_LOOK_BACK_DISTANCE,
+ INVALID_GZIP_HEADER,
+ INCORRECT_GZIP_TRAILER,
+ INFLATE_GENERAL_ERROR,
+
+ INVALID_FLUSH_ERROR,
+
+ OVERFLOW_TEST_ERROR,
+ RESULT_ERROR
+};
+
+static const int hdr_bytes = 300;
+
+static const uint8_t gzip_hdr[10] = {
+ 0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00,
+ 0x00, 0xff
+};
+
+static const uint32_t gzip_hdr_bytes = 10;
+/* static const uint32_t gzip_trl_bytes = 8; */
+
+static const int gzip_extra_bytes = 18; /* gzip_hdr_bytes + gzip_trl_bytes */
+
+int inflate_type = 0;
+
+struct isal_hufftables *hufftables = NULL;
+
+#define HISTORY_SIZE 32*1024
+#define MIN_LENGTH 3
+#define MIN_DIST 1
+
+/* Create random compressible data. This is achieved by randomly choosing a
+ * random character, or to repeat previous data in the stream for a random
+ * length and look back distance. The probability of a random character or a
+ * repeat being chosen is semi-randomly chosen by setting max_repeat_data to be
+ * differing values */
+void create_rand_repeat_data(uint8_t * data, int size)
+{
+ uint32_t next_data;
+ uint8_t *data_start = data;
+ uint32_t length, distance;
+ uint32_t max_repeat_data = 256;
+ uint32_t power = rand() % 32;
+ /* An array of the powers of 2 (except the final element which is 0) */
+ const uint32_t power_of_2_array[] = {
+ 0x00000001, 0x00000002, 0x00000004, 0x00000008,
+ 0x00000010, 0x00000020, 0x00000040, 0x00000080,
+ 0x00000100, 0x00000200, 0x00000400, 0x00000800,
+ 0x00001000, 0x00002000, 0x00004000, 0x00008000,
+ 0x00010000, 0x00020000, 0x00040000, 0x00080000,
+ 0x00100000, 0x00200000, 0x00400000, 0x00800000,
+ 0x01000000, 0x02000000, 0x04000000, 0x08000000,
+ 0x10000000, 0x20000000, 0x40000000, 0x00000000
+ };
+
+ max_repeat_data += power_of_2_array[power];
+
+ if (size-- > 0)
+ *data++ = rand();
+
+ while (size > 0) {
+ next_data = rand() % max_repeat_data;
+ if (next_data < 256) {
+ *data++ = next_data;
+ size--;
+ } else if (size < 3) {
+ *data++ = rand() % 256;
+ size--;
+ } else {
+ length = (rand() % 256) + MIN_LENGTH;
+ if (length > size)
+ length = (rand() % (size - 2)) + MIN_LENGTH;
+
+ distance = (rand() % HISTORY_SIZE) + MIN_DIST;
+ if (distance > data - data_start)
+ distance = (rand() % (data - data_start)) + MIN_DIST;
+
+ size -= length;
+ if (distance <= length) {
+ while (length-- > 0) {
+ *data = *(data - distance);
+ data++;
+ }
+ } else
+ memcpy(data, data - distance, length);
+ }
+ }
+}
+
+void print_error(int error_code)
+{
+ switch (error_code) {
+ case IGZIP_COMP_OK:
+ break;
+ case MALLOC_FAILED:
+ printf("error: failed to allocate memory\n");
+ break;
+ case FILE_READ_FAILED:
+ printf("error: failed to read in file\n");
+ break;
+ case COMPRESS_INCORRECT_STATE:
+ printf("error: incorrect stream internal state\n");
+ break;
+ case COMPRESS_INPUT_STREAM_INTEGRITY_ERROR:
+ printf("error: inconsistent stream input buffer\n");
+ break;
+ case COMPRESS_OUTPUT_STREAM_INTEGRITY_ERROR:
+ printf("error: inconsistent stream output buffer\n");
+ break;
+ case COMPRESS_END_OF_STREAM_NOT_SET:
+ printf("error: end of stream not set\n");
+ break;
+ case COMPRESS_ALL_INPUT_FAIL:
+ printf("error: not all input data compressed\n");
+ break;
+ case COMPRESS_OUT_BUFFER_OVERFLOW:
+ printf("error: output buffer overflow while compressing data\n");
+ break;
+ case COMPRESS_GENERAL_ERROR:
+ printf("error: compression failed\n");
+ break;
+ case INFLATE_END_OF_INPUT:
+ printf("error: did not decompress all input\n");
+ break;
+ case INFLATE_INVALID_BLOCK_HEADER:
+ printf("error: invalid header\n");
+ break;
+ case INFLATE_INVALID_SYMBOL:
+ printf("error: invalid symbol found when decompressing input\n");
+ break;
+ case INFLATE_OUT_BUFFER_OVERFLOW:
+ printf("error: output buffer overflow while decompressing data\n");
+ break;
+ case INFLATE_GENERAL_ERROR:
+ printf("error: decompression failed\n");
+ break;
+ case INFLATE_LEFTOVER_INPUT:
+ printf("error: the trailer of igzip output contains junk\n");
+ break;
+ case INFLATE_INCORRECT_OUTPUT_SIZE:
+ printf("error: incorrect amount of data was decompressed\n");
+ break;
+ case INFLATE_INVALID_LOOK_BACK_DISTANCE:
+ printf("error: invalid look back distance found while decompressing\n");
+ break;
+ case INVALID_GZIP_HEADER:
+ printf("error: incorrect gzip header found when inflating data\n");
+ break;
+ case INCORRECT_GZIP_TRAILER:
+ printf("error: incorrect gzip trailer found when inflating data\n");
+ break;
+ case INVALID_FLUSH_ERROR:
+ printf("error: invalid flush did not cause compression to error\n");
+ break;
+ case RESULT_ERROR:
+ printf("error: decompressed data is not the same as the compressed data\n");
+ break;
+ case OVERFLOW_TEST_ERROR:
+ printf("error: overflow undetected\n");
+ break;
+ default:
+ printf("error: unknown error code\n");
+ }
+}
+
+void print_uint8_t(uint8_t * array, uint64_t length)
+{
+ const int line_size = 16;
+ int i;
+
+ printf("Length = %lu", length);
+ for (i = 0; i < length; i++) {
+ if ((i % line_size) == 0)
+ printf("\n0x%08x\t", i);
+ else
+ printf(" ");
+ printf("0x%02x,", array[i]);
+ }
+ printf("\n");
+}
+
+uint32_t check_gzip_header(uint8_t * z_buf)
+{
+ /* These values are defined in RFC 1952 page 4 */
+ const uint8_t ID1 = 0x1f, ID2 = 0x8b, CM = 0x08, FLG = 0;
+ uint32_t ret = 0;
+ int i;
+ /* Verify that the gzip header is the one used in hufftables_c.c */
+ for (i = 0; i < gzip_hdr_bytes; i++)
+ if (z_buf[i] != gzip_hdr[i])
+ ret = INVALID_GZIP_HEADER;
+
+ /* Verify that the gzip header is a valid gzip header */
+ if (*z_buf++ != ID1)
+ ret = INVALID_GZIP_HEADER;
+
+ if (*z_buf++ != ID2)
+ ret = INVALID_GZIP_HEADER;
+
+ /* Verfiy compression method is Deflate */
+ if (*z_buf++ != CM)
+ ret = INVALID_GZIP_HEADER;
+
+ /* The following comparison is specific to how gzip headers are written in igzip */
+ /* Verify no extra flags are set */
+ if (*z_buf != FLG)
+ ret = INVALID_GZIP_HEADER;
+
+ /* The last 6 bytes in the gzip header do not contain any information
+ * important to decomrpessing the data */
+
+ return ret;
+}
+
+uint32_t check_gzip_trl(uint64_t gzip_trl, uint32_t inflate_crc, uint8_t * uncompress_buf,
+ uint32_t uncompress_len)
+{
+ uint64_t trl, ret = 0;
+ uint32_t crc;
+
+ crc = find_crc(uncompress_buf, uncompress_len);
+ trl = ((uint64_t) uncompress_len << 32) | crc;
+
+ if (crc != inflate_crc || trl != gzip_trl)
+ ret = INCORRECT_GZIP_TRAILER;
+
+ return ret;
+}
+
+int inflate_stateless_pass(uint8_t * compress_buf, uint64_t compress_len,
+ uint8_t * uncompress_buf, uint32_t * uncompress_len,
+ uint32_t gzip_flag)
+{
+ struct inflate_state state;
+ int ret = 0;
+
+ state.next_in = compress_buf;
+ state.avail_in = compress_len;
+ state.next_out = uncompress_buf;
+ state.avail_out = *uncompress_len;
+ state.crc_flag = gzip_flag;
+
+ ret = isal_inflate_stateless(&state);
+
+ *uncompress_len = state.total_out;
+
+ if (gzip_flag) {
+ if (!ret)
+ ret =
+ check_gzip_trl(*(uint64_t *) state.next_in, state.crc,
+ uncompress_buf, *uncompress_len);
+ state.avail_in -= 8;
+ }
+
+ if (ret == 0 && state.avail_in != 0)
+ ret = INFLATE_LEFTOVER_INPUT;
+
+ return ret;
+}
+
+int inflate_multi_pass(uint8_t * compress_buf, uint64_t compress_len,
+ uint8_t * uncompress_buf, uint32_t * uncompress_len, uint32_t gzip_flag)
+{
+ struct inflate_state *state = NULL;
+ int ret = 0;
+ uint8_t *comp_tmp = NULL, *uncomp_tmp = NULL;
+ uint32_t comp_tmp_size = 0, uncomp_tmp_size = 0;
+ uint32_t comp_processed = 0, uncomp_processed = 0;
+ int32_t read_in_old = 0;
+
+ state = malloc(sizeof(struct inflate_state));
+ if (state == NULL) {
+ printf("Failed to allocate memory\n");
+ exit(0);
+ }
+
+ isal_inflate_init(state);
+
+ state->next_in = NULL;
+ state->next_out = NULL;
+ state->avail_in = 0;
+ state->avail_out = 0;
+ state->crc_flag = gzip_flag;
+
+ if (gzip_flag)
+ compress_len -= 8;
+
+ while (1) {
+ if (state->avail_in == 0) {
+ comp_tmp_size = rand() % (compress_len + 1);
+
+ if (comp_tmp_size >= compress_len - comp_processed)
+ comp_tmp_size = compress_len - comp_processed;
+
+ if (comp_tmp_size != 0) {
+ if (comp_tmp != NULL) {
+ free(comp_tmp);
+ comp_tmp = NULL;
+ }
+
+ comp_tmp = malloc(comp_tmp_size);
+
+ if (comp_tmp == NULL) {
+ printf("Failed to allocate memory\n");
+ return MALLOC_FAILED;
+ }
+
+ memcpy(comp_tmp, compress_buf + comp_processed, comp_tmp_size);
+ comp_processed += comp_tmp_size;
+
+ state->next_in = comp_tmp;
+ state->avail_in = comp_tmp_size;
+ }
+ }
+
+ if (state->avail_out == 0) {
+ /* Save uncompressed data into uncompress_buf */
+ if (uncomp_tmp != NULL) {
+ memcpy(uncompress_buf + uncomp_processed, uncomp_tmp,
+ uncomp_tmp_size);
+ uncomp_processed += uncomp_tmp_size;
+ }
+
+ uncomp_tmp_size = rand() % (*uncompress_len + 1);
+
+ /* Limit size of buffer to be smaller than maximum */
+ if (uncomp_tmp_size > *uncompress_len - uncomp_processed)
+ uncomp_tmp_size = *uncompress_len - uncomp_processed;
+
+ if (uncomp_tmp_size != 0) {
+
+ if (uncomp_tmp != NULL) {
+ fflush(0);
+ free(uncomp_tmp);
+ uncomp_tmp = NULL;
+ }
+
+ uncomp_tmp = malloc(uncomp_tmp_size);
+ if (uncomp_tmp == NULL) {
+ printf("Failed to allocate memory\n");
+ return MALLOC_FAILED;
+ }
+
+ state->avail_out = uncomp_tmp_size;
+ state->next_out = uncomp_tmp;
+ }
+ }
+
+ ret = isal_inflate(state);
+
+ if (state->block_state == ISAL_BLOCK_FINISH || ret != 0) {
+ memcpy(uncompress_buf + uncomp_processed, uncomp_tmp, uncomp_tmp_size);
+ *uncompress_len = state->total_out;
+ break;
+ }
+
+ if (*uncompress_len - uncomp_processed == 0 && state->avail_out == 0
+ && state->tmp_out_valid - state->tmp_out_processed > 0) {
+ ret = ISAL_OUT_OVERFLOW;
+ break;
+ }
+
+ if (compress_len - comp_processed == 0 && state->avail_in == 0
+ && (state->block_state != ISAL_BLOCK_INPUT_DONE)
+ && state->tmp_out_valid - state->tmp_out_processed == 0) {
+ if (state->read_in_length == read_in_old) {
+ ret = ISAL_END_INPUT;
+ break;
+ }
+ read_in_old = state->read_in_length;
+ }
+ }
+
+ if (gzip_flag) {
+ if (!ret)
+ ret =
+ check_gzip_trl(*(uint64_t *) & compress_buf[compress_len],
+ state->crc, uncompress_buf, *uncompress_len);
+ }
+ if (ret == 0 && state->avail_in != 0)
+ ret = INFLATE_LEFTOVER_INPUT;
+
+ if (comp_tmp != NULL) {
+ free(comp_tmp);
+ comp_tmp = NULL;
+ }
+
+ if (uncomp_tmp != NULL) {
+ free(uncomp_tmp);
+ uncomp_tmp = NULL;
+ }
+
+ free(state);
+ return ret;
+}
+
+/* Inflate the compressed data and check that the decompressed data agrees with the input data */
+int inflate_check(uint8_t * z_buf, int z_size, uint8_t * in_buf, int in_size,
+ uint32_t gzip_flag)
+{
+ /* Test inflate with reference inflate */
+
+ int ret = 0;
+ uint32_t test_size = in_size;
+ uint8_t *test_buf = NULL;
+ int mem_result = 0;
+ int gzip_hdr_result = 0, gzip_trl_result = 0;
+
+ if (in_size > 0) {
+ assert(in_buf != NULL);
+ test_buf = malloc(test_size);
+ if (test_buf == NULL)
+ return MALLOC_FAILED;
+ }
+
+ if (test_buf != NULL)
+ memset(test_buf, 0xff, test_size);
+
+ if (gzip_flag == IGZIP_GZIP) {
+ gzip_hdr_result = check_gzip_header(z_buf);
+ z_buf += gzip_hdr_bytes;
+ z_size -= gzip_hdr_bytes;
+ }
+
+ if (inflate_type == 0) {
+ ret = inflate_stateless_pass(z_buf, z_size, test_buf, &test_size, gzip_flag);
+ inflate_type = 1;
+ } else {
+ ret = inflate_multi_pass(z_buf, z_size, test_buf, &test_size, gzip_flag);
+ inflate_type = 0;
+ }
+
+ if (test_buf != NULL)
+ mem_result = memcmp(in_buf, test_buf, in_size);
+
+#ifdef VERBOSE
+ int i;
+ if (mem_result)
+ for (i = 0; i < in_size; i++) {
+ if (in_buf[i] != test_buf[i]) {
+ printf
+ ("First incorrect data at 0x%x of 0x%x, 0x%x != 0x%x\n",
+ i, in_size, in_buf[i], test_buf[i]);
+ break;
+ }
+ }
+#endif
+
+ if (test_buf != NULL)
+ free(test_buf);
+ switch (ret) {
+ case 0:
+ break;
+ case ISAL_END_INPUT:
+ return INFLATE_END_OF_INPUT;
+ break;
+ case ISAL_INVALID_BLOCK:
+ return INFLATE_INVALID_BLOCK_HEADER;
+ break;
+ case ISAL_INVALID_SYMBOL:
+ return INFLATE_INVALID_SYMBOL;
+ break;
+ case ISAL_OUT_OVERFLOW:
+ return INFLATE_OUT_BUFFER_OVERFLOW;
+ break;
+ case ISAL_INVALID_LOOKBACK:
+ return INFLATE_INVALID_LOOK_BACK_DISTANCE;
+ break;
+ case INFLATE_LEFTOVER_INPUT:
+ return INFLATE_LEFTOVER_INPUT;
+ break;
+ case INCORRECT_GZIP_TRAILER:
+ gzip_trl_result = INCORRECT_GZIP_TRAILER;
+ break;
+
+ default:
+ return INFLATE_GENERAL_ERROR;
+ break;
+ }
+
+ if (test_size != in_size)
+ return INFLATE_INCORRECT_OUTPUT_SIZE;
+
+ if (mem_result)
+ return RESULT_ERROR;
+
+ if (gzip_flag) {
+ if (gzip_hdr_result)
+ return INVALID_GZIP_HEADER;
+
+ if (gzip_trl_result)
+ return INCORRECT_GZIP_TRAILER;
+ }
+
+ return 0;
+}
+
+/* Check if that the state of the data stream is consistent */
+int stream_valid_check(struct isal_zstream *stream, uint8_t * in_buf, uint32_t in_size,
+ uint8_t * out_buf, uint32_t out_size, uint32_t in_processed,
+ uint32_t out_processed, uint32_t data_size)
+{
+ uint32_t total_in, in_buffer_size, total_out, out_buffer_size;
+
+ total_in =
+ (in_size ==
+ 0) ? in_processed : (in_processed - in_size) + (stream->next_in - in_buf);
+ in_buffer_size = (in_size == 0) ? 0 : stream->next_in - in_buf + stream->avail_in;
+
+ /* Check for a consistent amount of data processed */
+ if (total_in != stream->total_in || in_buffer_size != in_size)
+ return COMPRESS_INPUT_STREAM_INTEGRITY_ERROR;
+
+ total_out =
+ (out_size == 0) ? out_processed : out_processed + (stream->next_out - out_buf);
+ out_buffer_size = (out_size == 0) ? 0 : stream->next_out - out_buf + stream->avail_out;
+
+ /* Check for a consistent amount of data compressed */
+ if (total_out != stream->total_out || out_buffer_size != out_size) {
+ return COMPRESS_OUTPUT_STREAM_INTEGRITY_ERROR;
+ }
+
+ return 0;
+}
+
+/* Performs compression with checks to discover and verify the state of the
+ * stream
+ * stream: compress data structure which has been initialized to use
+ * in_buf and out_buf as the buffers
+ * data_size: size of all input data
+ * compressed_size: size of all available output buffers
+ * in_buf: next buffer of data to be compressed
+ * in_size: size of in_buf
+ * out_buf: next out put buffer where data is stored
+ * out_size: size of out_buf
+ * in_processed: the amount of input data which has been loaded into buffers
+ * to be compressed, this includes the data in in_buf
+ * out_processed: the amount of output data which has been compressed and stored,
+ * this does not include the data in the current out_buf
+*/
+int isal_deflate_with_checks(struct isal_zstream *stream, uint32_t data_size,
+ uint32_t compressed_size, uint8_t * in_buf, uint32_t in_size,
+ uint32_t in_processed, uint8_t * out_buf, uint32_t out_size,
+ uint32_t out_processed)
+{
+ int ret, stream_check;
+ struct isal_zstate *state = &stream->internal_state;
+
+#ifdef VERBOSE
+ printf("Pre compression\n");
+ printf
+ ("data_size = 0x%05x, in_processed = 0x%05x, in_size = 0x%05x, avail_in = 0x%05x, total_in = 0x%05x\n",
+ data_size, in_processed, in_size, stream->avail_in, stream->total_in);
+ printf
+ ("compressed_size = 0x%05x, out_processed = 0x%05x, out_size = 0x%05x, avail_out = 0x%05x, total_out = 0x%05x\n",
+ compressed_size, out_processed, out_size, stream->avail_out, stream->total_out);
+#endif
+
+ ret = isal_deflate(stream);
+
+#ifdef VERBOSE
+ printf("Post compression\n");
+ printf
+ ("data_size = 0x%05x, in_processed = 0x%05x, in_size = 0x%05x, avail_in = 0x%05x, total_in = 0x%05x\n",
+ data_size, in_processed, in_size, stream->avail_in, stream->total_in);
+ printf
+ ("compressed_size = 0x%05x, out_processed = 0x%05x, out_size = 0x%05x, avail_out = 0x%05x, total_out = 0x%05x\n",
+ compressed_size, out_processed, out_size, stream->avail_out, stream->total_out);
+ printf("\n\n");
+#endif
+
+ /* Verify the stream is in a valid state */
+ stream_check = stream_valid_check(stream, in_buf, in_size, out_buf, out_size,
+ in_processed, out_processed, data_size);
+
+ if (stream_check != 0)
+ return stream_check;
+
+ if (ret != IGZIP_COMP_OK)
+ return COMPRESS_GENERAL_ERROR;
+
+ /* Check if the compression is completed */
+ if (state->state != ZSTATE_END)
+ if (compressed_size - out_processed - (out_size - stream->avail_out) <= 0)
+ return COMPRESS_OUT_BUFFER_OVERFLOW;
+
+ return ret;
+
+}
+
+void set_random_hufftable(struct isal_zstream *stream)
+{
+ isal_deflate_set_hufftables(stream, hufftables, rand() % 4);
+}
+
+/* Compress the input data into the output buffer where the input buffer and
+ * output buffer are randomly segmented to test state information for the
+ * compression*/
+int compress_multi_pass(uint8_t * data, uint32_t data_size, uint8_t * compressed_buf,
+ uint32_t * compressed_size, uint32_t flush_type, uint32_t gzip_flag,
+ uint32_t level)
+{
+ int ret = IGZIP_COMP_OK;
+ uint8_t *in_buf = NULL, *out_buf = NULL;
+ uint32_t in_size = 0, out_size = 0;
+ uint32_t in_processed = 0, out_processed = 0;
+ struct isal_zstream stream;
+ struct isal_zstate *state = &stream.internal_state;
+ uint32_t loop_count = 0;
+ uint32_t level_buf_size;
+ uint8_t *level_buf = NULL;
+
+#ifdef VERBOSE
+ printf("Starting Compress Multi Pass\n");
+#endif
+
+ create_rand_repeat_data((uint8_t *) & stream, sizeof(stream));
+
+ isal_deflate_init(&stream);
+
+ if (state->state != ZSTATE_NEW_HDR)
+ return COMPRESS_INCORRECT_STATE;
+
+ stream.flush = flush_type;
+ stream.end_of_stream = 0;
+
+ /* These are set here to allow the loop to run correctly */
+ stream.avail_in = 0;
+ stream.avail_out = 0;
+ stream.gzip_flag = gzip_flag;
+ stream.level = level;
+
+ if (level >= 1) {
+ level_buf_size = rand() % IBUF_SIZE + ISAL_DEF_LVL1_MIN;
+ level_buf = malloc(level_buf_size);
+ create_rand_repeat_data(level_buf, level_buf_size);
+ stream.level_buf = level_buf;
+ stream.level_buf_size = level_buf_size;
+ }
+
+ while (1) {
+ loop_count++;
+
+ /* Setup in buffer for next round of compression */
+ if (stream.avail_in == 0) {
+ if (flush_type == NO_FLUSH || state->state == ZSTATE_NEW_HDR) {
+ /* Randomly choose size of the next out buffer */
+ in_size = rand() % (data_size + 1);
+
+ /* Limit size of buffer to be smaller than maximum */
+ if (in_size >= data_size - in_processed) {
+ in_size = data_size - in_processed;
+ stream.end_of_stream = 1;
+ }
+
+ if (in_size != 0) {
+ if (in_buf != NULL) {
+ free(in_buf);
+ in_buf = NULL;
+ }
+
+ in_buf = malloc(in_size);
+ if (in_buf == NULL) {
+ ret = MALLOC_FAILED;
+ break;
+ }
+ memcpy(in_buf, data + in_processed, in_size);
+ in_processed += in_size;
+
+ stream.avail_in = in_size;
+ stream.next_in = in_buf;
+ }
+ }
+ }
+
+ /* Setup out buffer for next round of compression */
+ if (stream.avail_out == 0) {
+ /* Save compressed data inot compressed_buf */
+ if (out_buf != NULL) {
+ memcpy(compressed_buf + out_processed, out_buf,
+ out_size - stream.avail_out);
+ out_processed += out_size - stream.avail_out;
+ }
+
+ /* Randomly choose size of the next out buffer */
+ out_size = rand() % (*compressed_size + 1);
+
+ /* Limit size of buffer to be smaller than maximum */
+ if (out_size > *compressed_size - out_processed)
+ out_size = *compressed_size - out_processed;
+
+ if (out_size != 0) {
+ if (out_buf != NULL) {
+ free(out_buf);
+ out_buf = NULL;
+ }
+
+ out_buf = malloc(out_size);
+ if (out_buf == NULL) {
+ ret = MALLOC_FAILED;
+ break;
+ }
+
+ stream.avail_out = out_size;
+ stream.next_out = out_buf;
+ }
+ }
+
+ if (state->state == ZSTATE_NEW_HDR)
+ set_random_hufftable(&stream);
+
+ ret =
+ isal_deflate_with_checks(&stream, data_size, *compressed_size, in_buf,
+ in_size, in_processed, out_buf, out_size,
+ out_processed);
+
+ if (ret) {
+ if (ret == COMPRESS_OUT_BUFFER_OVERFLOW
+ || ret == COMPRESS_INCORRECT_STATE)
+ memcpy(compressed_buf + out_processed, out_buf, out_size);
+ break;
+ }
+
+ /* Check if the compression is completed */
+ if (state->state == ZSTATE_END) {
+ memcpy(compressed_buf + out_processed, out_buf, out_size);
+ *compressed_size = stream.total_out;
+ break;
+ }
+
+ }
+
+ if (level_buf != NULL)
+ free(level_buf);
+ if (in_buf != NULL)
+ free(in_buf);
+ if (out_buf != NULL)
+ free(out_buf);
+
+ if (ret == COMPRESS_OUT_BUFFER_OVERFLOW && flush_type == SYNC_FLUSH
+ && loop_count >= MAX_LOOPS)
+ ret = COMPRESS_LOOP_COUNT_OVERFLOW;
+
+ return ret;
+
+}
+
+/* Compress the input data into the outbuffer in one call to isal_deflate */
+int compress_single_pass(uint8_t * data, uint32_t data_size, uint8_t * compressed_buf,
+ uint32_t * compressed_size, uint32_t flush_type, uint32_t gzip_flag,
+ uint32_t level)
+{
+ int ret = IGZIP_COMP_OK;
+ struct isal_zstream stream;
+ struct isal_zstate *state = &stream.internal_state;
+ uint32_t level_buf_size;
+ uint8_t *level_buf = NULL;
+
+#ifdef VERBOSE
+ printf("Starting Compress Single Pass\n");
+#endif
+
+ create_rand_repeat_data((uint8_t *) & stream, sizeof(stream));
+
+ isal_deflate_init(&stream);
+
+ set_random_hufftable(&stream);
+
+ if (state->state != ZSTATE_NEW_HDR)
+ return COMPRESS_INCORRECT_STATE;
+
+ stream.flush = flush_type;
+ stream.avail_in = data_size;
+ stream.next_in = data;
+ stream.avail_out = *compressed_size;
+ stream.next_out = compressed_buf;
+ stream.end_of_stream = 1;
+ stream.gzip_flag = gzip_flag;
+ stream.level = level;
+
+ if (level >= 1) {
+ level_buf_size = rand() % IBUF_SIZE + ISAL_DEF_LVL1_MIN;
+ level_buf = malloc(level_buf_size);
+ create_rand_repeat_data(level_buf, level_buf_size);
+ stream.level_buf = level_buf;
+ stream.level_buf_size = level_buf_size;
+ }
+
+ ret =
+ isal_deflate_with_checks(&stream, data_size, *compressed_size, data, data_size,
+ data_size, compressed_buf, *compressed_size, 0);
+
+ if (level_buf != NULL)
+ free(level_buf);
+
+ /* Check if the compression is completed */
+ if (state->state == ZSTATE_END)
+ *compressed_size = stream.total_out;
+ else if (flush_type == SYNC_FLUSH && stream.avail_out < 16)
+ ret = COMPRESS_OUT_BUFFER_OVERFLOW;
+
+ return ret;
+
+}
+
+/* Statelessly compress the input buffer into the output buffer */
+int compress_stateless(uint8_t * data, uint32_t data_size, uint8_t * compressed_buf,
+ uint32_t * compressed_size, uint32_t flush_type, uint32_t gzip_flag,
+ uint32_t level)
+{
+ int ret = IGZIP_COMP_OK;
+ struct isal_zstream stream;
+ uint32_t level_buf_size;
+ uint8_t *level_buf = NULL;
+
+ create_rand_repeat_data((uint8_t *) & stream, sizeof(stream));
+
+ isal_deflate_stateless_init(&stream);
+
+ set_random_hufftable(&stream);
+
+ stream.avail_in = data_size;
+ stream.next_in = data;
+ stream.flush = flush_type;
+ if (flush_type != NO_FLUSH)
+ stream.end_of_stream = 1;
+ stream.avail_out = *compressed_size;
+ stream.next_out = compressed_buf;
+ stream.gzip_flag = gzip_flag;
+ stream.level = level;
+
+ if (level >= 1) {
+ level_buf_size = rand() % IBUF_SIZE;
+ if (level_buf_size >= ISAL_DEF_LVL1_MIN) {
+ level_buf = malloc(level_buf_size);
+ create_rand_repeat_data(level_buf, level_buf_size);
+ stream.level_buf = level_buf;
+ stream.level_buf_size = level_buf_size;
+ }
+ }
+
+ ret = isal_deflate_stateless(&stream);
+
+ if (level_buf != NULL)
+ free(level_buf);
+
+ /* verify the stream */
+ if (stream.next_in - data != stream.total_in ||
+ stream.total_in + stream.avail_in != data_size)
+ return COMPRESS_INPUT_STREAM_INTEGRITY_ERROR;
+
+ if (stream.next_out - compressed_buf != stream.total_out ||
+ stream.total_out + stream.avail_out != *compressed_size) {
+ return COMPRESS_OUTPUT_STREAM_INTEGRITY_ERROR;
+ }
+
+ if (ret != IGZIP_COMP_OK) {
+ if (ret == STATELESS_OVERFLOW)
+ return COMPRESS_OUT_BUFFER_OVERFLOW;
+ else if (ret == INVALID_FLUSH)
+ return INVALID_FLUSH_ERROR;
+ else
+ return COMPRESS_GENERAL_ERROR;
+ }
+
+ if (!stream.end_of_stream) {
+ return COMPRESS_END_OF_STREAM_NOT_SET;
+ }
+
+ if (stream.avail_in != 0)
+ return COMPRESS_ALL_INPUT_FAIL;
+
+ *compressed_size = stream.total_out;
+
+ return ret;
+
+}
+
+/* Statelessly compress the input buffer into the output buffer */
+int compress_stateless_full_flush(uint8_t * data, uint32_t data_size, uint8_t * compressed_buf,
+ uint32_t * compressed_size, uint32_t level)
+{
+ int ret = IGZIP_COMP_OK;
+ uint8_t *in_buf = NULL, *level_buf = NULL, *out_buf = compressed_buf;
+ uint32_t in_size = 0, level_buf_size;
+ uint32_t in_processed = 00;
+ struct isal_zstream stream;
+ uint32_t loop_count = 0;
+
+#ifdef VERBOSE
+ printf("Starting Stateless Compress Full Flush\n");
+#endif
+
+ create_rand_repeat_data((uint8_t *) & stream, sizeof(stream));
+
+ isal_deflate_stateless_init(&stream);
+
+ stream.flush = FULL_FLUSH;
+ stream.end_of_stream = 0;
+ stream.avail_out = *compressed_size;
+ stream.next_out = compressed_buf;
+ stream.level = level;
+
+ if (level >= 1) {
+ level_buf_size = rand() % IBUF_SIZE;
+ if (level_buf_size >= ISAL_DEF_LVL1_MIN) {
+ level_buf = malloc(level_buf_size);
+ create_rand_repeat_data(level_buf, level_buf_size);
+ stream.level_buf = level_buf;
+ stream.level_buf_size = level_buf_size;
+ }
+ }
+
+ while (1) {
+ loop_count++;
+
+ /* Randomly choose size of the next out buffer */
+ in_size = rand() % (data_size + 1);
+
+ /* Limit size of buffer to be smaller than maximum */
+ if (in_size >= data_size - in_processed) {
+ in_size = data_size - in_processed;
+ stream.end_of_stream = 1;
+ }
+
+ stream.avail_in = in_size;
+
+ if (in_size != 0) {
+ if (in_buf != NULL) {
+ free(in_buf);
+ in_buf = NULL;
+ }
+
+ in_buf = malloc(in_size);
+ if (in_buf == NULL) {
+ ret = MALLOC_FAILED;
+ break;
+ }
+ memcpy(in_buf, data + in_processed, in_size);
+ in_processed += in_size;
+
+ stream.next_in = in_buf;
+ }
+
+ out_buf = stream.next_out;
+
+ if (stream.internal_state.state == ZSTATE_NEW_HDR)
+ set_random_hufftable(&stream);
+
+ ret = isal_deflate_stateless(&stream);
+ assert(stream.internal_state.bitbuf.m_bit_count == 0);
+
+ assert(compressed_buf == stream.next_out - stream.total_out);
+ if (ret)
+ break;
+
+ /* Verify that blocks are independent */
+ ret = inflate_check(out_buf, stream.next_out - out_buf, in_buf, in_size, 0);
+
+ if (ret == INFLATE_INVALID_LOOK_BACK_DISTANCE) {
+ break;
+ } else
+ ret = 0;
+
+ /* Check if the compression is completed */
+ if (in_processed == data_size) {
+ *compressed_size = stream.total_out;
+ break;
+ }
+
+ }
+
+ if (level_buf != NULL)
+ free(level_buf);
+
+ if (in_buf != NULL)
+ free(in_buf);
+
+ if (ret == STATELESS_OVERFLOW && loop_count >= MAX_LOOPS)
+ ret = COMPRESS_LOOP_COUNT_OVERFLOW;
+
+ return ret;
+
+}
+
+/* Compress the input data into the output buffer where the input buffer and
+ * is randomly segmented to test for independence of blocks in full flush
+ * compression*/
+int compress_full_flush(uint8_t * data, uint32_t data_size, uint8_t * compressed_buf,
+ uint32_t * compressed_size, uint32_t gzip_flag, uint32_t level)
+{
+ int ret = IGZIP_COMP_OK;
+ uint8_t *in_buf = NULL, *out_buf = compressed_buf, *level_buf = NULL;
+ uint32_t in_size = 0, level_buf_size;
+ uint32_t in_processed = 00;
+ struct isal_zstream stream;
+ struct isal_zstate *state = &stream.internal_state;
+ uint32_t loop_count = 0;
+
+#ifdef VERBOSE
+ printf("Starting Compress Full Flush\n");
+#endif
+
+ create_rand_repeat_data((uint8_t *) & stream, sizeof(stream));
+
+ isal_deflate_init(&stream);
+
+ if (state->state != ZSTATE_NEW_HDR)
+ return COMPRESS_INCORRECT_STATE;
+
+ stream.flush = FULL_FLUSH;
+ stream.end_of_stream = 0;
+ stream.avail_out = *compressed_size;
+ stream.next_out = compressed_buf;
+ stream.total_out = 0;
+ stream.gzip_flag = gzip_flag;
+ stream.level = level;
+
+ if (level >= 1) {
+ level_buf_size = rand() % IBUF_SIZE + ISAL_DEF_LVL1_MIN;
+ if (level_buf_size >= ISAL_DEF_LVL1_MIN) {
+ level_buf = malloc(level_buf_size);
+ create_rand_repeat_data(level_buf, level_buf_size);
+ stream.level_buf = level_buf;
+ stream.level_buf_size = level_buf_size;
+ }
+ }
+
+ while (1) {
+ loop_count++;
+
+ /* Setup in buffer for next round of compression */
+ if (state->state == ZSTATE_NEW_HDR) {
+ /* Randomly choose size of the next out buffer */
+ in_size = rand() % (data_size + 1);
+
+ /* Limit size of buffer to be smaller than maximum */
+ if (in_size >= data_size - in_processed) {
+ in_size = data_size - in_processed;
+ stream.end_of_stream = 1;
+ }
+
+ stream.avail_in = in_size;
+
+ if (in_size != 0) {
+ if (in_buf != NULL) {
+ free(in_buf);
+ in_buf = NULL;
+ }
+
+ in_buf = malloc(in_size);
+ if (in_buf == NULL) {
+ ret = MALLOC_FAILED;
+ break;
+ }
+ memcpy(in_buf, data + in_processed, in_size);
+ in_processed += in_size;
+
+ stream.next_in = in_buf;
+ }
+
+ out_buf = stream.next_out;
+ }
+
+ if (state->state == ZSTATE_NEW_HDR)
+ set_random_hufftable(&stream);
+
+ ret = isal_deflate(&stream);
+
+ if (ret)
+ break;
+
+ /* Verify that blocks are independent */
+ if (state->state == ZSTATE_NEW_HDR || state->state == ZSTATE_END) {
+ ret =
+ inflate_check(out_buf, stream.next_out - out_buf, in_buf, in_size,
+ 0);
+
+ if (ret == INFLATE_INVALID_LOOK_BACK_DISTANCE)
+ break;
+ else
+ ret = 0;
+ }
+
+ /* Check if the compression is completed */
+ if (state->state == ZSTATE_END) {
+ *compressed_size = stream.total_out;
+ break;
+ }
+
+ }
+
+ if (level_buf != NULL)
+ free(level_buf);
+
+ if (in_buf != NULL)
+ free(in_buf);
+
+ if (ret == COMPRESS_OUT_BUFFER_OVERFLOW && loop_count >= MAX_LOOPS)
+ ret = COMPRESS_LOOP_COUNT_OVERFLOW;
+
+ return ret;
+
+}
+
+/*Compress the input buffer into the output buffer, but switch the flush type in
+ * the middle of the compression to test what happens*/
+int compress_swap_flush(uint8_t * data, uint32_t data_size, uint8_t * compressed_buf,
+ uint32_t * compressed_size, uint32_t flush_type, uint32_t gzip_flag)
+{
+ int ret = IGZIP_COMP_OK;
+ struct isal_zstream stream;
+ struct isal_zstate *state = &stream.internal_state;
+ uint32_t partial_size;
+
+#ifdef VERBOSE
+ printf("Starting Compress Swap Flush\n");
+#endif
+
+ isal_deflate_init(&stream);
+
+ set_random_hufftable(&stream);
+
+ if (state->state != ZSTATE_NEW_HDR)
+ return COMPRESS_INCORRECT_STATE;
+
+ partial_size = rand() % (data_size + 1);
+
+ stream.flush = flush_type;
+ stream.avail_in = partial_size;
+ stream.next_in = data;
+ stream.avail_out = *compressed_size;
+ stream.next_out = compressed_buf;
+ stream.end_of_stream = 0;
+ stream.gzip_flag = gzip_flag;
+
+ ret =
+ isal_deflate_with_checks(&stream, data_size, *compressed_size, data, partial_size,
+ partial_size, compressed_buf, *compressed_size, 0);
+
+ if (ret)
+ return ret;
+
+ if (state->state == ZSTATE_NEW_HDR)
+ set_random_hufftable(&stream);
+
+ flush_type = rand() % 3;
+
+ stream.flush = flush_type;
+ stream.avail_in = data_size - partial_size;
+ stream.next_in = data + partial_size;
+ stream.end_of_stream = 1;
+
+ ret =
+ isal_deflate_with_checks(&stream, data_size, *compressed_size, data + partial_size,
+ data_size - partial_size, data_size, compressed_buf,
+ *compressed_size, 0);
+
+ if (ret == COMPRESS_GENERAL_ERROR)
+ return INVALID_FLUSH_ERROR;
+
+ *compressed_size = stream.total_out;
+
+ return ret;
+}
+
+/* Test deflate_stateless */
+int test_compress_stateless(uint8_t * in_data, uint32_t in_size, uint32_t flush_type)
+{
+ int ret = IGZIP_COMP_OK;
+ uint32_t z_size, overflow, gzip_flag, level;
+ uint8_t *z_buf = NULL;
+ uint8_t *in_buf = NULL;
+
+ gzip_flag = rand() % 3;
+ level = rand() % 2;
+
+ if (in_size != 0) {
+ in_buf = malloc(in_size);
+
+ if (in_buf == NULL)
+ return MALLOC_FAILED;
+
+ memcpy(in_buf, in_data, in_size);
+ }
+
+ /* Test non-overflow case where a type 0 block is not written */
+ z_size = 2 * in_size + hdr_bytes;
+ if (gzip_flag)
+ z_size += gzip_extra_bytes;
+
+ z_buf = malloc(z_size);
+
+ if (z_buf == NULL)
+ return MALLOC_FAILED;
+
+ create_rand_repeat_data(z_buf, z_size);
+
+ /* If flush type is invalid */
+ if (flush_type != NO_FLUSH && flush_type != FULL_FLUSH) {
+ ret =
+ compress_stateless(in_buf, in_size, z_buf, &z_size, flush_type, gzip_flag,
+ level);
+
+ if (ret != INVALID_FLUSH_ERROR)
+ print_error(ret);
+ else
+ ret = 0;
+
+ if (z_buf != NULL)
+ free(z_buf);
+
+ if (in_buf != NULL)
+ free(in_buf);
+
+ return ret;
+ }
+
+ /* Else test valid flush type */
+ ret =
+ compress_stateless(in_buf, in_size, z_buf, &z_size, flush_type, gzip_flag, level);
+
+ if (!ret)
+ ret = inflate_check(z_buf, z_size, in_buf, in_size, gzip_flag);
+
+#ifdef VERBOSE
+ if (ret) {
+ printf("Compressed array at level %d with gzip flag %d: ", level, gzip_flag);
+ print_uint8_t(z_buf, z_size);
+ printf("\n");
+ printf("Data: ");
+ print_uint8_t(in_buf, in_size);
+ }
+#endif
+ if (z_buf != NULL) {
+ free(z_buf);
+ z_buf = NULL;
+ }
+
+ print_error(ret);
+ if (ret)
+ return ret;
+
+ /*Test non-overflow case where a type 0 block is possible to be written */
+ z_size = TYPE0_HDR_SIZE * ((in_size + TYPE0_MAX_SIZE - 1) / TYPE0_MAX_SIZE) + in_size;
+ if (gzip_flag)
+ z_size += gzip_extra_bytes;
+
+ if (z_size == gzip_extra_bytes)
+ z_size += TYPE0_HDR_SIZE;
+
+ if (z_size < 8)
+ z_size = 8;
+
+ z_buf = malloc(z_size);
+
+ if (z_buf == NULL)
+ return MALLOC_FAILED;
+
+ create_rand_repeat_data(z_buf, z_size);
+
+ ret =
+ compress_stateless(in_buf, in_size, z_buf, &z_size, flush_type, gzip_flag, level);
+ if (!ret)
+ ret = inflate_check(z_buf, z_size, in_buf, in_size, gzip_flag);
+#ifdef VERBOSE
+ if (ret) {
+ printf("Compressed array at level %d with gzip flag %d: ", level, gzip_flag);
+ print_uint8_t(z_buf, z_size);
+ printf("\n");
+ printf("Data: ");
+ print_uint8_t(in_buf, in_size);
+ }
+#endif
+
+ if (!ret) {
+ free(z_buf);
+ z_buf = NULL;
+
+ /* Test random overflow case */
+ z_size = rand() % z_size;
+
+ if (z_size > in_size)
+ z_size = rand() & in_size;
+
+ if (z_size > 0) {
+ z_buf = malloc(z_size);
+
+ if (z_buf == NULL)
+ return MALLOC_FAILED;
+ }
+
+ overflow =
+ compress_stateless(in_buf, in_size, z_buf, &z_size, flush_type, gzip_flag,
+ level);
+
+ if (overflow != COMPRESS_OUT_BUFFER_OVERFLOW) {
+#ifdef VERBOSE
+ printf("overflow error = %d\n", overflow);
+ print_error(overflow);
+ if (overflow == 0) {
+ overflow =
+ inflate_check(z_buf, z_size, in_buf, in_size, gzip_flag);
+ printf("inflate ret = %d\n", overflow);
+ print_error(overflow);
+ }
+ printf("Compressed array at level %d with gzip flag %d: ", level,
+ gzip_flag);
+ print_uint8_t(z_buf, z_size);
+ printf("\n");
+ printf("Data: ");
+ print_uint8_t(in_buf, in_size);
+#endif
+ }
+ }
+
+ print_error(ret);
+ if (ret) {
+ if (z_buf != NULL) {
+ free(z_buf);
+ z_buf = NULL;
+ }
+ if (in_buf != NULL)
+ free(in_buf);
+ return ret;
+ }
+
+ if (flush_type == FULL_FLUSH) {
+ if (z_buf != NULL)
+ free(z_buf);
+
+ z_size = 2 * in_size + MAX_LOOPS * (hdr_bytes + 5);
+
+ z_buf = malloc(z_size);
+
+ if (z_buf == NULL)
+ return MALLOC_FAILED;
+
+ create_rand_repeat_data(z_buf, z_size);
+
+ /* Else test valid flush type */
+ ret = compress_stateless_full_flush(in_buf, in_size, z_buf, &z_size, level);
+
+ if (!ret)
+ ret = inflate_check(z_buf, z_size, in_buf, in_size, 0);
+ else if (ret == COMPRESS_LOOP_COUNT_OVERFLOW)
+ ret = 0;
+
+ print_error(ret);
+#ifdef VERBOSE
+ if (ret) {
+ printf("Compressed array at level %d with gzip flag %d: ", level,
+ gzip_flag);
+ print_uint8_t(z_buf, z_size);
+ printf("\n");
+ printf("Data: ");
+ print_uint8_t(in_buf, in_size);
+ }
+#endif
+ }
+ if (z_buf != NULL)
+ free(z_buf);
+
+ if (in_buf != NULL)
+ free(in_buf);
+
+ return ret;
+}
+
+/* Test deflate */
+int test_compress(uint8_t * in_buf, uint32_t in_size, uint32_t flush_type)
+{
+ int ret = IGZIP_COMP_OK, fin_ret = IGZIP_COMP_OK;
+ uint32_t overflow = 0, gzip_flag, level;
+ uint32_t z_size, z_size_max, z_compressed_size;
+ uint8_t *z_buf = NULL;
+
+ /* Test a non overflow case */
+ if (flush_type == NO_FLUSH)
+ z_size_max = 2 * in_size + hdr_bytes + 2;
+ else if (flush_type == SYNC_FLUSH || flush_type == FULL_FLUSH)
+ z_size_max = 2 * in_size + MAX_LOOPS * (hdr_bytes + 5);
+ else {
+ printf("Invalid Flush Parameter\n");
+ return COMPRESS_GENERAL_ERROR;
+ }
+
+ gzip_flag = rand() % 3;
+ level = rand() % 2;
+ if (gzip_flag)
+ z_size_max += gzip_extra_bytes;
+
+ z_size = z_size_max;
+
+ z_buf = malloc(z_size);
+ if (z_buf == NULL) {
+ print_error(MALLOC_FAILED);
+ return MALLOC_FAILED;
+ }
+ create_rand_repeat_data(z_buf, z_size);
+
+ ret = compress_single_pass(in_buf, in_size, z_buf, &z_size, flush_type,
+ gzip_flag, level);
+
+ if (!ret)
+ ret = inflate_check(z_buf, z_size, in_buf, in_size, gzip_flag);
+
+ if (ret) {
+#ifdef VERBOSE
+ printf("Compressed array at level %d with gzip flag %d: ", level, gzip_flag);
+ print_uint8_t(z_buf, z_size);
+ printf("\n");
+ printf("Data: ");
+ print_uint8_t(in_buf, in_size);
+#endif
+ printf("Failed on compress single pass\n");
+ print_error(ret);
+ }
+
+ fin_ret |= ret;
+
+ z_compressed_size = z_size;
+ z_size = z_size_max;
+ create_rand_repeat_data(z_buf, z_size_max);
+
+ ret =
+ compress_multi_pass(in_buf, in_size, z_buf, &z_size, flush_type, gzip_flag, level);
+
+ if (!ret)
+ ret = inflate_check(z_buf, z_size, in_buf, in_size, gzip_flag);
+
+ if (ret) {
+#ifdef VERBOSE
+ printf("Compressed array at level %d with gzip flag %d: ", level, gzip_flag);
+ print_uint8_t(z_buf, z_size);
+ printf("\n");
+ printf("Data: ");
+ print_uint8_t(in_buf, in_size);
+#endif
+ printf("Failed on compress multi pass\n");
+ print_error(ret);
+ }
+
+ fin_ret |= ret;
+
+ ret = 0;
+
+ /* Test random overflow case */
+ if (flush_type == SYNC_FLUSH && z_compressed_size > in_size)
+ z_compressed_size = in_size + 1;
+
+ z_size = rand() % z_compressed_size;
+ create_rand_repeat_data(z_buf, z_size_max);
+
+ overflow = compress_single_pass(in_buf, in_size, z_buf, &z_size, flush_type,
+ gzip_flag, level);
+
+ if (overflow != COMPRESS_OUT_BUFFER_OVERFLOW) {
+ if (overflow == 0)
+ ret = inflate_check(z_buf, z_size, in_buf, in_size, gzip_flag);
+
+ /* Rarely single pass overflow will compresses data
+ * better than the initial run. This is to stop that
+ * case from erroring. */
+ if (overflow != 0 || ret != 0) {
+#ifdef VERBOSE
+ printf("overflow error = %d\n", overflow);
+ print_error(overflow);
+ printf("inflate ret = %d\n", ret);
+ print_error(overflow);
+
+ printf("Compressed array at level %d with gzip flag %d: ", level,
+ gzip_flag);
+ print_uint8_t(z_buf, z_size);
+ printf("\n");
+ printf("Data: ");
+ print_uint8_t(in_buf, in_size);
+#endif
+ printf("Failed on compress multi pass overflow\n");
+ print_error(ret);
+ ret = OVERFLOW_TEST_ERROR;
+ }
+ }
+
+ fin_ret |= ret;
+
+ if (flush_type == NO_FLUSH) {
+ create_rand_repeat_data(z_buf, z_size_max);
+
+ overflow =
+ compress_multi_pass(in_buf, in_size, z_buf, &z_size, flush_type,
+ gzip_flag, level);
+
+ if (overflow != COMPRESS_OUT_BUFFER_OVERFLOW) {
+ if (overflow == 0)
+ ret = inflate_check(z_buf, z_size, in_buf, in_size, gzip_flag);
+
+ /* Rarely multi pass overflow will compresses data
+ * better than the initial run. This is to stop that
+ * case from erroring */
+ if (overflow != 0 || ret != 0) {
+#ifdef VERBOSE
+ printf("overflow error = %d\n", overflow);
+ print_error(overflow);
+ printf("inflate ret = %d\n", ret);
+ print_error(overflow);
+
+ printf("Compressed array at level %d with gzip flag %d: ",
+ level, gzip_flag);
+ print_uint8_t(z_buf, z_size);
+ printf("\n");
+ printf("Data: ");
+ print_uint8_t(in_buf, in_size);
+#endif
+ printf("Failed on compress multi pass overflow\n");
+ print_error(ret);
+ ret = OVERFLOW_TEST_ERROR;
+ }
+ }
+ fin_ret |= ret;
+ }
+
+ free(z_buf);
+
+ return fin_ret;
+}
+
+/* Test swapping flush types in the middle of compression */
+int test_flush(uint8_t * in_buf, uint32_t in_size)
+{
+ int fin_ret = IGZIP_COMP_OK, ret;
+ uint32_t z_size, flush_type = 0, gzip_flag, level;
+ uint8_t *z_buf = NULL;
+
+ gzip_flag = rand() % 3;
+ level = rand() % 2;
+
+ z_size = 2 * in_size + 2 * hdr_bytes + 8;
+ if (gzip_flag)
+ z_size += gzip_extra_bytes;
+ z_buf = malloc(z_size);
+
+ if (z_buf == NULL)
+ return MALLOC_FAILED;
+
+ create_rand_repeat_data(z_buf, z_size);
+
+ while (flush_type < 3)
+ flush_type = rand();
+
+ /* Test invalid flush */
+ ret = compress_single_pass(in_buf, in_size, z_buf, &z_size, flush_type,
+ gzip_flag, level);
+
+ if (ret == COMPRESS_GENERAL_ERROR)
+ ret = 0;
+ else {
+ printf("Failed when passing invalid flush parameter\n");
+ ret = INVALID_FLUSH_ERROR;
+ }
+
+ fin_ret |= ret;
+ print_error(ret);
+
+ create_rand_repeat_data(z_buf, z_size);
+
+ /* Test swapping flush type */
+ ret = compress_swap_flush(in_buf, in_size, z_buf, &z_size, rand() % 3, gzip_flag);
+
+ if (!ret)
+ ret = inflate_check(z_buf, z_size, in_buf, in_size, gzip_flag);
+
+ if (ret) {
+#ifdef VERBOSE
+ printf("Compressed array at level %d with gzip flag %d: ", level, gzip_flag);
+ print_uint8_t(z_buf, z_size);
+ printf("\n");
+ printf("Data: ");
+ print_uint8_t(in_buf, in_size);
+#endif
+ printf("Failed on swapping flush type\n");
+ print_error(ret);
+ }
+
+ fin_ret |= ret;
+ print_error(ret);
+
+ return fin_ret;
+}
+
+/* Test there are no length distance pairs across full flushes */
+int test_full_flush(uint8_t * in_buf, uint32_t in_size)
+{
+ int ret = IGZIP_COMP_OK;
+ uint32_t z_size, gzip_flag, level;
+ uint8_t *z_buf = NULL;
+
+ gzip_flag = rand() % 3;
+ level = rand() % 2;
+ z_size = 2 * in_size + MAX_LOOPS * (hdr_bytes + 5);
+
+ if (gzip_flag)
+ z_size += gzip_extra_bytes;
+
+ z_buf = malloc(z_size);
+ if (z_buf == NULL) {
+ print_error(MALLOC_FAILED);
+ return MALLOC_FAILED;
+ }
+
+ create_rand_repeat_data(z_buf, z_size);
+
+ ret = compress_full_flush(in_buf, in_size, z_buf, &z_size, gzip_flag, level);
+
+ if (!ret)
+ ret = inflate_check(z_buf, z_size, in_buf, in_size, gzip_flag);
+
+ if (ret) {
+#ifdef VERBOSE
+ printf("Compressed array: ");
+ print_uint8_t(z_buf, z_size);
+ printf("\n");
+ printf("Data: ");
+ print_uint8_t(in_buf, in_size);
+#endif
+ printf("Failed on compress multi pass\n");
+ print_error(ret);
+ }
+
+ free(z_buf);
+
+ return ret;
+}
+
+int test_inflate(struct vect_result *in_vector)
+{
+ int ret = IGZIP_COMP_OK;
+ uint8_t *compress_buf = in_vector->vector, *out_buf = NULL;
+ uint64_t compress_len = in_vector->vector_length;
+ uint32_t out_size = 0;
+
+ out_size = 10 * in_vector->vector_length;
+ out_buf = malloc(out_size);
+ if (out_buf == NULL)
+ return MALLOC_FAILED;
+
+ ret = inflate_stateless_pass(compress_buf, compress_len, out_buf, &out_size, 0);
+
+ if (ret == INFLATE_LEFTOVER_INPUT)
+ ret = ISAL_DECOMP_OK;
+
+ if (ret != in_vector->expected_error)
+ printf("Inflate return value incorrect, %d != %d\n", ret,
+ in_vector->expected_error);
+ else
+ ret = IGZIP_COMP_OK;
+
+ if (!ret) {
+ ret = inflate_multi_pass(compress_buf, compress_len, out_buf, &out_size, 0);
+
+ if (ret == INFLATE_LEFTOVER_INPUT)
+ ret = ISAL_DECOMP_OK;
+
+ if (ret != in_vector->expected_error)
+ printf("Inflate return value incorrect, %d != %d\n", ret,
+ in_vector->expected_error);
+ else
+ ret = IGZIP_COMP_OK;
+ }
+
+ return ret;
+
+}
+
+int get_filesize(FILE * f)
+{
+ int curr, end;
+
+ curr = ftell(f); /* Save current position */
+ fseek(f, 0L, SEEK_END);
+ end = ftell(f);
+ fseek(f, curr, SEEK_SET); /* Restore position */
+ return end;
+}
+
+/* Run multiple compression tests on data stored in a file */
+int test_compress_file(char *file_name)
+{
+ int ret = IGZIP_COMP_OK;
+ uint32_t in_size;
+ uint8_t *in_buf = NULL;
+ FILE *in_file = NULL;
+
+ in_file = fopen(file_name, "rb");
+ if (!in_file)
+ return FILE_READ_FAILED;
+
+ in_size = get_filesize(in_file);
+ if (in_size != 0) {
+ in_buf = malloc(in_size);
+ if (in_buf == NULL)
+ return MALLOC_FAILED;
+ fread(in_buf, 1, in_size, in_file);
+ }
+
+ ret |= test_compress_stateless(in_buf, in_size, NO_FLUSH);
+ ret |= test_compress_stateless(in_buf, in_size, SYNC_FLUSH);
+ ret |= test_compress_stateless(in_buf, in_size, FULL_FLUSH);
+ ret |= test_compress(in_buf, in_size, NO_FLUSH);
+ ret |= test_compress(in_buf, in_size, SYNC_FLUSH);
+ ret |= test_compress(in_buf, in_size, FULL_FLUSH);
+ ret |= test_flush(in_buf, in_size);
+
+ if (ret)
+ printf("Failed on file %s\n", file_name);
+
+ if (in_buf != NULL)
+ free(in_buf);
+
+ return ret;
+}
+
+int create_custom_hufftables(struct isal_hufftables *hufftables_custom, int argc, char *argv[])
+{
+ long int file_length;
+ uint8_t *stream = NULL;
+ struct isal_huff_histogram histogram;
+ FILE *file;
+
+ memset(&histogram, 0, sizeof(histogram));
+
+ while (argc > 1) {
+ printf("Processing %s\n", argv[argc - 1]);
+ file = fopen(argv[argc - 1], "r");
+ if (file == NULL) {
+ printf("Error opening file\n");
+ return 1;
+ }
+ fseek(file, 0, SEEK_END);
+ file_length = ftell(file);
+ fseek(file, 0, SEEK_SET);
+ file_length -= ftell(file);
+
+ if (file_length > 0) {
+ stream = malloc(file_length);
+ if (stream == NULL) {
+ printf("Failed to allocate memory to read in file\n");
+ fclose(file);
+ return 1;
+ }
+ }
+
+ fread(stream, 1, file_length, file);
+
+ if (ferror(file)) {
+ printf("Error occurred when reading file\n");
+ fclose(file);
+ free(stream);
+ stream = NULL;
+ return 1;
+ }
+
+ /* Create a histogram of frequency of symbols found in stream to
+ * generate the huffman tree.*/
+ isal_update_histogram(stream, file_length, &histogram);
+
+ fclose(file);
+ if (stream != NULL) {
+ free(stream);
+ stream = NULL;
+ }
+ argc--;
+ }
+
+ return isal_create_hufftables(hufftables_custom, &histogram);
+
+}
+
+int main(int argc, char *argv[])
+{
+ int i = 0, ret = 0, fin_ret = 0;
+ uint32_t in_size = 0, offset = 0;
+ uint8_t *in_buf;
+ struct isal_hufftables hufftables_custom;
+
+#ifndef VERBOSE
+ setbuf(stdout, NULL);
+#endif
+
+ printf("Window Size: %d K\n", IGZIP_HIST_SIZE / 1024);
+ printf("Test Seed : %d\n", TEST_SEED);
+ printf("Randoms : %d\n", RANDOMS);
+ srand(TEST_SEED);
+
+ if (argc > 1) {
+ ret = create_custom_hufftables(&hufftables_custom, argc, argv);
+ if (ret == 0)
+ hufftables = &hufftables_custom;
+ else {
+ printf("Failed to generate custom hufftable");
+ return -1;
+ }
+ }
+
+ in_buf = malloc(IBUF_SIZE);
+ memset(in_buf, 0, IBUF_SIZE);
+
+ if (in_buf == NULL) {
+ fprintf(stderr, "Can't allocate in_buf memory\n");
+ return -1;
+ }
+
+ if (argc > 1) {
+ printf("igzip_rand_test files: ");
+
+ for (i = 1; i < argc; i++) {
+ ret |= test_compress_file(argv[i]);
+ if (ret)
+ return ret;
+ }
+
+ printf("................");
+ printf("%s\n", ret ? "Fail" : "Pass");
+ fin_ret |= ret;
+ }
+
+ printf("igzip_rand_test stateless: ");
+
+ ret = test_compress_stateless((uint8_t *) str1, sizeof(str1), NO_FLUSH);
+ if (ret)
+ return ret;
+
+ ret |= test_compress_stateless((uint8_t *) str2, sizeof(str2), NO_FLUSH);
+ if (ret)
+ return ret;
+
+ for (i = 0; i < RANDOMS; i++) {
+ in_size = rand() % (IBUF_SIZE + 1);
+ offset = rand() % (IBUF_SIZE + 1 - in_size);
+ in_buf += offset;
+
+ create_rand_repeat_data(in_buf, in_size);
+
+ ret |= test_compress_stateless(in_buf, in_size, NO_FLUSH);
+
+ in_buf -= offset;
+
+ if (i % (RANDOMS / 16) == 0)
+ printf(".");
+
+ if (ret)
+ return ret;
+ }
+
+ for (i = 0; i < RANDOMS / 16; i++) {
+ create_rand_repeat_data(in_buf, PAGE_SIZE);
+ ret |= test_compress_stateless(in_buf, PAGE_SIZE, NO_FLUSH); // good for efence
+ if (ret)
+ return ret;
+ }
+
+ fin_ret |= ret;
+
+ ret = test_compress_stateless((uint8_t *) str1, sizeof(str1), SYNC_FLUSH);
+ if (ret)
+ return ret;
+
+ ret |= test_compress_stateless((uint8_t *) str2, sizeof(str2), SYNC_FLUSH);
+ if (ret)
+ return ret;
+
+ for (i = 0; i < 16; i++) {
+ in_size = rand() % (IBUF_SIZE + 1);
+ offset = rand() % (IBUF_SIZE + 1 - in_size);
+ in_buf += offset;
+
+ create_rand_repeat_data(in_buf, in_size);
+
+ ret |= test_compress_stateless(in_buf, in_size, SYNC_FLUSH);
+
+ in_buf -= offset;
+
+ if (ret)
+ return ret;
+ }
+
+ fin_ret |= ret;
+
+ printf("%s\n", ret ? "Fail" : "Pass");
+
+ printf("igzip_rand_test stateless FULL_FLUSH: ");
+
+ ret = test_compress_stateless((uint8_t *) str1, sizeof(str1), FULL_FLUSH);
+ if (ret)
+ return ret;
+
+ ret |= test_compress_stateless((uint8_t *) str2, sizeof(str2), FULL_FLUSH);
+ if (ret)
+ return ret;
+
+ for (i = 0; i < RANDOMS; i++) {
+ in_size = rand() % (IBUF_SIZE + 1);
+ offset = rand() % (IBUF_SIZE + 1 - in_size);
+ in_buf += offset;
+
+ create_rand_repeat_data(in_buf, in_size);
+
+ ret |= test_compress_stateless(in_buf, in_size, FULL_FLUSH);
+
+ in_buf -= offset;
+
+ if (i % (RANDOMS / 16) == 0)
+ printf(".");
+
+ if (ret)
+ return ret;
+ }
+
+ for (i = 0; i < RANDOMS / 16; i++) {
+ create_rand_repeat_data(in_buf, PAGE_SIZE);
+ ret |= test_compress_stateless(in_buf, PAGE_SIZE, FULL_FLUSH); // good for efence
+ if (ret)
+ return ret;
+ }
+ fin_ret |= ret;
+
+ printf("%s\n", ret ? "Fail" : "Pass");
+
+ printf("igzip_rand_test stateful NO_FLUSH: ");
+
+ ret = test_compress((uint8_t *) str1, sizeof(str1), NO_FLUSH);
+ if (ret)
+ return ret;
+
+ ret |= test_compress((uint8_t *) str2, sizeof(str2), NO_FLUSH);
+ if (ret)
+ return ret;
+
+ for (i = 0; i < RANDOMS; i++) {
+ in_size = rand() % (IBUF_SIZE + 1);
+ offset = rand() % (IBUF_SIZE + 1 - in_size);
+ in_buf += offset;
+
+ create_rand_repeat_data(in_buf, in_size);
+
+ ret |= test_compress(in_buf, in_size, NO_FLUSH);
+
+ in_buf -= offset;
+
+ if (i % (RANDOMS / 16) == 0)
+ printf(".");
+ if (ret)
+ return ret;
+ }
+
+ fin_ret |= ret;
+
+ printf("%s\n", ret ? "Fail" : "Pass");
+
+ printf("igzip_rand_test stateful SYNC_FLUSH: ");
+
+ ret = test_compress((uint8_t *) str1, sizeof(str1), SYNC_FLUSH);
+ if (ret)
+ return ret;
+
+ ret |= test_compress((uint8_t *) str2, sizeof(str2), SYNC_FLUSH);
+ if (ret)
+ return ret;
+
+ for (i = 0; i < RANDOMS; i++) {
+ in_size = rand() % (IBUF_SIZE + 1);
+ offset = rand() % (IBUF_SIZE + 1 - in_size);
+ in_buf += offset;
+
+ create_rand_repeat_data(in_buf, in_size);
+
+ ret |= test_compress(in_buf, in_size, SYNC_FLUSH);
+
+ in_buf -= offset;
+
+ if (i % (RANDOMS / 16) == 0)
+ printf(".");
+ if (ret)
+ return ret;
+ }
+
+ fin_ret |= ret;
+
+ printf("%s\n", ret ? "Fail" : "Pass");
+
+ printf("igzip_rand_test stateful FULL_FLUSH: ");
+
+ ret = test_compress((uint8_t *) str1, sizeof(str1), FULL_FLUSH);
+ if (ret)
+ return ret;
+
+ ret |= test_compress((uint8_t *) str2, sizeof(str2), FULL_FLUSH);
+ if (ret)
+ return ret;
+
+ for (i = 0; i < RANDOMS; i++) {
+ in_size = rand() % (IBUF_SIZE + 1);
+ offset = rand() % (IBUF_SIZE + 1 - in_size);
+ in_buf += offset;
+
+ create_rand_repeat_data(in_buf, in_size);
+
+ ret |= test_compress(in_buf, in_size, FULL_FLUSH);
+
+ in_buf -= offset;
+
+ if (i % (RANDOMS / 16) == 0)
+ printf(".");
+ if (ret)
+ return ret;
+ }
+
+ for (i = 0; i < RANDOMS / 8; i++) {
+ in_size = rand() % (IBUF_SIZE + 1);
+ offset = rand() % (IBUF_SIZE + 1 - in_size);
+ in_buf += offset;
+
+ create_rand_repeat_data(in_buf, in_size);
+
+ ret |= test_full_flush(in_buf, in_size);
+
+ in_buf -= offset;
+
+ if (ret)
+ return ret;
+ }
+
+ fin_ret |= ret;
+
+ printf("%s\n", ret ? "Fail" : "Pass");
+
+ printf("igzip_rand_test stateful Change Flush: ");
+
+ ret = test_flush((uint8_t *) str1, sizeof(str1));
+ if (ret)
+ return ret;
+
+ ret |= test_flush((uint8_t *) str2, sizeof(str2));
+ if (ret)
+ return ret;
+
+ for (i = 0; i < RANDOMS / 4; i++) {
+ in_size = rand() % (IBUF_SIZE + 1);
+ offset = rand() % (IBUF_SIZE + 1 - in_size);
+ in_buf += offset;
+
+ create_rand_repeat_data(in_buf, in_size);
+
+ ret |= test_flush(in_buf, in_size);
+
+ in_buf -= offset;
+
+ if (i % ((RANDOMS / 4) / 16) == 0)
+ printf(".");
+ if (ret)
+ return ret;
+ }
+
+ fin_ret |= ret;
+
+ printf("%s\n", ret ? "Fail" : "Pass");
+
+ printf("igzip_rand_test inflate Std Vectors: ");
+
+ for (i = 0; i < sizeof(std_vect_array) / sizeof(struct vect_result); i++) {
+ ret = test_inflate(&std_vect_array[i]);
+ if (ret)
+ return ret;
+ }
+ printf("................");
+ printf("%s\n", ret ? "Fail" : "Pass");
+
+ printf("igzip rand test finished: %s\n",
+ fin_ret ? "Some tests failed" : "All tests passed");
+
+ return fin_ret != IGZIP_COMP_OK;
+}