forked from chromium/chromium
-
Notifications
You must be signed in to change notification settings - Fork 0
/
media_keyboard_hook_win.cc
139 lines (110 loc) · 4.29 KB
/
media_keyboard_hook_win.cc
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
130
131
132
133
134
135
136
137
138
139
// Copyright 2018 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 "ui/events/win/keyboard_hook_win_base.h"
#include "ui/events/event.h"
#include "ui/events/keycodes/dom/dom_code.h"
#include "ui/events/win/events_win_utils.h"
namespace ui {
namespace {
bool IsMediaKey(DWORD vk) {
return vk == VK_MEDIA_NEXT_TRACK || vk == VK_MEDIA_PREV_TRACK ||
vk == VK_MEDIA_PLAY_PAUSE || vk == VK_MEDIA_STOP;
}
class MediaKeyboardHookWinImpl : public KeyboardHookWinBase {
public:
MediaKeyboardHookWinImpl(KeyEventCallback callback,
bool enable_hook_registration);
~MediaKeyboardHookWinImpl() override;
// KeyboardHookWinBase implementation.
bool ProcessKeyEventMessage(WPARAM w_param,
DWORD vk,
DWORD scan_code,
DWORD time_stamp) override;
bool Register();
private:
static LRESULT CALLBACK ProcessKeyEvent(int code,
WPARAM w_param,
LPARAM l_param);
static MediaKeyboardHookWinImpl* instance_;
// Tracks the last non-located key down seen in order to determine if the
// current key event should be marked as a repeated key press.
DWORD last_key_down_ = 0;
DISALLOW_COPY_AND_ASSIGN(MediaKeyboardHookWinImpl);
};
// static
MediaKeyboardHookWinImpl* MediaKeyboardHookWinImpl::instance_ = nullptr;
MediaKeyboardHookWinImpl::MediaKeyboardHookWinImpl(
KeyEventCallback callback,
bool enable_hook_registration)
: KeyboardHookWinBase(
base::Optional<base::flat_set<DomCode>>(
{DomCode::MEDIA_PLAY_PAUSE, DomCode::MEDIA_STOP,
DomCode::MEDIA_TRACK_NEXT, DomCode::MEDIA_TRACK_PREVIOUS}),
std::move(callback),
enable_hook_registration) {}
MediaKeyboardHookWinImpl::~MediaKeyboardHookWinImpl() {
if (!enable_hook_registration())
return;
DCHECK_EQ(instance_, this);
instance_ = nullptr;
}
bool MediaKeyboardHookWinImpl::Register() {
// Only one instance of this class can be registered at a time.
DCHECK(!instance_);
instance_ = this;
return KeyboardHookWinBase::Register(
reinterpret_cast<HOOKPROC>(&MediaKeyboardHookWinImpl::ProcessKeyEvent));
}
// static
LRESULT CALLBACK MediaKeyboardHookWinImpl::ProcessKeyEvent(int code,
WPARAM w_param,
LPARAM l_param) {
return KeyboardHookWinBase::ProcessKeyEvent(instance_, code, w_param,
l_param);
}
bool MediaKeyboardHookWinImpl::ProcessKeyEventMessage(WPARAM w_param,
DWORD vk,
DWORD scan_code,
DWORD time_stamp) {
if (!IsMediaKey(vk))
return false;
bool is_repeat = false;
MSG msg = {nullptr, w_param, vk, GetLParamFromScanCode(scan_code),
time_stamp};
EventType event_type = EventTypeFromMSG(msg);
if (event_type == ET_KEY_PRESSED) {
is_repeat = (last_key_down_ == vk);
last_key_down_ = vk;
} else {
DCHECK_EQ(event_type, ET_KEY_RELEASED);
last_key_down_ = 0;
}
std::unique_ptr<KeyEvent> key_event =
std::make_unique<KeyEvent>(KeyEventFromMSG(msg));
if (is_repeat)
key_event->set_flags(key_event->flags() | EF_IS_REPEAT);
ForwardCapturedKeyEvent(key_event.get());
// If the event is handled, don't propagate to the OS.
return key_event->handled();
}
} // namespace
// static
std::unique_ptr<KeyboardHook> KeyboardHook::CreateMediaKeyboardHook(
KeyEventCallback callback) {
std::unique_ptr<MediaKeyboardHookWinImpl> keyboard_hook =
std::make_unique<MediaKeyboardHookWinImpl>(
std::move(callback),
/*enable_hook_registration=*/true);
if (!keyboard_hook->Register())
return nullptr;
return keyboard_hook;
}
std::unique_ptr<KeyboardHookWinBase>
KeyboardHookWinBase::CreateMediaKeyboardHookForTesting(
KeyEventCallback callback) {
return std::make_unique<MediaKeyboardHookWinImpl>(
std::move(callback),
/*enable_hook_registration=*/false);
}
} // namespace ui