Land Recent QUIC Changes until 05/13/2015

net/quic/quic_session.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_session.h"
 
 #include "base/stl_util.h"
 #include "net/quic/crypto/proof_verifier.h"
 #include "net/quic/quic_connection.h"
 #include "net/quic/quic_flow_controller.h"
@@ skipping 116 matching lines @@
   STLDeleteElements(&closed_streams_);
   STLDeleteValues(&stream_map_);
 
   DLOG_IF(WARNING,
           locally_closed_streams_highest_offset_.size() > max_open_streams_)
       << "Surprisingly high number of locally closed streams still waiting for "
          "final byte offset: " << locally_closed_streams_highest_offset_.size();
 }
 
 void QuicSession::OnStreamFrames(const vector<QuicStreamFrame>& frames) {
-  for (size_t i = 0; i < frames.size(); ++i) {
+  for (size_t i = 0; i < frames.size() && connection_->connected(); ++i) {
     // TODO(rch) deal with the error case of stream id 0.
     const QuicStreamFrame& frame = frames[i];
     QuicStreamId stream_id = frame.stream_id;
     ReliableQuicStream* stream = GetStream(stream_id);
     if (!stream) {
       // The stream no longer exists, but we may still be interested in the
       // final stream byte offset sent by the peer. A frame with a FIN can give
       // us this offset.
       if (frame.fin) {
         QuicStreamOffset final_byte_offset =
             frame.offset + frame.data.TotalBufferSize();
         UpdateFlowControlOnFinalReceivedByteOffset(stream_id,
                                                    final_byte_offset);
       }
 
       continue;
     }
     stream->OnStreamFrame(frames[i]);
+    if (!connection_->connected()) {
+      return;
+    }
   }
 }
 
 void QuicSession::OnStreamHeaders(QuicStreamId stream_id,
                                   StringPiece headers_data) {
   QuicDataStream* stream = GetDataStream(stream_id);
   if (!stream) {
     // It's quite possible to receive headers after a stream has been reset.
     return;
   }
@@ skipping 571 matching lines @@
   for (DataStreamMap::iterator it = stream_map_.begin();
        it != stream_map_.end(); ++it) {
     if (it->second->flow_controller()->IsBlocked()) {
       return true;
     }
   }
   return false;
 }
 
 }  // namespace net