1f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch/* 2f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch * libjingle 3f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch * Copyright 2004--2005, Google Inc. 4f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch * 5f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch * Redistribution and use in source and binary forms, with or without 6f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch * modification, are permitted provided that the following conditions are met: 7f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch * 8f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch * 1. Redistributions of source code must retain the above copyright notice, 9f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch * this list of conditions and the following disclaimer. 10f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch * 2. Redistributions in binary form must reproduce the above copyright notice, 11f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch * this list of conditions and the following disclaimer in the documentation 12f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch * and/or other materials provided with the distribution. 13f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch * 3. The name of the author may not be used to endorse or promote products 14f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch * derived from this software without specific prior written permission. 15f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch * 16f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR IMPLIED 17f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF 18f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO 19f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch * EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 20f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, 21f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; 22f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, 23f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR 24f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF 25f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 26f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch */ 27f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch 28f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch#if defined(_MSC_VER) && _MSC_VER < 1300 29f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch#pragma warning(disable:4786) 30f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch#endif 31f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch 32f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch#include "talk/base/asyncudpsocket.h" 33f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch#include "talk/base/logging.h" 34f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch 35f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdochnamespace talk_base { 36f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch 37dc0f95d653279beabeb9817299e2902918ba123eKristian Monsenstatic const int BUF_SIZE = 64 * 1024; 38dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen 39dc0f95d653279beabeb9817299e2902918ba123eKristian MonsenAsyncUDPSocket* AsyncUDPSocket::Create(SocketFactory* factory, 40dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen const SocketAddress& address) { 41dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen scoped_ptr<AsyncSocket> socket(factory->CreateAsyncSocket(SOCK_DGRAM)); 42dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen if (!socket.get()) 43dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen return NULL; 44dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen if (socket->Bind(address)) { 45dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen LOG(LS_INFO) << "Failed to bind UDP socket " << socket->GetError(); 46dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen return NULL; 47dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen } 48dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen return new AsyncUDPSocket(socket.release()); 49dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen} 50f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch 51f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen MurdochAsyncUDPSocket::AsyncUDPSocket(AsyncSocket* socket) 52dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen : socket_(socket) { 53dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen ASSERT(socket_.get() != NULL); 54f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch size_ = BUF_SIZE; 55f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch buf_ = new char[size_]; 56f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch 57f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch // The socket should start out readable but not writable. 58f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch socket_->SignalReadEvent.connect(this, &AsyncUDPSocket::OnReadEvent); 59f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch} 60f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch 61f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen MurdochAsyncUDPSocket::~AsyncUDPSocket() { 62f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch delete [] buf_; 63f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch} 64f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch 65dc0f95d653279beabeb9817299e2902918ba123eKristian MonsenSocketAddress AsyncUDPSocket::GetLocalAddress(bool* allocated) const { 66dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen if (allocated) 67dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen *allocated = true; 68dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen return socket_->GetLocalAddress(); 69dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen} 70dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen 71dc0f95d653279beabeb9817299e2902918ba123eKristian MonsenSocketAddress AsyncUDPSocket::GetRemoteAddress() const { 72dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen return socket_->GetRemoteAddress(); 73dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen} 74dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen 75dc0f95d653279beabeb9817299e2902918ba123eKristian Monsenint AsyncUDPSocket::Send(const void *pv, size_t cb) { 76dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen return socket_->Send(pv, cb); 77dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen} 78dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen 79dc0f95d653279beabeb9817299e2902918ba123eKristian Monsenint AsyncUDPSocket::SendTo( 80dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen const void *pv, size_t cb, const SocketAddress& addr) { 81dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen return socket_->SendTo(pv, cb, addr); 82dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen} 83dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen 84dc0f95d653279beabeb9817299e2902918ba123eKristian Monsenint AsyncUDPSocket::Close() { 85dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen return socket_->Close(); 86dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen} 87dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen 88dc0f95d653279beabeb9817299e2902918ba123eKristian MonsenSocket::ConnState AsyncUDPSocket::GetState() const { 89dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen return socket_->GetState(); 90dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen} 91dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen 92dc0f95d653279beabeb9817299e2902918ba123eKristian Monsenint AsyncUDPSocket::GetOption(Socket::Option opt, int* value) { 93dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen return socket_->GetOption(opt, value); 94dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen} 95dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen 96dc0f95d653279beabeb9817299e2902918ba123eKristian Monsenint AsyncUDPSocket::SetOption(Socket::Option opt, int value) { 97dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen return socket_->SetOption(opt, value); 98dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen} 99dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen 100dc0f95d653279beabeb9817299e2902918ba123eKristian Monsenint AsyncUDPSocket::GetError() const { 101dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen return socket_->GetError(); 102dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen} 103dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen 104dc0f95d653279beabeb9817299e2902918ba123eKristian Monsenvoid AsyncUDPSocket::SetError(int error) { 105dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen return socket_->SetError(error); 106dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen} 107dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen 108f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdochvoid AsyncUDPSocket::OnReadEvent(AsyncSocket* socket) { 109dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen ASSERT(socket_.get() == socket); 110f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch 111f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch SocketAddress remote_addr; 112f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch int len = socket_->RecvFrom(buf_, size_, &remote_addr); 113f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch if (len < 0) { 114f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch // An error here typically means we got an ICMP error in response to our 115f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch // send datagram, indicating the remote address was unreachable. 116f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch // When doing ICE, this kind of thing will often happen. 117731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick // TODO: Do something better like forwarding the error to the user. 118f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch SocketAddress local_addr = socket_->GetLocalAddress(); 119f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch LOG(LS_INFO) << "AsyncUDPSocket[" << local_addr.ToString() << "] " 120f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch << "receive failed with error " << socket_->GetError(); 121f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch return; 122f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch } 123f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch 124731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick // TODO: Make sure that we got all of the packet. 125f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch // If we did not, then we should resize our buffer to be large enough. 126dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen SignalReadPacket(this, buf_, (size_t)len, remote_addr); 127f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch} 128f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch 129f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch} // namespace talk_base 130