Commit 8d8311ed authored by Kurt Schwehr's avatar Kurt Schwehr Committed by MathisRosenhauer

Switch fuzzer to use a vector for the buffer and use C++ casts (#12)

parent 60ca6add
#include <cstddef> #include <cstddef>
#include <cstdint> #include <cstdint>
#include <cstdlib> #include <cstdlib>
#include <vector>
#include "libaec.h" #include "libaec.h"
extern "C" int LLVMFuzzerTestOneInput(const uint8_t *Data, size_t Size) { extern "C" int LLVMFuzzerTestOneInput(const uint8_t *Data, size_t Size) {
if (Size < 2) if (Size < 2)
return 0; return 0;
unsigned char *dest = static_cast<unsigned char *>(malloc(Size * 4)); std::vector<unsigned char> dest(Size * 4);
struct aec_stream strm; aec_stream strm;
strm.bits_per_sample = (Data[0] & 0x1f) | 1; strm.bits_per_sample = (Data[0] & 0x1f) | 1;
strm.block_size = 8 << (Data[1] & 3); strm.block_size = 8 << (Data[1] & 3);
strm.rsi = 2; strm.rsi = 2;
...@@ -24,19 +25,18 @@ extern "C" int LLVMFuzzerTestOneInput(const uint8_t *Data, size_t Size) { ...@@ -24,19 +25,18 @@ extern "C" int LLVMFuzzerTestOneInput(const uint8_t *Data, size_t Size) {
strm.flags |= AEC_DATA_3BYTE; strm.flags |= AEC_DATA_3BYTE;
// Decode data // Decode data
strm.next_in = (unsigned char *)(Data + 2); strm.next_in = reinterpret_cast<const unsigned char *>(Data) + 2;
strm.avail_in = Size - 2; strm.avail_in = Size - 2;
strm.next_out = dest; strm.next_out = dest.data();
strm.avail_out = (Size - 2) * 4; strm.avail_out = dest.size();
aec_buffer_encode(&strm); aec_buffer_encode(&strm);
// Encode data // Encode data
strm.next_in = (unsigned char *)(Data + 2); strm.next_in = reinterpret_cast<const unsigned char *>(Data) + 2;
strm.avail_in = Size - 2; strm.avail_in = Size - 2;
strm.next_out = dest; strm.next_out = dest.data();
strm.avail_out = (Size - 2) * 4; strm.avail_out = dest.size();
aec_buffer_decode(&strm); aec_buffer_decode(&strm);
free(dest);
return 0; return 0;
} }
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment