| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #ifndef NET_SOCKET_SOCKS_CLIENT_SOCKET_POOL_H_ | 5 #ifndef NET_SOCKET_SOCKS_CLIENT_SOCKET_POOL_H_ |
| 6 #define NET_SOCKET_SOCKS_CLIENT_SOCKET_POOL_H_ | 6 #define NET_SOCKET_SOCKS_CLIENT_SOCKET_POOL_H_ |
| 7 | 7 |
| 8 #include <string> | 8 #include <string> |
| 9 | 9 |
| 10 #include "base/basictypes.h" | 10 #include "base/basictypes.h" |
| (...skipping 45 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 56 class SOCKSConnectJob : public ConnectJob { | 56 class SOCKSConnectJob : public ConnectJob { |
| 57 public: | 57 public: |
| 58 SOCKSConnectJob(const std::string& group_name, | 58 SOCKSConnectJob(const std::string& group_name, |
| 59 RequestPriority priority, | 59 RequestPriority priority, |
| 60 const scoped_refptr<SOCKSSocketParams>& params, | 60 const scoped_refptr<SOCKSSocketParams>& params, |
| 61 const base::TimeDelta& timeout_duration, | 61 const base::TimeDelta& timeout_duration, |
| 62 TransportClientSocketPool* transport_pool, | 62 TransportClientSocketPool* transport_pool, |
| 63 HostResolver* host_resolver, | 63 HostResolver* host_resolver, |
| 64 Delegate* delegate, | 64 Delegate* delegate, |
| 65 NetLog* net_log); | 65 NetLog* net_log); |
| 66 virtual ~SOCKSConnectJob(); | 66 ~SOCKSConnectJob() override; |
| 67 | 67 |
| 68 // ConnectJob methods. | 68 // ConnectJob methods. |
| 69 virtual LoadState GetLoadState() const override; | 69 LoadState GetLoadState() const override; |
| 70 | 70 |
| 71 private: | 71 private: |
| 72 enum State { | 72 enum State { |
| 73 STATE_TRANSPORT_CONNECT, | 73 STATE_TRANSPORT_CONNECT, |
| 74 STATE_TRANSPORT_CONNECT_COMPLETE, | 74 STATE_TRANSPORT_CONNECT_COMPLETE, |
| 75 STATE_SOCKS_CONNECT, | 75 STATE_SOCKS_CONNECT, |
| 76 STATE_SOCKS_CONNECT_COMPLETE, | 76 STATE_SOCKS_CONNECT_COMPLETE, |
| 77 STATE_NONE, | 77 STATE_NONE, |
| 78 }; | 78 }; |
| 79 | 79 |
| 80 void OnIOComplete(int result); | 80 void OnIOComplete(int result); |
| 81 | 81 |
| 82 // Runs the state transition loop. | 82 // Runs the state transition loop. |
| 83 int DoLoop(int result); | 83 int DoLoop(int result); |
| 84 | 84 |
| 85 int DoTransportConnect(); | 85 int DoTransportConnect(); |
| 86 int DoTransportConnectComplete(int result); | 86 int DoTransportConnectComplete(int result); |
| 87 int DoSOCKSConnect(); | 87 int DoSOCKSConnect(); |
| 88 int DoSOCKSConnectComplete(int result); | 88 int DoSOCKSConnectComplete(int result); |
| 89 | 89 |
| 90 // Begins the transport connection and the SOCKS handshake. Returns OK on | 90 // Begins the transport connection and the SOCKS handshake. Returns OK on |
| 91 // success and ERR_IO_PENDING if it cannot immediately service the request. | 91 // success and ERR_IO_PENDING if it cannot immediately service the request. |
| 92 // Otherwise, it returns a net error code. | 92 // Otherwise, it returns a net error code. |
| 93 virtual int ConnectInternal() override; | 93 int ConnectInternal() override; |
| 94 | 94 |
| 95 scoped_refptr<SOCKSSocketParams> socks_params_; | 95 scoped_refptr<SOCKSSocketParams> socks_params_; |
| 96 TransportClientSocketPool* const transport_pool_; | 96 TransportClientSocketPool* const transport_pool_; |
| 97 HostResolver* const resolver_; | 97 HostResolver* const resolver_; |
| 98 | 98 |
| 99 State next_state_; | 99 State next_state_; |
| 100 CompletionCallback callback_; | 100 CompletionCallback callback_; |
| 101 scoped_ptr<ClientSocketHandle> transport_socket_handle_; | 101 scoped_ptr<ClientSocketHandle> transport_socket_handle_; |
| 102 scoped_ptr<StreamSocket> socket_; | 102 scoped_ptr<StreamSocket> socket_; |
| 103 | 103 |
| 104 DISALLOW_COPY_AND_ASSIGN(SOCKSConnectJob); | 104 DISALLOW_COPY_AND_ASSIGN(SOCKSConnectJob); |
| 105 }; | 105 }; |
| 106 | 106 |
| 107 class NET_EXPORT_PRIVATE SOCKSClientSocketPool | 107 class NET_EXPORT_PRIVATE SOCKSClientSocketPool |
| 108 : public ClientSocketPool, public HigherLayeredPool { | 108 : public ClientSocketPool, public HigherLayeredPool { |
| 109 public: | 109 public: |
| 110 typedef SOCKSSocketParams SocketParams; | 110 typedef SOCKSSocketParams SocketParams; |
| 111 | 111 |
| 112 SOCKSClientSocketPool( | 112 SOCKSClientSocketPool( |
| 113 int max_sockets, | 113 int max_sockets, |
| 114 int max_sockets_per_group, | 114 int max_sockets_per_group, |
| 115 ClientSocketPoolHistograms* histograms, | 115 ClientSocketPoolHistograms* histograms, |
| 116 HostResolver* host_resolver, | 116 HostResolver* host_resolver, |
| 117 TransportClientSocketPool* transport_pool, | 117 TransportClientSocketPool* transport_pool, |
| 118 NetLog* net_log); | 118 NetLog* net_log); |
| 119 | 119 |
| 120 virtual ~SOCKSClientSocketPool(); | 120 ~SOCKSClientSocketPool() override; |
| 121 | 121 |
| 122 // ClientSocketPool implementation. | 122 // ClientSocketPool implementation. |
| 123 virtual int RequestSocket(const std::string& group_name, | 123 int RequestSocket(const std::string& group_name, |
| 124 const void* connect_params, | 124 const void* connect_params, |
| 125 RequestPriority priority, | 125 RequestPriority priority, |
| 126 ClientSocketHandle* handle, | 126 ClientSocketHandle* handle, |
| 127 const CompletionCallback& callback, | 127 const CompletionCallback& callback, |
| 128 const BoundNetLog& net_log) override; | 128 const BoundNetLog& net_log) override; |
| 129 | 129 |
| 130 virtual void RequestSockets(const std::string& group_name, | 130 void RequestSockets(const std::string& group_name, |
| 131 const void* params, | 131 const void* params, |
| 132 int num_sockets, | 132 int num_sockets, |
| 133 const BoundNetLog& net_log) override; | 133 const BoundNetLog& net_log) override; |
| 134 | 134 |
| 135 virtual void CancelRequest(const std::string& group_name, | 135 void CancelRequest(const std::string& group_name, |
| 136 ClientSocketHandle* handle) override; | 136 ClientSocketHandle* handle) override; |
| 137 | 137 |
| 138 virtual void ReleaseSocket(const std::string& group_name, | 138 void ReleaseSocket(const std::string& group_name, |
| 139 scoped_ptr<StreamSocket> socket, | 139 scoped_ptr<StreamSocket> socket, |
| 140 int id) override; | 140 int id) override; |
| 141 | 141 |
| 142 virtual void FlushWithError(int error) override; | 142 void FlushWithError(int error) override; |
| 143 | 143 |
| 144 virtual void CloseIdleSockets() override; | 144 void CloseIdleSockets() override; |
| 145 | 145 |
| 146 virtual int IdleSocketCount() const override; | 146 int IdleSocketCount() const override; |
| 147 | 147 |
| 148 virtual int IdleSocketCountInGroup( | 148 int IdleSocketCountInGroup(const std::string& group_name) const override; |
| 149 const std::string& group_name) const override; | |
| 150 | 149 |
| 151 virtual LoadState GetLoadState( | 150 LoadState GetLoadState(const std::string& group_name, |
| 152 const std::string& group_name, | 151 const ClientSocketHandle* handle) const override; |
| 153 const ClientSocketHandle* handle) const override; | |
| 154 | 152 |
| 155 virtual base::DictionaryValue* GetInfoAsValue( | 153 base::DictionaryValue* GetInfoAsValue( |
| 156 const std::string& name, | 154 const std::string& name, |
| 157 const std::string& type, | 155 const std::string& type, |
| 158 bool include_nested_pools) const override; | 156 bool include_nested_pools) const override; |
| 159 | 157 |
| 160 virtual base::TimeDelta ConnectionTimeout() const override; | 158 base::TimeDelta ConnectionTimeout() const override; |
| 161 | 159 |
| 162 virtual ClientSocketPoolHistograms* histograms() const override; | 160 ClientSocketPoolHistograms* histograms() const override; |
| 163 | 161 |
| 164 // LowerLayeredPool implementation. | 162 // LowerLayeredPool implementation. |
| 165 virtual bool IsStalled() const override; | 163 bool IsStalled() const override; |
| 166 | 164 |
| 167 virtual void AddHigherLayeredPool(HigherLayeredPool* higher_pool) override; | 165 void AddHigherLayeredPool(HigherLayeredPool* higher_pool) override; |
| 168 | 166 |
| 169 virtual void RemoveHigherLayeredPool(HigherLayeredPool* higher_pool) override; | 167 void RemoveHigherLayeredPool(HigherLayeredPool* higher_pool) override; |
| 170 | 168 |
| 171 // HigherLayeredPool implementation. | 169 // HigherLayeredPool implementation. |
| 172 virtual bool CloseOneIdleConnection() override; | 170 bool CloseOneIdleConnection() override; |
| 173 | 171 |
| 174 private: | 172 private: |
| 175 typedef ClientSocketPoolBase<SOCKSSocketParams> PoolBase; | 173 typedef ClientSocketPoolBase<SOCKSSocketParams> PoolBase; |
| 176 | 174 |
| 177 class SOCKSConnectJobFactory : public PoolBase::ConnectJobFactory { | 175 class SOCKSConnectJobFactory : public PoolBase::ConnectJobFactory { |
| 178 public: | 176 public: |
| 179 SOCKSConnectJobFactory(TransportClientSocketPool* transport_pool, | 177 SOCKSConnectJobFactory(TransportClientSocketPool* transport_pool, |
| 180 HostResolver* host_resolver, | 178 HostResolver* host_resolver, |
| 181 NetLog* net_log) | 179 NetLog* net_log) |
| 182 : transport_pool_(transport_pool), | 180 : transport_pool_(transport_pool), |
| 183 host_resolver_(host_resolver), | 181 host_resolver_(host_resolver), |
| 184 net_log_(net_log) {} | 182 net_log_(net_log) {} |
| 185 | 183 |
| 186 virtual ~SOCKSConnectJobFactory() {} | 184 ~SOCKSConnectJobFactory() override {} |
| 187 | 185 |
| 188 // ClientSocketPoolBase::ConnectJobFactory methods. | 186 // ClientSocketPoolBase::ConnectJobFactory methods. |
| 189 virtual scoped_ptr<ConnectJob> NewConnectJob( | 187 scoped_ptr<ConnectJob> NewConnectJob( |
| 190 const std::string& group_name, | 188 const std::string& group_name, |
| 191 const PoolBase::Request& request, | 189 const PoolBase::Request& request, |
| 192 ConnectJob::Delegate* delegate) const override; | 190 ConnectJob::Delegate* delegate) const override; |
| 193 | 191 |
| 194 virtual base::TimeDelta ConnectionTimeout() const override; | 192 base::TimeDelta ConnectionTimeout() const override; |
| 195 | 193 |
| 196 private: | 194 private: |
| 197 TransportClientSocketPool* const transport_pool_; | 195 TransportClientSocketPool* const transport_pool_; |
| 198 HostResolver* const host_resolver_; | 196 HostResolver* const host_resolver_; |
| 199 NetLog* net_log_; | 197 NetLog* net_log_; |
| 200 | 198 |
| 201 DISALLOW_COPY_AND_ASSIGN(SOCKSConnectJobFactory); | 199 DISALLOW_COPY_AND_ASSIGN(SOCKSConnectJobFactory); |
| 202 }; | 200 }; |
| 203 | 201 |
| 204 TransportClientSocketPool* const transport_pool_; | 202 TransportClientSocketPool* const transport_pool_; |
| 205 PoolBase base_; | 203 PoolBase base_; |
| 206 | 204 |
| 207 DISALLOW_COPY_AND_ASSIGN(SOCKSClientSocketPool); | 205 DISALLOW_COPY_AND_ASSIGN(SOCKSClientSocketPool); |
| 208 }; | 206 }; |
| 209 | 207 |
| 210 } // namespace net | 208 } // namespace net |
| 211 | 209 |
| 212 #endif // NET_SOCKET_SOCKS_CLIENT_SOCKET_POOL_H_ | 210 #endif // NET_SOCKET_SOCKS_CLIENT_SOCKET_POOL_H_ |
| OLD | NEW |