Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(19)

Side by Side Diff: net/socket/socks_client_socket.h

Issue 992733002: Remove //net (except for Android test stuff) and sdch (Closed) Base URL: git@github.com:domokit/mojo.git@master
Patch Set: Created 5 years, 9 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
« no previous file with comments | « net/socket/socks5_client_socket_unittest.cc ('k') | net/socket/socks_client_socket.cc » ('j') | no next file with comments »
Toggle Intra-line Diffs ('i') | Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
OLDNEW
(Empty)
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
3 // found in the LICENSE file.
4
5 #ifndef NET_SOCKET_SOCKS_CLIENT_SOCKET_H_
6 #define NET_SOCKET_SOCKS_CLIENT_SOCKET_H_
7
8 #include <string>
9
10 #include "base/basictypes.h"
11 #include "base/gtest_prod_util.h"
12 #include "base/memory/ref_counted.h"
13 #include "base/memory/scoped_ptr.h"
14 #include "net/base/address_list.h"
15 #include "net/base/completion_callback.h"
16 #include "net/base/net_errors.h"
17 #include "net/base/net_log.h"
18 #include "net/dns/host_resolver.h"
19 #include "net/dns/single_request_host_resolver.h"
20 #include "net/socket/stream_socket.h"
21
22 namespace net {
23
24 class ClientSocketHandle;
25 class BoundNetLog;
26
27 // The SOCKS client socket implementation
28 class NET_EXPORT_PRIVATE SOCKSClientSocket : public StreamSocket {
29 public:
30 // |req_info| contains the hostname and port to which the socket above will
31 // communicate to via the socks layer. For testing the referrer is optional.
32 SOCKSClientSocket(scoped_ptr<ClientSocketHandle> transport_socket,
33 const HostResolver::RequestInfo& req_info,
34 RequestPriority priority,
35 HostResolver* host_resolver);
36
37 // On destruction Disconnect() is called.
38 ~SOCKSClientSocket() override;
39
40 // StreamSocket implementation.
41
42 // Does the SOCKS handshake and completes the protocol.
43 int Connect(const CompletionCallback& callback) override;
44 void Disconnect() override;
45 bool IsConnected() const override;
46 bool IsConnectedAndIdle() const override;
47 const BoundNetLog& NetLog() const override;
48 void SetSubresourceSpeculation() override;
49 void SetOmniboxSpeculation() override;
50 bool WasEverUsed() const override;
51 bool UsingTCPFastOpen() const override;
52 bool WasNpnNegotiated() const override;
53 NextProto GetNegotiatedProtocol() const override;
54 bool GetSSLInfo(SSLInfo* ssl_info) override;
55
56 // Socket implementation.
57 int Read(IOBuffer* buf,
58 int buf_len,
59 const CompletionCallback& callback) override;
60 int Write(IOBuffer* buf,
61 int buf_len,
62 const CompletionCallback& callback) override;
63
64 int SetReceiveBufferSize(int32 size) override;
65 int SetSendBufferSize(int32 size) override;
66
67 int GetPeerAddress(IPEndPoint* address) const override;
68 int GetLocalAddress(IPEndPoint* address) const override;
69
70 private:
71 FRIEND_TEST_ALL_PREFIXES(SOCKSClientSocketTest, CompleteHandshake);
72 FRIEND_TEST_ALL_PREFIXES(SOCKSClientSocketTest, SOCKS4AFailedDNS);
73 FRIEND_TEST_ALL_PREFIXES(SOCKSClientSocketTest, SOCKS4AIfDomainInIPv6);
74
75 enum State {
76 STATE_RESOLVE_HOST,
77 STATE_RESOLVE_HOST_COMPLETE,
78 STATE_HANDSHAKE_WRITE,
79 STATE_HANDSHAKE_WRITE_COMPLETE,
80 STATE_HANDSHAKE_READ,
81 STATE_HANDSHAKE_READ_COMPLETE,
82 STATE_NONE,
83 };
84
85 void DoCallback(int result);
86 void OnIOComplete(int result);
87 void OnReadWriteComplete(const CompletionCallback& callback, int result);
88
89 int DoLoop(int last_io_result);
90 int DoResolveHost();
91 int DoResolveHostComplete(int result);
92 int DoHandshakeRead();
93 int DoHandshakeReadComplete(int result);
94 int DoHandshakeWrite();
95 int DoHandshakeWriteComplete(int result);
96
97 const std::string BuildHandshakeWriteBuffer() const;
98
99 // Stores the underlying socket.
100 scoped_ptr<ClientSocketHandle> transport_;
101
102 State next_state_;
103
104 // Stores the callbacks to the layer above, called on completing Connect().
105 CompletionCallback user_callback_;
106
107 // This IOBuffer is used by the class to read and write
108 // SOCKS handshake data. The length contains the expected size to
109 // read or write.
110 scoped_refptr<IOBuffer> handshake_buf_;
111
112 // While writing, this buffer stores the complete write handshake data.
113 // While reading, it stores the handshake information received so far.
114 std::string buffer_;
115
116 // This becomes true when the SOCKS handshake has completed and the
117 // overlying connection is free to communicate.
118 bool completed_handshake_;
119
120 // These contain the bytes sent / received by the SOCKS handshake.
121 size_t bytes_sent_;
122 size_t bytes_received_;
123
124 // This becomes true when the socket is used to send or receive data.
125 bool was_ever_used_;
126
127 // Used to resolve the hostname to which the SOCKS proxy will connect.
128 SingleRequestHostResolver host_resolver_;
129 AddressList addresses_;
130 HostResolver::RequestInfo host_request_info_;
131 RequestPriority priority_;
132
133 BoundNetLog net_log_;
134
135 DISALLOW_COPY_AND_ASSIGN(SOCKSClientSocket);
136 };
137
138 } // namespace net
139
140 #endif // NET_SOCKET_SOCKS_CLIENT_SOCKET_H_
OLDNEW
« no previous file with comments | « net/socket/socks5_client_socket_unittest.cc ('k') | net/socket/socks_client_socket.cc » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698