OLD | NEW |
1 /* | 1 /* |
2 * Copyright 2004 The WebRTC Project Authors. All rights reserved. | 2 * Copyright 2004 The WebRTC Project Authors. All rights reserved. |
3 * | 3 * |
4 * Use of this source code is governed by a BSD-style license | 4 * Use of this source code is governed by a BSD-style license |
5 * that can be found in the LICENSE file in the root of the source | 5 * that can be found in the LICENSE file in the root of the source |
6 * tree. An additional intellectual property rights grant can be found | 6 * tree. An additional intellectual property rights grant can be found |
7 * in the file PATENTS. All contributing project authors may | 7 * in the file PATENTS. All contributing project authors may |
8 * be found in the AUTHORS file in the root of the source tree. | 8 * be found in the AUTHORS file in the root of the source tree. |
9 */ | 9 */ |
10 | 10 |
11 #include "webrtc/rtc_base/messagequeue.h" | 11 #include "webrtc/rtc_base/messagequeue.h" |
12 | 12 |
13 #include <functional> | 13 #include <functional> |
14 | 14 |
15 #include "webrtc/rtc_base/atomicops.h" | 15 #include "webrtc/rtc_base/atomicops.h" |
16 #include "webrtc/rtc_base/bind.h" | 16 #include "webrtc/rtc_base/bind.h" |
17 #include "webrtc/rtc_base/event.h" | 17 #include "webrtc/rtc_base/event.h" |
18 #include "webrtc/rtc_base/gunit.h" | 18 #include "webrtc/rtc_base/gunit.h" |
19 #include "webrtc/rtc_base/logging.h" | 19 #include "webrtc/rtc_base/logging.h" |
20 #include "webrtc/rtc_base/nullsocketserver.h" | 20 #include "webrtc/rtc_base/nullsocketserver.h" |
| 21 #include "webrtc/rtc_base/refcount.h" |
| 22 #include "webrtc/rtc_base/refcountedobject.h" |
21 #include "webrtc/rtc_base/thread.h" | 23 #include "webrtc/rtc_base/thread.h" |
22 #include "webrtc/rtc_base/timeutils.h" | 24 #include "webrtc/rtc_base/timeutils.h" |
23 | 25 |
24 using namespace rtc; | 26 using namespace rtc; |
25 | 27 |
26 class MessageQueueTest: public testing::Test, public MessageQueue { | 28 class MessageQueueTest: public testing::Test, public MessageQueue { |
27 public: | 29 public: |
28 MessageQueueTest() : MessageQueue(SocketServer::CreateDefault(), true) {} | 30 MessageQueueTest() : MessageQueue(SocketServer::CreateDefault(), true) {} |
29 bool IsLocked_Worker() { | 31 bool IsLocked_Worker() { |
30 if (!crit_.TryEnter()) { | 32 if (!crit_.TryEnter()) { |
(...skipping 177 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
208 FunctorMessageHandler<void, std::function<void()>> event_signaler( | 210 FunctorMessageHandler<void, std::function<void()>> event_signaler( |
209 [&entered_process_all_message_queues] { | 211 [&entered_process_all_message_queues] { |
210 entered_process_all_message_queues.Set(); | 212 entered_process_all_message_queues.Set(); |
211 }); | 213 }); |
212 | 214 |
213 // Post messages (both delayed and non delayed) to both threads. | 215 // Post messages (both delayed and non delayed) to both threads. |
214 t.Post(RTC_FROM_HERE, &clearer); | 216 t.Post(RTC_FROM_HERE, &clearer); |
215 rtc::Thread::Current()->Post(RTC_FROM_HERE, &event_signaler); | 217 rtc::Thread::Current()->Post(RTC_FROM_HERE, &event_signaler); |
216 MessageQueueManager::ProcessAllMessageQueues(); | 218 MessageQueueManager::ProcessAllMessageQueues(); |
217 } | 219 } |
| 220 |
| 221 class RefCountedHandler |
| 222 : public MessageHandler, |
| 223 public rtc::RefCountInterface { |
| 224 public: |
| 225 void OnMessage(Message* msg) override {} |
| 226 }; |
| 227 |
| 228 class EmptyHandler : public MessageHandler { |
| 229 public: |
| 230 void OnMessage(Message* msg) override {} |
| 231 }; |
| 232 |
| 233 TEST(MessageQueueManager, ClearReentrant) { |
| 234 Thread t; |
| 235 EmptyHandler handler; |
| 236 RefCountedHandler* inner_handler( |
| 237 new rtc::RefCountedObject<RefCountedHandler>()); |
| 238 // When the empty handler is destroyed, it will clear messages queued for |
| 239 // itself. The message to be cleared itself wraps a MessageHandler object |
| 240 // (RefCountedHandler) so this will cause the message queue to be cleared |
| 241 // again in a re-entrant fashion, which previously triggered a DCHECK. |
| 242 // The inner handler will be removed in a re-entrant fashion from the |
| 243 // message queue of the thread while the outer handler is removed, verifying |
| 244 // that the iterator is not invalidated in "MessageQueue::Clear". |
| 245 t.Post(RTC_FROM_HERE, inner_handler, 0); |
| 246 t.Post(RTC_FROM_HERE, &handler, 0, |
| 247 new ScopedRefMessageData<RefCountedHandler>(inner_handler)); |
| 248 } |
OLD | NEW |