OLD | NEW |
1 // Copyright 2013 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 // | 4 // |
5 // Provides a simple interface for QUIC tests to create a variety of packets. | 5 // Provides a simple interface for QUIC tests to create a variety of packets. |
6 | 6 |
7 #ifndef NET_QUIC_TEST_TOOLS_QUIC_TEST_PACKET_MAKER_H_ | 7 #ifndef NET_QUIC_TEST_TOOLS_QUIC_TEST_PACKET_MAKER_H_ |
8 #define NET_QUIC_TEST_TOOLS_QUIC_TEST_PACKET_MAKER_H_ | 8 #define NET_QUIC_TEST_TOOLS_QUIC_TEST_PACKET_MAKER_H_ |
9 | 9 |
10 #include <stddef.h> | 10 #include <stddef.h> |
(...skipping 66 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
77 QuicPacketNumber largest_received, | 77 QuicPacketNumber largest_received, |
78 QuicPacketNumber ack_least_unacked, | 78 QuicPacketNumber ack_least_unacked, |
79 QuicPacketNumber stop_least_unacked, | 79 QuicPacketNumber stop_least_unacked, |
80 bool send_feedback); | 80 bool send_feedback); |
81 scoped_ptr<QuicReceivedPacket> MakeDataPacket(QuicPacketNumber packet_number, | 81 scoped_ptr<QuicReceivedPacket> MakeDataPacket(QuicPacketNumber packet_number, |
82 QuicStreamId stream_id, | 82 QuicStreamId stream_id, |
83 bool should_include_version, | 83 bool should_include_version, |
84 bool fin, | 84 bool fin, |
85 QuicStreamOffset offset, | 85 QuicStreamOffset offset, |
86 base::StringPiece data); | 86 base::StringPiece data); |
| 87 scoped_ptr<QuicReceivedPacket> MakeMultipleDataFramesPacket( |
| 88 QuicPacketNumber packet_number, |
| 89 QuicStreamId stream_id, |
| 90 bool should_include_version, |
| 91 bool fin, |
| 92 QuicStreamOffset offset, |
| 93 const std::vector<std::string>& data_writes); |
87 scoped_ptr<QuicReceivedPacket> MakeAckAndDataPacket( | 94 scoped_ptr<QuicReceivedPacket> MakeAckAndDataPacket( |
88 QuicPacketNumber packet_number, | 95 QuicPacketNumber packet_number, |
89 bool include_version, | 96 bool include_version, |
90 QuicStreamId stream_id, | 97 QuicStreamId stream_id, |
91 QuicPacketNumber largest_received, | 98 QuicPacketNumber largest_received, |
92 QuicPacketNumber least_unacked, | 99 QuicPacketNumber least_unacked, |
93 bool fin, | 100 bool fin, |
94 QuicStreamOffset offset, | 101 QuicStreamOffset offset, |
95 base::StringPiece data); | 102 base::StringPiece data); |
96 | 103 |
| 104 scoped_ptr<QuicReceivedPacket> MakeRequestHeadersAndMultipleDataFramesPacket( |
| 105 QuicPacketNumber packet_number, |
| 106 QuicStreamId stream_id, |
| 107 bool should_include_version, |
| 108 bool fin, |
| 109 SpdyPriority priority, |
| 110 const SpdyHeaderBlock& headers, |
| 111 size_t* spdy_headers_frame_length, |
| 112 const std::vector<std::string>& data_writes); |
| 113 |
97 // If |spdy_headers_frame_length| is non-null, it will be set to the size of | 114 // If |spdy_headers_frame_length| is non-null, it will be set to the size of |
98 // the SPDY headers frame created for this packet. | 115 // the SPDY headers frame created for this packet. |
99 scoped_ptr<QuicReceivedPacket> MakeRequestHeadersPacket( | 116 scoped_ptr<QuicReceivedPacket> MakeRequestHeadersPacket( |
100 QuicPacketNumber packet_number, | 117 QuicPacketNumber packet_number, |
101 QuicStreamId stream_id, | 118 QuicStreamId stream_id, |
102 bool should_include_version, | 119 bool should_include_version, |
103 bool fin, | 120 bool fin, |
104 SpdyPriority priority, | 121 SpdyPriority priority, |
105 const SpdyHeaderBlock& headers, | 122 const SpdyHeaderBlock& headers, |
106 size_t* spdy_headers_frame_length); | 123 size_t* spdy_headers_frame_length); |
(...skipping 56 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
163 SpdyHeaderBlock GetResponseHeaders(const std::string& status, | 180 SpdyHeaderBlock GetResponseHeaders(const std::string& status, |
164 const std::string& alt_svc); | 181 const std::string& alt_svc); |
165 | 182 |
166 private: | 183 private: |
167 scoped_ptr<QuicReceivedPacket> MakePacket(const QuicPacketHeader& header, | 184 scoped_ptr<QuicReceivedPacket> MakePacket(const QuicPacketHeader& header, |
168 const QuicFrame& frame); | 185 const QuicFrame& frame); |
169 scoped_ptr<QuicReceivedPacket> MakeMultipleFramesPacket( | 186 scoped_ptr<QuicReceivedPacket> MakeMultipleFramesPacket( |
170 const QuicPacketHeader& header, | 187 const QuicPacketHeader& header, |
171 const QuicFrames& frames); | 188 const QuicFrames& frames); |
172 | 189 |
| 190 // Each of |data_writes| will be made into a data frame. |
| 191 QuicFrames MakeMultipleDataFrames( |
| 192 QuicStreamId stream_id, |
| 193 bool fin, |
| 194 QuicStreamOffset offset, |
| 195 const std::vector<std::string>& data_writes); |
| 196 |
173 void InitializeHeader(QuicPacketNumber packet_number, | 197 void InitializeHeader(QuicPacketNumber packet_number, |
174 bool should_include_version); | 198 bool should_include_version); |
175 | 199 |
176 QuicVersion version_; | 200 QuicVersion version_; |
177 QuicConnectionId connection_id_; | 201 QuicConnectionId connection_id_; |
178 MockClock* clock_; // Owned by QuicStreamFactory. | 202 MockClock* clock_; // Owned by QuicStreamFactory. |
179 std::string host_; | 203 std::string host_; |
180 SpdyFramer spdy_request_framer_; | 204 SpdyFramer spdy_request_framer_; |
181 SpdyFramer spdy_response_framer_; | 205 SpdyFramer spdy_response_framer_; |
182 MockRandom random_generator_; | 206 MockRandom random_generator_; |
183 QuicPacketHeader header_; | 207 QuicPacketHeader header_; |
184 | 208 |
185 DISALLOW_COPY_AND_ASSIGN(QuicTestPacketMaker); | 209 DISALLOW_COPY_AND_ASSIGN(QuicTestPacketMaker); |
186 }; | 210 }; |
187 | 211 |
188 } // namespace test | 212 } // namespace test |
189 } // namespace net | 213 } // namespace net |
190 | 214 |
191 #endif // NET_QUIC_TEST_TOOLS_QUIC_TEST_PACKET_MAKER_H_ | 215 #endif // NET_QUIC_TEST_TOOLS_QUIC_TEST_PACKET_MAKER_H_ |
OLD | NEW |