henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 +0000 | [diff] [blame] | 1 | /* |
| 2 | * Copyright 2004 The WebRTC Project Authors. All rights reserved. |
| 3 | * |
| 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 |
| 6 | * tree. An additional intellectual property rights grant can be found |
| 7 | * in the file PATENTS. All contributing project authors may |
| 8 | * be found in the AUTHORS file in the root of the source tree. |
| 9 | */ |
andresp@webrtc.org | ff689be | 2015-02-12 11:54:26 +0000 | [diff] [blame] | 10 | #include <algorithm> |
| 11 | |
Mirko Bonadei | 92ea95e | 2017-09-15 06:47:31 +0200 | [diff] [blame] | 12 | #include "rtc_base/atomicops.h" |
| 13 | #include "rtc_base/checks.h" |
| 14 | #include "rtc_base/logging.h" |
| 15 | #include "rtc_base/messagequeue.h" |
| 16 | #include "rtc_base/stringencode.h" |
| 17 | #include "rtc_base/thread.h" |
| 18 | #include "rtc_base/trace_event.h" |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 +0000 | [diff] [blame] | 19 | |
| 20 | namespace rtc { |
andresp | cdf6172 | 2016-07-08 02:45:40 -0700 | [diff] [blame] | 21 | namespace { |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 +0000 | [diff] [blame] | 22 | |
Honghai Zhang | 82d7862 | 2016-05-06 11:29:15 -0700 | [diff] [blame] | 23 | const int kMaxMsgLatency = 150; // 150 ms |
Taylor Brandstetter | 5d97a9a | 2016-06-10 14:17:27 -0700 | [diff] [blame] | 24 | const int kSlowDispatchLoggingThreshold = 50; // 50 ms |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 +0000 | [diff] [blame] | 25 | |
danilchap | 3c6abd2 | 2017-09-06 05:46:29 -0700 | [diff] [blame] | 26 | class RTC_SCOPED_LOCKABLE MarkProcessingCritScope { |
andresp | cdf6172 | 2016-07-08 02:45:40 -0700 | [diff] [blame] | 27 | public: |
jbauch | 5b36173 | 2017-07-06 23:51:37 -0700 | [diff] [blame] | 28 | MarkProcessingCritScope(const CriticalSection* cs, size_t* processing) |
danilchap | 3c6abd2 | 2017-09-06 05:46:29 -0700 | [diff] [blame] | 29 | RTC_EXCLUSIVE_LOCK_FUNCTION(cs) |
jbauch | 5b36173 | 2017-07-06 23:51:37 -0700 | [diff] [blame] | 30 | : cs_(cs), processing_(processing) { |
andresp | cdf6172 | 2016-07-08 02:45:40 -0700 | [diff] [blame] | 31 | cs_->Enter(); |
jbauch | 5b36173 | 2017-07-06 23:51:37 -0700 | [diff] [blame] | 32 | *processing_ += 1; |
andresp | cdf6172 | 2016-07-08 02:45:40 -0700 | [diff] [blame] | 33 | } |
| 34 | |
danilchap | 3c6abd2 | 2017-09-06 05:46:29 -0700 | [diff] [blame] | 35 | ~MarkProcessingCritScope() RTC_UNLOCK_FUNCTION() { |
jbauch | 5b36173 | 2017-07-06 23:51:37 -0700 | [diff] [blame] | 36 | *processing_ -= 1; |
andresp | cdf6172 | 2016-07-08 02:45:40 -0700 | [diff] [blame] | 37 | cs_->Leave(); |
| 38 | } |
| 39 | |
| 40 | private: |
| 41 | const CriticalSection* const cs_; |
jbauch | 5b36173 | 2017-07-06 23:51:37 -0700 | [diff] [blame] | 42 | size_t* processing_; |
andresp | cdf6172 | 2016-07-08 02:45:40 -0700 | [diff] [blame] | 43 | |
jbauch | 5b36173 | 2017-07-06 23:51:37 -0700 | [diff] [blame] | 44 | RTC_DISALLOW_COPY_AND_ASSIGN(MarkProcessingCritScope); |
andresp | cdf6172 | 2016-07-08 02:45:40 -0700 | [diff] [blame] | 45 | }; |
| 46 | } // namespace |
| 47 | |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 +0000 | [diff] [blame] | 48 | //------------------------------------------------------------------ |
| 49 | // MessageQueueManager |
| 50 | |
deadbeef | 37f5ecf | 2017-02-27 14:06:41 -0800 | [diff] [blame] | 51 | MessageQueueManager* MessageQueueManager::instance_ = nullptr; |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 +0000 | [diff] [blame] | 52 | |
| 53 | MessageQueueManager* MessageQueueManager::Instance() { |
| 54 | // Note: This is not thread safe, but it is first called before threads are |
| 55 | // spawned. |
| 56 | if (!instance_) |
| 57 | instance_ = new MessageQueueManager; |
| 58 | return instance_; |
| 59 | } |
| 60 | |
| 61 | bool MessageQueueManager::IsInitialized() { |
deadbeef | 37f5ecf | 2017-02-27 14:06:41 -0800 | [diff] [blame] | 62 | return instance_ != nullptr; |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 +0000 | [diff] [blame] | 63 | } |
| 64 | |
jbauch | 5b36173 | 2017-07-06 23:51:37 -0700 | [diff] [blame] | 65 | MessageQueueManager::MessageQueueManager() : processing_(0) {} |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 +0000 | [diff] [blame] | 66 | |
| 67 | MessageQueueManager::~MessageQueueManager() { |
| 68 | } |
| 69 | |
| 70 | void MessageQueueManager::Add(MessageQueue *message_queue) { |
| 71 | return Instance()->AddInternal(message_queue); |
| 72 | } |
| 73 | void MessageQueueManager::AddInternal(MessageQueue *message_queue) { |
jbauch | 5b36173 | 2017-07-06 23:51:37 -0700 | [diff] [blame] | 74 | CritScope cs(&crit_); |
| 75 | // Prevent changes while the list of message queues is processed. |
| 76 | RTC_DCHECK_EQ(processing_, 0); |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 +0000 | [diff] [blame] | 77 | message_queues_.push_back(message_queue); |
| 78 | } |
| 79 | |
| 80 | void MessageQueueManager::Remove(MessageQueue *message_queue) { |
| 81 | // If there isn't a message queue manager instance, then there isn't a queue |
| 82 | // to remove. |
| 83 | if (!instance_) return; |
| 84 | return Instance()->RemoveInternal(message_queue); |
| 85 | } |
| 86 | void MessageQueueManager::RemoveInternal(MessageQueue *message_queue) { |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 +0000 | [diff] [blame] | 87 | // If this is the last MessageQueue, destroy the manager as well so that |
| 88 | // we don't leak this object at program shutdown. As mentioned above, this is |
| 89 | // not thread-safe, but this should only happen at program termination (when |
| 90 | // the ThreadManager is destroyed, and threads are no longer active). |
| 91 | bool destroy = false; |
| 92 | { |
jbauch | 5b36173 | 2017-07-06 23:51:37 -0700 | [diff] [blame] | 93 | CritScope cs(&crit_); |
| 94 | // Prevent changes while the list of message queues is processed. |
| 95 | RTC_DCHECK_EQ(processing_, 0); |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 +0000 | [diff] [blame] | 96 | std::vector<MessageQueue *>::iterator iter; |
| 97 | iter = std::find(message_queues_.begin(), message_queues_.end(), |
| 98 | message_queue); |
| 99 | if (iter != message_queues_.end()) { |
| 100 | message_queues_.erase(iter); |
| 101 | } |
| 102 | destroy = message_queues_.empty(); |
| 103 | } |
| 104 | if (destroy) { |
deadbeef | 37f5ecf | 2017-02-27 14:06:41 -0800 | [diff] [blame] | 105 | instance_ = nullptr; |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 +0000 | [diff] [blame] | 106 | delete this; |
| 107 | } |
| 108 | } |
| 109 | |
| 110 | void MessageQueueManager::Clear(MessageHandler *handler) { |
| 111 | // If there isn't a message queue manager instance, then there aren't any |
| 112 | // queues to remove this handler from. |
| 113 | if (!instance_) return; |
| 114 | return Instance()->ClearInternal(handler); |
| 115 | } |
| 116 | void MessageQueueManager::ClearInternal(MessageHandler *handler) { |
jbauch | 5b36173 | 2017-07-06 23:51:37 -0700 | [diff] [blame] | 117 | // Deleted objects may cause re-entrant calls to ClearInternal. This is |
| 118 | // allowed as the list of message queues does not change while queues are |
| 119 | // cleared. |
| 120 | MarkProcessingCritScope cs(&crit_, &processing_); |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 +0000 | [diff] [blame] | 121 | std::vector<MessageQueue *>::iterator iter; |
jbauch | 5b36173 | 2017-07-06 23:51:37 -0700 | [diff] [blame] | 122 | for (MessageQueue* queue : message_queues_) { |
| 123 | queue->Clear(handler); |
| 124 | } |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 +0000 | [diff] [blame] | 125 | } |
| 126 | |
deadbeef | f5f03e8 | 2016-06-06 11:16:06 -0700 | [diff] [blame] | 127 | void MessageQueueManager::ProcessAllMessageQueues() { |
Taylor Brandstetter | b3c6810 | 2016-05-27 14:15:43 -0700 | [diff] [blame] | 128 | if (!instance_) { |
| 129 | return; |
| 130 | } |
deadbeef | f5f03e8 | 2016-06-06 11:16:06 -0700 | [diff] [blame] | 131 | return Instance()->ProcessAllMessageQueuesInternal(); |
Taylor Brandstetter | b3c6810 | 2016-05-27 14:15:43 -0700 | [diff] [blame] | 132 | } |
| 133 | |
deadbeef | f5f03e8 | 2016-06-06 11:16:06 -0700 | [diff] [blame] | 134 | void MessageQueueManager::ProcessAllMessageQueuesInternal() { |
Taylor Brandstetter | fe7d091 | 2016-09-15 17:47:42 -0700 | [diff] [blame] | 135 | // This works by posting a delayed message at the current time and waiting |
| 136 | // for it to be dispatched on all queues, which will ensure that all messages |
| 137 | // that came before it were also dispatched. |
| 138 | volatile int queues_not_done = 0; |
| 139 | |
| 140 | // This class is used so that whether the posted message is processed, or the |
| 141 | // message queue is simply cleared, queues_not_done gets decremented. |
| 142 | class ScopedIncrement : public MessageData { |
| 143 | public: |
| 144 | ScopedIncrement(volatile int* value) : value_(value) { |
| 145 | AtomicOps::Increment(value_); |
| 146 | } |
| 147 | ~ScopedIncrement() override { AtomicOps::Decrement(value_); } |
| 148 | |
| 149 | private: |
| 150 | volatile int* value_; |
| 151 | }; |
| 152 | |
deadbeef | f5f03e8 | 2016-06-06 11:16:06 -0700 | [diff] [blame] | 153 | { |
jbauch | 5b36173 | 2017-07-06 23:51:37 -0700 | [diff] [blame] | 154 | MarkProcessingCritScope cs(&crit_, &processing_); |
deadbeef | f5f03e8 | 2016-06-06 11:16:06 -0700 | [diff] [blame] | 155 | for (MessageQueue* queue : message_queues_) { |
pthatcher | 1749bc3 | 2017-02-08 13:18:00 -0800 | [diff] [blame] | 156 | if (!queue->IsProcessingMessages()) { |
| 157 | // If the queue is not processing messages, it can |
| 158 | // be ignored. If we tried to post a message to it, it would be dropped |
| 159 | // or ignored. |
Taylor Brandstetter | fe7d091 | 2016-09-15 17:47:42 -0700 | [diff] [blame] | 160 | continue; |
| 161 | } |
| 162 | queue->PostDelayed(RTC_FROM_HERE, 0, nullptr, MQID_DISPOSE, |
| 163 | new ScopedIncrement(&queues_not_done)); |
deadbeef | f5f03e8 | 2016-06-06 11:16:06 -0700 | [diff] [blame] | 164 | } |
Taylor Brandstetter | b3c6810 | 2016-05-27 14:15:43 -0700 | [diff] [blame] | 165 | } |
deadbeef | f5f03e8 | 2016-06-06 11:16:06 -0700 | [diff] [blame] | 166 | // Note: One of the message queues may have been on this thread, which is why |
| 167 | // we can't synchronously wait for queues_not_done to go to 0; we need to |
| 168 | // process messages as well. |
| 169 | while (AtomicOps::AcquireLoad(&queues_not_done) > 0) { |
| 170 | rtc::Thread::Current()->ProcessMessages(0); |
| 171 | } |
Taylor Brandstetter | b3c6810 | 2016-05-27 14:15:43 -0700 | [diff] [blame] | 172 | } |
| 173 | |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 +0000 | [diff] [blame] | 174 | //------------------------------------------------------------------ |
| 175 | // MessageQueue |
jbauch | 25d1f28 | 2016-02-05 00:25:02 -0800 | [diff] [blame] | 176 | MessageQueue::MessageQueue(SocketServer* ss, bool init_queue) |
André Susano Pinto | 02a5797 | 2016-07-22 13:30:05 +0200 | [diff] [blame] | 177 | : fPeekKeep_(false), |
| 178 | dmsgq_next_num_(0), |
| 179 | fInitialized_(false), |
| 180 | fDestroyed_(false), |
| 181 | stop_(0), |
| 182 | ss_(ss) { |
danilchap | bebf54c | 2016-04-28 01:32:48 -0700 | [diff] [blame] | 183 | RTC_DCHECK(ss); |
| 184 | // Currently, MessageQueue holds a socket server, and is the base class for |
| 185 | // Thread. It seems like it makes more sense for Thread to hold the socket |
| 186 | // server, and provide it to the MessageQueue, since the Thread controls |
| 187 | // the I/O model, and MQ is agnostic to those details. Anyway, this causes |
| 188 | // messagequeue_unittest to depend on network libraries... yuck. |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 +0000 | [diff] [blame] | 189 | ss_->SetMessageQueue(this); |
jbauch | 25d1f28 | 2016-02-05 00:25:02 -0800 | [diff] [blame] | 190 | if (init_queue) { |
| 191 | DoInit(); |
| 192 | } |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 +0000 | [diff] [blame] | 193 | } |
| 194 | |
danilchap | bebf54c | 2016-04-28 01:32:48 -0700 | [diff] [blame] | 195 | MessageQueue::MessageQueue(std::unique_ptr<SocketServer> ss, bool init_queue) |
| 196 | : MessageQueue(ss.get(), init_queue) { |
| 197 | own_ss_ = std::move(ss); |
| 198 | } |
| 199 | |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 +0000 | [diff] [blame] | 200 | MessageQueue::~MessageQueue() { |
jbauch | 25d1f28 | 2016-02-05 00:25:02 -0800 | [diff] [blame] | 201 | DoDestroy(); |
| 202 | } |
| 203 | |
| 204 | void MessageQueue::DoInit() { |
| 205 | if (fInitialized_) { |
| 206 | return; |
| 207 | } |
| 208 | |
| 209 | fInitialized_ = true; |
| 210 | MessageQueueManager::Add(this); |
| 211 | } |
| 212 | |
| 213 | void MessageQueue::DoDestroy() { |
| 214 | if (fDestroyed_) { |
| 215 | return; |
| 216 | } |
| 217 | |
| 218 | fDestroyed_ = true; |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 +0000 | [diff] [blame] | 219 | // The signal is done from here to ensure |
| 220 | // that it always gets called when the queue |
| 221 | // is going away. |
| 222 | SignalQueueDestroyed(); |
henrike@webrtc.org | 99b4162 | 2014-05-21 20:42:17 +0000 | [diff] [blame] | 223 | MessageQueueManager::Remove(this); |
deadbeef | 37f5ecf | 2017-02-27 14:06:41 -0800 | [diff] [blame] | 224 | Clear(nullptr); |
jbauch | 9ccedc3 | 2016-02-25 01:14:56 -0800 | [diff] [blame] | 225 | |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 +0000 | [diff] [blame] | 226 | if (ss_) { |
deadbeef | 37f5ecf | 2017-02-27 14:06:41 -0800 | [diff] [blame] | 227 | ss_->SetMessageQueue(nullptr); |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 +0000 | [diff] [blame] | 228 | } |
| 229 | } |
| 230 | |
jbauch | 9ccedc3 | 2016-02-25 01:14:56 -0800 | [diff] [blame] | 231 | SocketServer* MessageQueue::socketserver() { |
jbauch | 9ccedc3 | 2016-02-25 01:14:56 -0800 | [diff] [blame] | 232 | return ss_; |
| 233 | } |
| 234 | |
jbauch | 9ccedc3 | 2016-02-25 01:14:56 -0800 | [diff] [blame] | 235 | void MessageQueue::WakeUpSocketServer() { |
jbauch | 9ccedc3 | 2016-02-25 01:14:56 -0800 | [diff] [blame] | 236 | ss_->WakeUp(); |
| 237 | } |
| 238 | |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 +0000 | [diff] [blame] | 239 | void MessageQueue::Quit() { |
André Susano Pinto | 02a5797 | 2016-07-22 13:30:05 +0200 | [diff] [blame] | 240 | AtomicOps::ReleaseStore(&stop_, 1); |
jbauch | 9ccedc3 | 2016-02-25 01:14:56 -0800 | [diff] [blame] | 241 | WakeUpSocketServer(); |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 +0000 | [diff] [blame] | 242 | } |
| 243 | |
| 244 | bool MessageQueue::IsQuitting() { |
André Susano Pinto | 02a5797 | 2016-07-22 13:30:05 +0200 | [diff] [blame] | 245 | return AtomicOps::AcquireLoad(&stop_) != 0; |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 +0000 | [diff] [blame] | 246 | } |
| 247 | |
pthatcher | 1749bc3 | 2017-02-08 13:18:00 -0800 | [diff] [blame] | 248 | bool MessageQueue::IsProcessingMessages() { |
| 249 | return !IsQuitting(); |
| 250 | } |
| 251 | |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 +0000 | [diff] [blame] | 252 | void MessageQueue::Restart() { |
André Susano Pinto | 02a5797 | 2016-07-22 13:30:05 +0200 | [diff] [blame] | 253 | AtomicOps::ReleaseStore(&stop_, 0); |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 +0000 | [diff] [blame] | 254 | } |
| 255 | |
| 256 | bool MessageQueue::Peek(Message *pmsg, int cmsWait) { |
| 257 | if (fPeekKeep_) { |
| 258 | *pmsg = msgPeek_; |
| 259 | return true; |
| 260 | } |
| 261 | if (!Get(pmsg, cmsWait)) |
| 262 | return false; |
| 263 | msgPeek_ = *pmsg; |
| 264 | fPeekKeep_ = true; |
| 265 | return true; |
| 266 | } |
| 267 | |
| 268 | bool MessageQueue::Get(Message *pmsg, int cmsWait, bool process_io) { |
| 269 | // Return and clear peek if present |
| 270 | // Always return the peek if it exists so there is Peek/Get symmetry |
| 271 | |
| 272 | if (fPeekKeep_) { |
| 273 | *pmsg = msgPeek_; |
| 274 | fPeekKeep_ = false; |
| 275 | return true; |
| 276 | } |
| 277 | |
| 278 | // Get w/wait + timer scan / dispatch + socket / event multiplexer dispatch |
| 279 | |
Honghai Zhang | 82d7862 | 2016-05-06 11:29:15 -0700 | [diff] [blame] | 280 | int64_t cmsTotal = cmsWait; |
| 281 | int64_t cmsElapsed = 0; |
| 282 | int64_t msStart = TimeMillis(); |
| 283 | int64_t msCurrent = msStart; |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 +0000 | [diff] [blame] | 284 | while (true) { |
| 285 | // Check for sent messages |
| 286 | ReceiveSends(); |
| 287 | |
| 288 | // Check for posted events |
Honghai Zhang | 82d7862 | 2016-05-06 11:29:15 -0700 | [diff] [blame] | 289 | int64_t cmsDelayNext = kForever; |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 +0000 | [diff] [blame] | 290 | bool first_pass = true; |
| 291 | while (true) { |
| 292 | // All queue operations need to be locked, but nothing else in this loop |
| 293 | // (specifically handling disposed message) can happen inside the crit. |
| 294 | // Otherwise, disposed MessageHandlers will cause deadlocks. |
| 295 | { |
| 296 | CritScope cs(&crit_); |
| 297 | // On the first pass, check for delayed messages that have been |
| 298 | // triggered and calculate the next trigger time. |
| 299 | if (first_pass) { |
| 300 | first_pass = false; |
| 301 | while (!dmsgq_.empty()) { |
Honghai Zhang | 82d7862 | 2016-05-06 11:29:15 -0700 | [diff] [blame] | 302 | if (msCurrent < dmsgq_.top().msTrigger_) { |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 +0000 | [diff] [blame] | 303 | cmsDelayNext = TimeDiff(dmsgq_.top().msTrigger_, msCurrent); |
| 304 | break; |
| 305 | } |
| 306 | msgq_.push_back(dmsgq_.top().msg_); |
| 307 | dmsgq_.pop(); |
| 308 | } |
| 309 | } |
| 310 | // Pull a message off the message queue, if available. |
| 311 | if (msgq_.empty()) { |
| 312 | break; |
| 313 | } else { |
| 314 | *pmsg = msgq_.front(); |
| 315 | msgq_.pop_front(); |
| 316 | } |
| 317 | } // crit_ is released here. |
| 318 | |
| 319 | // Log a warning for time-sensitive messages that we're late to deliver. |
| 320 | if (pmsg->ts_sensitive) { |
Honghai Zhang | 82d7862 | 2016-05-06 11:29:15 -0700 | [diff] [blame] | 321 | int64_t delay = TimeDiff(msCurrent, pmsg->ts_sensitive); |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 +0000 | [diff] [blame] | 322 | if (delay > 0) { |
| 323 | LOG_F(LS_WARNING) << "id: " << pmsg->message_id << " delay: " |
| 324 | << (delay + kMaxMsgLatency) << "ms"; |
| 325 | } |
| 326 | } |
| 327 | // If this was a dispose message, delete it and skip it. |
| 328 | if (MQID_DISPOSE == pmsg->message_id) { |
deadbeef | 37f5ecf | 2017-02-27 14:06:41 -0800 | [diff] [blame] | 329 | RTC_DCHECK(nullptr == pmsg->phandler); |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 +0000 | [diff] [blame] | 330 | delete pmsg->pdata; |
| 331 | *pmsg = Message(); |
| 332 | continue; |
| 333 | } |
| 334 | return true; |
| 335 | } |
| 336 | |
André Susano Pinto | 02a5797 | 2016-07-22 13:30:05 +0200 | [diff] [blame] | 337 | if (IsQuitting()) |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 +0000 | [diff] [blame] | 338 | break; |
| 339 | |
| 340 | // Which is shorter, the delay wait or the asked wait? |
| 341 | |
Honghai Zhang | 82d7862 | 2016-05-06 11:29:15 -0700 | [diff] [blame] | 342 | int64_t cmsNext; |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 +0000 | [diff] [blame] | 343 | if (cmsWait == kForever) { |
| 344 | cmsNext = cmsDelayNext; |
| 345 | } else { |
Honghai Zhang | 82d7862 | 2016-05-06 11:29:15 -0700 | [diff] [blame] | 346 | cmsNext = std::max<int64_t>(0, cmsTotal - cmsElapsed); |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 +0000 | [diff] [blame] | 347 | if ((cmsDelayNext != kForever) && (cmsDelayNext < cmsNext)) |
| 348 | cmsNext = cmsDelayNext; |
| 349 | } |
| 350 | |
jbauch | 9ccedc3 | 2016-02-25 01:14:56 -0800 | [diff] [blame] | 351 | { |
| 352 | // Wait and multiplex in the meantime |
Honghai Zhang | 82d7862 | 2016-05-06 11:29:15 -0700 | [diff] [blame] | 353 | if (!ss_->Wait(static_cast<int>(cmsNext), process_io)) |
jbauch | 9ccedc3 | 2016-02-25 01:14:56 -0800 | [diff] [blame] | 354 | return false; |
| 355 | } |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 +0000 | [diff] [blame] | 356 | |
| 357 | // If the specified timeout expired, return |
| 358 | |
Honghai Zhang | 82d7862 | 2016-05-06 11:29:15 -0700 | [diff] [blame] | 359 | msCurrent = TimeMillis(); |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 +0000 | [diff] [blame] | 360 | cmsElapsed = TimeDiff(msCurrent, msStart); |
| 361 | if (cmsWait != kForever) { |
| 362 | if (cmsElapsed >= cmsWait) |
| 363 | return false; |
| 364 | } |
| 365 | } |
| 366 | return false; |
| 367 | } |
| 368 | |
| 369 | void MessageQueue::ReceiveSends() { |
| 370 | } |
| 371 | |
Taylor Brandstetter | 5d97a9a | 2016-06-10 14:17:27 -0700 | [diff] [blame] | 372 | void MessageQueue::Post(const Location& posted_from, |
| 373 | MessageHandler* phandler, |
Peter Boström | 0c4e06b | 2015-10-07 12:23:21 +0200 | [diff] [blame] | 374 | uint32_t id, |
| 375 | MessageData* pdata, |
| 376 | bool time_sensitive) { |
André Susano Pinto | 02a5797 | 2016-07-22 13:30:05 +0200 | [diff] [blame] | 377 | if (IsQuitting()) |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 +0000 | [diff] [blame] | 378 | return; |
| 379 | |
| 380 | // Keep thread safe |
| 381 | // Add the message to the end of the queue |
| 382 | // Signal for the multiplexer to return |
| 383 | |
jbauch | 9ccedc3 | 2016-02-25 01:14:56 -0800 | [diff] [blame] | 384 | { |
| 385 | CritScope cs(&crit_); |
| 386 | Message msg; |
Taylor Brandstetter | 5d97a9a | 2016-06-10 14:17:27 -0700 | [diff] [blame] | 387 | msg.posted_from = posted_from; |
jbauch | 9ccedc3 | 2016-02-25 01:14:56 -0800 | [diff] [blame] | 388 | msg.phandler = phandler; |
| 389 | msg.message_id = id; |
| 390 | msg.pdata = pdata; |
| 391 | if (time_sensitive) { |
Honghai Zhang | 82d7862 | 2016-05-06 11:29:15 -0700 | [diff] [blame] | 392 | msg.ts_sensitive = TimeMillis() + kMaxMsgLatency; |
jbauch | 9ccedc3 | 2016-02-25 01:14:56 -0800 | [diff] [blame] | 393 | } |
| 394 | msgq_.push_back(msg); |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 +0000 | [diff] [blame] | 395 | } |
jbauch | 9ccedc3 | 2016-02-25 01:14:56 -0800 | [diff] [blame] | 396 | WakeUpSocketServer(); |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 +0000 | [diff] [blame] | 397 | } |
| 398 | |
Taylor Brandstetter | 5d97a9a | 2016-06-10 14:17:27 -0700 | [diff] [blame] | 399 | void MessageQueue::PostDelayed(const Location& posted_from, |
| 400 | int cmsDelay, |
kwiberg@webrtc.org | 67186fe | 2015-03-09 22:21:53 +0000 | [diff] [blame] | 401 | MessageHandler* phandler, |
Peter Boström | 0c4e06b | 2015-10-07 12:23:21 +0200 | [diff] [blame] | 402 | uint32_t id, |
kwiberg@webrtc.org | 67186fe | 2015-03-09 22:21:53 +0000 | [diff] [blame] | 403 | MessageData* pdata) { |
Taylor Brandstetter | 5d97a9a | 2016-06-10 14:17:27 -0700 | [diff] [blame] | 404 | return DoDelayPost(posted_from, cmsDelay, TimeAfter(cmsDelay), phandler, id, |
| 405 | pdata); |
kwiberg@webrtc.org | 67186fe | 2015-03-09 22:21:53 +0000 | [diff] [blame] | 406 | } |
| 407 | |
Taylor Brandstetter | 5d97a9a | 2016-06-10 14:17:27 -0700 | [diff] [blame] | 408 | void MessageQueue::PostAt(const Location& posted_from, |
| 409 | uint32_t tstamp, |
kwiberg@webrtc.org | 67186fe | 2015-03-09 22:21:53 +0000 | [diff] [blame] | 410 | MessageHandler* phandler, |
Peter Boström | 0c4e06b | 2015-10-07 12:23:21 +0200 | [diff] [blame] | 411 | uint32_t id, |
kwiberg@webrtc.org | 67186fe | 2015-03-09 22:21:53 +0000 | [diff] [blame] | 412 | MessageData* pdata) { |
Honghai Zhang | 82d7862 | 2016-05-06 11:29:15 -0700 | [diff] [blame] | 413 | // This should work even if it is used (unexpectedly). |
Taylor Brandstetter | 2b3bf6b | 2016-05-19 14:57:31 -0700 | [diff] [blame] | 414 | int64_t delay = static_cast<uint32_t>(TimeMillis()) - tstamp; |
Taylor Brandstetter | 5d97a9a | 2016-06-10 14:17:27 -0700 | [diff] [blame] | 415 | return DoDelayPost(posted_from, delay, tstamp, phandler, id, pdata); |
Honghai Zhang | 82d7862 | 2016-05-06 11:29:15 -0700 | [diff] [blame] | 416 | } |
| 417 | |
Taylor Brandstetter | 5d97a9a | 2016-06-10 14:17:27 -0700 | [diff] [blame] | 418 | void MessageQueue::PostAt(const Location& posted_from, |
| 419 | int64_t tstamp, |
Honghai Zhang | 82d7862 | 2016-05-06 11:29:15 -0700 | [diff] [blame] | 420 | MessageHandler* phandler, |
| 421 | uint32_t id, |
| 422 | MessageData* pdata) { |
Taylor Brandstetter | 5d97a9a | 2016-06-10 14:17:27 -0700 | [diff] [blame] | 423 | return DoDelayPost(posted_from, TimeUntil(tstamp), tstamp, phandler, id, |
| 424 | pdata); |
kwiberg@webrtc.org | 67186fe | 2015-03-09 22:21:53 +0000 | [diff] [blame] | 425 | } |
| 426 | |
Taylor Brandstetter | 5d97a9a | 2016-06-10 14:17:27 -0700 | [diff] [blame] | 427 | void MessageQueue::DoDelayPost(const Location& posted_from, |
| 428 | int64_t cmsDelay, |
Honghai Zhang | 82d7862 | 2016-05-06 11:29:15 -0700 | [diff] [blame] | 429 | int64_t tstamp, |
Peter Boström | 0c4e06b | 2015-10-07 12:23:21 +0200 | [diff] [blame] | 430 | MessageHandler* phandler, |
| 431 | uint32_t id, |
| 432 | MessageData* pdata) { |
André Susano Pinto | 02a5797 | 2016-07-22 13:30:05 +0200 | [diff] [blame] | 433 | if (IsQuitting()) { |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 +0000 | [diff] [blame] | 434 | return; |
Taylor Brandstetter | 2b3bf6b | 2016-05-19 14:57:31 -0700 | [diff] [blame] | 435 | } |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 +0000 | [diff] [blame] | 436 | |
| 437 | // Keep thread safe |
| 438 | // Add to the priority queue. Gets sorted soonest first. |
| 439 | // Signal for the multiplexer to return. |
| 440 | |
jbauch | 9ccedc3 | 2016-02-25 01:14:56 -0800 | [diff] [blame] | 441 | { |
| 442 | CritScope cs(&crit_); |
| 443 | Message msg; |
Taylor Brandstetter | 5d97a9a | 2016-06-10 14:17:27 -0700 | [diff] [blame] | 444 | msg.posted_from = posted_from; |
jbauch | 9ccedc3 | 2016-02-25 01:14:56 -0800 | [diff] [blame] | 445 | msg.phandler = phandler; |
| 446 | msg.message_id = id; |
| 447 | msg.pdata = pdata; |
| 448 | DelayedMessage dmsg(cmsDelay, tstamp, dmsgq_next_num_, msg); |
| 449 | dmsgq_.push(dmsg); |
| 450 | // If this message queue processes 1 message every millisecond for 50 days, |
| 451 | // we will wrap this number. Even then, only messages with identical times |
| 452 | // will be misordered, and then only briefly. This is probably ok. |
nisse | 7ce109a | 2017-01-31 00:57:56 -0800 | [diff] [blame] | 453 | ++dmsgq_next_num_; |
| 454 | RTC_DCHECK_NE(0, dmsgq_next_num_); |
jbauch | 9ccedc3 | 2016-02-25 01:14:56 -0800 | [diff] [blame] | 455 | } |
| 456 | WakeUpSocketServer(); |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 +0000 | [diff] [blame] | 457 | } |
| 458 | |
| 459 | int MessageQueue::GetDelay() { |
| 460 | CritScope cs(&crit_); |
| 461 | |
| 462 | if (!msgq_.empty()) |
| 463 | return 0; |
| 464 | |
| 465 | if (!dmsgq_.empty()) { |
| 466 | int delay = TimeUntil(dmsgq_.top().msTrigger_); |
| 467 | if (delay < 0) |
| 468 | delay = 0; |
| 469 | return delay; |
| 470 | } |
| 471 | |
| 472 | return kForever; |
| 473 | } |
| 474 | |
Peter Boström | 0c4e06b | 2015-10-07 12:23:21 +0200 | [diff] [blame] | 475 | void MessageQueue::Clear(MessageHandler* phandler, |
| 476 | uint32_t id, |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 +0000 | [diff] [blame] | 477 | MessageList* removed) { |
| 478 | CritScope cs(&crit_); |
| 479 | |
| 480 | // Remove messages with phandler |
| 481 | |
| 482 | if (fPeekKeep_ && msgPeek_.Match(phandler, id)) { |
| 483 | if (removed) { |
| 484 | removed->push_back(msgPeek_); |
| 485 | } else { |
| 486 | delete msgPeek_.pdata; |
| 487 | } |
| 488 | fPeekKeep_ = false; |
| 489 | } |
| 490 | |
| 491 | // Remove from ordered message queue |
| 492 | |
| 493 | for (MessageList::iterator it = msgq_.begin(); it != msgq_.end();) { |
| 494 | if (it->Match(phandler, id)) { |
| 495 | if (removed) { |
| 496 | removed->push_back(*it); |
| 497 | } else { |
| 498 | delete it->pdata; |
| 499 | } |
| 500 | it = msgq_.erase(it); |
| 501 | } else { |
| 502 | ++it; |
| 503 | } |
| 504 | } |
| 505 | |
| 506 | // Remove from priority queue. Not directly iterable, so use this approach |
decurtis@webrtc.org | 2af3057 | 2015-02-21 01:59:50 +0000 | [diff] [blame] | 507 | |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 +0000 | [diff] [blame] | 508 | PriorityQueue::container_type::iterator new_end = dmsgq_.container().begin(); |
| 509 | for (PriorityQueue::container_type::iterator it = new_end; |
| 510 | it != dmsgq_.container().end(); ++it) { |
| 511 | if (it->msg_.Match(phandler, id)) { |
| 512 | if (removed) { |
| 513 | removed->push_back(it->msg_); |
| 514 | } else { |
| 515 | delete it->msg_.pdata; |
| 516 | } |
| 517 | } else { |
decurtis@webrtc.org | 2af3057 | 2015-02-21 01:59:50 +0000 | [diff] [blame] | 518 | *new_end++ = *it; |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 +0000 | [diff] [blame] | 519 | } |
| 520 | } |
| 521 | dmsgq_.container().erase(new_end, dmsgq_.container().end()); |
| 522 | dmsgq_.reheap(); |
| 523 | } |
| 524 | |
| 525 | void MessageQueue::Dispatch(Message *pmsg) { |
Taylor Brandstetter | 5d97a9a | 2016-06-10 14:17:27 -0700 | [diff] [blame] | 526 | TRACE_EVENT2("webrtc", "MessageQueue::Dispatch", "src_file_and_line", |
| 527 | pmsg->posted_from.file_and_line(), "src_func", |
| 528 | pmsg->posted_from.function_name()); |
| 529 | int64_t start_time = TimeMillis(); |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 +0000 | [diff] [blame] | 530 | pmsg->phandler->OnMessage(pmsg); |
Taylor Brandstetter | 5d97a9a | 2016-06-10 14:17:27 -0700 | [diff] [blame] | 531 | int64_t end_time = TimeMillis(); |
| 532 | int64_t diff = TimeDiff(end_time, start_time); |
| 533 | if (diff >= kSlowDispatchLoggingThreshold) { |
| 534 | LOG(LS_INFO) << "Message took " << diff << "ms to dispatch. Posted from: " |
| 535 | << pmsg->posted_from.ToString(); |
| 536 | } |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 +0000 | [diff] [blame] | 537 | } |
| 538 | |
henrike@webrtc.org | f048872 | 2014-05-13 18:00:26 +0000 | [diff] [blame] | 539 | } // namespace rtc |