SPDY - Handle incomplete headers during server push.

net/spdy/spdy_proxy_client_socket.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/spdy/spdy_proxy_client_socket.h"
 
 #include <algorithm>  // min
 
 #include "base/bind.h"
 #include "base/bind_helpers.h"
@@ skipping 471 matching lines @@
   // Save the response
   if (!SpdyHeadersToHttpResponse(
           response, spdy_stream_->GetProtocolVersion(), &response_))
       return ERR_INCOMPLETE_SPDY_HEADERS;
 
   OnIOComplete(status);
   return OK;
 }
 
 // Called when data is received.
-void SpdyProxyClientSocket::OnDataReceived(const char* data, int length) {
+int SpdyProxyClientSocket::OnDataReceived(const char* data, int length) {
   if (length > 0) {
     // Save the received data.
     scoped_refptr<IOBuffer> io_buffer(new IOBuffer(length));
     memcpy(io_buffer->data(), data, length);
     read_buffer_.push_back(
         make_scoped_refptr(new DrainableIOBuffer(io_buffer, length)));
   }
 
   if (!read_callback_.is_null()) {
     int rv = PopulateUserReadBuffer();
     CompletionCallback c = read_callback_;
     read_callback_.Reset();
     user_buffer_ = NULL;
     c.Run(rv);
   }
+  return OK;
 }
 
 void SpdyProxyClientSocket::OnDataSent(int length)  {
   DCHECK(!write_callback_.is_null());
 
   write_bytes_outstanding_ -= length;
 
   DCHECK_GE(write_bytes_outstanding_, 0);
 
   if (write_bytes_outstanding_ == 0) {
@@ skipping 34 matching lines @@
   } else if (!read_callback_.is_null()) {
     // If we have a read_callback_, the we need to make sure we call it back.
     OnDataReceived(NULL, 0);
   }
   // This may have been deleted by read_callback_, so check first.
   if (weak_ptr && !write_callback.is_null())
     write_callback.Run(ERR_CONNECTION_CLOSED);
 }
 
 }  // namespace net