src/content/browser/loader/resource_loader.h

   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 CONTENT_BROWSER_LOADER_RESOURCE_LOADER_H_
   6 #define CONTENT_BROWSER_LOADER_RESOURCE_LOADER_H_
   7
   8 #include "base/gtest_prod_util.h"
   9 #include "base/memory/scoped_ptr.h"
  10 #include "base/memory/weak_ptr.h"
  11 #include "content/browser/loader/resource_handler.h"
  12 #include "content/browser/ssl/ssl_error_handler.h"
  13 #include "content/common/content_export.h"
  14 #include "content/public/browser/resource_controller.h"
  15 #include "content/public/common/signed_certificate_timestamp_id_and_status.h"
  16 #include "net/url_request/url_request.h"
  17
  18 namespace content {
  19 class ResourceDispatcherHostLoginDelegate;
  20 class ResourceLoaderDelegate;
  21 class ResourceRequestInfoImpl;
  22 class SSLClientAuthHandler;
  23
  24 // This class is responsible for driving the URLRequest (i.e., calling Start,
  25 // Read, and servicing events).  It has a ResourceHandler, which is typically a
  26 // chain of ResourceHandlers, and is the ResourceController for its handler.
  27 class CONTENT_EXPORT ResourceLoader : public net::URLRequest::Delegate,
  28                                       public SSLErrorHandler::Delegate,
  29                                       public ResourceController {
  30  public:
  31   ResourceLoader(scoped_ptr<net::URLRequest> request,
  32                  scoped_ptr<ResourceHandler> handler,
  33                  ResourceLoaderDelegate* delegate);
  34   virtual ~ResourceLoader();
  35
  36   void StartRequest();
  37   void CancelRequest(bool from_renderer);
  38
  39   void ReportUploadProgress();
  40
  41   bool is_transferring() const { return is_transferring_; }
  42   void MarkAsTransferring();
  43   void CompleteTransfer();
  44
  45   net::URLRequest* request() { return request_.get(); }
  46   ResourceRequestInfoImpl* GetRequestInfo();
  47
  48   void ClearLoginDelegate();
  49   void ClearSSLClientAuthHandler();
  50
  51   // IPC message handlers:
  52   void OnUploadProgressACK();
  53
  54  private:
  55   FRIEND_TEST_ALL_PREFIXES(ResourceLoaderTest, ClientCertStoreLookup);
  56   FRIEND_TEST_ALL_PREFIXES(ResourceLoaderTest, ClientCertStoreNull);
  57
  58   // net::URLRequest::Delegate implementation:
  59   virtual void OnReceivedRedirect(net::URLRequest* request,
  60                                   const GURL& new_url,
  61                                   bool* defer) OVERRIDE;
  62   virtual void OnAuthRequired(net::URLRequest* request,
  63                               net::AuthChallengeInfo* info) OVERRIDE;
  64   virtual void OnCertificateRequested(net::URLRequest* request,
  65                                       net::SSLCertRequestInfo* info) OVERRIDE;
  66   virtual void OnSSLCertificateError(net::URLRequest* request,
  67                                      const net::SSLInfo& info,
  68                                      bool fatal) OVERRIDE;
  69   virtual void OnBeforeNetworkStart(net::URLRequest* request,
  70                                     bool* defer) OVERRIDE;
  71   virtual void OnResponseStarted(net::URLRequest* request) OVERRIDE;
  72   virtual void OnReadCompleted(net::URLRequest* request,
  73                                int bytes_read) OVERRIDE;
  74
  75   // SSLErrorHandler::Delegate implementation:
  76   virtual void CancelSSLRequest(const GlobalRequestID& id,
  77                                 int error,
  78                                 const net::SSLInfo* ssl_info) OVERRIDE;
  79   virtual void ContinueSSLRequest(const GlobalRequestID& id) OVERRIDE;
  80
  81   // ResourceController implementation:
  82   virtual void Resume() OVERRIDE;
  83   virtual void Cancel() OVERRIDE;
  84   virtual void CancelAndIgnore() OVERRIDE;
  85   virtual void CancelWithError(int error_code) OVERRIDE;
  86
  87   void StartRequestInternal();
  88   void CancelRequestInternal(int error, bool from_renderer);
  89   // Stores the SignedCertificateTimestamps held in |sct_list| in the
  90   // SignedCertificateTimestampStore singleton, associated with |process_id|.
  91   // On return, |sct_ids| contains the assigned ID and verification status of
  92   // each SignedCertificateTimestamp.
  93   void StoreSignedCertificateTimestamps(
  94       const net::SignedCertificateTimestampAndStatusList& sct_list,
  95       int process_id,
  96       SignedCertificateTimestampIDStatusList* sct_ids);
  97   void CompleteResponseStarted();
  98   void StartReading(bool is_continuation);
  99   void ResumeReading();
 100   void ReadMore(int* bytes_read);
 101   // Passes a read result to the handler.
 102   void CompleteRead(int bytes_read);
 103   void ResponseCompleted();
 104   void CallDidFinishLoading();
 105   void RecordHistograms();
 106
 107   bool is_deferred() const { return deferred_stage_ != DEFERRED_NONE; }
 108
 109   // Used for categorizing loading of prefetches for reporting in histograms.
 110   // NOTE: This enumeration is used in histograms, so please do not add entries
 111   // in the middle.
 112   enum PrefetchStatus {
 113     STATUS_UNDEFINED,
 114     STATUS_SUCCESS_FROM_CACHE,
 115     STATUS_SUCCESS_FROM_NETWORK,
 116     STATUS_CANCELED,
 117     STATUS_MAX,
 118   };
 119
 120   enum DeferredStage {
 121     DEFERRED_NONE,
 122     DEFERRED_START,
 123     DEFERRED_NETWORK_START,
 124     DEFERRED_REDIRECT,
 125     DEFERRED_READ,
 126     DEFERRED_FINISH
 127   };
 128   DeferredStage deferred_stage_;
 129
 130   scoped_ptr<net::URLRequest> request_;
 131   scoped_ptr<ResourceHandler> handler_;
 132   ResourceLoaderDelegate* delegate_;
 133
 134   scoped_refptr<ResourceDispatcherHostLoginDelegate> login_delegate_;
 135   scoped_refptr<SSLClientAuthHandler> ssl_client_auth_handler_;
 136
 137   uint64 last_upload_position_;
 138   bool waiting_for_upload_progress_ack_;
 139   base::TimeTicks last_upload_ticks_;
 140   base::TimeTicks read_deferral_start_time_;
 141
 142   // Indicates that we are in a state of being transferred to a new downstream
 143   // consumer.  We are waiting for a notification to complete the transfer, at
 144   // which point we'll receive a new ResourceHandler.
 145   bool is_transferring_;
 146
 147   base::WeakPtrFactory<ResourceLoader> weak_ptr_factory_;
 148
 149   DISALLOW_COPY_AND_ASSIGN(ResourceLoader);
 150 };
 151
 152 }  // namespace content
 153
 154 #endif  // CONTENT_BROWSER_LOADER_RESOURCE_LOADER_H_