local_hotkey_input_monitor_mac.mm 6.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240
  1. // Copyright 2018 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. #include "remoting/host/input_monitor/local_hotkey_input_monitor.h"
  5. #include "base/memory/raw_ptr.h"
  6. #import <AppKit/AppKit.h>
  7. #include <cstdint>
  8. #include <utility>
  9. #include "base/bind.h"
  10. #include "base/callback.h"
  11. #include "base/compiler_specific.h"
  12. #include "base/location.h"
  13. #include "base/logging.h"
  14. #include "base/mac/scoped_cftyperef.h"
  15. #include "base/memory/ptr_util.h"
  16. #include "base/memory/ref_counted.h"
  17. #include "base/sequence_checker.h"
  18. #include "base/synchronization/lock.h"
  19. #include "base/task/single_thread_task_runner.h"
  20. #import "third_party/google_toolbox_for_mac/src/AppKit/GTMCarbonEvent.h"
  21. #include "third_party/webrtc/modules/desktop_capture/desktop_geometry.h"
  22. // Esc Key Code is 53.
  23. // http://boredzo.org/blog/wp-content/uploads/2007/05/IMTx-virtual-keycodes.pdf
  24. static const NSUInteger kEscKeyCode = 53;
  25. namespace remoting {
  26. namespace {
  27. class LocalHotkeyInputMonitorMac : public LocalHotkeyInputMonitor {
  28. public:
  29. // Invoked by LocalHotkeyInputMonitorManager.
  30. class EventHandler {
  31. public:
  32. virtual ~EventHandler() = default;
  33. virtual void OnDisconnectShortcut() = 0;
  34. };
  35. LocalHotkeyInputMonitorMac(
  36. scoped_refptr<base::SingleThreadTaskRunner> caller_task_runner,
  37. scoped_refptr<base::SingleThreadTaskRunner> ui_task_runner,
  38. base::OnceClosure disconnect_callback);
  39. LocalHotkeyInputMonitorMac(const LocalHotkeyInputMonitorMac&) = delete;
  40. LocalHotkeyInputMonitorMac& operator=(const LocalHotkeyInputMonitorMac&) =
  41. delete;
  42. ~LocalHotkeyInputMonitorMac() override;
  43. private:
  44. // The implementation resides in LocalHotkeyInputMonitorMac::Core class.
  45. class Core;
  46. scoped_refptr<Core> core_;
  47. SEQUENCE_CHECKER(sequence_checker_);
  48. };
  49. } // namespace
  50. } // namespace remoting
  51. @interface LocalHotkeyInputMonitorManager : NSObject {
  52. @private
  53. GTMCarbonHotKey* _hotKey;
  54. raw_ptr<remoting::LocalHotkeyInputMonitorMac::EventHandler> _monitor;
  55. }
  56. - (instancetype)initWithMonitor:
  57. (remoting::LocalHotkeyInputMonitorMac::EventHandler*)monitor;
  58. // Called when the hotKey is hit.
  59. - (void)hotKeyHit:(GTMCarbonHotKey*)hotKey;
  60. // Must be called when the LocalHotkeyInputMonitorManager is no longer needed.
  61. // Similar to NSTimer in that more than a simple release is required.
  62. - (void)invalidate;
  63. @end
  64. @implementation LocalHotkeyInputMonitorManager
  65. - (instancetype)initWithMonitor:
  66. (remoting::LocalHotkeyInputMonitorMac::EventHandler*)monitor {
  67. if ((self = [super init])) {
  68. _monitor = monitor;
  69. GTMCarbonEventDispatcherHandler* handler =
  70. [GTMCarbonEventDispatcherHandler sharedEventDispatcherHandler];
  71. _hotKey = [handler
  72. registerHotKey:kEscKeyCode
  73. modifiers:(NSEventModifierFlagOption | NSEventModifierFlagControl)
  74. target:self
  75. action:@selector(hotKeyHit:)
  76. userInfo:nil
  77. whenPressed:YES];
  78. if (!_hotKey) {
  79. LOG(ERROR) << "registerHotKey failed.";
  80. [self release];
  81. return nil;
  82. }
  83. }
  84. return self;
  85. }
  86. - (void)hotKeyHit:(GTMCarbonHotKey*)hotKey {
  87. _monitor->OnDisconnectShortcut();
  88. }
  89. - (void)invalidate {
  90. if (_hotKey) {
  91. GTMCarbonEventDispatcherHandler* handler =
  92. [GTMCarbonEventDispatcherHandler sharedEventDispatcherHandler];
  93. [handler unregisterHotKey:_hotKey];
  94. _hotKey = nullptr;
  95. }
  96. }
  97. @end
  98. namespace remoting {
  99. namespace {
  100. class LocalHotkeyInputMonitorMac::Core
  101. : public base::RefCountedThreadSafe<Core>,
  102. public EventHandler {
  103. public:
  104. Core(scoped_refptr<base::SingleThreadTaskRunner> caller_task_runner,
  105. scoped_refptr<base::SingleThreadTaskRunner> ui_task_runner,
  106. base::OnceClosure disconnect_callback);
  107. Core(const Core&) = delete;
  108. Core& operator=(const Core&) = delete;
  109. void Start();
  110. void Stop();
  111. private:
  112. friend class base::RefCountedThreadSafe<Core>;
  113. ~Core() override;
  114. void StartOnUiThread();
  115. void StopOnUiThread();
  116. // EventHandler interface.
  117. void OnDisconnectShortcut() override;
  118. // Task runner on which public methods of this class must be called.
  119. scoped_refptr<base::SingleThreadTaskRunner> caller_task_runner_;
  120. // Task runner on which |window_| is created.
  121. scoped_refptr<base::SingleThreadTaskRunner> ui_task_runner_;
  122. LocalHotkeyInputMonitorManager* manager_;
  123. // Invoked in the |caller_task_runner_| thread to report session disconnect
  124. // requests.
  125. base::OnceClosure disconnect_callback_;
  126. };
  127. LocalHotkeyInputMonitorMac::LocalHotkeyInputMonitorMac(
  128. scoped_refptr<base::SingleThreadTaskRunner> caller_task_runner,
  129. scoped_refptr<base::SingleThreadTaskRunner> ui_task_runner,
  130. base::OnceClosure disconnect_callback)
  131. : core_(new Core(caller_task_runner,
  132. ui_task_runner,
  133. std::move(disconnect_callback))) {
  134. core_->Start();
  135. }
  136. LocalHotkeyInputMonitorMac::~LocalHotkeyInputMonitorMac() {
  137. DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
  138. core_->Stop();
  139. }
  140. LocalHotkeyInputMonitorMac::Core::Core(
  141. scoped_refptr<base::SingleThreadTaskRunner> caller_task_runner,
  142. scoped_refptr<base::SingleThreadTaskRunner> ui_task_runner,
  143. base::OnceClosure disconnect_callback)
  144. : caller_task_runner_(caller_task_runner),
  145. ui_task_runner_(ui_task_runner),
  146. manager_(nil),
  147. disconnect_callback_(std::move(disconnect_callback)) {
  148. DCHECK(disconnect_callback_);
  149. }
  150. void LocalHotkeyInputMonitorMac::Core::Start() {
  151. DCHECK(caller_task_runner_->BelongsToCurrentThread());
  152. ui_task_runner_->PostTask(FROM_HERE,
  153. base::BindOnce(&Core::StartOnUiThread, this));
  154. }
  155. void LocalHotkeyInputMonitorMac::Core::Stop() {
  156. DCHECK(caller_task_runner_->BelongsToCurrentThread());
  157. ui_task_runner_->PostTask(FROM_HERE,
  158. base::BindOnce(&Core::StopOnUiThread, this));
  159. }
  160. LocalHotkeyInputMonitorMac::Core::~Core() {
  161. DCHECK_EQ(manager_, nil);
  162. }
  163. void LocalHotkeyInputMonitorMac::Core::StartOnUiThread() {
  164. DCHECK(ui_task_runner_->BelongsToCurrentThread());
  165. manager_ = [[LocalHotkeyInputMonitorManager alloc] initWithMonitor:this];
  166. }
  167. void LocalHotkeyInputMonitorMac::Core::StopOnUiThread() {
  168. DCHECK(ui_task_runner_->BelongsToCurrentThread());
  169. [manager_ invalidate];
  170. [manager_ release];
  171. manager_ = nil;
  172. }
  173. void LocalHotkeyInputMonitorMac::Core::OnDisconnectShortcut() {
  174. if (disconnect_callback_) {
  175. caller_task_runner_->PostTask(FROM_HERE, std::move(disconnect_callback_));
  176. }
  177. }
  178. } // namespace
  179. std::unique_ptr<LocalHotkeyInputMonitor> LocalHotkeyInputMonitor::Create(
  180. scoped_refptr<base::SingleThreadTaskRunner> caller_task_runner,
  181. scoped_refptr<base::SingleThreadTaskRunner> input_task_runner,
  182. scoped_refptr<base::SingleThreadTaskRunner> ui_task_runner,
  183. base::OnceClosure disconnect_callback) {
  184. return std::make_unique<LocalHotkeyInputMonitorMac>(
  185. caller_task_runner, ui_task_runner, std::move(disconnect_callback));
  186. }
  187. } // namespace remoting