Project import generated by Copybara.
PiperOrigin-RevId: 224614037
Change-Id: I14e53449d4aeccb328f86828c76b5f09dea0d4b8
diff --git a/http2/test_tools/http2_random.cc b/http2/test_tools/http2_random.cc
new file mode 100644
index 0000000..fc577f4
--- /dev/null
+++ b/http2/test_tools/http2_random.cc
@@ -0,0 +1,72 @@
+#include "net/third_party/quiche/src/http2/test_tools/http2_random.h"
+
+#include "base/logging.h"
+#include "net/third_party/quiche/src/http2/platform/api/http2_string_utils.h"
+#include "third_party/boringssl/src/include/openssl/chacha.h"
+#include "third_party/boringssl/src/include/openssl/rand.h"
+
+static const uint8_t kZeroNonce[12] = {0};
+
+namespace http2 {
+namespace test {
+
+Http2Random::Http2Random() {
+ RAND_bytes(key_, sizeof(key_));
+
+ LOG(INFO) << "Initialized test RNG with the following key: " << Key();
+}
+
+Http2Random::Http2Random(Http2StringPiece key) {
+ Http2String decoded_key = Http2HexDecode(key);
+ CHECK_EQ(sizeof(key_), decoded_key.size());
+ memcpy(key_, decoded_key.data(), sizeof(key_));
+}
+
+Http2String Http2Random::Key() const {
+ return Http2HexEncode(key_, sizeof(key_));
+}
+
+void Http2Random::FillRandom(void* buffer, size_t buffer_size) {
+ memset(buffer, 0, buffer_size);
+ uint8_t* buffer_u8 = reinterpret_cast<uint8_t*>(buffer);
+ CRYPTO_chacha_20(buffer_u8, buffer_u8, buffer_size, key_, kZeroNonce,
+ counter_++);
+}
+
+Http2String Http2Random::RandString(int length) {
+ Http2String result;
+ result.resize(length);
+ FillRandom(&result[0], length);
+ return result;
+}
+
+uint64_t Http2Random::Rand64() {
+ union {
+ uint64_t number;
+ uint8_t bytes[sizeof(uint64_t)];
+ } result;
+ FillRandom(result.bytes, sizeof(result.bytes));
+ return result.number;
+}
+
+double Http2Random::RandDouble() {
+ union {
+ double f;
+ uint64_t i;
+ } value;
+ value.i = (1023ull << 52ull) | (Rand64() & 0xfffffffffffffu);
+ return value.f - 1.0;
+}
+
+Http2String Http2Random::RandStringWithAlphabet(int length,
+ Http2StringPiece alphabet) {
+ Http2String result;
+ result.resize(length);
+ for (int i = 0; i < length; i++) {
+ result[i] = alphabet[Uniform(alphabet.size())];
+ }
+ return result;
+}
+
+} // namespace test
+} // namespace http2
