src/chrome/browser/renderer_host/chrome_resource_dispatcher_host_delegate.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 CHROME_BROWSER_RENDERER_HOST_CHROME_RESOURCE_DISPATCHER_HOST_DELEGATE_H_
   6 #define CHROME_BROWSER_RENDERER_HOST_CHROME_RESOURCE_DISPATCHER_HOST_DELEGATE_H_
   7
   8 #include <map>
   9 #include <set>
  10
  11 #include "base/compiler_specific.h"
  12 #include "base/memory/ref_counted.h"
  13 #include "chrome/browser/external_protocol/external_protocol_handler.h"
  14 #include "content/public/browser/resource_dispatcher_host_delegate.h"
  15
  16 class DelayedResourceQueue;
  17 class DownloadRequestLimiter;
  18 class SafeBrowsingService;
  19
  20 namespace extensions {
  21 class UserScriptListener;
  22 }
  23
  24 namespace prerender {
  25 class PrerenderTracker;
  26 }
  27
  28 // Implements ResourceDispatcherHostDelegate. Currently used by the Prerender
  29 // system to abort requests and add to the load flags when a request begins.
  30 class ChromeResourceDispatcherHostDelegate
  31     : public content::ResourceDispatcherHostDelegate {
  32  public:
  33   // This class does not take ownership of the tracker but merely holds a
  34   // reference to it to avoid accessing g_browser_process.
  35   // |prerender_tracker| must outlive |this|.
  36   explicit ChromeResourceDispatcherHostDelegate(
  37       prerender::PrerenderTracker* prerender_tracker);
  38   virtual ~ChromeResourceDispatcherHostDelegate();
  39
  40   // ResourceDispatcherHostDelegate implementation.
  41   virtual bool ShouldBeginRequest(
  42       const std::string& method,
  43       const GURL& url,
  44       content::ResourceType resource_type,
  45       content::ResourceContext* resource_context) OVERRIDE;
  46   virtual void RequestBeginning(
  47       net::URLRequest* request,
  48       content::ResourceContext* resource_context,
  49       content::AppCacheService* appcache_service,
  50       content::ResourceType resource_type,
  51       ScopedVector<content::ResourceThrottle>* throttles) OVERRIDE;
  52   virtual void DownloadStarting(
  53       net::URLRequest* request,
  54       content::ResourceContext* resource_context,
  55       int child_id,
  56       int route_id,
  57       int request_id,
  58       bool is_content_initiated,
  59       bool must_download,
  60       ScopedVector<content::ResourceThrottle>* throttles) OVERRIDE;
  61   virtual content::ResourceDispatcherHostLoginDelegate* CreateLoginDelegate(
  62       net::AuthChallengeInfo* auth_info, net::URLRequest* request) OVERRIDE;
  63   virtual bool HandleExternalProtocol(const GURL& url,
  64                                       int child_id,
  65                                       int route_id) OVERRIDE;
  66   virtual bool ShouldForceDownloadResource(
  67       const GURL& url, const std::string& mime_type) OVERRIDE;
  68   virtual bool ShouldInterceptResourceAsStream(
  69       net::URLRequest* request,
  70       const std::string& mime_type,
  71       GURL* origin,
  72       std::string* payload) OVERRIDE;
  73   virtual void OnStreamCreated(
  74       net::URLRequest* request,
  75       scoped_ptr<content::StreamHandle> stream) OVERRIDE;
  76   virtual void OnResponseStarted(
  77       net::URLRequest* request,
  78       content::ResourceContext* resource_context,
  79       content::ResourceResponse* response,
  80       IPC::Sender* sender) OVERRIDE;
  81   virtual void OnRequestRedirected(
  82       const GURL& redirect_url,
  83       net::URLRequest* request,
  84       content::ResourceContext* resource_context,
  85       content::ResourceResponse* response) OVERRIDE;
  86   virtual void RequestComplete(net::URLRequest* url_request) OVERRIDE;
  87
  88   // Called on the UI thread. Allows switching out the
  89   // ExternalProtocolHandler::Delegate for testing code.
  90   static void SetExternalProtocolHandlerDelegateForTesting(
  91       ExternalProtocolHandler::Delegate* delegate);
  92
  93  private:
  94 #if defined(ENABLE_EXTENSIONS)
  95   struct StreamTargetInfo {
  96     std::string extension_id;
  97     std::string view_id;
  98   };
  99 #endif
 100
 101   void AppendStandardResourceThrottles(
 102       net::URLRequest* request,
 103       content::ResourceContext* resource_context,
 104       content::ResourceType resource_type,
 105       ScopedVector<content::ResourceThrottle>* throttles);
 106
 107 #if defined(ENABLE_ONE_CLICK_SIGNIN)
 108   // Append headers required to tell Gaia whether the sync interstitial
 109   // should be shown or not.  This header is only added for valid Gaia URLs.
 110   void AppendChromeSyncGaiaHeader(
 111       net::URLRequest* request,
 112       content::ResourceContext* resource_context);
 113 #endif
 114
 115   scoped_refptr<DownloadRequestLimiter> download_request_limiter_;
 116   scoped_refptr<SafeBrowsingService> safe_browsing_;
 117 #if defined(ENABLE_EXTENSIONS)
 118   scoped_refptr<extensions::UserScriptListener> user_script_listener_;
 119   std::map<net::URLRequest*, StreamTargetInfo> stream_target_info_;
 120 #endif
 121   prerender::PrerenderTracker* prerender_tracker_;
 122
 123   DISALLOW_COPY_AND_ASSIGN(ChromeResourceDispatcherHostDelegate);
 124 };
 125
 126 #endif  // CHROME_BROWSER_RENDERER_HOST_CHROME_RESOURCE_DISPATCHER_HOST_DELEGATE_H_