Allow trusted brokers to restrict connections for spawned applications to whitelisted applications …

mojo/shell/application_instance.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/application_instance.h"
 
 #include "base/bind.h"
 #include "base/stl_util.h"
 #include "mojo/application/public/interfaces/content_handler.mojom.h"
 #include "mojo/common/common_type_converters.h"
 #include "mojo/common/url_type_converters.h"
 #include "mojo/shell/application_manager.h"
 
 namespace mojo {
 namespace shell {
 
 ApplicationInstance::QueuedClientRequest::QueuedClientRequest() {
 }
 
 ApplicationInstance::QueuedClientRequest::~QueuedClientRequest() {
 }
 
 ApplicationInstance::ApplicationInstance(
     ApplicationPtr application,
     ApplicationManager* manager,
     const Identity& identity,
+    const CapabilityFilter& filter,
     const base::Closure& on_application_end)
     : manager_(manager),
       identity_(identity),
+      filter_(filter),
       on_application_end_(on_application_end),
       application_(application.Pass()),
       binding_(this),
       queue_requests_(false) {
   binding_.set_connection_error_handler([this]() { OnConnectionError(); });
 }
 
 ApplicationInstance::~ApplicationInstance() {
   STLDeleteElements(&queued_client_requests_);
 }
 
 void ApplicationInstance::InitializeApplication() {
   ShellPtr shell;
   binding_.Bind(GetProxy(&shell));
   application_->Initialize(shell.Pass(), identity_.url.spec());
 }
 
 void ApplicationInstance::ConnectToClient(
+    ApplicationInstance* originator,
     const GURL& requested_url,
     const GURL& requestor_url,
     InterfaceRequest<ServiceProvider> services,
-    ServiceProviderPtr exposed_services) {
+    ServiceProviderPtr exposed_services,
+    CapabilityFilterPtr filter) {
   if (queue_requests_) {
     QueuedClientRequest* queued_request = new QueuedClientRequest;
+    queued_request->originator = originator;
     queued_request->requested_url = requested_url;
     queued_request->requestor_url = requestor_url;
     queued_request->services = services.Pass();
     queued_request->exposed_services = exposed_services.Pass();
+    queued_request->filter = filter.Pass(),
     queued_client_requests_.push_back(queued_request);
     return;
   }
 
-  application_->AcceptConnection(requestor_url.spec(), services.Pass(),
-                                 exposed_services.Pass(), requested_url.spec());
+  CallAcceptConnection(originator, requestor_url, services.Pass(),
+                       exposed_services.Pass(), requested_url);
+}
+
+ApplicationInstance::AllowedInterfaces
+    ApplicationInstance::GetAllowedInterfaces(
+        const Identity& identity) const {
+  auto it = filter_.find(identity.url.spec());
+  return it != filter_.end() ? it->second : AllowedInterfaces();
 }
 
 // Shell implementation:
 void ApplicationInstance::ConnectToApplication(
-    mojo::URLRequestPtr app_request,
+    URLRequestPtr app_request,
     InterfaceRequest<ServiceProvider> services,
-    ServiceProviderPtr exposed_services) {
-  GURL app_gurl(app_request->url.To<std::string>());
-  if (!app_gurl.is_valid()) {
-    LOG(ERROR) << "Error: invalid URL: " << app_request;
+    ServiceProviderPtr exposed_services,
+    CapabilityFilterPtr filter) {
+  std::string url_string = app_request->url.To<std::string>();
+  if (!GURL(url_string).is_valid()) {
+    LOG(ERROR) << "Error: invalid URL: " << url_string;
     return;
   }
-  manager_->ConnectToApplication(app_request.Pass(), std::string(),
-                                 identity_.url, services.Pass(),
-                                 exposed_services.Pass(), base::Closure());
+  if (filter_.empty() || filter_.find(url_string) != filter_.end()) {
+    manager_->ConnectToApplication(this, app_request.Pass(), std::string(),
+                                   identity_.url, services.Pass(),
+                                   exposed_services.Pass(), filter.Pass(),
+                                   base::Closure());
+  } else {
+    DVLOG(2) << "CapabilityFilter prevented connection to: " << url_string;
+  }
 }
 
 void ApplicationInstance::QuitApplication() {
   queue_requests_ = true;
   application_->OnQuitRequested(
       base::Bind(&ApplicationInstance::OnQuitRequestedResult,
                  base::Unretained(this)));
 }
 
+void ApplicationInstance::CallAcceptConnection(
+    ApplicationInstance* originator,
+    const GURL& requestor_url,
+    InterfaceRequest<ServiceProvider> services,
+    ServiceProviderPtr exposed_services,
+    const GURL& requested_url) {
+  AllowedInterfaces interfaces;
+  if (originator)
+    interfaces = originator->GetAllowedInterfaces(identity_);
+  application_->AcceptConnection(requestor_url.spec(),
+                                 services.Pass(),
+                                 exposed_services.Pass(),
+                                 Array<String>::From(interfaces).Pass(),
+                                 requested_url.spec());
+}
+
 void ApplicationInstance::OnConnectionError() {
   std::vector<QueuedClientRequest*> queued_client_requests;
   queued_client_requests_.swap(queued_client_requests);
   auto manager = manager_;
   manager_->OnApplicationInstanceError(this);
   //|this| is deleted.
 
   // If any queued requests came to shell during time it was shutting down,
   // start them now.
   for (auto request : queued_client_requests) {
     mojo::URLRequestPtr url(mojo::URLRequest::New());
     url->url = mojo::String::From(request->requested_url.spec());
-    manager->ConnectToApplication(url.Pass(), std::string(),
+    manager->ConnectToApplication(this, url.Pass(), std::string(),
                                   request->requestor_url,
                                   request->services.Pass(),
                                   request->exposed_services.Pass(),
+                                  request->filter.Pass(),
                                   base::Closure());
   }
   STLDeleteElements(&queued_client_requests);
 }
 
 void ApplicationInstance::OnQuitRequestedResult(bool can_quit) {
   if (can_quit)
     return;
 
   queue_requests_ = false;
   for (auto request : queued_client_requests_) {
-    application_->AcceptConnection(request->requestor_url.spec(),
-                                   request->services.Pass(),
-                                   request->exposed_services.Pass(),
-                                   request->requested_url.spec());
+    CallAcceptConnection(request->originator,
+                         request->requestor_url,
+                         request->services.Pass(),
+                         request->exposed_services.Pass(),
+                         request->requested_url);
   }
   STLDeleteElements(&queued_client_requests_);
 }
 
 }  // namespace shell
 }  // namespace mojo