nodejs/deps/zlib/contrib/tests/fuzzers/compress_fuzzer.cc
Node.js GitHub Bot 0c17a3fb38 deps: update zlib to 1.3.0.1-motley-68e57e6
PR-URL: https://github.com/nodejs/node/pull/53464
Reviewed-By: Chemi Atlow <chemi@atlow.co.il>
Reviewed-By: Marco Ippolito <marcoippolito54@gmail.com>
Reviewed-By: Rafael Gonzaga <rafael.nunu@hotmail.com>
Reviewed-By: James M Snell <jasnell@gmail.com>
2024-08-12 06:12:07 +00:00

47 lines
1.8 KiB
C++

// Copyright 2024 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include <fuzzer/FuzzedDataProvider.h>
#include <vector>
#include "zlib.h"
// Fuzzer builds often have NDEBUG set, so roll our own assert macro.
#define ASSERT(cond) \
do { \
if (!(cond)) { \
fprintf(stderr, "%s:%d Assert failed: %s\n", __FILE__, __LINE__, #cond); \
exit(1); \
} \
} while (0)
extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size) {
FuzzedDataProvider fdp(data, size);
const int level = fdp.PickValueInArray({-1, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9});
const std::vector<uint8_t> src = fdp.ConsumeRemainingBytes<uint8_t>();
const unsigned long compress_bound = compressBound(src.size());
std::vector<uint8_t> compressed;
compressed.resize(compress_bound);
unsigned long compressed_size = compress_bound;
int ret = compress2(compressed.data(), &compressed_size, src.data(),
src.size(), level);
ASSERT(ret == Z_OK);
ASSERT(compressed_size <= compress_bound);
compressed.resize(compressed_size);
std::vector<uint8_t> uncompressed;
uncompressed.resize(src.size());
unsigned long uncompressed_size = uncompressed.size();
ret = uncompress(uncompressed.data(), &uncompressed_size, compressed.data(),
compressed.size());
ASSERT(ret == Z_OK);
ASSERT(uncompressed_size == src.size());
ASSERT(uncompressed == src);
return 0;
}