Add Socket::ReadIfReady()

net/spdy/spdy_session.cc

Index: net/spdy/spdy_session.cc
diff --git a/net/spdy/spdy_session.cc b/net/spdy/spdy_session.cc
index aa8397af8cfd23e89c47993d5ea9ed4dc3494020..231313d44a493f5ddedcb03cced3b2634dea4746 100644
--- a/net/spdy/spdy_session.cc
+++ b/net/spdy/spdy_session.cc
@@ -11,6 +11,7 @@
 
 #include "base/bind.h"
 #include "base/compiler_specific.h"
+#include "base/feature_list.h"
 #include "base/location.h"
 #include "base/logging.h"
 #include "base/memory/ptr_util.h"
@@ -45,6 +46,7 @@
 #include "net/log/net_log_source_type.h"
 #include "net/log/net_log_with_source.h"
 #include "net/proxy/proxy_server.h"
+#include "net/socket/socket.h"
 #include "net/socket/ssl_client_socket.h"
 #include "net/spdy/platform/api/spdy_estimate_memory_usage.h"
 #include "net/spdy/spdy_buffer_producer.h"
@@ -1803,6 +1805,32 @@ void SpdySession::PumpReadLoop(ReadState expected_read_state, int result) {
   ignore_result(DoReadLoop(expected_read_state, result));
 }
 
+int SpdySession::DoRead() {
+  CHECK(in_io_loop_);
+
+  CHECK(connection_);
+  CHECK(connection_->socket());
+  read_state_ = READ_STATE_DO_READ_COMPLETE;
+  int rv = ERR_READ_IF_READY_NOT_IMPLEMENTED;
+  if (base::FeatureList::IsEnabled(Socket::kReadIfReadyExperiment)) {
+    rv = connection_->socket()->ReadIfReady(
+        read_buffer_.get(), kReadBufferSize,
+        base::Bind(&SpdySession::PumpReadLoop, weak_factory_.GetWeakPtr(),
+                   READ_STATE_DO_READ));
+    // TODO(xunjieli): Follow-up CL to release |read_buffer_|. crbug.com/524258.
+  }
+  if (rv == ERR_READ_IF_READY_NOT_IMPLEMENTED) {
+    // Fallback to regular Read().
+    return connection_->socket()->Read(
+        read_buffer_.get(), kReadBufferSize,
+        base::Bind(&SpdySession::PumpReadLoop, weak_factory_.GetWeakPtr(),
+                   READ_STATE_DO_READ_COMPLETE));
+  }
+  if (rv == ERR_IO_PENDING)
+    read_state_ = READ_STATE_DO_READ;
+  return rv;
+}
+
 int SpdySession::DoReadLoop(ReadState expected_read_state, int result) {
   CHECK(!in_io_loop_);
   CHECK_EQ(read_state_, expected_read_state);
@@ -1856,18 +1884,6 @@ int SpdySession::DoReadLoop(ReadState expected_read_state, int result) {
   return result;
 }
 
-int SpdySession::DoRead() {
-  CHECK(in_io_loop_);
-
-  CHECK(connection_);
-  CHECK(connection_->socket());
-  read_state_ = READ_STATE_DO_READ_COMPLETE;
-  return connection_->socket()->Read(
-      read_buffer_.get(), kReadBufferSize,
-      base::Bind(&SpdySession::PumpReadLoop, weak_factory_.GetWeakPtr(),
-                 READ_STATE_DO_READ_COMPLETE));
-}
-
 int SpdySession::DoReadComplete(int result) {
   CHECK(in_io_loop_);