Project import generated by Copybara.
PiperOrigin-RevId: 237361882
Change-Id: I109a68f44db867b20f8c6a7732b0ce657133e52a
diff --git a/quic/core/chlo_extractor_test.cc b/quic/core/chlo_extractor_test.cc
new file mode 100644
index 0000000..1e0b6f5
--- /dev/null
+++ b/quic/core/chlo_extractor_test.cc
@@ -0,0 +1,168 @@
+// Copyright (c) 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/quic/core/chlo_extractor.h"
+
+#include <memory>
+
+#include "net/third_party/quiche/src/quic/core/quic_framer.h"
+#include "net/third_party/quiche/src/quic/core/quic_utils.h"
+#include "net/third_party/quiche/src/quic/platform/api/quic_arraysize.h"
+#include "net/third_party/quiche/src/quic/platform/api/quic_ptr_util.h"
+#include "net/third_party/quiche/src/quic/platform/api/quic_string.h"
+#include "net/third_party/quiche/src/quic/platform/api/quic_test.h"
+#include "net/third_party/quiche/src/quic/test_tools/crypto_test_utils.h"
+#include "net/third_party/quiche/src/quic/test_tools/quic_test_utils.h"
+
+namespace quic {
+namespace test {
+namespace {
+
+class TestDelegate : public ChloExtractor::Delegate {
+ public:
+ TestDelegate() = default;
+ ~TestDelegate() override = default;
+
+ // ChloExtractor::Delegate implementation
+ void OnChlo(QuicTransportVersion version,
+ QuicConnectionId connection_id,
+ const CryptoHandshakeMessage& chlo) override {
+ version_ = version;
+ connection_id_ = connection_id;
+ chlo_ = chlo.DebugString();
+ }
+
+ QuicConnectionId connection_id() const { return connection_id_; }
+ QuicTransportVersion transport_version() const { return version_; }
+ const QuicString& chlo() const { return chlo_; }
+
+ private:
+ QuicConnectionId connection_id_;
+ QuicTransportVersion version_;
+ QuicString chlo_;
+};
+
+class ChloExtractorTest : public QuicTest {
+ public:
+ ChloExtractorTest() {
+ header_.destination_connection_id = TestConnectionId();
+ header_.destination_connection_id_included = CONNECTION_ID_PRESENT;
+ header_.version_flag = true;
+ header_.version = AllSupportedVersions().front();
+ header_.reset_flag = false;
+ header_.packet_number_length = PACKET_4BYTE_PACKET_NUMBER;
+ header_.packet_number = QuicPacketNumber(1);
+ if (QuicVersionHasLongHeaderLengths(header_.version.transport_version)) {
+ header_.retry_token_length_length = VARIABLE_LENGTH_INTEGER_LENGTH_1;
+ header_.length_length = VARIABLE_LENGTH_INTEGER_LENGTH_2;
+ }
+ }
+
+ void MakePacket(ParsedQuicVersion version,
+ QuicStringPiece data,
+ bool munge_offset,
+ bool munge_stream_id) {
+ QuicFrames frames;
+ size_t offset = 0;
+ if (munge_offset) {
+ offset++;
+ }
+ QuicFramer framer(SupportedVersions(header_.version), QuicTime::Zero(),
+ Perspective::IS_CLIENT, kQuicDefaultConnectionIdLength);
+ if (version.transport_version < QUIC_VERSION_47 || munge_stream_id) {
+ QuicStreamId stream_id =
+ QuicUtils::GetCryptoStreamId(version.transport_version);
+ if (munge_stream_id) {
+ stream_id++;
+ }
+ frames.push_back(
+ QuicFrame(QuicStreamFrame(stream_id, false, offset, data)));
+ } else {
+ frames.push_back(
+ QuicFrame(new QuicCryptoFrame(ENCRYPTION_NONE, offset, data)));
+ }
+ std::unique_ptr<QuicPacket> packet(
+ BuildUnsizedDataPacket(&framer, header_, frames));
+ EXPECT_TRUE(packet != nullptr);
+ size_t encrypted_length =
+ framer.EncryptPayload(ENCRYPTION_NONE, header_.packet_number, *packet,
+ buffer_, QUIC_ARRAYSIZE(buffer_));
+ ASSERT_NE(0u, encrypted_length);
+ packet_ = QuicMakeUnique<QuicEncryptedPacket>(buffer_, encrypted_length);
+ EXPECT_TRUE(packet_ != nullptr);
+ DeleteFrames(&frames);
+ }
+
+ protected:
+ TestDelegate delegate_;
+ QuicPacketHeader header_;
+ std::unique_ptr<QuicEncryptedPacket> packet_;
+ char buffer_[kMaxPacketSize];
+};
+
+TEST_F(ChloExtractorTest, FindsValidChlo) {
+ CryptoHandshakeMessage client_hello;
+ client_hello.set_tag(kCHLO);
+
+ QuicString client_hello_str(client_hello.GetSerialized().AsStringPiece());
+ // Construct a CHLO with each supported version
+ for (ParsedQuicVersion version : AllSupportedVersions()) {
+ SCOPED_TRACE(version);
+ ParsedQuicVersionVector versions(SupportedVersions(version));
+ header_.version = version;
+ if (QuicVersionHasLongHeaderLengths(version.transport_version) &&
+ header_.version_flag) {
+ header_.retry_token_length_length = VARIABLE_LENGTH_INTEGER_LENGTH_1;
+ header_.length_length = VARIABLE_LENGTH_INTEGER_LENGTH_2;
+ } else {
+ header_.retry_token_length_length = VARIABLE_LENGTH_INTEGER_LENGTH_0;
+ header_.length_length = VARIABLE_LENGTH_INTEGER_LENGTH_0;
+ }
+ MakePacket(version, client_hello_str, /*munge_offset*/ false,
+ /*munge_stream_id*/ false);
+ EXPECT_TRUE(ChloExtractor::Extract(*packet_, versions, {}, &delegate_,
+ kQuicDefaultConnectionIdLength))
+ << ParsedQuicVersionToString(version);
+ EXPECT_EQ(version.transport_version, delegate_.transport_version());
+ EXPECT_EQ(header_.destination_connection_id, delegate_.connection_id());
+ EXPECT_EQ(client_hello.DebugString(), delegate_.chlo())
+ << ParsedQuicVersionToString(version);
+ }
+}
+
+TEST_F(ChloExtractorTest, DoesNotFindValidChloOnWrongStream) {
+ CryptoHandshakeMessage client_hello;
+ client_hello.set_tag(kCHLO);
+
+ QuicString client_hello_str(client_hello.GetSerialized().AsStringPiece());
+ MakePacket(AllSupportedVersions()[0], client_hello_str,
+ /*munge_offset*/ false, /*munge_stream_id*/ true);
+ EXPECT_FALSE(ChloExtractor::Extract(*packet_, AllSupportedVersions(), {},
+ &delegate_,
+ kQuicDefaultConnectionIdLength));
+}
+
+TEST_F(ChloExtractorTest, DoesNotFindValidChloOnWrongOffset) {
+ CryptoHandshakeMessage client_hello;
+ client_hello.set_tag(kCHLO);
+
+ QuicString client_hello_str(client_hello.GetSerialized().AsStringPiece());
+ MakePacket(AllSupportedVersions()[0], client_hello_str, /*munge_offset*/ true,
+ /*munge_stream_id*/ false);
+ EXPECT_FALSE(ChloExtractor::Extract(*packet_, AllSupportedVersions(), {},
+ &delegate_,
+ kQuicDefaultConnectionIdLength));
+}
+
+TEST_F(ChloExtractorTest, DoesNotFindInvalidChlo) {
+ MakePacket(AllSupportedVersions()[0], "foo", /*munge_offset*/ false,
+ /*munge_stream_id*/ true);
+ EXPECT_FALSE(ChloExtractor::Extract(*packet_, AllSupportedVersions(), {},
+ &delegate_,
+ kQuicDefaultConnectionIdLength));
+}
+
+} // namespace
+} // namespace test
+} // namespace quic
