tests-heatshrink.c
3.09 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
/*
* Copyright (C) 2016 Kaspar Schleiser <kaspar@schleiser.de>
*
* This file is subject to the terms and conditions of the GNU Lesser
* General Public License v2.1. See the file LICENSE in the top level
* directory for more details.
*/
/**
* @ingroup unittests
* @{
* @file
* @brief Tests for the heatshrink compression library package
* @author Kaspar Schleiser <kaspar@schleiser.de>
* @}
*/
#include <string.h>
#include <stdio.h>
#include "embUnit.h"
#include "heatshrink_encoder.h"
#include "heatshrink_decoder.h"
static const char *_data = "This is a test string TEST TEST TEST TEST!";
#define BUFSIZE (256U)
static uint8_t _buf[BUFSIZE];
static uint8_t _buf_res[BUFSIZE];
static heatshrink_encoder _encoder;
static heatshrink_decoder _decoder;
static int _comp_uncomp(const uint8_t *buf, size_t len)
{
memset(_buf, 0, sizeof(_buf));
memset(_buf_res, 0, sizeof(_buf_res));
heatshrink_encoder_reset(&_encoder);
heatshrink_decoder_reset(&_decoder);
size_t n = len;
uint8_t *inpos = (uint8_t*)buf;
uint8_t *outpos = _buf;
/* compress */
while(1) {
size_t n_sunk = 0;
if (n) {
heatshrink_encoder_sink(&_encoder, inpos, n, &n_sunk);
if (n_sunk) {
inpos += n_sunk;
n -= n_sunk;
}
size_t written = 0;
heatshrink_encoder_poll(&_encoder, outpos, (_buf + sizeof(_buf) - outpos), &written);
outpos += written;
}
else {
while (heatshrink_encoder_finish(&_encoder) == HSER_FINISH_MORE) {
size_t written = 0;
heatshrink_encoder_poll(&_encoder, outpos, (_buf + sizeof(_buf) - outpos), &written);
outpos += written;
}
break;
}
}
/* decompress */
n = (outpos - _buf);
inpos = (uint8_t*) _buf;
outpos = _buf_res;
while(1) {
size_t n_sunk = 0;
if (n) {
heatshrink_decoder_sink(&_decoder, inpos, n, &n_sunk);
if (n_sunk) {
inpos += n_sunk;
n -= n_sunk;
}
size_t written = 0;
heatshrink_decoder_poll(&_decoder, outpos, (_buf + sizeof(_buf) - outpos), &written);
outpos += written;
}
else {
while (heatshrink_decoder_finish(&_decoder) == HSDR_FINISH_MORE) {
size_t written = 0;
heatshrink_decoder_poll(&_decoder, outpos, (_buf_res + sizeof(_buf_res) - outpos), &written);
outpos += written;
}
break;
}
}
if ((outpos - _buf_res) - len) {
return -1;
}
return memcmp(buf, _buf_res, len);
}
static void test_heatshrink(void)
{
TEST_ASSERT_EQUAL_INT(_comp_uncomp((const uint8_t*)_data, strlen(_data)), 0);
}
TestRef test_heatshrink_all(void)
{
EMB_UNIT_TESTFIXTURES(fixtures) {
new_TestFixture(test_heatshrink),
};
EMB_UNIT_TESTCALLER(HeatshrinkTest, 0, 0, fixtures);
return (TestRef) & HeatshrinkTest;
}
void tests_heatshrink(void)
{
TESTS_RUN(test_heatshrink_all());
}