Index: content/browser/frame_host/form_submission_throttle.cc |
diff --git a/content/browser/frame_host/form_submission_throttle.cc b/content/browser/frame_host/form_submission_throttle.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..3aac4c4096ff125b5161cef99ae113d91fc6ae02 |
--- /dev/null |
+++ b/content/browser/frame_host/form_submission_throttle.cc |
@@ -0,0 +1,69 @@ |
+// Copyright 2016 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/frame_host/form_submission_throttle.h" |
+#include "content/browser/frame_host/navigation_handle_impl.h" |
+#include "content/public/browser/browser_thread.h" |
+#include "content/public/browser/navigation_handle.h" |
+#include "content/public/browser/navigation_throttle.h" |
+#include "content/public/common/browser_side_navigation_policy.h" |
+ |
+namespace content { |
+ |
+// static |
+std::unique_ptr<NavigationThrottle> |
+FormSubmissionThrottle::MaybeCreateThrottleFor(NavigationHandle* handle) { |
+ DCHECK_CURRENTLY_ON(content::BrowserThread::UI); |
+ |
+ if (!IsBrowserSideNavigationEnabled()) |
+ return nullptr; |
+ |
+ NavigationHandleImpl* handle_impl = |
+ static_cast<NavigationHandleImpl*>(handle); |
+ |
+ if (!handle_impl->is_form_submission()) |
+ return nullptr; |
+ |
+ return std::unique_ptr<NavigationThrottle>( |
+ new FormSubmissionThrottle(handle)); |
+} |
+ |
+FormSubmissionThrottle::~FormSubmissionThrottle() {} |
+ |
+NavigationThrottle::ThrottleCheckResult |
+FormSubmissionThrottle::CheckContentSecurityPolicyFormAction(bool is_redirect) { |
+ NavigationHandleImpl* handle = |
+ static_cast<NavigationHandleImpl*>(navigation_handle()); |
+ |
+ // Allow the request when it bypasses the CSP. |
+ if (handle->should_bypass_main_world_csp()) |
+ return NavigationThrottle::PROCEED; |
+ |
+ const GURL& url = handle->GetURL(); |
+ RenderFrameHostImpl* render_frame = |
+ handle->frame_tree_node()->current_frame_host(); |
+ |
+ if (!render_frame->csp_context()->Allow( |
+ render_frame->content_security_policies(), CSPDirective::FormAction, |
+ url, is_redirect)) { |
+ return NavigationThrottle::CANCEL; |
+ } |
+ |
+ return NavigationThrottle::PROCEED; |
Mike West
2017/02/22 15:36:07
It seems valuable to add some unit tests for this
arthursonzogni
2017/02/22 17:15:23
Acknowledged. It looks hard since it depends on th
arthursonzogni
2017/02/23 13:28:53
Tests added!
|
+} |
+ |
+NavigationThrottle::ThrottleCheckResult |
+FormSubmissionThrottle::WillStartRequest() { |
+ return CheckContentSecurityPolicyFormAction(false /* is_redirect */); |
Mike West
2017/02/22 15:36:07
I might be misunderstanding things, but I thought
arthursonzogni
2017/02/22 17:15:23
I confirm that WillStartRequest is not called twic
|
+} |
+ |
+NavigationThrottle::ThrottleCheckResult |
+FormSubmissionThrottle::WillRedirectRequest() { |
+ return CheckContentSecurityPolicyFormAction(true /* is_redirect */); |
+} |
+ |
+FormSubmissionThrottle::FormSubmissionThrottle(NavigationHandle* handle) |
+ : NavigationThrottle(handle) {} |
+ |
+} // namespace content |