Random: add getRandom(), returning std::vector<char>
This commit is contained in:
parent
d507c507e4
commit
9840dbbeff
37
random.cpp
37
random.cpp
@ -21,6 +21,7 @@ SOFTWARE.
|
|||||||
#include <sstream>
|
#include <sstream>
|
||||||
#include <exception>
|
#include <exception>
|
||||||
#include <memory>
|
#include <memory>
|
||||||
|
#include <vector>
|
||||||
#include "random.h"
|
#include "random.h"
|
||||||
#include "logger.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::string Random::getRandomHexString(unsigned int bytes)
|
||||||
{
|
{
|
||||||
std::stringstream stream;
|
std::stringstream stream;
|
||||||
|
|
||||||
auto buffer = std::make_unique<char[]>(bytes);
|
std::vector<char> random = getRandom(bytes);
|
||||||
int r = getrandom(buffer.get(), bytes, GRND_NONBLOCK);
|
for(size_t i = 0; i < random.size(); i++)
|
||||||
if(r != -1 && (size_t)r == bytes)
|
|
||||||
{
|
{
|
||||||
for(size_t i = 0; i < bytes; i++)
|
unsigned char c = (unsigned char)random[i];
|
||||||
{
|
stream << std::hex << (unsigned int)c;
|
||||||
unsigned char c = (unsigned char)buffer[i];
|
|
||||||
stream << std::hex << (unsigned int)c;
|
|
||||||
}
|
|
||||||
return stream.str();
|
|
||||||
}
|
}
|
||||||
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