Add a NavigationThrottle to the public content/ interface

content/browser/frame_host/navigation_handle_impl.h

 // 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.
 
 #ifndef CONTENT_BROWSER_FRAME_HOST_NAVIGATION_HANDLE_IMPL_H_
 #define CONTENT_BROWSER_FRAME_HOST_NAVIGATION_HANDLE_IMPL_H_
 
 #include "content/public/browser/navigation_handle.h"
 
 #include "base/macros.h"
 #include "base/memory/scoped_ptr.h"
+#include "base/memory/scoped_vector.h"
 #include "content/common/content_export.h"
+#include "content/public/browser/navigation_throttle.h"
 #include "url/gurl.h"
 
 namespace content {
 
 class NavigatorDelegate;
 struct NavigationRequestInfo;
 
 // This class keeps track of a single navigation. It is created upon receipt of
 // a DidStartProvisionalLoad IPC in a RenderFrameHost. The RenderFrameHost owns
 // the newly created NavigationHandleImpl as long as the navigation is ongoing.
@@ skipping 22 matching lines @@
 // NavigationRequest. It is then owned by the NavigationRequest until the
 // navigation is ready to commit. The NavigationHandleImpl ownership is then
 // transferred to the RenderFrameHost in which the navigation will commit.
 //
 // When PlzNavigate is enabled, the NavigationHandleImpl will never be reset
 // following the receipt of a DidStartProvisionalLoad IPC. There are also no
 // transferring navigations. The other causes of NavigationHandleImpl reset in
 // the RenderFrameHost still apply.
 class CONTENT_EXPORT NavigationHandleImpl : public NavigationHandle {
  public:
-  static scoped_ptr<NavigationHandleImpl> Create(const GURL& url,
-                                                 const bool is_main_frame,
-                                                 NavigatorDelegate* delegate);
-
   ~NavigationHandleImpl() override;
 
   // NavigationHandle implementation:
   const GURL& GetURL() const override;
+  const GURL& GetValidatedURL() const override;
+  WebContents* GetWebContents() const override;
+  bool IsInMainFrame() const override;
+  bool IsPost() const override;
+  const Referrer& GetReferrer() const override;
+  bool HasUserGesture() const override;
+  ui::PageTransition GetPageTransition() const override;
+  bool IsExternalProtocol() const override;
   net::Error GetNetErrorCode() const override;
-  bool IsInMainFrame() const override;
   bool HasCommittedDocument() const override;
   bool HasCommittedErrorPage() const override;
+  void RegisterThrottleForTesting(
+      scoped_ptr<NavigationThrottle> navigation_throttle) override;
 
   void set_net_error_code(net::Error net_error_code) {
     net_error_code_ = net_error_code;
   }
 
   // Returns whether the navigation is currently being transferred from one
   // RenderFrameHost to another. In particular, a DidStartProvisionalLoad IPC
   // for the navigation URL, received in the new RenderFrameHost, should not
   // indicate the start of a new navigation in that case.
   bool is_transferring() const { return is_transferring_; }
   void set_is_transferring(bool is_transferring) {
     is_transferring_ = is_transferring;
   }
 
+  void set_web_contents(WebContents* web_contents) {

[nasko, 2015/09/14 22:00:04]

Uh, this looks scary/weird. Why do we need it?

[clamy, 2015/09/14 22:12:10]

Because I cannot static_cast the NavigatorDelegate to a WebContentsImpl, and
still need a way for the handle to know about the WebContents (can't include the
web_contents_impl header file inside /frame_host).

+    web_contents_ = web_contents;
+  }
+
+  // Called when the URLRequest will start in the network stack.
+  NavigationThrottle::ThrottleCheckResult WillStartRequest(
+      bool is_post,
+      const Referrer& sanitized_referrer,
+      bool has_user_gesture,
+      ui::PageTransition transition,
+      bool is_external_protocol);
+
+  // Called when the URLRequest will be redirected in the network stack.
+  NavigationThrottle::ThrottleCheckResult WillRedirectRequest(
+      const GURL& new_url,
+      const GURL& new_validated_url,
+      bool new_method_is_post,
+      const GURL& new_referrer_url,
+      bool new_is_external_protocol);
+
   // Called when the navigation was redirected. This will update the |url_| and
   // inform the delegate.
   void DidRedirectNavigation(const GURL& new_url);
 
   // Called when the navigation was committed. This will update the |state_|
   // and inform the delegate,
   void DidCommitNavigation();
 
  private:
+  friend class NavigationHandleFactory;
+  friend class TestNavigationHandle;
+
   // Used to track the state the navigation is currently in.
   enum State {
     DID_START = 0,
+    WILL_SEND_REQUEST,
     DID_COMMIT,
     DID_COMMIT_ERROR_PAGE,
   };
 
   NavigationHandleImpl(const GURL& url,
+                       const GURL& validated_url,
                        const bool is_main_frame,
                        NavigatorDelegate* delegate);
 
   // See NavigationHandle for a description of those member variables.
   GURL url_;
+  GURL validated_url_;
+  const bool is_main_frame_;
+  bool is_post_;
+  Referrer sanitized_referrer_;
+  bool has_user_gesture_;
+  ui::PageTransition transition_;
+  bool is_external_protocol_;
   net::Error net_error_code_;
+
+  // The state the navigation is in.
   State state_;
-  const bool is_main_frame_;
 
   // Whether the navigation is in the middle of a transfer. Set to false when
   // the DidStartProvisionalLoad is received from the new renderer.
   bool is_transferring_;
 
   // The delegate that should be notified about events related to this
   // navigation.
   NavigatorDelegate* delegate_;
 
+  // A pointer to the WebContents the navigation is taking place in.

[nasko, 2015/09/14 22:00:04]

nit: Add to the comment to explain why it is ok to use a raw pointer here and
what the lifetime guarantees are.

[Charlie Reis, 2015/09/16 00:09:24]

It's probably also worth explaining (here or elsewhere) why this code in
frame_host depends on WebContents (due to the relationship between
WebContentsObserver and NavigationHandle).

[clamy, 2015/09/16 01:03:21]

Done.

+  WebContents* web_contents_;
+
+  // A list of Throttles registered for this navigation.
+  ScopedVector<NavigationThrottle> throttles_;
+
   DISALLOW_COPY_AND_ASSIGN(NavigationHandleImpl);
 };
 
 }  // namespace content
 
 #endif  // CONTENT_BROWSER_FRAME_HOST_NAVIGATION_HANDLE_IMPL_H_