Roll Chrome into Mojo.

net/quic/quic_connection_test.cc

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "net/quic/quic_connection.h"
 
 #include "base/basictypes.h"
 #include "base/bind.h"
 #include "base/stl_util.h"
 #include "net/base/net_errors.h"
 #include "net/quic/congestion_control/loss_detection_interface.h"
 #include "net/quic/congestion_control/receive_algorithm_interface.h"
 #include "net/quic/congestion_control/send_algorithm_interface.h"
 #include "net/quic/crypto/null_encrypter.h"
 #include "net/quic/crypto/quic_decrypter.h"
 #include "net/quic/crypto/quic_encrypter.h"
+#include "net/quic/quic_ack_notifier.h"
 #include "net/quic/quic_flags.h"
 #include "net/quic/quic_protocol.h"
 #include "net/quic/quic_utils.h"
 #include "net/quic/test_tools/mock_clock.h"
 #include "net/quic/test_tools/mock_random.h"
 #include "net/quic/test_tools/quic_config_peer.h"
 #include "net/quic/test_tools/quic_connection_peer.h"
 #include "net/quic/test_tools/quic_framer_peer.h"
 #include "net/quic/test_tools/quic_packet_creator_peer.h"
 #include "net/quic/test_tools/quic_sent_packet_manager_peer.h"
 #include "net/quic/test_tools/quic_test_utils.h"
 #include "net/quic/test_tools/simple_quic_framer.h"
+#include "net/test/gtest_util.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 using base::StringPiece;
 using std::map;
 using std::string;
 using std::vector;
 using testing::AnyNumber;
 using testing::AtLeast;
 using testing::ContainerEq;
@@ skipping 4090 matching lines @@
   ack_header.fec_group = 1;
 
   QuicPacket* packet =
       BuildUnsizedDataPacket(&framer_, ack_header, frames).packet;
 
   // Take the packet which contains the ACK frame, and construct and deliver an
   // FEC packet which allows the ACK packet to be recovered.
   ProcessFecPacket(2, 1, true, !kEntropyFlag, packet);
 }
 
-TEST_P(QuicConnectionTest, NetworkChangeVisitorCallbacksChangeFecState) {
+TEST_P(QuicConnectionTest, NetworkChangeVisitorCwndCallbackChangesFecState) {
   QuicPacketCreator* creator =
       QuicConnectionPeer::GetPacketCreator(&connection_);
   size_t max_packets_per_fec_group = creator->max_packets_per_fec_group();
 
   QuicSentPacketManager::NetworkChangeVisitor* visitor =
       QuicSentPacketManagerPeer::GetNetworkChangeVisitor(
           QuicConnectionPeer::GetSentPacketManager(&connection_));
   EXPECT_TRUE(visitor);
 
   // Increase FEC group size by increasing congestion window to a large number.
   EXPECT_CALL(*send_algorithm_, GetCongestionWindow()).WillRepeatedly(
       Return(1000 * kDefaultTCPMSS));
   visitor->OnCongestionWindowChange();
   EXPECT_LT(max_packets_per_fec_group, creator->max_packets_per_fec_group());
 }
 
+TEST_P(QuicConnectionTest, NetworkChangeVisitorConfigCallbackChangesFecState) {
+  QuicSentPacketManager* sent_packet_manager =
+      QuicConnectionPeer::GetSentPacketManager(&connection_);
+  QuicSentPacketManager::NetworkChangeVisitor* visitor =
+      QuicSentPacketManagerPeer::GetNetworkChangeVisitor(sent_packet_manager);
+  EXPECT_TRUE(visitor);
+
+  QuicPacketGenerator* generator =
+      QuicConnectionPeer::GetPacketGenerator(&connection_);
+  EXPECT_EQ(QuicTime::Delta::Zero(), generator->fec_timeout());
+
+  // Verify that sending a config with a new initial rtt changes fec timeout.
+  // Create and process a config with a non-zero initial RTT.
+  EXPECT_CALL(*send_algorithm_, SetFromConfig(_, _, _));
+  QuicConfig config;
+  config.SetInitialRoundTripTimeUsToSend(300000);
+  connection_.SetFromConfig(config);
+  EXPECT_LT(QuicTime::Delta::Zero(), generator->fec_timeout());
+}
+
+TEST_P(QuicConnectionTest, NetworkChangeVisitorRttCallbackChangesFecState) {
+  // Verify that sending a config with a new initial rtt changes fec timeout.
+  QuicSentPacketManager* sent_packet_manager =
+      QuicConnectionPeer::GetSentPacketManager(&connection_);
+  QuicSentPacketManager::NetworkChangeVisitor* visitor =
+      QuicSentPacketManagerPeer::GetNetworkChangeVisitor(sent_packet_manager);
+  EXPECT_TRUE(visitor);
+
+  QuicPacketGenerator* generator =
+      QuicConnectionPeer::GetPacketGenerator(&connection_);
+  EXPECT_EQ(QuicTime::Delta::Zero(), generator->fec_timeout());
+
+  // Increase FEC timeout by increasing RTT.
+  RttStats* rtt_stats =
+      QuicSentPacketManagerPeer::GetRttStats(sent_packet_manager);
+  rtt_stats->UpdateRtt(QuicTime::Delta::FromMilliseconds(300),
+                       QuicTime::Delta::Zero(), QuicTime::Zero());
+  visitor->OnRttChange();
+  EXPECT_LT(QuicTime::Delta::Zero(), generator->fec_timeout());
+}
+
 class MockQuicConnectionDebugVisitor
     : public QuicConnectionDebugVisitor {
  public:
   MOCK_METHOD1(OnFrameAddedToPacket,
                void(const QuicFrame&));
 
   MOCK_METHOD6(OnPacketSent,
                void(const SerializedPacket&,
                     QuicPacketSequenceNumber,
                     EncryptionLevel,
@@ skipping 75 matching lines @@
 
   // Cancel alarm, and try again with BLOCKED frame.
   ack_alarm->Cancel();
   QuicBlockedFrame blocked;
   blocked.stream_id = 3;
   EXPECT_CALL(visitor_, OnBlockedFrames(_));
   ProcessFramePacket(QuicFrame(&blocked));
   EXPECT_TRUE(ack_alarm->IsSet());
 }
 
+TEST_P(QuicConnectionTest, NoDataNoFin) {
+  ValueRestore<bool> old_flag(&FLAGS_quic_empty_data_no_fin_early_return, true);
+  // Make sure that a call to SendStreamWithData, with no data and no FIN, does
+  // not result in a QuicAckNotifier being used-after-free (fail under ASAN).
+  // Regression test for b/18594622
+  scoped_refptr<MockAckNotifierDelegate> delegate(new MockAckNotifierDelegate);
+  EXPECT_DFATAL(
+      connection_.SendStreamDataWithString(3, "", 0, !kFin, delegate.get()),
+      "Attempt to send empty stream frame");
+}
+
 }  // namespace
 }  // namespace test
 }  // namespace net