1#include <cstddef>
2#include <cstdint>
3#include <cstdio>
4#include <cstdlib>
5#include <vector>
6
7#include "gwp_asan/stack_trace_compressor.h"
8
9constexpr size_t kBytesForLargestVarInt = (sizeof(uintptr_t) * 8) / 7 + 1;
10
11extern "C" int LLVMFuzzerTestOneInput(const uint8_t *Data, size_t Size) {
12 size_t BufferSize = kBytesForLargestVarInt * Size / sizeof(uintptr_t);
13 std::vector<uint8_t> Buffer(BufferSize);
14 std::vector<uint8_t> Buffer2(BufferSize);
15
16 // Unpack the fuzz bytes.
17 gwp_asan::compression::unpack(Packed: Data, PackedSize: Size,
18 Unpacked: reinterpret_cast<uintptr_t *>(Buffer2.data()),
19 UnpackedMaxSize: BufferSize / sizeof(uintptr_t));
20
21 // Pack the fuzz bytes.
22 size_t BytesWritten = gwp_asan::compression::pack(
23 Unpacked: reinterpret_cast<const uintptr_t *>(Data), UnpackedSize: Size / sizeof(uintptr_t),
24 Packed: Buffer.data(), PackedMaxSize: BufferSize);
25
26 // Unpack the compressed buffer.
27 size_t DecodedElements = gwp_asan::compression::unpack(
28 Packed: Buffer.data(), PackedSize: BytesWritten,
29 Unpacked: reinterpret_cast<uintptr_t *>(Buffer2.data()),
30 UnpackedMaxSize: BufferSize / sizeof(uintptr_t));
31
32 // Ensure that every element was encoded and decoded properly.
33 if (DecodedElements != Size / sizeof(uintptr_t))
34 abort();
35
36 // Ensure that the compression and uncompression resulted in the same trace.
37 const uintptr_t *FuzzPtrs = reinterpret_cast<const uintptr_t *>(Data);
38 const uintptr_t *DecodedPtrs =
39 reinterpret_cast<const uintptr_t *>(Buffer2.data());
40 for (size_t i = 0; i < Size / sizeof(uintptr_t); ++i) {
41 if (FuzzPtrs[i] != DecodedPtrs[i]) {
42 fprintf(stderr, "FuzzPtrs[%zu] != DecodedPtrs[%zu] (0x%zx vs. 0x%zx)", i,
43 i, FuzzPtrs[i], DecodedPtrs[i]);
44 abort();
45 }
46 }
47
48 return 0;
49}
50

source code of compiler-rt/tools/gwp_asan/stack_trace_compressor_fuzzer.cpp