| OLD | NEW |
| (Empty) |
| 1 // Copyright (c) 2006-2008 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 | |
| 5 #include <stdint.h> | |
| 6 | |
| 7 #include "sandbox/win/src/win2k_threadpool.h" | |
| 8 #include "testing/gtest/include/gtest/gtest.h" | |
| 9 | |
| 10 void __stdcall EmptyCallBack(void*, unsigned char) { | |
| 11 } | |
| 12 | |
| 13 void __stdcall TestCallBack(void* context, unsigned char) { | |
| 14 HANDLE event = reinterpret_cast<HANDLE>(context); | |
| 15 ::SetEvent(event); | |
| 16 } | |
| 17 | |
| 18 namespace sandbox { | |
| 19 | |
| 20 // Test that register and unregister work, part 1. | |
| 21 TEST(IPCTest, ThreadPoolRegisterTest1) { | |
| 22 Win2kThreadPool thread_pool; | |
| 23 | |
| 24 EXPECT_EQ(0u, thread_pool.OutstandingWaits()); | |
| 25 | |
| 26 HANDLE event1 = ::CreateEventW(NULL, FALSE, FALSE, NULL); | |
| 27 HANDLE event2 = ::CreateEventW(NULL, FALSE, FALSE, NULL); | |
| 28 | |
| 29 uint32_t context = 0; | |
| 30 EXPECT_FALSE(thread_pool.RegisterWait(0, event1, EmptyCallBack, &context)); | |
| 31 EXPECT_EQ(0u, thread_pool.OutstandingWaits()); | |
| 32 | |
| 33 EXPECT_TRUE(thread_pool.RegisterWait(this, event1, EmptyCallBack, &context)); | |
| 34 EXPECT_EQ(1u, thread_pool.OutstandingWaits()); | |
| 35 EXPECT_TRUE(thread_pool.RegisterWait(this, event2, EmptyCallBack, &context)); | |
| 36 EXPECT_EQ(2u, thread_pool.OutstandingWaits()); | |
| 37 | |
| 38 EXPECT_TRUE(thread_pool.UnRegisterWaits(this)); | |
| 39 EXPECT_EQ(0u, thread_pool.OutstandingWaits()); | |
| 40 | |
| 41 EXPECT_EQ(TRUE, ::CloseHandle(event1)); | |
| 42 EXPECT_EQ(TRUE, ::CloseHandle(event2)); | |
| 43 } | |
| 44 | |
| 45 // Test that register and unregister work, part 2. | |
| 46 TEST(IPCTest, ThreadPoolRegisterTest2) { | |
| 47 Win2kThreadPool thread_pool; | |
| 48 | |
| 49 HANDLE event1 = ::CreateEventW(NULL, FALSE, FALSE, NULL); | |
| 50 HANDLE event2 = ::CreateEventW(NULL, FALSE, FALSE, NULL); | |
| 51 | |
| 52 uint32_t context = 0; | |
| 53 uint32_t c1 = 0; | |
| 54 uint32_t c2 = 0; | |
| 55 | |
| 56 EXPECT_TRUE(thread_pool.RegisterWait(&c1, event1, EmptyCallBack, &context)); | |
| 57 EXPECT_EQ(1u, thread_pool.OutstandingWaits()); | |
| 58 EXPECT_TRUE(thread_pool.RegisterWait(&c2, event2, EmptyCallBack, &context)); | |
| 59 EXPECT_EQ(2u, thread_pool.OutstandingWaits()); | |
| 60 | |
| 61 EXPECT_TRUE(thread_pool.UnRegisterWaits(&c2)); | |
| 62 EXPECT_EQ(1u, thread_pool.OutstandingWaits()); | |
| 63 EXPECT_TRUE(thread_pool.UnRegisterWaits(&c2)); | |
| 64 EXPECT_EQ(1u, thread_pool.OutstandingWaits()); | |
| 65 | |
| 66 EXPECT_TRUE(thread_pool.UnRegisterWaits(&c1)); | |
| 67 EXPECT_EQ(0u, thread_pool.OutstandingWaits()); | |
| 68 | |
| 69 EXPECT_EQ(TRUE, ::CloseHandle(event1)); | |
| 70 EXPECT_EQ(TRUE, ::CloseHandle(event2)); | |
| 71 } | |
| 72 | |
| 73 // Test that the thread pool has at least a thread that services an event. | |
| 74 // Test that when the event is un-registered is no longer serviced. | |
| 75 TEST(IPCTest, ThreadPoolSignalAndWaitTest) { | |
| 76 Win2kThreadPool thread_pool; | |
| 77 | |
| 78 // The events are auto reset and start not signaled. | |
| 79 HANDLE event1 = ::CreateEventW(NULL, FALSE, FALSE, NULL); | |
| 80 HANDLE event2 = ::CreateEventW(NULL, FALSE, FALSE, NULL); | |
| 81 | |
| 82 EXPECT_TRUE(thread_pool.RegisterWait(this, event1, TestCallBack, event2)); | |
| 83 | |
| 84 EXPECT_EQ(WAIT_OBJECT_0, ::SignalObjectAndWait(event1, event2, 5000, FALSE)); | |
| 85 EXPECT_EQ(WAIT_OBJECT_0, ::SignalObjectAndWait(event1, event2, 5000, FALSE)); | |
| 86 | |
| 87 EXPECT_TRUE(thread_pool.UnRegisterWaits(this)); | |
| 88 EXPECT_EQ(0u, thread_pool.OutstandingWaits()); | |
| 89 | |
| 90 EXPECT_EQ(static_cast<DWORD>(WAIT_TIMEOUT), | |
| 91 ::SignalObjectAndWait(event1, event2, 1000, FALSE)); | |
| 92 | |
| 93 EXPECT_EQ(TRUE, ::CloseHandle(event1)); | |
| 94 EXPECT_EQ(TRUE, ::CloseHandle(event2)); | |
| 95 } | |
| 96 | |
| 97 } // namespace sandbox | |
| OLD | NEW |