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 "talk/p2p/client/socketmonitor.h"
29
30#include "webrtc/base/common.h"
31
32namespace cricket {
33
34enum {
35  MSG_MONITOR_POLL,
36  MSG_MONITOR_START,
37  MSG_MONITOR_STOP,
38  MSG_MONITOR_SIGNAL
39};
40
41SocketMonitor::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
50SocketMonitor::~SocketMonitor() {
51  channel_thread_->Clear(this);
52  monitoring_thread_->Clear(this);
53}
54
55void SocketMonitor::Start(int milliseconds) {
56  rate_ = milliseconds;
57  if (rate_ < 250)
58    rate_ = 250;
59  channel_thread_->Post(this, MSG_MONITOR_START);
60}
61
62void SocketMonitor::Stop() {
63  channel_thread_->Post(this, MSG_MONITOR_STOP);
64}
65
66void 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
101void 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
115