Simplify resolution of mojo: URLs.

shell/context.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 "shell/context.h"
 
 #include <vector>
 
 #include "base/base_switches.h"
 #include "base/bind.h"
 #include "base/command_line.h"
 #include "base/files/file_path.h"
 #include "base/lazy_instance.h"
 #include "base/macros.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/memory/scoped_vector.h"
+#include "base/path_service.h"
 #include "base/strings/string_split.h"
 #include "base/strings/string_util.h"
 #include "build/build_config.h"
 #include "mojo/common/tracing_impl.h"
 #include "mojo/edk/embedder/embedder.h"
 #include "mojo/edk/embedder/simple_platform_support.h"
 #include "mojo/public/cpp/application/application_connection.h"
 #include "mojo/public/cpp/application/application_delegate.h"
 #include "mojo/public/cpp/application/application_impl.h"
 #include "services/tracing/tracing.mojom.h"
 #include "shell/application_manager/application_loader.h"
 #include "shell/application_manager/application_manager.h"
 #include "shell/dynamic_application_loader.h"
 #include "shell/external_application_listener.h"
+#include "shell/filename_util.h"
 #include "shell/in_process_dynamic_service_runner.h"
 #include "shell/out_of_process_dynamic_service_runner.h"
 #include "shell/switches.h"
 #include "url/gurl.h"
 
 namespace mojo {
 namespace shell {
 namespace {
 
-// These mojo: URLs are loaded directly from the local filesystem. They
-// correspond to shared libraries bundled alongside the mojo_shell.
-const char* kLocalMojoURLs[] = {
-    "mojo:network_service",
-};
-
 // Used to ensure we only init once.
 class Setup {
  public:
   Setup() {
     embedder::Init(scoped_ptr<mojo::embedder::PlatformSupport>(
         new mojo::embedder::SimplePlatformSupport()));
   }
 
   ~Setup() {}
 
@@ skipping 43 matching lines @@
       LOG(ERROR) << "Invalid value for switch " << switches::kContentHandlers
                  << ": '" << parts[i + 1] << "' is not a valid URL.";
       return;
     }
     // TODO(eseidel): We should also validate that the mimetype is valid
     // net/base/mime_util.h could do this, but we don't want to depend on net.
     loader->RegisterContentHandler(parts[i], url);
   }
 }
 
-bool ConfigureURLMappings(const std::string& mappings,
-                          mojo::shell::MojoURLResolver* resolver) {
-  base::StringPairs pairs;
-  if (!base::SplitStringIntoKeyValuePairs(mappings, '=', ',', &pairs))
+bool ConfigureURLMappings(base::CommandLine* command_line,
+                          Context* context) {
+  MojoURLResolver* resolver = context->mojo_url_resolver();
+
+  // Configure the resolution of unknown mojo: URLs.
+  GURL base_url;
+  if (command_line->HasSwitch(switches::kOrigin))
+    base_url = GURL(command_line->GetSwitchValueASCII(switches::kOrigin));
+  else
+    // Use the shell's file root if the base was not specified.
+    base_url = context->ResolveShellFileURL("");
+
+  if (!base_url.is_valid())
     return false;
-  using StringPair = std::pair<std::string, std::string>;
-  for (const StringPair& pair : pairs) {
-    const GURL from(pair.first);
-    const GURL to(pair.second);
-    if (!from.is_valid() || !to.is_valid())
+
+  resolver->SetBaseURL(base_url);
+
+  // The network service must be loaded from the filesystem.
+  // This mapping is done before the command line URL mapping are processed, so
+  // that it can be overridden.
+  resolver->AddCustomMapping(
+      GURL("mojo:network_service"),
+      context->ResolveShellFileURL("file:network_service.mojo"));
+
+  // Command line URL mapping.
+  if (command_line->HasSwitch(switches::kURLMappings)) {
+    const std::string mappings =
+      command_line->GetSwitchValueASCII(switches::kURLMappings);
+
+    base::StringPairs pairs;
+    if (!base::SplitStringIntoKeyValuePairs(mappings, '=', ',', &pairs))
       return false;
-    resolver->AddCustomMapping(from, to);
+    using StringPair = std::pair<std::string, std::string>;
+    for (const StringPair& pair : pairs) {
+      const GURL from(pair.first);
+      const GURL to(pair.second);
+      if (!from.is_valid() || !to.is_valid())
+        return false;
+      resolver->AddCustomMapping(from, to);
+    }
   }
   return true;
 }
 
 }  // namespace
 
 Context::Context() : application_manager_(this) {
   DCHECK(!base::MessageLoop::current());
+
+  // By default assume that the local apps reside alongside the shell.
+  // TODO(ncbray): really, this should be passed in rather than defaulting.
+  // This default makes sense for desktop but not Android.
+  base::FilePath shell_dir;
+  PathService::Get(base::DIR_MODULE, &shell_dir);
+  SetShellFileRoot(shell_dir);
 }
 
 Context::~Context() {
   DCHECK(!base::MessageLoop::current());
 }
 
 void Context::EnsureEmbedderIsInitialized() {
   setup.Get();
 }
 
+void Context::SetShellFileRoot(const base::FilePath& path) {
+  shell_file_root_ = AddTrailingSlashIfNeeded(FilePathToFileURL(path));
+}
+
+GURL Context::ResolveShellFileURL(const std::string& path) {
+  return shell_file_root_.Resolve(path);
+}
+
 bool Context::Init() {
   base::CommandLine* command_line = base::CommandLine::ForCurrentProcess();
 
   if (command_line->HasSwitch(switches::kWaitForDebugger))
     base::debug::WaitForDebugger(60, true);
 
   EnsureEmbedderIsInitialized();
   task_runners_.reset(
       new TaskRunners(base::MessageLoop::current()->message_loop_proxy()));
 
-  for (size_t i = 0; i < arraysize(kLocalMojoURLs); ++i)
-    mojo_url_resolver_.AddLocalFileMapping(GURL(kLocalMojoURLs[i]));
+  if (!shell_file_root_.is_valid())
+    return false;
+  if (!ConfigureURLMappings(command_line, this))
+    return false;
 
   if (command_line->HasSwitch(switches::kEnableExternalApplications)) {
     listener_.reset(new ExternalApplicationListener(
         task_runners_->shell_runner(), task_runners_->io_runner()));
 
     base::FilePath socket_path =
         command_line->GetSwitchValuePath(switches::kEnableExternalApplications);
     if (socket_path.empty())
       socket_path = ExternalApplicationListener::ConstructDefaultSocketPath();
 
     listener_->ListenInBackground(
         socket_path,
         base::Bind(&ApplicationManager::RegisterExternalApplication,
                    base::Unretained(&application_manager_)));
   }
-  if (command_line->HasSwitch(switches::kOrigin)) {
-    mojo_url_resolver()->SetBaseURL(
-        GURL(command_line->GetSwitchValueASCII(switches::kOrigin)));
-  }
-  if (command_line->HasSwitch(switches::kURLMappings) &&
-      !ConfigureURLMappings(
-          command_line->GetSwitchValueASCII(switches::kURLMappings),
-          mojo_url_resolver())) {
-    return false;
-  }
 
   scoped_ptr<DynamicServiceRunnerFactory> runner_factory;
   if (command_line->HasSwitch(switches::kEnableMultiprocess))
     runner_factory.reset(new OutOfProcessDynamicServiceRunnerFactory());
   else
     runner_factory.reset(new InProcessDynamicServiceRunnerFactory());
 
   DynamicApplicationLoader* dynamic_application_loader =
       new DynamicApplicationLoader(this, runner_factory.Pass());
   InitContentHandlers(dynamic_application_loader, command_line);
@@ skipping 39 matching lines @@
 ScopedMessagePipeHandle Context::ConnectToServiceByName(
     const GURL& application_url,
     const std::string& service_name) {
   app_urls_.insert(application_url);
   return application_manager_.ConnectToServiceByName(application_url,
                                                      service_name).Pass();
 }
 
 }  // namespace shell
 }  // namespace mojo