Index: net/socket/tcp_client_socket.h |
diff --git a/net/socket/tcp_client_socket.h b/net/socket/tcp_client_socket.h |
index 8a2c0cd73f06f944ae5aa7f5ba4520542a5b9b72..0e27f81a02f00a264b95d75fe763c0a5bef2e406 100644 |
--- a/net/socket/tcp_client_socket.h |
+++ b/net/socket/tcp_client_socket.h |
@@ -8,21 +8,25 @@ |
#include "build/build_config.h" |
#include "net/base/net_export.h" |
+// TODO(yzshen): Switch OS_POSIX to use the same platform-independent |
+// TCPClientSocket. |
#if defined(OS_WIN) |
-#include "net/socket/tcp_client_socket_win.h" |
+ |
+#include "base/basictypes.h" |
+#include "base/compiler_specific.h" |
+#include "base/memory/scoped_ptr.h" |
+#include "net/base/address_list.h" |
+#include "net/base/completion_callback.h" |
+#include "net/base/net_log.h" |
+#include "net/socket/stream_socket.h" |
+#include "net/socket/tcp_socket.h" |
+ |
#elif defined(OS_POSIX) |
akalin
2013/09/05 22:52:28
test for OS_POSIX first to match block below?
yzshen1
2013/09/06 00:57:44
Done. It will be removed in the next CL.
|
#include "net/socket/tcp_client_socket_libevent.h" |
#endif |
namespace net { |
-// A client socket that uses TCP as the transport layer. |
-#if defined(OS_WIN) |
-typedef TCPClientSocketWin TCPClientSocket; |
-#elif defined(OS_POSIX) |
-typedef TCPClientSocketLibevent TCPClientSocket; |
-#endif |
- |
// Enable/disable experimental TCP FastOpen option. |
// Not thread safe. Must be called during initialization/startup only. |
NET_EXPORT void SetTCPFastOpenEnabled(bool value); |
@@ -30,6 +34,113 @@ NET_EXPORT void SetTCPFastOpenEnabled(bool value); |
// Check if the TCP FastOpen option is enabled. |
bool IsTCPFastOpenEnabled(); |
+// A client socket that uses TCP as the transport layer. |
+#if defined(OS_POSIX) |
+typedef TCPClientSocketLibevent TCPClientSocket; |
+#elif defined(OS_WIN) |
+ |
+class NET_EXPORT TCPClientSocket : public StreamSocket { |
+ public: |
+ // The IP address(es) and port number to connect to. The TCP socket will try |
+ // each IP address in the list until it succeeds in establishing a |
+ // connection. |
+ TCPClientSocket(const AddressList& addresses, |
+ net::NetLog* net_log, |
+ const net::NetLog::Source& source); |
+ |
+ // Adopts the given, connected socket. This object takes ownership of the |
+ // given socket and then acts as if Connect() had been called. This function |
akalin
2013/09/05 22:52:28
No need to mention ownership, since scoped_ptr<> i
yzshen1
2013/09/06 00:57:44
Done.
|
+ // is used by TCPServerSocket to adopt accepted connections and for testing. |
+ explicit TCPClientSocket(scoped_ptr<TCPSocket> connected_socket, |
akalin
2013/09/05 22:52:28
no need for explicit
yzshen1
2013/09/06 00:57:44
Done.
|
+ const IPEndPoint& peer_address); |
+ |
+ virtual ~TCPClientSocket(); |
+ |
+ // Binds the socket to a local IP address and port. |
+ int Bind(const IPEndPoint& address); |
+ |
+ // StreamSocket implementation. |
+ virtual int Connect(const CompletionCallback& callback); |
+ virtual void Disconnect(); |
+ virtual bool IsConnected() const; |
+ virtual bool IsConnectedAndIdle() const; |
+ virtual int GetPeerAddress(IPEndPoint* address) const; |
+ virtual int GetLocalAddress(IPEndPoint* address) const; |
+ virtual const BoundNetLog& NetLog() const { return socket_->net_log(); } |
+ virtual void SetSubresourceSpeculation(); |
+ virtual void SetOmniboxSpeculation(); |
+ virtual bool WasEverUsed() const; |
+ virtual bool UsingTCPFastOpen() const; |
+ virtual bool WasNpnNegotiated() const OVERRIDE; |
+ virtual NextProto GetNegotiatedProtocol() const OVERRIDE; |
+ virtual bool GetSSLInfo(SSLInfo* ssl_info) OVERRIDE; |
+ |
+ // Socket implementation. |
+ // Multiple outstanding requests are not supported. |
+ // Full duplex mode (reading and writing at the same time) is supported |
akalin
2013/09/05 22:52:28
period at end
yzshen1
2013/09/06 00:57:44
Done.
|
+ virtual int Read(IOBuffer* buf, int buf_len, |
+ const CompletionCallback& callback) OVERRIDE; |
+ virtual int Write(IOBuffer* buf, int buf_len, |
+ const CompletionCallback& callback) OVERRIDE; |
+ virtual bool SetReceiveBufferSize(int32 size) OVERRIDE; |
+ virtual bool SetSendBufferSize(int32 size) OVERRIDE; |
+ |
+ virtual bool SetKeepAlive(bool enable, int delay); |
+ virtual bool SetNoDelay(bool no_delay); |
+ |
+ private: |
+ // State machine for connecting the socket. |
+ enum ConnectState { |
+ CONNECT_STATE_CONNECT, |
+ CONNECT_STATE_CONNECT_COMPLETE, |
+ CONNECT_STATE_NONE, |
+ }; |
+ |
+ // State machine used by Connect(). |
+ int DoConnectLoop(int result); |
+ int DoConnect(); |
+ int DoConnectComplete(int result); |
+ |
+ // Helper used by Disconnect(), which disconnects minus resetting |
+ // current_address_index_ and bind_address_. |
+ void DoDisconnect(); |
+ |
+ void DidCompleteConnect(int result); |
+ void DidCompleteReadWrite(const CompletionCallback& forward_callback, |
+ int result); |
+ |
+ int CreateSocket(AddressFamily family); |
+ |
+ scoped_ptr<TCPSocket> socket_; |
+ |
+ // Local IP address and port we are bound to. Set to NULL if Bind() |
+ // was't called (in that cases OS chooses address/port). |
akalin
2013/09/05 22:52:28
was't -> wasn't
in that cases -> in that case
yzshen1
2013/09/06 00:57:44
Done.
|
+ scoped_ptr<IPEndPoint> bind_address_; |
+ |
+ // The list of addresses we should try in order to establish a connection. |
+ AddressList addresses_; |
+ |
+ // Where we are in above list. Set to -1 if uninitialized. |
+ int current_address_index_; |
+ |
+ // External callback; called when connect is complete. |
+ CompletionCallback connect_callback_; |
+ |
+ // The next state for the Connect() state machine. |
+ ConnectState next_connect_state_; |
+ |
+ // This socket was previously disconnected and has not been re-connected. |
+ bool previously_disconnected_; |
+ |
+ // Record of connectivity and transmissions, for use in speculative connection |
+ // histograms. |
+ UseHistory use_history_; |
+ |
+ DISALLOW_COPY_AND_ASSIGN(TCPClientSocket); |
+}; |
+ |
+#endif |
+ |
} // namespace net |
#endif // NET_SOCKET_TCP_CLIENT_SOCKET_H_ |