src/third_party/libjingle/source/talk/p2p/client/socketmonitor.cc

   1 /*
   2  * libjingle
   3  * Copyright 2004--2005, Google Inc.
   4  *
   5  * Redistribution and use in source and binary forms, with or without
   6  * modification, are permitted provided that the following conditions are met:
   7  *
   8  *  1. Redistributions of source code must retain the above copyright notice,
   9  *     this list of conditions and the following disclaimer.
  10  *  2. Redistributions in binary form must reproduce the above copyright notice,
  11  *     this list of conditions and the following disclaimer in the documentation
  12  *     and/or other materials provided with the distribution.
  13  *  3. The name of the author may not be used to endorse or promote products
  14  *     derived from this software without specific prior written permission.
  15  *
  16  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR IMPLIED
  17  * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
  18  * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO
  19  * EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  20  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
  21  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
  22  * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
  23  * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
  24  * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
  25  * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  26  */
  27
  28 #include "webrtc/p2p/client/socketmonitor.h"
  29
  30 #include "webrtc/base/common.h"
  31
  32 namespace cricket {
  33
  34 enum {
  35   MSG_MONITOR_POLL,
  36   MSG_MONITOR_START,
  37   MSG_MONITOR_STOP,
  38   MSG_MONITOR_SIGNAL
  39 };
  40
  41 SocketMonitor::SocketMonitor(TransportChannel* channel,
  42                              rtc::Thread* worker_thread,
  43                              rtc::Thread* monitor_thread) {
  44   channel_ = channel;
  45   channel_thread_ = worker_thread;
  46   monitoring_thread_ = monitor_thread;
  47   monitoring_ = false;
  48 }
  49
  50 SocketMonitor::~SocketMonitor() {
  51   channel_thread_->Clear(this);
  52   monitoring_thread_->Clear(this);
  53 }
  54
  55 void SocketMonitor::Start(int milliseconds) {
  56   rate_ = milliseconds;
  57   if (rate_ < 250)
  58     rate_ = 250;
  59   channel_thread_->Post(this, MSG_MONITOR_START);
  60 }
  61
  62 void SocketMonitor::Stop() {
  63   channel_thread_->Post(this, MSG_MONITOR_STOP);
  64 }
  65
  66 void SocketMonitor::OnMessage(rtc::Message *message) {
  67   rtc::CritScope cs(&crit_);
  68   switch (message->message_id) {
  69     case MSG_MONITOR_START:
  70       ASSERT(rtc::Thread::Current() == channel_thread_);
  71       if (!monitoring_) {
  72         monitoring_ = true;
  73         PollSocket(true);
  74       }
  75       break;
  76
  77     case MSG_MONITOR_STOP:
  78       ASSERT(rtc::Thread::Current() == channel_thread_);
  79       if (monitoring_) {
  80         monitoring_ = false;
  81         channel_thread_->Clear(this);
  82       }
  83       break;
  84
  85     case MSG_MONITOR_POLL:
  86       ASSERT(rtc::Thread::Current() == channel_thread_);
  87       PollSocket(true);
  88       break;
  89
  90     case MSG_MONITOR_SIGNAL: {
  91       ASSERT(rtc::Thread::Current() == monitoring_thread_);
  92       std::vector<ConnectionInfo> infos = connection_infos_;
  93       crit_.Leave();
  94       SignalUpdate(this, infos);
  95       crit_.Enter();
  96       break;
  97     }
  98   }
  99 }
 100
 101 void SocketMonitor::PollSocket(bool poll) {
 102   ASSERT(rtc::Thread::Current() == channel_thread_);
 103   rtc::CritScope cs(&crit_);
 104
 105   // Gather connection infos
 106   channel_->GetStats(&connection_infos_);
 107
 108   // Signal the monitoring thread, start another poll timer
 109   monitoring_thread_->Post(this, MSG_MONITOR_SIGNAL);
 110   if (poll)
 111     channel_thread_->PostDelayed(rate_, this, MSG_MONITOR_POLL);
 112 }
 113
 114 }  // namespace cricket