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
|
// Copyright 2018 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "content/browser/screenlock_monitor/screenlock_monitor_device_source.h"
#include <memory>
#include "base/functional/bind.h"
#include "base/functional/callback_helpers.h"
#include "base/logging.h"
#include "base/task/thread_pool.h"
#include "ui/gfx/win/singleton_hwnd.h"
#include "ui/gfx/win/singleton_hwnd_observer.h"
namespace content {
namespace {
bool RegisterForSessionNotifications(HWND hwnd, DWORD flag) {
base::OnceClosure wts_register = base::BindOnce(
base::IgnoreResult(&::WTSRegisterSessionNotification), hwnd, flag);
base::ThreadPool::CreateCOMSTATaskRunner({})->PostTask(
FROM_HERE, std::move(wts_register));
return true;
}
bool UnregisterFromSessionNotifications(HWND hwnd) {
return ::WTSUnRegisterSessionNotification(hwnd);
}
} // namespace
// Set static member function pointers with default (non-fake) APIs.
ScreenlockMonitorDeviceSource::WTSRegisterSessionNotificationFunction
ScreenlockMonitorDeviceSource::SessionMessageWindow::
register_session_notification_function_ =
&RegisterForSessionNotifications;
ScreenlockMonitorDeviceSource::WTSUnRegisterSessionNotificationFunction
ScreenlockMonitorDeviceSource::SessionMessageWindow::
unregister_session_notification_function_ =
&UnregisterFromSessionNotifications;
// static
void ScreenlockMonitorDeviceSource::SetFakeNotificationAPIsForTesting(
WTSRegisterSessionNotificationFunction register_function,
WTSUnRegisterSessionNotificationFunction unregister_function) {
SessionMessageWindow::SetFakeNotificationAPIsForTesting(register_function,
unregister_function);
}
// static
void ScreenlockMonitorDeviceSource::SessionMessageWindow::
SetFakeNotificationAPIsForTesting(
WTSRegisterSessionNotificationFunction register_function,
WTSUnRegisterSessionNotificationFunction unregister_function) {
register_session_notification_function_ = register_function;
unregister_session_notification_function_ = unregister_function;
}
ScreenlockMonitorDeviceSource::SessionMessageWindow::SessionMessageWindow() {
// Create a singleton observer for receiving session change notifications.
// base:Unretained() is safe because the observer handles the correct
// cleanup if either the SingletonHwnd or forwarded object is destroyed
// first.
singleton_hwnd_observer_ =
std::make_unique<gfx::SingletonHwndObserver>(base::BindRepeating(
&ScreenlockMonitorDeviceSource::SessionMessageWindow::OnWndProc,
base::Unretained(this)));
// Use NOTIFY_FOR_THIS_SESSION so we only receive events from the current
// session, and not from other users connected to the same session host.
bool registered = register_session_notification_function_(
gfx::SingletonHwnd::GetInstance()->hwnd(), NOTIFY_FOR_THIS_SESSION);
DCHECK(registered);
}
ScreenlockMonitorDeviceSource::SessionMessageWindow::~SessionMessageWindow() {}
void ScreenlockMonitorDeviceSource::SessionMessageWindow::OnWndProc(
HWND hwnd,
UINT message,
WPARAM wparam,
LPARAM lparam) {
if (message == WM_WTSSESSION_CHANGE) {
ProcessWTSSessionLockMessage(wparam);
}
}
void ScreenlockMonitorDeviceSource::SessionMessageWindow::
ProcessWTSSessionLockMessage(WPARAM event_id) {
ScreenlockEvent screenlock_event;
switch (event_id) {
case WTS_SESSION_LOCK:
screenlock_event = SCREEN_LOCK_EVENT;
break;
case WTS_SESSION_UNLOCK:
screenlock_event = SCREEN_UNLOCK_EVENT;
break;
default:
return;
}
ProcessScreenlockEvent(screenlock_event);
}
} // namespace content
|
