1 files changed, 12 insertions, 10 deletions
diff --git a/net/socket/socks_client_socket.h b/net/socket/socks_client_socket.h
index 03925ba..575f263 100644
--- a/net/socket/socks_client_socket.h
+++ b/net/socket/socks_client_socket.h
@@ -2,8 +2,8 @@
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
-#ifndef NET_BASE_SOCKS_CLIENT_SOCKET_H_
-#define NET_BASE_SOCKS_CLIENT_SOCKET_H_
+#ifndef NET_SOCKET_SOCKS_CLIENT_SOCKET_H_
+#define NET_SOCKET_SOCKS_CLIENT_SOCKET_H_
 
 #include <string>
 
@@ -54,6 +54,10 @@ class SOCKSClientSocket : public ClientSocket {
 #endif
 
  private:
+  FRIEND_TEST(SOCKSClientSocketTest, CompleteHandshake);
+  FRIEND_TEST(SOCKSClientSocketTest, SOCKS4AFailedDNS);
+  FRIEND_TEST(SOCKSClientSocketTest, SOCKS4AIfDomainInIPv6);
+
   enum State {
     STATE_RESOLVE_HOST,
     STATE_RESOLVE_HOST_COMPLETE,
@@ -85,7 +89,7 @@ class SOCKSClientSocket : public ClientSocket {
   int DoHandshakeWrite();
   int DoHandshakeWriteComplete(int result);
 
-  void BuildHandshakeWriteBuffer();
+  const std::string BuildHandshakeWriteBuffer() const;
 
   CompletionCallbackImpl<SOCKSClientSocket> io_callback_;
 
@@ -102,23 +106,21 @@ class SOCKSClientSocket : public ClientSocket {
   // SOCKS handshake data. The length contains the expected size to
   // read or write.
   scoped_refptr<IOBuffer> handshake_buf_;
-  int handshake_buf_len_;
 
   // While writing, this buffer stores the complete write handshake data.
   // While reading, it stores the handshake information received so far.
-  scoped_array<char> buffer_;
-  int buffer_len_;
+  std::string buffer_;
 
   // This becomes true when the SOCKS handshake has completed and the
   // overlying connection is free to communicate.
   bool completed_handshake_;
 
   // These contain the bytes sent / received by the SOCKS handshake.
-  int bytes_sent_;
-  int bytes_received_;
+  size_t bytes_sent_;
+  size_t bytes_received_;
 
   // Used to resolve the hostname to which the SOCKS proxy will connect.
-  SingleRequestHostResolver resolver_;
+  SingleRequestHostResolver host_resolver_;
   AddressList addresses_;
   HostResolver::RequestInfo host_request_info_;
 
@@ -127,5 +129,5 @@ class SOCKSClientSocket : public ClientSocket {
 
 }  // namespace net
 
-#endif  // NET_BASE_SOCKS_CLIENT_SOCKET_H_
+#endif  // NET_SOCKET_SOCKS_CLIENT_SOCKET_H_