Project import generated by Copybara.
PiperOrigin-RevId: 224614037
Change-Id: I14e53449d4aeccb328f86828c76b5f09dea0d4b8
diff --git a/http2/tools/http2_frame_builder.cc b/http2/tools/http2_frame_builder.cc
new file mode 100644
index 0000000..1dfcdeb
--- /dev/null
+++ b/http2/tools/http2_frame_builder.cc
@@ -0,0 +1,181 @@
+// Copyright 2016 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include "net/third_party/quiche/src/http2/tools/http2_frame_builder.h"
+
+#ifdef WIN32
+#include <winsock2.h> // for htonl() functions
+#else
+#include <arpa/inet.h>
+#include <netinet/in.h> // for htonl, htons
+#endif
+
+#include "testing/gtest/include/gtest/gtest.h"
+#include "net/third_party/quiche/src/http2/platform/api/http2_string_utils.h"
+
+namespace http2 {
+namespace test {
+
+Http2FrameBuilder::Http2FrameBuilder(Http2FrameType type,
+ uint8_t flags,
+ uint32_t stream_id) {
+ AppendUInt24(0); // Frame payload length, unknown so far.
+ Append(type);
+ AppendUInt8(flags);
+ AppendUInt31(stream_id);
+}
+
+Http2FrameBuilder::Http2FrameBuilder(const Http2FrameHeader& v) {
+ Append(v);
+}
+
+void Http2FrameBuilder::Append(Http2StringPiece s) {
+ Http2StrAppend(&buffer_, s);
+}
+
+void Http2FrameBuilder::AppendBytes(const void* data, uint32_t num_bytes) {
+ Append(Http2StringPiece(static_cast<const char*>(data), num_bytes));
+}
+
+void Http2FrameBuilder::AppendZeroes(size_t num_zero_bytes) {
+ char zero = 0;
+ buffer_.append(num_zero_bytes, zero);
+}
+
+void Http2FrameBuilder::AppendUInt8(uint8_t value) {
+ AppendBytes(&value, 1);
+}
+
+void Http2FrameBuilder::AppendUInt16(uint16_t value) {
+ value = htons(value);
+ AppendBytes(&value, 2);
+}
+
+void Http2FrameBuilder::AppendUInt24(uint32_t value) {
+ // Doesn't make sense to try to append a larger value, as that doesn't
+ // simulate something an encoder could do (i.e. the other 8 bits simply aren't
+ // there to be occupied).
+ EXPECT_EQ(value, value & 0xffffff);
+ value = htonl(value);
+ AppendBytes(reinterpret_cast<char*>(&value) + 1, 3);
+}
+
+void Http2FrameBuilder::AppendUInt31(uint32_t value) {
+ // If you want to test the high-bit being set, call AppendUInt32 instead.
+ uint32_t tmp = value & StreamIdMask();
+ EXPECT_EQ(value, value & StreamIdMask())
+ << "High-bit of uint32_t should be clear.";
+ value = htonl(tmp);
+ AppendBytes(&value, 4);
+}
+
+void Http2FrameBuilder::AppendUInt32(uint32_t value) {
+ value = htonl(value);
+ AppendBytes(&value, sizeof(value));
+}
+
+void Http2FrameBuilder::Append(Http2ErrorCode error_code) {
+ AppendUInt32(static_cast<uint32_t>(error_code));
+}
+
+void Http2FrameBuilder::Append(Http2FrameType type) {
+ AppendUInt8(static_cast<uint8_t>(type));
+}
+
+void Http2FrameBuilder::Append(Http2SettingsParameter parameter) {
+ AppendUInt16(static_cast<uint16_t>(parameter));
+}
+
+void Http2FrameBuilder::Append(const Http2FrameHeader& v) {
+ AppendUInt24(v.payload_length);
+ Append(v.type);
+ AppendUInt8(v.flags);
+ AppendUInt31(v.stream_id);
+}
+
+void Http2FrameBuilder::Append(const Http2PriorityFields& v) {
+ // The EXCLUSIVE flag is the high-bit of the 32-bit stream dependency field.
+ uint32_t tmp = v.stream_dependency & StreamIdMask();
+ EXPECT_EQ(tmp, v.stream_dependency);
+ if (v.is_exclusive) {
+ tmp |= 0x80000000;
+ }
+ AppendUInt32(tmp);
+
+ // The PRIORITY frame's weight field is logically in the range [1, 256],
+ // but is encoded as a byte in the range [0, 255].
+ ASSERT_LE(1u, v.weight);
+ ASSERT_LE(v.weight, 256u);
+ AppendUInt8(v.weight - 1);
+}
+
+void Http2FrameBuilder::Append(const Http2RstStreamFields& v) {
+ Append(v.error_code);
+}
+
+void Http2FrameBuilder::Append(const Http2SettingFields& v) {
+ Append(v.parameter);
+ AppendUInt32(v.value);
+}
+
+void Http2FrameBuilder::Append(const Http2PushPromiseFields& v) {
+ AppendUInt31(v.promised_stream_id);
+}
+
+void Http2FrameBuilder::Append(const Http2PingFields& v) {
+ AppendBytes(v.opaque_bytes, sizeof Http2PingFields::opaque_bytes);
+}
+
+void Http2FrameBuilder::Append(const Http2GoAwayFields& v) {
+ AppendUInt31(v.last_stream_id);
+ Append(v.error_code);
+}
+
+void Http2FrameBuilder::Append(const Http2WindowUpdateFields& v) {
+ EXPECT_NE(0u, v.window_size_increment) << "Increment must be non-zero.";
+ AppendUInt31(v.window_size_increment);
+}
+
+void Http2FrameBuilder::Append(const Http2AltSvcFields& v) {
+ AppendUInt16(v.origin_length);
+}
+
+// Methods for changing existing buffer contents.
+
+void Http2FrameBuilder::WriteAt(Http2StringPiece s, size_t offset) {
+ ASSERT_LE(offset, buffer_.size());
+ size_t len = offset + s.size();
+ if (len > buffer_.size()) {
+ buffer_.resize(len);
+ }
+ for (size_t ndx = 0; ndx < s.size(); ++ndx) {
+ buffer_[offset + ndx] = s[ndx];
+ }
+}
+
+void Http2FrameBuilder::WriteBytesAt(const void* data,
+ uint32_t num_bytes,
+ size_t offset) {
+ WriteAt(Http2StringPiece(static_cast<const char*>(data), num_bytes), offset);
+}
+
+void Http2FrameBuilder::WriteUInt24At(uint32_t value, size_t offset) {
+ ASSERT_LT(value, static_cast<uint32_t>(1 << 24));
+ value = htonl(value);
+ WriteBytesAt(reinterpret_cast<char*>(&value) + 1, sizeof(value) - 1, offset);
+}
+
+void Http2FrameBuilder::SetPayloadLength(uint32_t payload_length) {
+ WriteUInt24At(payload_length, 0);
+}
+
+size_t Http2FrameBuilder::SetPayloadLength() {
+ EXPECT_GE(size(), Http2FrameHeader::EncodedSize());
+ uint32_t payload_length = size() - Http2FrameHeader::EncodedSize();
+ SetPayloadLength(payload_length);
+ return payload_length;
+}
+
+} // namespace test
+} // namespace http2
diff --git a/http2/tools/http2_frame_builder.h b/http2/tools/http2_frame_builder.h
new file mode 100644
index 0000000..c36cb56
--- /dev/null
+++ b/http2/tools/http2_frame_builder.h
@@ -0,0 +1,101 @@
+// Copyright 2016 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#ifndef QUICHE_HTTP2_TOOLS_HTTP2_FRAME_BUILDER_H_
+#define QUICHE_HTTP2_TOOLS_HTTP2_FRAME_BUILDER_H_
+
+// Http2FrameBuilder builds wire-format HTTP/2 frames (or fragments thereof)
+// from components.
+//
+// For now, this is only intended for use in tests, and thus has EXPECT* in the
+// code. If desired to use it in an encoder, it will need optimization work,
+// especially w.r.t memory mgmt, and the EXPECT* will need to be removed or
+// replaced with DCHECKs.
+
+#include <stddef.h> // for size_t
+
+#include <cstdint>
+
+#include "net/third_party/quiche/src/http2/http2_constants.h"
+#include "net/third_party/quiche/src/http2/http2_structures.h"
+#include "net/third_party/quiche/src/http2/platform/api/http2_string.h"
+#include "net/third_party/quiche/src/http2/platform/api/http2_string_piece.h"
+
+namespace http2 {
+namespace test {
+
+class Http2FrameBuilder {
+ public:
+ Http2FrameBuilder(Http2FrameType type, uint8_t flags, uint32_t stream_id);
+ explicit Http2FrameBuilder(const Http2FrameHeader& v);
+ Http2FrameBuilder() {}
+ ~Http2FrameBuilder() {}
+
+ size_t size() const { return buffer_.size(); }
+ const Http2String& buffer() const { return buffer_; }
+
+ //----------------------------------------------------------------------------
+ // Methods for appending to the end of the buffer.
+
+ // Append a sequence of bytes from various sources.
+ void Append(Http2StringPiece s);
+ void AppendBytes(const void* data, uint32_t num_bytes);
+
+ // Append an array of type T[N] to the string. Intended for tests with arrays
+ // initialized from literals, such as:
+ // const char kData[] = {0x12, 0x23, ...};
+ // builder.AppendBytes(kData);
+ template <typename T, size_t N>
+ void AppendBytes(T (&buf)[N]) {
+ AppendBytes(buf, N * sizeof(buf[0]));
+ }
+
+ // Support for appending padding. Does not read or write the Pad Length field.
+ void AppendZeroes(size_t num_zero_bytes);
+
+ // Append various sizes of unsigned integers.
+ void AppendUInt8(uint8_t value);
+ void AppendUInt16(uint16_t value);
+ void AppendUInt24(uint32_t value);
+ void AppendUInt31(uint32_t value);
+ void AppendUInt32(uint32_t value);
+
+ // Append various enums.
+ void Append(Http2ErrorCode error_code);
+ void Append(Http2FrameType type);
+ void Append(Http2SettingsParameter parameter);
+
+ // Append various structures.
+ void Append(const Http2FrameHeader& v);
+ void Append(const Http2PriorityFields& v);
+ void Append(const Http2RstStreamFields& v);
+ void Append(const Http2SettingFields& v);
+ void Append(const Http2PushPromiseFields& v);
+ void Append(const Http2PingFields& v);
+ void Append(const Http2GoAwayFields& v);
+ void Append(const Http2WindowUpdateFields& v);
+ void Append(const Http2AltSvcFields& v);
+
+ // Methods for changing existing buffer contents (mostly focused on updating
+ // the payload length).
+
+ void WriteAt(Http2StringPiece s, size_t offset);
+ void WriteBytesAt(const void* data, uint32_t num_bytes, size_t offset);
+ void WriteUInt24At(uint32_t value, size_t offset);
+
+ // Set the payload length to the specified size.
+ void SetPayloadLength(uint32_t payload_length);
+
+ // Sets the payload length to the size of the buffer minus the size of
+ // the frame header.
+ size_t SetPayloadLength();
+
+ private:
+ Http2String buffer_;
+};
+
+} // namespace test
+} // namespace http2
+
+#endif // QUICHE_HTTP2_TOOLS_HTTP2_FRAME_BUILDER_H_
diff --git a/http2/tools/random_decoder_test.cc b/http2/tools/random_decoder_test.cc
new file mode 100644
index 0000000..623487c
--- /dev/null
+++ b/http2/tools/random_decoder_test.cc
@@ -0,0 +1,167 @@
+// Copyright 2016 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include "net/third_party/quiche/src/http2/tools/random_decoder_test.h"
+
+#include <stddef.h>
+
+#include <algorithm>
+#include <memory>
+
+#include "base/logging.h"
+#include "testing/gtest/include/gtest/gtest.h"
+#include "net/third_party/quiche/src/http2/decoder/decode_buffer.h"
+#include "net/third_party/quiche/src/http2/decoder/decode_status.h"
+#include "net/third_party/quiche/src/http2/http2_constants.h"
+#include "net/third_party/quiche/src/http2/platform/api/http2_test_helpers.h"
+
+using ::testing::AssertionFailure;
+using ::testing::AssertionResult;
+using ::testing::AssertionSuccess;
+
+namespace http2 {
+namespace test {
+
+RandomDecoderTest::RandomDecoderTest() = default;
+
+bool RandomDecoderTest::StopDecodeOnDone() {
+ return stop_decode_on_done_;
+}
+
+DecodeStatus RandomDecoderTest::DecodeSegments(DecodeBuffer* original,
+ const SelectSize& select_size) {
+ DecodeStatus status = DecodeStatus::kDecodeInProgress;
+ bool first = true;
+ VLOG(2) << "DecodeSegments: input size=" << original->Remaining();
+ while (first || original->HasData()) {
+ size_t remaining = original->Remaining();
+ size_t size =
+ std::min(remaining, select_size(first, original->Offset(), remaining));
+ DecodeBuffer db(original->cursor(), size);
+ VLOG(2) << "Decoding " << size << " bytes of " << remaining << " remaining";
+ if (first) {
+ first = false;
+ status = StartDecoding(&db);
+ } else {
+ status = ResumeDecoding(&db);
+ }
+ // A decoder MUST consume some input (if any is available), else we could
+ // get stuck in infinite loops.
+ if (db.Offset() == 0 && db.HasData() &&
+ status != DecodeStatus::kDecodeError) {
+ ADD_FAILURE() << "Decoder didn't make any progress; db.FullSize="
+ << db.FullSize()
+ << " original.Offset=" << original->Offset();
+ return DecodeStatus::kDecodeError;
+ }
+ original->AdvanceCursor(db.Offset());
+ switch (status) {
+ case DecodeStatus::kDecodeDone:
+ if (original->Empty() || StopDecodeOnDone()) {
+ return DecodeStatus::kDecodeDone;
+ }
+ continue;
+ case DecodeStatus::kDecodeInProgress:
+ continue;
+ case DecodeStatus::kDecodeError:
+ return DecodeStatus::kDecodeError;
+ }
+ }
+ return status;
+}
+
+// Decode |original| multiple times, with different segmentations, validating
+// after each decode, returning on the first failure.
+AssertionResult RandomDecoderTest::DecodeAndValidateSeveralWays(
+ DecodeBuffer* original,
+ bool return_non_zero_on_first,
+ const Validator& validator) {
+ const uint32_t original_remaining = original->Remaining();
+ VLOG(1) << "DecodeAndValidateSeveralWays - Start, remaining = "
+ << original_remaining;
+ uint32_t first_consumed;
+ {
+ // Fast decode (no stopping unless decoder does so).
+ DecodeBuffer input(original->cursor(), original_remaining);
+ VLOG(2) << "DecodeSegmentsAndValidate with SelectRemaining";
+ VERIFY_SUCCESS(
+ DecodeSegmentsAndValidate(&input, SelectRemaining(), validator))
+ << "\nFailed with SelectRemaining; input.Offset=" << input.Offset()
+ << "; input.Remaining=" << input.Remaining();
+ first_consumed = input.Offset();
+ }
+ if (original_remaining <= 30) {
+ // Decode again, one byte at a time.
+ DecodeBuffer input(original->cursor(), original_remaining);
+ VLOG(2) << "DecodeSegmentsAndValidate with SelectOne";
+ VERIFY_SUCCESS(DecodeSegmentsAndValidate(&input, SelectOne(), validator))
+ << "\nFailed with SelectOne; input.Offset=" << input.Offset()
+ << "; input.Remaining=" << input.Remaining();
+ VERIFY_EQ(first_consumed, input.Offset()) << "\nFailed with SelectOne";
+ }
+ if (original_remaining <= 20) {
+ // Decode again, one or zero bytes at a time.
+ DecodeBuffer input(original->cursor(), original_remaining);
+ VLOG(2) << "DecodeSegmentsAndValidate with SelectZeroAndOne";
+ VERIFY_SUCCESS(DecodeSegmentsAndValidate(
+ &input, SelectZeroAndOne(return_non_zero_on_first), validator))
+ << "\nFailed with SelectZeroAndOne";
+ VERIFY_EQ(first_consumed, input.Offset())
+ << "\nFailed with SelectZeroAndOne; input.Offset=" << input.Offset()
+ << "; input.Remaining=" << input.Remaining();
+ }
+ {
+ // Decode again, with randomly selected segment sizes.
+ DecodeBuffer input(original->cursor(), original_remaining);
+ VLOG(2) << "DecodeSegmentsAndValidate with SelectRandom";
+ VERIFY_SUCCESS(DecodeSegmentsAndValidate(
+ &input, SelectRandom(return_non_zero_on_first), validator))
+ << "\nFailed with SelectRandom; input.Offset=" << input.Offset()
+ << "; input.Remaining=" << input.Remaining();
+ VERIFY_EQ(first_consumed, input.Offset()) << "\nFailed with SelectRandom";
+ }
+ VERIFY_EQ(original_remaining, original->Remaining());
+ original->AdvanceCursor(first_consumed);
+ VLOG(1) << "DecodeAndValidateSeveralWays - SUCCESS";
+ return ::testing::AssertionSuccess();
+}
+
+// static
+RandomDecoderTest::SelectSize RandomDecoderTest::SelectZeroAndOne(
+ bool return_non_zero_on_first) {
+ std::shared_ptr<bool> zero_next(new bool);
+ *zero_next = !return_non_zero_on_first;
+ return [zero_next](bool first, size_t offset, size_t remaining) -> size_t {
+ if (*zero_next) {
+ *zero_next = false;
+ return 0;
+ } else {
+ *zero_next = true;
+ return 1;
+ }
+ };
+}
+
+RandomDecoderTest::SelectSize RandomDecoderTest::SelectRandom(
+ bool return_non_zero_on_first) {
+ return [this, return_non_zero_on_first](bool first, size_t offset,
+ size_t remaining) -> size_t {
+ uint32_t r = random_.Rand32();
+ if (first && return_non_zero_on_first) {
+ CHECK_LT(0u, remaining);
+ if (remaining == 1) {
+ return 1;
+ }
+ return 1 + (r % remaining); // size in range [1, remaining).
+ }
+ return r % (remaining + 1); // size in range [0, remaining].
+ };
+}
+
+uint32_t RandomDecoderTest::RandStreamId() {
+ return random_.Rand32() & StreamIdMask();
+}
+
+} // namespace test
+} // namespace http2
diff --git a/http2/tools/random_decoder_test.h b/http2/tools/random_decoder_test.h
new file mode 100644
index 0000000..36f3196
--- /dev/null
+++ b/http2/tools/random_decoder_test.h
@@ -0,0 +1,258 @@
+// Copyright 2016 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#ifndef QUICHE_HTTP2_TOOLS_RANDOM_DECODER_TEST_H_
+#define QUICHE_HTTP2_TOOLS_RANDOM_DECODER_TEST_H_
+
+// RandomDecoderTest is a base class for tests of decoding various kinds
+// of HTTP/2 and HPACK encodings.
+
+// TODO(jamessynge): Move more methods into .cc file.
+
+#include <stddef.h>
+
+#include <cstdint>
+#include <functional>
+#include <memory>
+#include <type_traits>
+
+#include "base/logging.h"
+#include "testing/gtest/include/gtest/gtest.h"
+#include "net/third_party/quiche/src/http2/decoder/decode_buffer.h"
+#include "net/third_party/quiche/src/http2/decoder/decode_status.h"
+#include "net/third_party/quiche/src/http2/platform/api/http2_string.h"
+#include "net/third_party/quiche/src/http2/platform/api/http2_string_piece.h"
+#include "net/third_party/quiche/src/http2/platform/api/http2_test_helpers.h"
+#include "net/third_party/quiche/src/http2/test_tools/http2_random.h"
+
+namespace http2 {
+namespace test {
+
+// Some helpers.
+
+template <typename T, size_t N>
+Http2StringPiece ToStringPiece(T (&data)[N]) {
+ return Http2StringPiece(reinterpret_cast<const char*>(data), N * sizeof(T));
+}
+
+// Overwrite the enum with some random value, probably not a valid value for
+// the enum type, but which fits into its storage.
+template <typename T,
+ typename E = typename std::enable_if<std::is_enum<T>::value>::type>
+void CorruptEnum(T* out, Http2Random* rng) {
+ // Per cppreference.com, if the destination type of a static_cast is
+ // smaller than the source type (i.e. type of r and uint32 below), the
+ // resulting value is the smallest unsigned value equal to the source value
+ // modulo 2^n, where n is the number of bits used to represent the
+ // destination type unsigned U.
+ using underlying_type_T = typename std::underlying_type<T>::type;
+ using unsigned_underlying_type_T =
+ typename std::make_unsigned<underlying_type_T>::type;
+ auto r = static_cast<unsigned_underlying_type_T>(rng->Rand32());
+ *out = static_cast<T>(r);
+}
+
+// Base class for tests of the ability to decode a sequence of bytes with
+// various boundaries between the DecodeBuffers provided to the decoder.
+class RandomDecoderTest : public ::testing::Test {
+ public:
+ // SelectSize returns the size of the next DecodeBuffer to be passed to the
+ // decoder. Note that RandomDecoderTest allows that size to be zero, though
+ // some decoders can't deal with that on the first byte, hence the |first|
+ // parameter.
+ typedef std::function<size_t(bool first, size_t offset, size_t remaining)>
+ SelectSize;
+
+ // Validator returns an AssertionResult so test can do:
+ // EXPECT_THAT(DecodeAndValidate(..., validator));
+ typedef ::testing::AssertionResult AssertionResult;
+ typedef std::function<AssertionResult(const DecodeBuffer& input,
+ DecodeStatus status)>
+ Validator;
+ typedef std::function<AssertionResult()> NoArgValidator;
+
+ RandomDecoderTest();
+
+ protected:
+ // TODO(jamessynge): Modify StartDecoding, etc. to (somehow) return
+ // AssertionResult so that the VERIFY_* methods exported from
+ // gunit_helpers.h can be widely used.
+
+ // Start decoding; call allows sub-class to Reset the decoder, or deal with
+ // the first byte if that is done in a unique fashion. Might be called with
+ // a zero byte buffer.
+ virtual DecodeStatus StartDecoding(DecodeBuffer* db) = 0;
+
+ // Resume decoding of the input after a prior call to StartDecoding, and
+ // possibly many calls to ResumeDecoding.
+ virtual DecodeStatus ResumeDecoding(DecodeBuffer* db) = 0;
+
+ // Return true if a decode status of kDecodeDone indicates that
+ // decoding should stop.
+ virtual bool StopDecodeOnDone();
+
+ // Decode buffer |original| until we run out of input, or kDecodeDone is
+ // returned by the decoder AND StopDecodeOnDone() returns true. Segments
+ // (i.e. cuts up) the original DecodeBuffer into (potentially) smaller buffers
+ // by calling |select_size| to decide how large each buffer should be.
+ // We do this to test the ability to deal with arbitrary boundaries, as might
+ // happen in transport.
+ // Returns the final DecodeStatus.
+ DecodeStatus DecodeSegments(DecodeBuffer* original,
+ const SelectSize& select_size);
+
+ // Decode buffer |original| until we run out of input, or kDecodeDone is
+ // returned by the decoder AND StopDecodeOnDone() returns true. Segments
+ // (i.e. cuts up) the original DecodeBuffer into (potentially) smaller buffers
+ // by calling |select_size| to decide how large each buffer should be.
+ // We do this to test the ability to deal with arbitrary boundaries, as might
+ // happen in transport.
+ // Invokes |validator| with the final decode status and the original decode
+ // buffer, with the cursor advanced as far as has been consumed by the decoder
+ // and returns validator's result.
+ ::testing::AssertionResult DecodeSegmentsAndValidate(
+ DecodeBuffer* original,
+ const SelectSize& select_size,
+ const Validator& validator) {
+ DecodeStatus status = DecodeSegments(original, select_size);
+ VERIFY_AND_RETURN_SUCCESS(validator(*original, status));
+ }
+
+ // Returns a SelectSize function for fast decoding, i.e. passing all that
+ // is available to the decoder.
+ static SelectSize SelectRemaining() {
+ return [](bool first, size_t offset, size_t remaining) -> size_t {
+ return remaining;
+ };
+ }
+
+ // Returns a SelectSize function for decoding a single byte at a time.
+ static SelectSize SelectOne() {
+ return
+ [](bool first, size_t offset, size_t remaining) -> size_t { return 1; };
+ }
+
+ // Returns a SelectSize function for decoding a single byte at a time, where
+ // zero byte buffers are also allowed. Alternates between zero and one.
+ static SelectSize SelectZeroAndOne(bool return_non_zero_on_first);
+
+ // Returns a SelectSize function for decoding random sized segments.
+ SelectSize SelectRandom(bool return_non_zero_on_first);
+
+ // Decode |original| multiple times, with different segmentations of the
+ // decode buffer, validating after each decode, and confirming that they
+ // each decode the same amount. Returns on the first failure, else returns
+ // success.
+ AssertionResult DecodeAndValidateSeveralWays(DecodeBuffer* original,
+ bool return_non_zero_on_first,
+ const Validator& validator);
+
+ static Validator ToValidator(std::nullptr_t) {
+ return [](const DecodeBuffer& input, DecodeStatus status) {
+ return ::testing::AssertionSuccess();
+ };
+ }
+
+ static Validator ToValidator(const Validator& validator) {
+ if (validator == nullptr) {
+ return ToValidator(nullptr);
+ }
+ return validator;
+ }
+
+ static Validator ToValidator(const NoArgValidator& validator) {
+ if (validator == nullptr) {
+ return ToValidator(nullptr);
+ }
+ return [validator](const DecodeBuffer& input, DecodeStatus status) {
+ return validator();
+ };
+ }
+
+ // Wraps a validator with another validator
+ // that first checks that the DecodeStatus is kDecodeDone and
+ // that the DecodeBuffer is empty.
+ // TODO(jamessynge): Replace this overload with the next, as using this method
+ // usually means that the wrapped function doesn't need to be passed the
+ // DecodeBuffer nor the DecodeStatus.
+ static Validator ValidateDoneAndEmpty(const Validator& wrapped) {
+ return [wrapped](const DecodeBuffer& input,
+ DecodeStatus status) -> AssertionResult {
+ VERIFY_EQ(status, DecodeStatus::kDecodeDone);
+ VERIFY_EQ(0u, input.Remaining()) << "\nOffset=" << input.Offset();
+ if (wrapped) {
+ return wrapped(input, status);
+ }
+ return ::testing::AssertionSuccess();
+ };
+ }
+ static Validator ValidateDoneAndEmpty(NoArgValidator wrapped) {
+ return [wrapped](const DecodeBuffer& input,
+ DecodeStatus status) -> AssertionResult {
+ VERIFY_EQ(status, DecodeStatus::kDecodeDone);
+ VERIFY_EQ(0u, input.Remaining()) << "\nOffset=" << input.Offset();
+ if (wrapped) {
+ return wrapped();
+ }
+ return ::testing::AssertionSuccess();
+ };
+ }
+ static Validator ValidateDoneAndEmpty() {
+ NoArgValidator validator;
+ return ValidateDoneAndEmpty(validator);
+ }
+
+ // Wraps a validator with another validator
+ // that first checks that the DecodeStatus is kDecodeDone and
+ // that the DecodeBuffer has the expected offset.
+ // TODO(jamessynge): Replace this overload with the next, as using this method
+ // usually means that the wrapped function doesn't need to be passed the
+ // DecodeBuffer nor the DecodeStatus.
+ static Validator ValidateDoneAndOffset(uint32_t offset,
+ const Validator& wrapped) {
+ return [wrapped, offset](const DecodeBuffer& input,
+ DecodeStatus status) -> AssertionResult {
+ VERIFY_EQ(status, DecodeStatus::kDecodeDone);
+ VERIFY_EQ(offset, input.Offset()) << "\nRemaining=" << input.Remaining();
+ if (wrapped) {
+ return wrapped(input, status);
+ }
+ return ::testing::AssertionSuccess();
+ };
+ }
+ static Validator ValidateDoneAndOffset(uint32_t offset,
+ NoArgValidator wrapped) {
+ return [wrapped, offset](const DecodeBuffer& input,
+ DecodeStatus status) -> AssertionResult {
+ VERIFY_EQ(status, DecodeStatus::kDecodeDone);
+ VERIFY_EQ(offset, input.Offset()) << "\nRemaining=" << input.Remaining();
+ if (wrapped) {
+ return wrapped();
+ }
+ return ::testing::AssertionSuccess();
+ };
+ }
+ static Validator ValidateDoneAndOffset(uint32_t offset) {
+ NoArgValidator validator;
+ return ValidateDoneAndOffset(offset, validator);
+ }
+
+ // Expose |random_| as Http2Random so callers don't have to care about which
+ // sub-class of Http2Random is used, nor can they rely on the specific
+ // sub-class that RandomDecoderTest uses.
+ Http2Random& Random() { return random_; }
+ Http2Random* RandomPtr() { return &random_; }
+
+ uint32_t RandStreamId();
+
+ bool stop_decode_on_done_ = true;
+
+ private:
+ Http2Random random_;
+};
+
+} // namespace test
+} // namespace http2
+
+#endif // QUICHE_HTTP2_TOOLS_RANDOM_DECODER_TEST_H_
diff --git a/http2/tools/random_util.cc b/http2/tools/random_util.cc
new file mode 100644
index 0000000..82c3edd
--- /dev/null
+++ b/http2/tools/random_util.cc
@@ -0,0 +1,39 @@
+// Copyright 2016 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include "net/third_party/quiche/src/http2/tools/random_util.h"
+
+#include <cmath>
+
+namespace http2 {
+namespace test {
+
+// Here "word" means something that starts with a lower-case letter, and has
+// zero or more additional characters that are numbers or lower-case letters.
+Http2String GenerateHttp2HeaderName(size_t len, Http2Random* rng) {
+ Http2StringPiece alpha_lc = "abcdefghijklmnopqrstuvwxyz";
+ // If the name is short, just make it one word.
+ if (len < 8) {
+ return rng->RandStringWithAlphabet(len, alpha_lc);
+ }
+ // If the name is longer, ensure it starts with a word, and after that may
+ // have any character in alphanumdash_lc. 4 is arbitrary, could be as low
+ // as 1.
+ Http2StringPiece alphanumdash_lc = "abcdefghijklmnopqrstuvwxyz0123456789-";
+ return rng->RandStringWithAlphabet(4, alpha_lc) +
+ rng->RandStringWithAlphabet(len - 4, alphanumdash_lc);
+}
+
+Http2String GenerateWebSafeString(size_t len, Http2Random* rng) {
+ static const char* kWebsafe64 =
+ "0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ-_";
+ return rng->RandStringWithAlphabet(len, kWebsafe64);
+}
+
+Http2String GenerateWebSafeString(size_t lo, size_t hi, Http2Random* rng) {
+ return GenerateWebSafeString(rng->UniformInRange(lo, hi), rng);
+}
+
+} // namespace test
+} // namespace http2
diff --git a/http2/tools/random_util.h b/http2/tools/random_util.h
new file mode 100644
index 0000000..a2107b7
--- /dev/null
+++ b/http2/tools/random_util.h
@@ -0,0 +1,29 @@
+// Copyright 2016 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#ifndef QUICHE_HTTP2_TOOLS_RANDOM_UTIL_H_
+#define QUICHE_HTTP2_TOOLS_RANDOM_UTIL_H_
+
+#include <stddef.h>
+
+#include "net/third_party/quiche/src/http2/platform/api/http2_string.h"
+#include "net/third_party/quiche/src/http2/test_tools/http2_random.h"
+
+namespace http2 {
+namespace test {
+
+// Generate a string with the allowed character set for HTTP/2 / HPACK header
+// names.
+Http2String GenerateHttp2HeaderName(size_t len, Http2Random* rng);
+
+// Generate a string with the web-safe string character set of specified len.
+Http2String GenerateWebSafeString(size_t len, Http2Random* rng);
+
+// Generate a string with the web-safe string character set of length [lo, hi).
+Http2String GenerateWebSafeString(size_t lo, size_t hi, Http2Random* rng);
+
+} // namespace test
+} // namespace http2
+
+#endif // QUICHE_HTTP2_TOOLS_RANDOM_UTIL_H_
