Reland of Add a NavigationThrottle to the public content/ interface

content/browser/loader/navigation_resource_throttle.cc

+// Copyright 2015 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include "content/browser/loader/navigation_resource_throttle.h"
+
+#include "base/callback.h"
+#include "content/browser/frame_host/navigation_handle_impl.h"
+#include "content/browser/frame_host/render_frame_host_impl.h"
+#include "content/public/browser/resource_context.h"
+#include "content/public/browser/resource_controller.h"
+#include "content/public/browser/resource_request_info.h"
+#include "content/public/common/referrer.h"
+#include "net/url_request/redirect_info.h"
+#include "net/url_request/url_request.h"
+#include "net/url_request/url_request_context.h"
+#include "net/url_request/url_request_job_factory.h"
+#include "ui/base/page_transition_types.h"
+
+namespace content {
+
+namespace {
+typedef base::Callback<void(NavigationThrottle::ThrottleCheckResult)>
+    UIChecksPerformedCallback;
+
+void CheckWillStartRequestOnUIThread(UIChecksPerformedCallback callback,
+                                     int render_process_id,
+                                     int render_frame_host_id,
+                                     bool is_post,
+                                     const Referrer& sanitized_referrer,
+                                     bool has_user_gesture,
+                                     ui::PageTransition transition,
+                                     bool is_external_protocol) {
+  DCHECK_CURRENTLY_ON(BrowserThread::UI);
+  NavigationThrottle::ThrottleCheckResult result = NavigationThrottle::PROCEED;
+  RenderFrameHostImpl* render_frame_host =
+      RenderFrameHostImpl::FromID(render_process_id, render_frame_host_id);
+  if (render_frame_host) {
+    NavigationHandleImpl* navigation_handle =
+        render_frame_host->navigation_handle();
+    if (navigation_handle) {
+      result = navigation_handle->WillStartRequest(is_post, sanitized_referrer,
+                                                   has_user_gesture, transition,
+                                                   is_external_protocol);
+    }
+  }
+  BrowserThread::PostTask(BrowserThread::IO, FROM_HERE,
+                          base::Bind(callback, result));
+}
+
+void CheckWillRedirectRequestOnUIThread(UIChecksPerformedCallback callback,
+                                        int render_process_id,
+                                        int render_frame_host_id,
+                                        const GURL& new_url,
+                                        bool new_method_is_post,
+                                        const GURL& new_referrer_url,
+                                        bool new_is_external_protocol) {
+  DCHECK_CURRENTLY_ON(BrowserThread::UI);
+  NavigationThrottle::ThrottleCheckResult result = NavigationThrottle::PROCEED;
+  RenderFrameHostImpl* render_frame_host =
+      RenderFrameHostImpl::FromID(render_process_id, render_frame_host_id);
+  if (render_frame_host) {
+    NavigationHandleImpl* navigation_handle =
+        render_frame_host->navigation_handle();
+    if (navigation_handle) {
+      RenderProcessHost* rph = RenderProcessHost::FromID(render_process_id);
+      GURL new_validated_url = new_url;
+      rph->FilterURL(false, &new_validated_url);
+      result = navigation_handle->WillRedirectRequest(
+          new_validated_url, new_method_is_post, new_referrer_url,
+          new_is_external_protocol);
+    }
+  }
+  BrowserThread::PostTask(BrowserThread::IO, FROM_HERE,
+                          base::Bind(callback, result));
+}
+}  // namespace
+
+NavigationResourceThrottle::NavigationResourceThrottle(net::URLRequest* request)
+    : request_(request), weak_ptr_factory_(this) {}
+
+NavigationResourceThrottle::~NavigationResourceThrottle() {}
+
+void NavigationResourceThrottle::WillStartRequest(bool* defer) {
+  const ResourceRequestInfo* info = ResourceRequestInfo::ForRequest(request_);
+  if (!info)
+    return;
+
+  int render_process_id, render_frame_id;
+  if (!info->GetAssociatedRenderFrame(&render_process_id, &render_frame_id))
+    return;
+
+  bool is_external_protocol =
+      !info->GetContext()->GetRequestContext()->job_factory()->IsHandledURL(
+          request_->url());
+  UIChecksPerformedCallback callback =
+      base::Bind(&NavigationResourceThrottle::OnUIChecksPerformed,
+                 weak_ptr_factory_.GetWeakPtr());
+  DCHECK(request_->method() == "POST" || request_->method() == "GET");
+  BrowserThread::PostTask(
+      BrowserThread::UI, FROM_HERE,
+      base::Bind(&CheckWillStartRequestOnUIThread, callback, render_process_id,
+                 render_frame_id, request_->method() == "POST",
+                 Referrer::SanitizeForRequest(
+                     request_->url(), Referrer(GURL(request_->referrer()),
+                                               info->GetReferrerPolicy())),
+                 info->HasUserGesture(), info->GetPageTransition(),
+                 is_external_protocol));
+  *defer = true;
+}
+
+void NavigationResourceThrottle::WillRedirectRequest(
+    const net::RedirectInfo& redirect_info,
+    bool* defer) {
+  const ResourceRequestInfo* info = ResourceRequestInfo::ForRequest(request_);
+  if (!info)
+    return;
+
+  int render_process_id, render_frame_id;
+  if (!info->GetAssociatedRenderFrame(&render_process_id, &render_frame_id))
+    return;
+
+  bool new_is_external_protocol =
+      !info->GetContext()->GetRequestContext()->job_factory()->IsHandledURL(
+          request_->url());
+  DCHECK(redirect_info.new_method == "POST" ||
+         redirect_info.new_method == "GET");
+  UIChecksPerformedCallback callback =
+      base::Bind(&NavigationResourceThrottle::OnUIChecksPerformed,
+                 weak_ptr_factory_.GetWeakPtr());
+  BrowserThread::PostTask(
+      BrowserThread::UI, FROM_HERE,
+      base::Bind(&CheckWillRedirectRequestOnUIThread, callback,
+                 render_process_id, render_frame_id, redirect_info.new_url,
+                 redirect_info.new_method == "POST",
+                 GURL(redirect_info.new_referrer), new_is_external_protocol));
+  *defer = true;
+}
+
+const char* NavigationResourceThrottle::GetNameForLogging() const {
+  return "NavigationResourceThrottle";
+}
+
+void NavigationResourceThrottle::OnUIChecksPerformed(
+    NavigationThrottle::ThrottleCheckResult result) {
+  DCHECK_CURRENTLY_ON(BrowserThread::IO);
+  if (result == NavigationThrottle::CANCEL_AND_IGNORE) {
+    controller()->CancelAndIgnore();
+  } else {
+    controller()->Resume();
+  }
+}
+
+}  // namespace content