Index: mojo/shell/context.cc |
diff --git a/mojo/shell/context.cc b/mojo/shell/context.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..fbde07dd2ab0e6510a0905073a08a30e56ffae25 |
--- /dev/null |
+++ b/mojo/shell/context.cc |
@@ -0,0 +1,328 @@ |
+// 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 "mojo/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/run_loop.h" |
+#include "base/strings/string_split.h" |
+#include "base/strings/string_util.h" |
+#include "base/trace_event/trace_event.h" |
+#include "build/build_config.h" |
+#include "mojo/common/trace_controller_impl.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 "mojo/services/tracing/tracing.mojom.h" |
+#include "mojo/shell/application_manager/application_loader.h" |
+#include "mojo/shell/application_manager/application_manager.h" |
+#include "mojo/shell/command_line_util.h" |
+#include "mojo/shell/filename_util.h" |
+#include "mojo/shell/in_process_native_runner.h" |
+#include "mojo/shell/out_of_process_native_runner.h" |
+#include "mojo/shell/switches.h" |
+#include "url/gurl.h" |
+ |
+namespace mojo { |
+namespace shell { |
+namespace { |
+ |
+// Used to ensure we only init once. |
+class Setup { |
+ public: |
+ Setup() { |
+ embedder::Init(make_scoped_ptr(new embedder::SimplePlatformSupport())); |
+ } |
+ |
+ ~Setup() {} |
+ |
+ private: |
+ DISALLOW_COPY_AND_ASSIGN(Setup); |
+}; |
+ |
+bool ConfigureURLMappings(const base::CommandLine& command_line, |
+ Context* context) { |
+ URLResolver* resolver = context->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; |
+ |
+ resolver->SetMojoBaseURL(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->AddURLMapping( |
+ GURL("mojo:network_service"), |
+ context->ResolveShellFileURL("file:network_service.mojo")); |
+ |
+ // Command line URL mapping. |
+ std::vector<URLResolver::OriginMapping> origin_mappings = |
+ URLResolver::GetOriginMappings(command_line.argv()); |
+ for (const auto& origin_mapping : origin_mappings) |
+ resolver->AddOriginMapping(GURL(origin_mapping.origin), |
+ GURL(origin_mapping.base_url)); |
+ |
+ 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; |
+ using StringPair = std::pair<std::string, std::string>; |
+ for (const StringPair& pair : pairs) { |
+ const GURL from(pair.first); |
+ const GURL to = context->ResolveCommandLineURL(pair.second); |
+ if (!from.is_valid() || !to.is_valid()) |
+ return false; |
+ resolver->AddURLMapping(from, to); |
+ } |
+ } |
+ return true; |
+} |
+ |
+void InitContentHandlers(ApplicationManager* manager, |
+ const base::CommandLine& command_line) { |
+ // Default content handlers. |
+ manager->RegisterContentHandler("application/pdf", GURL("mojo:pdf_viewer")); |
+ manager->RegisterContentHandler("image/png", GURL("mojo:png_viewer")); |
+ manager->RegisterContentHandler("text/html", GURL("mojo:html_viewer")); |
+ |
+ // Command-line-specified content handlers. |
+ std::string handlers_spec = |
+ command_line.GetSwitchValueASCII(switches::kContentHandlers); |
+ if (handlers_spec.empty()) |
+ return; |
+ |
+#if defined(OS_ANDROID) |
+ // TODO(eseidel): On Android we pass command line arguments is via the |
+ // 'parameters' key on the intent, which we specify during 'am shell start' |
+ // via --esa, however that expects comma-separated values and says: |
+ // am shell --help: |
+ // [--esa <EXTRA_KEY> <EXTRA_STRING_VALUE>[,<EXTRA_STRING_VALUE...]] |
+ // (to embed a comma into a string escape it using "\,") |
+ // Whatever takes 'parameters' and constructs a CommandLine is failing to |
+ // un-escape the commas, we need to move this fix to that file. |
+ ReplaceSubstringsAfterOffset(&handlers_spec, 0, "\\,", ","); |
+#endif |
+ |
+ std::vector<std::string> parts; |
+ base::SplitString(handlers_spec, ',', &parts); |
+ if (parts.size() % 2 != 0) { |
+ LOG(ERROR) << "Invalid value for switch " << switches::kContentHandlers |
+ << ": must be a comma-separated list of mimetype/url pairs." |
+ << handlers_spec; |
+ return; |
+ } |
+ |
+ for (size_t i = 0; i < parts.size(); i += 2) { |
+ GURL url(parts[i + 1]); |
+ if (!url.is_valid()) { |
+ 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. |
+ manager->RegisterContentHandler(parts[i], url); |
+ } |
+} |
+ |
+void InitNativeOptions(ApplicationManager* manager, |
+ const base::CommandLine& command_line) { |
+ std::vector<std::string> force_in_process_url_list; |
+ base::SplitString(command_line.GetSwitchValueASCII(switches::kForceInProcess), |
+ ',', &force_in_process_url_list); |
+ for (const auto& force_in_process_url : force_in_process_url_list) { |
+ GURL gurl(force_in_process_url); |
+ if (!gurl.is_valid()) { |
+ LOG(ERROR) << "Invalid value for switch " << switches::kForceInProcess |
+ << ": '" << force_in_process_url << "'is not a valid URL."; |
+ return; |
+ } |
+ |
+ NativeRunnerFactory::Options options; |
+ options.force_in_process = true; |
+ manager->SetNativeOptionsForURL(options, gurl); |
+ } |
+} |
+ |
+class TracingServiceProvider : public ServiceProvider { |
+ public: |
+ explicit TracingServiceProvider(InterfaceRequest<ServiceProvider> request) |
+ : binding_(this, request.Pass()) {} |
+ ~TracingServiceProvider() override {} |
+ |
+ void ConnectToService(const String& service_name, |
+ ScopedMessagePipeHandle client_handle) override { |
+ if (service_name == tracing::TraceController::Name_) { |
+ new TraceControllerImpl( |
+ MakeRequest<tracing::TraceController>(client_handle.Pass())); |
+ } |
+ } |
+ |
+ private: |
+ StrongBinding<ServiceProvider> binding_; |
+ |
+ DISALLOW_COPY_AND_ASSIGN(TracingServiceProvider); |
+}; |
+ |
+} // 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); |
+ |
+ base::FilePath cwd; |
+ PathService::Get(base::DIR_CURRENT, &cwd); |
+ SetCommandLineCWD(cwd); |
+} |
+ |
+Context::~Context() { |
+ DCHECK(!base::MessageLoop::current()); |
+} |
+ |
+// static |
+void Context::EnsureEmbedderIsInitialized() { |
+ static base::LazyInstance<Setup>::Leaky setup = LAZY_INSTANCE_INITIALIZER; |
+ 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); |
+} |
+ |
+void Context::SetCommandLineCWD(const base::FilePath& path) { |
+ command_line_cwd_ = AddTrailingSlashIfNeeded(FilePathToFileURL(path)); |
+} |
+ |
+GURL Context::ResolveCommandLineURL(const std::string& path) { |
+ return command_line_cwd_.Resolve(path); |
+} |
+ |
+bool Context::Init() { |
+ TRACE_EVENT0("mojo_shell", "Context::Init"); |
+ const 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())); |
+ |
+ // TODO(vtl): Probably these failures should be checked before |Init()|, and |
+ // this function simply shouldn't fail. |
+ if (!shell_file_root_.is_valid()) |
+ return false; |
+ if (!ConfigureURLMappings(command_line, this)) |
+ return false; |
+ |
+ // TODO(vtl): This should be MASTER, not NONE. |
+ embedder::InitIPCSupport( |
+ embedder::ProcessType::NONE, task_runners_->shell_runner(), this, |
+ task_runners_->io_runner(), embedder::ScopedPlatformHandle()); |
+ |
+ scoped_ptr<NativeRunnerFactory> runner_factory; |
+ if (command_line.HasSwitch(switches::kEnableMultiprocess)) |
+ runner_factory.reset(new OutOfProcessNativeRunnerFactory(this)); |
+ else |
+ runner_factory.reset(new InProcessNativeRunnerFactory(this)); |
+ application_manager_.set_blocking_pool(task_runners_->blocking_pool()); |
+ application_manager_.set_native_runner_factory(runner_factory.Pass()); |
+ application_manager_.set_disable_cache( |
+ base::CommandLine::ForCurrentProcess()->HasSwitch( |
+ switches::kDisableCache)); |
+ |
+ InitContentHandlers(&application_manager_, command_line); |
+ InitNativeOptions(&application_manager_, command_line); |
+ |
+ ServiceProviderPtr tracing_service_provider_ptr; |
+ new TracingServiceProvider(GetProxy(&tracing_service_provider_ptr)); |
+ application_manager_.ConnectToApplication( |
+ GURL("mojo:tracing"), GURL(""), nullptr, |
+ tracing_service_provider_ptr.Pass(), base::Closure()); |
+ |
+ return true; |
+} |
+ |
+void Context::Shutdown() { |
+ TRACE_EVENT0("mojo_shell", "Context::Shutdown"); |
+ DCHECK_EQ(base::MessageLoop::current()->task_runner(), |
+ task_runners_->shell_runner()); |
+ embedder::ShutdownIPCSupport(); |
+ // We'll quit when we get OnShutdownComplete(). |
+ base::MessageLoop::current()->Run(); |
+} |
+ |
+GURL Context::ResolveURL(const GURL& url) { |
+ return url_resolver_.ResolveMojoURL(url); |
+} |
+ |
+GURL Context::ResolveMappings(const GURL& url) { |
+ return url_resolver_.ApplyMappings(url); |
+} |
+ |
+void Context::OnShutdownComplete() { |
+ DCHECK_EQ(base::MessageLoop::current()->task_runner(), |
+ task_runners_->shell_runner()); |
+ base::MessageLoop::current()->Quit(); |
+} |
+ |
+void Context::Run(const GURL& url) { |
+ ServiceProviderPtr services; |
+ ServiceProviderPtr exposed_services; |
+ |
+ app_urls_.insert(url); |
+ application_manager_.ConnectToApplication( |
+ url, GURL(), GetProxy(&services), exposed_services.Pass(), |
+ base::Bind(&Context::OnApplicationEnd, base::Unretained(this), url)); |
+} |
+ |
+void Context::OnApplicationEnd(const GURL& url) { |
+ if (app_urls_.find(url) != app_urls_.end()) { |
+ app_urls_.erase(url); |
+ if (app_urls_.empty() && base::MessageLoop::current()->is_running()) { |
+ DCHECK_EQ(base::MessageLoop::current()->task_runner(), |
+ task_runners_->shell_runner()); |
+ base::MessageLoop::current()->Quit(); |
+ } |
+ } |
+} |
+ |
+} // namespace shell |
+} // namespace mojo |