75 lines
1.4 KiB
C++
75 lines
1.4 KiB
C++
#include "std_include.hpp"
|
|
#include "crypto_key.hpp"
|
|
#include "console.hpp"
|
|
|
|
#include "utils/io.hpp"
|
|
|
|
namespace crypto_key
|
|
{
|
|
namespace
|
|
{
|
|
bool load_key(utils::cryptography::ecc::key& key)
|
|
{
|
|
std::string data{};
|
|
if (!utils::io::read_file("./private.key", &data))
|
|
{
|
|
return false;
|
|
}
|
|
|
|
key.deserialize(data);
|
|
if (!key.is_valid())
|
|
{
|
|
console::warn("Loaded key is invalid!");
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
utils::cryptography::ecc::key generate_key()
|
|
{
|
|
auto key = utils::cryptography::ecc::generate_key(512);
|
|
if (!key.is_valid())
|
|
{
|
|
throw std::runtime_error("Failed to generate server key!");
|
|
}
|
|
|
|
if (!utils::io::write_file("./private.key", key.serialize()))
|
|
{
|
|
throw std::runtime_error("Failed to write server key!");
|
|
}
|
|
|
|
console::info("Generated cryptographic key: %llX", key.get_hash());
|
|
return key;
|
|
}
|
|
|
|
utils::cryptography::ecc::key load_or_generate_key()
|
|
{
|
|
utils::cryptography::ecc::key key{};
|
|
if (load_key(key))
|
|
{
|
|
console::log("Loaded cryptographic key: %llX", key.get_hash());
|
|
return key;
|
|
}
|
|
|
|
return generate_key();
|
|
}
|
|
|
|
utils::cryptography::ecc::key get_key_internal()
|
|
{
|
|
auto key = load_or_generate_key();
|
|
if (!utils::io::write_file("./public.key", key.get_public_key()))
|
|
{
|
|
console::error("Failed to write public key!");
|
|
}
|
|
return key;
|
|
}
|
|
}
|
|
|
|
const utils::cryptography::ecc::key& get()
|
|
{
|
|
static auto key = get_key_internal();
|
|
return key;
|
|
}
|
|
}
|