| Index: components/navigation_interception/intercept_navigation_resource_throttle.cc
|
| diff --git a/components/navigation_interception/intercept_navigation_resource_throttle.cc b/components/navigation_interception/intercept_navigation_resource_throttle.cc
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..1b0b89f43d396926c777878c2cf92179ed2e61e6
|
| --- /dev/null
|
| +++ b/components/navigation_interception/intercept_navigation_resource_throttle.cc
|
| @@ -0,0 +1,146 @@
|
| +// Copyright (c) 2012 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 "components/navigation_interception/intercept_navigation_resource_throttle.h"
|
| +
|
| +#include "components/navigation_interception/navigation_params.h"
|
| +#include "content/public/browser/browser_thread.h"
|
| +#include "content/public/browser/child_process_security_policy.h"
|
| +#include "content/public/browser/render_frame_host.h"
|
| +#include "content/public/browser/render_process_host.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/browser/web_contents.h"
|
| +#include "content/public/common/referrer.h"
|
| +#include "net/http/http_response_headers.h"
|
| +#include "net/url_request/redirect_info.h"
|
| +#include "net/url_request/url_request_context.h"
|
| +#include "net/url_request/url_request_job_factory.h"
|
| +#include "net/url_request/url_request.h"
|
| +#include "ui/base/page_transition_types.h"
|
| +
|
| +using content::BrowserThread;
|
| +using content::ChildProcessSecurityPolicy;
|
| +using ui::PageTransition;
|
| +using content::Referrer;
|
| +using content::RenderProcessHost;
|
| +using content::ResourceRequestInfo;
|
| +
|
| +namespace navigation_interception {
|
| +
|
| +namespace {
|
| +
|
| +void CheckIfShouldIgnoreNavigationOnUIThread(
|
| + int render_process_id,
|
| + int render_frame_id,
|
| + const NavigationParams& navigation_params,
|
| + InterceptNavigationResourceThrottle::CheckOnUIThreadCallback
|
| + should_ignore_callback,
|
| + base::Callback<void(bool)> callback) {
|
| + bool should_ignore_navigation = false;
|
| + RenderProcessHost* rph = RenderProcessHost::FromID(render_process_id);
|
| + if (rph) {
|
| + NavigationParams validated_params(navigation_params);
|
| + rph->FilterURL(false, &validated_params.url());
|
| +
|
| + content::RenderFrameHost* render_frame_host =
|
| + content::RenderFrameHost::FromID(render_process_id, render_frame_id);
|
| + content::WebContents* web_contents =
|
| + content::WebContents::FromRenderFrameHost(render_frame_host);
|
| +
|
| + if (web_contents) {
|
| + should_ignore_navigation = should_ignore_callback.Run(web_contents,
|
| + validated_params);
|
| + }
|
| + }
|
| +
|
| + BrowserThread::PostTask(
|
| + BrowserThread::IO,
|
| + FROM_HERE,
|
| + base::Bind(callback, should_ignore_navigation));
|
| +}
|
| +
|
| +} // namespace
|
| +
|
| +InterceptNavigationResourceThrottle::InterceptNavigationResourceThrottle(
|
| + net::URLRequest* request,
|
| + CheckOnUIThreadCallback should_ignore_callback)
|
| + : request_(request),
|
| + should_ignore_callback_(should_ignore_callback),
|
| + weak_ptr_factory_(this) {
|
| +}
|
| +
|
| +InterceptNavigationResourceThrottle::~InterceptNavigationResourceThrottle() {
|
| + DCHECK_CURRENTLY_ON(BrowserThread::IO);
|
| +}
|
| +
|
| +void InterceptNavigationResourceThrottle::WillStartRequest(bool* defer) {
|
| + *defer =
|
| + CheckIfShouldIgnoreNavigation(request_->url(), request_->method(), false);
|
| +}
|
| +
|
| +void InterceptNavigationResourceThrottle::WillRedirectRequest(
|
| + const net::RedirectInfo& redirect_info,
|
| + bool* defer) {
|
| + *defer = CheckIfShouldIgnoreNavigation(redirect_info.new_url,
|
| + redirect_info.new_method, true);
|
| +}
|
| +
|
| +const char* InterceptNavigationResourceThrottle::GetNameForLogging() const {
|
| + return "InterceptNavigationResourceThrottle";
|
| +}
|
| +
|
| +bool InterceptNavigationResourceThrottle::CheckIfShouldIgnoreNavigation(
|
| + const GURL& url,
|
| + const std::string& method,
|
| + bool is_redirect) {
|
| + const ResourceRequestInfo* info = ResourceRequestInfo::ForRequest(request_);
|
| + if (!info)
|
| + return false;
|
| +
|
| + int render_process_id, render_frame_id;
|
| + if (!info->GetAssociatedRenderFrame(&render_process_id, &render_frame_id))
|
| + return false;
|
| +
|
| + bool is_external_protocol =
|
| + !info->GetContext()->GetRequestContext()->job_factory()->IsHandledURL(
|
| + url);
|
| + NavigationParams navigation_params(
|
| + url,
|
| + Referrer::SanitizeForRequest(
|
| + url, Referrer(GURL(request_->referrer()), info->GetReferrerPolicy())),
|
| + info->HasUserGesture(), method == "POST", info->GetPageTransition(),
|
| + is_redirect, is_external_protocol, true);
|
| +
|
| + BrowserThread::PostTask(
|
| + BrowserThread::UI,
|
| + FROM_HERE,
|
| + base::Bind(
|
| + &CheckIfShouldIgnoreNavigationOnUIThread,
|
| + render_process_id,
|
| + render_frame_id,
|
| + navigation_params,
|
| + should_ignore_callback_,
|
| + base::Bind(
|
| + &InterceptNavigationResourceThrottle::OnResultObtained,
|
| + weak_ptr_factory_.GetWeakPtr())));
|
| +
|
| + // Defer request while we wait for the UI thread to check if the navigation
|
| + // should be ignored.
|
| + return true;
|
| +}
|
| +
|
| +void InterceptNavigationResourceThrottle::OnResultObtained(
|
| + bool should_ignore_navigation) {
|
| + DCHECK_CURRENTLY_ON(BrowserThread::IO);
|
| +
|
| + if (should_ignore_navigation) {
|
| + controller()->CancelAndIgnore();
|
| + } else {
|
| + controller()->Resume();
|
| + }
|
| +}
|
| +
|
| +} // namespace navigation_interception
|
|
|