Tests: Add MicroLZMA test.

2 files changed, 551 insertions, 1 deletions

diff --git a/tests/Makefile.am b/tests/Makefile.am
index 9cfc068b..759bd71a 100644
--- a/tests/Makefile.am
+++ b/tests/Makefile.am
@@ -42,7 +42,8 @@ check_PROGRAMS = \
 	test_bcj_exact_size \
 	test_memlimit \
 	test_lzip_decoder \
-	test_vli
+	test_vli \
+	test_microlzma
 
 TESTS = \
 	test_check \
@@ -57,6 +58,7 @@ TESTS = \
 	test_memlimit \
 	test_lzip_decoder \
 	test_vli \
+	test_microlzma \
 	test_files.sh \
 	test_suffix.sh \
 	test_compress_prepared_bcj_sparc \
diff --git a/tests/test_microlzma.c b/tests/test_microlzma.c
new file mode 100644
index 00000000..0fba9192
--- /dev/null
+++ b/tests/test_microlzma.c
@@ -0,0 +1,548 @@
+///////////////////////////////////////////////////////////////////////////////
+//
+/// \file       test_microlzma.c
+/// \brief      Tests MicroLZMA encoding and decoding
+//
+//  Author:     Jia Tan
+//
+//  This file has been put into the public domain.
+//  You can do whatever you want with this file.
+//
+///////////////////////////////////////////////////////////////////////////////
+
+#include "tests.h"
+
+#ifdef HAVE_MICROLZMA
+
+#define BUFFER_SIZE 1024
+
+#ifdef HAVE_ENCODER_LZMA1
+
+// MicroLZMA encoded "Hello\nWorld\n" output size in bytes.
+#define ENCODED_OUTPUT_SIZE 17
+
+// Byte array of "Hello\nWorld\n". This is used for various encoder tests.
+static const uint8_t hello_world[] = { 0x48, 0x65, 0x6C, 0x6C, 0x6F, 0x0A,
+		0x57, 0x6F, 0x72, 0x6C, 0x64, 0x0A };
+
+// This is the CRC32 value of the MicroLZMA encoding of "Hello\nWorld\n".
+// The settings used were based on LZMA_PRESET_DEFAULT as of liblzma 5.6.0.
+// This assumes MicroLZMA is correct in liblzma 5.6.0, which is safe
+// considering the encoded "Hello\nWorld\n" can successfully be decoded at
+// this time. This is to test for regressions that cause MicroLZMA output
+// to change.
+static const uint32_t hello_world_encoded_crc = 0x3CDE40A8;
+
+// Function implementation borrowed from lzma_decoder.c. It is needed to
+// ensure the first byte of a MicroLZMA stream is set correctly with the
+// negation of the LZMA properties.
+static bool
+lzma_lzma_lclppb_decode(lzma_options_lzma *options, uint8_t byte)
+{
+	if (byte > (4 * 5 + 4) * 9 + 8)
+		return true;
+
+	// See the file format specification to understand this.
+	options->pb = byte / (9 * 5);
+	byte -= options->pb * 9 * 5;
+	options->lp = byte / 9;
+	options->lc = byte - options->lp * 9;
+
+	return options->lc + options->lp > LZMA_LCLP_MAX;
+}
+
+
+///////////////////
+// Encoder tests //
+///////////////////
+
+// This tests a few of the basic options. These options are not unique to
+// MicroLZMA in any way, its mostly ensuring that the options are actually
+// being checked before initializing the decoder internals.
+static void
+test_encode_options(void)
+{
+	lzma_stream strm = LZMA_STREAM_INIT;
+	lzma_options_lzma opt_lzma;
+
+	// Initialize with default options.
+	assert_false(lzma_lzma_preset(&opt_lzma, LZMA_PRESET_DEFAULT));
+
+	// NULL stream
+	assert_lzma_ret(lzma_microlzma_encoder(NULL, &opt_lzma),
+			LZMA_PROG_ERROR);
+
+	// lc/lp/pb = 5/0/2 (lc invalid)
+	opt_lzma.lc = 5;
+	opt_lzma.lp = 0;
+	opt_lzma.pb = 2;
+	assert_lzma_ret(lzma_microlzma_encoder(&strm, &opt_lzma),
+			LZMA_OPTIONS_ERROR);
+
+	// lc/lp/pb = 0/5/2 (lp invalid)
+	opt_lzma.lc = 0;
+	opt_lzma.lp = 5;
+	opt_lzma.pb = 2;
+	assert_lzma_ret(lzma_microlzma_encoder(&strm, &opt_lzma),
+			LZMA_OPTIONS_ERROR);
+
+	// lc/lp/pb = 3/2/2 (lc + lp invalid)
+	opt_lzma.lc = 3;
+	opt_lzma.lp = 2;
+	opt_lzma.pb = 2;
+	assert_lzma_ret(lzma_microlzma_encoder(&strm, &opt_lzma),
+			LZMA_OPTIONS_ERROR);
+
+	// lc/lp/pb = 3/0/5 (pb invalid)
+	opt_lzma.lc = 3;
+	opt_lzma.lp = 0;
+	opt_lzma.pb = 5;
+	assert_lzma_ret(lzma_microlzma_encoder(&strm, &opt_lzma),
+			LZMA_OPTIONS_ERROR);
+
+	// Zero out lp, pb, lc options to not interfere with later tests.
+	opt_lzma.lp = 0;
+	opt_lzma.pb = 0;
+	opt_lzma.lc = 0;
+
+	// Set invalid dictionary size.
+	opt_lzma.dict_size = LZMA_DICT_SIZE_MIN - 1;
+	assert_lzma_ret(lzma_microlzma_encoder(&strm, &opt_lzma),
+			LZMA_OPTIONS_ERROR);
+
+	// Maximum dictionary size for the encoder, as described in lzma12.h
+	// is 1.5 GiB.
+	opt_lzma.dict_size = (UINT32_C(1) << 30) + (UINT32_C(1) << 29) + 1;
+	assert_lzma_ret(lzma_microlzma_encoder(&strm, &opt_lzma),
+			LZMA_OPTIONS_ERROR);
+
+	lzma_end(&strm);
+}
+
+
+static void
+test_encode_basic(void)
+{
+	lzma_stream strm = LZMA_STREAM_INIT;
+	lzma_options_lzma opt_lzma;
+
+	assert_false(lzma_lzma_preset(&opt_lzma, LZMA_PRESET_DEFAULT));
+
+	// Initialize the encoder using the default options.
+	assert_lzma_ret(lzma_microlzma_encoder(&strm, &opt_lzma), LZMA_OK);
+
+	uint8_t output[BUFFER_SIZE];
+
+	strm.next_in = hello_world;
+	strm.avail_in = ARRAY_SIZE(hello_world);
+	strm.next_out = output;
+	strm.avail_out = BUFFER_SIZE;
+
+	// Everything must be encoded in one lzma_code() call.
+	assert_lzma_ret(lzma_code(&strm, LZMA_FINISH), LZMA_STREAM_END);
+
+	// Check entire input was consumed.
+	assert_uint_eq(strm.total_in, ARRAY_SIZE(hello_world));
+
+	// Check that the first byte in the output stream is not 0x0.
+	// In regular .lzma, the first byte is always 0x0. Instead, a
+	// feature of MicroLZMA is the first byte is the bitwise-negation
+	// of the LZMA properties.
+	assert_uint(output[0], !=, 0x0);
+
+	uint8_t props = ~output[0];
+
+	lzma_options_lzma test_options;
+	assert_false(lzma_lzma_lclppb_decode(&test_options, props));
+
+	assert_uint_eq(opt_lzma.lc, test_options.lc);
+	assert_uint_eq(opt_lzma.lp, test_options.lp);
+	assert_uint_eq(opt_lzma.pb, test_options.pb);
+
+	// Compute the check over the output data. This is compared to
+	// the expected check value.
+	uint32_t check_val = lzma_crc32(output, strm.total_out, 0);
+
+	assert_uint_eq(check_val, hello_world_encoded_crc);
+
+	lzma_end(&strm);
+}
+
+
+// This tests the behavior when strm.avail_out is so small it cannot hold
+// the header plus 1 encoded byte (< 6).
+static void
+test_encode_small_out(void)
+{
+	lzma_stream strm = LZMA_STREAM_INIT;
+	lzma_options_lzma opt_lzma;
+
+	assert_false(lzma_lzma_preset(&opt_lzma, LZMA_PRESET_DEFAULT));
+
+	assert_lzma_ret(lzma_microlzma_encoder(&strm, &opt_lzma), LZMA_OK);
+
+	uint8_t output[BUFFER_SIZE];
+
+	strm.next_in = hello_world;
+	strm.avail_in = ARRAY_SIZE(hello_world);
+	strm.next_out = output;
+	strm.avail_out = 5;
+
+	// LZMA_PROG_ERROR is expected when strm.avail_out < 6
+	assert_lzma_ret(lzma_code(&strm, LZMA_FINISH), LZMA_PROG_ERROR);
+
+	// The encoder must be reset because coders cannot be used again
+	// after returning LZMA_PROG_ERROR.
+	assert_lzma_ret(lzma_microlzma_encoder(&strm, &opt_lzma), LZMA_OK);
+
+	// Reset strm.avail_out to be > 6, but not enough to hold all of the
+	// compressed data.
+	strm.avail_out = ENCODED_OUTPUT_SIZE - 1;
+
+	// Encoding should not return an error now.
+	assert_lzma_ret(lzma_code(&strm, LZMA_FINISH), LZMA_STREAM_END);
+	assert_uint(strm.total_in, <, ARRAY_SIZE(hello_world));
+
+	lzma_end(&strm);
+}
+
+
+// LZMA_FINISH is the only supported action. All others must
+// return LZMA_PROG_ERROR.
+static void
+test_encode_actions(void)
+{
+	lzma_stream strm = LZMA_STREAM_INIT;
+	lzma_options_lzma opt_lzma;
+
+	assert_false(lzma_lzma_preset(&opt_lzma, LZMA_PRESET_DEFAULT));
+
+	lzma_action actions[] = { LZMA_RUN, LZMA_SYNC_FLUSH,
+			LZMA_FULL_FLUSH, LZMA_FULL_BARRIER };
+
+	for (uint32_t i = 0; i < ARRAY_SIZE(actions); i++) {
+		assert_lzma_ret(lzma_microlzma_encoder(&strm, &opt_lzma),
+				LZMA_OK);
+
+		uint8_t output[BUFFER_SIZE];
+
+		strm.next_in = hello_world;
+		strm.avail_in = ARRAY_SIZE(hello_world);
+		strm.next_out = output;
+		strm.avail_out = BUFFER_SIZE;
+
+		assert_lzma_ret(lzma_code(&strm, actions[i]),
+				LZMA_PROG_ERROR);
+	}
+
+	lzma_end(&strm);
+}
+#endif
+
+
+///////////////////
+// Decoder tests //
+///////////////////
+
+#if defined(HAVE_DECODER_LZMA1) && defined(HAVE_ENCODER_LZMA1)
+
+// Byte array of "Goodbye World!". This is used for various decoder tests.
+static const uint8_t goodbye_world[] = { 0x47, 0x6f, 0x6f, 0x64, 0x62,
+		0x79, 0x65, 0x20, 0x57, 0x6f, 0x72, 0x6c, 0x64, 0x21 };
+
+static uint8_t *goodbye_world_encoded = NULL;
+static uint32_t goodbye_world_encoded_size = 0;
+
+
+// Helper function to encode data and return the compressed size.
+static uint32_t
+basic_microlzma_encode(const uint8_t *input, uint32_t in_size,
+		uint8_t **compressed)
+{
+	lzma_stream strm = LZMA_STREAM_INIT;
+	lzma_options_lzma opt_lzma;
+
+	// Lazy way to set the output size since the input should never
+	// inflate by much in these simple test cases. This is tested to
+	// be large enough after encoding to fit the entire input, so if
+	// this assumption does not hold then this will fail.
+	const uint32_t out_size = in_size << 1;
+
+	*compressed = tuktest_malloc(out_size);
+
+	// Always encode with the default options for simplicity.
+	if (lzma_lzma_preset(&opt_lzma, LZMA_PRESET_DEFAULT))
+		goto decoder_setup_error;
+
+	if (lzma_microlzma_encoder(&strm, &opt_lzma) != LZMA_OK)
+		goto decoder_setup_error;
+
+	strm.next_in = input;
+	strm.avail_in = in_size;
+	strm.next_out = *compressed;
+	strm.avail_out = out_size;
+
+	if (lzma_code(&strm, LZMA_FINISH) != LZMA_STREAM_END)
+		goto decoder_setup_error;
+
+	// Check the entire input was consumed and fit into the output buffer.
+	if (strm.total_in != in_size)
+		goto decoder_setup_error;
+
+	const uint64_t encoded_count = strm.total_out;
+
+	lzma_end(&strm);
+
+	return encoded_count;
+
+decoder_setup_error:
+	tuktest_error("Failed to initialize decoder tests");
+	return 0;
+}
+
+
+static void
+test_decode_options(void)
+{
+	// NULL stream
+	assert_lzma_ret(lzma_microlzma_decoder(NULL, BUFFER_SIZE,
+			ARRAY_SIZE(hello_world), true,
+			LZMA_DICT_SIZE_DEFAULT), LZMA_PROG_ERROR);
+
+	// Uncompressed size larger than max
+	lzma_stream strm = LZMA_STREAM_INIT;
+	assert_lzma_ret(lzma_microlzma_decoder(&strm, BUFFER_SIZE,
+			LZMA_VLI_MAX + 1, true, LZMA_DICT_SIZE_DEFAULT),
+			LZMA_OPTIONS_ERROR);
+}
+
+
+// Test decoding succeeds when uncomp_size is correct regardless of
+// the value of uncomp_size_is_exact.
+static void
+test_decode_uncomp_size_is_exact(void)
+{
+	lzma_stream strm = LZMA_STREAM_INIT;
+
+	assert_lzma_ret(lzma_microlzma_decoder(&strm,
+			goodbye_world_encoded_size,
+			ARRAY_SIZE(goodbye_world), true,
+			LZMA_DICT_SIZE_DEFAULT), LZMA_OK);
+
+	uint8_t output[BUFFER_SIZE];
+
+	strm.next_in = goodbye_world_encoded;
+	strm.next_out = output;
+	strm.avail_out = BUFFER_SIZE;
+	strm.avail_in = goodbye_world_encoded_size;
+
+	assert_lzma_ret(lzma_code(&strm, LZMA_RUN), LZMA_STREAM_END);
+	assert_uint_eq(strm.total_in, goodbye_world_encoded_size);
+
+	assert_uint_eq(strm.total_out, ARRAY_SIZE(goodbye_world));
+	assert_array_eq(goodbye_world, output, strm.total_out);
+
+	// Reset decoder with uncomp_size_is_exact set to false and
+	// uncomp_size set to correct value. Also test using the
+	// uncompressed size as the dictionary size.
+	assert_lzma_ret(lzma_microlzma_decoder(&strm,
+			goodbye_world_encoded_size,
+			ARRAY_SIZE(goodbye_world), false,
+			ARRAY_SIZE(goodbye_world)), LZMA_OK);
+
+	strm.next_in = goodbye_world_encoded;
+	strm.next_out = output;
+	strm.avail_out = BUFFER_SIZE;
+	strm.avail_in = goodbye_world_encoded_size;
+
+	assert_lzma_ret(lzma_code(&strm, LZMA_RUN), LZMA_STREAM_END);
+	assert_uint_eq(strm.total_in, goodbye_world_encoded_size);
+
+	assert_uint_eq(strm.total_out, ARRAY_SIZE(goodbye_world));
+	assert_array_eq(goodbye_world, output, strm.total_out);
+
+	lzma_end(&strm);
+}
+
+
+// This tests decoding when MicroLZMA decoder is called with
+// an incorrect uncompressed size.
+static void
+test_decode_uncomp_size_wrong(void)
+{
+	lzma_stream strm = LZMA_STREAM_INIT;
+	assert_lzma_ret(lzma_microlzma_decoder(&strm,
+			goodbye_world_encoded_size,
+			ARRAY_SIZE(goodbye_world) + 1, false,
+			LZMA_DICT_SIZE_DEFAULT), LZMA_OK);
+
+	uint8_t output[BUFFER_SIZE];
+
+	strm.next_in = goodbye_world_encoded;
+	strm.next_out = output;
+	strm.avail_out = BUFFER_SIZE;
+	strm.avail_in = goodbye_world_encoded_size;
+
+	// LZMA_OK should be returned because the input size given was
+	// larger than the actual encoded size. The decoder is expecting
+	// more input to possibly fill the uncompressed size that was set.
+	assert_lzma_ret(lzma_code(&strm, LZMA_FINISH), LZMA_OK);
+
+	assert_uint_eq(strm.total_out, ARRAY_SIZE(goodbye_world));
+
+	assert_array_eq(goodbye_world, output, strm.total_out);
+
+	// Next, test with uncomp_size_is_exact set.
+	assert_lzma_ret(lzma_microlzma_decoder(&strm,
+			goodbye_world_encoded_size,
+			ARRAY_SIZE(goodbye_world) + 1, true,
+			LZMA_DICT_SIZE_DEFAULT), LZMA_OK);
+
+	strm.next_in = goodbye_world_encoded;
+	strm.next_out = output;
+	strm.avail_out = BUFFER_SIZE;
+	strm.avail_in = goodbye_world_encoded_size;
+
+	// No error detected, even though all input was consumed and there
+	// is more room in the output buffer.
+	assert_lzma_ret(lzma_code(&strm, LZMA_FINISH), LZMA_OK);
+
+	assert_uint_eq(strm.total_out, ARRAY_SIZE(goodbye_world));
+	assert_array_eq(goodbye_world, output, strm.total_out);
+
+	// Reset stream with uncomp_size smaller than the real
+	// uncompressed size.
+	assert_lzma_ret(lzma_microlzma_decoder(&strm,
+			goodbye_world_encoded_size,
+			ARRAY_SIZE(hello_world) - 1, true,
+			LZMA_DICT_SIZE_DEFAULT), LZMA_OK);
+
+	strm.next_in = goodbye_world_encoded;
+	strm.next_out = output;
+	strm.avail_out = BUFFER_SIZE;
+	strm.avail_in = goodbye_world_encoded_size;
+
+	// This case actually results in an error since it decodes the full
+	// uncompressed size but the range coder is not in the proper state
+	// for the stream to end.
+	assert_lzma_ret(lzma_code(&strm, LZMA_RUN), LZMA_DATA_ERROR);
+
+	lzma_end(&strm);
+}
+
+
+static void
+test_decode_comp_size_wrong(void)
+{
+	lzma_stream strm = LZMA_STREAM_INIT;
+	assert_lzma_ret(lzma_microlzma_decoder(&strm,
+			goodbye_world_encoded_size + 1,
+			ARRAY_SIZE(goodbye_world), true,
+			LZMA_DICT_SIZE_DEFAULT), LZMA_OK);
+
+	uint8_t output[BUFFER_SIZE];
+
+	strm.next_in = goodbye_world_encoded;
+	strm.next_out = output;
+	strm.avail_out = BUFFER_SIZE;
+	strm.avail_in = goodbye_world_encoded_size;
+
+	// When uncomp_size_is_exact is set, the compressed size must be
+	// correct or else LZMA_DATA_ERROR is returned.
+	assert_lzma_ret(lzma_code(&strm, LZMA_FINISH), LZMA_DATA_ERROR);
+
+	assert_lzma_ret(lzma_microlzma_decoder(&strm,
+			goodbye_world_encoded_size + 1,
+			ARRAY_SIZE(goodbye_world), false,
+			LZMA_DICT_SIZE_DEFAULT), LZMA_OK);
+
+	strm.next_in = goodbye_world_encoded;
+	strm.next_out = output;
+	strm.avail_out = BUFFER_SIZE;
+	strm.avail_in = goodbye_world_encoded_size;
+
+	// When uncomp_size_is_exact is not set, the decoder does not
+	// detect when the compressed size is wrong as long as all of the
+	// expected output has been decoded.
+	assert_lzma_ret(lzma_code(&strm, LZMA_FINISH), LZMA_STREAM_END);
+
+	lzma_end(&strm);
+}
+
+
+static void
+test_decode_bad_lzma_properties(void)
+{
+	// Alter first byte to encode invalid LZMA properties.
+	uint8_t *compressed = tuktest_malloc(goodbye_world_encoded_size);
+	memcpy(compressed, goodbye_world_encoded, goodbye_world_encoded_size);
+
+	// lc=3, lp=2, pb=2
+	compressed[0] = (uint8_t)~0x6FU;
+
+	lzma_stream strm = LZMA_STREAM_INIT;
+	assert_lzma_ret(lzma_microlzma_decoder(&strm,
+			goodbye_world_encoded_size,
+			ARRAY_SIZE(goodbye_world), false,
+			LZMA_DICT_SIZE_DEFAULT), LZMA_OK);
+
+	uint8_t output[BUFFER_SIZE];
+
+	strm.next_in = compressed;
+	strm.next_out = output;
+	strm.avail_out = BUFFER_SIZE;
+	strm.avail_in = goodbye_world_encoded_size;
+
+	assert_lzma_ret(lzma_code(&strm, LZMA_RUN), LZMA_OPTIONS_ERROR);
+
+	// Use valid, but incorrect LZMA properties.
+	// lc=3, lp=1, pb=2
+	compressed[0] = (uint8_t)~0x66;
+
+	assert_lzma_ret(lzma_microlzma_decoder(&strm,
+			goodbye_world_encoded_size,
+			ARRAY_SIZE(goodbye_world), true,
+			LZMA_DICT_SIZE_DEFAULT), LZMA_OK);
+
+	strm.next_in = compressed;
+	strm.next_out = output;
+	strm.avail_out = BUFFER_SIZE;
+	strm.avail_in = goodbye_world_encoded_size;
+
+	assert_lzma_ret(lzma_code(&strm, LZMA_RUN), LZMA_DATA_ERROR);
+
+	lzma_end(&strm);
+}
+#endif
+#endif
+
+
+extern int
+main(int argc, char **argv)
+{
+	tuktest_start(argc, argv);
+
+#ifndef HAVE_MICROLZMA
+	tuktest_early_skip("MicroLZMA disabled");
+#else
+#	ifdef HAVE_ENCODER_LZMA1
+	tuktest_run(test_encode_options);
+	tuktest_run(test_encode_basic);
+	tuktest_run(test_encode_small_out);
+	tuktest_run(test_encode_actions);
+#	endif
+
+#	if defined(HAVE_DECODER_LZMA1) && defined(HAVE_ENCODER_LZMA1)
+	goodbye_world_encoded_size = basic_microlzma_encode(goodbye_world,
+			ARRAY_SIZE(goodbye_world), &goodbye_world_encoded);
+
+	tuktest_run(test_decode_options);
+	tuktest_run(test_decode_uncomp_size_is_exact);
+	tuktest_run(test_decode_uncomp_size_wrong);
+	tuktest_run(test_decode_comp_size_wrong);
+	tuktest_run(test_decode_bad_lzma_properties);
+#	endif
+
+	return tuktest_end();
+#endif
+}