[Extensions] Extension Port Ids and Initialization 2.0

extensions/renderer/extension_frame_helper.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 "extensions/renderer/extension_frame_helper.h"
 
 #include "base/metrics/histogram_macros.h"
 #include "base/strings/string_util.h"
 #include "base/timer/elapsed_timer.h"
 #include "content/public/renderer/render_frame.h"
 #include "extensions/common/api/messaging/message.h"
+#include "extensions/common/api/messaging/port_id.h"
 #include "extensions/common/constants.h"
 #include "extensions/common/extension_messages.h"
 #include "extensions/common/manifest_handlers/background_info.h"
 #include "extensions/renderer/console.h"
 #include "extensions/renderer/content_watcher.h"
 #include "extensions/renderer/dispatcher.h"
 #include "extensions/renderer/messaging_bindings.h"
 #include "extensions/renderer/script_context.h"
 #include "third_party/WebKit/public/platform/WebSecurityOrigin.h"
 #include "third_party/WebKit/public/web/WebConsoleMessage.h"
@@ skipping 61 matching lines @@
 }
 
 enum class PortType {
   EXTENSION,
   TAB,
   NATIVE_APP,
 };
 
 }  // namespace
 
-struct ExtensionFrameHelper::PendingPortRequest {
-  PendingPortRequest(PortType type, const base::Callback<void(int)>& callback)
-      : type(type), callback(callback) {}
-  ~PendingPortRequest() {}
-
-  base::ElapsedTimer timer;
-  PortType type;
-  base::Callback<void(int)> callback;
-
- private:
-  DISALLOW_COPY_AND_ASSIGN(PendingPortRequest);
-};
-
 ExtensionFrameHelper::ExtensionFrameHelper(content::RenderFrame* render_frame,
                                            Dispatcher* extension_dispatcher)
     : content::RenderFrameObserver(render_frame),
       content::RenderFrameObserverTracker<ExtensionFrameHelper>(render_frame),
       view_type_(VIEW_TYPE_INVALID),
       tab_id_(-1),
       browser_window_id_(-1),
       extension_dispatcher_(extension_dispatcher),
       did_create_current_document_element_(false),
-      next_port_request_id_(0),
       weak_ptr_factory_(this) {
   g_frame_helpers.Get().insert(this);
 }
 
 ExtensionFrameHelper::~ExtensionFrameHelper() {
   g_frame_helpers.Get().erase(this);
 }
 
 // static
 std::vector<content::RenderFrame*> ExtensionFrameHelper::GetExtensionFrames(
@@ skipping 61 matching lines @@
 void ExtensionFrameHelper::ScheduleAtDocumentStart(
     const base::Closure& callback) {
   document_element_created_callbacks_.push_back(callback);
 }
 
 void ExtensionFrameHelper::ScheduleAtDocumentEnd(
     const base::Closure& callback) {
   document_load_finished_callbacks_.push_back(callback);
 }
 
-void ExtensionFrameHelper::RequestPortId(
-    const ExtensionMsg_ExternalConnectionInfo& info,
-    const std::string& channel_name,
-    bool include_tls_channel_id,
-    const base::Callback<void(int)>& callback) {
-  int port_request_id = next_port_request_id_++;
-  pending_port_requests_[port_request_id] =
-      base::MakeUnique<PendingPortRequest>(PortType::EXTENSION, callback);
-  {
-    SCOPED_UMA_HISTOGRAM_TIMER(
-        "Extensions.Messaging.GetPortIdSyncTime.Extension");
-    render_frame()->Send(new ExtensionHostMsg_OpenChannelToExtension(
-        render_frame()->GetRoutingID(), info, channel_name,
-        include_tls_channel_id, port_request_id));
-  }
-}
-
-void ExtensionFrameHelper::RequestTabPortId(
-    const ExtensionMsg_TabTargetConnectionInfo& info,
-    const std::string& extension_id,
-    const std::string& channel_name,
-    const base::Callback<void(int)>& callback) {
-  int port_request_id = next_port_request_id_++;
-  pending_port_requests_[port_request_id] =
-      base::MakeUnique<PendingPortRequest>(PortType::TAB, callback);
-  {
-    SCOPED_UMA_HISTOGRAM_TIMER("Extensions.Messaging.GetPortIdSyncTime.Tab");
-    render_frame()->Send(new ExtensionHostMsg_OpenChannelToTab(
-        render_frame()->GetRoutingID(), info, extension_id, channel_name,
-        port_request_id));
-  }
-}
-
-void ExtensionFrameHelper::RequestNativeAppPortId(
-    const std::string& native_app_name,
-    const base::Callback<void(int)>& callback) {
-  int port_request_id = next_port_request_id_++;
-  pending_port_requests_[port_request_id] =
-      base::MakeUnique<PendingPortRequest>(PortType::NATIVE_APP, callback);
-  {
-    SCOPED_UMA_HISTOGRAM_TIMER(
-        "Extensions.Messaging.GetPortIdSyncTime.NativeApp");
-    render_frame()->Send(new ExtensionHostMsg_OpenChannelToNativeApp(
-        render_frame()->GetRoutingID(), native_app_name, port_request_id));
-  }
-}
-
-int ExtensionFrameHelper::RequestSyncPortId(
-    const ExtensionMsg_ExternalConnectionInfo& info,
-    const std::string& channel_name,
-    bool include_tls_channel_id) {
-  int port_id = 0;
-  render_frame()->Send(new ExtensionHostMsg_OpenChannelToExtensionSync(
-      render_frame()->GetRoutingID(), info, channel_name,
-      include_tls_channel_id, &port_id));
-  return port_id;
-}
-
 void ExtensionFrameHelper::DidMatchCSS(
     const blink::WebVector<blink::WebString>& newly_matching_selectors,
     const blink::WebVector<blink::WebString>& stopped_matching_selectors) {
   extension_dispatcher_->content_watcher()->DidMatchCSS(
       render_frame()->GetWebFrame(), newly_matching_selectors,
       stopped_matching_selectors);
 }
 
 void ExtensionFrameHelper::DidStartProvisionalLoad() {
   if (!delayed_main_world_script_initialization_)
@@ skipping 45 matching lines @@
     IPC_MESSAGE_HANDLER(ExtensionMsg_DeliverMessage, OnExtensionDeliverMessage)
     IPC_MESSAGE_HANDLER(ExtensionMsg_DispatchOnDisconnect,
                         OnExtensionDispatchOnDisconnect)
     IPC_MESSAGE_HANDLER(ExtensionMsg_SetTabId, OnExtensionSetTabId)
     IPC_MESSAGE_HANDLER(ExtensionMsg_UpdateBrowserWindowId,
                         OnUpdateBrowserWindowId)
     IPC_MESSAGE_HANDLER(ExtensionMsg_NotifyRenderViewType,
                         OnNotifyRendererViewType)
     IPC_MESSAGE_HANDLER(ExtensionMsg_Response, OnExtensionResponse)
     IPC_MESSAGE_HANDLER(ExtensionMsg_MessageInvoke, OnExtensionMessageInvoke)
-    IPC_MESSAGE_HANDLER(ExtensionMsg_AssignPortId, OnAssignPortId)
     IPC_MESSAGE_UNHANDLED(handled = false)
   IPC_END_MESSAGE_MAP()
   return handled;
 }
 
-void ExtensionFrameHelper::OnExtensionValidateMessagePort(int port_id) {
+void ExtensionFrameHelper::OnExtensionValidateMessagePort(const PortId& id) {
   MessagingBindings::ValidateMessagePort(
-      extension_dispatcher_->script_context_set(), port_id, render_frame());
+      extension_dispatcher_->script_context_set(), id, render_frame());
 }
 
 void ExtensionFrameHelper::OnExtensionDispatchOnConnect(
-    int target_port_id,
+    const PortId& target_port_id,
     const std::string& channel_name,
     const ExtensionMsg_TabConnectionInfo& source,
     const ExtensionMsg_ExternalConnectionInfo& info,
     const std::string& tls_channel_id) {
   MessagingBindings::DispatchOnConnect(
       extension_dispatcher_->script_context_set(),
       target_port_id,
       channel_name,
       source,
       info,
       tls_channel_id,
       render_frame());
 }
 
-void ExtensionFrameHelper::OnExtensionDeliverMessage(int target_id,
+void ExtensionFrameHelper::OnExtensionDeliverMessage(const PortId& target_id,
                                                      const Message& message) {
   MessagingBindings::DeliverMessage(
       extension_dispatcher_->script_context_set(), target_id, message,
       render_frame());
 }
 
 void ExtensionFrameHelper::OnExtensionDispatchOnDisconnect(
-    int port_id,
+    const PortId& id,
     const std::string& error_message) {
   MessagingBindings::DispatchOnDisconnect(
-      extension_dispatcher_->script_context_set(), port_id, error_message,
+      extension_dispatcher_->script_context_set(), id, error_message,
       render_frame());
 }
 
 void ExtensionFrameHelper::OnExtensionSetTabId(int tab_id) {
   CHECK_EQ(tab_id_, -1);
   CHECK_GE(tab_id, 0);
   tab_id_ = tab_id;
 }
 
 void ExtensionFrameHelper::OnUpdateBrowserWindowId(int browser_window_id) {
@@ skipping 18 matching lines @@
 
 void ExtensionFrameHelper::OnExtensionMessageInvoke(
     const std::string& extension_id,
     const std::string& module_name,
     const std::string& function_name,
     const base::ListValue& args) {
   extension_dispatcher_->InvokeModuleSystemMethod(
       render_frame(), extension_id, module_name, function_name, args);
 }
 
-void ExtensionFrameHelper::OnAssignPortId(int port_id, int request_id) {
-  auto iter = pending_port_requests_.find(request_id);
-  DCHECK(iter != pending_port_requests_.end());
-  PendingPortRequest& request = *iter->second;
-  switch (request.type) {
-    case PortType::EXTENSION: {
-      UMA_HISTOGRAM_TIMES("Extensions.Messaging.GetPortIdAsyncTime.Extension",
-                          request.timer.Elapsed());
-      break;
-    }
-    case PortType::TAB: {
-      UMA_HISTOGRAM_TIMES("Extensions.Messaging.GetPortIdAsyncTime.Tab",
-                          request.timer.Elapsed());
-      break;
-    }
-    case PortType::NATIVE_APP: {
-      UMA_HISTOGRAM_TIMES("Extensions.Messaging.GetPortIdAsyncTime.NativeApp",
-                          request.timer.Elapsed());
-      break;
-    }
-  }
-  request.callback.Run(port_id);
-  pending_port_requests_.erase(iter);
-}
-
 void ExtensionFrameHelper::OnDestruct() {
   delete this;
 }
 
 }  // namespace extensions