| OLD | NEW |
| (Empty) |
| 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 | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 #ifndef BASE_SYNCHRONIZATION_WAITABLE_EVENT_H_ | |
| 6 #define BASE_SYNCHRONIZATION_WAITABLE_EVENT_H_ | |
| 7 | |
| 8 #include "base/base_export.h" | |
| 9 #include "base/basictypes.h" | |
| 10 | |
| 11 #if defined(OS_WIN) | |
| 12 #include "base/win/scoped_handle.h" | |
| 13 #endif | |
| 14 | |
| 15 #if defined(OS_POSIX) | |
| 16 #include <list> | |
| 17 #include <utility> | |
| 18 #include "base/memory/ref_counted.h" | |
| 19 #include "base/synchronization/lock.h" | |
| 20 #endif | |
| 21 | |
| 22 namespace base { | |
| 23 | |
| 24 class TimeDelta; | |
| 25 | |
| 26 // A WaitableEvent can be a useful thread synchronization tool when you want to | |
| 27 // allow one thread to wait for another thread to finish some work. For | |
| 28 // non-Windows systems, this can only be used from within a single address | |
| 29 // space. | |
| 30 // | |
| 31 // Use a WaitableEvent when you would otherwise use a Lock+ConditionVariable to | |
| 32 // protect a simple boolean value. However, if you find yourself using a | |
| 33 // WaitableEvent in conjunction with a Lock to wait for a more complex state | |
| 34 // change (e.g., for an item to be added to a queue), then you should probably | |
| 35 // be using a ConditionVariable instead of a WaitableEvent. | |
| 36 // | |
| 37 // NOTE: On Windows, this class provides a subset of the functionality afforded | |
| 38 // by a Windows event object. This is intentional. If you are writing Windows | |
| 39 // specific code and you need other features of a Windows event, then you might | |
| 40 // be better off just using an Windows event directly. | |
| 41 class BASE_EXPORT WaitableEvent { | |
| 42 public: | |
| 43 // If manual_reset is true, then to set the event state to non-signaled, a | |
| 44 // consumer must call the Reset method. If this parameter is false, then the | |
| 45 // system automatically resets the event state to non-signaled after a single | |
| 46 // waiting thread has been released. | |
| 47 WaitableEvent(bool manual_reset, bool initially_signaled); | |
| 48 | |
| 49 #if defined(OS_WIN) | |
| 50 // Create a WaitableEvent from an Event HANDLE which has already been | |
| 51 // created. This objects takes ownership of the HANDLE and will close it when | |
| 52 // deleted. | |
| 53 explicit WaitableEvent(win::ScopedHandle event_handle); | |
| 54 #endif | |
| 55 | |
| 56 ~WaitableEvent(); | |
| 57 | |
| 58 // Put the event in the un-signaled state. | |
| 59 void Reset(); | |
| 60 | |
| 61 // Put the event in the signaled state. Causing any thread blocked on Wait | |
| 62 // to be woken up. | |
| 63 void Signal(); | |
| 64 | |
| 65 // Returns true if the event is in the signaled state, else false. If this | |
| 66 // is not a manual reset event, then this test will cause a reset. | |
| 67 bool IsSignaled(); | |
| 68 | |
| 69 // Wait indefinitely for the event to be signaled. Wait's return "happens | |
| 70 // after" |Signal| has completed. This means that it's safe for a | |
| 71 // WaitableEvent to synchronise its own destruction, like this: | |
| 72 // | |
| 73 // WaitableEvent *e = new WaitableEvent; | |
| 74 // SendToOtherThread(e); | |
| 75 // e->Wait(); | |
| 76 // delete e; | |
| 77 void Wait(); | |
| 78 | |
| 79 // Wait up until max_time has passed for the event to be signaled. Returns | |
| 80 // true if the event was signaled. If this method returns false, then it | |
| 81 // does not necessarily mean that max_time was exceeded. | |
| 82 // | |
| 83 // TimedWait can synchronise its own destruction like |Wait|. | |
| 84 bool TimedWait(const TimeDelta& max_time); | |
| 85 | |
| 86 #if defined(OS_WIN) | |
| 87 HANDLE handle() const { return handle_.Get(); } | |
| 88 #endif | |
| 89 | |
| 90 // Wait, synchronously, on multiple events. | |
| 91 // waitables: an array of WaitableEvent pointers | |
| 92 // count: the number of elements in @waitables | |
| 93 // | |
| 94 // returns: the index of a WaitableEvent which has been signaled. | |
| 95 // | |
| 96 // You MUST NOT delete any of the WaitableEvent objects while this wait is | |
| 97 // happening, however WaitMany's return "happens after" the |Signal| call | |
| 98 // that caused it has completed, like |Wait|. | |
| 99 static size_t WaitMany(WaitableEvent** waitables, size_t count); | |
| 100 | |
| 101 // For asynchronous waiting, see WaitableEventWatcher | |
| 102 | |
| 103 // This is a private helper class. It's here because it's used by friends of | |
| 104 // this class (such as WaitableEventWatcher) to be able to enqueue elements | |
| 105 // of the wait-list | |
| 106 class Waiter { | |
| 107 public: | |
| 108 // Signal the waiter to wake up. | |
| 109 // | |
| 110 // Consider the case of a Waiter which is in multiple WaitableEvent's | |
| 111 // wait-lists. Each WaitableEvent is automatic-reset and two of them are | |
| 112 // signaled at the same time. Now, each will wake only the first waiter in | |
| 113 // the wake-list before resetting. However, if those two waiters happen to | |
| 114 // be the same object (as can happen if another thread didn't have a chance | |
| 115 // to dequeue the waiter from the other wait-list in time), two auto-resets | |
| 116 // will have happened, but only one waiter has been signaled! | |
| 117 // | |
| 118 // Because of this, a Waiter may "reject" a wake by returning false. In | |
| 119 // this case, the auto-reset WaitableEvent shouldn't act as if anything has | |
| 120 // been notified. | |
| 121 virtual bool Fire(WaitableEvent* signaling_event) = 0; | |
| 122 | |
| 123 // Waiters may implement this in order to provide an extra condition for | |
| 124 // two Waiters to be considered equal. In WaitableEvent::Dequeue, if the | |
| 125 // pointers match then this function is called as a final check. See the | |
| 126 // comments in ~Handle for why. | |
| 127 virtual bool Compare(void* tag) = 0; | |
| 128 | |
| 129 protected: | |
| 130 virtual ~Waiter() {} | |
| 131 }; | |
| 132 | |
| 133 private: | |
| 134 friend class WaitableEventWatcher; | |
| 135 | |
| 136 #if defined(OS_WIN) | |
| 137 win::ScopedHandle handle_; | |
| 138 #else | |
| 139 // On Windows, one can close a HANDLE which is currently being waited on. The | |
| 140 // MSDN documentation says that the resulting behaviour is 'undefined', but | |
| 141 // it doesn't crash. However, if we were to include the following members | |
| 142 // directly then, on POSIX, one couldn't use WaitableEventWatcher to watch an | |
| 143 // event which gets deleted. This mismatch has bitten us several times now, | |
| 144 // so we have a kernel of the WaitableEvent, which is reference counted. | |
| 145 // WaitableEventWatchers may then take a reference and thus match the Windows | |
| 146 // behaviour. | |
| 147 struct WaitableEventKernel : | |
| 148 public RefCountedThreadSafe<WaitableEventKernel> { | |
| 149 public: | |
| 150 WaitableEventKernel(bool manual_reset, bool initially_signaled); | |
| 151 | |
| 152 bool Dequeue(Waiter* waiter, void* tag); | |
| 153 | |
| 154 base::Lock lock_; | |
| 155 const bool manual_reset_; | |
| 156 bool signaled_; | |
| 157 std::list<Waiter*> waiters_; | |
| 158 | |
| 159 private: | |
| 160 friend class RefCountedThreadSafe<WaitableEventKernel>; | |
| 161 ~WaitableEventKernel(); | |
| 162 }; | |
| 163 | |
| 164 typedef std::pair<WaitableEvent*, size_t> WaiterAndIndex; | |
| 165 | |
| 166 // When dealing with arrays of WaitableEvent*, we want to sort by the address | |
| 167 // of the WaitableEvent in order to have a globally consistent locking order. | |
| 168 // In that case we keep them, in sorted order, in an array of pairs where the | |
| 169 // second element is the index of the WaitableEvent in the original, | |
| 170 // unsorted, array. | |
| 171 static size_t EnqueueMany(WaiterAndIndex* waitables, | |
| 172 size_t count, Waiter* waiter); | |
| 173 | |
| 174 bool SignalAll(); | |
| 175 bool SignalOne(); | |
| 176 void Enqueue(Waiter* waiter); | |
| 177 | |
| 178 scoped_refptr<WaitableEventKernel> kernel_; | |
| 179 #endif | |
| 180 | |
| 181 DISALLOW_COPY_AND_ASSIGN(WaitableEvent); | |
| 182 }; | |
| 183 | |
| 184 } // namespace base | |
| 185 | |
| 186 #endif // BASE_SYNCHRONIZATION_WAITABLE_EVENT_H_ | |
| OLD | NEW |