xz/tests/test_block_header.c

241 lines
5.8 KiB
C
Raw Normal View History

2007-12-08 22:42:33 +00:00
///////////////////////////////////////////////////////////////////////////////
//
/// \file test_block_header.c
/// \brief Tests Block Header coders
//
// Author: Lasse Collin
2007-12-08 22:42:33 +00:00
//
// This file has been put into the public domain.
// You can do whatever you want with this file.
2007-12-08 22:42:33 +00:00
//
///////////////////////////////////////////////////////////////////////////////
#include "tests.h"
static uint8_t buf[LZMA_BLOCK_HEADER_SIZE_MAX];
static lzma_block known_options;
static lzma_block decoded_options;
2007-12-08 22:42:33 +00:00
static lzma_options_lzma opt_lzma;
static lzma_filter filters_none[1] = {
2007-12-08 22:42:33 +00:00
{
.id = LZMA_VLI_UNKNOWN,
2007-12-08 22:42:33 +00:00
},
};
static lzma_filter filters_one[2] = {
2007-12-08 22:42:33 +00:00
{
.id = LZMA_FILTER_LZMA2,
.options = &opt_lzma,
2007-12-08 22:42:33 +00:00
}, {
.id = LZMA_VLI_UNKNOWN,
}
2007-12-08 22:42:33 +00:00
};
static lzma_filter filters_four[5] = {
2007-12-08 22:42:33 +00:00
{
.id = LZMA_FILTER_X86,
.options = NULL,
2007-12-08 22:42:33 +00:00
}, {
.id = LZMA_FILTER_X86,
2007-12-08 22:42:33 +00:00
.options = NULL,
}, {
.id = LZMA_FILTER_X86,
2007-12-08 22:42:33 +00:00
.options = NULL,
}, {
.id = LZMA_FILTER_LZMA2,
.options = &opt_lzma,
}, {
.id = LZMA_VLI_UNKNOWN,
}
2007-12-08 22:42:33 +00:00
};
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,
}
};
2007-12-08 22:42:33 +00:00
static void
code(void)
2007-12-08 22:42:33 +00:00
{
expect(lzma_block_header_encode(&known_options, buf) == LZMA_OK);
2007-12-08 22:42:33 +00:00
lzma_filter filters[LZMA_FILTERS_MAX + 1];
memcrap(filters, sizeof(filters));
2007-12-08 22:42:33 +00:00
memcrap(&decoded_options, sizeof(decoded_options));
decoded_options.header_size = known_options.header_size;
decoded_options.check = known_options.check;
decoded_options.filters = filters;
expect(lzma_block_header_decode(&decoded_options, NULL, buf)
== LZMA_OK);
2007-12-08 22:42:33 +00:00
expect(known_options.compressed_size
== decoded_options.compressed_size);
expect(known_options.uncompressed_size
== decoded_options.uncompressed_size);
2007-12-08 22:42:33 +00:00
for (size_t i = 0; known_options.filters[i].id
!= LZMA_VLI_UNKNOWN; ++i)
expect(known_options.filters[i].id == filters[i].id);
2007-12-08 22:42:33 +00:00
for (size_t i = 0; i < LZMA_FILTERS_MAX; ++i)
free(decoded_options.filters[i].options);
2007-12-08 22:42:33 +00:00
}
static void
test1(void)
{
known_options = (lzma_block){
.check = LZMA_CHECK_NONE,
.compressed_size = LZMA_VLI_UNKNOWN,
.uncompressed_size = LZMA_VLI_UNKNOWN,
.filters = NULL,
2007-12-08 22:42:33 +00:00
};
expect(lzma_block_header_size(&known_options) == LZMA_PROG_ERROR);
known_options.filters = filters_none;
expect(lzma_block_header_size(&known_options) == LZMA_PROG_ERROR);
2007-12-08 22:42:33 +00:00
known_options.filters = filters_five;
expect(lzma_block_header_size(&known_options) == LZMA_PROG_ERROR);
2007-12-08 22:42:33 +00:00
known_options.filters = filters_one;
expect(lzma_block_header_size(&known_options) == LZMA_OK);
2007-12-08 22:42:33 +00:00
known_options.check = 999; // Some invalid value, which gets ignored.
expect(lzma_block_header_size(&known_options) == LZMA_OK);
2007-12-08 22:42:33 +00:00
known_options.compressed_size = 5;
expect(lzma_block_header_size(&known_options) == LZMA_OK);
2007-12-08 22:42:33 +00:00
known_options.compressed_size = 0; // Cannot be zero.
expect(lzma_block_header_size(&known_options) == LZMA_PROG_ERROR);
2007-12-08 22:42:33 +00:00
// 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;
expect(lzma_block_header_size(&known_options) == LZMA_OK);
known_options.compressed_size = LZMA_VLI_UNKNOWN;
known_options.uncompressed_size = 0;
expect(lzma_block_header_size(&known_options) == LZMA_OK);
2007-12-08 22:42:33 +00:00
known_options.uncompressed_size = LZMA_VLI_MAX + 1;
expect(lzma_block_header_size(&known_options) == LZMA_PROG_ERROR);
2007-12-08 22:42:33 +00:00
}
static void
test2(void)
2007-12-08 22:42:33 +00:00
{
known_options = (lzma_block){
.check = LZMA_CHECK_CRC32,
.compressed_size = LZMA_VLI_UNKNOWN,
.uncompressed_size = LZMA_VLI_UNKNOWN,
.filters = filters_four,
2007-12-08 22:42:33 +00:00
};
expect(lzma_block_header_size(&known_options) == LZMA_OK);
code();
2007-12-08 22:42:33 +00:00
known_options.compressed_size = 123456;
known_options.uncompressed_size = 234567;
expect(lzma_block_header_size(&known_options) == LZMA_OK);
code();
2007-12-08 22:42:33 +00:00
// We can make the sizes smaller while keeping the header size
// the same.
known_options.compressed_size = 12;
known_options.uncompressed_size = 23;
code();
2007-12-08 22:42:33 +00:00
}
static void
test3(void)
2007-12-08 22:42:33 +00:00
{
known_options = (lzma_block){
.check = LZMA_CHECK_CRC32,
.compressed_size = LZMA_VLI_UNKNOWN,
.uncompressed_size = LZMA_VLI_UNKNOWN,
.filters = filters_one,
2007-12-08 22:42:33 +00:00
};
expect(lzma_block_header_size(&known_options) == LZMA_OK);
known_options.header_size += 4;
expect(lzma_block_header_encode(&known_options, buf) == LZMA_OK);
2007-12-08 22:42:33 +00:00
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;
2007-12-08 22:42:33 +00:00
// Wrong size
++buf[0];
expect(lzma_block_header_decode(&decoded_options, NULL, buf)
== LZMA_PROG_ERROR);
--buf[0];
// Wrong CRC32
buf[known_options.header_size - 1] ^= 1;
expect(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;
integer_write_32(buf + known_options.header_size - 4,
lzma_crc32(buf, known_options.header_size - 4, 0));
expect(lzma_block_header_decode(&decoded_options, NULL, buf)
== LZMA_OPTIONS_ERROR);
buf[2] ^= 0x1F;
2007-12-08 22:42:33 +00:00
// Non-nul Padding
buf[known_options.header_size - 4 - 1] ^= 1;
integer_write_32(buf + known_options.header_size - 4,
lzma_crc32(buf, known_options.header_size - 4, 0));
expect(lzma_block_header_decode(&decoded_options, NULL, buf)
== LZMA_OPTIONS_ERROR);
buf[known_options.header_size - 4 - 1] ^= 1;
2007-12-08 22:42:33 +00:00
}
int
2007-12-09 09:03:28 +00:00
main(void)
2007-12-08 22:42:33 +00:00
{
succeed(lzma_lzma_preset(&opt_lzma, 1));
2007-12-08 22:42:33 +00:00
test1();
test2();
test3();
return 0;
}