66 lines
1.8 KiB
C++
66 lines
1.8 KiB
C++
// Copyright 2018 Google Inc.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
#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(strcmp(randombytes_implementation_name(), "fake_random") == 0);
|
|
int initialized = sodium_init();
|
|
assert(initialized >= 0);
|
|
}
|
|
|
|
#endif // FAKE_RANDOM_H_
|