| // 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 |