2 files changed, 6 insertions, 0 deletions
diff --git a/ash/mus/sysui_application.cc b/ash/mus/sysui_application.cc
index 16585a3..15a5186 100644
--- a/ash/mus/sysui_application.cc
+++ b/ash/mus/sysui_application.cc
@@ -16,6 +16,7 @@
 #include "ash/shell_window_ids.h"
 #include "base/bind.h"
 #include "base/files/file_path.h"
+#include "base/message_loop/message_loop.h"
 #include "base/path_service.h"
 #include "base/threading/sequenced_worker_pool.h"
 #include "components/mus/public/cpp/property_type_converters.h"
@@ -288,5 +289,9 @@ bool SysUIApplication::AcceptConnection(mojo::Connection* connection) {
   return true;
 }
 
+void SysUIApplication::ShellConnectionLost() {
+  base::MessageLoop::current()->QuitWhenIdle();
+}
+
 }  // namespace sysui
 }  // namespace ash
diff --git a/ash/mus/sysui_application.h b/ash/mus/sysui_application.h
index dba0f5e..2535c5e0 100644
--- a/ash/mus/sysui_application.h
+++ b/ash/mus/sysui_application.h
@@ -26,6 +26,7 @@ class SysUIApplication : public mojo::ShellClient {
                   const mojo::Identity& identity,
                   uint32_t id) override;
   bool AcceptConnection(mojo::Connection* connection) override;
+  void ShellConnectionLost() override;
 
   mojo::TracingImpl tracing_;
   scoped_ptr<AshInit> ash_init_;