Adds a way to determine id of content handler that created app

mojo/shell/content_handler_connection.cc

 // Copyright 2015 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 "mojo/shell/content_handler_connection.h"
 
+#include "base/bind.h"
 #include "mojo/shell/application_manager.h"
 
 namespace mojo {
 namespace shell {
 
 ContentHandlerConnection::ContentHandlerConnection(
     ApplicationInstance* originator,
     ApplicationManager* manager,
     const GURL& content_handler_url,
     const GURL& requestor_url,
     const std::string& qualifier,
-    const CapabilityFilter& filter)
+    const CapabilityFilter& filter,
+    uint32_t id)
     : manager_(manager),
       content_handler_url_(content_handler_url),
       content_handler_qualifier_(qualifier),
-      connection_closed_(false) {
+      connection_closed_(false),
+      id_(id),
+      got_nested_content_handler_id_(false),
+      nested_content_handler_id_(0u),
+      weak_ptr_factory_(this) {
   ServiceProviderPtr services;
   mojo::URLRequestPtr request(mojo::URLRequest::New());
   request->url = mojo::String::From(content_handler_url.spec());
   manager->ConnectToApplication(
-      originator, request.Pass(), qualifier, requestor_url, GetProxy(&services),
-      nullptr, filter, base::Closure());
+      originator, request.Pass(), qualifier, requestor_url, id,
+      GetProxy(&services), nullptr, filter, base::Closure(),
+      base::Bind(&ContentHandlerConnection::OnGotNestedContentHandlerID,
+                 weak_ptr_factory_.GetWeakPtr()));
   MessagePipe pipe;
   content_handler_.Bind(
       InterfacePtrInfo<ContentHandler>(pipe.handle0.Pass(), 0u));
   services->ConnectToService(ContentHandler::Name_, pipe.handle1.Pass());
   content_handler_.set_connection_error_handler(
       [this]() { CloseConnection(); });
 }
 
 void ContentHandlerConnection::CloseConnection() {
   if (connection_closed_)
     return;
   connection_closed_ = true;
   manager_->OnContentHandlerConnectionClosed(this);
   delete this;
 }
 
+void ContentHandlerConnection::ScheduleTargetIdCallback(
+    const Shell::ConnectToApplicationCallback& connect_callback) {
+  if (got_nested_content_handler_id_)
+    connect_callback.Run(content_handler_id_for_callback());
+  else
+    pending_content_handler_id_callbacks_.push_back(connect_callback);
+}
+
 ContentHandlerConnection::~ContentHandlerConnection() {
   // If this DCHECK fails then something has tried to delete this object without
   // calling CloseConnection.
   DCHECK(connection_closed_);
 }
 
+void ContentHandlerConnection::OnGotNestedContentHandlerID(
+    uint32_t content_handler_id) {
+  DCHECK(!got_nested_content_handler_id_);
+  got_nested_content_handler_id_ = true;
+  nested_content_handler_id_ = content_handler_id;
+  for (auto callback : pending_content_handler_id_callbacks_)
+    callback.Run(content_handler_id_for_callback());
+  pending_content_handler_id_callbacks_.clear();
+}
+
+uint32_t ContentHandlerConnection::content_handler_id_for_callback() const {
+  return nested_content_handler_id_ ==
+                 ApplicationManager::kInvalidContentHandlerID
+             ? id_
+             : nested_content_handler_id_;
+}
+
 }  // namespace shell
 }  // namespace mojo