| // Copyright (c) 2012 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. |
| |
| // Common utilities for Quic tests |
| |
| #ifndef QUICHE_QUIC_TEST_TOOLS_QUIC_TEST_UTILS_H_ |
| #define QUICHE_QUIC_TEST_TOOLS_QUIC_TEST_UTILS_H_ |
| |
| #include <cstdint> |
| #include <memory> |
| #include <string> |
| #include <utility> |
| #include <vector> |
| |
| #include "absl/strings/str_cat.h" |
| #include "absl/strings/string_view.h" |
| #include "quic/core/congestion_control/loss_detection_interface.h" |
| #include "quic/core/congestion_control/send_algorithm_interface.h" |
| #include "quic/core/crypto/transport_parameters.h" |
| #include "quic/core/http/quic_client_push_promise_index.h" |
| #include "quic/core/http/quic_server_session_base.h" |
| #include "quic/core/http/quic_spdy_session.h" |
| #include "quic/core/quic_connection.h" |
| #include "quic/core/quic_connection_id.h" |
| #include "quic/core/quic_framer.h" |
| #include "quic/core/quic_packet_writer.h" |
| #include "quic/core/quic_path_validator.h" |
| #include "quic/core/quic_sent_packet_manager.h" |
| #include "quic/core/quic_server_id.h" |
| #include "quic/core/quic_simple_buffer_allocator.h" |
| #include "quic/core/quic_types.h" |
| #include "quic/core/quic_utils.h" |
| #include "quic/platform/api/quic_mem_slice_storage.h" |
| #include "quic/platform/api/quic_test.h" |
| #include "quic/test_tools/mock_clock.h" |
| #include "quic/test_tools/mock_quic_session_visitor.h" |
| #include "quic/test_tools/mock_random.h" |
| #include "quic/test_tools/quic_framer_peer.h" |
| #include "quic/test_tools/simple_quic_framer.h" |
| |
| namespace quic { |
| |
| namespace test { |
| |
| // A generic predictable connection ID suited for testing. |
| QuicConnectionId TestConnectionId(); |
| |
| // A generic predictable connection ID suited for testing, generated from a |
| // given number, such as an index. |
| QuicConnectionId TestConnectionId(uint64_t connection_number); |
| |
| // A generic predictable connection ID suited for testing, generated from a |
| // given number, such as an index. Guaranteed to be 9 bytes long. |
| QuicConnectionId TestConnectionIdNineBytesLong(uint64_t connection_number); |
| |
| // Extracts the connection number passed to TestConnectionId(). |
| uint64_t TestConnectionIdToUInt64(QuicConnectionId connection_id); |
| |
| enum : uint16_t { kTestPort = 12345 }; |
| enum : uint32_t { |
| kMaxDatagramFrameSizeForTest = 1333, |
| kMaxPacketSizeForTest = 9001, |
| kInitialStreamFlowControlWindowForTest = 1024 * 1024, // 1 MB |
| kInitialSessionFlowControlWindowForTest = 1536 * 1024, // 1.5 MB |
| }; |
| |
| enum : uint64_t { |
| kAckDelayExponentForTest = 10, |
| kMaxAckDelayForTest = 51, // ms |
| kActiveConnectionIdLimitForTest = 52, |
| kMinAckDelayUsForTest = 1000 |
| }; |
| |
| // Create an arbitrary stateless reset token, same across multiple calls. |
| std::vector<uint8_t> CreateStatelessResetTokenForTest(); |
| |
| // A hostname useful for testing, returns "test.example.org". |
| std::string TestHostname(); |
| |
| // A server ID useful for testing, returns test.example.org:12345. |
| QuicServerId TestServerId(); |
| |
| // Returns the test peer IP address. |
| QuicIpAddress TestPeerIPAddress(); |
| |
| // Upper limit on versions we support. |
| ParsedQuicVersion QuicVersionMax(); |
| |
| // Lower limit on versions we support. |
| ParsedQuicVersion QuicVersionMin(); |
| |
| // Disables all flags that enable QUIC versions that use TLS. |
| // This is only meant as a temporary measure to prevent some broken tests |
| // from running with TLS. |
| void DisableQuicVersionsWithTls(); |
| |
| // Create an encrypted packet for testing. |
| // If versions == nullptr, uses &AllSupportedVersions(). |
| // Note that the packet is encrypted with NullEncrypter, so to decrypt the |
| // constructed packet, the framer must be set to use NullDecrypter. |
| QuicEncryptedPacket* ConstructEncryptedPacket( |
| QuicConnectionId destination_connection_id, |
| QuicConnectionId source_connection_id, |
| bool version_flag, |
| bool reset_flag, |
| uint64_t packet_number, |
| const std::string& data, |
| bool full_padding, |
| QuicConnectionIdIncluded destination_connection_id_included, |
| QuicConnectionIdIncluded source_connection_id_included, |
| QuicPacketNumberLength packet_number_length, |
| ParsedQuicVersionVector* versions, |
| Perspective perspective); |
| |
| QuicEncryptedPacket* ConstructEncryptedPacket( |
| QuicConnectionId destination_connection_id, |
| QuicConnectionId source_connection_id, |
| bool version_flag, |
| bool reset_flag, |
| uint64_t packet_number, |
| const std::string& data, |
| bool full_padding, |
| QuicConnectionIdIncluded destination_connection_id_included, |
| QuicConnectionIdIncluded source_connection_id_included, |
| QuicPacketNumberLength packet_number_length, |
| ParsedQuicVersionVector* versions); |
| |
| // Create an encrypted packet for testing. |
| // If versions == nullptr, uses &AllSupportedVersions(). |
| // Note that the packet is encrypted with NullEncrypter, so to decrypt the |
| // constructed packet, the framer must be set to use NullDecrypter. |
| QuicEncryptedPacket* ConstructEncryptedPacket( |
| QuicConnectionId destination_connection_id, |
| QuicConnectionId source_connection_id, |
| bool version_flag, |
| bool reset_flag, |
| uint64_t packet_number, |
| const std::string& data, |
| QuicConnectionIdIncluded destination_connection_id_included, |
| QuicConnectionIdIncluded source_connection_id_included, |
| QuicPacketNumberLength packet_number_length, |
| ParsedQuicVersionVector* versions); |
| |
| // This form assumes |versions| == nullptr. |
| QuicEncryptedPacket* ConstructEncryptedPacket( |
| QuicConnectionId destination_connection_id, |
| QuicConnectionId source_connection_id, |
| bool version_flag, |
| bool reset_flag, |
| uint64_t packet_number, |
| const std::string& data, |
| QuicConnectionIdIncluded destination_connection_id_included, |
| QuicConnectionIdIncluded source_connection_id_included, |
| QuicPacketNumberLength packet_number_length); |
| |
| // This form assumes |connection_id_length| == PACKET_8BYTE_CONNECTION_ID, |
| // |packet_number_length| == PACKET_4BYTE_PACKET_NUMBER and |
| // |versions| == nullptr. |
| QuicEncryptedPacket* ConstructEncryptedPacket( |
| QuicConnectionId destination_connection_id, |
| QuicConnectionId source_connection_id, |
| bool version_flag, |
| bool reset_flag, |
| uint64_t packet_number, |
| const std::string& data); |
| |
| // Creates a client-to-server ZERO-RTT packet that will fail to decrypt. |
| std::unique_ptr<QuicEncryptedPacket> GetUndecryptableEarlyPacket( |
| const ParsedQuicVersion& version, |
| const QuicConnectionId& server_connection_id); |
| |
| // Constructs a received packet for testing. The caller must take ownership |
| // of the returned pointer. |
| QuicReceivedPacket* ConstructReceivedPacket( |
| const QuicEncryptedPacket& encrypted_packet, |
| QuicTime receipt_time); |
| |
| // Create an encrypted packet for testing whose data portion erroneous. |
| // The specific way the data portion is erroneous is not specified, but |
| // it is an error that QuicFramer detects. |
| // Note that the packet is encrypted with NullEncrypter, so to decrypt the |
| // constructed packet, the framer must be set to use NullDecrypter. |
| QuicEncryptedPacket* ConstructMisFramedEncryptedPacket( |
| QuicConnectionId destination_connection_id, |
| QuicConnectionId source_connection_id, |
| bool version_flag, |
| bool reset_flag, |
| uint64_t packet_number, |
| const std::string& data, |
| QuicConnectionIdIncluded destination_connection_id_included, |
| QuicConnectionIdIncluded source_connection_id_included, |
| QuicPacketNumberLength packet_number_length, |
| ParsedQuicVersion version, |
| Perspective perspective); |
| |
| // Returns QuicConfig set to default values. |
| QuicConfig DefaultQuicConfig(); |
| |
| ParsedQuicVersionVector SupportedVersions(ParsedQuicVersion version); |
| |
| struct QuicAckBlock { |
| QuicPacketNumber start; // Included |
| QuicPacketNumber limit; // Excluded |
| }; |
| |
| // Testing convenience method to construct a QuicAckFrame with arbitrary ack |
| // blocks. Each block is given by a (closed-open) range of packet numbers. e.g.: |
| // InitAckFrame({{1, 10}}) |
| // => 1 ack block acking packet numbers 1 to 9. |
| // |
| // InitAckFrame({{1, 2}, {3, 4}}) |
| // => 2 ack blocks acking packet 1 and 3. Packet 2 is missing. |
| QuicAckFrame InitAckFrame(const std::vector<QuicAckBlock>& ack_blocks); |
| |
| // Testing convenience method to construct a QuicAckFrame with 1 ack block which |
| // covers packet number range [1, |largest_acked| + 1). |
| // Equivalent to InitAckFrame({{1, largest_acked + 1}}) |
| QuicAckFrame InitAckFrame(uint64_t largest_acked); |
| QuicAckFrame InitAckFrame(QuicPacketNumber largest_acked); |
| |
| // Testing convenience method to construct a QuicAckFrame with |num_ack_blocks| |
| // ack blocks of width 1 packet, starting from |least_unacked| + 2. |
| QuicAckFrame MakeAckFrameWithAckBlocks(size_t num_ack_blocks, |
| uint64_t least_unacked); |
| |
| // Testing convenice method to construct a QuicAckFrame with |largest_acked|, |
| // ack blocks of width 1 packet and |gap_size|. |
| QuicAckFrame MakeAckFrameWithGaps(uint64_t gap_size, |
| size_t max_num_gaps, |
| uint64_t largest_acked); |
| |
| // Returns the encryption level that corresponds to the header type in |
| // |header|. If the header is for GOOGLE_QUIC_PACKET instead of an |
| // IETF-invariants packet, this function returns ENCRYPTION_INITIAL. |
| EncryptionLevel HeaderToEncryptionLevel(const QuicPacketHeader& header); |
| |
| // Returns a QuicPacket that is owned by the caller, and |
| // is populated with the fields in |header| and |frames|, or is nullptr if the |
| // packet could not be created. |
| std::unique_ptr<QuicPacket> BuildUnsizedDataPacket( |
| QuicFramer* framer, |
| const QuicPacketHeader& header, |
| const QuicFrames& frames); |
| // Returns a QuicPacket that is owned by the caller, and of size |packet_size|. |
| std::unique_ptr<QuicPacket> BuildUnsizedDataPacket( |
| QuicFramer* framer, |
| const QuicPacketHeader& header, |
| const QuicFrames& frames, |
| size_t packet_size); |
| |
| // Compute SHA-1 hash of the supplied std::string. |
| std::string Sha1Hash(absl::string_view data); |
| |
| // Delete |frame| and return true. |
| bool ClearControlFrame(const QuicFrame& frame); |
| bool ClearControlFrameWithTransmissionType(const QuicFrame& frame, |
| TransmissionType type); |
| |
| // Simple random number generator used to compute random numbers suitable |
| // for pseudo-randomly dropping packets in tests. |
| class SimpleRandom : public QuicRandom { |
| public: |
| SimpleRandom() { set_seed(0); } |
| SimpleRandom(const SimpleRandom&) = delete; |
| SimpleRandom& operator=(const SimpleRandom&) = delete; |
| ~SimpleRandom() override {} |
| |
| // Returns a random number in the range [0, kuint64max]. |
| uint64_t RandUint64() override; |
| |
| void RandBytes(void* data, size_t len) override; |
| |
| void set_seed(uint64_t seed); |
| |
| private: |
| uint8_t buffer_[4096]; |
| size_t buffer_offset_; |
| uint8_t key_[32]; |
| |
| void FillBuffer(); |
| }; |
| |
| class MockFramerVisitor : public QuicFramerVisitorInterface { |
| public: |
| MockFramerVisitor(); |
| MockFramerVisitor(const MockFramerVisitor&) = delete; |
| MockFramerVisitor& operator=(const MockFramerVisitor&) = delete; |
| ~MockFramerVisitor() override; |
| |
| MOCK_METHOD(void, OnError, (QuicFramer*), (override)); |
| // The constructor sets this up to return false by default. |
| MOCK_METHOD(bool, |
| OnProtocolVersionMismatch, |
| (ParsedQuicVersion version), |
| (override)); |
| MOCK_METHOD(void, OnPacket, (), (override)); |
| MOCK_METHOD(void, |
| OnPublicResetPacket, |
| (const QuicPublicResetPacket& header), |
| (override)); |
| MOCK_METHOD(void, |
| OnVersionNegotiationPacket, |
| (const QuicVersionNegotiationPacket& packet), |
| (override)); |
| MOCK_METHOD(void, |
| OnRetryPacket, |
| (QuicConnectionId original_connection_id, |
| QuicConnectionId new_connection_id, |
| absl::string_view retry_token, |
| absl::string_view retry_integrity_tag, |
| absl::string_view retry_without_tag), |
| (override)); |
| // The constructor sets this up to return true by default. |
| MOCK_METHOD(bool, |
| OnUnauthenticatedHeader, |
| (const QuicPacketHeader& header), |
| (override)); |
| // The constructor sets this up to return true by default. |
| MOCK_METHOD(bool, |
| OnUnauthenticatedPublicHeader, |
| (const QuicPacketHeader& header), |
| (override)); |
| MOCK_METHOD(void, |
| OnDecryptedPacket, |
| (size_t length, EncryptionLevel level), |
| (override)); |
| MOCK_METHOD(bool, |
| OnPacketHeader, |
| (const QuicPacketHeader& header), |
| (override)); |
| MOCK_METHOD(void, |
| OnCoalescedPacket, |
| (const QuicEncryptedPacket& packet), |
| (override)); |
| MOCK_METHOD(void, |
| OnUndecryptablePacket, |
| (const QuicEncryptedPacket& packet, |
| EncryptionLevel decryption_level, |
| bool has_decryption_key), |
| (override)); |
| MOCK_METHOD(bool, OnStreamFrame, (const QuicStreamFrame& frame), (override)); |
| MOCK_METHOD(bool, OnCryptoFrame, (const QuicCryptoFrame& frame), (override)); |
| MOCK_METHOD(bool, |
| OnAckFrameStart, |
| (QuicPacketNumber, QuicTime::Delta), |
| (override)); |
| MOCK_METHOD(bool, |
| OnAckRange, |
| (QuicPacketNumber, QuicPacketNumber), |
| (override)); |
| MOCK_METHOD(bool, OnAckTimestamp, (QuicPacketNumber, QuicTime), (override)); |
| MOCK_METHOD(bool, OnAckFrameEnd, (QuicPacketNumber), (override)); |
| MOCK_METHOD(bool, |
| OnStopWaitingFrame, |
| (const QuicStopWaitingFrame& frame), |
| (override)); |
| MOCK_METHOD(bool, |
| OnPaddingFrame, |
| (const QuicPaddingFrame& frame), |
| (override)); |
| MOCK_METHOD(bool, OnPingFrame, (const QuicPingFrame& frame), (override)); |
| MOCK_METHOD(bool, |
| OnRstStreamFrame, |
| (const QuicRstStreamFrame& frame), |
| (override)); |
| MOCK_METHOD(bool, |
| OnConnectionCloseFrame, |
| (const QuicConnectionCloseFrame& frame), |
| (override)); |
| MOCK_METHOD(bool, |
| OnNewConnectionIdFrame, |
| (const QuicNewConnectionIdFrame& frame), |
| (override)); |
| MOCK_METHOD(bool, |
| OnRetireConnectionIdFrame, |
| (const QuicRetireConnectionIdFrame& frame), |
| (override)); |
| MOCK_METHOD(bool, |
| OnNewTokenFrame, |
| (const QuicNewTokenFrame& frame), |
| (override)); |
| MOCK_METHOD(bool, |
| OnStopSendingFrame, |
| (const QuicStopSendingFrame& frame), |
| (override)); |
| MOCK_METHOD(bool, |
| OnPathChallengeFrame, |
| (const QuicPathChallengeFrame& frame), |
| (override)); |
| MOCK_METHOD(bool, |
| OnPathResponseFrame, |
| (const QuicPathResponseFrame& frame), |
| (override)); |
| MOCK_METHOD(bool, OnGoAwayFrame, (const QuicGoAwayFrame& frame), (override)); |
| MOCK_METHOD(bool, |
| OnMaxStreamsFrame, |
| (const QuicMaxStreamsFrame& frame), |
| (override)); |
| MOCK_METHOD(bool, |
| OnStreamsBlockedFrame, |
| (const QuicStreamsBlockedFrame& frame), |
| (override)); |
| MOCK_METHOD(bool, |
| OnWindowUpdateFrame, |
| (const QuicWindowUpdateFrame& frame), |
| (override)); |
| MOCK_METHOD(bool, |
| OnBlockedFrame, |
| (const QuicBlockedFrame& frame), |
| (override)); |
| MOCK_METHOD(bool, |
| OnMessageFrame, |
| (const QuicMessageFrame& frame), |
| (override)); |
| MOCK_METHOD(bool, |
| OnHandshakeDoneFrame, |
| (const QuicHandshakeDoneFrame& frame), |
| (override)); |
| MOCK_METHOD(bool, |
| OnAckFrequencyFrame, |
| (const QuicAckFrequencyFrame& frame), |
| (override)); |
| MOCK_METHOD(void, OnPacketComplete, (), (override)); |
| MOCK_METHOD(bool, |
| IsValidStatelessResetToken, |
| (QuicUint128), |
| (const, override)); |
| MOCK_METHOD(void, |
| OnAuthenticatedIetfStatelessResetPacket, |
| (const QuicIetfStatelessResetPacket&), |
| (override)); |
| MOCK_METHOD(void, OnKeyUpdate, (KeyUpdateReason), (override)); |
| MOCK_METHOD(void, OnDecryptedFirstPacketInKeyPhase, (), (override)); |
| MOCK_METHOD(std::unique_ptr<QuicDecrypter>, |
| AdvanceKeysAndCreateCurrentOneRttDecrypter, |
| (), |
| (override)); |
| MOCK_METHOD(std::unique_ptr<QuicEncrypter>, |
| CreateCurrentOneRttEncrypter, |
| (), |
| (override)); |
| }; |
| |
| class NoOpFramerVisitor : public QuicFramerVisitorInterface { |
| public: |
| NoOpFramerVisitor() {} |
| NoOpFramerVisitor(const NoOpFramerVisitor&) = delete; |
| NoOpFramerVisitor& operator=(const NoOpFramerVisitor&) = delete; |
| |
| void OnError(QuicFramer* /*framer*/) override {} |
| void OnPacket() override {} |
| void OnPublicResetPacket(const QuicPublicResetPacket& /*packet*/) override {} |
| void OnVersionNegotiationPacket( |
| const QuicVersionNegotiationPacket& /*packet*/) override {} |
| void OnRetryPacket(QuicConnectionId /*original_connection_id*/, |
| QuicConnectionId /*new_connection_id*/, |
| absl::string_view /*retry_token*/, |
| absl::string_view /*retry_integrity_tag*/, |
| absl::string_view /*retry_without_tag*/) override {} |
| bool OnProtocolVersionMismatch(ParsedQuicVersion version) override; |
| bool OnUnauthenticatedHeader(const QuicPacketHeader& header) override; |
| bool OnUnauthenticatedPublicHeader(const QuicPacketHeader& header) override; |
| void OnDecryptedPacket(size_t /*length*/, |
| EncryptionLevel /*level*/) override {} |
| bool OnPacketHeader(const QuicPacketHeader& header) override; |
| void OnCoalescedPacket(const QuicEncryptedPacket& packet) override; |
| void OnUndecryptablePacket(const QuicEncryptedPacket& packet, |
| EncryptionLevel decryption_level, |
| bool has_decryption_key) override; |
| bool OnStreamFrame(const QuicStreamFrame& frame) override; |
| bool OnCryptoFrame(const QuicCryptoFrame& frame) override; |
| bool OnAckFrameStart(QuicPacketNumber largest_acked, |
| QuicTime::Delta ack_delay_time) override; |
| bool OnAckRange(QuicPacketNumber start, QuicPacketNumber end) override; |
| bool OnAckTimestamp(QuicPacketNumber packet_number, |
| QuicTime timestamp) override; |
| bool OnAckFrameEnd(QuicPacketNumber start) override; |
| bool OnStopWaitingFrame(const QuicStopWaitingFrame& frame) override; |
| bool OnPaddingFrame(const QuicPaddingFrame& frame) override; |
| bool OnPingFrame(const QuicPingFrame& frame) override; |
| bool OnRstStreamFrame(const QuicRstStreamFrame& frame) override; |
| bool OnConnectionCloseFrame(const QuicConnectionCloseFrame& frame) override; |
| bool OnNewConnectionIdFrame(const QuicNewConnectionIdFrame& frame) override; |
| bool OnRetireConnectionIdFrame( |
| const QuicRetireConnectionIdFrame& frame) override; |
| bool OnNewTokenFrame(const QuicNewTokenFrame& frame) override; |
| bool OnStopSendingFrame(const QuicStopSendingFrame& frame) override; |
| bool OnPathChallengeFrame(const QuicPathChallengeFrame& frame) override; |
| bool OnPathResponseFrame(const QuicPathResponseFrame& frame) override; |
| bool OnGoAwayFrame(const QuicGoAwayFrame& frame) override; |
| bool OnMaxStreamsFrame(const QuicMaxStreamsFrame& frame) override; |
| bool OnStreamsBlockedFrame(const QuicStreamsBlockedFrame& frame) override; |
| bool OnWindowUpdateFrame(const QuicWindowUpdateFrame& frame) override; |
| bool OnBlockedFrame(const QuicBlockedFrame& frame) override; |
| bool OnMessageFrame(const QuicMessageFrame& frame) override; |
| bool OnHandshakeDoneFrame(const QuicHandshakeDoneFrame& frame) override; |
| bool OnAckFrequencyFrame(const QuicAckFrequencyFrame& frame) override; |
| void OnPacketComplete() override {} |
| bool IsValidStatelessResetToken(QuicUint128 token) const override; |
| void OnAuthenticatedIetfStatelessResetPacket( |
| const QuicIetfStatelessResetPacket& /*packet*/) override {} |
| void OnKeyUpdate(KeyUpdateReason /*reason*/) override {} |
| void OnDecryptedFirstPacketInKeyPhase() override {} |
| std::unique_ptr<QuicDecrypter> AdvanceKeysAndCreateCurrentOneRttDecrypter() |
| override { |
| return nullptr; |
| } |
| std::unique_ptr<QuicEncrypter> CreateCurrentOneRttEncrypter() override { |
| return nullptr; |
| } |
| }; |
| |
| class MockQuicConnectionVisitor : public QuicConnectionVisitorInterface { |
| public: |
| MockQuicConnectionVisitor(); |
| MockQuicConnectionVisitor(const MockQuicConnectionVisitor&) = delete; |
| MockQuicConnectionVisitor& operator=(const MockQuicConnectionVisitor&) = |
| delete; |
| ~MockQuicConnectionVisitor() override; |
| |
| MOCK_METHOD(void, OnStreamFrame, (const QuicStreamFrame& frame), (override)); |
| MOCK_METHOD(void, OnCryptoFrame, (const QuicCryptoFrame& frame), (override)); |
| MOCK_METHOD(void, |
| OnWindowUpdateFrame, |
| (const QuicWindowUpdateFrame& frame), |
| (override)); |
| MOCK_METHOD(void, |
| OnBlockedFrame, |
| (const QuicBlockedFrame& frame), |
| (override)); |
| MOCK_METHOD(void, OnRstStream, (const QuicRstStreamFrame& frame), (override)); |
| MOCK_METHOD(void, OnGoAway, (const QuicGoAwayFrame& frame), (override)); |
| MOCK_METHOD(void, OnMessageReceived, (absl::string_view message), (override)); |
| MOCK_METHOD(void, OnHandshakeDoneReceived, (), (override)); |
| MOCK_METHOD(void, OnNewTokenReceived, (absl::string_view token), (override)); |
| MOCK_METHOD(void, |
| OnConnectionClosed, |
| (const QuicConnectionCloseFrame& frame, |
| ConnectionCloseSource source), |
| (override)); |
| MOCK_METHOD(void, OnWriteBlocked, (), (override)); |
| MOCK_METHOD(void, OnCanWrite, (), (override)); |
| MOCK_METHOD(bool, SendProbingData, (), (override)); |
| MOCK_METHOD(bool, |
| ValidateStatelessReset, |
| (const quic::QuicSocketAddress&, const quic::QuicSocketAddress&), |
| (override)); |
| MOCK_METHOD(void, OnCongestionWindowChange, (QuicTime now), (override)); |
| MOCK_METHOD(void, |
| OnConnectionMigration, |
| (AddressChangeType type), |
| (override)); |
| MOCK_METHOD(void, OnPathDegrading, (), (override)); |
| MOCK_METHOD(void, OnForwardProgressMadeAfterPathDegrading, (), (override)); |
| MOCK_METHOD(bool, WillingAndAbleToWrite, (), (const, override)); |
| MOCK_METHOD(bool, ShouldKeepConnectionAlive, (), (const, override)); |
| MOCK_METHOD(std::string, GetStreamsInfoForLogging, (), (const, override)); |
| MOCK_METHOD(void, |
| OnSuccessfulVersionNegotiation, |
| (const ParsedQuicVersion& version), |
| (override)); |
| MOCK_METHOD(void, |
| OnPacketReceived, |
| (const QuicSocketAddress& self_address, |
| const QuicSocketAddress& peer_address, |
| bool is_connectivity_probe), |
| (override)); |
| MOCK_METHOD(void, OnAckNeedsRetransmittableFrame, (), (override)); |
| MOCK_METHOD(void, |
| SendAckFrequency, |
| (const QuicAckFrequencyFrame& frame), |
| (override)); |
| MOCK_METHOD(bool, AllowSelfAddressChange, (), (const, override)); |
| MOCK_METHOD(HandshakeState, GetHandshakeState, (), (const, override)); |
| MOCK_METHOD(bool, |
| OnMaxStreamsFrame, |
| (const QuicMaxStreamsFrame& frame), |
| (override)); |
| MOCK_METHOD(bool, |
| OnStreamsBlockedFrame, |
| (const QuicStreamsBlockedFrame& frame), |
| (override)); |
| MOCK_METHOD(void, |
| OnStopSendingFrame, |
| (const QuicStopSendingFrame& frame), |
| (override)); |
| MOCK_METHOD(void, OnPacketDecrypted, (EncryptionLevel), (override)); |
| MOCK_METHOD(void, OnOneRttPacketAcknowledged, (), (override)); |
| MOCK_METHOD(void, OnHandshakePacketSent, (), (override)); |
| MOCK_METHOD(void, OnKeyUpdate, (KeyUpdateReason), (override)); |
| MOCK_METHOD(std::unique_ptr<QuicDecrypter>, |
| AdvanceKeysAndCreateCurrentOneRttDecrypter, |
| (), |
| (override)); |
| MOCK_METHOD(std::unique_ptr<QuicEncrypter>, |
| CreateCurrentOneRttEncrypter, |
| (), |
| (override)); |
| MOCK_METHOD(void, BeforeConnectionCloseSent, (), (override)); |
| MOCK_METHOD(bool, ValidateToken, (absl::string_view), (const, override)); |
| MOCK_METHOD(void, MaybeSendAddressToken, (), (override)); |
| }; |
| |
| class MockQuicConnectionHelper : public QuicConnectionHelperInterface { |
| public: |
| MockQuicConnectionHelper(); |
| MockQuicConnectionHelper(const MockQuicConnectionHelper&) = delete; |
| MockQuicConnectionHelper& operator=(const MockQuicConnectionHelper&) = delete; |
| ~MockQuicConnectionHelper() override; |
| const QuicClock* GetClock() const override; |
| QuicRandom* GetRandomGenerator() override; |
| QuicBufferAllocator* GetStreamSendBufferAllocator() override; |
| void AdvanceTime(QuicTime::Delta delta); |
| |
| private: |
| MockClock clock_; |
| MockRandom random_generator_; |
| SimpleBufferAllocator buffer_allocator_; |
| }; |
| |
| class MockAlarmFactory : public QuicAlarmFactory { |
| public: |
| QuicAlarm* CreateAlarm(QuicAlarm::Delegate* delegate) override; |
| QuicArenaScopedPtr<QuicAlarm> CreateAlarm( |
| QuicArenaScopedPtr<QuicAlarm::Delegate> delegate, |
| QuicConnectionArena* arena) override; |
| |
| // No-op alarm implementation |
| class TestAlarm : public QuicAlarm { |
| public: |
| explicit TestAlarm(QuicArenaScopedPtr<QuicAlarm::Delegate> delegate) |
| : QuicAlarm(std::move(delegate)) {} |
| |
| void SetImpl() override {} |
| void CancelImpl() override {} |
| |
| using QuicAlarm::Fire; |
| }; |
| |
| void FireAlarm(QuicAlarm* alarm) { |
| reinterpret_cast<TestAlarm*>(alarm)->Fire(); |
| } |
| }; |
| |
| class MockQuicConnection : public QuicConnection { |
| public: |
| // Uses a ConnectionId of 42 and 127.0.0.1:123. |
| MockQuicConnection(MockQuicConnectionHelper* helper, |
| MockAlarmFactory* alarm_factory, |
| Perspective perspective); |
| |
| // Uses a ConnectionId of 42. |
| MockQuicConnection(QuicSocketAddress address, |
| MockQuicConnectionHelper* helper, |
| MockAlarmFactory* alarm_factory, |
| Perspective perspective); |
| |
| // Uses 127.0.0.1:123. |
| MockQuicConnection(QuicConnectionId connection_id, |
| MockQuicConnectionHelper* helper, |
| MockAlarmFactory* alarm_factory, |
| Perspective perspective); |
| |
| // Uses a ConnectionId of 42, and 127.0.0.1:123. |
| MockQuicConnection(MockQuicConnectionHelper* helper, |
| MockAlarmFactory* alarm_factory, |
| Perspective perspective, |
| const ParsedQuicVersionVector& supported_versions); |
| |
| MockQuicConnection(QuicConnectionId connection_id, |
| QuicSocketAddress address, |
| MockQuicConnectionHelper* helper, |
| MockAlarmFactory* alarm_factory, |
| Perspective perspective, |
| const ParsedQuicVersionVector& supported_versions); |
| MockQuicConnection(const MockQuicConnection&) = delete; |
| MockQuicConnection& operator=(const MockQuicConnection&) = delete; |
| |
| ~MockQuicConnection() override; |
| |
| // If the constructor that uses a MockQuicConnectionHelper has been used then |
| // this method |
| // will advance the time of the MockClock. |
| void AdvanceTime(QuicTime::Delta delta); |
| |
| MOCK_METHOD(void, |
| ProcessUdpPacket, |
| (const QuicSocketAddress& self_address, |
| const QuicSocketAddress& peer_address, |
| const QuicReceivedPacket& packet), |
| (override)); |
| MOCK_METHOD(void, |
| CloseConnection, |
| (QuicErrorCode error, |
| const std::string& details, |
| ConnectionCloseBehavior connection_close_behavior), |
| (override)); |
| MOCK_METHOD(void, |
| SendConnectionClosePacket, |
| (QuicErrorCode error, const std::string& details), |
| (override)); |
| MOCK_METHOD(void, OnCanWrite, (), (override)); |
| MOCK_METHOD(void, |
| SendConnectivityProbingResponsePacket, |
| (const QuicSocketAddress& peer_address), |
| (override)); |
| MOCK_METHOD(bool, |
| SendConnectivityProbingPacket, |
| (QuicPacketWriter*, const QuicSocketAddress& peer_address), |
| (override)); |
| |
| MOCK_METHOD(void, |
| OnSendConnectionState, |
| (const CachedNetworkParameters&), |
| (override)); |
| MOCK_METHOD(void, |
| ResumeConnectionState, |
| (const CachedNetworkParameters&, bool), |
| (override)); |
| MOCK_METHOD(void, SetMaxPacingRate, (QuicBandwidth), (override)); |
| |
| MOCK_METHOD(void, |
| OnStreamReset, |
| (QuicStreamId, QuicRstStreamErrorCode), |
| (override)); |
| MOCK_METHOD(bool, SendControlFrame, (const QuicFrame& frame), (override)); |
| MOCK_METHOD(MessageStatus, |
| SendMessage, |
| (QuicMessageId, QuicMemSliceSpan, bool), |
| (override)); |
| MOCK_METHOD(bool, |
| SendPathChallenge, |
| (const QuicPathFrameBuffer&, |
| const QuicSocketAddress&, |
| const QuicSocketAddress&, |
| const QuicSocketAddress&, |
| QuicPacketWriter*), |
| (override)); |
| |
| MOCK_METHOD(void, OnError, (QuicFramer*), (override)); |
| void QuicConnection_OnError(QuicFramer* framer) { |
| QuicConnection::OnError(framer); |
| } |
| |
| void ReallyOnCanWrite() { QuicConnection::OnCanWrite(); } |
| |
| void ReallyCloseConnection( |
| QuicErrorCode error, |
| const std::string& details, |
| ConnectionCloseBehavior connection_close_behavior) { |
| QuicConnection::CloseConnection(error, details, connection_close_behavior); |
| } |
| |
| void ReallySendConnectionClosePacket(QuicErrorCode error, |
| const std::string& details) { |
| QuicConnection::SendConnectionClosePacket(error, details); |
| } |
| |
| void ReallyProcessUdpPacket(const QuicSocketAddress& self_address, |
| const QuicSocketAddress& peer_address, |
| const QuicReceivedPacket& packet) { |
| QuicConnection::ProcessUdpPacket(self_address, peer_address, packet); |
| } |
| |
| bool OnProtocolVersionMismatch(ParsedQuicVersion version) override; |
| |
| bool ReallySendControlFrame(const QuicFrame& frame) { |
| return QuicConnection::SendControlFrame(frame); |
| } |
| |
| bool ReallySendConnectivityProbingPacket( |
| QuicPacketWriter* probing_writer, |
| const QuicSocketAddress& peer_address) { |
| return QuicConnection::SendConnectivityProbingPacket(probing_writer, |
| peer_address); |
| } |
| |
| void ReallySendConnectivityProbingResponsePacket( |
| const QuicSocketAddress& peer_address) { |
| QuicConnection::SendConnectivityProbingResponsePacket(peer_address); |
| } |
| |
| bool ReallyOnPathResponseFrame(const QuicPathResponseFrame& frame) { |
| return QuicConnection::OnPathResponseFrame(frame); |
| } |
| |
| MOCK_METHOD(bool, |
| OnPathResponseFrame, |
| (const QuicPathResponseFrame&), |
| (override)); |
| MOCK_METHOD(bool, |
| OnStopSendingFrame, |
| (const QuicStopSendingFrame& frame), |
| (override)); |
| MOCK_METHOD(size_t, |
| SendCryptoData, |
| (EncryptionLevel, size_t, QuicStreamOffset), |
| (override)); |
| size_t QuicConnection_SendCryptoData(EncryptionLevel level, |
| size_t write_length, |
| QuicStreamOffset offset) { |
| return QuicConnection::SendCryptoData(level, write_length, offset); |
| } |
| }; |
| |
| class PacketSavingConnection : public MockQuicConnection { |
| public: |
| PacketSavingConnection(MockQuicConnectionHelper* helper, |
| MockAlarmFactory* alarm_factory, |
| Perspective perspective); |
| |
| PacketSavingConnection(MockQuicConnectionHelper* helper, |
| MockAlarmFactory* alarm_factory, |
| Perspective perspective, |
| const ParsedQuicVersionVector& supported_versions); |
| PacketSavingConnection(const PacketSavingConnection&) = delete; |
| PacketSavingConnection& operator=(const PacketSavingConnection&) = delete; |
| |
| ~PacketSavingConnection() override; |
| |
| void SendOrQueuePacket(SerializedPacket packet) override; |
| |
| MOCK_METHOD(void, OnPacketSent, (EncryptionLevel, TransmissionType)); |
| |
| std::vector<std::unique_ptr<QuicEncryptedPacket>> encrypted_packets_; |
| MockClock clock_; |
| }; |
| |
| class MockQuicSession : public QuicSession { |
| public: |
| // Takes ownership of |connection|. |
| MockQuicSession(QuicConnection* connection, bool create_mock_crypto_stream); |
| |
| // Takes ownership of |connection|. |
| explicit MockQuicSession(QuicConnection* connection); |
| MockQuicSession(const MockQuicSession&) = delete; |
| MockQuicSession& operator=(const MockQuicSession&) = delete; |
| ~MockQuicSession() override; |
| |
| QuicCryptoStream* GetMutableCryptoStream() override; |
| const QuicCryptoStream* GetCryptoStream() const override; |
| void SetCryptoStream(QuicCryptoStream* crypto_stream); |
| |
| MOCK_METHOD(void, |
| OnConnectionClosed, |
| (const QuicConnectionCloseFrame& frame, |
| ConnectionCloseSource source), |
| (override)); |
| MOCK_METHOD(QuicStream*, CreateIncomingStream, (QuicStreamId id), (override)); |
| MOCK_METHOD(QuicSpdyStream*, |
| CreateIncomingStream, |
| (PendingStream*), |
| (override)); |
| MOCK_METHOD(QuicConsumedData, |
| WritevData, |
| (QuicStreamId id, |
| size_t write_length, |
| QuicStreamOffset offset, |
| StreamSendingState state, |
| TransmissionType type, |
| absl::optional<EncryptionLevel> level), |
| (override)); |
| MOCK_METHOD(bool, |
| WriteControlFrame, |
| (const QuicFrame& frame, TransmissionType type), |
| (override)); |
| |
| MOCK_METHOD(void, |
| SendRstStream, |
| (QuicStreamId stream_id, |
| QuicRstStreamErrorCode error, |
| QuicStreamOffset bytes_written, |
| bool send_rst_only), |
| (override)); |
| MOCK_METHOD(void, |
| MaybeSendRstStreamFrame, |
| (QuicStreamId stream_id, |
| QuicRstStreamErrorCode error, |
| QuicStreamOffset bytes_written), |
| (override)); |
| MOCK_METHOD(void, |
| MaybeSendStopSendingFrame, |
| (QuicStreamId stream_id, QuicRstStreamErrorCode error), |
| (override)); |
| |
| MOCK_METHOD(bool, ShouldKeepConnectionAlive, (), (const, override)); |
| MOCK_METHOD(void, |
| SendStopSending, |
| (QuicRstStreamErrorCode code, QuicStreamId stream_id), |
| (override)); |
| MOCK_METHOD(std::vector<std::string>, GetAlpnsToOffer, (), (const, override)); |
| MOCK_METHOD(std::vector<absl::string_view>::const_iterator, |
| SelectAlpn, |
| (const std::vector<absl::string_view>&), |
| (const, override)); |
| MOCK_METHOD(void, OnAlpnSelected, (absl::string_view), (override)); |
| |
| using QuicSession::ActivateStream; |
| |
| // Returns a QuicConsumedData that indicates all of |write_length| (and |fin| |
| // if set) has been consumed. |
| QuicConsumedData ConsumeData(QuicStreamId id, |
| size_t write_length, |
| QuicStreamOffset offset, |
| StreamSendingState state, |
| TransmissionType type, |
| absl::optional<EncryptionLevel> level); |
| |
| void ReallySendRstStream(QuicStreamId id, |
| QuicRstStreamErrorCode error, |
| QuicStreamOffset bytes_written, |
| bool send_rst_only) { |
| QuicSession::SendRstStream(id, error, bytes_written, send_rst_only); |
| } |
| |
| void ReallyMaybeSendRstStreamFrame(QuicStreamId id, |
| QuicRstStreamErrorCode error, |
| QuicStreamOffset bytes_written) { |
| QuicSession::MaybeSendRstStreamFrame(id, error, bytes_written); |
| } |
| |
| private: |
| std::unique_ptr<QuicCryptoStream> crypto_stream_; |
| }; |
| |
| class MockQuicCryptoStream : public QuicCryptoStream { |
| public: |
| explicit MockQuicCryptoStream(QuicSession* session); |
| |
| ~MockQuicCryptoStream() override; |
| |
| ssl_early_data_reason_t EarlyDataReason() const override; |
| bool encryption_established() const override; |
| bool one_rtt_keys_available() const override; |
| const QuicCryptoNegotiatedParameters& crypto_negotiated_params() |
| const override; |
| CryptoMessageParser* crypto_message_parser() override; |
| void OnPacketDecrypted(EncryptionLevel /*level*/) override {} |
| void OnOneRttPacketAcknowledged() override {} |
| void OnHandshakePacketSent() override {} |
| void OnHandshakeDoneReceived() override {} |
| void OnNewTokenReceived(absl::string_view /*token*/) override {} |
| std::string GetAddressToken() const override { return ""; } |
| bool ValidateAddressToken(absl::string_view /*token*/) const override { |
| return true; |
| } |
| void OnConnectionClosed(QuicErrorCode /*error*/, |
| ConnectionCloseSource /*source*/) override {} |
| HandshakeState GetHandshakeState() const override { return HANDSHAKE_START; } |
| void SetServerApplicationStateForResumption( |
| std::unique_ptr<ApplicationState> /*application_state*/) override {} |
| bool KeyUpdateSupportedLocally() const override { return false; } |
| std::unique_ptr<QuicDecrypter> AdvanceKeysAndCreateCurrentOneRttDecrypter() |
| override { |
| return nullptr; |
| } |
| std::unique_ptr<QuicEncrypter> CreateCurrentOneRttEncrypter() override { |
| return nullptr; |
| } |
| |
| private: |
| QuicReferenceCountedPointer<QuicCryptoNegotiatedParameters> params_; |
| CryptoFramer crypto_framer_; |
| }; |
| |
| class MockQuicSpdySession : public QuicSpdySession { |
| public: |
| // Takes ownership of |connection|. |
| explicit MockQuicSpdySession(QuicConnection* connection); |
| // Takes ownership of |connection|. |
| MockQuicSpdySession(QuicConnection* connection, |
| bool create_mock_crypto_stream); |
| MockQuicSpdySession(const MockQuicSpdySession&) = delete; |
| MockQuicSpdySession& operator=(const MockQuicSpdySession&) = delete; |
| ~MockQuicSpdySession() override; |
| |
| QuicCryptoStream* GetMutableCryptoStream() override; |
| const QuicCryptoStream* GetCryptoStream() const override; |
| void SetCryptoStream(QuicCryptoStream* crypto_stream); |
| |
| void ReallyOnConnectionClosed(const QuicConnectionCloseFrame& frame, |
| ConnectionCloseSource source) { |
| QuicSession::OnConnectionClosed(frame, source); |
| } |
| |
| // From QuicSession. |
| MOCK_METHOD(void, |
| OnConnectionClosed, |
| (const QuicConnectionCloseFrame& frame, |
| ConnectionCloseSource source), |
| (override)); |
| MOCK_METHOD(QuicSpdyStream*, |
| CreateIncomingStream, |
| (QuicStreamId id), |
| (override)); |
| MOCK_METHOD(QuicSpdyStream*, |
| CreateIncomingStream, |
| (PendingStream*), |
| (override)); |
| MOCK_METHOD(QuicSpdyStream*, |
| CreateOutgoingBidirectionalStream, |
| (), |
| (override)); |
| MOCK_METHOD(QuicSpdyStream*, |
| CreateOutgoingUnidirectionalStream, |
| (), |
| (override)); |
| MOCK_METHOD(bool, ShouldCreateIncomingStream, (QuicStreamId id), (override)); |
| MOCK_METHOD(bool, ShouldCreateOutgoingBidirectionalStream, (), (override)); |
| MOCK_METHOD(bool, ShouldCreateOutgoingUnidirectionalStream, (), (override)); |
| MOCK_METHOD(QuicConsumedData, |
| WritevData, |
| (QuicStreamId id, |
| size_t write_length, |
| QuicStreamOffset offset, |
| StreamSendingState state, |
| TransmissionType type, |
| absl::optional<EncryptionLevel> level), |
| (override)); |
| MOCK_METHOD(void, |
| SendRstStream, |
| (QuicStreamId stream_id, |
| QuicRstStreamErrorCode error, |
| QuicStreamOffset bytes_written, |
| bool send_rst_only), |
| (override)); |
| MOCK_METHOD(void, |
| MaybeSendRstStreamFrame, |
| (QuicStreamId stream_id, |
| QuicRstStreamErrorCode error, |
| QuicStreamOffset bytes_written), |
| (override)); |
| MOCK_METHOD(void, |
| MaybeSendStopSendingFrame, |
| (QuicStreamId stream_id, QuicRstStreamErrorCode error), |
| (override)); |
| MOCK_METHOD(void, |
| SendWindowUpdate, |
| (QuicStreamId id, QuicStreamOffset byte_offset), |
| (override)); |
| MOCK_METHOD(void, SendBlocked, (QuicStreamId id), (override)); |
| MOCK_METHOD(void, |
| OnStreamHeadersPriority, |
| (QuicStreamId stream_id, |
| const spdy::SpdyStreamPrecedence& precedence), |
| (override)); |
| MOCK_METHOD(void, |
| OnStreamHeaderList, |
| (QuicStreamId stream_id, |
| bool fin, |
| size_t frame_len, |
| const QuicHeaderList& header_list), |
| (override)); |
| MOCK_METHOD(void, |
| OnPromiseHeaderList, |
| (QuicStreamId stream_id, |
| QuicStreamId promised_stream_id, |
| size_t frame_len, |
| const QuicHeaderList& header_list), |
| (override)); |
| MOCK_METHOD(void, |
| OnPriorityFrame, |
| (QuicStreamId id, const spdy::SpdyStreamPrecedence& precedence), |
| (override)); |
| MOCK_METHOD(void, OnCongestionWindowChange, (QuicTime now), (override)); |
| |
| // Returns a QuicConsumedData that indicates all of |write_length| (and |fin| |
| // if set) has been consumed. |
| QuicConsumedData ConsumeData(QuicStreamId id, |
| size_t write_length, |
| QuicStreamOffset offset, |
| StreamSendingState state, |
| TransmissionType type, |
| absl::optional<EncryptionLevel> level); |
| |
| using QuicSession::ActivateStream; |
| |
| private: |
| std::unique_ptr<QuicCryptoStream> crypto_stream_; |
| }; |
| |
| class MockHttp3DebugVisitor : public Http3DebugVisitor { |
| public: |
| MOCK_METHOD(void, OnControlStreamCreated, (QuicStreamId), (override)); |
| MOCK_METHOD(void, OnQpackEncoderStreamCreated, (QuicStreamId), (override)); |
| MOCK_METHOD(void, OnQpackDecoderStreamCreated, (QuicStreamId), (override)); |
| MOCK_METHOD(void, OnPeerControlStreamCreated, (QuicStreamId), (override)); |
| MOCK_METHOD(void, |
| OnPeerQpackEncoderStreamCreated, |
| (QuicStreamId), |
| (override)); |
| MOCK_METHOD(void, |
| OnPeerQpackDecoderStreamCreated, |
| (QuicStreamId), |
| (override)); |
| |
| MOCK_METHOD(void, |
| OnCancelPushFrameReceived, |
| (const CancelPushFrame&), |
| (override)); |
| MOCK_METHOD(void, |
| OnSettingsFrameReceived, |
| (const SettingsFrame&), |
| (override)); |
| MOCK_METHOD(void, OnGoAwayFrameReceived, (const GoAwayFrame&), (override)); |
| MOCK_METHOD(void, |
| OnMaxPushIdFrameReceived, |
| (const MaxPushIdFrame&), |
| (override)); |
| MOCK_METHOD(void, |
| OnPriorityUpdateFrameReceived, |
| (const PriorityUpdateFrame&), |
| (override)); |
| MOCK_METHOD(void, |
| OnAcceptChFrameReceived, |
| (const AcceptChFrame&), |
| (override)); |
| |
| MOCK_METHOD(void, |
| OnDataFrameReceived, |
| (QuicStreamId, QuicByteCount), |
| (override)); |
| MOCK_METHOD(void, |
| OnHeadersFrameReceived, |
| (QuicStreamId, QuicByteCount), |
| (override)); |
| MOCK_METHOD(void, |
| OnHeadersDecoded, |
| (QuicStreamId, QuicHeaderList), |
| (override)); |
| MOCK_METHOD(void, |
| OnPushPromiseFrameReceived, |
| (QuicStreamId, QuicStreamId, QuicByteCount), |
| (override)); |
| MOCK_METHOD(void, |
| OnPushPromiseDecoded, |
| (QuicStreamId, QuicStreamId, QuicHeaderList), |
| (override)); |
| MOCK_METHOD(void, |
| OnUnknownFrameReceived, |
| (QuicStreamId, uint64_t, QuicByteCount), |
| (override)); |
| |
| MOCK_METHOD(void, OnSettingsFrameSent, (const SettingsFrame&), (override)); |
| MOCK_METHOD(void, OnGoAwayFrameSent, (QuicStreamId), (override)); |
| MOCK_METHOD(void, OnMaxPushIdFrameSent, (const MaxPushIdFrame&), (override)); |
| MOCK_METHOD(void, |
| OnPriorityUpdateFrameSent, |
| (const PriorityUpdateFrame&), |
| (override)); |
| |
| MOCK_METHOD(void, OnDataFrameSent, (QuicStreamId, QuicByteCount), (override)); |
| MOCK_METHOD(void, |
| OnHeadersFrameSent, |
| (QuicStreamId, const spdy::SpdyHeaderBlock&), |
| (override)); |
| MOCK_METHOD(void, |
| OnPushPromiseFrameSent, |
| (QuicStreamId, QuicStreamId, const spdy::SpdyHeaderBlock&), |
| (override)); |
| }; |
| |
| class TestQuicSpdyServerSession : public QuicServerSessionBase { |
| public: |
| // Takes ownership of |connection|. |
| TestQuicSpdyServerSession(QuicConnection* connection, |
| const QuicConfig& config, |
| const ParsedQuicVersionVector& supported_versions, |
| const QuicCryptoServerConfig* crypto_config, |
| QuicCompressedCertsCache* compressed_certs_cache); |
| TestQuicSpdyServerSession(const TestQuicSpdyServerSession&) = delete; |
| TestQuicSpdyServerSession& operator=(const TestQuicSpdyServerSession&) = |
| delete; |
| ~TestQuicSpdyServerSession() override; |
| |
| MOCK_METHOD(QuicSpdyStream*, |
| CreateIncomingStream, |
| (QuicStreamId id), |
| (override)); |
| MOCK_METHOD(QuicSpdyStream*, |
| CreateIncomingStream, |
| (PendingStream*), |
| (override)); |
| MOCK_METHOD(QuicSpdyStream*, |
| CreateOutgoingBidirectionalStream, |
| (), |
| (override)); |
| MOCK_METHOD(QuicSpdyStream*, |
| CreateOutgoingUnidirectionalStream, |
| (), |
| (override)); |
| MOCK_METHOD(std::vector<absl::string_view>::const_iterator, |
| SelectAlpn, |
| (const std::vector<absl::string_view>&), |
| (const, override)); |
| MOCK_METHOD(void, OnAlpnSelected, (absl::string_view), (override)); |
| std::unique_ptr<QuicCryptoServerStreamBase> CreateQuicCryptoServerStream( |
| const QuicCryptoServerConfig* crypto_config, |
| QuicCompressedCertsCache* compressed_certs_cache) override; |
| |
| QuicCryptoServerStreamBase* GetMutableCryptoStream() override; |
| |
| const QuicCryptoServerStreamBase* GetCryptoStream() const override; |
| |
| MockQuicCryptoServerStreamHelper* helper() { return &helper_; } |
| |
| private: |
| MockQuicSessionVisitor visitor_; |
| MockQuicCryptoServerStreamHelper helper_; |
| }; |
| |
| // A test implementation of QuicClientPushPromiseIndex::Delegate. |
| class TestPushPromiseDelegate : public QuicClientPushPromiseIndex::Delegate { |
| public: |
| // |match| sets the validation result for checking whether designated header |
| // fields match for promise request and client request. |
| explicit TestPushPromiseDelegate(bool match); |
| |
| bool CheckVary(const spdy::SpdyHeaderBlock& client_request, |
| const spdy::SpdyHeaderBlock& promise_request, |
| const spdy::SpdyHeaderBlock& promise_response) override; |
| |
| void OnRendezvousResult(QuicSpdyStream* stream) override; |
| |
| QuicSpdyStream* rendezvous_stream() { return rendezvous_stream_; } |
| bool rendezvous_fired() { return rendezvous_fired_; } |
| |
| private: |
| bool match_; |
| bool rendezvous_fired_; |
| QuicSpdyStream* rendezvous_stream_; |
| }; |
| |
| class TestQuicSpdyClientSession : public QuicSpdyClientSessionBase { |
| public: |
| TestQuicSpdyClientSession(QuicConnection* connection, |
| const QuicConfig& config, |
| const ParsedQuicVersionVector& supported_versions, |
| const QuicServerId& server_id, |
| QuicCryptoClientConfig* crypto_config); |
| TestQuicSpdyClientSession(const TestQuicSpdyClientSession&) = delete; |
| TestQuicSpdyClientSession& operator=(const TestQuicSpdyClientSession&) = |
| delete; |
| ~TestQuicSpdyClientSession() override; |
| |
| bool IsAuthorized(const std::string& authority) override; |
| |
| // QuicSpdyClientSessionBase |
| MOCK_METHOD(void, |
| OnProofValid, |
| (const QuicCryptoClientConfig::CachedState& cached), |
| (override)); |
| MOCK_METHOD(void, |
| OnProofVerifyDetailsAvailable, |
| (const ProofVerifyDetails& verify_details), |
| (override)); |
| |
| // TestQuicSpdyClientSession |
| MOCK_METHOD(QuicSpdyStream*, |
| CreateIncomingStream, |
| (QuicStreamId id), |
| (override)); |
| MOCK_METHOD(QuicSpdyStream*, |
| CreateIncomingStream, |
| (PendingStream*), |
| (override)); |
| MOCK_METHOD(QuicSpdyStream*, |
| CreateOutgoingBidirectionalStream, |
| (), |
| (override)); |
| MOCK_METHOD(QuicSpdyStream*, |
| CreateOutgoingUnidirectionalStream, |
| (), |
| (override)); |
| MOCK_METHOD(bool, ShouldCreateIncomingStream, (QuicStreamId id), (override)); |
| MOCK_METHOD(bool, ShouldCreateOutgoingBidirectionalStream, (), (override)); |
| MOCK_METHOD(bool, ShouldCreateOutgoingUnidirectionalStream, (), (override)); |
| MOCK_METHOD(std::vector<std::string>, GetAlpnsToOffer, (), (const, override)); |
| MOCK_METHOD(void, OnAlpnSelected, (absl::string_view), (override)); |
| MOCK_METHOD(void, OnConfigNegotiated, (), (override)); |
| |
| QuicCryptoClientStream* GetMutableCryptoStream() override; |
| const QuicCryptoClientStream* GetCryptoStream() const override; |
| |
| // Override to save sent crypto handshake messages. |
| void OnCryptoHandshakeMessageSent( |
| const CryptoHandshakeMessage& message) override { |
| sent_crypto_handshake_messages_.push_back(message); |
| } |
| |
| const std::vector<CryptoHandshakeMessage>& sent_crypto_handshake_messages() |
| const { |
| return sent_crypto_handshake_messages_; |
| } |
| |
| private: |
| // Calls the parent class's OnConfigNegotiated method. Used to set the default |
| // mock behavior for OnConfigNegotiated. |
| void RealOnConfigNegotiated(); |
| |
| std::unique_ptr<QuicCryptoClientStream> crypto_stream_; |
| QuicClientPushPromiseIndex push_promise_index_; |
| std::vector<CryptoHandshakeMessage> sent_crypto_handshake_messages_; |
| }; |
| |
| class MockPacketWriter : public QuicPacketWriter { |
| public: |
| MockPacketWriter(); |
| MockPacketWriter(const MockPacketWriter&) = delete; |
| MockPacketWriter& operator=(const MockPacketWriter&) = delete; |
| ~MockPacketWriter() override; |
| |
| MOCK_METHOD(WriteResult, |
| WritePacket, |
| (const char*, |
| size_t buf_len, |
| const QuicIpAddress& self_address, |
| const QuicSocketAddress& peer_address, |
| PerPacketOptions*), |
| (override)); |
| MOCK_METHOD(bool, IsWriteBlocked, (), (const, override)); |
| MOCK_METHOD(void, SetWritable, (), (override)); |
| MOCK_METHOD(QuicByteCount, |
| GetMaxPacketSize, |
| (const QuicSocketAddress& peer_address), |
| (const, override)); |
| MOCK_METHOD(bool, SupportsReleaseTime, (), (const, override)); |
| MOCK_METHOD(bool, IsBatchMode, (), (const, override)); |
| MOCK_METHOD(QuicPacketBuffer, |
| GetNextWriteLocation, |
| (const QuicIpAddress& self_address, |
| const QuicSocketAddress& peer_address), |
| (override)); |
| MOCK_METHOD(WriteResult, Flush, (), (override)); |
| }; |
| |
| class MockSendAlgorithm : public SendAlgorithmInterface { |
| public: |
| MockSendAlgorithm(); |
| MockSendAlgorithm(const MockSendAlgorithm&) = delete; |
| MockSendAlgorithm& operator=(const MockSendAlgorithm&) = delete; |
| ~MockSendAlgorithm() override; |
| |
| MOCK_METHOD(void, |
| SetFromConfig, |
| (const QuicConfig& config, Perspective perspective), |
| (override)); |
| MOCK_METHOD(void, |
| ApplyConnectionOptions, |
| (const QuicTagVector& connection_options), |
| (override)); |
| MOCK_METHOD(void, |
| SetInitialCongestionWindowInPackets, |
| (QuicPacketCount packets), |
| (override)); |
| MOCK_METHOD(void, |
| OnCongestionEvent, |
| (bool rtt_updated, |
| QuicByteCount bytes_in_flight, |
| QuicTime event_time, |
| const AckedPacketVector& acked_packets, |
| const LostPacketVector& lost_packets), |
| (override)); |
| MOCK_METHOD(void, |
| OnPacketSent, |
| (QuicTime, |
| QuicByteCount, |
| QuicPacketNumber, |
| QuicByteCount, |
| HasRetransmittableData), |
| (override)); |
| MOCK_METHOD(void, OnPacketNeutered, (QuicPacketNumber), (override)); |
| MOCK_METHOD(void, OnRetransmissionTimeout, (bool), (override)); |
| MOCK_METHOD(void, OnConnectionMigration, (), (override)); |
| MOCK_METHOD(bool, CanSend, (QuicByteCount), (override)); |
| MOCK_METHOD(QuicBandwidth, PacingRate, (QuicByteCount), (const, override)); |
| MOCK_METHOD(QuicBandwidth, BandwidthEstimate, (), (const, override)); |
| MOCK_METHOD(QuicByteCount, GetCongestionWindow, (), (const, override)); |
| MOCK_METHOD(std::string, GetDebugState, (), (const, override)); |
| MOCK_METHOD(bool, InSlowStart, (), (const, override)); |
| MOCK_METHOD(bool, InRecovery, (), (const, override)); |
| MOCK_METHOD(bool, ShouldSendProbingPacket, (), (const, override)); |
| MOCK_METHOD(QuicByteCount, GetSlowStartThreshold, (), (const, override)); |
| MOCK_METHOD(CongestionControlType, |
| GetCongestionControlType, |
| (), |
| (const, override)); |
| MOCK_METHOD(void, |
| AdjustNetworkParameters, |
| (const NetworkParams&), |
| (override)); |
| MOCK_METHOD(void, OnApplicationLimited, (QuicByteCount), (override)); |
| MOCK_METHOD(void, |
| PopulateConnectionStats, |
| (QuicConnectionStats*), |
| (const, override)); |
| }; |
| |
| class MockLossAlgorithm : public LossDetectionInterface { |
| public: |
| MockLossAlgorithm(); |
| MockLossAlgorithm(const MockLossAlgorithm&) = delete; |
| MockLossAlgorithm& operator=(const MockLossAlgorithm&) = delete; |
| ~MockLossAlgorithm() override; |
| |
| MOCK_METHOD(void, |
| SetFromConfig, |
| (const QuicConfig& config, Perspective perspective), |
| (override)); |
| |
| MOCK_METHOD(DetectionStats, |
| DetectLosses, |
| (const QuicUnackedPacketMap& unacked_packets, |
| QuicTime time, |
| const RttStats& rtt_stats, |
| QuicPacketNumber largest_recently_acked, |
| const AckedPacketVector& packets_acked, |
| LostPacketVector*), |
| (override)); |
| MOCK_METHOD(QuicTime, GetLossTimeout, (), (const, override)); |
| MOCK_METHOD(void, |
| SpuriousLossDetected, |
| (const QuicUnackedPacketMap&, |
| const RttStats&, |
| QuicTime, |
| QuicPacketNumber, |
| QuicPacketNumber), |
| (override)); |
| |
| MOCK_METHOD(void, OnConfigNegotiated, (), (override)); |
| MOCK_METHOD(void, OnMinRttAvailable, (), (override)); |
| MOCK_METHOD(void, OnUserAgentIdKnown, (), (override)); |
| MOCK_METHOD(void, OnConnectionClosed, (), (override)); |
| MOCK_METHOD(void, OnReorderingDetected, (), (override)); |
| }; |
| |
| class MockAckListener : public QuicAckListenerInterface { |
| public: |
| MockAckListener(); |
| MockAckListener(const MockAckListener&) = delete; |
| MockAckListener& operator=(const MockAckListener&) = delete; |
| |
| MOCK_METHOD(void, |
| OnPacketAcked, |
| (int acked_bytes, QuicTime::Delta ack_delay_time), |
| (override)); |
| |
| MOCK_METHOD(void, |
| OnPacketRetransmitted, |
| (int retransmitted_bytes), |
| (override)); |
| |
| protected: |
| // Object is ref counted. |
| ~MockAckListener() override; |
| }; |
| |
| class MockNetworkChangeVisitor |
| : public QuicSentPacketManager::NetworkChangeVisitor { |
| public: |
| MockNetworkChangeVisitor(); |
| MockNetworkChangeVisitor(const MockNetworkChangeVisitor&) = delete; |
| MockNetworkChangeVisitor& operator=(const MockNetworkChangeVisitor&) = delete; |
| ~MockNetworkChangeVisitor() override; |
| |
| MOCK_METHOD(void, OnCongestionChange, (), (override)); |
| MOCK_METHOD(void, OnPathMtuIncreased, (QuicPacketLength), (override)); |
| }; |
| |
| class MockQuicConnectionDebugVisitor : public QuicConnectionDebugVisitor { |
| public: |
| MockQuicConnectionDebugVisitor(); |
| ~MockQuicConnectionDebugVisitor() override; |
| |
| MOCK_METHOD(void, |
| OnPacketSent, |
| (QuicPacketNumber, |
| QuicPacketLength, |
| bool, |
| TransmissionType, |
| EncryptionLevel, |
| const QuicFrames&, |
| const QuicFrames&, |
| QuicTime), |
| (override)); |
| |
| MOCK_METHOD(void, |
| OnCoalescedPacketSent, |
| (const QuicCoalescedPacket&, size_t), |
| (override)); |
| |
| MOCK_METHOD(void, OnPingSent, (), (override)); |
| |
| MOCK_METHOD(void, |
| OnPacketReceived, |
| (const QuicSocketAddress&, |
| const QuicSocketAddress&, |
| const QuicEncryptedPacket&), |
| (override)); |
| |
| MOCK_METHOD(void, OnIncorrectConnectionId, (QuicConnectionId), (override)); |
| |
| MOCK_METHOD(void, OnProtocolVersionMismatch, (ParsedQuicVersion), (override)); |
| |
| MOCK_METHOD(void, |
| OnPacketHeader, |
| (const QuicPacketHeader& header, |
| QuicTime receive_time, |
| EncryptionLevel level), |
| (override)); |
| |
| MOCK_METHOD(void, |
| OnSuccessfulVersionNegotiation, |
| (const ParsedQuicVersion&), |
| (override)); |
| |
| MOCK_METHOD(void, OnStreamFrame, (const QuicStreamFrame&), (override)); |
| |
| MOCK_METHOD(void, OnCryptoFrame, (const QuicCryptoFrame&), (override)); |
| |
| MOCK_METHOD(void, |
| OnStopWaitingFrame, |
| (const QuicStopWaitingFrame&), |
| (override)); |
| |
| MOCK_METHOD(void, OnRstStreamFrame, (const QuicRstStreamFrame&), (override)); |
| |
| MOCK_METHOD(void, |
| OnConnectionCloseFrame, |
| (const QuicConnectionCloseFrame&), |
| (override)); |
| |
| MOCK_METHOD(void, OnBlockedFrame, (const QuicBlockedFrame&), (override)); |
| |
| MOCK_METHOD(void, |
| OnNewConnectionIdFrame, |
| (const QuicNewConnectionIdFrame&), |
| (override)); |
| |
| MOCK_METHOD(void, |
| OnRetireConnectionIdFrame, |
| (const QuicRetireConnectionIdFrame&), |
| (override)); |
| |
| MOCK_METHOD(void, OnNewTokenFrame, (const QuicNewTokenFrame&), (override)); |
| |
| MOCK_METHOD(void, OnMessageFrame, (const QuicMessageFrame&), (override)); |
| |
| MOCK_METHOD(void, |
| OnStopSendingFrame, |
| (const QuicStopSendingFrame&), |
| (override)); |
| |
| MOCK_METHOD(void, |
| OnPathChallengeFrame, |
| (const QuicPathChallengeFrame&), |
| (override)); |
| |
| MOCK_METHOD(void, |
| OnPathResponseFrame, |
| (const QuicPathResponseFrame&), |
| (override)); |
| |
| MOCK_METHOD(void, |
| OnPublicResetPacket, |
| (const QuicPublicResetPacket&), |
| (override)); |
| |
| MOCK_METHOD(void, |
| OnVersionNegotiationPacket, |
| (const QuicVersionNegotiationPacket&), |
| (override)); |
| |
| MOCK_METHOD(void, |
| OnTransportParametersSent, |
| (const TransportParameters&), |
| (override)); |
| |
| MOCK_METHOD(void, |
| OnTransportParametersReceived, |
| (const TransportParameters&), |
| (override)); |
| |
| MOCK_METHOD(void, OnZeroRttRejected, (int), (override)); |
| MOCK_METHOD(void, OnZeroRttPacketAcked, (), (override)); |
| }; |
| |
| class MockReceivedPacketManager : public QuicReceivedPacketManager { |
| public: |
| explicit MockReceivedPacketManager(QuicConnectionStats* stats); |
| ~MockReceivedPacketManager() override; |
| |
| MOCK_METHOD(void, |
| RecordPacketReceived, |
| (const QuicPacketHeader& header, QuicTime receipt_time), |
| (override)); |
| MOCK_METHOD(bool, IsMissing, (QuicPacketNumber packet_number), (override)); |
| MOCK_METHOD(bool, |
| IsAwaitingPacket, |
| (QuicPacketNumber packet_number), |
| (const, override)); |
| MOCK_METHOD(bool, HasNewMissingPackets, (), (const, override)); |
| MOCK_METHOD(bool, ack_frame_updated, (), (const, override)); |
| }; |
| |
| class MockPacketCreatorDelegate : public QuicPacketCreator::DelegateInterface { |
| public: |
| MockPacketCreatorDelegate(); |
| MockPacketCreatorDelegate(const MockPacketCreatorDelegate&) = delete; |
| MockPacketCreatorDelegate& operator=(const MockPacketCreatorDelegate&) = |
| delete; |
| ~MockPacketCreatorDelegate() override; |
| |
| MOCK_METHOD(QuicPacketBuffer, GetPacketBuffer, (), (override)); |
| MOCK_METHOD(void, OnSerializedPacket, (SerializedPacket), (override)); |
| MOCK_METHOD(void, |
| OnUnrecoverableError, |
| (QuicErrorCode, const std::string&), |
| (override)); |
| MOCK_METHOD(bool, |
| ShouldGeneratePacket, |
| (HasRetransmittableData retransmittable, IsHandshake handshake), |
| (override)); |
| MOCK_METHOD(const QuicFrames, |
| MaybeBundleAckOpportunistically, |
| (), |
| (override)); |
| MOCK_METHOD(SerializedPacketFate, |
| GetSerializedPacketFate, |
| (bool, EncryptionLevel), |
| (override)); |
| }; |
| |
| class MockSessionNotifier : public SessionNotifierInterface { |
| public: |
| MockSessionNotifier(); |
| ~MockSessionNotifier() override; |
| |
| MOCK_METHOD(bool, |
| OnFrameAcked, |
| (const QuicFrame&, QuicTime::Delta, QuicTime), |
| (override)); |
| MOCK_METHOD(void, |
| OnStreamFrameRetransmitted, |
| (const QuicStreamFrame&), |
| (override)); |
| MOCK_METHOD(void, OnFrameLost, (const QuicFrame&), (override)); |
| MOCK_METHOD(void, |
| RetransmitFrames, |
| (const QuicFrames&, TransmissionType type), |
| (override)); |
| MOCK_METHOD(bool, IsFrameOutstanding, (const QuicFrame&), (const, override)); |
| MOCK_METHOD(bool, HasUnackedCryptoData, (), (const, override)); |
| MOCK_METHOD(bool, HasUnackedStreamData, (), (const, override)); |
| }; |
| |
| class MockQuicPathValidationContext : public QuicPathValidationContext { |
| public: |
| MockQuicPathValidationContext(const QuicSocketAddress& self_address, |
| const QuicSocketAddress& peer_address, |
| const QuicSocketAddress& effective_peer_address, |
| QuicPacketWriter* writer) |
| : QuicPathValidationContext(self_address, |
| peer_address, |
| effective_peer_address), |
| writer_(writer) {} |
| QuicPacketWriter* WriterToUse() override { return writer_; } |
| |
| private: |
| QuicPacketWriter* writer_; |
| }; |
| |
| class MockQuicPathValidationResultDelegate |
| : public QuicPathValidator::ResultDelegate { |
| public: |
| MOCK_METHOD(void, |
| OnPathValidationSuccess, |
| (std::unique_ptr<QuicPathValidationContext>), |
| (override)); |
| |
| MOCK_METHOD(void, |
| OnPathValidationFailure, |
| (std::unique_ptr<QuicPathValidationContext>), |
| (override)); |
| }; |
| |
| class QuicCryptoClientStreamPeer { |
| public: |
| QuicCryptoClientStreamPeer() = delete; |
| |
| static QuicCryptoClientStream::HandshakerInterface* GetHandshaker( |
| QuicCryptoClientStream* stream); |
| }; |
| |
| // Creates a client session for testing. |
| // |
| // server_id: The server id associated with this stream. |
| // connection_start_time: The time to set for the connection clock. |
| // Needed for strike-register nonce verification. The client |
| // connection_start_time should be synchronized witht the server |
| // start time, otherwise nonce verification will fail. |
| // supported_versions: Set of QUIC versions this client supports. |
| // helper: Pointer to the MockQuicConnectionHelper to use for the session. |
| // crypto_client_config: Pointer to the crypto client config. |
| // client_connection: Pointer reference for newly created |
| // connection. This object will be owned by the |
| // client_session. |
| // client_session: Pointer reference for the newly created client |
| // session. The new object will be owned by the caller. |
| void CreateClientSessionForTest( |
| QuicServerId server_id, |
| QuicTime::Delta connection_start_time, |
| const ParsedQuicVersionVector& supported_versions, |
| MockQuicConnectionHelper* helper, |
| MockAlarmFactory* alarm_factory, |
| QuicCryptoClientConfig* crypto_client_config, |
| PacketSavingConnection** client_connection, |
| TestQuicSpdyClientSession** client_session); |
| |
| // Creates a server session for testing. |
| // |
| // server_id: The server id associated with this stream. |
| // connection_start_time: The time to set for the connection clock. |
| // Needed for strike-register nonce verification. The server |
| // connection_start_time should be synchronized witht the client |
| // start time, otherwise nonce verification will fail. |
| // supported_versions: Set of QUIC versions this server supports. |
| // helper: Pointer to the MockQuicConnectionHelper to use for the session. |
| // crypto_server_config: Pointer to the crypto server config. |
| // server_connection: Pointer reference for newly created |
| // connection. This object will be owned by the |
| // server_session. |
| // server_session: Pointer reference for the newly created server |
| // session. The new object will be owned by the caller. |
| void CreateServerSessionForTest( |
| QuicServerId server_id, |
| QuicTime::Delta connection_start_time, |
| ParsedQuicVersionVector supported_versions, |
| MockQuicConnectionHelper* helper, |
| MockAlarmFactory* alarm_factory, |
| QuicCryptoServerConfig* crypto_server_config, |
| QuicCompressedCertsCache* compressed_certs_cache, |
| PacketSavingConnection** server_connection, |
| TestQuicSpdyServerSession** server_session); |
| |
| // Verifies that the relative error of |actual| with respect to |expected| is |
| // no more than |margin|. |
| // Please use EXPECT_APPROX_EQ, a wrapper around this function, for better error |
| // report. |
| template <typename T> |
| void ExpectApproxEq(T expected, T actual, float relative_margin) { |
| // If |relative_margin| > 1 and T is an unsigned type, the comparison will |
| // underflow. |
| ASSERT_LE(relative_margin, 1); |
| ASSERT_GE(relative_margin, 0); |
| |
| T absolute_margin = expected * relative_margin; |
| |
| EXPECT_GE(expected + absolute_margin, actual) << "actual value too big"; |
| EXPECT_LE(expected - absolute_margin, actual) << "actual value too small"; |
| } |
| |
| #define EXPECT_APPROX_EQ(expected, actual, relative_margin) \ |
| do { \ |
| SCOPED_TRACE(testing::Message() << "relative_margin:" << relative_margin); \ |
| quic::test::ExpectApproxEq(expected, actual, relative_margin); \ |
| } while (0) |
| |
| template <typename T> |
| QuicHeaderList AsHeaderList(const T& container) { |
| QuicHeaderList l; |
| l.OnHeaderBlockStart(); |
| size_t total_size = 0; |
| for (auto p : container) { |
| total_size += p.first.size() + p.second.size(); |
| l.OnHeader(p.first, p.second); |
| } |
| l.OnHeaderBlockEnd(total_size, total_size); |
| return l; |
| } |
| |
| // Utility function that stores |str|'s data in |iov|. |
| inline void MakeIOVector(absl::string_view str, struct iovec* iov) { |
| iov->iov_base = const_cast<char*>(str.data()); |
| iov->iov_len = static_cast<size_t>(str.size()); |
| } |
| |
| // Helper functions for stream ids, to allow test logic to abstract over the |
| // HTTP stream numbering scheme (i.e. whether one or two QUIC streams are used |
| // per HTTP transaction). |
| QuicStreamId GetNthClientInitiatedBidirectionalStreamId( |
| QuicTransportVersion version, |
| int n); |
| QuicStreamId GetNthServerInitiatedBidirectionalStreamId( |
| QuicTransportVersion version, |
| int n); |
| QuicStreamId GetNthServerInitiatedUnidirectionalStreamId( |
| QuicTransportVersion version, |
| int n); |
| QuicStreamId GetNthClientInitiatedUnidirectionalStreamId( |
| QuicTransportVersion version, |
| int n); |
| |
| StreamType DetermineStreamType(QuicStreamId id, |
| ParsedQuicVersion version, |
| Perspective perspective, |
| bool is_incoming, |
| StreamType default_type); |
| |
| // Utility function that stores message_data in |storage| and returns a |
| // QuicMemSliceSpan. |
| QuicMemSliceSpan MakeSpan(QuicBufferAllocator* allocator, |
| absl::string_view message_data, |
| QuicMemSliceStorage* storage); |
| |
| // Creates a MemSlice using a singleton trivial buffer allocator. Performs a |
| // copy. |
| QuicMemSlice MemSliceFromString(absl::string_view data); |
| |
| // Used to compare ReceivedPacketInfo. |
| MATCHER_P(ReceivedPacketInfoEquals, info, "") { |
| return info.ToString() == arg.ToString(); |
| } |
| |
| MATCHER_P(ReceivedPacketInfoConnectionIdEquals, destination_connection_id, "") { |
| return arg.destination_connection_id == destination_connection_id; |
| } |
| |
| MATCHER_P2(InRange, min, max, "") { |
| return arg >= min && arg <= max; |
| } |
| |
| // A GMock matcher that prints expected and actual QuicErrorCode strings |
| // upon failure. Example usage: |
| // EXPECT_THAT(stream_->connection_error(), IsError(QUIC_INTERNAL_ERROR)); |
| MATCHER_P(IsError, |
| expected, |
| absl::StrCat(negation ? "isn't equal to " : "is equal to ", |
| QuicErrorCodeToString(expected))) { |
| *result_listener << QuicErrorCodeToString(static_cast<QuicErrorCode>(arg)); |
| return arg == expected; |
| } |
| |
| // Shorthand for IsError(QUIC_NO_ERROR). |
| // Example usage: EXPECT_THAT(stream_->connection_error(), IsQuicNoError()); |
| MATCHER(IsQuicNoError, |
| absl::StrCat(negation ? "isn't equal to " : "is equal to ", |
| QuicErrorCodeToString(QUIC_NO_ERROR))) { |
| *result_listener << QuicErrorCodeToString(arg); |
| return arg == QUIC_NO_ERROR; |
| } |
| |
| // A GMock matcher that prints expected and actual QuicRstStreamErrorCode |
| // strings upon failure. Example usage: |
| // EXPECT_THAT(stream_->stream_error(), IsStreamError(QUIC_INTERNAL_ERROR)); |
| MATCHER_P(IsStreamError, |
| expected, |
| absl::StrCat(negation ? "isn't equal to " : "is equal to ", |
| QuicRstStreamErrorCodeToString(expected))) { |
| *result_listener << QuicRstStreamErrorCodeToString(arg); |
| return arg == expected; |
| } |
| |
| // Shorthand for IsStreamError(QUIC_STREAM_NO_ERROR). Example usage: |
| // EXPECT_THAT(stream_->stream_error(), IsQuicStreamNoError()); |
| MATCHER(IsQuicStreamNoError, |
| absl::StrCat(negation ? "isn't equal to " : "is equal to ", |
| QuicRstStreamErrorCodeToString(QUIC_STREAM_NO_ERROR))) { |
| *result_listener << QuicRstStreamErrorCodeToString(arg); |
| return arg == QUIC_STREAM_NO_ERROR; |
| } |
| |
| // TaggingEncrypter appends kTagSize bytes of |tag| to the end of each message. |
| class TaggingEncrypter : public QuicEncrypter { |
| public: |
| explicit TaggingEncrypter(uint8_t tag) : tag_(tag) {} |
| TaggingEncrypter(const TaggingEncrypter&) = delete; |
| TaggingEncrypter& operator=(const TaggingEncrypter&) = delete; |
| |
| ~TaggingEncrypter() override {} |
| |
| // QuicEncrypter interface. |
| bool SetKey(absl::string_view /*key*/) override { return true; } |
| |
| bool SetNoncePrefix(absl::string_view /*nonce_prefix*/) override { |
| return true; |
| } |
| |
| bool SetIV(absl::string_view /*iv*/) override { return true; } |
| |
| bool SetHeaderProtectionKey(absl::string_view /*key*/) override { |
| return true; |
| } |
| |
| bool EncryptPacket(uint64_t packet_number, |
| absl::string_view associated_data, |
| absl::string_view plaintext, |
| char* output, |
| size_t* output_length, |
| size_t max_output_length) override; |
| |
| std::string GenerateHeaderProtectionMask( |
| absl::string_view /*sample*/) override { |
| return std::string(5, 0); |
| } |
| |
| size_t GetKeySize() const override { return 0; } |
| size_t GetNoncePrefixSize() const override { return 0; } |
| size_t GetIVSize() const override { return 0; } |
| |
| size_t GetMaxPlaintextSize(size_t ciphertext_size) const override { |
| return ciphertext_size - kTagSize; |
| } |
| |
| size_t GetCiphertextSize(size_t plaintext_size) const override { |
| return plaintext_size + kTagSize; |
| } |
| |
| QuicPacketCount GetConfidentialityLimit() const override { |
| return std::numeric_limits<QuicPacketCount>::max(); |
| } |
| |
| absl::string_view GetKey() const override { return absl::string_view(); } |
| |
| absl::string_view GetNoncePrefix() const override { |
| return absl::string_view(); |
| } |
| |
| private: |
| enum { |
| kTagSize = 12, |
| }; |
| |
| const uint8_t tag_; |
| }; |
| |
| // TaggingDecrypter ensures that the final kTagSize bytes of the message all |
| // have the same value and then removes them. |
| class TaggingDecrypter : public QuicDecrypter { |
| public: |
| ~TaggingDecrypter() override {} |
| |
| // QuicDecrypter interface |
| bool SetKey(absl::string_view /*key*/) override { return true; } |
| |
| bool SetNoncePrefix(absl::string_view /*nonce_prefix*/) override { |
| return true; |
| } |
| |
| bool SetIV(absl::string_view /*iv*/) override { return true; } |
| |
| bool SetHeaderProtectionKey(absl::string_view /*key*/) override { |
| return true; |
| } |
| |
| bool SetPreliminaryKey(absl::string_view /*key*/) override { |
| QUIC_BUG << "should not be called"; |
| return false; |
| } |
| |
| bool SetDiversificationNonce(const DiversificationNonce& /*key*/) override { |
| return true; |
| } |
| |
| bool DecryptPacket(uint64_t packet_number, |
| absl::string_view associated_data, |
| absl::string_view ciphertext, |
| char* output, |
| size_t* output_length, |
| size_t max_output_length) override; |
| |
| std::string GenerateHeaderProtectionMask( |
| QuicDataReader* /*sample_reader*/) override { |
| return std::string(5, 0); |
| } |
| |
| size_t GetKeySize() const override { return 0; } |
| size_t GetNoncePrefixSize() const override { return 0; } |
| size_t GetIVSize() const override { return 0; } |
| absl::string_view GetKey() const override { return absl::string_view(); } |
| absl::string_view GetNoncePrefix() const override { |
| return absl::string_view(); |
| } |
| // Use a distinct value starting with 0xFFFFFF, which is never used by TLS. |
| uint32_t cipher_id() const override { return 0xFFFFFFF0; } |
| QuicPacketCount GetIntegrityLimit() const override { |
| return std::numeric_limits<QuicPacketCount>::max(); |
| } |
| |
| protected: |
| virtual uint8_t GetTag(absl::string_view ciphertext) { |
| return ciphertext.data()[ciphertext.size() - 1]; |
| } |
| |
| private: |
| enum { |
| kTagSize = 12, |
| }; |
| |
| bool CheckTag(absl::string_view ciphertext, uint8_t tag); |
| }; |
| |
| // StringTaggingDecrypter ensures that the final kTagSize bytes of the message |
| // match the expected value. |
| class StrictTaggingDecrypter : public TaggingDecrypter { |
| public: |
| explicit StrictTaggingDecrypter(uint8_t tag) : tag_(tag) {} |
| ~StrictTaggingDecrypter() override {} |
| |
| // TaggingQuicDecrypter |
| uint8_t GetTag(absl::string_view /*ciphertext*/) override { return tag_; } |
| |
| // Use a distinct value starting with 0xFFFFFF, which is never used by TLS. |
| uint32_t cipher_id() const override { return 0xFFFFFFF1; } |
| |
| private: |
| const uint8_t tag_; |
| }; |
| |
| class TestPacketWriter : public QuicPacketWriter { |
| struct PacketBuffer { |
| ABSL_CACHELINE_ALIGNED char buffer[1500]; |
| bool in_use = false; |
| }; |
| |
| public: |
| TestPacketWriter(ParsedQuicVersion version, |
| MockClock* clock, |
| Perspective perspective); |
| |
| TestPacketWriter(const TestPacketWriter&) = delete; |
| TestPacketWriter& operator=(const TestPacketWriter&) = delete; |
| |
| ~TestPacketWriter() override; |
| |
| // QuicPacketWriter interface |
| WriteResult WritePacket(const char* buffer, |
| size_t buf_len, |
| const QuicIpAddress& self_address, |
| const QuicSocketAddress& peer_address, |
| PerPacketOptions* options) override; |
| |
| bool ShouldWriteFail() { return write_should_fail_; } |
| |
| bool IsWriteBlocked() const override { return write_blocked_; } |
| |
| void SetWriteBlocked() { write_blocked_ = true; } |
| |
| void SetWritable() override { write_blocked_ = false; } |
| |
| void SetShouldWriteFail() { write_should_fail_ = true; } |
| |
| void SetWriteError(int error_code) { write_error_code_ = error_code; } |
| |
| QuicByteCount GetMaxPacketSize( |
| const QuicSocketAddress& /*peer_address*/) const override { |
| return max_packet_size_; |
| } |
| |
| bool SupportsReleaseTime() const override { return supports_release_time_; } |
| |
| bool IsBatchMode() const override { return is_batch_mode_; } |
| |
| QuicPacketBuffer GetNextWriteLocation( |
| const QuicIpAddress& /*self_address*/, |
| const QuicSocketAddress& /*peer_address*/) override; |
| |
| WriteResult Flush() override; |
| |
| void BlockOnNextFlush() { block_on_next_flush_ = true; } |
| |
| void BlockOnNextWrite() { block_on_next_write_ = true; } |
| |
| void SimulateNextPacketTooLarge() { next_packet_too_large_ = true; } |
| |
| void AlwaysGetPacketTooLarge() { always_get_packet_too_large_ = true; } |
| |
| // Sets the amount of time that the writer should before the actual write. |
| void SetWritePauseTimeDelta(QuicTime::Delta delta) { |
| write_pause_time_delta_ = delta; |
| } |
| |
| void SetBatchMode(bool new_value) { is_batch_mode_ = new_value; } |
| |
| const QuicPacketHeader& header() { return framer_.header(); } |
| |
| size_t frame_count() const { return framer_.num_frames(); } |
| |
| const std::vector<QuicAckFrame>& ack_frames() const { |
| return framer_.ack_frames(); |
| } |
| |
| const std::vector<QuicStopWaitingFrame>& stop_waiting_frames() const { |
| return framer_.stop_waiting_frames(); |
| } |
| |
| const std::vector<QuicConnectionCloseFrame>& connection_close_frames() const { |
| return framer_.connection_close_frames(); |
| } |
| |
| const std::vector<QuicRstStreamFrame>& rst_stream_frames() const { |
| return framer_.rst_stream_frames(); |
| } |
| |
| const std::vector<std::unique_ptr<QuicStreamFrame>>& stream_frames() const { |
| return framer_.stream_frames(); |
| } |
| |
| const std::vector<std::unique_ptr<QuicCryptoFrame>>& crypto_frames() const { |
| return framer_.crypto_frames(); |
| } |
| |
| const std::vector<QuicPingFrame>& ping_frames() const { |
| return framer_.ping_frames(); |
| } |
| |
| const std::vector<QuicMessageFrame>& message_frames() const { |
| return framer_.message_frames(); |
| } |
| |
| const std::vector<QuicWindowUpdateFrame>& window_update_frames() const { |
| return framer_.window_update_frames(); |
| } |
| |
| const std::vector<QuicPaddingFrame>& padding_frames() const { |
| return framer_.padding_frames(); |
| } |
| |
| const std::vector<QuicPathChallengeFrame>& path_challenge_frames() const { |
| return framer_.path_challenge_frames(); |
| } |
| |
| const std::vector<QuicPathResponseFrame>& path_response_frames() const { |
| return framer_.path_response_frames(); |
| } |
| |
| const QuicEncryptedPacket* coalesced_packet() const { |
| return framer_.coalesced_packet(); |
| } |
| |
| size_t last_packet_size() { return last_packet_size_; } |
| |
| const QuicPacketHeader& last_packet_header() const { |
| return last_packet_header_; |
| } |
| |
| const QuicVersionNegotiationPacket* version_negotiation_packet() { |
| return framer_.version_negotiation_packet(); |
| } |
| |
| void set_is_write_blocked_data_buffered(bool buffered) { |
| is_write_blocked_data_buffered_ = buffered; |
| } |
| |
| void set_perspective(Perspective perspective) { |
| // We invert perspective here, because the framer needs to parse packets |
| // we send. |
| QuicFramerPeer::SetPerspective(framer_.framer(), |
| QuicUtils::InvertPerspective(perspective)); |
| framer_.framer()->SetInitialObfuscators(TestConnectionId()); |
| } |
| |
| // final_bytes_of_last_packet_ returns the last four bytes of the previous |
| // packet as a little-endian, uint32_t. This is intended to be used with a |
| // TaggingEncrypter so that tests can determine which encrypter was used for |
| // a given packet. |
| uint32_t final_bytes_of_last_packet() { return final_bytes_of_last_packet_; } |
| |
| // Returns the final bytes of the second to last packet. |
| uint32_t final_bytes_of_previous_packet() { |
| return final_bytes_of_previous_packet_; |
| } |
| |
| void use_tagging_decrypter() { use_tagging_decrypter_ = true; } |
| |
| uint32_t packets_write_attempts() const { return packets_write_attempts_; } |
| |
| uint32_t flush_attempts() const { return flush_attempts_; } |
| |
| uint32_t connection_close_packets() const { |
| return connection_close_packets_; |
| } |
| |
| void Reset() { framer_.Reset(); } |
| |
| void SetSupportedVersions(const ParsedQuicVersionVector& versions) { |
| framer_.SetSupportedVersions(versions); |
| } |
| |
| void set_max_packet_size(QuicByteCount max_packet_size) { |
| max_packet_size_ = max_packet_size; |
| } |
| |
| void set_supports_release_time(bool supports_release_time) { |
| supports_release_time_ = supports_release_time; |
| } |
| |
| SimpleQuicFramer* framer() { return &framer_; } |
| |
| const QuicIpAddress& last_write_source_address() const { |
| return last_write_source_address_; |
| } |
| |
| const QuicSocketAddress& last_write_peer_address() const { |
| return last_write_peer_address_; |
| } |
| |
| private: |
| char* AllocPacketBuffer(); |
| |
| void FreePacketBuffer(const char* buffer); |
| |
| ParsedQuicVersion version_; |
| SimpleQuicFramer framer_; |
| size_t last_packet_size_ = 0; |
| QuicPacketHeader last_packet_header_; |
| bool write_blocked_ = false; |
| bool write_should_fail_ = false; |
| bool block_on_next_flush_ = false; |
| bool block_on_next_write_ = false; |
| bool next_packet_too_large_ = false; |
| bool always_get_packet_too_large_ = false; |
| bool is_write_blocked_data_buffered_ = false; |
| bool is_batch_mode_ = false; |
| // Number of times Flush() was called. |
| uint32_t flush_attempts_ = 0; |
| // (Batch mode only) Number of bytes buffered in writer. It is used as the |
| // return value of a successful Flush(). |
| uint32_t bytes_buffered_ = 0; |
| uint32_t final_bytes_of_last_packet_ = 0; |
| uint32_t final_bytes_of_previous_packet_ = 0; |
| bool use_tagging_decrypter_ = false; |
| uint32_t packets_write_attempts_ = 0; |
| uint32_t connection_close_packets_ = 0; |
| MockClock* clock_ = nullptr; |
| // If non-zero, the clock will pause during WritePacket for this amount of |
| // time. |
| QuicTime::Delta write_pause_time_delta_ = QuicTime::Delta::Zero(); |
| QuicByteCount max_packet_size_ = kMaxOutgoingPacketSize; |
| bool supports_release_time_ = false; |
| // Used to verify writer-allocated packet buffers are properly released. |
| std::vector<PacketBuffer*> packet_buffer_pool_; |
| // Buffer address => Address of the owning PacketBuffer. |
| QuicHashMap<char*, PacketBuffer*> packet_buffer_pool_index_; |
| // Indices in packet_buffer_pool_ that are not allocated. |
| std::list<PacketBuffer*> packet_buffer_free_list_; |
| // The soruce/peer address passed into WritePacket(). |
| QuicIpAddress last_write_source_address_; |
| QuicSocketAddress last_write_peer_address_; |
| int write_error_code_{0}; |
| }; |
| |
| // Parses a packet generated by |
| // QuicFramer::WriteClientVersionNegotiationProbePacket. |
| // |packet_bytes| must point to |packet_length| bytes in memory which represent |
| // the packet. This method will fill in |destination_connection_id_bytes| |
| // which must point to at least |*destination_connection_id_length_out| bytes in |
| // memory. |*destination_connection_id_length_out| will contain the length of |
| // the received destination connection ID, which on success will match the |
| // contents of the destination connection ID passed in to |
| // WriteClientVersionNegotiationProbePacket. |
| bool ParseClientVersionNegotiationProbePacket( |
| const char* packet_bytes, |
| size_t packet_length, |
| char* destination_connection_id_bytes, |
| uint8_t* destination_connection_id_length_out); |
| |
| // Writes an array of bytes that correspond to a QUIC version negotiation packet |
| // that a QUIC server would send in response to a probe created by |
| // QuicFramer::WriteClientVersionNegotiationProbePacket. |
| // The bytes will be written to |packet_bytes|, which must point to |
| // |*packet_length_out| bytes of memory. |*packet_length_out| will contain the |
| // length of the created packet. |source_connection_id_bytes| will be sent as |
| // the source connection ID, and must point to |source_connection_id_length| |
| // bytes of memory. |
| bool WriteServerVersionNegotiationProbeResponse( |
| char* packet_bytes, |
| size_t* packet_length_out, |
| const char* source_connection_id_bytes, |
| uint8_t source_connection_id_length); |
| |
| } // namespace test |
| } // namespace quic |
| |
| #endif // QUICHE_QUIC_TEST_TOOLS_QUIC_TEST_UTILS_H_ |