Add elevation to the managed mode navigation observer.

chrome/browser/managed_mode/managed_mode_navigation_observer.h

 // 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.
 
 #ifndef CHROME_BROWSER_MANAGED_MODE_MANAGED_MODE_NAVIGATION_OBSERVER_H_
 #define CHROME_BROWSER_MANAGED_MODE_MANAGED_MODE_NAVIGATION_OBSERVER_H_
 
 #include <set>
 
 #include "base/values.h"
@@ skipping 22 matching lines @@
   bool CanTemporarilyNavigateHost(const GURL& url);
 
   // Clears the state recorded in the observer.
   void ClearObserverState();
 
   // Whitelists exact URLs for redirects and host patterns for the final URL.
   // If the URL uses HTTPS, whitelists only the host on HTTPS. Clears the
   // observer state after adding the URLs.
   void AddSavedURLsToWhitelistAndClearState();
 
+  // Returns the elevation state for the corresponding WebContents.
+  bool is_elevated() const;
+
+  // Set the elevation state for the corresponding WebContents.
+  void set_elevated(bool is_elevated);
+
  private:
   // An observer can be in one of the following states:
   // - RECORDING_URLS_BEFORE_PREVIEW: This is the initial state when the user
   // navigates to a new page. In this state the navigated URLs that should be
   // blocked are recorded. The Observer moves to the next state when an
   // interstitial was shown and the user clicked "Preview" on the interstitial.
   // - RECORDING_URLS_AFTER_PREVIEW: URLs that should be blocked are
   // still recorded while the page redirects. The Observer moves to the next
   // state after the page finished redirecting (DidNavigateMainFrame gets
   // called).
@@ skipping 27 matching lines @@
   //  -> ProvisionalChangeToMainFrameUrl http://www.google.com (redirect)
   //  -> DidCommitProvisionalLoadForFrame http://www.google.com (redirect)
   //  -> DidNavigateMainFrame http://www.google.com
   // 3. Page is shown.
   virtual void NavigateToPendingEntry(
       const GURL& url,
       content::NavigationController::ReloadType reload_type) OVERRIDE;
   virtual void DidNavigateMainFrame(
       const content::LoadCommittedDetails& details,
       const content::FrameNavigateParams& params) OVERRIDE;
-  virtual void DidStartProvisionalLoadForFrame(
-      int64 frame_id,
-      int64 parent_frame_id,
-      bool is_main_frame,
-      const GURL& validated_url,
-      bool is_error_page,
-      bool is_iframe_srcdoc,
-      content::RenderViewHost* render_view_host) OVERRIDE;
   virtual void ProvisionalChangeToMainFrameUrl(
       const GURL& url,
       content::RenderViewHost* render_view_host) OVERRIDE;
   virtual void DidCommitProvisionalLoadForFrame(
       int64 frame_id,
       bool is_main_frame,
       const GURL& url,
       content::PageTransition transition_type,
       content::RenderViewHost* render_view_host) OVERRIDE;
   virtual void DidGetUserGesture() OVERRIDE;
 
+  // Returns whether the user would stay in elevated state if he visits this
+  // URL.
+  bool ShouldStayElevatedForURL(const GURL& url);
+
   // Owned by the profile, so outlives us.
   ManagedUserService* managed_user_service_;
 
   // Owned by ManagedUserService.
   const ManagedModeURLFilter* url_filter_;
 
   // Owned by the InfoBarService, which has the same lifetime as this object.
   InfoBarDelegate* warn_infobar_delegate_;
   InfoBarDelegate* preview_infobar_delegate_;
 
   // Whether we received a user gesture.
   // The goal is to allow automatic redirects (in order not to break the flow
   // or show too many interstitials) while not allowing the user to navigate
   // to blocked pages. We consider a redirect to be automatic if we did
   // not get a user gesture.
   bool got_user_gesture_;
   ObserverState state_;
   std::set<GURL> navigated_urls_;
   GURL last_url_;
 
+  // The elevation state corresponding to the current WebContents.
+  // Will be set to true for non-managed users.
+  bool is_elevated_;
+
   int last_allowed_page_;
 
   DISALLOW_COPY_AND_ASSIGN(ManagedModeNavigationObserver);
 };
 
 #endif  // CHROME_BROWSER_MANAGED_MODE_MANAGED_MODE_NAVIGATION_OBSERVER_H_