content/browser/loader/cross_site_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_CROSS_SITE_RESOURCE_HANDLER_H_
   6 #define CONTENT_BROWSER_LOADER_CROSS_SITE_RESOURCE_HANDLER_H_
   7
   8 #include "base/memory/ref_counted.h"
   9 #include "base/memory/weak_ptr.h"
  10 #include "content/browser/loader/layered_resource_handler.h"
  11 #include "content/common/content_export.h"
  12 #include "net/url_request/url_request_status.h"
  13
  14 namespace net {
  15 class URLRequest;
  16 }
  17
  18 namespace content {
  19
  20 // Ensures that cross-site responses are delayed until the onunload handler of
  21 // the previous page is allowed to run.  This handler wraps an
  22 // AsyncEventHandler, and it sits inside SafeBrowsing and Buffered event
  23 // handlers.  This is important, so that it can intercept OnResponseStarted
  24 // after we determine that a response is safe and not a download.
  25 class CrossSiteResourceHandler : public LayeredResourceHandler {
  26  public:
  27   CrossSiteResourceHandler(scoped_ptr<ResourceHandler> next_handler,
  28                            net::URLRequest* request);
  29   virtual ~CrossSiteResourceHandler();
  30
  31   // ResourceHandler implementation:
  32   virtual bool OnRequestRedirected(const GURL& new_url,
  33                                    ResourceResponse* response,
  34                                    bool* defer) OVERRIDE;
  35   virtual bool OnResponseStarted(ResourceResponse* response,
  36                                  bool* defer) OVERRIDE;
  37   virtual bool OnReadCompleted(int bytes_read,
  38                                bool* defer) OVERRIDE;
  39   virtual void OnResponseCompleted(const net::URLRequestStatus& status,
  40                                    const std::string& security_info,
  41                                    bool* defer) OVERRIDE;
  42
  43   // We can now send the response to the new renderer, which will cause
  44   // WebContentsImpl to swap in the new renderer and destroy the old one.
  45   void ResumeResponse();
  46
  47   // When set to true, requests are leaked when they can't be passed to a
  48   // RenderViewHost, for unit tests.
  49   CONTENT_EXPORT static void SetLeakRequestsForTesting(
  50       bool leak_requests_for_testing);
  51
  52  private:
  53   // Prepare to render the cross-site response in a new RenderViewHost, by
  54   // telling the old RenderViewHost to run its onunload handler.
  55   void StartCrossSiteTransition(ResourceResponse* response,
  56                                 bool should_transfer);
  57
  58   // Defer the navigation to the UI thread to check whether transfer is required
  59   // or not. Currently only used in --site-per-process.
  60   bool DeferForNavigationPolicyCheck(ResourceRequestInfoImpl* info,
  61                                      ResourceResponse* response,
  62                                      bool* defer);
  63
  64   void ResumeOrTransfer(bool is_transfer);
  65   void ResumeIfDeferred();
  66
  67   // Called when about to defer a request.  Sets |did_defer_| and logs the
  68   // defferral
  69   void OnDidDefer();
  70
  71   bool has_started_response_;
  72   bool in_cross_site_transition_;
  73   bool completed_during_transition_;
  74   bool did_defer_;
  75   net::URLRequestStatus completed_status_;
  76   std::string completed_security_info_;
  77   scoped_refptr<ResourceResponse> response_;
  78
  79   // TODO(nasko): WeakPtr is needed in --site-per-process, since all navigations
  80   // are deferred to the UI thread and come back to IO thread via
  81   // PostTaskAndReplyWithResult. If a transfer is needed, it goes back to the UI
  82   // thread. This can be removed once the code is changed to only do one hop.
  83   base::WeakPtrFactory<CrossSiteResourceHandler> weak_ptr_factory_;
  84
  85   DISALLOW_COPY_AND_ASSIGN(CrossSiteResourceHandler);
  86 };
  87
  88 }  // namespace content
  89
  90 #endif  // CONTENT_BROWSER_LOADER_CROSS_SITE_RESOURCE_HANDLER_H_