GuestView: Fix message routing across embedder navigations

extensions/browser/guest_view/guest_view_manager.cc

 // Copyright 2014 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 "extensions/browser/guest_view/guest_view_manager.h"
 
 #include "base/strings/stringprintf.h"
 #include "content/public/browser/browser_context.h"
 #include "content/public/browser/render_frame_host.h"
 #include "content/public/browser/render_process_host.h"
@@ skipping 52 matching lines @@
 content::WebContents* GuestViewManager::GetGuestByInstanceIDSafely(
     int guest_instance_id,
     int embedder_render_process_id) {
   if (!CanEmbedderAccessInstanceIDMaybeKill(embedder_render_process_id,
                                             guest_instance_id)) {
     return nullptr;
   }
   return GetGuestByInstanceID(guest_instance_id);
 }
 
-void GuestViewManager::AttachGuest(
-    int embedder_render_process_id,
-    int embedder_routing_id,
-    int element_instance_id,
-    int guest_instance_id,
-    const base::DictionaryValue& attach_params) {
-  content::WebContents* guest_web_contents =
-      GetGuestByInstanceIDSafely(guest_instance_id, embedder_render_process_id);
-  if (!guest_web_contents)
+void GuestViewManager::AttachGuest(int embedder_process_id,
+                                   int element_instance_id,
+                                   int guest_instance_id,
+                                   const base::DictionaryValue& attach_params) {
+  auto guest_view = GuestViewBase::From(embedder_process_id, guest_instance_id);
+  if (!guest_view)
     return;
 
-  auto guest_view = GuestViewBase::FromWebContents(guest_web_contents);
-  DCHECK(guest_view);
-
-  auto rvh = content::RenderViewHost::FromID(embedder_render_process_id,
-                                             embedder_routing_id);
-  // We need to check that rvh is not nullptr because there may be a race
-  // between AttachGuest and destroying the embedder (i.e. when the embedder is
-  // destroyed immediately after the guest is created).
-  if (!rvh)
-    return;
-  auto owner_web_contents = content::WebContents::FromRenderViewHost(rvh);
-  if (!owner_web_contents)
-    return;
-  ElementInstanceKey key(owner_web_contents, element_instance_id);
-
+  ElementInstanceKey key(embedder_process_id, element_instance_id);
   auto it = instance_id_map_.find(key);
+  // If there is an existing guest attached to the element, then destroy the
+  // existing guest.
   if (it != instance_id_map_.end()) {
     int old_guest_instance_id = it->second;
-    // Reattachment to the same guest is not currently supported.
     if (old_guest_instance_id == guest_instance_id)
       return;
 
-    auto old_guest_web_contents =
-        GetGuestByInstanceIDSafely(old_guest_instance_id,
-                                   embedder_render_process_id);
-    if (!old_guest_web_contents)
-      return;
-
-    auto old_guest_view =
-        GuestViewBase::FromWebContents(old_guest_web_contents);
-
+    auto old_guest_view = GuestViewBase::From(embedder_process_id,
+                                              old_guest_instance_id);
     old_guest_view->Destroy();
   }
   instance_id_map_[key] = guest_instance_id;
   reverse_instance_id_map_[guest_instance_id] = key;
   guest_view->SetAttachParams(attach_params);
 }
 
-void GuestViewManager::DetachGuest(GuestViewBase* guest,
-                                   int element_instance_id) {
+void GuestViewManager::DetachGuest(GuestViewBase* guest) {
   if (!guest->attached())
     return;
 
-  ElementInstanceKey key(guest->owner_web_contents(), element_instance_id);
-  auto it = instance_id_map_.find(key);
-  // There's nothing to do if this key does not exist in the map.
-  if (it == instance_id_map_.end())
+  auto reverse_it = reverse_instance_id_map_.find(guest->guest_instance_id());
+  if (reverse_it == reverse_instance_id_map_.end())
     return;
 
-  int guest_instance_id = it->second;
-  instance_id_map_.erase(key);
+  const ElementInstanceKey& key = reverse_it->second;
 
-  auto reverse_it = reverse_instance_id_map_.find(guest->guest_instance_id());
-  DCHECK(reverse_it != reverse_instance_id_map_.end());
-  DCHECK(reverse_it->second == key);
-  reverse_instance_id_map_.erase(guest_instance_id);
+  auto it = instance_id_map_.find(key);
+  DCHECK(it != instance_id_map_.end());
+
+  reverse_instance_id_map_.erase(reverse_it);
+  instance_id_map_.erase(it);
 }
 
 int GuestViewManager::GetNextInstanceID() {
   return ++current_instance_id_;
 }
 
 void GuestViewManager::CreateGuest(const std::string& view_type,
                                    content::WebContents* owner_web_contents,
                                    const base::DictionaryValue& create_params,
                                    const WebContentsCreatedCallback& callback) {
@@ skipping 13 matching lines @@
   if (!guest)
     return nullptr;
   content::WebContents::CreateParams guest_create_params(create_params);
   guest_create_params.guest_delegate = guest;
   auto guest_web_contents = WebContents::Create(guest_create_params);
   guest->InitWithWebContents(base::DictionaryValue(), guest_web_contents);
   return guest_web_contents;
 }
 
 content::WebContents* GuestViewManager::GetGuestByInstanceID(
-    content::WebContents* owner_web_contents,
+    int owner_process_id,
     int element_instance_id) {
-  int guest_instance_id = GetGuestInstanceIDForElementID(owner_web_contents,
+  int guest_instance_id = GetGuestInstanceIDForElementID(owner_process_id,
                                                          element_instance_id);
   if (guest_instance_id == guestview::kInstanceIDNone)
     return nullptr;
 
   return GetGuestByInstanceID(guest_instance_id);
 }
 
-int GuestViewManager::GetGuestInstanceIDForElementID(
-    content::WebContents* owner_web_contents,
-    int element_instance_id) {
+int GuestViewManager::GetGuestInstanceIDForElementID(int owner_process_id,
+                                                     int element_instance_id) {
   auto iter = instance_id_map_.find(
-      ElementInstanceKey(owner_web_contents, element_instance_id));
+      ElementInstanceKey(owner_process_id, element_instance_id));
   if (iter == instance_id_map_.end())
     return guestview::kInstanceIDNone;
   return iter->second;
 }
 
 SiteInstance* GuestViewManager::GetGuestSiteInstance(
     const GURL& guest_site) {
   for (const auto& guest : guest_web_contents_by_instance_id_) {
     if (guest.second->GetSiteInstance()->GetSiteURL() == guest_site)
       return guest.second->GetSiteInstance();
@@ skipping 109 matching lines @@
 
   auto guest_view = GuestViewBase::FromWebContents(it->second);
   if (!guest_view)
     return false;
 
   return embedder_render_process_id ==
       guest_view->owner_web_contents()->GetRenderProcessHost()->GetID();
 }
 
 }  // namespace extensions