Introducing remoting::Stoppable helper base class implementing asynchronous shutdown on a specific t

remoting/host/host_service_win.h

 // Copyright (c) 2012 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.
 
 #ifndef REMOTING_HOST_HOST_SERVICE_WIN_H_
 #define REMOTING_HOST_HOST_SERVICE_WIN_H_
 
 #include <windows.h>
 
+#include "base/memory/ref_counted.h"
 #include "base/file_path.h"
 #include "base/memory/singleton.h"
 #include "base/observer_list.h"
 #include "base/synchronization/waitable_event.h"
-
 #include "remoting/host/wts_console_monitor_win.h"
 
 class CommandLine;
 class MessageLoop;
 
 namespace remoting {
 
+class AutoMessageLoop;
+class Shutdownable;
 class WtsConsoleObserver;
+class WtsSessionProcessLauncher;
 
 class HostService : public WtsConsoleMonitor {
  public:
   static HostService* GetInstance();
 
   // This function parses the command line and selects the action routine.
   bool InitWithCommandLine(const CommandLine* command_line);
 
   // Invoke the choosen action routine.
   int Run();
 
   // WtsConsoleMonitor implementation
   virtual void AddWtsConsoleObserver(WtsConsoleObserver* observer) OVERRIDE;
   virtual void RemoveWtsConsoleObserver(
                    WtsConsoleObserver* observer) OVERRIDE;
 
  private:
   HostService();
   ~HostService();
 
+  void OnLauncherShutdown(Shutdownable* launcher);
+
   // Notifies the service of changes in session state.
   void OnSessionChange();
 
+  bool BeforeMessageLoop();
+
   // This is a common entry point to the main service loop called by both
   // RunAsService() and RunInConsole().
   void RunMessageLoop();
 
   // This function handshakes with the service control manager and starts
   // the service.
   int RunAsService();
 
   // This function starts the service in interactive mode (i.e. as a plain
   // console application).
@@ skipping 15 matching lines @@
                                                         WPARAM wparam,
                                                         LPARAM lparam);
 
   // Current physical console session id.
   uint32 console_session_id_;
 
   // The list of observers receiving notifications about any session attached
   // to the physical console.
   ObserverList<WtsConsoleObserver> console_observers_;
 
+  scoped_ptr<WtsSessionProcessLauncher> launcher_;
+
   // The host binary name.
   FilePath host_binary_;
 
   // Service message loop.
-  MessageLoop* message_loop_;
+  scoped_refptr<AutoMessageLoop> message_loop_;
 
   // The action routine to be executed.
   int (HostService::*run_routine_)();
 
   // The service name.
   std::wstring service_name_;
 
   // The service status handle.
   SERVICE_STATUS_HANDLE service_status_handle_;
 
-  // True if the service is being stopped.
-  bool shutting_down_;
-
   // A waitable event that is used to wait until the service is stopped.
   base::WaitableEvent stopped_event_;
 
   // Singleton.
   friend struct DefaultSingletonTraits<HostService>;
 
   DISALLOW_COPY_AND_ASSIGN(HostService);
 };
 
 }  // namespace remoting
 
 #endif  // REMOTING_HOST_HOST_SERVICE_WIN_H_