| OLD | NEW |
| (Empty) |
| 1 // Copyright 2013 The Chromium Authors. All rights reserved. | |
| 2 // Use of this source code is governed by a BSD-style license that can be | |
| 3 // found in the LICENSE file. | |
| 4 // | |
| 5 // Provides a simple interface for QUIC tests to create a variety of packets. | |
| 6 | |
| 7 #ifndef NET_QUIC_TEST_TOOLS_QUIC_TEST_PACKET_MAKER_H_ | |
| 8 #define NET_QUIC_TEST_TOOLS_QUIC_TEST_PACKET_MAKER_H_ | |
| 9 | |
| 10 #include "base/memory/scoped_ptr.h" | |
| 11 #include "net/base/request_priority.h" | |
| 12 #include "net/quic/quic_protocol.h" | |
| 13 #include "net/quic/test_tools/mock_clock.h" | |
| 14 #include "net/quic/test_tools/mock_random.h" | |
| 15 #include "net/spdy/spdy_framer.h" | |
| 16 #include "net/spdy/spdy_protocol.h" | |
| 17 | |
| 18 namespace net { | |
| 19 namespace test { | |
| 20 | |
| 21 class QuicTestPacketMaker { | |
| 22 public: | |
| 23 QuicTestPacketMaker(QuicVersion version, | |
| 24 QuicConnectionId connection_id, | |
| 25 MockClock* clock); | |
| 26 ~QuicTestPacketMaker(); | |
| 27 | |
| 28 scoped_ptr<QuicEncryptedPacket> MakeRstPacket( | |
| 29 QuicPacketSequenceNumber num, | |
| 30 bool include_version, | |
| 31 QuicStreamId stream_id, | |
| 32 QuicRstStreamErrorCode error_code); | |
| 33 scoped_ptr<QuicEncryptedPacket> MakeAckAndRstPacket( | |
| 34 QuicPacketSequenceNumber num, | |
| 35 bool include_version, | |
| 36 QuicStreamId stream_id, | |
| 37 QuicRstStreamErrorCode error_code, | |
| 38 QuicPacketSequenceNumber largest_received, | |
| 39 QuicPacketSequenceNumber least_unacked, | |
| 40 bool send_feedback); | |
| 41 scoped_ptr<QuicEncryptedPacket> MakeConnectionClosePacket( | |
| 42 QuicPacketSequenceNumber num); | |
| 43 scoped_ptr<QuicEncryptedPacket> MakeAckPacket( | |
| 44 QuicPacketSequenceNumber sequence_number, | |
| 45 QuicPacketSequenceNumber largest_received, | |
| 46 QuicPacketSequenceNumber least_unacked, | |
| 47 bool send_feedback); | |
| 48 scoped_ptr<QuicEncryptedPacket> MakeDataPacket( | |
| 49 QuicPacketSequenceNumber sequence_number, | |
| 50 QuicStreamId stream_id, | |
| 51 bool should_include_version, | |
| 52 bool fin, | |
| 53 QuicStreamOffset offset, | |
| 54 base::StringPiece data); | |
| 55 scoped_ptr<QuicEncryptedPacket> MakeRequestHeadersPacket( | |
| 56 QuicPacketSequenceNumber sequence_number, | |
| 57 QuicStreamId stream_id, | |
| 58 bool should_include_version, | |
| 59 bool fin, | |
| 60 QuicPriority priority, | |
| 61 const SpdyHeaderBlock& headers); | |
| 62 scoped_ptr<QuicEncryptedPacket> MakeResponseHeadersPacket( | |
| 63 QuicPacketSequenceNumber sequence_number, | |
| 64 QuicStreamId stream_id, | |
| 65 bool should_include_version, | |
| 66 bool fin, | |
| 67 const SpdyHeaderBlock& headers); | |
| 68 | |
| 69 SpdyHeaderBlock GetRequestHeaders(const std::string& method, | |
| 70 const std::string& scheme, | |
| 71 const std::string& path); | |
| 72 SpdyHeaderBlock GetResponseHeaders(const std::string& status); | |
| 73 | |
| 74 private: | |
| 75 scoped_ptr<QuicEncryptedPacket> MakePacket( | |
| 76 const QuicPacketHeader& header, | |
| 77 const QuicFrame& frame); | |
| 78 | |
| 79 void InitializeHeader(QuicPacketSequenceNumber sequence_number, | |
| 80 bool should_include_version); | |
| 81 | |
| 82 QuicVersion version_; | |
| 83 QuicConnectionId connection_id_; | |
| 84 MockClock* clock_; // Owned by QuicStreamFactory. | |
| 85 SpdyFramer spdy_request_framer_; | |
| 86 SpdyFramer spdy_response_framer_; | |
| 87 MockRandom random_generator_; | |
| 88 QuicPacketHeader header_; | |
| 89 | |
| 90 DISALLOW_COPY_AND_ASSIGN(QuicTestPacketMaker); | |
| 91 }; | |
| 92 | |
| 93 } // namespace test | |
| 94 } // namespace net | |
| 95 | |
| 96 #endif // NET_QUIC_TEST_TOOLS_QUIC_TEST_PACKET_MAKER_H_ | |
| OLD | NEW |