1*27162e4eSAndroid Build Coastguard Worker /**
2*27162e4eSAndroid Build Coastguard Worker * This fuzz target performs a lz4 round-trip test (compress & decompress),
3*27162e4eSAndroid Build Coastguard Worker * compares the result with the original, and calls abort() on corruption.
4*27162e4eSAndroid Build Coastguard Worker */
5*27162e4eSAndroid Build Coastguard Worker
6*27162e4eSAndroid Build Coastguard Worker #include <stddef.h>
7*27162e4eSAndroid Build Coastguard Worker #include <stdint.h>
8*27162e4eSAndroid Build Coastguard Worker #include <stdlib.h>
9*27162e4eSAndroid Build Coastguard Worker #include <string.h>
10*27162e4eSAndroid Build Coastguard Worker
11*27162e4eSAndroid Build Coastguard Worker #include "fuzz_helpers.h"
12*27162e4eSAndroid Build Coastguard Worker #include "fuzz_data_producer.h"
13*27162e4eSAndroid Build Coastguard Worker #include "lz4.h"
14*27162e4eSAndroid Build Coastguard Worker #include "lz4hc.h"
15*27162e4eSAndroid Build Coastguard Worker
LLVMFuzzerTestOneInput(const uint8_t * data,size_t size)16*27162e4eSAndroid Build Coastguard Worker int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size)
17*27162e4eSAndroid Build Coastguard Worker {
18*27162e4eSAndroid Build Coastguard Worker FUZZ_dataProducer_t *producer = FUZZ_dataProducer_create(data, size);
19*27162e4eSAndroid Build Coastguard Worker int const level = FUZZ_dataProducer_range32(producer,
20*27162e4eSAndroid Build Coastguard Worker LZ4HC_CLEVEL_MIN, LZ4HC_CLEVEL_MAX);
21*27162e4eSAndroid Build Coastguard Worker size = FUZZ_dataProducer_remainingBytes(producer);
22*27162e4eSAndroid Build Coastguard Worker
23*27162e4eSAndroid Build Coastguard Worker size_t const dstCapacity = LZ4_compressBound(size);
24*27162e4eSAndroid Build Coastguard Worker char* const dst = (char*)malloc(dstCapacity);
25*27162e4eSAndroid Build Coastguard Worker char* const rt = (char*)malloc(size);
26*27162e4eSAndroid Build Coastguard Worker
27*27162e4eSAndroid Build Coastguard Worker FUZZ_ASSERT(dst);
28*27162e4eSAndroid Build Coastguard Worker FUZZ_ASSERT(rt);
29*27162e4eSAndroid Build Coastguard Worker
30*27162e4eSAndroid Build Coastguard Worker /* Compression must succeed and round trip correctly. */
31*27162e4eSAndroid Build Coastguard Worker int const dstSize = LZ4_compress_HC((const char*)data, dst, size,
32*27162e4eSAndroid Build Coastguard Worker dstCapacity, level);
33*27162e4eSAndroid Build Coastguard Worker FUZZ_ASSERT(dstSize > 0);
34*27162e4eSAndroid Build Coastguard Worker
35*27162e4eSAndroid Build Coastguard Worker int const rtSize = LZ4_decompress_safe(dst, rt, dstSize, size);
36*27162e4eSAndroid Build Coastguard Worker FUZZ_ASSERT_MSG(rtSize == size, "Incorrect size");
37*27162e4eSAndroid Build Coastguard Worker FUZZ_ASSERT_MSG(!memcmp(data, rt, size), "Corruption!");
38*27162e4eSAndroid Build Coastguard Worker
39*27162e4eSAndroid Build Coastguard Worker free(dst);
40*27162e4eSAndroid Build Coastguard Worker free(rt);
41*27162e4eSAndroid Build Coastguard Worker FUZZ_dataProducer_free(producer);
42*27162e4eSAndroid Build Coastguard Worker
43*27162e4eSAndroid Build Coastguard Worker return 0;
44*27162e4eSAndroid Build Coastguard Worker }
45