Send PING to check the status of the SPDY connection.

net/spdy/spdy_http_stream.cc

 // Copyright (c) 2011 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_http_stream.h"
 
 #include <algorithm>
 #include <list>
 #include <string>
 
 #include "base/logging.h"
 #include "base/message_loop.h"
 #include "net/base/address_list.h"
 #include "net/base/host_port_pair.h"
 #include "net/base/load_flags.h"
 #include "net/base/net_util.h"
 #include "net/http/http_request_headers.h"
 #include "net/http/http_request_info.h"
 #include "net/http/http_response_info.h"
 #include "net/http/http_util.h"
 #include "net/spdy/spdy_http_utils.h"
 #include "net/spdy/spdy_session.h"
 
 namespace net {
 
 SpdyHttpStream::SpdyHttpStream(SpdySession* spdy_session,
                                bool direct)
     : ALLOW_THIS_IN_INITIALIZER_LIST(read_callback_factory_(this)),
+      ALLOW_THIS_IN_INITIALIZER_LIST(ping_method_factory_(this)),
       stream_(NULL),
       spdy_session_(spdy_session),
       response_info_(NULL),
       download_finished_(false),
       response_headers_received_(false),
       user_callback_(NULL),
       user_buffer_len_(0),
       buffered_read_callback_pending_(false),
       more_read_data_pending_(false),
       direct_(direct) { }
@@ skipping 221 matching lines @@
 
 void SpdyHttpStream::Cancel() {
   if (spdy_session_)
     spdy_session_->CancelPendingCreateStreams(&stream_);
   user_callback_ = NULL;
   if (stream_)
     stream_->Cancel();
 }
 
 bool SpdyHttpStream::OnSendHeadersComplete(int status) {
+  // Send the spdy ping message here.
+  stream_->SendPing();
+  const int kPingTimeMs = 1000;
+  MessageLoop::current()->PostDelayedTask(FROM_HERE, ping_method_factory_.
+      NewRunnableMethod(&SpdyHttpStream::CheckPingResponse),
+      kPingTimeMs);
+
   if (user_callback_)
     DoCallback(status);
   return request_body_stream_.get() == NULL;
 }
 
+void SpdyHttpStream::CheckPingResponse() {
+  // TODO(rtenneti): we should send a different error code.
+  if (!spdy_session_->DidLastPingWork())
+    DoCallback(ERR_CONNECTION_ABORTED);

[willchan no longer on Chromium, 2011/10/12 05:41:06]

Need a different error code.

[ramant (doing other things), 2011/10/13 21:41:14]

Done.

+}
+
 int SpdyHttpStream::OnSendBody() {
   CHECK(request_body_stream_.get());
 
   int buf_len = static_cast<int>(request_body_stream_->buf_len());
   if (!buf_len)
     return OK;
   bool is_chunked = request_body_stream_->is_chunked();
   // TODO(satish): For non-chunked POST data, we set DATA_FLAG_FIN for all
   // blocks of data written out. This is wrong if the POST data was larger than
   // UploadDataStream::kBufSize as that is the largest buffer that
@@ skipping 181 matching lines @@
     SSLCertRequestInfo* cert_request_info) {
   DCHECK(stream_);
   stream_->GetSSLCertRequestInfo(cert_request_info);
 }
 
 bool SpdyHttpStream::IsSpdyHttpStream() const {
   return true;
 }
 
 }  // namespace net