15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2012 The Chromium Authors. All rights reserved. 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file. 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#ifndef NET_UDP_UDP_SOCKET_LIBEVENT_H_ 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#define NET_UDP_UDP_SOCKET_LIBEVENT_H_ 75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/memory/ref_counted.h" 95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/memory/scoped_ptr.h" 109ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch#include "base/message_loop/message_loop.h" 115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/threading/non_thread_safe.h" 125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/base/completion_callback.h" 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/base/io_buffer.h" 145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/base/ip_endpoint.h" 159ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch#include "net/base/net_export.h" 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/base/net_log.h" 179ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch#include "net/base/rand_callback.h" 185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/udp/datagram_socket.h" 195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace net { 215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class NET_EXPORT UDPSocketLibevent : public base::NonThreadSafe { 235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) UDPSocketLibevent(DatagramSocket::BindType bind_type, 255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const RandIntCallback& rand_int_cb, 265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) net::NetLog* net_log, 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const net::NetLog::Source& source); 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual ~UDPSocketLibevent(); 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Connect the socket to connect with a certain |address|. 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Returns a net error code. 325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int Connect(const IPEndPoint& address); 335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Bind the address/port for this socket to |address|. This is generally 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // only used on a server. 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Returns a net error code. 375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int Bind(const IPEndPoint& address); 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Close the socket. 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void Close(); 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Copy the remote udp address into |address| and return a network error code. 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int GetPeerAddress(IPEndPoint* address) const; 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Copy the local udp address into |address| and return a network error code. 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // (similar to getsockname) 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int GetLocalAddress(IPEndPoint* address) const; 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // IO: 505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Multiple outstanding read requests are not supported. 515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Full duplex mode (reading and writing at the same time) is supported 525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Read from the socket. 545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Only usable from the client-side of a UDP socket, after the socket 555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // has been connected. 565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int Read(IOBuffer* buf, int buf_len, const CompletionCallback& callback); 575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Write to the socket. 595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Only usable from the client-side of a UDP socket, after the socket 605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // has been connected. 615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int Write(IOBuffer* buf, int buf_len, const CompletionCallback& callback); 625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Read from a socket and receive sender address information. 645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // |buf| is the buffer to read data into. 655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // |buf_len| is the maximum amount of data to read. 665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // |address| is a buffer provided by the caller for receiving the sender 675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // address information about the received data. This buffer must be kept 685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // alive by the caller until the callback is placed. 695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // |address_length| is a ptr to the length of the |address| buffer. This 705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // is an input parameter containing the maximum size |address| can hold 715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // and also an output parameter for the size of |address| upon completion. 725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // |callback| the callback on completion of the Recv. 735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Returns a net error code, or ERR_IO_PENDING if the IO is in progress. 745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // If ERR_IO_PENDING is returned, the caller must keep |buf|, |address|, 755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // and |address_length| alive until the callback is called. 765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int RecvFrom(IOBuffer* buf, 775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int buf_len, 785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) IPEndPoint* address, 795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const CompletionCallback& callback); 805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Send to a socket with a particular destination. 825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // |buf| is the buffer to send 835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // |buf_len| is the number of bytes to send 845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // |address| is the recipient address. 855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // |address_length| is the size of the recipient address 865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // |callback| is the user callback function to call on complete. 875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Returns a net error code, or ERR_IO_PENDING if the IO is in progress. 885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // If ERR_IO_PENDING is returned, the caller must keep |buf| and |address| 895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // alive until the callback is called. 905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int SendTo(IOBuffer* buf, 915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int buf_len, 925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const IPEndPoint& address, 935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const CompletionCallback& callback); 945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Set the receive buffer size (in bytes) for the socket. 965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool SetReceiveBufferSize(int32 size); 975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Set the send buffer size (in bytes) for the socket. 995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool SetSendBufferSize(int32 size); 1005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Returns true if the socket is already connected or bound. 1025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool is_connected() const { return socket_ != kInvalidSocket; } 1035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const BoundNetLog& NetLog() const { return net_log_; } 1055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Sets corresponding flags in |socket_options_| to allow the socket 1075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // to share the local address to which the socket will be bound with 1085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // other processes. Should be called before Bind(). 1095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void AllowAddressReuse(); 1105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Sets corresponding flags in |socket_options_| to allow sending 1125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // and receiving packets to and from broadcast addresses. Should be 1135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // called before Bind(). 1145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void AllowBroadcast(); 1155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 116c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Join the multicast group. 117c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // |group_address| is the group address to join, could be either 118c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // an IPv4 or IPv6 address. 119c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Return a network error code. 120c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) int JoinGroup(const IPAddressNumber& group_address) const; 121c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 122c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Leave the multicast group. 123c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // |group_address| is the group address to leave, could be either 124c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // an IPv4 or IPv6 address. If the socket hasn't joined the group, 125c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // it will be ignored. 126c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // It's optional to leave the multicast group before destroying 127c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // the socket. It will be done by the OS. 128c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Return a network error code. 129c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) int LeaveGroup(const IPAddressNumber& group_address) const; 130c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 131c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Set the time-to-live option for UDP packets sent to the multicast 132c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // group address. The default value of this option is 1. 133c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Cannot be negative or more than 255. 134c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Should be called before Bind(). 135c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Return a network error code. 136c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) int SetMulticastTimeToLive(int time_to_live); 137c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 138c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Set the loopback flag for UDP socket. If this flag is true, the host 139c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // will receive packets sent to the joined group from itself. 140c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // The default value of this option is true. 141c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Should be called before Bind(). 142c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Return a network error code. 143c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // 144c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Note: the behavior of |SetMulticastLoopbackMode| is slightly 145c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // different between Windows and Unix-like systems. The inconsistency only 146c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // happens when there are more than one applications on the same host 147c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // joined to the same multicast group while having different settings on 148c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // multicast loopback mode. On Windows, the applications with loopback off 149c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // will not RECEIVE the loopback packets; while on Unix-like systems, the 150c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // applications with loopback off will not SEND the loopback packets to 151c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // other applications on the same host. See MSDN: http://goo.gl/6vqbj 152c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) int SetMulticastLoopbackMode(bool loopback); 153c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 1545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) private: 1555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) static const int kInvalidSocket = -1; 1565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) enum SocketOptions { 158c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) SOCKET_OPTION_REUSE_ADDRESS = 1 << 0, 159c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) SOCKET_OPTION_BROADCAST = 1 << 1, 160c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) SOCKET_OPTION_MULTICAST_LOOP = 1 << 2 1615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) }; 1625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 16390dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) class ReadWatcher : public base::MessageLoopForIO::Watcher { 1645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 1655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) explicit ReadWatcher(UDPSocketLibevent* socket) : socket_(socket) {} 1665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // MessageLoopForIO::Watcher methods 1685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void OnFileCanReadWithoutBlocking(int /* fd */) OVERRIDE; 1705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void OnFileCanWriteWithoutBlocking(int /* fd */) OVERRIDE {} 1725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) private: 1745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) UDPSocketLibevent* const socket_; 1755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DISALLOW_COPY_AND_ASSIGN(ReadWatcher); 1775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) }; 1785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 17990dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) class WriteWatcher : public base::MessageLoopForIO::Watcher { 1805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 1815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) explicit WriteWatcher(UDPSocketLibevent* socket) : socket_(socket) {} 1825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // MessageLoopForIO::Watcher methods 1845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void OnFileCanReadWithoutBlocking(int /* fd */) OVERRIDE {} 1865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void OnFileCanWriteWithoutBlocking(int /* fd */) OVERRIDE; 1885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) private: 1905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) UDPSocketLibevent* const socket_; 1915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DISALLOW_COPY_AND_ASSIGN(WriteWatcher); 1935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) }; 1945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void DoReadCallback(int rv); 1965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void DoWriteCallback(int rv); 1975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void DidCompleteRead(); 1985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void DidCompleteWrite(); 1995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Handles stats and logging. |result| is the number of bytes transferred, on 2015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // success, or the net error code on failure. On success, LogRead takes in a 2025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // sockaddr and its length, which are mandatory, while LogWrite takes in an 2035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // optional IPEndPoint. 2045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void LogRead(int result, const char* bytes, socklen_t addr_len, 2055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const sockaddr* addr) const; 2065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void LogWrite(int result, const char* bytes, const IPEndPoint* address) const; 2075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Returns the OS error code (or 0 on success). 2095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int CreateSocket(const IPEndPoint& address); 2105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Same as SendTo(), except that address is passed by pointer 2125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // instead of by reference. It is called from Write() with |address| 2135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // set to NULL. 2145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int SendToOrWrite(IOBuffer* buf, 2155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int buf_len, 2165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const IPEndPoint* address, 2175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const CompletionCallback& callback); 2185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int InternalConnect(const IPEndPoint& address); 2205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int InternalRecvFrom(IOBuffer* buf, int buf_len, IPEndPoint* address); 2215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int InternalSendTo(IOBuffer* buf, int buf_len, const IPEndPoint* address); 2225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Applies |socket_options_| to |socket_|. Should be called before 2245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Bind(). 2255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int SetSocketOptions(); 2265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int DoBind(const IPEndPoint& address); 2275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int RandomBind(const IPEndPoint& address); 2285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int socket_; 230c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) int addr_family_; 2315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Bitwise-or'd combination of SocketOptions. Specifies the set of 2335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // options that should be applied to |socket_| before Bind(). 2345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int socket_options_; 2355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 236c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Multicast socket options cached for SetSocketOption. 237c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Cannot be used after Bind(). 238c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) int multicast_time_to_live_; 239c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 2405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // How to do source port binding, used only when UDPSocket is part of 2415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // UDPClientSocket, since UDPServerSocket provides Bind. 2425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DatagramSocket::BindType bind_type_; 2435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // PRNG function for generating port numbers. 2455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) RandIntCallback rand_int_cb_; 2465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // These are mutable since they're just cached copies to make 2485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // GetPeerAddress/GetLocalAddress smarter. 2495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) mutable scoped_ptr<IPEndPoint> local_address_; 2505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) mutable scoped_ptr<IPEndPoint> remote_address_; 2515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // The socket's libevent wrappers 25390dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) base::MessageLoopForIO::FileDescriptorWatcher read_socket_watcher_; 25490dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) base::MessageLoopForIO::FileDescriptorWatcher write_socket_watcher_; 2555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // The corresponding watchers for reads and writes. 2575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ReadWatcher read_watcher_; 2585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) WriteWatcher write_watcher_; 2595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // The buffer used by InternalRead() to retry Read requests 2615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_refptr<IOBuffer> read_buf_; 2625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int read_buf_len_; 2635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) IPEndPoint* recv_from_address_; 2645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // The buffer used by InternalWrite() to retry Write requests 2665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_refptr<IOBuffer> write_buf_; 2675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int write_buf_len_; 2685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_ptr<IPEndPoint> send_to_address_; 2695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // External callback; called when read is complete. 2715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CompletionCallback read_callback_; 2725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // External callback; called when write is complete. 2745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CompletionCallback write_callback_; 2755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) BoundNetLog net_log_; 2775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DISALLOW_COPY_AND_ASSIGN(UDPSocketLibevent); 2795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 2805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace net 2825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif // NET_UDP_UDP_SOCKET_LIBEVENT_H_ 284