Plumbing for TCP FastOpen for SSL sockets.

net/socket/transport_client_socket_pool_test_util.cc

 // Copyright 2014 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/socket/transport_client_socket_pool_test_util.h"
 
 #include <string>
 
 #include "base/logging.h"
 #include "base/memory/weak_ptr.h"
@@ skipping 16 matching lines @@
   CHECK(ParseIPLiteralToNumber(ip, &number));
   return number;
 }
 
 // A StreamSocket which connects synchronously and successfully.
 class MockConnectClientSocket : public StreamSocket {
  public:
   MockConnectClientSocket(const AddressList& addrlist, net::NetLog* net_log)
       : connected_(false),
         addrlist_(addrlist),
-        net_log_(BoundNetLog::Make(net_log, NetLog::SOURCE_SOCKET)) {}
+        net_log_(BoundNetLog::Make(net_log, NetLog::SOURCE_SOCKET)),
+        use_tcp_fastopen_(false) {}
 
   // StreamSocket implementation.
   virtual int Connect(const CompletionCallback& callback) OVERRIDE {
     connected_ = true;
     return OK;
   }
   virtual void Disconnect() OVERRIDE { connected_ = false; }
   virtual bool IsConnected() const OVERRIDE { return connected_; }
   virtual bool IsConnectedAndIdle() const OVERRIDE { return connected_; }
 
   virtual int GetPeerAddress(IPEndPoint* address) const OVERRIDE {
     *address = addrlist_.front();
     return OK;
   }
   virtual int GetLocalAddress(IPEndPoint* address) const OVERRIDE {
     if (!connected_)
       return ERR_SOCKET_NOT_CONNECTED;
     if (addrlist_.front().GetFamily() == ADDRESS_FAMILY_IPV4)
       SetIPv4Address(address);
     else
       SetIPv6Address(address);
     return OK;
   }
   virtual const BoundNetLog& NetLog() const OVERRIDE { return net_log_; }
 
   virtual void SetSubresourceSpeculation() OVERRIDE {}
   virtual void SetOmniboxSpeculation() OVERRIDE {}
   virtual bool WasEverUsed() const OVERRIDE { return false; }
-  virtual bool UsingTCPFastOpen() const OVERRIDE { return false; }
+  virtual void EnableTCPFastOpen() OVERRIDE { use_tcp_fastopen_ = true; }
+  virtual bool UsingTCPFastOpen() const OVERRIDE { return use_tcp_fastopen_; }
   virtual bool WasNpnNegotiated() const OVERRIDE { return false; }
   virtual NextProto GetNegotiatedProtocol() const OVERRIDE {
     return kProtoUnknown;
   }
   virtual bool GetSSLInfo(SSLInfo* ssl_info) OVERRIDE { return false; }
 
   // Socket implementation.
   virtual int Read(IOBuffer* buf,
                    int buf_len,
                    const CompletionCallback& callback) OVERRIDE {
     return ERR_FAILED;
   }
   virtual int Write(IOBuffer* buf,
                     int buf_len,
                     const CompletionCallback& callback) OVERRIDE {
     return ERR_FAILED;
   }
   virtual int SetReceiveBufferSize(int32 size) OVERRIDE { return OK; }
   virtual int SetSendBufferSize(int32 size) OVERRIDE { return OK; }
 
  private:
   bool connected_;
   const AddressList addrlist_;
   BoundNetLog net_log_;
+  bool use_tcp_fastopen_;
 
   DISALLOW_COPY_AND_ASSIGN(MockConnectClientSocket);
 };
 
 class MockFailingClientSocket : public StreamSocket {
  public:
   MockFailingClientSocket(const AddressList& addrlist, net::NetLog* net_log)
       : addrlist_(addrlist),
-        net_log_(BoundNetLog::Make(net_log, NetLog::SOURCE_SOCKET)) {}
+        net_log_(BoundNetLog::Make(net_log, NetLog::SOURCE_SOCKET)),
+        use_tcp_fastopen_(false) {}
 
   // StreamSocket implementation.
   virtual int Connect(const CompletionCallback& callback) OVERRIDE {
     return ERR_CONNECTION_FAILED;
   }
 
   virtual void Disconnect() OVERRIDE {}
 
   virtual bool IsConnected() const OVERRIDE { return false; }
   virtual bool IsConnectedAndIdle() const OVERRIDE { return false; }
   virtual int GetPeerAddress(IPEndPoint* address) const OVERRIDE {
     return ERR_UNEXPECTED;
   }
   virtual int GetLocalAddress(IPEndPoint* address) const OVERRIDE {
     return ERR_UNEXPECTED;
   }
   virtual const BoundNetLog& NetLog() const OVERRIDE { return net_log_; }
 
   virtual void SetSubresourceSpeculation() OVERRIDE {}
   virtual void SetOmniboxSpeculation() OVERRIDE {}
   virtual bool WasEverUsed() const OVERRIDE { return false; }
-  virtual bool UsingTCPFastOpen() const OVERRIDE { return false; }
+  virtual void EnableTCPFastOpen() OVERRIDE { use_tcp_fastopen_ = true; }
+  virtual bool UsingTCPFastOpen() const OVERRIDE { return use_tcp_fastopen_; }
   virtual bool WasNpnNegotiated() const OVERRIDE { return false; }
   virtual NextProto GetNegotiatedProtocol() const OVERRIDE {
     return kProtoUnknown;
   }
   virtual bool GetSSLInfo(SSLInfo* ssl_info) OVERRIDE { return false; }
 
   // Socket implementation.
   virtual int Read(IOBuffer* buf,
                    int buf_len,
                    const CompletionCallback& callback) OVERRIDE {
     return ERR_FAILED;
   }
 
   virtual int Write(IOBuffer* buf,
                     int buf_len,
                     const CompletionCallback& callback) OVERRIDE {
     return ERR_FAILED;
   }
   virtual int SetReceiveBufferSize(int32 size) OVERRIDE { return OK; }
   virtual int SetSendBufferSize(int32 size) OVERRIDE { return OK; }
 
  private:
   const AddressList addrlist_;
   BoundNetLog net_log_;
+  bool use_tcp_fastopen_;
 
   DISALLOW_COPY_AND_ASSIGN(MockFailingClientSocket);
 };
 
 class MockTriggerableClientSocket : public StreamSocket {
  public:
   // |should_connect| indicates whether the socket should successfully complete
   // or fail.
   MockTriggerableClientSocket(const AddressList& addrlist,
                               bool should_connect,
                               net::NetLog* net_log)
       : should_connect_(should_connect),
         is_connected_(false),
         addrlist_(addrlist),
         net_log_(BoundNetLog::Make(net_log, NetLog::SOURCE_SOCKET)),
+        use_tcp_fastopen_(false),
         weak_factory_(this) {}
 
   // Call this method to get a closure which will trigger the connect callback
   // when called. The closure can be called even after the socket is deleted; it
   // will safely do nothing.
   base::Closure GetConnectCallback() {
     return base::Bind(&MockTriggerableClientSocket::DoCallback,
                       weak_factory_.GetWeakPtr());
   }
 
@@ skipping 50 matching lines @@
       SetIPv4Address(address);
     else
       SetIPv6Address(address);
     return OK;
   }
   virtual const BoundNetLog& NetLog() const OVERRIDE { return net_log_; }
 
   virtual void SetSubresourceSpeculation() OVERRIDE {}
   virtual void SetOmniboxSpeculation() OVERRIDE {}
   virtual bool WasEverUsed() const OVERRIDE { return false; }
-  virtual bool UsingTCPFastOpen() const OVERRIDE { return false; }
+  virtual void EnableTCPFastOpen() OVERRIDE { use_tcp_fastopen_ = true; }
+  virtual bool UsingTCPFastOpen() const OVERRIDE { return use_tcp_fastopen_; }
   virtual bool WasNpnNegotiated() const OVERRIDE { return false; }
   virtual NextProto GetNegotiatedProtocol() const OVERRIDE {
     return kProtoUnknown;
   }
   virtual bool GetSSLInfo(SSLInfo* ssl_info) OVERRIDE { return false; }
 
   // Socket implementation.
   virtual int Read(IOBuffer* buf,
                    int buf_len,
                    const CompletionCallback& callback) OVERRIDE {
@@ skipping 12 matching lines @@
   void DoCallback() {
     is_connected_ = should_connect_;
     callback_.Run(is_connected_ ? OK : ERR_CONNECTION_FAILED);
   }
 
   bool should_connect_;
   bool is_connected_;
   const AddressList addrlist_;
   BoundNetLog net_log_;
   CompletionCallback callback_;
+  bool use_tcp_fastopen_;
 
   base::WeakPtrFactory<MockTriggerableClientSocket> weak_factory_;
 
   DISALLOW_COPY_AND_ASSIGN(MockTriggerableClientSocket);
 };
 
 }  // namespace
 
 void TestLoadTimingInfoConnectedReused(const ClientSocketHandle& handle) {
   LoadTimingInfo load_timing_info;
@@ skipping 138 matching lines @@
     run_loop_quit_closure_ = run_loop.QuitClosure();
     run_loop.Run();
     run_loop_quit_closure_.Reset();
   }
   base::Closure trigger = triggerable_sockets_.front();
   triggerable_sockets_.pop();
   return trigger;
 }
 
 }  // namespace net