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 "talk/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
SocketMonitor(TransportChannel * channel,talk_base::Thread * worker_thread,talk_base::Thread * monitor_thread)41 SocketMonitor::SocketMonitor(TransportChannel* channel,
42 talk_base::Thread* worker_thread,
43 talk_base::Thread* monitor_thread) {
44 channel_ = channel;
45 channel_thread_ = worker_thread;
46 monitoring_thread_ = monitor_thread;
47 monitoring_ = false;
48 }
49
~SocketMonitor()50 SocketMonitor::~SocketMonitor() {
51 channel_thread_->Clear(this);
52 monitoring_thread_->Clear(this);
53 }
54
Start(int milliseconds)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
Stop()62 void SocketMonitor::Stop() {
63 channel_thread_->Post(this, MSG_MONITOR_STOP);
64 }
65
OnMessage(talk_base::Message * message)66 void SocketMonitor::OnMessage(talk_base::Message *message) {
67 talk_base::CritScope cs(&crit_);
68 switch (message->message_id) {
69 case MSG_MONITOR_START:
70 ASSERT(talk_base::Thread::Current() == channel_thread_);
71 if (!monitoring_) {
72 monitoring_ = true;
73 PollSocket(true);
74 }
75 break;
76
77 case MSG_MONITOR_STOP:
78 ASSERT(talk_base::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(talk_base::Thread::Current() == channel_thread_);
87 PollSocket(true);
88 break;
89
90 case MSG_MONITOR_SIGNAL: {
91 ASSERT(talk_base::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
PollSocket(bool poll)101 void SocketMonitor::PollSocket(bool poll) {
102 ASSERT(talk_base::Thread::Current() == channel_thread_);
103 talk_base::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