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 net {
  19 class X509Certificate;
  20 }
  21
  22 namespace content {
  23 class ResourceDispatcherHostLoginDelegate;
  24 class ResourceLoaderDelegate;
  25 class ResourceRequestInfoImpl;
  26 class SSLClientAuthHandler;
  27
  28 // This class is responsible for driving the URLRequest (i.e., calling Start,
  29 // Read, and servicing events).  It has a ResourceHandler, which is typically a
  30 // chain of ResourceHandlers, and is the ResourceController for its handler.
  31 class CONTENT_EXPORT ResourceLoader : public net::URLRequest::Delegate,
  32                                       public SSLErrorHandler::Delegate,
  33                                       public ResourceController {
  34  public:
  35   ResourceLoader(scoped_ptr<net::URLRequest> request,
  36                  scoped_ptr<ResourceHandler> handler,
  37                  ResourceLoaderDelegate* delegate);
  38   ~ResourceLoader() override;
  39
  40   void StartRequest();
  41   void CancelRequest(bool from_renderer);
  42
  43   void ReportUploadProgress();
  44
  45   bool is_transferring() const { return is_transferring_; }
  46   void MarkAsTransferring();
  47   void CompleteTransfer();
  48
  49   net::URLRequest* request() { return request_.get(); }
  50   ResourceRequestInfoImpl* GetRequestInfo();
  51
  52   void ClearLoginDelegate();
  53
  54   // IPC message handlers:
  55   void OnUploadProgressACK();
  56
  57  private:
  58   FRIEND_TEST_ALL_PREFIXES(ResourceLoaderTest, ClientCertStoreLookup);
  59   FRIEND_TEST_ALL_PREFIXES(ResourceLoaderTest, ClientCertStoreNull);
  60   FRIEND_TEST_ALL_PREFIXES(ResourceLoaderTest, ClientCertStoreAsyncCancel);
  61
  62   // net::URLRequest::Delegate implementation:
  63   void OnReceivedRedirect(net::URLRequest* request,
  64                           const net::RedirectInfo& redirect_info,
  65                           bool* defer) override;
  66   void OnAuthRequired(net::URLRequest* request,
  67                       net::AuthChallengeInfo* info) override;
  68   void OnCertificateRequested(net::URLRequest* request,
  69                               net::SSLCertRequestInfo* info) override;
  70   void OnSSLCertificateError(net::URLRequest* request,
  71                              const net::SSLInfo& info,
  72                              bool fatal) override;
  73   void OnBeforeNetworkStart(net::URLRequest* request, bool* defer) override;
  74   void OnResponseStarted(net::URLRequest* request) override;
  75   void OnReadCompleted(net::URLRequest* request, int bytes_read) override;
  76
  77   // SSLErrorHandler::Delegate implementation:
  78   void CancelSSLRequest(int error, const net::SSLInfo* ssl_info) override;
  79   void ContinueSSLRequest() override;
  80
  81   // ResourceController implementation:
  82   void Resume() override;
  83   void Cancel() override;
  84   void CancelAndIgnore() override;
  85   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   void ContinueWithCertificate(net::X509Certificate* cert);
 107
 108   bool is_deferred() const { return deferred_stage_ != DEFERRED_NONE; }
 109
 110   // Used for categorizing loading of prefetches for reporting in histograms.
 111   // NOTE: This enumeration is used in histograms, so please do not add entries
 112   // in the middle.
 113   enum PrefetchStatus {
 114     STATUS_UNDEFINED,
 115     STATUS_SUCCESS_FROM_CACHE,
 116     STATUS_SUCCESS_FROM_NETWORK,
 117     STATUS_CANCELED,
 118     STATUS_MAX,
 119   };
 120
 121   enum DeferredStage {
 122     DEFERRED_NONE,
 123     DEFERRED_START,
 124     DEFERRED_NETWORK_START,
 125     DEFERRED_REDIRECT,
 126     DEFERRED_READ,
 127     DEFERRED_RESPONSE_COMPLETE,
 128     DEFERRED_FINISH
 129   };
 130   DeferredStage deferred_stage_;
 131
 132   scoped_ptr<net::URLRequest> request_;
 133   scoped_ptr<ResourceHandler> handler_;
 134   ResourceLoaderDelegate* delegate_;
 135
 136   scoped_refptr<ResourceDispatcherHostLoginDelegate> login_delegate_;
 137   scoped_ptr<SSLClientAuthHandler> ssl_client_auth_handler_;
 138
 139   uint64 last_upload_position_;
 140   bool waiting_for_upload_progress_ack_;
 141   base::TimeTicks last_upload_ticks_;
 142   base::TimeTicks read_deferral_start_time_;
 143
 144   // Indicates that we are in a state of being transferred to a new downstream
 145   // consumer.  We are waiting for a notification to complete the transfer, at
 146   // which point we'll receive a new ResourceHandler.
 147   bool is_transferring_;
 148
 149   base::WeakPtrFactory<ResourceLoader> weak_ptr_factory_;
 150
 151   DISALLOW_COPY_AND_ASSIGN(ResourceLoader);
 152 };
 153
 154 }  // namespace content
 155
 156 #endif  // CONTENT_BROWSER_LOADER_RESOURCE_LOADER_H_