summaryrefslogtreecommitdiffstats
path: root/remoting/host/ipc_desktop_environment_factory.cc
blob: 927c1681b6ed2be177a21bfed6d6726ec01be529 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
// 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.

#include "remoting/host/ipc_desktop_environment_factory.h"

#include <utility>

#include "ipc/ipc_channel_proxy.h"
#include "base/platform_file.h"
#include "remoting/host/audio_capturer.h"
#include "remoting/host/chromoting_host.h"
#include "remoting/host/chromoting_host_context.h"
#include "remoting/host/chromoting_messages.h"
#include "remoting/host/desktop_session_connector.h"
#include "remoting/host/desktop_session_proxy.h"
#include "remoting/host/event_executor.h"
#include "remoting/host/ipc_desktop_environment.h"
#include "remoting/host/video_frame_capturer.h"

namespace remoting {

IpcDesktopEnvironmentFactory::IpcDesktopEnvironmentFactory(
    IPC::ChannelProxy* daemon_channel,
    scoped_refptr<base::SingleThreadTaskRunner> input_task_runner,
    scoped_refptr<base::SingleThreadTaskRunner> network_task_runner,
    scoped_refptr<base::SingleThreadTaskRunner> ui_task_runner,
    scoped_refptr<base::SingleThreadTaskRunner> video_capture_task_runner)
    : DesktopEnvironmentFactory(input_task_runner, ui_task_runner),
      daemon_channel_(daemon_channel),
      network_task_runner_(network_task_runner),
      video_capture_task_runner_(video_capture_task_runner),
      next_id_(0) {
}

IpcDesktopEnvironmentFactory::~IpcDesktopEnvironmentFactory() {
}

scoped_ptr<DesktopEnvironment> IpcDesktopEnvironmentFactory::Create(
    ClientSession* client) {
  DCHECK(network_task_runner_->BelongsToCurrentThread());

  scoped_refptr<DesktopSessionProxy> desktop_session_proxy(
      new DesktopSessionProxy(network_task_runner_,
                              video_capture_task_runner_));

  return scoped_ptr<DesktopEnvironment>(new IpcDesktopEnvironment(
      input_task_runner_, network_task_runner_, ui_task_runner_,
      this, desktop_session_proxy, client));
}

void IpcDesktopEnvironmentFactory::ConnectTerminal(
    IpcDesktopEnvironment* desktop_environment) {
  DCHECK(network_task_runner_->BelongsToCurrentThread());

  int id = next_id_++;
  bool inserted = active_connections_.insert(
      std::make_pair(id, desktop_environment)).second;
  CHECK(inserted);

  VLOG(1) << "Network: registered desktop environment " << id;
  daemon_channel_->Send(new ChromotingNetworkHostMsg_ConnectTerminal(id));
}

void IpcDesktopEnvironmentFactory::DisconnectTerminal(
    IpcDesktopEnvironment* desktop_environment) {
  DCHECK(network_task_runner_->BelongsToCurrentThread());

  ActiveConnectionsList::iterator i;
  for (i = active_connections_.begin(); i != active_connections_.end(); ++i) {
    if (i->second == desktop_environment)
      break;
  }

  if (i != active_connections_.end()) {
    int id = i->first;
    active_connections_.erase(i);

    VLOG(1) << "Network: unregistered desktop environment " << id;
    daemon_channel_->Send(new ChromotingNetworkHostMsg_DisconnectTerminal(id));
  }
}

void IpcDesktopEnvironmentFactory::OnDesktopSessionAgentAttached(
    int terminal_id,
    IPC::PlatformFileForTransit desktop_process,
    IPC::PlatformFileForTransit desktop_pipe) {
  if (!network_task_runner_->BelongsToCurrentThread()) {
    network_task_runner_->PostTask(FROM_HERE, base::Bind(
        &IpcDesktopEnvironmentFactory::OnDesktopSessionAgentAttached,
        base::Unretained(this), terminal_id, desktop_process, desktop_pipe));
    return;
  }

  ActiveConnectionsList::iterator i = active_connections_.find(terminal_id);
  if (i != active_connections_.end()) {
    i->second->OnDesktopSessionAgentAttached(desktop_process, desktop_pipe);
  } else {
#if defined(OS_POSIX)
    DCHECK(desktop_process.auto_close);
    DCHECK(desktop_pipe.auto_close);

    base::ClosePlatformFile(desktop_process.fd);
    base::ClosePlatformFile(desktop_pipe.fd);
#elif defined(OS_WIN)
    base::ClosePlatformFile(desktop_process);
#endif  // defined(OS_WIN)
  }
}

void IpcDesktopEnvironmentFactory::OnTerminalDisconnected(int terminal_id) {
  if (!network_task_runner_->BelongsToCurrentThread()) {
    network_task_runner_->PostTask(FROM_HERE, base::Bind(
        &IpcDesktopEnvironmentFactory::OnTerminalDisconnected,
        base::Unretained(this), terminal_id));
    return;
  }

  ActiveConnectionsList::iterator i = active_connections_.find(terminal_id);
  if (i != active_connections_.end()) {
    IpcDesktopEnvironment* desktop_environment = i->second;
    active_connections_.erase(i);

    // Disconnect the client for the given desktop environment.
    desktop_environment->DisconnectClient();
  }
}

}  // namespace remoting