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
|
// Copyright 2019 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef CHROME_BROWSER_UI_VIEWS_ACCESSIBILITY_UIA_ACCESSIBILITY_EVENT_WAITER_H_
#define CHROME_BROWSER_UI_VIEWS_ACCESSIBILITY_UIA_ACCESSIBILITY_EVENT_WAITER_H_
#include <ole2.h>
#include <stdint.h>
#include <wrl/client.h>
#include <map>
#include <string>
#include <vector>
#include "base/memory/raw_ptr.h"
#include "base/process/process_handle.h"
#include "base/run_loop.h"
#include "base/synchronization/waitable_event.h"
#include "base/threading/platform_thread.h"
#include "base/time/time.h"
#include "base/win/atl.h"
#include "ui/views/accessibility/view_accessibility.h"
#include <uiautomation.h>
struct UiaAccessibilityWaiterInfo {
HWND hwnd;
std::wstring role;
std::wstring name;
ax::mojom::Event event;
};
class UiaAccessibilityEventWaiter {
public:
explicit UiaAccessibilityEventWaiter(UiaAccessibilityWaiterInfo info);
~UiaAccessibilityEventWaiter();
void Wait();
void WaitWithTimeout(base::TimeDelta timeout);
private:
// All UIA calls need to be made on a secondary MTA thread to avoid sporadic
// test hangs / timeouts.
class Thread : public base::PlatformThread::Delegate {
public:
Thread();
~Thread() override;
void Init(UiaAccessibilityEventWaiter* owner,
const UiaAccessibilityWaiterInfo& info,
base::OnceClosure initialization_loop,
base::OnceClosure shutdown_loop);
void SendShutdownSignal();
void ThreadMain() override;
protected:
UiaAccessibilityWaiterInfo info_;
private:
raw_ptr<UiaAccessibilityEventWaiter> owner_ = nullptr;
Microsoft::WRL::ComPtr<IUIAutomation> uia_;
Microsoft::WRL::ComPtr<IUIAutomationElement> root_;
Microsoft::WRL::ComPtr<IUIAutomationCacheRequest> cache_request_;
// Thread synchronization members.
base::OnceClosure initialization_complete_;
base::OnceClosure shutdown_complete_;
base::WaitableEvent shutdown_signal_;
// An implementation of various UIA interfaces that forward event
// notifications to the waiter.
class EventHandler : public CComObjectRootEx<CComMultiThreadModel>,
public IUIAutomationFocusChangedEventHandler,
public IUIAutomationPropertyChangedEventHandler,
public IUIAutomationStructureChangedEventHandler,
public IUIAutomationEventHandler {
public:
EventHandler();
EventHandler(const EventHandler&) = delete;
EventHandler& operator=(const EventHandler&) = delete;
virtual ~EventHandler();
void Init(UiaAccessibilityEventWaiter::Thread* owner,
Microsoft::WRL::ComPtr<IUIAutomationElement> root);
void CleanUp();
BEGIN_COM_MAP(EventHandler)
COM_INTERFACE_ENTRY(IUIAutomationFocusChangedEventHandler)
COM_INTERFACE_ENTRY(IUIAutomationPropertyChangedEventHandler)
COM_INTERFACE_ENTRY(IUIAutomationStructureChangedEventHandler)
COM_INTERFACE_ENTRY(IUIAutomationEventHandler)
END_COM_MAP()
// IUIAutomationFocusChangedEventHandler interface.
IFACEMETHODIMP HandleFocusChangedEvent(
IUIAutomationElement* sender) override;
// IUIAutomationPropertyChangedEventHandler interface.
IFACEMETHODIMP HandlePropertyChangedEvent(IUIAutomationElement* sender,
PROPERTYID property_id,
VARIANT new_value) override;
// IUIAutomationStructureChangedEventHandler interface.
IFACEMETHODIMP HandleStructureChangedEvent(
IUIAutomationElement* sender,
StructureChangeType change_type,
SAFEARRAY* runtime_id) override;
// IUIAutomationEventHandler interface.
IFACEMETHODIMP HandleAutomationEvent(IUIAutomationElement* sender,
EVENTID event_id) override;
// Points to the waiter to receive notifications.
raw_ptr<UiaAccessibilityEventWaiter::Thread> owner_ = nullptr;
private:
bool MatchesNameRole(IUIAutomationElement* sender);
Microsoft::WRL::ComPtr<IUIAutomationElement> root_;
};
Microsoft::WRL::ComPtr<CComObject<EventHandler>> uia_event_handler_;
};
Thread thread_;
base::RunLoop shutdown_loop_;
base::PlatformThreadHandle thread_handle_;
};
#endif // CHROME_BROWSER_UI_VIEWS_ACCESSIBILITY_UIA_ACCESSIBILITY_EVENT_WAITER_H_
|
