| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #ifndef REMOTING_HOST_DAEMON_PROCESS_H_ | 5 #ifndef REMOTING_HOST_DAEMON_PROCESS_H_ |
| 6 #define REMOTING_HOST_DAEMON_PROCESS_H_ | 6 #define REMOTING_HOST_DAEMON_PROCESS_H_ |
| 7 | 7 |
| 8 #include <list> | 8 #include <list> |
| 9 | 9 |
| 10 #include "base/basictypes.h" | 10 #include "base/basictypes.h" |
| 11 #include "base/compiler_specific.h" | 11 #include "base/compiler_specific.h" |
| 12 #include "base/location.h" | 12 #include "base/location.h" |
| 13 #include "base/memory/ref_counted.h" | 13 #include "base/memory/ref_counted.h" |
| 14 #include "base/memory/scoped_ptr.h" | 14 #include "base/memory/scoped_ptr.h" |
| 15 #include "base/process.h" |
| 15 #include "ipc/ipc_channel.h" | 16 #include "ipc/ipc_channel.h" |
| 16 #include "ipc/ipc_channel_proxy.h" | 17 #include "ipc/ipc_channel_proxy.h" |
| 18 #include "ipc/ipc_platform_file.h" |
| 17 #include "remoting/base/stoppable.h" | 19 #include "remoting/base/stoppable.h" |
| 18 #include "remoting/host/config_file_watcher.h" | 20 #include "remoting/host/config_file_watcher.h" |
| 19 #include "remoting/host/worker_process_ipc_delegate.h" | 21 #include "remoting/host/worker_process_ipc_delegate.h" |
| 20 | 22 |
| 21 class FilePath; | 23 class FilePath; |
| 22 | 24 |
| 23 namespace base { | |
| 24 class SingleThreadTaskRunner; | |
| 25 } // namespace base | |
| 26 | |
| 27 namespace remoting { | 25 namespace remoting { |
| 28 | 26 |
| 27 class AutoThreadTaskRunner; |
| 29 class DesktopSession; | 28 class DesktopSession; |
| 30 | 29 |
| 31 // This class implements core of the daemon process. It manages the networking | 30 // This class implements core of the daemon process. It manages the networking |
| 32 // process running at lower privileges and maintains the list of desktop | 31 // process running at lower privileges and maintains the list of desktop |
| 33 // sessions. | 32 // sessions. |
| 34 class DaemonProcess | 33 class DaemonProcess |
| 35 : public Stoppable, | 34 : public Stoppable, |
| 36 public ConfigFileWatcher::Delegate, | 35 public ConfigFileWatcher::Delegate, |
| 37 public WorkerProcessIpcDelegate { | 36 public WorkerProcessIpcDelegate { |
| 38 public: | 37 public: |
| 39 typedef std::list<DesktopSession*> DesktopSessionList; | 38 typedef std::list<DesktopSession*> DesktopSessionList; |
| 40 | 39 |
| 41 virtual ~DaemonProcess(); | 40 virtual ~DaemonProcess(); |
| 42 | 41 |
| 43 // Creates a platform-specific implementation of the daemon process object | 42 // Creates a platform-specific implementation of the daemon process object |
| 44 // passing relevant task runners. Public methods of this class must be called | 43 // passing relevant task runners. Public methods of this class must be called |
| 45 // on the |caller_task_runner| thread. |io_task_runner| is used to handle IPC | 44 // on the |caller_task_runner| thread. |io_task_runner| is used to handle IPC |
| 46 // and background I/O tasks. | 45 // and background I/O tasks. |
| 47 static scoped_ptr<DaemonProcess> Create( | 46 static scoped_ptr<DaemonProcess> Create( |
| 48 scoped_refptr<base::SingleThreadTaskRunner> caller_task_runner, | 47 scoped_refptr<AutoThreadTaskRunner> caller_task_runner, |
| 49 scoped_refptr<base::SingleThreadTaskRunner> io_task_runner, | 48 scoped_refptr<AutoThreadTaskRunner> io_task_runner, |
| 50 const base::Closure& stopped_callback); | 49 const base::Closure& stopped_callback); |
| 51 | 50 |
| 52 // ConfigFileWatcher::Delegate | 51 // ConfigFileWatcher::Delegate |
| 53 virtual void OnConfigUpdated(const std::string& serialized_config) OVERRIDE; | 52 virtual void OnConfigUpdated(const std::string& serialized_config) OVERRIDE; |
| 54 virtual void OnConfigWatcherError() OVERRIDE; | 53 virtual void OnConfigWatcherError() OVERRIDE; |
| 55 | 54 |
| 56 // WorkerProcessIpcDelegate implementation. | 55 // WorkerProcessIpcDelegate implementation. |
| 57 virtual void OnChannelConnected(int32 peer_pid) OVERRIDE; | 56 virtual void OnChannelConnected(int32 peer_pid) OVERRIDE; |
| 58 virtual bool OnMessageReceived(const IPC::Message& message) OVERRIDE; | 57 virtual bool OnMessageReceived(const IPC::Message& message) OVERRIDE; |
| 59 virtual void OnPermanentError() OVERRIDE; | 58 virtual void OnPermanentError() OVERRIDE; |
| 60 | 59 |
| 61 // Sends an IPC message to the network process. The message will be dropped | 60 // Sends an IPC message to the network process. The message will be dropped |
| 62 // unless the network process is connected over the IPC channel. | 61 // unless the network process is connected over the IPC channel. |
| 63 virtual void SendToNetwork(IPC::Message* message) = 0; | 62 virtual void SendToNetwork(IPC::Message* message) = 0; |
| 64 | 63 |
| 64 // Called when a desktop integration process attaches to |terminal_id|. |
| 65 // |desktop_process| is a handle of the desktop integration process. |
| 66 // |desktop_pipe| specifies the client end of the desktop pipe. Returns true |
| 67 // on success, false otherwise. |
| 68 virtual bool OnDesktopSessionAgentAttached( |
| 69 int terminal_id, |
| 70 base::ProcessHandle desktop_process, |
| 71 IPC::PlatformFileForTransit desktop_pipe) = 0; |
| 72 |
| 65 // Closes the desktop session identified by |terminal_id|. | 73 // Closes the desktop session identified by |terminal_id|. |
| 66 void CloseDesktopSession(int terminal_id); | 74 void CloseDesktopSession(int terminal_id); |
| 67 | 75 |
| 68 protected: | 76 protected: |
| 69 DaemonProcess(scoped_refptr<base::SingleThreadTaskRunner> caller_task_runner, | 77 DaemonProcess(scoped_refptr<AutoThreadTaskRunner> caller_task_runner, |
| 70 scoped_refptr<base::SingleThreadTaskRunner> io_task_runner, | 78 scoped_refptr<AutoThreadTaskRunner> io_task_runner, |
| 71 const base::Closure& stopped_callback); | 79 const base::Closure& stopped_callback); |
| 72 | 80 |
| 73 // Creates a desktop session and assigns a unique ID to it. | 81 // Creates a desktop session and assigns a unique ID to it. |
| 74 void CreateDesktopSession(int terminal_id); | 82 void CreateDesktopSession(int terminal_id); |
| 75 | 83 |
| 76 // Requests the network process to crash. | 84 // Requests the network process to crash. |
| 77 void CrashNetworkProcess(const tracked_objects::Location& location); | 85 void CrashNetworkProcess(const tracked_objects::Location& location); |
| 78 | 86 |
| 79 // Reads the host configuration and launches the network process. | 87 // Reads the host configuration and launches the network process. |
| 80 void Initialize(); | 88 void Initialize(); |
| 81 | 89 |
| 82 // Returns true if |terminal_id| is considered to be known. I.e. it is | 90 // Returns true if |terminal_id| is considered to be known. I.e. it is |
| 83 // less or equal to the highest ID we have seen so far. | 91 // less or equal to the highest ID we have seen so far. |
| 84 bool IsTerminalIdKnown(int terminal_id); | 92 bool IsTerminalIdKnown(int terminal_id); |
| 85 | 93 |
| 86 // Stoppable implementation. | 94 // Stoppable implementation. |
| 87 virtual void DoStop() OVERRIDE; | 95 virtual void DoStop() OVERRIDE; |
| 88 | 96 |
| 89 // Creates a platform-specific desktop session and assigns a unique ID to it. | 97 // Creates a platform-specific desktop session and assigns a unique ID to it. |
| 90 virtual scoped_ptr<DesktopSession> DoCreateDesktopSession( | 98 virtual scoped_ptr<DesktopSession> DoCreateDesktopSession( |
| 91 int terminal_id) = 0; | 99 int terminal_id) = 0; |
| 92 | 100 |
| 93 // Launches the network process and establishes an IPC channel with it. | 101 // Launches the network process and establishes an IPC channel with it. |
| 94 virtual void LaunchNetworkProcess() = 0; | 102 virtual void LaunchNetworkProcess() = 0; |
| 95 | 103 |
| 96 scoped_refptr<base::SingleThreadTaskRunner> caller_task_runner() { | 104 scoped_refptr<AutoThreadTaskRunner> caller_task_runner() { |
| 97 return caller_task_runner_; | 105 return caller_task_runner_; |
| 98 } | 106 } |
| 99 | 107 |
| 100 scoped_refptr<base::SingleThreadTaskRunner> io_task_runner() { | 108 scoped_refptr<AutoThreadTaskRunner> io_task_runner() { |
| 101 return io_task_runner_; | 109 return io_task_runner_; |
| 102 } | 110 } |
| 103 | 111 |
| 104 // Let the test code analyze the list of desktop sessions. | 112 // Let the test code analyze the list of desktop sessions. |
| 105 friend class DaemonProcessTest; | 113 friend class DaemonProcessTest; |
| 106 const DesktopSessionList& desktop_sessions() const { | 114 const DesktopSessionList& desktop_sessions() const { |
| 107 return desktop_sessions_; | 115 return desktop_sessions_; |
| 108 } | 116 } |
| 109 | 117 |
| 110 private: | 118 private: |
| 111 // Deletes all desktop sessions. | 119 // Deletes all desktop sessions. |
| 112 void DeleteAllDesktopSessions(); | 120 void DeleteAllDesktopSessions(); |
| 113 | 121 |
| 114 // Task runner on which public methods of this class must be called. | 122 // Task runner on which public methods of this class must be called. |
| 115 scoped_refptr<base::SingleThreadTaskRunner> caller_task_runner_; | 123 scoped_refptr<AutoThreadTaskRunner> caller_task_runner_; |
| 116 | 124 |
| 117 // Handles IPC and background I/O tasks. | 125 // Handles IPC and background I/O tasks. |
| 118 scoped_refptr<base::SingleThreadTaskRunner> io_task_runner_; | 126 scoped_refptr<AutoThreadTaskRunner> io_task_runner_; |
| 119 | 127 |
| 120 scoped_ptr<ConfigFileWatcher> config_watcher_; | 128 scoped_ptr<ConfigFileWatcher> config_watcher_; |
| 121 | 129 |
| 122 // The configuration file contents. | 130 // The configuration file contents. |
| 123 std::string serialized_config_; | 131 std::string serialized_config_; |
| 124 | 132 |
| 125 // The list of active desktop sessions. | 133 // The list of active desktop sessions. |
| 126 DesktopSessionList desktop_sessions_; | 134 DesktopSessionList desktop_sessions_; |
| 127 | 135 |
| 128 // The highest desktop session ID that has been seen so far. | 136 // The highest desktop session ID that has been seen so far. |
| 129 int next_terminal_id_; | 137 int next_terminal_id_; |
| 130 | 138 |
| 131 DISALLOW_COPY_AND_ASSIGN(DaemonProcess); | 139 DISALLOW_COPY_AND_ASSIGN(DaemonProcess); |
| 132 }; | 140 }; |
| 133 | 141 |
| 134 } // namespace remoting | 142 } // namespace remoting |
| 135 | 143 |
| 136 #endif // REMOTING_HOST_DAEMON_PROCESS_H_ | 144 #endif // REMOTING_HOST_DAEMON_PROCESS_H_ |
| OLD | NEW |