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
|
// Copyright 2012 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "remoting/host/curtain_mode.h"
#include <windows.h>
#include "base/logging.h"
#include "base/memory/ptr_util.h"
#include "base/task/single_thread_task_runner.h"
#include "remoting/host/client_session_control.h"
namespace remoting {
class CurtainModeWin : public CurtainMode {
public:
CurtainModeWin();
CurtainModeWin(const CurtainModeWin&) = delete;
CurtainModeWin& operator=(const CurtainModeWin&) = delete;
// Overriden from CurtainMode.
bool Activate() override;
};
CurtainModeWin::CurtainModeWin() {}
bool CurtainModeWin::Activate() {
DWORD session_id;
if (!ProcessIdToSessionId(GetCurrentProcessId(), &session_id)) {
PLOG(ERROR) << "Failed to map the current PID to session ID";
return false;
}
// The current session is curtained if it is not attached to the local
// console.
return WTSGetActiveConsoleSessionId() != session_id;
}
// static
std::unique_ptr<CurtainMode> CurtainMode::Create(
scoped_refptr<base::SingleThreadTaskRunner> caller_task_runner,
scoped_refptr<base::SingleThreadTaskRunner> ui_task_runner,
base::WeakPtr<ClientSessionControl> client_session_control) {
// |client_session_control| is not used because the client session is
// disconnected as soon as the session is re-attached to the local console.
// See RdpDesktopSession for more details.
return base::WrapUnique(new CurtainModeWin());
}
} // namespace remoting
|
