Adds routed interface support between RenderFrameHost and RenderFrame

content/browser/frame_host/render_frame_host_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/render_frame_host_impl.h"
 
 #include <utility>
 
 #include "base/bind.h"
 #include "base/command_line.h"
@@ skipping 69 matching lines @@
 #include "content/public/common/file_chooser_params.h"
 #include "content/public/common/isolated_world_ids.h"
 #include "content/public/common/mojo_shell_connection.h"
 #include "content/public/common/url_constants.h"
 #include "content/public/common/url_utils.h"
 #include "device/generic_sensor/sensor_provider_impl.h"
 #include "device/geolocation/geolocation_service_context.h"
 #include "device/vibration/vibration_manager_impl.h"
 #include "media/mojo/interfaces/media_service.mojom.h"
 #include "media/mojo/interfaces/service_factory.mojom.h"
+#include "mojo/public/cpp/bindings/associated_interface_ptr.h"
+#include "mojo/public/cpp/bindings/associated_interface_request.h"
 #include "services/shell/public/cpp/connector.h"
 #include "services/shell/public/cpp/interface_provider.h"
 #include "ui/accessibility/ax_tree.h"
 #include "ui/accessibility/ax_tree_update.h"
 #include "ui/gfx/geometry/quad_f.h"
 #include "url/gurl.h"
 
 #if defined(OS_ANDROID)
 #include "content/browser/mojo/interface_registrar_android.h"
 #if defined(ENABLE_MOJO_CDM)
@@ skipping 160 matching lines @@
       browser_plugin_embedder_ax_tree_id_(AXTreeIDRegistry::kNoAXTreeID),
       no_create_browser_accessibility_manager_for_testing_(false),
       web_ui_type_(WebUI::kNoWebUI),
       pending_web_ui_type_(WebUI::kNoWebUI),
       should_reuse_web_ui_(false),
       last_navigation_lofi_state_(LOFI_UNSPECIFIED),
       frame_host_binding_(this),
       weak_ptr_factory_(this) {
   frame_tree_->AddRenderViewHostRef(render_view_host_);
   GetProcess()->AddRoute(routing_id_, this);
+  GetProcess()->AddRoutedInterfaces(routing_id_, this);
   g_routing_id_frame_map.Get().insert(std::make_pair(
       RenderFrameHostID(GetProcess()->GetID(), routing_id_),
       this));
   site_instance_->AddObserver(this);
   GetSiteInstance()->IncrementActiveFrameCount();
 
   if (frame_tree_node_->parent()) {
     // Keep track of the parent RenderFrameHost, which shouldn't change even if
     // this RenderFrameHost is on the pending deletion list and the parent
     // FrameTreeNode has changed its current RenderFrameHost.
@@ skipping 213 matching lines @@
 }
 
 shell::InterfaceRegistry* RenderFrameHostImpl::GetInterfaceRegistry() {
   return interface_registry_.get();
 }
 
 shell::InterfaceProvider* RenderFrameHostImpl::GetRemoteInterfaces() {
   return remote_interfaces_.get();
 }
 
+void RenderFrameHostImpl::AddRoutedInterface(
+    const base::StringPiece& name,
+    const RoutedInterfaceFactory& factory) {
+  auto result =
+      routed_interfaces_.insert(std::make_pair(name.as_string(), factory));
+  DCHECK(result.second);
+}
+
+void RenderFrameHostImpl::RemoveRoutedInterface(const base::StringPiece& name) {
+  auto it = routed_interfaces_.find(name.as_string());
+  DCHECK(it != routed_interfaces_.end());
+  routed_interfaces_.erase(it);
+}
+
+void RenderFrameHostImpl::GetRemoteRoutedInterface(
+    const base::StringPiece& name,
+    mojo::ScopedInterfaceEndpointHandle handle) {
+  mojom::RoutedInterfaceAssociatedRequest request;
+  request.Bind(std::move(handle));
+  GetRemoteRoutedInterfaces()->GetRoutedInterface(name.as_string(),
+                                                  std::move(request));
+}
+
 blink::WebPageVisibilityState RenderFrameHostImpl::GetVisibilityState() {
   // Works around the crashes seen in https://crbug.com/501863, where the
   // active WebContents from a browser iterator may contain a render frame
   // detached from the frame tree. This tries to find a RenderWidgetHost
   // attached to an ancestor frame, and defaults to visibility hidden if
   // it fails.
   // TODO(yfriedman, peter): Ideally this would never be called on an
   // unattached frame and we could omit this check. See
   // https://crbug.com/615867.
   RenderFrameHostImpl* frame = this;
@@ skipping 943 matching lines @@
   // reset.
   SetRenderFrameCreated(false);
   InvalidateMojoConnection();
 
   // Execute any pending AX tree snapshot callbacks with an empty response,
   // since we're never going to get a response from this renderer.
   for (const auto& iter : ax_tree_snapshot_callbacks_)
     iter.second.Run(ui::AXTreeUpdate());
   ax_tree_snapshot_callbacks_.clear();
 
+  // Ensure that future remote interface requests are associated with the new
+  // process's channel.
+  remote_routed_interfaces_.reset();
+
   if (!is_active()) {
     // If the process has died, we don't need to wait for the swap out ack from
     // this RenderFrame if it is pending deletion.  Complete the swap out to
     // destroy it.
     OnSwappedOut();
   } else {
     // If this was the current pending or speculative RFH dying, cancel and
     // destroy it.
     frame_tree_node_->render_manager()->CancelPendingIfNecessary(this);
   }
@@ skipping 1578 matching lines @@
   // handler after it's destroyed so it can't run after the RFHI is destroyed.
   web_bluetooth_service_->SetClientConnectionErrorHandler(base::Bind(
       &RenderFrameHostImpl::DeleteWebBluetoothService, base::Unretained(this)));
   return web_bluetooth_service_.get();
 }
 
 void RenderFrameHostImpl::DeleteWebBluetoothService() {
   web_bluetooth_service_.reset();
 }
 
+mojom::RoutedInterfaceProvider*
+RenderFrameHostImpl::GetRemoteRoutedInterfaces() {
+  if (!remote_routed_interfaces_) {
+    GetProcess()->GetRemoteRouteProvider()->GetRoutedInterfaces(
+        GetRoutingID(),
+        mojo::GetProxy(&remote_routed_interfaces_,
+                       GetProcess()->GetChannel()->GetAssociatedGroup()));
+  }
+  return remote_routed_interfaces_.get();
+}
+
+void RenderFrameHostImpl::GetRoutedInterface(
+    const std::string& name,
+    mojom::RoutedInterfaceAssociatedRequest request) {
+  auto it = routed_interfaces_.find(name);
+  if (it != routed_interfaces_.end())
+    it->second.Run(request.PassHandle());
+}
+
 }  // namespace content