| 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 |