| 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/chromium/quic_test_packet_maker.h" | 5 #include "net/quic/chromium/quic_test_packet_maker.h" |
| 6 | 6 |
| 7 #include <list> | 7 #include <list> |
| 8 #include <utility> | 8 #include <utility> |
| 9 | 9 |
| 10 #include "base/memory/ptr_util.h" | 10 #include "base/memory/ptr_util.h" |
| (...skipping 100 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 111 DVLOG(1) << "Adding frame: " << frames[1]; | 111 DVLOG(1) << "Adding frame: " << frames[1]; |
| 112 | 112 |
| 113 QuicRstStreamFrame rst(stream_id, error_code, 0); | 113 QuicRstStreamFrame rst(stream_id, error_code, 0); |
| 114 frames.push_back(QuicFrame(&rst)); | 114 frames.push_back(QuicFrame(&rst)); |
| 115 DVLOG(1) << "Adding frame: " << frames[2]; | 115 DVLOG(1) << "Adding frame: " << frames[2]; |
| 116 | 116 |
| 117 QuicFramer framer(SupportedVersions(version_), clock_->Now(), perspective_); | 117 QuicFramer framer(SupportedVersions(version_), clock_->Now(), perspective_); |
| 118 std::unique_ptr<QuicPacket> packet( | 118 std::unique_ptr<QuicPacket> packet( |
| 119 BuildUnsizedDataPacket(&framer, header, frames)); | 119 BuildUnsizedDataPacket(&framer, header, frames)); |
| 120 char buffer[kMaxPacketSize]; | 120 char buffer[kMaxPacketSize]; |
| 121 size_t encrypted_size = framer.EncryptPayload(ENCRYPTION_NONE, /*path_id=*/0u, | 121 size_t encrypted_size = framer.EncryptPayload( |
| 122 header.packet_number, *packet, | 122 ENCRYPTION_NONE, header.packet_number, *packet, buffer, kMaxPacketSize); |
| 123 buffer, kMaxPacketSize); | |
| 124 EXPECT_NE(0u, encrypted_size); | 123 EXPECT_NE(0u, encrypted_size); |
| 125 QuicReceivedPacket encrypted(buffer, encrypted_size, QuicTime::Zero(), false); | 124 QuicReceivedPacket encrypted(buffer, encrypted_size, QuicTime::Zero(), false); |
| 126 return std::unique_ptr<QuicReceivedPacket>(encrypted.Clone()); | 125 return std::unique_ptr<QuicReceivedPacket>(encrypted.Clone()); |
| 127 } | 126 } |
| 128 | 127 |
| 129 std::unique_ptr<QuicReceivedPacket> | 128 std::unique_ptr<QuicReceivedPacket> |
| 130 QuicTestPacketMaker::MakeAckAndConnectionClosePacket( | 129 QuicTestPacketMaker::MakeAckAndConnectionClosePacket( |
| 131 QuicPacketNumber num, | 130 QuicPacketNumber num, |
| 132 bool include_version, | 131 bool include_version, |
| 133 QuicTime::Delta ack_delay_time, | 132 QuicTime::Delta ack_delay_time, |
| (...skipping 29 matching lines...) Expand all Loading... |
| 163 close.error_code = quic_error; | 162 close.error_code = quic_error; |
| 164 close.error_details = quic_error_details; | 163 close.error_details = quic_error_details; |
| 165 | 164 |
| 166 frames.push_back(QuicFrame(&close)); | 165 frames.push_back(QuicFrame(&close)); |
| 167 DVLOG(1) << "Adding frame: " << frames[2]; | 166 DVLOG(1) << "Adding frame: " << frames[2]; |
| 168 | 167 |
| 169 QuicFramer framer(SupportedVersions(version_), clock_->Now(), perspective_); | 168 QuicFramer framer(SupportedVersions(version_), clock_->Now(), perspective_); |
| 170 std::unique_ptr<QuicPacket> packet( | 169 std::unique_ptr<QuicPacket> packet( |
| 171 BuildUnsizedDataPacket(&framer, header, frames)); | 170 BuildUnsizedDataPacket(&framer, header, frames)); |
| 172 char buffer[kMaxPacketSize]; | 171 char buffer[kMaxPacketSize]; |
| 173 size_t encrypted_size = framer.EncryptPayload(ENCRYPTION_NONE, /*path_id=*/0u, | 172 size_t encrypted_size = framer.EncryptPayload( |
| 174 header.packet_number, *packet, | 173 ENCRYPTION_NONE, header.packet_number, *packet, buffer, kMaxPacketSize); |
| 175 buffer, kMaxPacketSize); | |
| 176 EXPECT_NE(0u, encrypted_size); | 174 EXPECT_NE(0u, encrypted_size); |
| 177 QuicReceivedPacket encrypted(buffer, encrypted_size, clock_->Now(), false); | 175 QuicReceivedPacket encrypted(buffer, encrypted_size, clock_->Now(), false); |
| 178 return std::unique_ptr<QuicReceivedPacket>(encrypted.Clone()); | 176 return std::unique_ptr<QuicReceivedPacket>(encrypted.Clone()); |
| 179 } | 177 } |
| 180 | 178 |
| 181 std::unique_ptr<QuicReceivedPacket> | 179 std::unique_ptr<QuicReceivedPacket> |
| 182 QuicTestPacketMaker::MakeConnectionClosePacket(QuicPacketNumber num) { | 180 QuicTestPacketMaker::MakeConnectionClosePacket(QuicPacketNumber num) { |
| 183 QuicPacketHeader header; | 181 QuicPacketHeader header; |
| 184 header.public_header.connection_id = connection_id_; | 182 header.public_header.connection_id = connection_id_; |
| 185 header.public_header.reset_flag = false; | 183 header.public_header.reset_flag = false; |
| (...skipping 66 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 252 DVLOG(1) << "Adding frame: " << ack_frame; | 250 DVLOG(1) << "Adding frame: " << ack_frame; |
| 253 frames.push_back(ack_frame); | 251 frames.push_back(ack_frame); |
| 254 | 252 |
| 255 QuicStopWaitingFrame stop_waiting; | 253 QuicStopWaitingFrame stop_waiting; |
| 256 stop_waiting.least_unacked = stop_least_unacked; | 254 stop_waiting.least_unacked = stop_least_unacked; |
| 257 frames.push_back(QuicFrame(&stop_waiting)); | 255 frames.push_back(QuicFrame(&stop_waiting)); |
| 258 | 256 |
| 259 std::unique_ptr<QuicPacket> packet( | 257 std::unique_ptr<QuicPacket> packet( |
| 260 BuildUnsizedDataPacket(&framer, header, frames)); | 258 BuildUnsizedDataPacket(&framer, header, frames)); |
| 261 char buffer[kMaxPacketSize]; | 259 char buffer[kMaxPacketSize]; |
| 262 size_t encrypted_size = framer.EncryptPayload(ENCRYPTION_NONE, /*path_id=*/0u, | 260 size_t encrypted_size = framer.EncryptPayload( |
| 263 header.packet_number, *packet, | 261 ENCRYPTION_NONE, header.packet_number, *packet, buffer, kMaxPacketSize); |
| 264 buffer, kMaxPacketSize); | |
| 265 EXPECT_NE(0u, encrypted_size); | 262 EXPECT_NE(0u, encrypted_size); |
| 266 QuicReceivedPacket encrypted(buffer, encrypted_size, clock_->Now(), false); | 263 QuicReceivedPacket encrypted(buffer, encrypted_size, clock_->Now(), false); |
| 267 return std::unique_ptr<QuicReceivedPacket>(encrypted.Clone()); | 264 return std::unique_ptr<QuicReceivedPacket>(encrypted.Clone()); |
| 268 } | 265 } |
| 269 | 266 |
| 270 // Returns a newly created packet to send kData on stream 1. | 267 // Returns a newly created packet to send kData on stream 1. |
| 271 std::unique_ptr<QuicReceivedPacket> QuicTestPacketMaker::MakeDataPacket( | 268 std::unique_ptr<QuicReceivedPacket> QuicTestPacketMaker::MakeDataPacket( |
| 272 QuicPacketNumber packet_number, | 269 QuicPacketNumber packet_number, |
| 273 QuicStreamId stream_id, | 270 QuicStreamId stream_id, |
| 274 bool should_include_version, | 271 bool should_include_version, |
| (...skipping 335 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 610 return MakeMultipleFramesPacket(header, frames); | 607 return MakeMultipleFramesPacket(header, frames); |
| 611 } | 608 } |
| 612 | 609 |
| 613 std::unique_ptr<QuicReceivedPacket> | 610 std::unique_ptr<QuicReceivedPacket> |
| 614 QuicTestPacketMaker::MakeMultipleFramesPacket(const QuicPacketHeader& header, | 611 QuicTestPacketMaker::MakeMultipleFramesPacket(const QuicPacketHeader& header, |
| 615 const QuicFrames& frames) { | 612 const QuicFrames& frames) { |
| 616 QuicFramer framer(SupportedVersions(version_), clock_->Now(), perspective_); | 613 QuicFramer framer(SupportedVersions(version_), clock_->Now(), perspective_); |
| 617 std::unique_ptr<QuicPacket> packet( | 614 std::unique_ptr<QuicPacket> packet( |
| 618 BuildUnsizedDataPacket(&framer, header, frames)); | 615 BuildUnsizedDataPacket(&framer, header, frames)); |
| 619 char buffer[kMaxPacketSize]; | 616 char buffer[kMaxPacketSize]; |
| 620 size_t encrypted_size = framer.EncryptPayload(ENCRYPTION_NONE, /*path_id=*/0u, | 617 size_t encrypted_size = framer.EncryptPayload( |
| 621 header.packet_number, *packet, | 618 ENCRYPTION_NONE, header.packet_number, *packet, buffer, kMaxPacketSize); |
| 622 buffer, kMaxPacketSize); | |
| 623 EXPECT_NE(0u, encrypted_size); | 619 EXPECT_NE(0u, encrypted_size); |
| 624 QuicReceivedPacket encrypted(buffer, encrypted_size, clock_->Now(), false); | 620 QuicReceivedPacket encrypted(buffer, encrypted_size, clock_->Now(), false); |
| 625 return std::unique_ptr<QuicReceivedPacket>(encrypted.Clone()); | 621 return std::unique_ptr<QuicReceivedPacket>(encrypted.Clone()); |
| 626 } | 622 } |
| 627 | 623 |
| 628 void QuicTestPacketMaker::InitializeHeader(QuicPacketNumber packet_number, | 624 void QuicTestPacketMaker::InitializeHeader(QuicPacketNumber packet_number, |
| 629 bool should_include_version) { | 625 bool should_include_version) { |
| 630 header_.public_header.connection_id = connection_id_; | 626 header_.public_header.connection_id = connection_id_; |
| 631 header_.public_header.reset_flag = false; | 627 header_.public_header.reset_flag = false; |
| 632 header_.public_header.version_flag = should_include_version; | 628 header_.public_header.version_flag = should_include_version; |
| (...skipping 19 matching lines...) Expand all Loading... |
| 652 *offset += spdy_frame.size(); | 648 *offset += spdy_frame.size(); |
| 653 return MakePacket(header_, QuicFrame(&quic_frame)); | 649 return MakePacket(header_, QuicFrame(&quic_frame)); |
| 654 } | 650 } |
| 655 QuicStreamFrame quic_frame(kHeadersStreamId, false, 0, | 651 QuicStreamFrame quic_frame(kHeadersStreamId, false, 0, |
| 656 StringPiece(spdy_frame.data(), spdy_frame.size())); | 652 StringPiece(spdy_frame.data(), spdy_frame.size())); |
| 657 return MakePacket(header_, QuicFrame(&quic_frame)); | 653 return MakePacket(header_, QuicFrame(&quic_frame)); |
| 658 } | 654 } |
| 659 | 655 |
| 660 } // namespace test | 656 } // namespace test |
| 661 } // namespace net | 657 } // namespace net |
| OLD | NEW |