| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 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_utils.h" | 5 #include "net/quic/test_tools/quic_test_utils.h" |
| 6 | 6 |
| 7 #include "base/sha1.h" | 7 #include "base/sha1.h" |
| 8 #include "base/stl_util.h" | 8 #include "base/stl_util.h" |
| 9 #include "base/strings/string_number_conversions.h" | 9 #include "base/strings/string_number_conversions.h" |
| 10 #include "net/quic/crypto/crypto_framer.h" | 10 #include "net/quic/crypto/crypto_framer.h" |
| (...skipping 284 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 295 bool is_server, | 295 bool is_server, |
| 296 const QuicVersionVector& supported_versions) | 296 const QuicVersionVector& supported_versions) |
| 297 : MockConnection(is_server, supported_versions) { | 297 : MockConnection(is_server, supported_versions) { |
| 298 } | 298 } |
| 299 | 299 |
| 300 PacketSavingConnection::~PacketSavingConnection() { | 300 PacketSavingConnection::~PacketSavingConnection() { |
| 301 STLDeleteElements(&packets_); | 301 STLDeleteElements(&packets_); |
| 302 STLDeleteElements(&encrypted_packets_); | 302 STLDeleteElements(&encrypted_packets_); |
| 303 } | 303 } |
| 304 | 304 |
| 305 bool PacketSavingConnection::SendOrQueuePacket( | 305 void PacketSavingConnection::SendOrQueuePacket(QueuedPacket packet) { |
| 306 EncryptionLevel level, | 306 packets_.push_back(packet.serialized_packet.packet); |
| 307 const SerializedPacket& packet, | |
| 308 TransmissionType transmission_type) { | |
| 309 packets_.push_back(packet.packet); | |
| 310 QuicEncryptedPacket* encrypted = QuicConnectionPeer::GetFramer(this)-> | 307 QuicEncryptedPacket* encrypted = QuicConnectionPeer::GetFramer(this)-> |
| 311 EncryptPacket(level, packet.sequence_number, *packet.packet); | 308 EncryptPacket(packet.encryption_level, |
| 309 packet.serialized_packet.sequence_number, |
| 310 *packet.serialized_packet.packet); |
| 312 encrypted_packets_.push_back(encrypted); | 311 encrypted_packets_.push_back(encrypted); |
| 313 return true; | 312 // Transfer ownership of the packet to the SentPacketManager and the |
| 313 // ack notifier to the AckNotifierManager. |
| 314 sent_packet_manager_.OnSerializedPacket(packet.serialized_packet); |
| 314 } | 315 } |
| 315 | 316 |
| 316 MockSession::MockSession(QuicConnection* connection) | 317 MockSession::MockSession(QuicConnection* connection) |
| 317 : QuicSession(connection, DefaultQuicConfig()) { | 318 : QuicSession(connection, DefaultQuicConfig()) { |
| 318 InitializeSession(); | 319 InitializeSession(); |
| 319 ON_CALL(*this, WritevData(_, _, _, _, _, _)) | 320 ON_CALL(*this, WritevData(_, _, _, _, _, _)) |
| 320 .WillByDefault(testing::Return(QuicConsumedData(0, false))); | 321 .WillByDefault(testing::Return(QuicConsumedData(0, false))); |
| 321 } | 322 } |
| 322 | 323 |
| 323 MockSession::~MockSession() { | 324 MockSession::~MockSession() { |
| (...skipping 342 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 666 // different way, so TestWriterFactory::OnPacketSent might never be called. | 667 // different way, so TestWriterFactory::OnPacketSent might never be called. |
| 667 factory_->current_writer_ = this; | 668 factory_->current_writer_ = this; |
| 668 return QuicPerConnectionPacketWriter::WritePacket(buffer, | 669 return QuicPerConnectionPacketWriter::WritePacket(buffer, |
| 669 buf_len, | 670 buf_len, |
| 670 self_address, | 671 self_address, |
| 671 peer_address); | 672 peer_address); |
| 672 } | 673 } |
| 673 | 674 |
| 674 } // namespace test | 675 } // namespace test |
| 675 } // namespace net | 676 } // namespace net |
| OLD | NEW |