Make QUIC stateless rejects work in the face of packet loss by storing the SREJ packet(s) in the Qu…

net/quic/test_tools/quic_connection_peer.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/test_tools/quic_connection_peer.h"
 
 #include "base/stl_util.h"
 #include "net/quic/congestion_control/send_algorithm_interface.h"
 #include "net/quic/quic_connection.h"
 #include "net/quic/quic_packet_writer.h"
@@ skipping 201 matching lines @@
 }
 
 // static
 void QuicConnectionPeer::CloseConnection(QuicConnection* connection) {
   connection->connected_ = false;
 }
 
 // static
 QuicEncryptedPacket* QuicConnectionPeer::GetConnectionClosePacket(
     QuicConnection* connection) {
-  return connection->connection_close_packet_.get();
+  if (connection->termination_packets_ == nullptr ||
+      connection->termination_packets_->empty()) {
+    return nullptr;
+  }
+  return (*connection->termination_packets_)[0];
 }
 
 // static
 QuicPacketHeader* QuicConnectionPeer::GetLastHeader(
     QuicConnection* connection) {
   return &connection->last_header_;
 }
 
 // static
 void QuicConnectionPeer::SetPacketNumberOfLastSentPacket(
@@ skipping 20 matching lines @@
 }
 
 // static
 void QuicConnectionPeer::SetNextMtuProbeAt(QuicConnection* connection,
                                            QuicPacketNumber number) {
   connection->next_mtu_probe_at_ = number;
 }
 
 }  // namespace test
 }  // namespace net