Respect cookies set in a 401 responses when restarting the http transaction....

net/http/http_network_transaction.h

Index: net/http/http_network_transaction.h
===================================================================
--- net/http/http_network_transaction.h	(revision 12402)
+++ net/http/http_network_transaction.h	(working copy)
@@ -41,6 +41,11 @@
   virtual int RestartWithAuth(const std::wstring& username,
                               const std::wstring& password,
                               CompletionCallback* callback);
+  virtual bool IsReadyToRestartForAuth() {
+    return pending_auth_target_ != HttpAuth::AUTH_NONE &&
+        HaveAuth(pending_auth_target_);
+  }
+
   virtual int Read(IOBuffer* buf, int buf_len, CompletionCallback* callback);
   virtual const HttpResponseInfo* GetResponseInfo() const;
   virtual LoadState GetLoadState() const;
@@ -152,10 +157,8 @@
   void AddAuthorizationHeader(HttpAuth::Target target);
 
   // Handles HTTP status code 401 or 407.
-  // HandleAuthChallenge() returns a network error code, or OK, or
-  // WILL_RESTART_TRANSACTION. The latter indicates that the state machine has
-  // been updated to restart the transaction with a new auth attempt.
-  enum { WILL_RESTART_TRANSACTION = 1 };
+  // HandleAuthChallenge() returns a network error code, or OK on success.
+  // May update |pending_auth_target_| or |response_.auth_challenge|.
   int HandleAuthChallenge();
 
   // Populates response_.auth_challenge with the challenge information, so that
@@ -177,10 +180,6 @@
   // auth_handler_[target] with the cache entry's data and returns true.
   bool SelectPreemptiveAuth(HttpAuth::Target target);
 
-  bool NeedAuth(HttpAuth::Target target) const {
-    return auth_handler_[target].get() && auth_identity_[target].invalid;
-  }
-
   bool HaveAuth(HttpAuth::Target target) const {
     return auth_handler_[target].get() && !auth_identity_[target].invalid;
   }
@@ -206,6 +205,11 @@
   // a number of places (url, cache, prompt).
   HttpAuth::Identity auth_identity_[2];
 
+  // Whether this transaction is waiting for proxy auth, server auth, or is
+  // not waiting for any auth at all. |pending_auth_target_| is read and
+  // cleared by RestartWithAuth().
+  HttpAuth::Target pending_auth_target_;
+
   CompletionCallbackImpl<HttpNetworkTransaction> io_callback_;
   CompletionCallback* user_callback_;