1 /**
2 * Copyright (c) Meta Platforms, Inc. and affiliates.
3 * All rights reserved.
4 *
5 * This source code is licensed under both the BSD-style license (found in the
6 * LICENSE file in the root directory of this source tree) and the GPLv2 (found
7 * in the COPYING file in the root directory of this source tree).
8 * You may select, at your option, one of the above-listed licenses.
9 */
10
11 /**
12 * This fuzz target attempts to decompress the fuzzed data with the simple
13 * decompression function to ensure the decompressor never crashes.
14 */
15
16 #include "fuzz_data_producer.h"
17 #define ZSTD_STATIC_LINKING_ONLY
18
19 #include <stddef.h>
20 #include <stdlib.h>
21 #include <stdio.h>
22 #include "fuzz_helpers.h"
23 #include "zstd.h"
24
25 static ZSTD_DCtx *dctx = NULL;
26 static void* rBuf = NULL;
27 static size_t bufSize = 0;
28
LLVMFuzzerTestOneInput(const uint8_t * src,size_t size)29 int LLVMFuzzerTestOneInput(const uint8_t *src, size_t size)
30 {
31 size_t const neededBufSize = ZSTD_BLOCKSIZE_MAX;
32 FUZZ_dataProducer_t *producer = FUZZ_dataProducer_create(src, size);
33
34 /* Allocate all buffers and contexts if not already allocated */
35 if (neededBufSize > bufSize) {
36 free(rBuf);
37 rBuf = FUZZ_malloc_rand(neededBufSize, producer);
38 bufSize = neededBufSize;
39 }
40 if (!dctx) {
41 dctx = ZSTD_createDCtx();
42 FUZZ_ASSERT(dctx);
43 }
44 ZSTD_decompressBegin(dctx);
45 ZSTD_decompressBlock(dctx, rBuf, neededBufSize, src, size);
46
47 FUZZ_dataProducer_free(producer);
48
49 #ifndef STATEFUL_FUZZING
50 ZSTD_freeDCtx(dctx); dctx = NULL;
51 #endif
52 return 0;
53 }
54