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
3 // found in the LICENSE file.
4
5 #include "remoting/host/ipc_desktop_environment.h"
6
7 #include <utility>
8
9 #include "base/compiler_specific.h"
10 #include "base/logging.h"
11 #include "base/process/process_handle.h"
12 #include "base/single_thread_task_runner.h"
13 #include "ipc/ipc_sender.h"
14 #include "remoting/host/audio_capturer.h"
15 #include "remoting/host/chromoting_messages.h"
16 #include "remoting/host/client_session_control.h"
17 #include "remoting/host/desktop_session.h"
18 #include "remoting/host/desktop_session_proxy.h"
19 #include "remoting/host/input_injector.h"
20 #include "remoting/host/screen_controls.h"
21 #include "third_party/webrtc/modules/desktop_capture/screen_capturer.h"
22
23 namespace remoting {
24
IpcDesktopEnvironment(scoped_refptr<base::SingleThreadTaskRunner> audio_task_runner,scoped_refptr<base::SingleThreadTaskRunner> caller_task_runner,scoped_refptr<base::SingleThreadTaskRunner> capture_task_runner,scoped_refptr<base::SingleThreadTaskRunner> io_task_runner,base::WeakPtr<ClientSessionControl> client_session_control,base::WeakPtr<DesktopSessionConnector> desktop_session_connector,bool virtual_terminal)25 IpcDesktopEnvironment::IpcDesktopEnvironment(
26 scoped_refptr<base::SingleThreadTaskRunner> audio_task_runner,
27 scoped_refptr<base::SingleThreadTaskRunner> caller_task_runner,
28 scoped_refptr<base::SingleThreadTaskRunner> capture_task_runner,
29 scoped_refptr<base::SingleThreadTaskRunner> io_task_runner,
30 base::WeakPtr<ClientSessionControl> client_session_control,
31 base::WeakPtr<DesktopSessionConnector> desktop_session_connector,
32 bool virtual_terminal) {
33 DCHECK(caller_task_runner->BelongsToCurrentThread());
34
35 desktop_session_proxy_ = new DesktopSessionProxy(audio_task_runner,
36 caller_task_runner,
37 io_task_runner,
38 capture_task_runner,
39 client_session_control,
40 desktop_session_connector,
41 virtual_terminal);
42 }
43
~IpcDesktopEnvironment()44 IpcDesktopEnvironment::~IpcDesktopEnvironment() {
45 }
46
CreateAudioCapturer()47 scoped_ptr<AudioCapturer> IpcDesktopEnvironment::CreateAudioCapturer() {
48 return desktop_session_proxy_->CreateAudioCapturer();
49 }
50
CreateInputInjector()51 scoped_ptr<InputInjector> IpcDesktopEnvironment::CreateInputInjector() {
52 return desktop_session_proxy_->CreateInputInjector();
53 }
54
CreateScreenControls()55 scoped_ptr<ScreenControls> IpcDesktopEnvironment::CreateScreenControls() {
56 return desktop_session_proxy_->CreateScreenControls();
57 }
58
59 scoped_ptr<webrtc::ScreenCapturer>
CreateVideoCapturer()60 IpcDesktopEnvironment::CreateVideoCapturer() {
61 return desktop_session_proxy_->CreateVideoCapturer();
62 }
63
GetCapabilities() const64 std::string IpcDesktopEnvironment::GetCapabilities() const {
65 return desktop_session_proxy_->GetCapabilities();
66 }
67
SetCapabilities(const std::string & capabilities)68 void IpcDesktopEnvironment::SetCapabilities(const std::string& capabilities) {
69 return desktop_session_proxy_->SetCapabilities(capabilities);
70 }
71
IpcDesktopEnvironmentFactory(scoped_refptr<base::SingleThreadTaskRunner> audio_task_runner,scoped_refptr<base::SingleThreadTaskRunner> caller_task_runner,scoped_refptr<base::SingleThreadTaskRunner> capture_task_runner,scoped_refptr<base::SingleThreadTaskRunner> io_task_runner,IPC::Sender * daemon_channel)72 IpcDesktopEnvironmentFactory::IpcDesktopEnvironmentFactory(
73 scoped_refptr<base::SingleThreadTaskRunner> audio_task_runner,
74 scoped_refptr<base::SingleThreadTaskRunner> caller_task_runner,
75 scoped_refptr<base::SingleThreadTaskRunner> capture_task_runner,
76 scoped_refptr<base::SingleThreadTaskRunner> io_task_runner,
77 IPC::Sender* daemon_channel)
78 : audio_task_runner_(audio_task_runner),
79 caller_task_runner_(caller_task_runner),
80 capture_task_runner_(capture_task_runner),
81 io_task_runner_(io_task_runner),
82 curtain_enabled_(false),
83 daemon_channel_(daemon_channel),
84 connector_factory_(this),
85 next_id_(0) {
86 }
87
~IpcDesktopEnvironmentFactory()88 IpcDesktopEnvironmentFactory::~IpcDesktopEnvironmentFactory() {
89 }
90
Create(base::WeakPtr<ClientSessionControl> client_session_control)91 scoped_ptr<DesktopEnvironment> IpcDesktopEnvironmentFactory::Create(
92 base::WeakPtr<ClientSessionControl> client_session_control) {
93 DCHECK(caller_task_runner_->BelongsToCurrentThread());
94
95 return scoped_ptr<DesktopEnvironment>(
96 new IpcDesktopEnvironment(audio_task_runner_,
97 caller_task_runner_,
98 capture_task_runner_,
99 io_task_runner_,
100 client_session_control,
101 connector_factory_.GetWeakPtr(),
102 curtain_enabled_));
103 }
104
SetEnableCurtaining(bool enable)105 void IpcDesktopEnvironmentFactory::SetEnableCurtaining(bool enable) {
106 DCHECK(caller_task_runner_->BelongsToCurrentThread());
107
108 curtain_enabled_ = enable;
109 }
110
SupportsAudioCapture() const111 bool IpcDesktopEnvironmentFactory::SupportsAudioCapture() const {
112 DCHECK(caller_task_runner_->BelongsToCurrentThread());
113
114 return AudioCapturer::IsSupported();
115 }
116
ConnectTerminal(DesktopSessionProxy * desktop_session_proxy,const ScreenResolution & resolution,bool virtual_terminal)117 void IpcDesktopEnvironmentFactory::ConnectTerminal(
118 DesktopSessionProxy* desktop_session_proxy,
119 const ScreenResolution& resolution,
120 bool virtual_terminal) {
121 DCHECK(caller_task_runner_->BelongsToCurrentThread());
122
123 int id = next_id_++;
124 bool inserted = active_connections_.insert(
125 std::make_pair(id, desktop_session_proxy)).second;
126 CHECK(inserted);
127
128 VLOG(1) << "Network: registered desktop environment " << id;
129
130 daemon_channel_->Send(new ChromotingNetworkHostMsg_ConnectTerminal(
131 id, resolution, virtual_terminal));
132 }
133
DisconnectTerminal(DesktopSessionProxy * desktop_session_proxy)134 void IpcDesktopEnvironmentFactory::DisconnectTerminal(
135 DesktopSessionProxy* desktop_session_proxy) {
136 DCHECK(caller_task_runner_->BelongsToCurrentThread());
137
138 ActiveConnectionsList::iterator i;
139 for (i = active_connections_.begin(); i != active_connections_.end(); ++i) {
140 if (i->second == desktop_session_proxy)
141 break;
142 }
143
144 if (i != active_connections_.end()) {
145 int id = i->first;
146 active_connections_.erase(i);
147
148 VLOG(1) << "Network: unregistered desktop environment " << id;
149 daemon_channel_->Send(new ChromotingNetworkHostMsg_DisconnectTerminal(id));
150 }
151 }
152
SetScreenResolution(DesktopSessionProxy * desktop_session_proxy,const ScreenResolution & resolution)153 void IpcDesktopEnvironmentFactory::SetScreenResolution(
154 DesktopSessionProxy* desktop_session_proxy,
155 const ScreenResolution& resolution) {
156 DCHECK(caller_task_runner_->BelongsToCurrentThread());
157
158 ActiveConnectionsList::iterator i;
159 for (i = active_connections_.begin(); i != active_connections_.end(); ++i) {
160 if (i->second == desktop_session_proxy)
161 break;
162 }
163
164 if (i != active_connections_.end()) {
165 daemon_channel_->Send(new ChromotingNetworkDaemonMsg_SetScreenResolution(
166 i->first, resolution));
167 }
168 }
169
OnDesktopSessionAgentAttached(int terminal_id,base::ProcessHandle desktop_process,IPC::PlatformFileForTransit desktop_pipe)170 void IpcDesktopEnvironmentFactory::OnDesktopSessionAgentAttached(
171 int terminal_id,
172 base::ProcessHandle desktop_process,
173 IPC::PlatformFileForTransit desktop_pipe) {
174 if (!caller_task_runner_->BelongsToCurrentThread()) {
175 caller_task_runner_->PostTask(FROM_HERE, base::Bind(
176 &IpcDesktopEnvironmentFactory::OnDesktopSessionAgentAttached,
177 base::Unretained(this), terminal_id, desktop_process, desktop_pipe));
178 return;
179 }
180
181 ActiveConnectionsList::iterator i = active_connections_.find(terminal_id);
182 if (i != active_connections_.end()) {
183 i->second->DetachFromDesktop();
184 i->second->AttachToDesktop(desktop_process, desktop_pipe);
185 } else {
186 base::CloseProcessHandle(desktop_process);
187
188 #if defined(OS_POSIX)
189 DCHECK(desktop_pipe.auto_close);
190
191 base::ClosePlatformFile(desktop_pipe.fd);
192 #endif // defined(OS_POSIX)
193 }
194 }
195
OnTerminalDisconnected(int terminal_id)196 void IpcDesktopEnvironmentFactory::OnTerminalDisconnected(int terminal_id) {
197 if (!caller_task_runner_->BelongsToCurrentThread()) {
198 caller_task_runner_->PostTask(FROM_HERE, base::Bind(
199 &IpcDesktopEnvironmentFactory::OnTerminalDisconnected,
200 base::Unretained(this), terminal_id));
201 return;
202 }
203
204 ActiveConnectionsList::iterator i = active_connections_.find(terminal_id);
205 if (i != active_connections_.end()) {
206 DesktopSessionProxy* desktop_session_proxy = i->second;
207 active_connections_.erase(i);
208
209 // Disconnect the client session.
210 desktop_session_proxy->DisconnectSession();
211 }
212 }
213
214 } // namespace remoting
215