Allow a main interstitial frame to act as an ExtensionMessagePort opener

content/browser/frame_host/interstitial_page_impl.cc

 // Copyright 2013 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/interstitial_page_impl.h"
 
 #include <utility>
 #include <vector>
 
 #include "base/bind.h"
@@ skipping 130 matching lines @@
     WebContents* web_contents) {
   InitInterstitialPageMap();
   InterstitialPageMap::const_iterator iter =
       g_web_contents_to_interstitial_page->find(web_contents);
   if (iter == g_web_contents_to_interstitial_page->end())
     return NULL;
 
   return iter->second;
 }
 
+InterstitialPage* InterstitialPage::FromRenderFrameHost(RenderFrameHost* rfh) {
+  if (!rfh)
+    return nullptr;
+  return static_cast<RenderFrameHostImpl*>(rfh)
+      ->delegate()
+      ->GetAsInterstitialPage();
+}
+
 InterstitialPageImpl::InterstitialPageImpl(
     WebContents* web_contents,
     RenderWidgetHostDelegate* render_widget_host_delegate,
     bool new_navigation,
     const GURL& url,
     InterstitialPageDelegate* delegate)
     : underlying_content_observer_(web_contents, this),
       web_contents_(web_contents),
       controller_(static_cast<NavigationControllerImpl*>(
           &web_contents->GetController())),
@@ skipping 256 matching lines @@
   if (!new_navigation_ && !should_revert_web_contents_title_) {
     original_web_contents_title_ = entry->GetTitle();
     should_revert_web_contents_title_ = true;
   }
   // TODO(evan): make use of title_direction.
   // http://code.google.com/p/chromium/issues/detail?id=27094
   entry->SetTitle(title);
   controller_->delegate()->NotifyNavigationStateChanged(INVALIDATE_TYPE_TITLE);
 }
 
+InterstitialPage* InterstitialPageImpl::GetAsInterstitialPage() {
+  return this;
+}
+
 AccessibilityMode InterstitialPageImpl::GetAccessibilityMode() const {
   if (web_contents_)
     return static_cast<WebContentsImpl*>(web_contents_)->GetAccessibilityMode();
   else
     return AccessibilityModeOff;
 }
 
 void InterstitialPageImpl::Cut() {
   FrameTreeNode* focused_node = frame_tree_.GetFocusedFrame();
   if (!focused_node)
@@ skipping 31 matching lines @@
 
   focused_node->current_frame_host()->Send(new InputMsg_SelectAll(
       focused_node->current_frame_host()->GetRoutingID()));
   RecordAction(base::UserMetricsAction("SelectAll"));
 }
 
 RenderViewHostDelegateView* InterstitialPageImpl::GetDelegateView() {
   return rvh_delegate_view_.get();
 }
 
+WebContents* InterstitialPageImpl::GetWebContents() const {
+  return web_contents();
+}
+
 const GURL& InterstitialPageImpl::GetMainFrameLastCommittedURL() const {
   return url_;
 }
 
 void InterstitialPageImpl::RenderViewTerminated(
     RenderViewHost* render_view_host,
     base::TerminationStatus status,
     int error_code) {
   // Our renderer died. This should not happen in normal cases.
   // If we haven't already started shutdown, just dismiss the interstitial.
@@ skipping 453 matching lines @@
 void InterstitialPageImpl::UnderlyingContentObserver::NavigationEntryCommitted(
     const LoadCommittedDetails& load_details) {
   interstitial_->OnNavigatingAwayOrTabClosing();
 }
 
 void InterstitialPageImpl::UnderlyingContentObserver::WebContentsDestroyed() {
   interstitial_->OnNavigatingAwayOrTabClosing();
 }
 
 }  // namespace content