blob: 26d3c80fb5ae618ed1c271c132f29d3154c7fc67 [file] [log] [blame]
Tom Rini10e47792018-05-06 17:58:06 -04001// SPDX-License-Identifier: GPL-2.0+
Kees Cook5d3bca82013-08-16 07:59:11 -07002/*
3 * Copyright (c) 2013, The Chromium Authors
Kees Cook5d3bca82013-08-16 07:59:11 -07004 */
5
Kees Cook5d3bca82013-08-16 07:59:11 -07006#include <common.h>
Simon Glass6d084ce2014-12-02 13:17:35 -07007#include <bootm.h>
Kees Cook5d3bca82013-08-16 07:59:11 -07008#include <command.h>
Simon Glass1a974af2019-08-01 09:46:36 -06009#include <gzip.h>
Simon Glass2dc9c342020-05-10 11:40:01 -060010#include <image.h>
Simon Glass0f2af882020-05-10 11:40:05 -060011#include <log.h>
Kees Cook5d3bca82013-08-16 07:59:11 -070012#include <malloc.h>
Joe Hershberger65b905b2015-03-22 17:08:59 -050013#include <mapmem.h>
Simon Glass6d084ce2014-12-02 13:17:35 -070014#include <asm/io.h>
Kees Cook5d3bca82013-08-16 07:59:11 -070015
Simon Glasscaf62672021-10-09 09:28:21 -060016#include <u-boot/lz4.h>
Kees Cook5d3bca82013-08-16 07:59:11 -070017#include <u-boot/zlib.h>
18#include <bzlib.h>
19
20#include <lzma/LzmaTypes.h>
21#include <lzma/LzmaDec.h>
22#include <lzma/LzmaTools.h>
23
24#include <linux/lzo.h>
Simon Glass1edaed02017-11-25 11:57:33 -070025#include <test/compression.h>
26#include <test/suites.h>
27#include <test/ut.h>
Kees Cook5d3bca82013-08-16 07:59:11 -070028
29static const char plain[] =
30 "I am a highly compressable bit of text.\n"
31 "I am a highly compressable bit of text.\n"
32 "I am a highly compressable bit of text.\n"
33 "There are many like me, but this one is mine.\n"
34 "If I were any shorter, there wouldn't be much sense in\n"
35 "compressing me in the first place. At least with lzo, anyway,\n"
36 "which appears to behave poorly in the face of short text\n"
37 "messages.\n";
38
39/* bzip2 -c /tmp/plain.txt > /tmp/plain.bz2 */
40static const char bzip2_compressed[] =
41 "\x42\x5a\x68\x39\x31\x41\x59\x26\x53\x59\xe5\x63\xdd\x09\x00\x00"
42 "\x28\x57\x80\x00\x10\x40\x85\x20\x20\x04\x00\x3f\xef\xdf\xf0\x30"
43 "\x00\xd6\xd0\x34\x91\x89\xa6\xf5\x4d\x19\x1a\x19\x0d\x02\x34\xd4"
44 "\xc9\x00\x34\x34\x00\x02\x48\x41\x35\x4f\xd4\xc6\x88\xd3\x50\x3d"
45 "\x4f\x51\x82\x4f\x88\xc3\x0d\x05\x62\x4f\x91\xa3\x52\x1b\xd0\x52"
46 "\x41\x4a\xa3\x98\xc2\x6b\xca\xa3\x82\xa5\xac\x8b\x15\x99\x68\xad"
47 "\xdf\x29\xd6\xf1\xf7\x5a\x10\xcd\x8c\x26\x61\x94\x95\xfe\x9e\x16"
48 "\x18\x28\x69\xd4\x23\x64\xcc\x2b\xe5\xe8\x5f\x00\xa4\x70\x26\x2c"
49 "\xee\xbd\x59\x6d\x6a\xec\xfc\x31\xda\x59\x0a\x14\x2a\x60\x1c\xf0"
50 "\x04\x86\x73\x9a\xc5\x5b\x87\x3f\x5b\x4c\x93\xe6\xb5\x35\x0d\xa6"
51 "\xb1\x2e\x62\x7b\xab\x67\xe7\x99\x2a\x14\x5e\x9f\x64\xcb\x96\xf4"
52 "\x0d\x65\xd4\x39\xe6\x8b\x7e\xea\x1c\x03\x69\x97\x83\x58\x91\x96"
53 "\xe1\xf0\x9d\xa4\x15\x8b\xb8\xc6\x93\xdc\x3d\xd9\x3c\x22\x55\xef"
54 "\xfb\xbb\x2a\xd3\x87\xa2\x8b\x04\xd9\x19\xf8\xe2\xfd\x4f\xdb\x1a"
55 "\x07\xc8\x60\xa3\x3f\xf8\xbb\x92\x29\xc2\x84\x87\x2b\x1e\xe8\x48";
56static const unsigned long bzip2_compressed_size = 240;
57
58/* lzma -z -c /tmp/plain.txt > /tmp/plain.lzma */
59static const char lzma_compressed[] =
60 "\x5d\x00\x00\x80\x00\xff\xff\xff\xff\xff\xff\xff\xff\x00\x24\x88"
61 "\x08\x26\xd8\x41\xff\x99\xc8\xcf\x66\x3d\x80\xac\xba\x17\xf1\xc8"
62 "\xb9\xdf\x49\x37\xb1\x68\xa0\x2a\xdd\x63\xd1\xa7\xa3\x66\xf8\x15"
63 "\xef\xa6\x67\x8a\x14\x18\x80\xcb\xc7\xb1\xcb\x84\x6a\xb2\x51\x16"
64 "\xa1\x45\xa0\xd6\x3e\x55\x44\x8a\x5c\xa0\x7c\xe5\xa8\xbd\x04\x57"
65 "\x8f\x24\xfd\xb9\x34\x50\x83\x2f\xf3\x46\x3e\xb9\xb0\x00\x1a\xf5"
66 "\xd3\x86\x7e\x8f\x77\xd1\x5d\x0e\x7c\xe1\xac\xde\xf8\x65\x1f\x4d"
67 "\xce\x7f\xa7\x3d\xaa\xcf\x26\xa7\x58\x69\x1e\x4c\xea\x68\x8a\xe5"
68 "\x89\xd1\xdc\x4d\xc7\xe0\x07\x42\xbf\x0c\x9d\x06\xd7\x51\xa2\x0b"
69 "\x7c\x83\x35\xe1\x85\xdf\xee\xfb\xa3\xee\x2f\x47\x5f\x8b\x70\x2b"
70 "\xe1\x37\xf3\x16\xf6\x27\x54\x8a\x33\x72\x49\xea\x53\x7d\x60\x0b"
71 "\x21\x90\x66\xe7\x9e\x56\x61\x5d\xd8\xdc\x59\xf0\xac\x2f\xd6\x49"
72 "\x6b\x85\x40\x08\x1f\xdf\x26\x25\x3b\x72\x44\xb0\xb8\x21\x2f\xb3"
73 "\xd7\x9b\x24\x30\x78\x26\x44\x07\xc3\x33\xd1\x4d\x03\x1b\xe1\xff"
74 "\xfd\xf5\x50\x8d\xca";
75static const unsigned long lzma_compressed_size = 229;
76
77/* lzop -c /tmp/plain.txt > /tmp/plain.lzo */
78static const char lzo_compressed[] =
79 "\x89\x4c\x5a\x4f\x00\x0d\x0a\x1a\x0a\x10\x30\x20\x60\x09\x40\x01"
80 "\x05\x03\x00\x00\x09\x00\x00\x81\xb4\x52\x09\x54\xf1\x00\x00\x00"
81 "\x00\x09\x70\x6c\x61\x69\x6e\x2e\x74\x78\x74\x65\xb1\x07\x9c\x00"
82 "\x00\x01\x5e\x00\x00\x01\x0f\xc3\xc7\x7a\xe0\x00\x16\x49\x20\x61"
83 "\x6d\x20\x61\x20\x68\x69\x67\x68\x6c\x79\x20\x63\x6f\x6d\x70\x72"
84 "\x65\x73\x73\x61\x62\x6c\x65\x20\x62\x69\x74\x20\x6f\x66\x20\x74"
85 "\x65\x78\x74\x2e\x0a\x20\x2f\x9c\x00\x00\x22\x54\x68\x65\x72\x65"
86 "\x20\x61\x72\x65\x20\x6d\x61\x6e\x79\x20\x6c\x69\x6b\x65\x20\x6d"
87 "\x65\x2c\x20\x62\x75\x74\x20\x74\x68\x69\x73\x20\x6f\x6e\x65\x20"
88 "\x69\x73\x20\x6d\x69\x6e\x65\x2e\x0a\x49\x66\x20\x49\x20\x77\x84"
89 "\x06\x0a\x6e\x79\x20\x73\x68\x6f\x72\x74\x65\x72\x2c\x20\x74\x90"
90 "\x08\x00\x08\x77\x6f\x75\x6c\x64\x6e\x27\x74\x20\x62\x65\x20\x6d"
91 "\x75\x63\x68\x20\x73\x65\x6e\x73\x65\x20\x69\x6e\x0a\xf8\x19\x02"
92 "\x69\x6e\x67\x20\x6d\x64\x02\x64\x06\x00\x5a\x20\x66\x69\x72\x73"
93 "\x74\x20\x70\x6c\x61\x63\x65\x2e\x20\x41\x74\x20\x6c\x65\x61\x73"
94 "\x74\x20\x77\x69\x74\x68\x20\x6c\x7a\x6f\x2c\x20\x61\x6e\x79\x77"
95 "\x61\x79\x2c\x0a\x77\x68\x69\x63\x68\x20\x61\x70\x70\x65\x61\x72"
96 "\x73\x20\x74\x6f\x20\x62\x65\x68\x61\x76\x65\x20\x70\x6f\x6f\x72"
97 "\x6c\x79\x20\x69\x6e\x20\x74\x68\x65\x20\x66\x61\x63\x65\x20\x6f"
98 "\x66\x20\x73\x68\x6f\x72\x74\x20\x74\x65\x78\x74\x0a\x6d\x65\x73"
99 "\x73\x61\x67\x65\x73\x2e\x0a\x11\x00\x00\x00\x00\x00\x00";
100static const unsigned long lzo_compressed_size = 334;
101
Julius Wernerf41a3ca2015-10-06 20:03:53 -0700102/* lz4 -z /tmp/plain.txt > /tmp/plain.lz4 */
103static const char lz4_compressed[] =
104 "\x04\x22\x4d\x18\x64\x70\xb9\x01\x01\x00\x00\xff\x19\x49\x20\x61"
105 "\x6d\x20\x61\x20\x68\x69\x67\x68\x6c\x79\x20\x63\x6f\x6d\x70\x72"
106 "\x65\x73\x73\x61\x62\x6c\x65\x20\x62\x69\x74\x20\x6f\x66\x20\x74"
107 "\x65\x78\x74\x2e\x0a\x28\x00\x3d\xf1\x25\x54\x68\x65\x72\x65\x20"
108 "\x61\x72\x65\x20\x6d\x61\x6e\x79\x20\x6c\x69\x6b\x65\x20\x6d\x65"
109 "\x2c\x20\x62\x75\x74\x20\x74\x68\x69\x73\x20\x6f\x6e\x65\x20\x69"
110 "\x73\x20\x6d\x69\x6e\x65\x2e\x0a\x49\x66\x20\x49\x20\x77\x32\x00"
111 "\xd1\x6e\x79\x20\x73\x68\x6f\x72\x74\x65\x72\x2c\x20\x74\x45\x00"
112 "\xf4\x0b\x77\x6f\x75\x6c\x64\x6e\x27\x74\x20\x62\x65\x20\x6d\x75"
113 "\x63\x68\x20\x73\x65\x6e\x73\x65\x20\x69\x6e\x0a\xcf\x00\x50\x69"
114 "\x6e\x67\x20\x6d\x12\x00\x00\x32\x00\xf0\x11\x20\x66\x69\x72\x73"
115 "\x74\x20\x70\x6c\x61\x63\x65\x2e\x20\x41\x74\x20\x6c\x65\x61\x73"
116 "\x74\x20\x77\x69\x74\x68\x20\x6c\x7a\x6f\x2c\x63\x00\xf5\x14\x77"
117 "\x61\x79\x2c\x0a\x77\x68\x69\x63\x68\x20\x61\x70\x70\x65\x61\x72"
118 "\x73\x20\x74\x6f\x20\x62\x65\x68\x61\x76\x65\x20\x70\x6f\x6f\x72"
119 "\x6c\x79\x4e\x00\x30\x61\x63\x65\x27\x01\x01\x95\x00\x01\x2d\x01"
120 "\xb0\x0a\x6d\x65\x73\x73\x61\x67\x65\x73\x2e\x0a\x00\x00\x00\x00"
121 "\x9d\x12\x8c\x9d";
122static const unsigned long lz4_compressed_size = 276;
123
Kees Cook5d3bca82013-08-16 07:59:11 -0700124
125#define TEST_BUFFER_SIZE 512
126
Simon Glass1edaed02017-11-25 11:57:33 -0700127typedef int (*mutate_func)(struct unit_test_state *uts, void *, unsigned long,
128 void *, unsigned long, unsigned long *);
Kees Cook5d3bca82013-08-16 07:59:11 -0700129
Simon Glass1edaed02017-11-25 11:57:33 -0700130static int compress_using_gzip(struct unit_test_state *uts,
131 void *in, unsigned long in_size,
Kees Cook5d3bca82013-08-16 07:59:11 -0700132 void *out, unsigned long out_max,
133 unsigned long *out_size)
134{
135 int ret;
136 unsigned long inout_size = out_max;
137
138 ret = gzip(out, &inout_size, in, in_size);
139 if (out_size)
140 *out_size = inout_size;
141
142 return ret;
143}
144
Simon Glass1edaed02017-11-25 11:57:33 -0700145static int uncompress_using_gzip(struct unit_test_state *uts,
146 void *in, unsigned long in_size,
Kees Cook5d3bca82013-08-16 07:59:11 -0700147 void *out, unsigned long out_max,
148 unsigned long *out_size)
149{
150 int ret;
151 unsigned long inout_size = in_size;
152
153 ret = gunzip(out, out_max, in, &inout_size);
154 if (out_size)
155 *out_size = inout_size;
156
157 return ret;
158}
159
Simon Glass1edaed02017-11-25 11:57:33 -0700160static int compress_using_bzip2(struct unit_test_state *uts,
161 void *in, unsigned long in_size,
Wolfgang Denkec7fbf52013-10-04 17:43:24 +0200162 void *out, unsigned long out_max,
163 unsigned long *out_size)
Kees Cook5d3bca82013-08-16 07:59:11 -0700164{
165 /* There is no bzip2 compression in u-boot, so fake it. */
Simon Glass1edaed02017-11-25 11:57:33 -0700166 ut_asserteq(in_size, strlen(plain));
Simon Glassa3186e62020-05-10 12:52:45 -0600167 ut_asserteq_mem(plain, in, in_size);
Kees Cook5d3bca82013-08-16 07:59:11 -0700168
169 if (bzip2_compressed_size > out_max)
170 return -1;
171
172 memcpy(out, bzip2_compressed, bzip2_compressed_size);
173 if (out_size)
174 *out_size = bzip2_compressed_size;
175
176 return 0;
177}
178
Simon Glass1edaed02017-11-25 11:57:33 -0700179static int uncompress_using_bzip2(struct unit_test_state *uts,
180 void *in, unsigned long in_size,
Kees Cook5d3bca82013-08-16 07:59:11 -0700181 void *out, unsigned long out_max,
182 unsigned long *out_size)
183{
184 int ret;
185 unsigned int inout_size = out_max;
186
187 ret = BZ2_bzBuffToBuffDecompress(out, &inout_size, in, in_size,
188 CONFIG_SYS_MALLOC_LEN < (4096 * 1024), 0);
189 if (out_size)
190 *out_size = inout_size;
191
192 return (ret != BZ_OK);
193}
194
Simon Glass1edaed02017-11-25 11:57:33 -0700195static int compress_using_lzma(struct unit_test_state *uts,
196 void *in, unsigned long in_size,
Kees Cook5d3bca82013-08-16 07:59:11 -0700197 void *out, unsigned long out_max,
198 unsigned long *out_size)
199{
200 /* There is no lzma compression in u-boot, so fake it. */
Simon Glass1edaed02017-11-25 11:57:33 -0700201 ut_asserteq(in_size, strlen(plain));
Simon Glassa3186e62020-05-10 12:52:45 -0600202 ut_asserteq_mem(plain, in, in_size);
Kees Cook5d3bca82013-08-16 07:59:11 -0700203
204 if (lzma_compressed_size > out_max)
205 return -1;
206
207 memcpy(out, lzma_compressed, lzma_compressed_size);
208 if (out_size)
209 *out_size = lzma_compressed_size;
210
211 return 0;
212}
213
Simon Glass1edaed02017-11-25 11:57:33 -0700214static int uncompress_using_lzma(struct unit_test_state *uts,
215 void *in, unsigned long in_size,
Kees Cook5d3bca82013-08-16 07:59:11 -0700216 void *out, unsigned long out_max,
217 unsigned long *out_size)
218{
219 int ret;
220 SizeT inout_size = out_max;
221
222 ret = lzmaBuffToBuffDecompress(out, &inout_size, in, in_size);
223 if (out_size)
224 *out_size = inout_size;
225
226 return (ret != SZ_OK);
227}
228
Simon Glass1edaed02017-11-25 11:57:33 -0700229static int compress_using_lzo(struct unit_test_state *uts,
230 void *in, unsigned long in_size,
Kees Cook5d3bca82013-08-16 07:59:11 -0700231 void *out, unsigned long out_max,
232 unsigned long *out_size)
233{
234 /* There is no lzo compression in u-boot, so fake it. */
Simon Glass1edaed02017-11-25 11:57:33 -0700235 ut_asserteq(in_size, strlen(plain));
Simon Glassa3186e62020-05-10 12:52:45 -0600236 ut_asserteq_mem(plain, in, in_size);
Kees Cook5d3bca82013-08-16 07:59:11 -0700237
238 if (lzo_compressed_size > out_max)
239 return -1;
240
241 memcpy(out, lzo_compressed, lzo_compressed_size);
242 if (out_size)
243 *out_size = lzo_compressed_size;
244
245 return 0;
246}
247
Simon Glass1edaed02017-11-25 11:57:33 -0700248static int uncompress_using_lzo(struct unit_test_state *uts,
249 void *in, unsigned long in_size,
Kees Cook5d3bca82013-08-16 07:59:11 -0700250 void *out, unsigned long out_max,
251 unsigned long *out_size)
252{
253 int ret;
254 size_t input_size = in_size;
255 size_t output_size = out_max;
256
257 ret = lzop_decompress(in, input_size, out, &output_size);
258 if (out_size)
259 *out_size = output_size;
260
261 return (ret != LZO_E_OK);
262}
263
Simon Glass1edaed02017-11-25 11:57:33 -0700264static int compress_using_lz4(struct unit_test_state *uts,
265 void *in, unsigned long in_size,
Julius Wernerf41a3ca2015-10-06 20:03:53 -0700266 void *out, unsigned long out_max,
267 unsigned long *out_size)
268{
269 /* There is no lz4 compression in u-boot, so fake it. */
Simon Glass1edaed02017-11-25 11:57:33 -0700270 ut_asserteq(in_size, strlen(plain));
Simon Glassa3186e62020-05-10 12:52:45 -0600271 ut_asserteq_mem(plain, in, in_size);
Julius Wernerf41a3ca2015-10-06 20:03:53 -0700272
273 if (lz4_compressed_size > out_max)
274 return -1;
275
276 memcpy(out, lz4_compressed, lz4_compressed_size);
277 if (out_size)
278 *out_size = lz4_compressed_size;
279
280 return 0;
281}
282
Simon Glass1edaed02017-11-25 11:57:33 -0700283static int uncompress_using_lz4(struct unit_test_state *uts,
284 void *in, unsigned long in_size,
Julius Wernerf41a3ca2015-10-06 20:03:53 -0700285 void *out, unsigned long out_max,
286 unsigned long *out_size)
287{
288 int ret;
289 size_t input_size = in_size;
290 size_t output_size = out_max;
291
292 ret = ulz4fn(in, input_size, out, &output_size);
293 if (out_size)
294 *out_size = output_size;
295
296 return (ret != 0);
297}
298
Kees Cook5d3bca82013-08-16 07:59:11 -0700299#define errcheck(statement) if (!(statement)) { \
300 fprintf(stderr, "\tFailed: %s\n", #statement); \
301 ret = 1; \
302 goto out; \
303}
304
Simon Glass053f8012017-11-25 11:57:31 -0700305struct buf_state {
306 ulong orig_size;
307 ulong compressed_size;
308 ulong uncompressed_size;
Kees Cook5d3bca82013-08-16 07:59:11 -0700309 void *orig_buf;
Simon Glass053f8012017-11-25 11:57:31 -0700310 void *compressed_buf;
311 void *uncompressed_buf;
312 void *compare_buf;
313};
Kees Cook5d3bca82013-08-16 07:59:11 -0700314
Simon Glass1edaed02017-11-25 11:57:33 -0700315static int run_test_internal(struct unit_test_state *uts, char *name,
Simon Glass053f8012017-11-25 11:57:31 -0700316 mutate_func compress, mutate_func uncompress,
317 struct buf_state *buf)
318{
319 int ret;
Kees Cook5d3bca82013-08-16 07:59:11 -0700320
321 /* Compress works as expected. */
Simon Glass053f8012017-11-25 11:57:31 -0700322 printf("\torig_size:%lu\n", buf->orig_size);
323 memset(buf->compressed_buf, 'A', TEST_BUFFER_SIZE);
Simon Glass1edaed02017-11-25 11:57:33 -0700324 errcheck(compress(uts, buf->orig_buf, buf->orig_size,
325 buf->compressed_buf, buf->compressed_size,
326 &buf->compressed_size) == 0);
Simon Glass053f8012017-11-25 11:57:31 -0700327 printf("\tcompressed_size:%lu\n", buf->compressed_size);
328 errcheck(buf->compressed_size > 0);
329 errcheck(buf->compressed_size < buf->orig_size);
330 errcheck(((char *)buf->compressed_buf)[buf->compressed_size - 1] !=
331 'A');
332 errcheck(((char *)buf->compressed_buf)[buf->compressed_size] == 'A');
Kees Cook5d3bca82013-08-16 07:59:11 -0700333
334 /* Uncompresses with space remaining. */
Simon Glass1edaed02017-11-25 11:57:33 -0700335 errcheck(uncompress(uts, buf->compressed_buf, buf->compressed_size,
Simon Glass053f8012017-11-25 11:57:31 -0700336 buf->uncompressed_buf, buf->uncompressed_size,
337 &buf->uncompressed_size) == 0);
338 printf("\tuncompressed_size:%lu\n", buf->uncompressed_size);
339 errcheck(buf->uncompressed_size == buf->orig_size);
340 errcheck(memcmp(buf->orig_buf, buf->uncompressed_buf,
341 buf->orig_size) == 0);
Kees Cook5d3bca82013-08-16 07:59:11 -0700342
343 /* Uncompresses with exactly the right size output buffer. */
Simon Glass053f8012017-11-25 11:57:31 -0700344 memset(buf->uncompressed_buf, 'A', TEST_BUFFER_SIZE);
Simon Glass1edaed02017-11-25 11:57:33 -0700345 errcheck(uncompress(uts, buf->compressed_buf, buf->compressed_size,
Simon Glass053f8012017-11-25 11:57:31 -0700346 buf->uncompressed_buf, buf->orig_size,
347 &buf->uncompressed_size) == 0);
348 errcheck(buf->uncompressed_size == buf->orig_size);
349 errcheck(memcmp(buf->orig_buf, buf->uncompressed_buf,
350 buf->orig_size) == 0);
351 errcheck(((char *)buf->uncompressed_buf)[buf->orig_size] == 'A');
Kees Cook5d3bca82013-08-16 07:59:11 -0700352
353 /* Make sure compression does not over-run. */
Simon Glass053f8012017-11-25 11:57:31 -0700354 memset(buf->compare_buf, 'A', TEST_BUFFER_SIZE);
Simon Glass1edaed02017-11-25 11:57:33 -0700355 ret = compress(uts, buf->orig_buf, buf->orig_size,
Simon Glass053f8012017-11-25 11:57:31 -0700356 buf->compare_buf, buf->compressed_size - 1,
Kees Cook5d3bca82013-08-16 07:59:11 -0700357 NULL);
Simon Glass053f8012017-11-25 11:57:31 -0700358 errcheck(((char *)buf->compare_buf)[buf->compressed_size] == 'A');
Kees Cook5d3bca82013-08-16 07:59:11 -0700359 errcheck(ret != 0);
360 printf("\tcompress does not overrun\n");
361
362 /* Make sure decompression does not over-run. */
Simon Glass053f8012017-11-25 11:57:31 -0700363 memset(buf->compare_buf, 'A', TEST_BUFFER_SIZE);
Simon Glass1edaed02017-11-25 11:57:33 -0700364 ret = uncompress(uts, buf->compressed_buf, buf->compressed_size,
Simon Glass053f8012017-11-25 11:57:31 -0700365 buf->compare_buf, buf->uncompressed_size - 1,
Kees Cook5d3bca82013-08-16 07:59:11 -0700366 NULL);
Simon Glass053f8012017-11-25 11:57:31 -0700367 errcheck(((char *)buf->compare_buf)[buf->uncompressed_size - 1] == 'A');
Kees Cook5d3bca82013-08-16 07:59:11 -0700368 errcheck(ret != 0);
369 printf("\tuncompress does not overrun\n");
370
371 /* Got here, everything is fine. */
372 ret = 0;
373
Simon Glass053f8012017-11-25 11:57:31 -0700374out:
375 return ret;
376}
377
Simon Glass1edaed02017-11-25 11:57:33 -0700378static int run_test(struct unit_test_state *uts, char *name,
379 mutate_func compress, mutate_func uncompress)
Simon Glass053f8012017-11-25 11:57:31 -0700380{
381 struct buf_state sbuf, *buf = &sbuf;
382 int ret;
383
384 printf(" testing %s ...\n", name);
385
386 buf->orig_buf = (void *)plain;
387 buf->orig_size = strlen(buf->orig_buf); /* Trailing NUL not included */
388 errcheck(buf->orig_size > 0);
389
390 buf->compressed_size = TEST_BUFFER_SIZE;
391 buf->uncompressed_size = TEST_BUFFER_SIZE;
392 buf->compressed_buf = malloc(buf->compressed_size);
393 errcheck(buf->compressed_buf);
394 buf->uncompressed_buf = malloc(buf->uncompressed_size);
395 errcheck(buf->uncompressed_buf);
396 buf->compare_buf = malloc(buf->uncompressed_size);
397 errcheck(buf->compare_buf);
398
Simon Glass1edaed02017-11-25 11:57:33 -0700399 ret = run_test_internal(uts, name, compress, uncompress, buf);
Kees Cook5d3bca82013-08-16 07:59:11 -0700400out:
401 printf(" %s: %s\n", name, ret == 0 ? "ok" : "FAILED");
402
Simon Glass053f8012017-11-25 11:57:31 -0700403 free(buf->compare_buf);
404 free(buf->uncompressed_buf);
405 free(buf->compressed_buf);
Kees Cook5d3bca82013-08-16 07:59:11 -0700406
407 return ret;
408}
409
Simon Glass1edaed02017-11-25 11:57:33 -0700410static int compression_test_gzip(struct unit_test_state *uts)
Kees Cook5d3bca82013-08-16 07:59:11 -0700411{
Simon Glass1edaed02017-11-25 11:57:33 -0700412 return run_test(uts, "gzip", compress_using_gzip,
413 uncompress_using_gzip);
414}
415COMPRESSION_TEST(compression_test_gzip, 0);
Kees Cook5d3bca82013-08-16 07:59:11 -0700416
Simon Glass1edaed02017-11-25 11:57:33 -0700417static int compression_test_bzip2(struct unit_test_state *uts)
418{
419 return run_test(uts, "bzip2", compress_using_bzip2,
420 uncompress_using_bzip2);
421}
422COMPRESSION_TEST(compression_test_bzip2, 0);
423
424static int compression_test_lzma(struct unit_test_state *uts)
425{
426 return run_test(uts, "lzma", compress_using_lzma,
427 uncompress_using_lzma);
428}
429COMPRESSION_TEST(compression_test_lzma, 0);
Kees Cook5d3bca82013-08-16 07:59:11 -0700430
Simon Glass1edaed02017-11-25 11:57:33 -0700431static int compression_test_lzo(struct unit_test_state *uts)
432{
433 return run_test(uts, "lzo", compress_using_lzo, uncompress_using_lzo);
434}
435COMPRESSION_TEST(compression_test_lzo, 0);
Kees Cook5d3bca82013-08-16 07:59:11 -0700436
Simon Glass1edaed02017-11-25 11:57:33 -0700437static int compression_test_lz4(struct unit_test_state *uts)
438{
439 return run_test(uts, "lz4", compress_using_lz4, uncompress_using_lz4);
Kees Cook5d3bca82013-08-16 07:59:11 -0700440}
Simon Glass1edaed02017-11-25 11:57:33 -0700441COMPRESSION_TEST(compression_test_lz4, 0);
Kees Cook5d3bca82013-08-16 07:59:11 -0700442
Simon Glass1edaed02017-11-25 11:57:33 -0700443static int compress_using_none(struct unit_test_state *uts,
444 void *in, unsigned long in_size,
Simon Glass6d084ce2014-12-02 13:17:35 -0700445 void *out, unsigned long out_max,
446 unsigned long *out_size)
447{
448 /* Here we just copy */
449 memcpy(out, in, in_size);
450 *out_size = in_size;
451
452 return 0;
453}
454
455/**
Heinrich Schuchardtd3855cf2020-03-23 18:47:47 +0100456 * run_bootm_test() - Run tests on the bootm decompression function
Simon Glass6d084ce2014-12-02 13:17:35 -0700457 *
458 * @comp_type: Compression type to test
459 * @compress: Our function to compress data
460 * @return 0 if OK, non-zero on failure
461 */
Simon Glass1edaed02017-11-25 11:57:33 -0700462static int run_bootm_test(struct unit_test_state *uts, int comp_type,
463 mutate_func compress)
Simon Glass6d084ce2014-12-02 13:17:35 -0700464{
465 ulong compress_size = 1024;
466 void *compress_buff;
467 int unc_len;
468 int err = 0;
469 const ulong image_start = 0;
470 const ulong load_addr = 0x1000;
471 ulong load_end;
472
473 printf("Testing: %s\n", genimg_get_comp_name(comp_type));
474 compress_buff = map_sysmem(image_start, 0);
475 unc_len = strlen(plain);
Simon Glass1edaed02017-11-25 11:57:33 -0700476 compress(uts, (void *)plain, unc_len, compress_buff, compress_size,
Simon Glass6d084ce2014-12-02 13:17:35 -0700477 &compress_size);
Julius Werner47ef9862019-07-24 19:37:54 -0700478 err = image_decomp(comp_type, load_addr, image_start,
479 IH_TYPE_KERNEL, map_sysmem(load_addr, 0),
480 compress_buff, compress_size, unc_len,
481 &load_end);
Simon Glass1edaed02017-11-25 11:57:33 -0700482 ut_assertok(err);
Julius Werner47ef9862019-07-24 19:37:54 -0700483 err = image_decomp(comp_type, load_addr, image_start,
484 IH_TYPE_KERNEL, map_sysmem(load_addr, 0),
485 compress_buff, compress_size, unc_len - 1,
486 &load_end);
Simon Glass1edaed02017-11-25 11:57:33 -0700487 ut_assert(err);
Simon Glass6d084ce2014-12-02 13:17:35 -0700488
489 /* We can't detect corruption when not decompressing */
490 if (comp_type == IH_COMP_NONE)
491 return 0;
492 memset(compress_buff + compress_size / 2, '\x49',
493 compress_size / 2);
Julius Werner47ef9862019-07-24 19:37:54 -0700494 err = image_decomp(comp_type, load_addr, image_start,
495 IH_TYPE_KERNEL, map_sysmem(load_addr, 0),
496 compress_buff, compress_size, 0x10000,
497 &load_end);
Simon Glass1edaed02017-11-25 11:57:33 -0700498 ut_assert(err);
Simon Glass6d084ce2014-12-02 13:17:35 -0700499
500 return 0;
501}
502
Simon Glass1edaed02017-11-25 11:57:33 -0700503static int compression_test_bootm_gzip(struct unit_test_state *uts)
Simon Glass6d084ce2014-12-02 13:17:35 -0700504{
Simon Glass1edaed02017-11-25 11:57:33 -0700505 return run_bootm_test(uts, IH_COMP_GZIP, compress_using_gzip);
506}
507COMPRESSION_TEST(compression_test_bootm_gzip, 0);
Simon Glass6d084ce2014-12-02 13:17:35 -0700508
Simon Glass1edaed02017-11-25 11:57:33 -0700509static int compression_test_bootm_bzip2(struct unit_test_state *uts)
510{
511 return run_bootm_test(uts, IH_COMP_BZIP2, compress_using_bzip2);
512}
513COMPRESSION_TEST(compression_test_bootm_bzip2, 0);
Simon Glass6d084ce2014-12-02 13:17:35 -0700514
Simon Glass1edaed02017-11-25 11:57:33 -0700515static int compression_test_bootm_lzma(struct unit_test_state *uts)
516{
517 return run_bootm_test(uts, IH_COMP_LZMA, compress_using_lzma);
518}
519COMPRESSION_TEST(compression_test_bootm_lzma, 0);
Simon Glass6d084ce2014-12-02 13:17:35 -0700520
Simon Glass1edaed02017-11-25 11:57:33 -0700521static int compression_test_bootm_lzo(struct unit_test_state *uts)
522{
523 return run_bootm_test(uts, IH_COMP_LZO, compress_using_lzo);
Simon Glass6d084ce2014-12-02 13:17:35 -0700524}
Simon Glass1edaed02017-11-25 11:57:33 -0700525COMPRESSION_TEST(compression_test_bootm_lzo, 0);
Simon Glass6d084ce2014-12-02 13:17:35 -0700526
Simon Glass1edaed02017-11-25 11:57:33 -0700527static int compression_test_bootm_lz4(struct unit_test_state *uts)
528{
529 return run_bootm_test(uts, IH_COMP_LZ4, compress_using_lz4);
530}
531COMPRESSION_TEST(compression_test_bootm_lz4, 0);
Simon Glass6d084ce2014-12-02 13:17:35 -0700532
Simon Glass1edaed02017-11-25 11:57:33 -0700533static int compression_test_bootm_none(struct unit_test_state *uts)
534{
535 return run_bootm_test(uts, IH_COMP_NONE, compress_using_none);
536}
537COMPRESSION_TEST(compression_test_bootm_none, 0);
538
Simon Glassed38aef2020-05-10 11:40:03 -0600539int do_ut_compression(struct cmd_tbl *cmdtp, int flag, int argc,
540 char *const argv[])
Simon Glass1edaed02017-11-25 11:57:33 -0700541{
Simon Glassb50211f2021-03-07 17:35:10 -0700542 struct unit_test *tests = UNIT_TEST_SUITE_START(compression_test);
543 const int n_ents = UNIT_TEST_SUITE_COUNT(compression_test);
Simon Glass1edaed02017-11-25 11:57:33 -0700544
Philippe Reynes1f99f842019-12-17 19:07:04 +0100545 return cmd_ut_category("compression", "compression_test_",
546 tests, n_ents, argc, argv);
Simon Glass1edaed02017-11-25 11:57:33 -0700547}