mirror of https://github.com/google/oss-fuzz.git
52 lines
1.2 KiB
C++
52 lines
1.2 KiB
C++
#ifndef FAKE_RANDOM_H_
|
|
#define FAKE_RANDOM_H_
|
|
|
|
#include <sodium.h>
|
|
#include <assert.h>
|
|
#include <stdint.h>
|
|
#include <string.h>
|
|
#include <algorithm>
|
|
|
|
static const unsigned char * SEED_DATA;
|
|
static size_t SEED_SIZE;
|
|
|
|
static const char *
|
|
fake_implementation_name(void) {
|
|
return "fake_random";
|
|
}
|
|
|
|
static void
|
|
fake_random_buffer(void * const buf, const size_t size) {
|
|
static unsigned char seed[randombytes_SEEDBYTES];
|
|
memset(seed, '0', randombytes_SEEDBYTES);
|
|
|
|
size_t boundary = std::min((size_t) randombytes_SEEDBYTES, SEED_SIZE);
|
|
memcpy(&seed, SEED_DATA, boundary);
|
|
|
|
randombytes_buf_deterministic(buf, size, seed);
|
|
}
|
|
|
|
struct randombytes_implementation fake_random = {
|
|
.implementation_name = fake_implementation_name,
|
|
.random = NULL,
|
|
.stir = NULL,
|
|
.uniform = NULL,
|
|
.buf = fake_random_buffer,
|
|
.close = NULL
|
|
};
|
|
|
|
void
|
|
setup_fake_random(const unsigned char * seed, const size_t seed_size) {
|
|
SEED_DATA = seed;
|
|
SEED_SIZE = seed_size;
|
|
|
|
int fake_random_set = randombytes_set_implementation(&fake_random);
|
|
assert(fake_random_set == 0);
|
|
|
|
assert(randombytes_implementation_name() == "fake_random");
|
|
int initialized = sodium_init();
|
|
assert(initialized >= 0);
|
|
}
|
|
|
|
#endif // FAKE_RANDOM_H_
|