Perform HTTP authentication over a keep-alive connection....

net/http/http_network_transaction.h

 // Copyright (c) 2006-2008 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_HTTP_HTTP_NETWORK_TRANSACTION_H_
 #define NET_HTTP_HTTP_NETWORK_TRANSACTION_H_
 
 #include <string>
 
 #include "base/ref_counted.h"
@@ skipping 57 matching lines @@
   int DoSSLConnectOverTunnel();
   int DoSSLConnectOverTunnelComplete(int result);
   int DoWriteHeaders();
   int DoWriteHeadersComplete(int result);
   int DoWriteBody();
   int DoWriteBodyComplete(int result);
   int DoReadHeaders();
   int DoReadHeadersComplete(int result);
   int DoReadBody();
   int DoReadBodyComplete(int result);
+  int DoDrainBodyForAuthRestart();
+  int DoDrainBodyForAuthRestartComplete(int result);
 
   // Record histogram of latency (first byte sent till last byte received) as
   // well as effective bandwidth used.
   void LogTransactionMetrics() const;
 
   // Called when header_buf_ contains the complete response headers.
   int DidReadResponseHeaders();
 
   // Called to handle a certificate error.  Returns OK if the error should be
   // ignored.  Otherwise, stores the certificate in response_.ssl_info and
@@ skipping 31 matching lines @@
   // Return true if based on the bytes read so far, the start of the
   // status line is known. This is used to distingish between HTTP/0.9
   // responses (which have no status line) and HTTP/1.x responses.
   bool has_found_status_line_start() const {
     return header_buf_http_offset_ != -1;
   }
 
   // Sets up the state machine to restart the transaction with auth.
   void PrepareForAuthRestart(HttpAuth::Target target);
 
+  // Called when we don't need to drain the response body or have drained it.
+  // Resets |connection_| unless |keep_alive| is true, then calls
+  // ResetStateForRestart.  Sets |next_state_| appropriately.
+  void DidDrainBodyForAuthRestart(bool keep_alive);
+
   // Resets the members of the transaction so it can be restarted.
   void ResetStateForRestart();
 
   // Attach any credentials needed for the proxy server or origin server.
   void ApplyAuth();
 
   // Helper used by ApplyAuth(). Adds either the proxy auth header, or the
   // origin server auth header, as specified by |target|
   void AddAuthorizationHeader(HttpAuth::Target target);
 
@@ skipping 99 matching lines @@
   // The number of bytes by which the header buffer is grown when it reaches
   // capacity.
   enum { kHeaderBufInitialSize = 4096 };
 
   // |kMaxHeaderBufSize| is the number of bytes that the response headers can
   // grow to. If the body start is not found within this range of the
   // response, the transaction will fail with ERR_RESPONSE_HEADERS_TOO_BIG.
   // Note: |kMaxHeaderBufSize| should be a multiple of |kHeaderBufInitialSize|.
   enum { kMaxHeaderBufSize = 32768 };  // 32 kilobytes.
 
+  // The size in bytes of the buffer we use to drain the response body that
+  // we want to throw away.  The response body is typically a small error
+  // page just a few hundred bytes long.
+  enum { kDrainBodyBufferSize = 1024 };
+
   // The position where status line starts; -1 if not found yet.
   int header_buf_http_offset_;
 
   // Indicates the content length remaining to read.  If this value is less
   // than zero (and chunked_decoder_ is null), then we read until the server
   // closes the connection.
   int64 content_length_;
 
   // Keeps track of the number of response body bytes read so far.
   int64 content_read_;
@@ skipping 16 matching lines @@
     STATE_SSL_CONNECT_OVER_TUNNEL,
     STATE_SSL_CONNECT_OVER_TUNNEL_COMPLETE,
     STATE_WRITE_HEADERS,
     STATE_WRITE_HEADERS_COMPLETE,
     STATE_WRITE_BODY,
     STATE_WRITE_BODY_COMPLETE,
     STATE_READ_HEADERS,
     STATE_READ_HEADERS_COMPLETE,
     STATE_READ_BODY,
     STATE_READ_BODY_COMPLETE,
+    STATE_DRAIN_BODY_FOR_AUTH_RESTART,
+    STATE_DRAIN_BODY_FOR_AUTH_RESTART_COMPLETE,
     STATE_NONE
   };
   State next_state_;
 };
 
 }  // namespace net
 
 #endif  // NET_HTTP_HTTP_NETWORK_TRANSACTION_H_