OLD | NEW |
---|---|
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 // | 4 // |
5 // Unit tests for event trace consumer base class. | 5 // Unit tests for event trace consumer base class. |
6 #include "base/win/event_trace_consumer.h" | 6 #include "base/win/event_trace_consumer.h" |
7 | 7 |
8 #include <list> | 8 #include <list> |
9 | 9 |
10 #include <objbase.h> | 10 #include <objbase.h> |
(...skipping 25 matching lines...) Expand all Loading... | |
36 sank_event_.Set(::CreateEvent(NULL, TRUE, FALSE, NULL)); | 36 sank_event_.Set(::CreateEvent(NULL, TRUE, FALSE, NULL)); |
37 ClearQueue(); | 37 ClearQueue(); |
38 } | 38 } |
39 | 39 |
40 ~TestConsumer() { | 40 ~TestConsumer() { |
41 ClearQueue(); | 41 ClearQueue(); |
42 sank_event_.Close(); | 42 sank_event_.Close(); |
43 } | 43 } |
44 | 44 |
45 void ClearQueue() { | 45 void ClearQueue() { |
46 EventQueue::const_iterator it(events_.begin()), end(events_.end()); | 46 for (EventQueue::const_iterator it(events_.begin()), end(events_.end()); |
47 | 47 it != end; ++it) { |
48 for (; it != end; ++it) { | 48 delete[] it->MofData; |
49 delete [] it->MofData; | |
50 } | 49 } |
51 | 50 |
52 events_.clear(); | 51 events_.clear(); |
53 } | 52 } |
54 | 53 |
55 static void EnqueueEvent(EVENT_TRACE* event) { | 54 static void EnqueueEvent(EVENT_TRACE* event) { |
56 events_.push_back(*event); | 55 events_.push_back(*event); |
57 EVENT_TRACE& back = events_.back(); | 56 EVENT_TRACE& back = events_.back(); |
58 | 57 |
59 if (NULL != event->MofData && 0 != event->MofLength) { | 58 if (event->MofData != NULL && event->MofLength != 0) { |
Peter Kasting
2014/03/18 03:17:09
(From our style guide: "Prefer (foo == 0) to (0 ==
| |
60 back.MofData = new char[event->MofLength]; | 59 back.MofData = new char[event->MofLength]; |
61 memcpy(back.MofData, event->MofData, event->MofLength); | 60 memcpy(back.MofData, event->MofData, event->MofLength); |
62 } | 61 } |
63 } | 62 } |
64 | 63 |
65 static void ProcessEvent(EVENT_TRACE* event) { | 64 static void ProcessEvent(EVENT_TRACE* event) { |
66 EnqueueEvent(event); | 65 EnqueueEvent(event); |
67 ::SetEvent(sank_event_.Get()); | 66 ::SetEvent(sank_event_.Get()); |
68 } | 67 } |
69 | 68 |
(...skipping 35 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
105 }; | 104 }; |
106 | 105 |
107 } // namespace | 106 } // namespace |
108 | 107 |
109 TEST_F(EtwTraceConsumerBaseTest, Initialize) { | 108 TEST_F(EtwTraceConsumerBaseTest, Initialize) { |
110 TestConsumer consumer_; | 109 TestConsumer consumer_; |
111 } | 110 } |
112 | 111 |
113 TEST_F(EtwTraceConsumerBaseTest, OpenRealtimeSucceedsWhenNoSession) { | 112 TEST_F(EtwTraceConsumerBaseTest, OpenRealtimeSucceedsWhenNoSession) { |
114 TestConsumer consumer_; | 113 TestConsumer consumer_; |
115 | |
116 ASSERT_HRESULT_SUCCEEDED( | 114 ASSERT_HRESULT_SUCCEEDED( |
117 consumer_.OpenRealtimeSession(session_name_.c_str())); | 115 consumer_.OpenRealtimeSession(session_name_.c_str())); |
118 } | 116 } |
119 | 117 |
120 TEST_F(EtwTraceConsumerBaseTest, ConsumerImmediateFailureWhenNoSession) { | 118 TEST_F(EtwTraceConsumerBaseTest, ConsumerImmediateFailureWhenNoSession) { |
121 TestConsumer consumer_; | 119 TestConsumer consumer_; |
122 | |
123 ASSERT_HRESULT_SUCCEEDED( | 120 ASSERT_HRESULT_SUCCEEDED( |
124 consumer_.OpenRealtimeSession(session_name_.c_str())); | 121 consumer_.OpenRealtimeSession(session_name_.c_str())); |
125 ASSERT_HRESULT_FAILED(consumer_.Consume()); | 122 ASSERT_HRESULT_FAILED(consumer_.Consume()); |
126 } | 123 } |
127 | 124 |
128 namespace { | 125 namespace { |
129 | 126 |
130 class EtwTraceConsumerRealtimeTest: public EtwTraceConsumerBaseTest { | 127 class EtwTraceConsumerRealtimeTest: public EtwTraceConsumerBaseTest { |
131 public: | 128 public: |
132 virtual void SetUp() { | 129 virtual void SetUp() { |
133 EtwTraceConsumerBaseTest::SetUp(); | 130 EtwTraceConsumerBaseTest::SetUp(); |
134 | |
135 ASSERT_HRESULT_SUCCEEDED( | 131 ASSERT_HRESULT_SUCCEEDED( |
136 consumer_.OpenRealtimeSession(session_name_.c_str())); | 132 consumer_.OpenRealtimeSession(session_name_.c_str())); |
137 } | 133 } |
138 | 134 |
139 virtual void TearDown() { | 135 virtual void TearDown() { |
140 consumer_.Close(); | 136 consumer_.Close(); |
141 | |
142 EtwTraceConsumerBaseTest::TearDown(); | 137 EtwTraceConsumerBaseTest::TearDown(); |
143 } | 138 } |
144 | 139 |
145 DWORD ConsumerThread() { | 140 DWORD ConsumerThread() { |
146 ::SetEvent(consumer_ready_.Get()); | 141 ::SetEvent(consumer_ready_.Get()); |
147 | 142 return consumer_.Consume(); |
148 HRESULT hr = consumer_.Consume(); | |
149 return hr; | |
150 } | 143 } |
151 | 144 |
152 static DWORD WINAPI ConsumerThreadMainProc(void* arg) { | 145 static DWORD WINAPI ConsumerThreadMainProc(void* arg) { |
153 return reinterpret_cast<EtwTraceConsumerRealtimeTest*>(arg)-> | 146 return reinterpret_cast<EtwTraceConsumerRealtimeTest*>(arg)-> |
154 ConsumerThread(); | 147 ConsumerThread(); |
155 } | 148 } |
156 | 149 |
157 HRESULT StartConsumerThread() { | 150 HRESULT StartConsumerThread() { |
158 consumer_ready_.Set(::CreateEvent(NULL, TRUE, FALSE, NULL)); | 151 consumer_ready_.Set(::CreateEvent(NULL, TRUE, FALSE, NULL)); |
159 EXPECT_TRUE(consumer_ready_ != NULL); | 152 EXPECT_TRUE(consumer_ready_ != NULL); |
160 consumer_thread_.Set(::CreateThread(NULL, 0, ConsumerThreadMainProc, | 153 consumer_thread_.Set(::CreateThread(NULL, 0, ConsumerThreadMainProc, this, |
161 this, 0, NULL)); | 154 0, NULL)); |
162 if (NULL == consumer_thread_.Get()) | 155 if (consumer_thread_.Get() == NULL) |
163 return HRESULT_FROM_WIN32(::GetLastError()); | 156 return HRESULT_FROM_WIN32(::GetLastError()); |
164 | 157 |
165 HRESULT hr = S_OK; | |
166 HANDLE events[] = { consumer_ready_, consumer_thread_ }; | 158 HANDLE events[] = { consumer_ready_, consumer_thread_ }; |
167 DWORD result = ::WaitForMultipleObjects(arraysize(events), events, | 159 DWORD result = |
168 FALSE, INFINITE); | 160 ::WaitForMultipleObjects(arraysize(events), events, FALSE, INFINITE); |
169 switch (result) { | 161 switch (result) { |
170 case WAIT_OBJECT_0: | 162 case WAIT_OBJECT_0: |
171 // The event was set, the consumer_ is ready. | 163 // The event was set, the consumer_ is ready. |
172 return S_OK; | 164 return S_OK; |
165 | |
173 case WAIT_OBJECT_0 + 1: { | 166 case WAIT_OBJECT_0 + 1: { |
174 // The thread finished. This may race with the event, so check | 167 // The thread finished. This may race with the event, so check |
175 // explicitly for the event here, before concluding there's trouble. | 168 // explicitly for the event here, before concluding there's trouble. |
176 if (WAIT_OBJECT_0 == ::WaitForSingleObject(consumer_ready_, 0)) | 169 if (::WaitForSingleObject(consumer_ready_, 0) == WAIT_OBJECT_0) |
177 return S_OK; | 170 return S_OK; |
178 DWORD exit_code = 0; | 171 DWORD exit_code = 0; |
179 if (::GetExitCodeThread(consumer_thread_, &exit_code)) | 172 return ::GetExitCodeThread(consumer_thread_, &exit_code) ? |
180 return exit_code; | 173 exit_code : HRESULT_FROM_WIN32(::GetLastError()); |
181 else | |
182 return HRESULT_FROM_WIN32(::GetLastError()); | |
183 break; | |
184 } | 174 } |
175 | |
185 default: | 176 default: |
186 return E_UNEXPECTED; | 177 return E_UNEXPECTED; |
187 break; | |
188 } | 178 } |
189 | |
190 return hr; | |
191 } | 179 } |
192 | 180 |
193 // Waits for consumer_ thread to exit, and returns its exit code. | 181 // Waits for consumer_ thread to exit, and returns its exit code. |
194 HRESULT JoinConsumerThread() { | 182 HRESULT JoinConsumerThread() { |
195 if (WAIT_OBJECT_0 != ::WaitForSingleObject(consumer_thread_, INFINITE)) | 183 if (::WaitForSingleObject(consumer_thread_, INFINITE) != WAIT_OBJECT_0) |
196 return HRESULT_FROM_WIN32(::GetLastError()); | 184 return HRESULT_FROM_WIN32(::GetLastError()); |
197 | 185 |
198 DWORD exit_code = 0; | 186 DWORD exit_code = 0; |
199 if (::GetExitCodeThread(consumer_thread_, &exit_code)) | 187 return ::GetExitCodeThread(consumer_thread_, &exit_code) ? |
200 return exit_code; | 188 exit_code : HRESULT_FROM_WIN32(::GetLastError()); |
201 | |
202 return HRESULT_FROM_WIN32(::GetLastError()); | |
203 } | 189 } |
204 | 190 |
205 TestConsumer consumer_; | 191 TestConsumer consumer_; |
206 ScopedHandle consumer_ready_; | 192 ScopedHandle consumer_ready_; |
207 ScopedHandle consumer_thread_; | 193 ScopedHandle consumer_thread_; |
208 }; | 194 }; |
209 | 195 |
210 } // namespace | 196 } // namespace |
211 | 197 |
212 TEST_F(EtwTraceConsumerRealtimeTest, ConsumerReturnsWhenSessionClosed) { | 198 TEST_F(EtwTraceConsumerRealtimeTest, ConsumerReturnsWhenSessionClosed) { |
213 EtwTraceController controller; | 199 EtwTraceController controller; |
214 | 200 if (controller.StartRealtimeSession(session_name_.c_str(), 100 * 1024) == |
215 HRESULT hr = controller.StartRealtimeSession(session_name_.c_str(), | 201 E_ACCESSDENIED) { |
216 100 * 1024); | |
217 if (hr == E_ACCESSDENIED) { | |
218 VLOG(1) << "You must be an administrator to run this test on Vista"; | 202 VLOG(1) << "You must be an administrator to run this test on Vista"; |
219 return; | 203 return; |
220 } | 204 } |
221 | 205 |
222 // Start the consumer_. | 206 // Start the consumer_. |
223 ASSERT_HRESULT_SUCCEEDED(StartConsumerThread()); | 207 ASSERT_HRESULT_SUCCEEDED(StartConsumerThread()); |
224 | 208 |
225 // Wait around for the consumer_ thread a bit. | 209 // Wait around for the consumer_ thread a bit. |
226 ASSERT_EQ(WAIT_TIMEOUT, ::WaitForSingleObject(consumer_thread_, 50)); | 210 ASSERT_EQ(WAIT_TIMEOUT, ::WaitForSingleObject(consumer_thread_, 50)); |
227 | |
228 ASSERT_HRESULT_SUCCEEDED(controller.Stop(NULL)); | 211 ASSERT_HRESULT_SUCCEEDED(controller.Stop(NULL)); |
229 | 212 |
230 // The consumer_ returns success on session stop. | 213 // The consumer_ returns success on session stop. |
231 ASSERT_HRESULT_SUCCEEDED(JoinConsumerThread()); | 214 ASSERT_HRESULT_SUCCEEDED(JoinConsumerThread()); |
232 } | 215 } |
233 | 216 |
234 namespace { | 217 namespace { |
235 | 218 |
236 // {57E47923-A549-476f-86CA-503D57F59E62} | 219 // {57E47923-A549-476f-86CA-503D57F59E62} |
237 DEFINE_GUID(kTestEventType, | 220 DEFINE_GUID( |
238 0x57e47923, 0xa549, 0x476f, 0x86, 0xca, 0x50, 0x3d, 0x57, 0xf5, 0x9e, 0x62); | 221 kTestEventType, |
222 0x57e47923, 0xa549, 0x476f, 0x86, 0xca, 0x50, 0x3d, 0x57, 0xf5, 0x9e, 0x62); | |
239 | 223 |
240 } // namespace | 224 } // namespace |
241 | 225 |
242 TEST_F(EtwTraceConsumerRealtimeTest, ConsumeEvent) { | 226 TEST_F(EtwTraceConsumerRealtimeTest, ConsumeEvent) { |
243 EtwTraceController controller; | 227 EtwTraceController controller; |
244 HRESULT hr = controller.StartRealtimeSession(session_name_.c_str(), | 228 if (controller.StartRealtimeSession(session_name_.c_str(), 100 * 1024) == |
245 100 * 1024); | 229 E_ACCESSDENIED) { |
246 if (hr == E_ACCESSDENIED) { | |
247 VLOG(1) << "You must be an administrator to run this test on Vista"; | 230 VLOG(1) << "You must be an administrator to run this test on Vista"; |
248 return; | 231 return; |
249 } | 232 } |
250 | 233 |
251 ASSERT_HRESULT_SUCCEEDED(controller.EnableProvider(test_provider_, | 234 ASSERT_HRESULT_SUCCEEDED(controller.EnableProvider( |
252 TRACE_LEVEL_VERBOSE, 0xFFFFFFFF)); | 235 test_provider_, TRACE_LEVEL_VERBOSE, 0xFFFFFFFF)); |
253 | 236 |
254 EtwTraceProvider provider(test_provider_); | 237 EtwTraceProvider provider(test_provider_); |
255 ASSERT_EQ(ERROR_SUCCESS, provider.Register()); | 238 ASSERT_EQ(ERROR_SUCCESS, provider.Register()); |
256 | 239 |
257 // Start the consumer_. | 240 // Start the consumer_. |
258 ASSERT_HRESULT_SUCCEEDED(StartConsumerThread()); | 241 ASSERT_HRESULT_SUCCEEDED(StartConsumerThread()); |
259 | |
260 ASSERT_EQ(0, TestConsumer::events_.size()); | 242 ASSERT_EQ(0, TestConsumer::events_.size()); |
261 | 243 |
262 EtwMofEvent<1> event(kTestEventType, 1, TRACE_LEVEL_ERROR); | 244 EtwMofEvent<1> event(kTestEventType, 1, TRACE_LEVEL_ERROR); |
263 EXPECT_EQ(ERROR_SUCCESS, provider.Log(&event.header)); | 245 EXPECT_EQ(ERROR_SUCCESS, provider.Log(&event.header)); |
264 | |
265 EXPECT_EQ(WAIT_OBJECT_0, ::WaitForSingleObject(TestConsumer::sank_event_, | 246 EXPECT_EQ(WAIT_OBJECT_0, ::WaitForSingleObject(TestConsumer::sank_event_, |
266 INFINITE)); | 247 INFINITE)); |
267 ASSERT_HRESULT_SUCCEEDED(controller.Stop(NULL)); | 248 ASSERT_HRESULT_SUCCEEDED(controller.Stop(NULL)); |
268 ASSERT_HRESULT_SUCCEEDED(JoinConsumerThread()); | 249 ASSERT_HRESULT_SUCCEEDED(JoinConsumerThread()); |
269 ASSERT_NE(0u, TestConsumer::events_.size()); | 250 ASSERT_NE(0u, TestConsumer::events_.size()); |
270 } | 251 } |
271 | 252 |
272 namespace { | 253 namespace { |
273 | 254 |
274 // We run events through a file session to assert that | 255 // We run events through a file session to assert that |
(...skipping 24 matching lines...) Expand all Loading... | |
299 HRESULT LogEventToTempSession(PEVENT_TRACE_HEADER header) { | 280 HRESULT LogEventToTempSession(PEVENT_TRACE_HEADER header) { |
300 EtwTraceController controller; | 281 EtwTraceController controller; |
301 | 282 |
302 // Set up a file session. | 283 // Set up a file session. |
303 HRESULT hr = controller.StartFileSession(session_name_.c_str(), | 284 HRESULT hr = controller.StartFileSession(session_name_.c_str(), |
304 temp_file_.value().c_str()); | 285 temp_file_.value().c_str()); |
305 if (FAILED(hr)) | 286 if (FAILED(hr)) |
306 return hr; | 287 return hr; |
307 | 288 |
308 // Enable our provider. | 289 // Enable our provider. |
309 EXPECT_HRESULT_SUCCEEDED(controller.EnableProvider(test_provider_, | 290 EXPECT_HRESULT_SUCCEEDED(controller.EnableProvider( |
310 TRACE_LEVEL_VERBOSE, 0xFFFFFFFF)); | 291 test_provider_, TRACE_LEVEL_VERBOSE, 0xFFFFFFFF)); |
311 | 292 |
312 EtwTraceProvider provider(test_provider_); | 293 EtwTraceProvider provider(test_provider_); |
313 // Then register our provider, means we get a session handle immediately. | 294 // Then register our provider, means we get a session handle immediately. |
314 EXPECT_EQ(ERROR_SUCCESS, provider.Register()); | 295 EXPECT_EQ(ERROR_SUCCESS, provider.Register()); |
315 // Trace the event, it goes to the temp file. | 296 // Trace the event, it goes to the temp file. |
316 EXPECT_EQ(ERROR_SUCCESS, provider.Log(header)); | 297 EXPECT_EQ(ERROR_SUCCESS, provider.Log(header)); |
317 EXPECT_HRESULT_SUCCEEDED(controller.DisableProvider(test_provider_)); | 298 EXPECT_HRESULT_SUCCEEDED(controller.DisableProvider(test_provider_)); |
318 EXPECT_HRESULT_SUCCEEDED(provider.Unregister()); | 299 EXPECT_HRESULT_SUCCEEDED(provider.Unregister()); |
319 EXPECT_HRESULT_SUCCEEDED(controller.Flush(NULL)); | 300 EXPECT_HRESULT_SUCCEEDED(controller.Flush(NULL)); |
320 EXPECT_HRESULT_SUCCEEDED(controller.Stop(NULL)); | 301 EXPECT_HRESULT_SUCCEEDED(controller.Stop(NULL)); |
(...skipping 46 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
367 event.fields[0].DataPtr = reinterpret_cast<ULONG64>(kData); | 348 event.fields[0].DataPtr = reinterpret_cast<ULONG64>(kData); |
368 event.fields[0].Length = sizeof(kData); | 349 event.fields[0].Length = sizeof(kData); |
369 | 350 |
370 PEVENT_TRACE trace = NULL; | 351 PEVENT_TRACE trace = NULL; |
371 HRESULT hr = RoundTripEvent(&event.header, &trace); | 352 HRESULT hr = RoundTripEvent(&event.header, &trace); |
372 if (hr == E_ACCESSDENIED) { | 353 if (hr == E_ACCESSDENIED) { |
373 VLOG(1) << "You must be an administrator to run this test on Vista"; | 354 VLOG(1) << "You must be an administrator to run this test on Vista"; |
374 return; | 355 return; |
375 } | 356 } |
376 ASSERT_HRESULT_SUCCEEDED(hr) << "RoundTripEvent failed"; | 357 ASSERT_HRESULT_SUCCEEDED(hr) << "RoundTripEvent failed"; |
377 ASSERT_TRUE(NULL != trace); | 358 ASSERT_TRUE(trace != NULL); |
378 ASSERT_EQ(sizeof(kData), trace->MofLength); | 359 ASSERT_EQ(sizeof(kData), trace->MofLength); |
379 ASSERT_STREQ(kData, reinterpret_cast<const char*>(trace->MofData)); | 360 ASSERT_STREQ(kData, reinterpret_cast<const char*>(trace->MofData)); |
380 } | 361 } |
381 | 362 |
382 } // namespace win | 363 } // namespace win |
383 } // namespace base | 364 } // namespace base |
OLD | NEW |