 d4e598d57a
			
		
	
	d4e598d57a
	
	
	
		
			
			git-svn-id: http://webrtc.googlecode.com/svn/trunk@6799 4adac7df-926f-26a2-2b94-8c16560cd09d
		
			
				
	
	
		
			115 lines
		
	
	
		
			3.5 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			115 lines
		
	
	
		
			3.5 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
| /*
 | |
|  * libjingle
 | |
|  * Copyright 2004--2005, Google Inc.
 | |
|  *
 | |
|  * Redistribution and use in source and binary forms, with or without
 | |
|  * modification, are permitted provided that the following conditions are met:
 | |
|  *
 | |
|  *  1. Redistributions of source code must retain the above copyright notice,
 | |
|  *     this list of conditions and the following disclaimer.
 | |
|  *  2. Redistributions in binary form must reproduce the above copyright notice,
 | |
|  *     this list of conditions and the following disclaimer in the documentation
 | |
|  *     and/or other materials provided with the distribution.
 | |
|  *  3. The name of the author may not be used to endorse or promote products
 | |
|  *     derived from this software without specific prior written permission.
 | |
|  *
 | |
|  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR IMPLIED
 | |
|  * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
 | |
|  * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO
 | |
|  * EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 | |
|  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
 | |
|  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
 | |
|  * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
 | |
|  * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
 | |
|  * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
 | |
|  * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 | |
|  */
 | |
| 
 | |
| #include "talk/p2p/client/socketmonitor.h"
 | |
| 
 | |
| #include "webrtc/base/common.h"
 | |
| 
 | |
| namespace cricket {
 | |
| 
 | |
| enum {
 | |
|   MSG_MONITOR_POLL,
 | |
|   MSG_MONITOR_START,
 | |
|   MSG_MONITOR_STOP,
 | |
|   MSG_MONITOR_SIGNAL
 | |
| };
 | |
| 
 | |
| SocketMonitor::SocketMonitor(TransportChannel* channel,
 | |
|                              rtc::Thread* worker_thread,
 | |
|                              rtc::Thread* monitor_thread) {
 | |
|   channel_ = channel;
 | |
|   channel_thread_ = worker_thread;
 | |
|   monitoring_thread_ = monitor_thread;
 | |
|   monitoring_ = false;
 | |
| }
 | |
| 
 | |
| SocketMonitor::~SocketMonitor() {
 | |
|   channel_thread_->Clear(this);
 | |
|   monitoring_thread_->Clear(this);
 | |
| }
 | |
| 
 | |
| void SocketMonitor::Start(int milliseconds) {
 | |
|   rate_ = milliseconds;
 | |
|   if (rate_ < 250)
 | |
|     rate_ = 250;
 | |
|   channel_thread_->Post(this, MSG_MONITOR_START);
 | |
| }
 | |
| 
 | |
| void SocketMonitor::Stop() {
 | |
|   channel_thread_->Post(this, MSG_MONITOR_STOP);
 | |
| }
 | |
| 
 | |
| void SocketMonitor::OnMessage(rtc::Message *message) {
 | |
|   rtc::CritScope cs(&crit_);
 | |
|   switch (message->message_id) {
 | |
|     case MSG_MONITOR_START:
 | |
|       ASSERT(rtc::Thread::Current() == channel_thread_);
 | |
|       if (!monitoring_) {
 | |
|         monitoring_ = true;
 | |
|         PollSocket(true);
 | |
|       }
 | |
|       break;
 | |
| 
 | |
|     case MSG_MONITOR_STOP:
 | |
|       ASSERT(rtc::Thread::Current() == channel_thread_);
 | |
|       if (monitoring_) {
 | |
|         monitoring_ = false;
 | |
|         channel_thread_->Clear(this);
 | |
|       }
 | |
|       break;
 | |
| 
 | |
|     case MSG_MONITOR_POLL:
 | |
|       ASSERT(rtc::Thread::Current() == channel_thread_);
 | |
|       PollSocket(true);
 | |
|       break;
 | |
| 
 | |
|     case MSG_MONITOR_SIGNAL: {
 | |
|       ASSERT(rtc::Thread::Current() == monitoring_thread_);
 | |
|       std::vector<ConnectionInfo> infos = connection_infos_;
 | |
|       crit_.Leave();
 | |
|       SignalUpdate(this, infos);
 | |
|       crit_.Enter();
 | |
|       break;
 | |
|     }
 | |
|   }
 | |
| }
 | |
| 
 | |
| void SocketMonitor::PollSocket(bool poll) {
 | |
|   ASSERT(rtc::Thread::Current() == channel_thread_);
 | |
|   rtc::CritScope cs(&crit_);
 | |
| 
 | |
|   // Gather connection infos
 | |
|   channel_->GetStats(&connection_infos_);
 | |
| 
 | |
|   // Signal the monitoring thread, start another poll timer
 | |
|   monitoring_thread_->Post(this, MSG_MONITOR_SIGNAL);
 | |
|   if (poll)
 | |
|     channel_thread_->PostDelayed(rate_, this, MSG_MONITOR_POLL);
 | |
| }
 | |
| 
 | |
| }  // namespace cricket
 |