89 lines
2.2 KiB
C++
89 lines
2.2 KiB
C++
|
#ifndef C_TOXCORE_TOXCORE_CRYPTO_CORE_TEST_UTIL_H
|
||
|
|
#define C_TOXCORE_TOXCORE_CRYPTO_CORE_TEST_UTIL_H
|
||
|
|
|
||
|
|
#include <algorithm>
|
||
|
|
#include <array>
|
||
|
|
#include <iosfwd>
|
||
|
|
#include <random>
|
||
|
|
|
||
|
|
#include "crypto_core.h"
|
||
|
|
#include "test_util.hh"
|
||
|
|
|
||
|
|
struct Random_Class {
|
||
|
|
static Random_Funcs const vtable;
|
||
|
|
Random const self;
|
||
|
|
|
||
|
|
operator Random const *() const { return &self; }
|
||
|
|
|
||
|
|
Random_Class(Random_Class const &) = default;
|
||
|
|
Random_Class()
|
||
|
|
: self{&vtable, this}
|
||
|
|
{
|
||
|
|
}
|
||
|
|
|
||
|
|
virtual ~Random_Class();
|
||
|
|
virtual crypto_random_bytes_cb random_bytes = 0;
|
||
|
|
virtual crypto_random_uniform_cb random_uniform = 0;
|
||
|
|
};
|
||
|
|
|
||
|
|
/**
|
||
|
|
* A very simple, fast, and deterministic PRNG just for testing.
|
||
|
|
*
|
||
|
|
* We generally don't want to use system_random(), since it's a
|
||
|
|
* cryptographically secure PRNG and we don't need that in unit tests.
|
||
|
|
*/
|
||
|
|
class Test_Random : public Random_Class {
|
||
|
|
std::minstd_rand lcg;
|
||
|
|
|
||
|
|
void random_bytes(void *obj, uint8_t *bytes, size_t length) override;
|
||
|
|
uint32_t random_uniform(void *obj, uint32_t upper_bound) override;
|
||
|
|
};
|
||
|
|
|
||
|
|
struct PublicKey : private std::array<uint8_t, CRYPTO_PUBLIC_KEY_SIZE> {
|
||
|
|
using Base = std::array<uint8_t, CRYPTO_PUBLIC_KEY_SIZE>;
|
||
|
|
|
||
|
|
using Base::begin;
|
||
|
|
using Base::data;
|
||
|
|
using Base::end;
|
||
|
|
using Base::size;
|
||
|
|
using Base::operator[];
|
||
|
|
|
||
|
|
PublicKey() = default;
|
||
|
|
explicit PublicKey(uint8_t const (&arr)[CRYPTO_PUBLIC_KEY_SIZE])
|
||
|
|
: PublicKey(to_array(arr))
|
||
|
|
{
|
||
|
|
}
|
||
|
|
explicit PublicKey(std::array<uint8_t, CRYPTO_PUBLIC_KEY_SIZE> const &arr)
|
||
|
|
{
|
||
|
|
std::copy(arr.begin(), arr.end(), begin());
|
||
|
|
}
|
||
|
|
|
||
|
|
PublicKey(std::initializer_list<uint8_t> const &arr)
|
||
|
|
{
|
||
|
|
std::copy(arr.begin(), arr.end(), begin());
|
||
|
|
}
|
||
|
|
|
||
|
|
Base const &base() const { return *this; }
|
||
|
|
};
|
||
|
|
|
||
|
|
inline bool operator!=(PublicKey const &pk1, PublicKey const &pk2)
|
||
|
|
{
|
||
|
|
return pk1.base() != pk2.base();
|
||
|
|
}
|
||
|
|
|
||
|
|
inline bool operator==(PublicKey const &pk1, PublicKey const &pk2)
|
||
|
|
{
|
||
|
|
return pk1.base() == pk2.base();
|
||
|
|
}
|
||
|
|
|
||
|
|
inline bool operator==(PublicKey::Base const &pk1, PublicKey const &pk2)
|
||
|
|
{
|
||
|
|
return pk1 == pk2.base();
|
||
|
|
}
|
||
|
|
|
||
|
|
std::ostream &operator<<(std::ostream &out, PublicKey const &pk);
|
||
|
|
|
||
|
|
PublicKey random_pk(const Random *rng);
|
||
|
|
|
||
|
|
#endif // C_TOXCORE_TOXCORE_CRYPTO_CORE_TEST_UTIL_H
|