OLD | NEW |
1 // Copyright (c) 2009 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2009 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_H_ | 5 #ifndef NET_SOCKET_SOCKS_CLIENT_SOCKET_H_ |
6 #define NET_SOCKET_SOCKS_CLIENT_SOCKET_H_ | 6 #define NET_SOCKET_SOCKS_CLIENT_SOCKET_H_ |
7 | 7 |
8 #include <string> | 8 #include <string> |
9 | 9 |
10 #include "base/logging.h" | 10 #include "base/logging.h" |
11 #include "base/ref_counted.h" | 11 #include "base/ref_counted.h" |
12 #include "base/scoped_ptr.h" | 12 #include "base/scoped_ptr.h" |
13 #include "googleurl/src/gurl.h" | 13 #include "googleurl/src/gurl.h" |
14 #include "net/base/address_list.h" | 14 #include "net/base/address_list.h" |
15 #include "net/base/completion_callback.h" | 15 #include "net/base/completion_callback.h" |
16 #include "net/base/host_resolver.h" | 16 #include "net/base/host_resolver.h" |
17 #include "net/base/net_errors.h" | 17 #include "net/base/net_errors.h" |
| 18 #include "net/base/net_log.h" |
18 #include "net/socket/client_socket.h" | 19 #include "net/socket/client_socket.h" |
19 #include "testing/gtest/include/gtest/gtest_prod.h" | 20 #include "testing/gtest/include/gtest/gtest_prod.h" |
20 | 21 |
21 namespace net { | 22 namespace net { |
22 | 23 |
23 class LoadLog; | 24 class BoundNetLog; |
24 | 25 |
25 // The SOCKS client socket implementation | 26 // The SOCKS client socket implementation |
26 class SOCKSClientSocket : public ClientSocket { | 27 class SOCKSClientSocket : public ClientSocket { |
27 public: | 28 public: |
28 // Takes ownership of the |transport_socket|, which should already be | 29 // Takes ownership of the |transport_socket|, which should already be |
29 // connected by the time Connect() is called. | 30 // connected by the time Connect() is called. |
30 // | 31 // |
31 // |req_info| contains the hostname and port to which the socket above will | 32 // |req_info| contains the hostname and port to which the socket above will |
32 // communicate to via the socks layer. For testing the referrer is optional. | 33 // communicate to via the socks layer. For testing the referrer is optional. |
33 SOCKSClientSocket(ClientSocket* transport_socket, | 34 SOCKSClientSocket(ClientSocket* transport_socket, |
34 const HostResolver::RequestInfo& req_info, | 35 const HostResolver::RequestInfo& req_info, |
35 HostResolver* host_resolver); | 36 HostResolver* host_resolver); |
36 | 37 |
37 // On destruction Disconnect() is called. | 38 // On destruction Disconnect() is called. |
38 virtual ~SOCKSClientSocket(); | 39 virtual ~SOCKSClientSocket(); |
39 | 40 |
40 // ClientSocket methods: | 41 // ClientSocket methods: |
41 | 42 |
42 // Does the SOCKS handshake and completes the protocol. | 43 // Does the SOCKS handshake and completes the protocol. |
43 virtual int Connect(CompletionCallback* callback, LoadLog* load_log); | 44 virtual int Connect(CompletionCallback* callback, const BoundNetLog& net_log); |
44 virtual void Disconnect(); | 45 virtual void Disconnect(); |
45 virtual bool IsConnected() const; | 46 virtual bool IsConnected() const; |
46 virtual bool IsConnectedAndIdle() const; | 47 virtual bool IsConnectedAndIdle() const; |
47 | 48 |
48 // Socket methods: | 49 // Socket methods: |
49 virtual int Read(IOBuffer* buf, int buf_len, CompletionCallback* callback); | 50 virtual int Read(IOBuffer* buf, int buf_len, CompletionCallback* callback); |
50 virtual int Write(IOBuffer* buf, int buf_len, CompletionCallback* callback); | 51 virtual int Write(IOBuffer* buf, int buf_len, CompletionCallback* callback); |
51 | 52 |
52 virtual bool SetReceiveBufferSize(int32 size); | 53 virtual bool SetReceiveBufferSize(int32 size); |
53 virtual bool SetSendBufferSize(int32 size); | 54 virtual bool SetSendBufferSize(int32 size); |
(...skipping 64 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
118 | 119 |
119 // These contain the bytes sent / received by the SOCKS handshake. | 120 // These contain the bytes sent / received by the SOCKS handshake. |
120 size_t bytes_sent_; | 121 size_t bytes_sent_; |
121 size_t bytes_received_; | 122 size_t bytes_received_; |
122 | 123 |
123 // Used to resolve the hostname to which the SOCKS proxy will connect. | 124 // Used to resolve the hostname to which the SOCKS proxy will connect. |
124 SingleRequestHostResolver host_resolver_; | 125 SingleRequestHostResolver host_resolver_; |
125 AddressList addresses_; | 126 AddressList addresses_; |
126 HostResolver::RequestInfo host_request_info_; | 127 HostResolver::RequestInfo host_request_info_; |
127 | 128 |
128 scoped_refptr<LoadLog> load_log_; | 129 BoundNetLog net_log_; |
129 | 130 |
130 DISALLOW_COPY_AND_ASSIGN(SOCKSClientSocket); | 131 DISALLOW_COPY_AND_ASSIGN(SOCKSClientSocket); |
131 }; | 132 }; |
132 | 133 |
133 } // namespace net | 134 } // namespace net |
134 | 135 |
135 #endif // NET_SOCKET_SOCKS_CLIENT_SOCKET_H_ | 136 #endif // NET_SOCKET_SOCKS_CLIENT_SOCKET_H_ |
OLD | NEW |