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 #include "net/quic/test_tools/quic_test_packet_maker.h" | 5 #include "net/quic/test_tools/quic_test_packet_maker.h" |
6 | 6 |
7 #include "net/quic/quic_framer.h" | 7 #include "net/quic/quic_framer.h" |
8 #include "net/quic/quic_http_utils.h" | 8 #include "net/quic/quic_http_utils.h" |
9 #include "net/quic/quic_utils.h" | 9 #include "net/quic/quic_utils.h" |
10 #include "net/quic/test_tools/quic_test_utils.h" | 10 #include "net/quic/test_tools/quic_test_utils.h" |
(...skipping 66 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
77 if (version_ > QUIC_VERSION_15) { | 77 if (version_ > QUIC_VERSION_15) { |
78 stop_waiting.least_unacked = least_unacked; | 78 stop_waiting.least_unacked = least_unacked; |
79 frames.push_back(QuicFrame(&stop_waiting)); | 79 frames.push_back(QuicFrame(&stop_waiting)); |
80 } | 80 } |
81 | 81 |
82 QuicRstStreamFrame rst(stream_id, error_code, 0); | 82 QuicRstStreamFrame rst(stream_id, error_code, 0); |
83 frames.push_back(QuicFrame(&rst)); | 83 frames.push_back(QuicFrame(&rst)); |
84 | 84 |
85 QuicFramer framer(SupportedVersions(version_), QuicTime::Zero(), false); | 85 QuicFramer framer(SupportedVersions(version_), QuicTime::Zero(), false); |
86 scoped_ptr<QuicPacket> packet( | 86 scoped_ptr<QuicPacket> packet( |
87 framer.BuildUnsizedDataPacket(header, frames).packet); | 87 BuildUnsizedDataPacket(&framer, header, frames).packet); |
88 return scoped_ptr<QuicEncryptedPacket>(framer.EncryptPacket( | 88 return scoped_ptr<QuicEncryptedPacket>(framer.EncryptPacket( |
89 ENCRYPTION_NONE, header.packet_sequence_number, *packet)); | 89 ENCRYPTION_NONE, header.packet_sequence_number, *packet)); |
90 } | 90 } |
91 | 91 |
92 scoped_ptr<QuicEncryptedPacket> QuicTestPacketMaker::MakeConnectionClosePacket( | 92 scoped_ptr<QuicEncryptedPacket> QuicTestPacketMaker::MakeConnectionClosePacket( |
93 QuicPacketSequenceNumber num) { | 93 QuicPacketSequenceNumber num) { |
94 QuicPacketHeader header; | 94 QuicPacketHeader header; |
95 header.public_header.connection_id = connection_id_; | 95 header.public_header.connection_id = connection_id_; |
96 header.public_header.reset_flag = false; | 96 header.public_header.reset_flag = false; |
97 header.public_header.version_flag = false; | 97 header.public_header.version_flag = false; |
(...skipping 38 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
136 frames.push_back(QuicFrame(&feedback)); | 136 frames.push_back(QuicFrame(&feedback)); |
137 } | 137 } |
138 | 138 |
139 QuicStopWaitingFrame stop_waiting; | 139 QuicStopWaitingFrame stop_waiting; |
140 if (version_ > QUIC_VERSION_15) { | 140 if (version_ > QUIC_VERSION_15) { |
141 stop_waiting.least_unacked = least_unacked; | 141 stop_waiting.least_unacked = least_unacked; |
142 frames.push_back(QuicFrame(&stop_waiting)); | 142 frames.push_back(QuicFrame(&stop_waiting)); |
143 } | 143 } |
144 | 144 |
145 scoped_ptr<QuicPacket> packet( | 145 scoped_ptr<QuicPacket> packet( |
146 framer.BuildUnsizedDataPacket(header, frames).packet); | 146 BuildUnsizedDataPacket(&framer, header, frames).packet); |
147 return scoped_ptr<QuicEncryptedPacket>(framer.EncryptPacket( | 147 return scoped_ptr<QuicEncryptedPacket>(framer.EncryptPacket( |
148 ENCRYPTION_NONE, header.packet_sequence_number, *packet)); | 148 ENCRYPTION_NONE, header.packet_sequence_number, *packet)); |
149 } | 149 } |
150 | 150 |
151 // Returns a newly created packet to send kData on stream 1. | 151 // Returns a newly created packet to send kData on stream 1. |
152 scoped_ptr<QuicEncryptedPacket> QuicTestPacketMaker::MakeDataPacket( | 152 scoped_ptr<QuicEncryptedPacket> QuicTestPacketMaker::MakeDataPacket( |
153 QuicPacketSequenceNumber sequence_number, | 153 QuicPacketSequenceNumber sequence_number, |
154 QuicStreamId stream_id, | 154 QuicStreamId stream_id, |
155 bool should_include_version, | 155 bool should_include_version, |
156 bool fin, | 156 bool fin, |
(...skipping 63 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
220 return headers; | 220 return headers; |
221 } | 221 } |
222 | 222 |
223 scoped_ptr<QuicEncryptedPacket> QuicTestPacketMaker::MakePacket( | 223 scoped_ptr<QuicEncryptedPacket> QuicTestPacketMaker::MakePacket( |
224 const QuicPacketHeader& header, | 224 const QuicPacketHeader& header, |
225 const QuicFrame& frame) { | 225 const QuicFrame& frame) { |
226 QuicFramer framer(SupportedVersions(version_), QuicTime::Zero(), false); | 226 QuicFramer framer(SupportedVersions(version_), QuicTime::Zero(), false); |
227 QuicFrames frames; | 227 QuicFrames frames; |
228 frames.push_back(frame); | 228 frames.push_back(frame); |
229 scoped_ptr<QuicPacket> packet( | 229 scoped_ptr<QuicPacket> packet( |
230 framer.BuildUnsizedDataPacket(header, frames).packet); | 230 BuildUnsizedDataPacket(&framer, header, frames).packet); |
231 return scoped_ptr<QuicEncryptedPacket>(framer.EncryptPacket( | 231 return scoped_ptr<QuicEncryptedPacket>(framer.EncryptPacket( |
232 ENCRYPTION_NONE, header.packet_sequence_number, *packet)); | 232 ENCRYPTION_NONE, header.packet_sequence_number, *packet)); |
233 } | 233 } |
234 | 234 |
235 void QuicTestPacketMaker::InitializeHeader( | 235 void QuicTestPacketMaker::InitializeHeader( |
236 QuicPacketSequenceNumber sequence_number, | 236 QuicPacketSequenceNumber sequence_number, |
237 bool should_include_version) { | 237 bool should_include_version) { |
238 header_.public_header.connection_id = connection_id_; | 238 header_.public_header.connection_id = connection_id_; |
239 header_.public_header.reset_flag = false; | 239 header_.public_header.reset_flag = false; |
240 header_.public_header.version_flag = should_include_version; | 240 header_.public_header.version_flag = should_include_version; |
241 header_.public_header.sequence_number_length = PACKET_1BYTE_SEQUENCE_NUMBER; | 241 header_.public_header.sequence_number_length = PACKET_1BYTE_SEQUENCE_NUMBER; |
242 header_.packet_sequence_number = sequence_number; | 242 header_.packet_sequence_number = sequence_number; |
243 header_.fec_group = 0; | 243 header_.fec_group = 0; |
244 header_.entropy_flag = false; | 244 header_.entropy_flag = false; |
245 header_.fec_flag = false; | 245 header_.fec_flag = false; |
246 } | 246 } |
247 | 247 |
248 } // namespace test | 248 } // namespace test |
249 } // namespace net | 249 } // namespace net |
OLD | NEW |