| OLD | NEW |
| (Empty) |
| 1 // Copyright 2016 The Chromium Authors. All rights reserved. | |
| 2 // Use of this source code is governed by a BSD-style license that can be | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 #include "mojo/shell/background/background_shell.h" | |
| 6 | |
| 7 #include "base/bind.h" | |
| 8 #include "base/bind_helpers.h" | |
| 9 #include "base/command_line.h" | |
| 10 #include "base/logging.h" | |
| 11 #include "base/path_service.h" | |
| 12 #include "base/synchronization/waitable_event.h" | |
| 13 #include "base/threading/simple_thread.h" | |
| 14 #include "mojo/message_pump/message_pump_mojo.h" | |
| 15 #include "mojo/services/catalog/store.h" | |
| 16 #include "mojo/shell/connect_params.h" | |
| 17 #include "mojo/shell/loader.h" | |
| 18 #include "mojo/shell/public/cpp/shell_client.h" | |
| 19 #include "mojo/shell/public/cpp/shell_connection.h" | |
| 20 #include "mojo/shell/shell.h" | |
| 21 #include "mojo/shell/standalone/context.h" | |
| 22 | |
| 23 namespace mojo { | |
| 24 namespace shell { | |
| 25 namespace { | |
| 26 | |
| 27 scoped_ptr<base::MessagePump> CreateMessagePumpMojo() { | |
| 28 return make_scoped_ptr(new common::MessagePumpMojo); | |
| 29 } | |
| 30 | |
| 31 // Used to obtain the ShellClientRequest for an application. When Loader::Load() | |
| 32 // is called a callback is run with the ShellClientRequest. | |
| 33 class BackgroundLoader : public Loader { | |
| 34 public: | |
| 35 using Callback = base::Callback<void(mojom::ShellClientRequest)>; | |
| 36 | |
| 37 explicit BackgroundLoader(const Callback& callback) : callback_(callback) {} | |
| 38 ~BackgroundLoader() override {} | |
| 39 | |
| 40 // Loader: | |
| 41 void Load(const std::string& name, | |
| 42 mojom::ShellClientRequest request) override { | |
| 43 DCHECK(!callback_.is_null()); // Callback should only be run once. | |
| 44 Callback callback = callback_; | |
| 45 callback_.Reset(); | |
| 46 callback.Run(std::move(request)); | |
| 47 } | |
| 48 | |
| 49 private: | |
| 50 Callback callback_; | |
| 51 | |
| 52 DISALLOW_COPY_AND_ASSIGN(BackgroundLoader); | |
| 53 }; | |
| 54 | |
| 55 class MojoMessageLoop : public base::MessageLoop { | |
| 56 public: | |
| 57 MojoMessageLoop() | |
| 58 : base::MessageLoop(base::MessageLoop::TYPE_CUSTOM, | |
| 59 base::Bind(&CreateMessagePumpMojo)) {} | |
| 60 ~MojoMessageLoop() override {} | |
| 61 | |
| 62 void BindToCurrentThread() { base::MessageLoop::BindToCurrentThread(); } | |
| 63 | |
| 64 private: | |
| 65 DISALLOW_COPY_AND_ASSIGN(MojoMessageLoop); | |
| 66 }; | |
| 67 | |
| 68 } // namespace | |
| 69 | |
| 70 // Manages the thread to startup mojo. | |
| 71 class BackgroundShell::MojoThread : public base::SimpleThread { | |
| 72 public: | |
| 73 explicit MojoThread(scoped_ptr<BackgroundShell::InitParams> init_params) | |
| 74 : SimpleThread("mojo-background-shell"), | |
| 75 init_params_(std::move(init_params)) {} | |
| 76 ~MojoThread() override {} | |
| 77 | |
| 78 void CreateShellClientRequest(base::WaitableEvent* signal, | |
| 79 scoped_ptr<ConnectParams> params, | |
| 80 mojom::ShellClientRequest* request) { | |
| 81 // Only valid to call this on the background thread. | |
| 82 DCHECK_EQ(message_loop_, base::MessageLoop::current()); | |
| 83 | |
| 84 // Ownership of |loader| passes to Shell. | |
| 85 const std::string name = params->target().name(); | |
| 86 BackgroundLoader* loader = new BackgroundLoader( | |
| 87 base::Bind(&MojoThread::OnGotApplicationRequest, base::Unretained(this), | |
| 88 name, signal, request)); | |
| 89 context_->shell()->SetLoaderForName(make_scoped_ptr(loader), name); | |
| 90 context_->shell()->Connect(std::move(params)); | |
| 91 // The request is asynchronously processed. When processed | |
| 92 // OnGotApplicationRequest() is called and we'll signal |signal|. | |
| 93 } | |
| 94 | |
| 95 base::MessageLoop* message_loop() { return message_loop_; } | |
| 96 | |
| 97 // Stops the background thread. | |
| 98 void Stop() { | |
| 99 DCHECK_NE(message_loop_, base::MessageLoop::current()); | |
| 100 message_loop_->task_runner()->PostTask( | |
| 101 FROM_HERE, base::MessageLoop::QuitWhenIdleClosure()); | |
| 102 Join(); | |
| 103 } | |
| 104 | |
| 105 void RunShellCallback(const BackgroundShell::ShellThreadCallback& callback) { | |
| 106 DCHECK_EQ(message_loop_, base::MessageLoop::current()); | |
| 107 callback.Run(context_->shell()); | |
| 108 } | |
| 109 | |
| 110 // base::SimpleThread: | |
| 111 void Start() override { | |
| 112 DCHECK(!message_loop_); | |
| 113 message_loop_ = new MojoMessageLoop; | |
| 114 base::SimpleThread::Start(); | |
| 115 } | |
| 116 void Run() override { | |
| 117 // The construction/destruction order is very finicky and has to be done | |
| 118 // in the order here. | |
| 119 scoped_ptr<base::MessageLoop> message_loop(message_loop_); | |
| 120 | |
| 121 scoped_ptr<mojo::shell::Context::InitParams> context_init_params( | |
| 122 new mojo::shell::Context::InitParams); | |
| 123 if (init_params_) { | |
| 124 context_init_params->catalog_store = | |
| 125 std::move(init_params_->catalog_store); | |
| 126 context_init_params->native_runner_delegate = | |
| 127 init_params_->native_runner_delegate; | |
| 128 context_init_params->init_edk = init_params_->init_edk; | |
| 129 } | |
| 130 if (context_init_params->init_edk) | |
| 131 Context::EnsureEmbedderIsInitialized(); | |
| 132 | |
| 133 message_loop_->BindToCurrentThread(); | |
| 134 | |
| 135 scoped_ptr<Context> context(new Context); | |
| 136 context_ = context.get(); | |
| 137 context_->Init(std::move(context_init_params)); | |
| 138 | |
| 139 message_loop_->Run(); | |
| 140 | |
| 141 // Has to happen after run, but while messageloop still valid. | |
| 142 context_->Shutdown(); | |
| 143 | |
| 144 // Context has to be destroyed after the MessageLoop has been destroyed. | |
| 145 message_loop.reset(); | |
| 146 context_ = nullptr; | |
| 147 } | |
| 148 | |
| 149 private: | |
| 150 void OnGotApplicationRequest(const std::string& name, | |
| 151 base::WaitableEvent* signal, | |
| 152 mojom::ShellClientRequest* request_result, | |
| 153 mojom::ShellClientRequest actual_request) { | |
| 154 *request_result = std::move(actual_request); | |
| 155 // Trigger destruction of the loader. | |
| 156 context_->shell()->SetLoaderForName(nullptr, name); | |
| 157 signal->Signal(); | |
| 158 } | |
| 159 | |
| 160 // We own this. It's created on the main thread, but destroyed on the | |
| 161 // background thread. | |
| 162 MojoMessageLoop* message_loop_ = nullptr; | |
| 163 // Created in Run() on the background thread. | |
| 164 Context* context_ = nullptr; | |
| 165 | |
| 166 scoped_ptr<BackgroundShell::InitParams> init_params_; | |
| 167 | |
| 168 DISALLOW_COPY_AND_ASSIGN(MojoThread); | |
| 169 }; | |
| 170 | |
| 171 BackgroundShell::InitParams::InitParams() {} | |
| 172 BackgroundShell::InitParams::~InitParams() {} | |
| 173 | |
| 174 BackgroundShell::BackgroundShell() {} | |
| 175 | |
| 176 BackgroundShell::~BackgroundShell() { | |
| 177 thread_->Stop(); | |
| 178 } | |
| 179 | |
| 180 void BackgroundShell::Init(scoped_ptr<InitParams> init_params) { | |
| 181 DCHECK(!thread_); | |
| 182 thread_.reset(new MojoThread(std::move(init_params))); | |
| 183 thread_->Start(); | |
| 184 } | |
| 185 | |
| 186 mojom::ShellClientRequest BackgroundShell::CreateShellClientRequest( | |
| 187 const std::string& name) { | |
| 188 scoped_ptr<ConnectParams> params(new ConnectParams); | |
| 189 params->set_source(CreateShellIdentity()); | |
| 190 params->set_target(Identity(name, mojom::kRootUserID)); | |
| 191 mojom::ShellClientRequest request; | |
| 192 base::WaitableEvent signal(true, false); | |
| 193 thread_->message_loop()->task_runner()->PostTask( | |
| 194 FROM_HERE, base::Bind(&MojoThread::CreateShellClientRequest, | |
| 195 base::Unretained(thread_.get()), &signal, | |
| 196 base::Passed(¶ms), &request)); | |
| 197 signal.Wait(); | |
| 198 return request; | |
| 199 } | |
| 200 | |
| 201 void BackgroundShell::ExecuteOnShellThread( | |
| 202 const ShellThreadCallback& callback) { | |
| 203 thread_->message_loop()->task_runner()->PostTask( | |
| 204 FROM_HERE, base::Bind(&MojoThread::RunShellCallback, | |
| 205 base::Unretained(thread_.get()), callback)); | |
| 206 } | |
| 207 | |
| 208 } // namespace shell | |
| 209 } // namespace mojo | |
| OLD | NEW |