forked from OSchip/llvm-project
50 lines
1.7 KiB
C++
50 lines
1.7 KiB
C++
#include <cstddef>
|
|
#include <cstdint>
|
|
#include <cstdio>
|
|
#include <cstdlib>
|
|
#include <vector>
|
|
|
|
#include "gwp_asan/stack_trace_compressor.h"
|
|
|
|
constexpr size_t kBytesForLargestVarInt = (sizeof(uintptr_t) * 8) / 7 + 1;
|
|
|
|
extern "C" int LLVMFuzzerTestOneInput(const uint8_t *Data, size_t Size) {
|
|
size_t BufferSize = kBytesForLargestVarInt * Size / sizeof(uintptr_t);
|
|
std::vector<uint8_t> Buffer(BufferSize);
|
|
std::vector<uint8_t> Buffer2(BufferSize);
|
|
|
|
// Unpack the fuzz bytes.
|
|
gwp_asan::compression::unpack(Data, Size,
|
|
reinterpret_cast<uintptr_t *>(Buffer2.data()),
|
|
BufferSize / sizeof(uintptr_t));
|
|
|
|
// Pack the fuzz bytes.
|
|
size_t BytesWritten = gwp_asan::compression::pack(
|
|
reinterpret_cast<const uintptr_t *>(Data), Size / sizeof(uintptr_t),
|
|
Buffer.data(), BufferSize);
|
|
|
|
// Unpack the compressed buffer.
|
|
size_t DecodedElements = gwp_asan::compression::unpack(
|
|
Buffer.data(), BytesWritten,
|
|
reinterpret_cast<uintptr_t *>(Buffer2.data()),
|
|
BufferSize / sizeof(uintptr_t));
|
|
|
|
// Ensure that every element was encoded and decoded properly.
|
|
if (DecodedElements != Size / sizeof(uintptr_t))
|
|
abort();
|
|
|
|
// Ensure that the compression and uncompression resulted in the same trace.
|
|
const uintptr_t *FuzzPtrs = reinterpret_cast<const uintptr_t *>(Data);
|
|
const uintptr_t *DecodedPtrs =
|
|
reinterpret_cast<const uintptr_t *>(Buffer2.data());
|
|
for (size_t i = 0; i < Size / sizeof(uintptr_t); ++i) {
|
|
if (FuzzPtrs[i] != DecodedPtrs[i]) {
|
|
fprintf(stderr, "FuzzPtrs[%zu] != DecodedPtrs[%zu] (0x%zx vs. 0x%zx)", i,
|
|
i, FuzzPtrs[i], DecodedPtrs[i]);
|
|
abort();
|
|
}
|
|
}
|
|
|
|
return 0;
|
|
}
|