OLD | NEW |
1 // Copyright 2015 The Chromium Authors. All rights reserved. | 1 // Copyright 2015 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "content/browser/loader/navigation_resource_throttle.h" | 5 #include "content/browser/loader/navigation_resource_throttle.h" |
6 | 6 |
7 #include "base/bind.h" | 7 #include "base/bind.h" |
8 #include "base/callback.h" | 8 #include "base/callback.h" |
9 #include "base/location.h" | 9 #include "base/location.h" |
10 #include "base/logging.h" | 10 #include "base/logging.h" |
| 11 #include "base/memory/ref_counted.h" |
11 #include "content/browser/frame_host/navigation_handle_impl.h" | 12 #include "content/browser/frame_host/navigation_handle_impl.h" |
12 #include "content/browser/frame_host/render_frame_host_impl.h" | 13 #include "content/browser/frame_host/render_frame_host_impl.h" |
| 14 #include "content/browser/loader/resource_request_info_impl.h" |
| 15 #include "content/common/resource_request_body.h" |
13 #include "content/public/browser/browser_thread.h" | 16 #include "content/public/browser/browser_thread.h" |
14 #include "content/public/browser/resource_context.h" | 17 #include "content/public/browser/resource_context.h" |
15 #include "content/public/browser/resource_controller.h" | 18 #include "content/public/browser/resource_controller.h" |
16 #include "content/public/browser/resource_request_info.h" | 19 #include "content/public/browser/resource_request_info.h" |
17 #include "content/public/common/referrer.h" | 20 #include "content/public/common/referrer.h" |
18 #include "net/url_request/redirect_info.h" | 21 #include "net/url_request/redirect_info.h" |
19 #include "net/url_request/url_request.h" | 22 #include "net/url_request/url_request.h" |
20 #include "net/url_request/url_request_context.h" | 23 #include "net/url_request/url_request_context.h" |
21 #include "net/url_request/url_request_job_factory.h" | 24 #include "net/url_request/url_request_job_factory.h" |
22 #include "ui/base/page_transition_types.h" | 25 #include "ui/base/page_transition_types.h" |
23 | 26 |
24 namespace content { | 27 namespace content { |
25 | 28 |
26 namespace { | 29 namespace { |
27 typedef base::Callback<void(NavigationThrottle::ThrottleCheckResult)> | 30 typedef base::Callback<void(NavigationThrottle::ThrottleCheckResult)> |
28 UIChecksPerformedCallback; | 31 UIChecksPerformedCallback; |
29 | 32 |
30 void SendCheckResultToIOThread(UIChecksPerformedCallback callback, | 33 void SendCheckResultToIOThread(UIChecksPerformedCallback callback, |
31 NavigationThrottle::ThrottleCheckResult result) { | 34 NavigationThrottle::ThrottleCheckResult result) { |
32 DCHECK_CURRENTLY_ON(BrowserThread::UI); | 35 DCHECK_CURRENTLY_ON(BrowserThread::UI); |
33 DCHECK_NE(result, NavigationThrottle::DEFER); | 36 DCHECK_NE(result, NavigationThrottle::DEFER); |
34 BrowserThread::PostTask(BrowserThread::IO, FROM_HERE, | 37 BrowserThread::PostTask(BrowserThread::IO, FROM_HERE, |
35 base::Bind(callback, result)); | 38 base::Bind(callback, result)); |
36 } | 39 } |
37 | 40 |
38 void CheckWillStartRequestOnUIThread(UIChecksPerformedCallback callback, | 41 void CheckWillStartRequestOnUIThread( |
39 int render_process_id, | 42 UIChecksPerformedCallback callback, |
40 int render_frame_host_id, | 43 int render_process_id, |
41 const std::string& method, | 44 int render_frame_host_id, |
42 const Referrer& sanitized_referrer, | 45 const std::string& method, |
43 bool has_user_gesture, | 46 const scoped_refptr<content::ResourceRequestBody>& resource_request_body, |
44 ui::PageTransition transition, | 47 const Referrer& sanitized_referrer, |
45 bool is_external_protocol) { | 48 bool has_user_gesture, |
| 49 ui::PageTransition transition, |
| 50 bool is_external_protocol) { |
46 DCHECK_CURRENTLY_ON(BrowserThread::UI); | 51 DCHECK_CURRENTLY_ON(BrowserThread::UI); |
47 RenderFrameHostImpl* render_frame_host = | 52 RenderFrameHostImpl* render_frame_host = |
48 RenderFrameHostImpl::FromID(render_process_id, render_frame_host_id); | 53 RenderFrameHostImpl::FromID(render_process_id, render_frame_host_id); |
49 if (!render_frame_host) { | 54 if (!render_frame_host) { |
50 SendCheckResultToIOThread(callback, NavigationThrottle::PROCEED); | 55 SendCheckResultToIOThread(callback, NavigationThrottle::PROCEED); |
51 return; | 56 return; |
52 } | 57 } |
53 | 58 |
54 NavigationHandleImpl* navigation_handle = | 59 NavigationHandleImpl* navigation_handle = |
55 render_frame_host->navigation_handle(); | 60 render_frame_host->navigation_handle(); |
56 if (!navigation_handle) { | 61 if (!navigation_handle) { |
57 SendCheckResultToIOThread(callback, NavigationThrottle::PROCEED); | 62 SendCheckResultToIOThread(callback, NavigationThrottle::PROCEED); |
58 return; | 63 return; |
59 } | 64 } |
60 | 65 |
61 navigation_handle->WillStartRequest( | 66 navigation_handle->WillStartRequest( |
62 method, sanitized_referrer, has_user_gesture, transition, | 67 method, resource_request_body, sanitized_referrer, has_user_gesture, |
63 is_external_protocol, base::Bind(&SendCheckResultToIOThread, callback)); | 68 transition, is_external_protocol, |
| 69 base::Bind(&SendCheckResultToIOThread, callback)); |
64 } | 70 } |
65 | 71 |
66 void CheckWillRedirectRequestOnUIThread( | 72 void CheckWillRedirectRequestOnUIThread( |
67 UIChecksPerformedCallback callback, | 73 UIChecksPerformedCallback callback, |
68 int render_process_id, | 74 int render_process_id, |
69 int render_frame_host_id, | 75 int render_frame_host_id, |
70 const GURL& new_url, | 76 const GURL& new_url, |
71 const std::string& new_method, | 77 const std::string& new_method, |
72 const GURL& new_referrer_url, | 78 const GURL& new_referrer_url, |
73 bool new_is_external_protocol, | 79 bool new_is_external_protocol, |
(...skipping 48 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
122 | 128 |
123 } // namespace | 129 } // namespace |
124 | 130 |
125 NavigationResourceThrottle::NavigationResourceThrottle(net::URLRequest* request) | 131 NavigationResourceThrottle::NavigationResourceThrottle(net::URLRequest* request) |
126 : request_(request), weak_ptr_factory_(this) {} | 132 : request_(request), weak_ptr_factory_(this) {} |
127 | 133 |
128 NavigationResourceThrottle::~NavigationResourceThrottle() {} | 134 NavigationResourceThrottle::~NavigationResourceThrottle() {} |
129 | 135 |
130 void NavigationResourceThrottle::WillStartRequest(bool* defer) { | 136 void NavigationResourceThrottle::WillStartRequest(bool* defer) { |
131 DCHECK_CURRENTLY_ON(BrowserThread::IO); | 137 DCHECK_CURRENTLY_ON(BrowserThread::IO); |
132 const ResourceRequestInfo* info = ResourceRequestInfo::ForRequest(request_); | 138 const ResourceRequestInfoImpl* info = |
| 139 ResourceRequestInfoImpl::ForRequest(request_); |
133 if (!info) | 140 if (!info) |
134 return; | 141 return; |
135 | 142 |
136 int render_process_id, render_frame_id; | 143 int render_process_id, render_frame_id; |
137 if (!info->GetAssociatedRenderFrame(&render_process_id, &render_frame_id)) | 144 if (!info->GetAssociatedRenderFrame(&render_process_id, &render_frame_id)) |
138 return; | 145 return; |
139 | 146 |
140 bool is_external_protocol = | 147 bool is_external_protocol = |
141 !info->GetContext()->GetRequestContext()->job_factory()->IsHandledURL( | 148 !info->GetContext()->GetRequestContext()->job_factory()->IsHandledURL( |
142 request_->url()); | 149 request_->url()); |
143 UIChecksPerformedCallback callback = | 150 UIChecksPerformedCallback callback = |
144 base::Bind(&NavigationResourceThrottle::OnUIChecksPerformed, | 151 base::Bind(&NavigationResourceThrottle::OnUIChecksPerformed, |
145 weak_ptr_factory_.GetWeakPtr()); | 152 weak_ptr_factory_.GetWeakPtr()); |
146 DCHECK(request_->method() == "POST" || request_->method() == "GET"); | 153 DCHECK(request_->method() == "POST" || request_->method() == "GET"); |
147 BrowserThread::PostTask( | 154 BrowserThread::PostTask( |
148 BrowserThread::UI, FROM_HERE, | 155 BrowserThread::UI, FROM_HERE, |
149 base::Bind(&CheckWillStartRequestOnUIThread, callback, render_process_id, | 156 base::Bind(&CheckWillStartRequestOnUIThread, callback, render_process_id, |
150 render_frame_id, request_->method(), | 157 render_frame_id, request_->method(), info->body(), |
151 Referrer::SanitizeForRequest( | 158 Referrer::SanitizeForRequest( |
152 request_->url(), Referrer(GURL(request_->referrer()), | 159 request_->url(), Referrer(GURL(request_->referrer()), |
153 info->GetReferrerPolicy())), | 160 info->GetReferrerPolicy())), |
154 info->HasUserGesture(), info->GetPageTransition(), | 161 info->HasUserGesture(), info->GetPageTransition(), |
155 is_external_protocol)); | 162 is_external_protocol)); |
156 *defer = true; | 163 *defer = true; |
157 } | 164 } |
158 | 165 |
159 void NavigationResourceThrottle::WillRedirectRequest( | 166 void NavigationResourceThrottle::WillRedirectRequest( |
160 const net::RedirectInfo& redirect_info, | 167 const net::RedirectInfo& redirect_info, |
(...skipping 83 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
244 // block main frame navigations in the future, we'll need to carefully | 251 // block main frame navigations in the future, we'll need to carefully |
245 // consider the right thing to do here. | 252 // consider the right thing to do here. |
246 DCHECK(!ResourceRequestInfo::ForRequest(request_)->IsMainFrame()); | 253 DCHECK(!ResourceRequestInfo::ForRequest(request_)->IsMainFrame()); |
247 controller()->CancelWithError(net::ERR_BLOCKED_BY_RESPONSE); | 254 controller()->CancelWithError(net::ERR_BLOCKED_BY_RESPONSE); |
248 } else { | 255 } else { |
249 controller()->Resume(); | 256 controller()->Resume(); |
250 } | 257 } |
251 } | 258 } |
252 | 259 |
253 } // namespace content | 260 } // namespace content |
OLD | NEW |