Random: add getRandom(), returning std::vector<char>
This commit is contained in:
父節點
d507c507e4
當前提交
9840dbbeff
37
random.cpp
37
random.cpp
@ -21,6 +21,7 @@ SOFTWARE.
|
||||
#include <sstream>
|
||||
#include <exception>
|
||||
#include <memory>
|
||||
#include <vector>
|
||||
#include "random.h"
|
||||
#include "logger.h"
|
||||
|
||||
@ -28,25 +29,33 @@ Random::Random()
|
||||
{
|
||||
}
|
||||
|
||||
std::vector<char> Random::getRandom(unsigned int bytes)
|
||||
{
|
||||
auto buffer = std::make_unique<char[]>(bytes);
|
||||
int r = getrandom(buffer.get(), bytes, GRND_NONBLOCK);
|
||||
if(r != -1 && (size_t)r == bytes)
|
||||
{
|
||||
char *ptr = buffer.get();
|
||||
return { ptr, ptr + bytes };
|
||||
}
|
||||
else
|
||||
{
|
||||
Logger::error() << "Random generator failed to get bytes: " + std::to_string(r);
|
||||
throw std::runtime_error("Random generator failed");
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
std::string Random::getRandomHexString(unsigned int bytes)
|
||||
{
|
||||
std::stringstream stream;
|
||||
|
||||
auto buffer = std::make_unique<char[]>(bytes);
|
||||
int r = getrandom(buffer.get(), bytes, GRND_NONBLOCK);
|
||||
if(r != -1 && (size_t)r == bytes)
|
||||
std::vector<char> random = getRandom(bytes);
|
||||
for(size_t i = 0; i < random.size(); i++)
|
||||
{
|
||||
for(size_t i = 0; i < bytes; i++)
|
||||
{
|
||||
unsigned char c = (unsigned char)buffer[i];
|
||||
stream << std::hex << (unsigned int)c;
|
||||
}
|
||||
return stream.str();
|
||||
unsigned char c = (unsigned char)random[i];
|
||||
stream << std::hex << (unsigned int)c;
|
||||
}
|
||||
else
|
||||
{
|
||||
return stream.str();
|
||||
|
||||
Logger::error() << "Random generator failed to get bytes: " + std::to_string(r);
|
||||
throw std::runtime_error("Random generator failed");
|
||||
}
|
||||
}
|
||||
|
Loading…
Reference in New Issue
Block a user