mirror of
https://git.tukaani.org/xz.git
synced 2026-10-31 03:14:56 +00:00
97 lines
2.1 KiB
C
97 lines
2.1 KiB
C
// SPDX-License-Identifier: 0BSD
|
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
/// \file test_alone_decoder.c
|
|
/// \brief Test lzma_alone_decoder()
|
|
//
|
|
// Author: Lasse Collin
|
|
//
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
|
|
#include "tests.h"
|
|
|
|
|
|
#define MEMLIMIT (16U << 20)
|
|
#define TOO_BIG_ALLOC (1U << 20)
|
|
|
|
|
|
#ifdef HAVE_DECODERS
|
|
static void *
|
|
my_alloc(void *opaque, size_t nmemb, size_t size)
|
|
{
|
|
(void)opaque;
|
|
(void)nmemb;
|
|
|
|
if (size >= TOO_BIG_ALLOC)
|
|
return NULL;
|
|
|
|
return malloc(size);
|
|
}
|
|
|
|
|
|
static lzma_allocator my_allocator = { &my_alloc, NULL, NULL };
|
|
|
|
|
|
static void
|
|
test_reuse_after_failure(void)
|
|
{
|
|
// 4 KiB dictionary
|
|
size_t small_dict_size;
|
|
uint8_t *small_dict = tuktest_file_from_srcdir(
|
|
"files/good-unknown_size-with_eopm.lzma",
|
|
&small_dict_size);
|
|
|
|
// 8 MiB dictionary
|
|
size_t big_dict_size;
|
|
uint8_t *big_dict = tuktest_file_from_srcdir(
|
|
"files/good-unknown_size-with_eopm.lzma",
|
|
&big_dict_size);
|
|
big_dict[2] = 0x00;
|
|
big_dict[3] = 0x80;
|
|
|
|
uint8_t out[256];
|
|
|
|
lzma_stream strm = LZMA_STREAM_INIT;
|
|
strm.allocator = &my_allocator;
|
|
|
|
assert_lzma_ret(lzma_alone_decoder(&strm, MEMLIMIT), LZMA_OK);
|
|
strm.next_in = small_dict;
|
|
strm.avail_in = small_dict_size;
|
|
strm.next_out = out;
|
|
strm.avail_out = sizeof(out);
|
|
assert_lzma_ret(lzma_code(&strm, LZMA_FINISH), LZMA_STREAM_END);
|
|
|
|
assert_lzma_ret(lzma_alone_decoder(&strm, MEMLIMIT), LZMA_OK);
|
|
strm.next_in = big_dict;
|
|
strm.avail_in = big_dict_size;
|
|
strm.next_out = out;
|
|
strm.avail_out = sizeof(out);
|
|
assert_lzma_ret(lzma_code(&strm, LZMA_FINISH), LZMA_MEM_ERROR);
|
|
|
|
assert_lzma_ret(lzma_alone_decoder(&strm, MEMLIMIT), LZMA_OK);
|
|
strm.next_in = small_dict;
|
|
strm.avail_in = small_dict_size;
|
|
strm.next_out = out;
|
|
strm.avail_out = sizeof(out);
|
|
assert_lzma_ret(lzma_code(&strm, LZMA_FINISH), LZMA_STREAM_END);
|
|
|
|
lzma_end(&strm);
|
|
}
|
|
#endif
|
|
|
|
|
|
extern int
|
|
main(int argc, char **argv)
|
|
{
|
|
tuktest_start(argc, argv);
|
|
|
|
#ifndef HAVE_DECODERS
|
|
tuktest_early_skip("Decoder support is disabled");
|
|
#else
|
|
tuktest_run(test_reuse_after_failure);
|
|
#endif
|
|
|
|
return tuktest_end();
|
|
}
|