Index: mojo/application/public/cpp/lib/application_runner.cc |
diff --git a/mojo/application/public/cpp/lib/application_runner.cc b/mojo/application/public/cpp/lib/application_runner.cc |
index c1d4c0a361c6e17d64e90081ce60100cc35dd72f..a31bc4316be37342df758fc1bb5b5d1120596c04 100644 |
--- a/mojo/application/public/cpp/lib/application_runner.cc |
+++ b/mojo/application/public/cpp/lib/application_runner.cc |
@@ -4,36 +4,77 @@ |
#include "mojo/application/public/cpp/application_runner.h" |
+#include "base/at_exit.h" |
+#include "base/command_line.h" |
+#include "base/debug/stack_trace.h" |
+#include "base/memory/scoped_ptr.h" |
+#include "base/message_loop/message_loop.h" |
#include "mojo/application/public/cpp/application_delegate.h" |
#include "mojo/application/public/cpp/application_impl.h" |
-#include "mojo/public/cpp/environment/environment.h" |
-#include "mojo/public/cpp/utility/run_loop.h" |
+#include "mojo/common/message_pump_mojo.h" |
+ |
+int g_argc; |
+const char* const* g_argv; |
+#if !defined(OS_WIN) |
+extern "C" { |
+__attribute__((visibility("default"))) void InitCommandLineArgs( |
+ int argc, const char* const* argv) { |
+ g_argc = argc; |
+ g_argv = argv; |
+} |
+} |
+#endif |
namespace mojo { |
// static |
void ApplicationImpl::Terminate() { |
- RunLoop::current()->Quit(); |
+ if (base::MessageLoop::current()->is_running()) |
+ base::MessageLoop::current()->Quit(); |
} |
ApplicationRunner::ApplicationRunner(ApplicationDelegate* delegate) |
- : delegate_(delegate) { |
+ : delegate_(scoped_ptr<ApplicationDelegate>(delegate)), |
+ message_loop_type_(base::MessageLoop::TYPE_CUSTOM), |
+ has_run_(false) {} |
+ |
+ApplicationRunner::~ApplicationRunner() {} |
+ |
+void ApplicationRunner::InitBaseCommandLine() { |
+ base::CommandLine::Init(g_argc, g_argv); |
} |
-ApplicationRunner::~ApplicationRunner() { |
- assert(!delegate_); |
+ |
+void ApplicationRunner::set_message_loop_type(base::MessageLoop::Type type) { |
+ DCHECK_NE(base::MessageLoop::TYPE_CUSTOM, type); |
+ DCHECK(!has_run_); |
+ |
+ message_loop_type_ = type; |
} |
-MojoResult ApplicationRunner::Run(MojoHandle app_request_handle) { |
- Environment env; |
+MojoResult ApplicationRunner::Run(MojoHandle application_request_handle) { |
+ DCHECK(!has_run_); |
+ has_run_ = true; |
+ |
+ InitBaseCommandLine(); |
+ base::AtExitManager at_exit; |
+ |
+#ifndef NDEBUG |
+ base::debug::EnableInProcessStackDumping(); |
+#endif |
+ |
{ |
- RunLoop loop; |
- ApplicationImpl app(delegate_, MakeRequest<Application>(MakeScopedHandle( |
- MessagePipeHandle(app_request_handle)))); |
- loop.Run(); |
- } |
+ scoped_ptr<base::MessageLoop> loop; |
+ if (message_loop_type_ == base::MessageLoop::TYPE_CUSTOM) |
+ loop.reset(new base::MessageLoop(common::MessagePumpMojo::Create())); |
+ else |
+ loop.reset(new base::MessageLoop(message_loop_type_)); |
- delete delegate_; |
- delegate_ = nullptr; |
+ ApplicationImpl impl(delegate_.get(), |
+ MakeRequest<Application>(MakeScopedHandle( |
+ MessagePipeHandle(application_request_handle)))); |
+ loop->Run(); |
+ } |
+ delegate_.reset(); |
return MOJO_RESULT_OK; |
} |