Moved code implementing message-only windows to a dedicated class (Windows only).

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 <list>
 
 #include "base/memory/ref_counted.h"
 #include "base/memory/singleton.h"
 #include "base/synchronization/waitable_event.h"
 #include "net/base/ip_endpoint.h"
+#include "remoting/host/win/message_window.h"
 #include "remoting/host/win/wts_terminal_monitor.h"
 
 class CommandLine;
 
 namespace base {
 class SingleThreadTaskRunner;
 }  // namespace base
 
 namespace remoting {
 
 class AutoThreadTaskRunner;
 class Stoppable;
 class WtsTerminalObserver;
 
-class HostService : public WtsTerminalMonitor {
+class HostService : public win::MessageWindow::Delegate,
+                    public WtsTerminalMonitor {
  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();
 
   // WtsTerminalMonitor implementation
@@ skipping 20 matching lines @@
 
   // Runs the service on the service thread. A separate routine is used to make
   // sure all local objects are destoyed by the time |stopped_event_| is
   // signalled.
   void RunAsServiceImpl();
 
   // This function starts the service in interactive mode (i.e. as a plain
   // console application).
   int RunInConsole();
 
+  // win::MessageWindow::Delegate interface.
+  virtual bool HandleMessage(HWND hwnd,
+                             UINT message,
+                             WPARAM wparam,
+                             LPARAM lparam,
+                             LRESULT* result) OVERRIDE;
+
   static BOOL WINAPI ConsoleControlHandler(DWORD event);
 
   // The control handler of the service.
   static DWORD WINAPI ServiceControlHandler(DWORD control,
                                             DWORD event_type,
                                             LPVOID event_data,
                                             LPVOID context);
 
   // The main service entry point.
   static VOID WINAPI ServiceMain(DWORD argc, WCHAR* argv[]);
 
-  static LRESULT CALLBACK SessionChangeNotificationProc(HWND hwnd,
-                                                        UINT message,
-                                                        WPARAM wparam,
-                                                        LPARAM lparam);
-
   struct RegisteredObserver {
     // Specifies the client address of an RDP connection or IPEndPoint() for
     // the physical console.
     net::IPEndPoint client_endpoint;
 
     // Specifies ID of the attached session or |kInvalidSession| if no session
     // is attached to the WTS console.
     uint32 session_id;
 
     // Points to the observer receiving notifications about the WTS console
@@ skipping 20 matching lines @@
 
   // Singleton.
   friend struct DefaultSingletonTraits<HostService>;
 
   DISALLOW_COPY_AND_ASSIGN(HostService);
 };
 
 }  // namespace remoting
 
 #endif  // REMOTING_HOST_WIN_HOST_SERVICE_H_