xz/tests/tests.h

119 lines
2.5 KiB
C
Raw Normal View History

2007-12-08 22:42:33 +00:00
///////////////////////////////////////////////////////////////////////////////
//
/// \file tests.h
/// \brief Common definitions for test applications
//
// 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
//
///////////////////////////////////////////////////////////////////////////////
#ifndef LZMA_TESTS_H
#define LZMA_TESTS_H
#include "sysdefs.h"
#include "tuklib_integer.h"
#include "lzma.h"
#include "tuktest.h"
2007-12-08 22:42:33 +00:00
#define memcrap(buf, size) memset(buf, 0xFD, size)
// TODO: Remove these three macros once all tests have been converted.
2009-09-11 06:25:09 +00:00
#define expect(test) ((test) ? 0 : (fprintf(stderr, "%s:%d: %s\n", \
__FILE__, __LINE__, #test), abort(), 0))
2007-12-08 22:42:33 +00:00
#define succeed(test) expect(!(test))
2007-12-08 22:42:33 +00:00
#define fail(test) expect(test)
// This table and macro allow getting more readable error messages when
// comparing the lzma_ret enumeration values.
static const char *enum_strings_lzma_ret[] = {
"LZMA_OK",
"LZMA_STREAM_END",
"LZMA_NO_CHECK",
"LZMA_UNSUPPORTED_CHECK",
"LZMA_GET_CHECK",
"LZMA_MEM_ERROR",
"LZMA_MEMLIMIT_ERROR",
"LZMA_FORMAT_ERROR",
"LZMA_OPTIONS_ERROR",
"LZMA_DATA_ERROR",
"LZMA_BUF_ERROR",
"LZMA_PROG_ERROR",
"LZMA_SEEK_NEEDED",
};
#define assert_lzma_ret(test_expr, ref_val) \
assert_enum_eq(test_expr, ref_val, enum_strings_lzma_ret)
2007-12-08 22:42:33 +00:00
static inline bool
coder_loop(lzma_stream *strm, uint8_t *in, size_t in_size,
uint8_t *out, size_t out_size,
lzma_ret expected_ret, lzma_action finishing_action)
{
size_t in_left = in_size;
size_t out_left = out_size > 0 ? out_size + 1 : 0;
lzma_action action = LZMA_RUN;
lzma_ret ret;
strm->next_in = NULL;
strm->avail_in = 0;
strm->next_out = NULL;
strm->avail_out = 0;
while (true) {
if (in_left > 0) {
if (--in_left == 0)
action = finishing_action;
strm->next_in = in++;
strm->avail_in = 1;
}
if (out_left > 0) {
--out_left;
strm->next_out = out++;
strm->avail_out = 1;
}
ret = lzma_code(strm, action);
if (ret != LZMA_OK)
break;
}
bool error = false;
if (ret != expected_ret)
error = true;
if (strm->total_in != in_size || strm->total_out != out_size)
error = true;
2007-12-08 22:42:33 +00:00
return error;
}
static inline bool
decoder_loop_ret(lzma_stream *strm, uint8_t *in, size_t in_size,
lzma_ret expected_ret)
{
return coder_loop(strm, in, in_size, NULL, 0, expected_ret, LZMA_RUN);
}
static inline bool
decoder_loop(lzma_stream *strm, uint8_t *in, size_t in_size)
{
return coder_loop(strm, in, in_size, NULL, 0,
LZMA_STREAM_END, LZMA_RUN);
}
#endif