Land Recent QUIC Changes.

net/quic/quic_unacked_packet_map.cc

Index: net/quic/quic_unacked_packet_map.cc
diff --git a/net/quic/quic_unacked_packet_map.cc b/net/quic/quic_unacked_packet_map.cc
index e9dd1368f6aa7dc062b058a7b77fa5c8eb3b1231..563b53dfe92d1da8dd141d729b36def31d342aa4 100644
--- a/net/quic/quic_unacked_packet_map.cc
+++ b/net/quic/quic_unacked_packet_map.cc
@@ -320,4 +320,21 @@ void QuicUnackedPacketMap::SetSent(QuicPacketSequenceNumber sequence_number,
   }
 }
 
+void QuicUnackedPacketMap::RestoreInFlight(
+    QuicPacketSequenceNumber sequence_number) {
+  DCHECK_LT(0u, sequence_number);
+  UnackedPacketMap::iterator it = unacked_packets_.find(sequence_number);
+  if (it == unacked_packets_.end()) {
+    LOG(DFATAL) << "OnPacketSent called for packet that is not unacked: "
+                << sequence_number;
+    return;
+  }
+  DCHECK(!it->second.in_flight);
+  DCHECK_NE(0u, it->second.bytes_sent);
+  DCHECK(it->second.sent_time.IsInitialized());
+
+  bytes_in_flight_ += it->second.bytes_sent;
+  it->second.in_flight = true;
+}
+
 }  // namespace net