Add proxy basic auth support in net/socket_stream.

net/socket_stream/socket_stream.h

 // Copyright (c) 2009 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #ifndef NET_SOCKET_STREAM_SOCKET_STREAM_H_
 #define NET_SOCKET_STREAM_SOCKET_STREAM_H_
 
 #include <deque>
+#include <map>
 #include <string>
 #include <vector>
 
 #include "base/linked_ptr.h"
 #include "base/ref_counted.h"
 #include "base/scoped_ptr.h"
 #include "base/task.h"
 #include "net/base/address_list.h"
 #include "net/base/completion_callback.h"
 #include "net/base/io_buffer.h"
+#include "net/http/http_auth.h"
+#include "net/http/http_auth_cache.h"
+#include "net/http/http_auth_handler.h"
 #include "net/proxy/proxy_service.h"
 #include "net/socket/tcp_client_socket.h"
 #include "net/url_request/url_request_context.h"
 
 namespace net {
 
+class AuthChallengeInfo;
 class ClientSocketFactory;
 class HostResolver;
 class SSLConfigService;
 class SingleRequestHostResolver;
 
+// SocketStream is used to implement Web Sockets.
+// It provides plain full-duplex stream with proxy and SSL support.
+// For proxy authentication, only basic mechanisum is supported.  It will try
+// authentication identity for proxy URL first.  If server requires proxy
+// authentication, it will try authentication identity for realm that server
+// requests.
 class SocketStream : public base::RefCountedThreadSafe<SocketStream> {
  public:
   // Derive from this class and add your own data members to associate extra
   // information with a SocketStream.  Use GetUserData(key) and
   // SetUserData(key, data).
   class UserData {
    public:
     UserData() {}
     virtual ~UserData() {}
   };
@@ skipping 12 matching lines @@
     // Called when |amount_sent| bytes of data are sent.
     virtual void OnSentData(SocketStream* socket,
                             int amount_sent) = 0;
 
     // Called when |len| bytes of |data| are received.
     virtual void OnReceivedData(SocketStream* socket,
                                 const char* data, int len) = 0;
 
     // Called when the socket stream has been closed.
     virtual void OnClose(SocketStream* socket) = 0;
+
+    // Called when proxy authentication required.
+    // The delegate should call RestartWithAuth() if credential for |auth_info|
+    // is found in password database, or call Close() to close the connection.
+    virtual void OnAuthRequired(SocketStream* socket,
+                                AuthChallengeInfo* auth_info) {
+      // By default, no credential is available and close the connection.
+      socket->Close();
+    }
   };
 
   SocketStream(const GURL& url, Delegate* delegate);
 
   // The user data allows the clients to associate data with this job.
   // Multiple user data values can be stored under different keys.
   // This job will TAKE OWNERSHIP of the given data pointer, and will
   // delete the object if it is changed or the job is destroyed.
   UserData* GetUserData(const void* key) const;
   void SetUserData(const void* key, UserData* data);
@@ skipping 12 matching lines @@
   // Requests to send |len| bytes of |data| on the connection.
   // Returns true if |data| is buffered in the job.
   // Returns false if size of buffered data would exceeds
   // |max_pending_send_allowed_| and |data| is not sent at all.
   bool SendData(const char* data, int len);
 
   // Requests to close the connection.
   // Once the connection is closed, calls delegate's OnClose.
   void Close();
 
+  // Restarts with authentication info.
+  // Should be used for response of OnAuthRequired.
+  void RestartWithAuth(
+      const std::wstring& username,
+      const std::wstring& password);
+
   // Detach delegate.  Call before delegate is deleted.
   // Once delegate is detached, close the socket stream and never call delegate
   // back.
   void DetachDelegate();
 
   // Sets an alternative HostResolver. For testing purposes only.
   void SetHostResolver(HostResolver* host_resolver);
 
   // Sets an alternative ClientSocketFactory.  Doesn't take ownership of
   // |factory|.  For testing purposes only.
@@ skipping 34 matching lines @@
     STATE_TCP_CONNECT,
     STATE_TCP_CONNECT_COMPLETE,
     STATE_WRITE_TUNNEL_HEADERS,
     STATE_WRITE_TUNNEL_HEADERS_COMPLETE,
     STATE_READ_TUNNEL_HEADERS,
     STATE_READ_TUNNEL_HEADERS_COMPLETE,
     STATE_SOCKS_CONNECT,
     STATE_SOCKS_CONNECT_COMPLETE,
     STATE_SSL_CONNECT,
     STATE_SSL_CONNECT_COMPLETE,
-    STATE_CONNECTION_ESTABLISHED,
     STATE_READ_WRITE
   };
 
   enum ProxyMode {
     kDirectConnection,  // If using a direct connection
     kTunnelProxy,  // If using a tunnel (CONNECT method as HTTPS)
     kSOCKSProxy,  // If using a SOCKS proxy
   };
 
   typedef std::deque< scoped_refptr<IOBufferWithSize> > PendingDataQueue;
@@ skipping 23 matching lines @@
   int DoWriteTunnelHeaders();
   int DoWriteTunnelHeadersComplete(int result);
   int DoReadTunnelHeaders();
   int DoReadTunnelHeadersComplete(int result);
   int DoSOCKSConnect();
   int DoSOCKSConnectComplete(int result);
   int DoSSLConnect();
   int DoSSLConnectComplete(int result);
   int DoReadWrite(int result);
 
+  GURL ProxyAuthOrigin() const;
+  int HandleAuthChallenge(const HttpResponseHeaders* headers);
+  void DoAuthRequired();
+  void DoRestartWithAuth();
+
   int HandleCertificateError(int result);
 
   bool is_secure() const;
   SSLConfigService* ssl_config_service() const;
   ProxyService* proxy_service() const;
 
   GURL url_;
   Delegate* delegate_;
   int max_pending_send_allowed_;
   scoped_refptr<URLRequestContext> context_;
 
   typedef std::map<const void*, linked_ptr<UserData> > UserDataMap;
   UserDataMap user_data_;
 
   State next_state_;
   scoped_refptr<HostResolver> host_resolver_;
   ClientSocketFactory* factory_;
 
   ProxyMode proxy_mode_;
 
   ProxyService::PacRequest* pac_request_;
   ProxyInfo proxy_info_;
 
+  HttpAuthCache auth_cache_;
+  scoped_refptr<HttpAuthHandler> auth_handler_;
+  HttpAuth::Identity auth_identity_;
+  scoped_refptr<AuthChallengeInfo> auth_info_;
+
   scoped_refptr<RequestHeaders> tunnel_request_headers_;
   size_t tunnel_request_headers_bytes_sent_;
   scoped_refptr<ResponseHeaders> tunnel_response_headers_;
   int tunnel_response_headers_capacity_;
   int tunnel_response_headers_len_;
 
   // Use the same number as HttpNetworkTransaction::kMaxHeaderBufSize.
   enum { kMaxTunnelResponseHeadersSize = 32768 };  // 32 kilobytes.
 
   scoped_ptr<SingleRequestHostResolver> resolver_;
@@ skipping 20 matching lines @@
   int write_buf_offset_;
   int write_buf_size_;
   PendingDataQueue pending_write_bufs_;
 
   DISALLOW_COPY_AND_ASSIGN(SocketStream);
 };
 
 }  // namespace net
 
 #endif  // NET_SOCKET_STREAM_SOCKET_STREAM_H_