Chromium Code Reviews| 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 |