52 lines
1.3 KiB
C++
52 lines
1.3 KiB
C++
#ifndef HELPERS_H
|
|
#define HELPERS_H
|
|
|
|
/* Common helpers for fuzzing */
|
|
|
|
#include <functional>
|
|
#include <string_view>
|
|
#include <cstring>
|
|
|
|
/* We use this to pad the fuzz */
|
|
static inline const uint8_t *makePadded(const uint8_t *data, size_t size) {
|
|
static int paddedLength = 512 * 1024;
|
|
static char *padded = new char[128 + paddedLength + 128];
|
|
|
|
/* Increase landing area if required */
|
|
if (paddedLength < size) {
|
|
delete [] padded;
|
|
paddedLength = size;
|
|
padded = new char [128 + paddedLength + 128];
|
|
}
|
|
|
|
memcpy(padded + 128, data, size);
|
|
|
|
return (uint8_t *) padded + 128;
|
|
}
|
|
|
|
/* Splits the fuzz data in one or many chunks */
|
|
static inline void makeChunked(const uint8_t *data, size_t size, std::function<void(const uint8_t *data, size_t size)> cb) {
|
|
/* First byte determines chunk size; 0 is all that remains, 1-255 is small chunk */
|
|
for (int i = 0; i < size; ) {
|
|
unsigned int chunkSize = data[i++];
|
|
if (!chunkSize) {
|
|
chunkSize = size - i;
|
|
} else {
|
|
chunkSize = std::min<int>(chunkSize, size - i);
|
|
}
|
|
|
|
cb(data + i, chunkSize);
|
|
i += chunkSize;
|
|
}
|
|
}
|
|
|
|
/* Reads all bytes to trigger invalid reads */
|
|
static inline void readBytes(std::string_view s) {
|
|
volatile int sum = 0;
|
|
for (int i = 0; i < s.size(); i++) {
|
|
sum += s[i];
|
|
}
|
|
}
|
|
|
|
#endif
|