[Chromoting] Introducing refcount-based life time management of the message loops in the service (d…

remoting/host/win/host_service.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_WIN_HOST_SERVICE_H_
 #define REMOTING_HOST_WIN_HOST_SERVICE_H_
 
 #include <windows.h>
 
 #include "base/memory/ref_counted.h"
 #include "base/memory/singleton.h"
 #include "base/observer_list.h"
 #include "base/synchronization/waitable_event.h"
 #include "remoting/host/win/wts_console_monitor.h"
 
 class CommandLine;
 class MessageLoop;
 
 namespace base {
 class SingleThreadTaskRunner;
 }  // namespace base
 
 namespace remoting {
 
-#if defined(REMOTING_MULTI_PROCESS)
-class DaemonProcess;
-#endif  // defined(REMOTING_MULTI_PROCESS)
-
+class AutoThreadTaskRunner;
 class Stoppable;
 class WtsConsoleObserver;
 
-#if !defined(REMOTING_MULTI_PROCESS)
+#if defined(REMOTING_MULTI_PROCESS)
+class DaemonProcess;
+#else  // !defined(REMOTING_MULTI_PROCESS)
 class WtsSessionProcessLauncher;
 #endif  // !defined(REMOTING_MULTI_PROCESS)
 
 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 OnChildStopped();
 
   // Notifies the service of changes in session state.
   void OnSessionChange();
 
+  // Creates the process launcher.
+  void CreateLauncher(
+    scoped_refptr<base::SingleThreadTaskRunner> io_task_runner);
+
   // This is a common entry point to the main service loop called by both
   // RunAsService() and RunInConsole().
   void RunMessageLoop(MessageLoop* message_loop);
 
   // Runs the binary specified by the command line, elevated.
   int Elevate();
 
   // This function handshakes with the service control manager and starts
   // the service.
   int RunAsService();
@@ skipping 21 matching lines @@
   // 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<Stoppable> child_;
 
   // Service message loop.
-  scoped_refptr<base::SingleThreadTaskRunner> main_task_runner_;
+  scoped_refptr<AutoThreadTaskRunner> main_task_runner_;
 
   // The action routine to be executed.
   int (HostService::*run_routine_)();
 
   // The service status handle.
   SERVICE_STATUS_HANDLE service_status_handle_;
 
   // 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_WIN_HOST_SERVICE_H_