257 lines
6.3 KiB
C
257 lines
6.3 KiB
C
///////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
/// \file test_block_header.c
|
|
/// \brief Tests Block Header coders
|
|
//
|
|
// Author: Lasse Collin
|
|
//
|
|
// This file has been put into the public domain.
|
|
// You can do whatever you want with this file.
|
|
//
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
|
|
#include "tests.h"
|
|
|
|
|
|
static uint8_t buf[LZMA_BLOCK_HEADER_SIZE_MAX];
|
|
static lzma_block known_options;
|
|
static lzma_block decoded_options;
|
|
|
|
static lzma_options_lzma opt_lzma;
|
|
|
|
static lzma_filter filters_none[1] = {
|
|
{
|
|
.id = LZMA_VLI_UNKNOWN,
|
|
},
|
|
};
|
|
|
|
|
|
static lzma_filter filters_one[2] = {
|
|
{
|
|
.id = LZMA_FILTER_LZMA2,
|
|
.options = &opt_lzma,
|
|
}, {
|
|
.id = LZMA_VLI_UNKNOWN,
|
|
}
|
|
};
|
|
|
|
|
|
static lzma_filter filters_four[5] = {
|
|
{
|
|
.id = LZMA_FILTER_X86,
|
|
.options = NULL,
|
|
}, {
|
|
.id = LZMA_FILTER_X86,
|
|
.options = NULL,
|
|
}, {
|
|
.id = LZMA_FILTER_X86,
|
|
.options = NULL,
|
|
}, {
|
|
.id = LZMA_FILTER_LZMA2,
|
|
.options = &opt_lzma,
|
|
}, {
|
|
.id = LZMA_VLI_UNKNOWN,
|
|
}
|
|
};
|
|
|
|
|
|
static lzma_filter filters_five[6] = {
|
|
{
|
|
.id = LZMA_FILTER_X86,
|
|
.options = NULL,
|
|
}, {
|
|
.id = LZMA_FILTER_X86,
|
|
.options = NULL,
|
|
}, {
|
|
.id = LZMA_FILTER_X86,
|
|
.options = NULL,
|
|
}, {
|
|
.id = LZMA_FILTER_X86,
|
|
.options = NULL,
|
|
}, {
|
|
.id = LZMA_FILTER_LZMA2,
|
|
.options = &opt_lzma,
|
|
}, {
|
|
.id = LZMA_VLI_UNKNOWN,
|
|
}
|
|
};
|
|
|
|
|
|
static void
|
|
code(void)
|
|
{
|
|
assert_lzma_ret(lzma_block_header_encode(&known_options, buf),
|
|
LZMA_OK);
|
|
|
|
lzma_filter filters[LZMA_FILTERS_MAX + 1];
|
|
memcrap(filters, sizeof(filters));
|
|
memcrap(&decoded_options, sizeof(decoded_options));
|
|
|
|
decoded_options.header_size = known_options.header_size;
|
|
decoded_options.check = known_options.check;
|
|
decoded_options.filters = filters;
|
|
assert_lzma_ret(lzma_block_header_decode(&decoded_options, NULL, buf),
|
|
LZMA_OK);
|
|
|
|
assert_uint_eq(decoded_options.compressed_size,
|
|
known_options.compressed_size);
|
|
assert_uint_eq(decoded_options.uncompressed_size,
|
|
known_options.uncompressed_size);
|
|
|
|
for (size_t i = 0; known_options.filters[i].id
|
|
!= LZMA_VLI_UNKNOWN; ++i)
|
|
assert_uint_eq(filters[i].id, known_options.filters[i].id);
|
|
|
|
for (size_t i = 0; i < LZMA_FILTERS_MAX; ++i)
|
|
free(decoded_options.filters[i].options);
|
|
}
|
|
|
|
|
|
static void
|
|
test1(void)
|
|
{
|
|
known_options = (lzma_block){
|
|
.check = LZMA_CHECK_NONE,
|
|
.compressed_size = LZMA_VLI_UNKNOWN,
|
|
.uncompressed_size = LZMA_VLI_UNKNOWN,
|
|
.filters = NULL,
|
|
};
|
|
|
|
assert_lzma_ret(lzma_block_header_size(&known_options),
|
|
LZMA_PROG_ERROR);
|
|
|
|
known_options.filters = filters_none;
|
|
assert_lzma_ret(lzma_block_header_size(&known_options),
|
|
LZMA_PROG_ERROR);
|
|
|
|
known_options.filters = filters_five;
|
|
assert_lzma_ret(lzma_block_header_size(&known_options),
|
|
LZMA_PROG_ERROR);
|
|
|
|
known_options.filters = filters_one;
|
|
assert_lzma_ret(lzma_block_header_size(&known_options), LZMA_OK);
|
|
|
|
// Some invalid value, which gets ignored.
|
|
known_options.check = (lzma_check)(99);
|
|
assert_lzma_ret(lzma_block_header_size(&known_options), LZMA_OK);
|
|
|
|
known_options.compressed_size = 5;
|
|
assert_lzma_ret(lzma_block_header_size(&known_options), LZMA_OK);
|
|
|
|
known_options.compressed_size = 0; // Cannot be zero.
|
|
assert_lzma_ret(lzma_block_header_size(&known_options),
|
|
LZMA_PROG_ERROR);
|
|
|
|
// LZMA_VLI_MAX is too big to keep the total size of the Block
|
|
// a valid VLI, but lzma_block_header_size() is not meant
|
|
// to validate it. (lzma_block_header_encode() must validate it.)
|
|
known_options.compressed_size = LZMA_VLI_MAX;
|
|
assert_lzma_ret(lzma_block_header_size(&known_options), LZMA_OK);
|
|
|
|
known_options.compressed_size = LZMA_VLI_UNKNOWN;
|
|
known_options.uncompressed_size = 0;
|
|
assert_lzma_ret(lzma_block_header_size(&known_options), LZMA_OK);
|
|
|
|
known_options.uncompressed_size = LZMA_VLI_MAX + 1;
|
|
assert_lzma_ret(lzma_block_header_size(&known_options),
|
|
LZMA_PROG_ERROR);
|
|
}
|
|
|
|
|
|
static void
|
|
test2(void)
|
|
{
|
|
known_options = (lzma_block){
|
|
.check = LZMA_CHECK_CRC32,
|
|
.compressed_size = LZMA_VLI_UNKNOWN,
|
|
.uncompressed_size = LZMA_VLI_UNKNOWN,
|
|
.filters = filters_four,
|
|
};
|
|
|
|
assert_lzma_ret(lzma_block_header_size(&known_options), LZMA_OK);
|
|
code();
|
|
|
|
known_options.compressed_size = 123456;
|
|
known_options.uncompressed_size = 234567;
|
|
assert_lzma_ret(lzma_block_header_size(&known_options), LZMA_OK);
|
|
code();
|
|
|
|
// We can make the sizes smaller while keeping the header size
|
|
// the same.
|
|
known_options.compressed_size = 12;
|
|
known_options.uncompressed_size = 23;
|
|
code();
|
|
}
|
|
|
|
|
|
static void
|
|
test3(void)
|
|
{
|
|
known_options = (lzma_block){
|
|
.check = LZMA_CHECK_CRC32,
|
|
.compressed_size = LZMA_VLI_UNKNOWN,
|
|
.uncompressed_size = LZMA_VLI_UNKNOWN,
|
|
.filters = filters_one,
|
|
};
|
|
|
|
assert_lzma_ret(lzma_block_header_size(&known_options), LZMA_OK);
|
|
known_options.header_size += 4;
|
|
assert_lzma_ret(lzma_block_header_encode(&known_options, buf),
|
|
LZMA_OK);
|
|
|
|
lzma_filter filters[LZMA_FILTERS_MAX + 1];
|
|
decoded_options.header_size = known_options.header_size;
|
|
decoded_options.check = known_options.check;
|
|
decoded_options.filters = filters;
|
|
|
|
// Wrong size
|
|
++buf[0];
|
|
assert_lzma_ret(lzma_block_header_decode(&decoded_options, NULL, buf),
|
|
LZMA_PROG_ERROR);
|
|
--buf[0];
|
|
|
|
// Wrong CRC32
|
|
buf[known_options.header_size - 1] ^= 1;
|
|
assert_lzma_ret(lzma_block_header_decode(&decoded_options, NULL, buf),
|
|
LZMA_DATA_ERROR);
|
|
buf[known_options.header_size - 1] ^= 1;
|
|
|
|
// Unsupported filter
|
|
// NOTE: This may need updating when new IDs become supported.
|
|
buf[2] ^= 0x1F;
|
|
write32le(buf + known_options.header_size - 4,
|
|
lzma_crc32(buf, known_options.header_size - 4, 0));
|
|
assert_lzma_ret(lzma_block_header_decode(&decoded_options, NULL, buf),
|
|
LZMA_OPTIONS_ERROR);
|
|
buf[2] ^= 0x1F;
|
|
|
|
// Non-nul Padding
|
|
buf[known_options.header_size - 4 - 1] ^= 1;
|
|
write32le(buf + known_options.header_size - 4,
|
|
lzma_crc32(buf, known_options.header_size - 4, 0));
|
|
assert_lzma_ret(lzma_block_header_decode(&decoded_options, NULL, buf),
|
|
LZMA_OPTIONS_ERROR);
|
|
buf[known_options.header_size - 4 - 1] ^= 1;
|
|
}
|
|
|
|
|
|
extern int
|
|
main(int argc, char **argv)
|
|
{
|
|
tuktest_start(argc, argv);
|
|
|
|
if (!lzma_filter_encoder_is_supported(LZMA_FILTER_X86)
|
|
|| !lzma_filter_decoder_is_supported(LZMA_FILTER_X86))
|
|
tuktest_early_skip("x86 BCJ encoder and/or decoder "
|
|
"is disabled");
|
|
|
|
if (lzma_lzma_preset(&opt_lzma, 1))
|
|
tuktest_error("lzma_lzma_preset() failed");
|
|
|
|
tuktest_run(test1);
|
|
tuktest_run(test2);
|
|
tuktest_run(test3);
|
|
|
|
return tuktest_end();
|
|
}
|