[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 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 object to be run on the main message loop right before entering
+  // the loop.
+  bool BeforeMessageLoop();
+
   // 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 41 matching lines @@
 
   // Singleton.
   friend struct DefaultSingletonTraits<HostService>;
 
   DISALLOW_COPY_AND_ASSIGN(HostService);
 };
 
 }  // namespace remoting
 
 #endif  // REMOTING_HOST_WIN_HOST_SERVICE_H_