<webview>: Store <webview> embedder info in ExtensionRendererState

chrome/browser/extensions/extension_renderer_state.cc

 // Copyright (c) 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 "chrome/browser/extensions/extension_renderer_state.h"
 
 #include "base/bind.h"
 #include "base/bind_helpers.h"
 #include "chrome/browser/sessions/session_tab_helper.h"
 #include "chrome/browser/tab_contents/retargeting_details.h"
@@ skipping 33 matching lines @@
                        const content::NotificationDetails& details);
 
   content::NotificationRegistrar registrar_;
 };
 
 ExtensionRendererState::TabObserver::TabObserver() {
   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
   registrar_.Add(this,
                  content::NOTIFICATION_WEB_CONTENTS_RENDER_VIEW_HOST_CREATED,
                  content::NotificationService::AllBrowserContextsAndSources());
+  registrar_.Add(this,
+                 content::NOTIFICATION_WEB_CONTENTS_CONNECTED,
+                 content::NotificationService::AllBrowserContextsAndSources());
   registrar_.Add(this, content::NOTIFICATION_RENDER_VIEW_HOST_DELETED,
                  content::NotificationService::AllBrowserContextsAndSources());
   registrar_.Add(this, chrome::NOTIFICATION_TAB_PARENTED,
                  content::NotificationService::AllBrowserContextsAndSources());
   registrar_.Add(this, chrome::NOTIFICATION_RETARGETING,
                  content::NotificationService::AllBrowserContextsAndSources());
-  registrar_.Add(this,
-                 content::NOTIFICATION_RENDERER_PROCESS_CREATED,
-                 content::NotificationService::AllBrowserContextsAndSources());
-  registrar_.Add(this,
-                 content::NOTIFICATION_RENDERER_PROCESS_TERMINATED,
-                 content::NotificationService::AllBrowserContextsAndSources());
-  registrar_.Add(this,
-                 content::NOTIFICATION_RENDERER_PROCESS_CLOSED,
-                 content::NotificationService::AllBrowserContextsAndSources());
 }
 
 ExtensionRendererState::TabObserver::~TabObserver() {
   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
 }
 
 void ExtensionRendererState::TabObserver::Observe(
     int type, const content::NotificationSource& source,
     const content::NotificationDetails& details) {
   switch (type) {
     case content::NOTIFICATION_WEB_CONTENTS_RENDER_VIEW_HOST_CREATED: {
       WebContents* web_contents = content::Source<WebContents>(source).ptr();
       SessionTabHelper* session_tab_helper =
           SessionTabHelper::FromWebContents(web_contents);
       if (!session_tab_helper)
         return;
       RenderViewHost* host = content::Details<RenderViewHost>(details).ptr();
       // TODO(mpcomplete): How can we tell if window_id is bogus? It may not
       // have been set yet.
       BrowserThread::PostTask(
           BrowserThread::IO, FROM_HERE,
           base::Bind(
               &ExtensionRendererState::SetTabAndWindowId,
               base::Unretained(ExtensionRendererState::GetInstance()),
               host->GetProcess()->GetID(), host->GetRoutingID(),
               session_tab_helper->session_id().id(),
               session_tab_helper->window_id().id()));
       break;
     }
+    case content::NOTIFICATION_WEB_CONTENTS_CONNECTED: {
+      WebContents* web_contents = content::Source<WebContents>(source).ptr();
+      if (!web_contents->GetRenderProcessHost()->IsGuest())
+        return;
+
+      WebContents* embedder_web_contents =
+          web_contents->GetEmbedderWebContents();
+      WebViewInfo web_view_info;
+      web_view_info.embedder_render_process_host_id =
+          embedder_web_contents->GetRenderProcessHost()->GetID();
+      web_view_info.embedder_routing_id =
+          embedder_web_contents->GetRoutingID();
+      web_view_info.web_view_instance_id = web_contents->GetInstanceID();
+
+      BrowserThread::PostTask(
+          BrowserThread::IO, FROM_HERE,
+          base::Bind(
+              &ExtensionRendererState::AddWebView,
+              base::Unretained(ExtensionRendererState::GetInstance()),
+              web_contents->GetRenderProcessHost()->GetID(),
+              web_contents->GetRoutingID(),
+              web_view_info));
+      break;
+    }
     case chrome::NOTIFICATION_TAB_PARENTED: {
       WebContents* web_contents = content::Source<WebContents>(source).ptr();
       SessionTabHelper* session_tab_helper =
           SessionTabHelper::FromWebContents(web_contents);
       if (!session_tab_helper)
         return;
       RenderViewHost* host = web_contents->GetRenderViewHost();
       BrowserThread::PostTask(
           BrowserThread::IO, FROM_HERE,
           base::Bind(
@@ skipping 18 matching lines @@
           base::Bind(
               &ExtensionRendererState::SetTabAndWindowId,
               base::Unretained(ExtensionRendererState::GetInstance()),
               host->GetProcess()->GetID(), host->GetRoutingID(),
               session_tab_helper->session_id().id(),
               session_tab_helper->window_id().id()));
       break;
     }
     case content::NOTIFICATION_RENDER_VIEW_HOST_DELETED: {
       RenderViewHost* host = content::Source<RenderViewHost>(source).ptr();
+      if (host->GetProcess()->IsGuest()) {
+        BrowserThread::PostTask(
+            BrowserThread::IO, FROM_HERE,
+            base::Bind(
+                &ExtensionRendererState::RemoveWebView,
+                base::Unretained(ExtensionRendererState::GetInstance()),
+                host->GetProcess()->GetID(),
+                host->GetRoutingID()));
+        return;
+      }
       BrowserThread::PostTask(
           BrowserThread::IO, FROM_HERE,
           base::Bind(
               &ExtensionRendererState::ClearTabAndWindowId,
               base::Unretained(ExtensionRendererState::GetInstance()),
               host->GetProcess()->GetID(), host->GetRoutingID()));
       break;
     }
-    case content::NOTIFICATION_RENDERER_PROCESS_CREATED: {
-      RenderProcessHost* render_process_host =
-          content::Source<RenderProcessHost>(source).ptr();
-      if (!render_process_host->IsGuest())
-        return;
-      BrowserThread::PostTask(
-          BrowserThread::IO, FROM_HERE,
-          base::Bind(
-              &ExtensionRendererState::AddGuestProcess,
-              base::Unretained(ExtensionRendererState::GetInstance()),
-              render_process_host->GetID()));
-      break;
-    }
-    case content::NOTIFICATION_RENDERER_PROCESS_TERMINATED:
-    case content::NOTIFICATION_RENDERER_PROCESS_CLOSED: {
-      RenderProcessHost* render_process_host =
-          content::Source<RenderProcessHost>(source).ptr();
-      if (!render_process_host->IsGuest())
-        return;
-      BrowserThread::PostTask(
-          BrowserThread::IO, FROM_HERE,
-          base::Bind(
-              &ExtensionRendererState::RemoveGuestProcess,
-              base::Unretained(ExtensionRendererState::GetInstance()),
-              render_process_host->GetID()));
-      break;
-    }
     default:
       NOTREACHED();
       return;
   }
 }
 
 //
 // ExtensionRendererState
 //
 
@@ skipping 36 matching lines @@
   RenderId render_id(render_process_host_id, routing_id);
   TabAndWindowIdMap::iterator iter = map_.find(render_id);
   if (iter != map_.end()) {
     *tab_id = iter->second.first;
     *window_id = iter->second.second;
     return true;
   }
   return false;
 }
 
-void ExtensionRendererState::AddGuestProcess(int render_process_host_id) {
+void ExtensionRendererState::AddWebView(int render_process_host_id,
+                                        int routing_id,
+                                        const WebViewInfo& web_view_info) {
   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
-  guest_set_.insert(render_process_host_id);
+  RenderId render_id(render_process_host_id, routing_id);
+  web_view_info_map_[render_id] = web_view_info;
 }
 
-void ExtensionRendererState::RemoveGuestProcess(int render_process_host_id) {
+void ExtensionRendererState::RemoveWebView(int render_process_host_id,
+                                           int routing_id) {
   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
-  guest_set_.erase(render_process_host_id);
+  RenderId render_id(render_process_host_id, routing_id);
+  web_view_info_map_.erase(render_id);
 }
 
-bool ExtensionRendererState::IsGuestProcess(int render_process_host_id) {
+bool ExtensionRendererState::GetWebViewInfo(int render_process_host_id,
+                                            int routing_id,
+                                            WebViewInfo* web_view_info) {
   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
-  return guest_set_.count(render_process_host_id) > 0;
+  RenderId render_id(render_process_host_id, routing_id);
+  WebViewInfoMap::iterator iter = web_view_info_map_.find(render_id);
+  if (iter != web_view_info_map_.end()) {
+    *web_view_info = iter->second;
+    return true;
+  }
+  return false;
 }