09feb0f3d9
We currently have the situation where the |tool| and |bssl_shim| code includes scoped_types.h from crypto/test and ssl/test. That's weird and shouldn't happen. Also, our C++ consumers might quite like to have access to the scoped types. Thus this change moves some of the template code to base.h and puts it all in a |bssl| namespace to prepare for scattering these types into their respective headers. In order that all the existing test code be able to access these types, it's all moved into the same namespace. Change-Id: I3207e29474dc5fcc344ace43119df26dae04eabb Reviewed-on: https://boringssl-review.googlesource.com/8730 Reviewed-by: David Benjamin <davidben@google.com>
100 lines
2.5 KiB
C++
100 lines
2.5 KiB
C++
/* Copyright (c) 2015, Google Inc.
|
|
*
|
|
* Permission to use, copy, modify, and/or distribute this software for any
|
|
* purpose with or without fee is hereby granted, provided that the above
|
|
* copyright notice and this permission notice appear in all copies.
|
|
*
|
|
* THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
|
|
* WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
|
|
* MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY
|
|
* SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
|
|
* WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION
|
|
* OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN
|
|
* CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. */
|
|
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
#include <stdint.h>
|
|
#include <stdlib.h>
|
|
|
|
#include <openssl/rand.h>
|
|
|
|
#include "internal.h"
|
|
|
|
|
|
namespace bssl {
|
|
|
|
static const struct argument kArguments[] = {
|
|
{
|
|
"-hex", kBooleanArgument,
|
|
"Hex encoded output."
|
|
},
|
|
{
|
|
"", kOptionalArgument, "",
|
|
},
|
|
};
|
|
|
|
bool Rand(const std::vector<std::string> &args) {
|
|
bool forever = true, hex = false;
|
|
size_t len = 0;
|
|
|
|
if (!args.empty()) {
|
|
std::vector<std::string> args_copy(args);
|
|
const std::string &last_arg = args.back();
|
|
|
|
if (last_arg.size() > 0 && last_arg[0] != '-') {
|
|
char *endptr;
|
|
unsigned long long num = strtoull(last_arg.c_str(), &endptr, 10);
|
|
if (*endptr == 0) {
|
|
len = num;
|
|
forever = false;
|
|
args_copy.pop_back();
|
|
}
|
|
}
|
|
|
|
std::map<std::string, std::string> args_map;
|
|
if (!ParseKeyValueArguments(&args_map, args_copy, kArguments)) {
|
|
PrintUsage(kArguments);
|
|
return false;
|
|
}
|
|
|
|
hex = args_map.count("-hex") > 0;
|
|
}
|
|
|
|
uint8_t buf[4096];
|
|
uint8_t hex_buf[8192];
|
|
|
|
size_t done = 0;
|
|
while (forever || done < len) {
|
|
size_t todo = sizeof(buf);
|
|
if (!forever && todo > len - done) {
|
|
todo = len - done;
|
|
}
|
|
RAND_bytes(buf, todo);
|
|
if (hex) {
|
|
static const char hextable[16 + 1] = "0123456789abcdef";
|
|
for (unsigned i = 0; i < todo; i++) {
|
|
hex_buf[i*2] = hextable[buf[i] >> 4];
|
|
hex_buf[i*2 + 1] = hextable[buf[i] & 0xf];
|
|
}
|
|
if (fwrite(hex_buf, todo*2, 1, stdout) != 1) {
|
|
return false;
|
|
}
|
|
} else {
|
|
if (fwrite(buf, todo, 1, stdout) != 1) {
|
|
return false;
|
|
}
|
|
}
|
|
done += todo;
|
|
}
|
|
|
|
if (hex && fwrite("\n", 1, 1, stdout) != 1) {
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
} // namespace bssl
|