src/content/browser/loader/throttling_resource_handler.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_THROTTLING_RESOURCE_HANDLER_H_
   6 #define CONTENT_BROWSER_LOADER_THROTTLING_RESOURCE_HANDLER_H_
   7
   8 #include "base/memory/ref_counted.h"
   9 #include "base/memory/scoped_vector.h"
  10 #include "content/browser/loader/layered_resource_handler.h"
  11 #include "content/public/browser/resource_controller.h"
  12 #include "net/url_request/redirect_info.h"
  13 #include "url/gurl.h"
  14
  15 namespace net {
  16 class URLRequest;
  17 }
  18
  19 namespace content {
  20
  21 class ResourceThrottle;
  22 struct ResourceResponse;
  23
  24 // Used to apply a list of ResourceThrottle instances to an URLRequest.
  25 class ThrottlingResourceHandler : public LayeredResourceHandler,
  26                                   public ResourceController {
  27  public:
  28   // Takes ownership of the ResourceThrottle instances.
  29   ThrottlingResourceHandler(scoped_ptr<ResourceHandler> next_handler,
  30                             net::URLRequest* request,
  31                             ScopedVector<ResourceThrottle> throttles);
  32   ~ThrottlingResourceHandler() override;
  33
  34   // LayeredResourceHandler overrides:
  35   bool OnRequestRedirected(const net::RedirectInfo& redirect_info,
  36                            ResourceResponse* response,
  37                            bool* defer) override;
  38   bool OnResponseStarted(ResourceResponse* response, bool* defer) override;
  39   bool OnWillStart(const GURL& url, bool* defer) override;
  40   bool OnBeforeNetworkStart(const GURL& url, bool* defer) override;
  41
  42   // ResourceController implementation:
  43   void Cancel() override;
  44   void CancelAndIgnore() override;
  45   void CancelWithError(int error_code) override;
  46   void Resume() override;
  47
  48  private:
  49   void ResumeStart();
  50   void ResumeNetworkStart();
  51   void ResumeRedirect();
  52   void ResumeResponse();
  53
  54   // Called when the throttle at |throttle_index| defers a request.  Logs the
  55   // name of the throttle that delayed the request.
  56   void OnRequestDefered(int throttle_index);
  57
  58   enum DeferredStage {
  59     DEFERRED_NONE,
  60     DEFERRED_START,
  61     DEFERRED_NETWORK_START,
  62     DEFERRED_REDIRECT,
  63     DEFERRED_RESPONSE
  64   };
  65   DeferredStage deferred_stage_;
  66
  67   ScopedVector<ResourceThrottle> throttles_;
  68   size_t next_index_;
  69
  70   GURL deferred_url_;
  71   net::RedirectInfo deferred_redirect_;
  72   scoped_refptr<ResourceResponse> deferred_response_;
  73
  74   bool cancelled_by_resource_throttle_;
  75 };
  76
  77 }  // namespace content
  78
  79 #endif  // CONTENT_BROWSER_LOADER_THROTTLING_RESOURCE_HANDLER_H_