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" 18424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles)#include "net/socket/socket_descriptor.h" 195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/udp/datagram_socket.h" 205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace net { 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class NET_EXPORT UDPSocketLibevent : public base::NonThreadSafe { 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) UDPSocketLibevent(DatagramSocket::BindType bind_type, 265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const RandIntCallback& rand_int_cb, 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) net::NetLog* net_log, 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const net::NetLog::Source& source); 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual ~UDPSocketLibevent(); 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Connect the socket to connect with a certain |address|. 325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Returns a net error code. 335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int Connect(const IPEndPoint& address); 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Bind the address/port for this socket to |address|. This is generally 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // only used on a server. 375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Returns a net error code. 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int Bind(const IPEndPoint& address); 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Close the socket. 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void Close(); 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Copy the remote udp address into |address| and return a network error code. 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int GetPeerAddress(IPEndPoint* address) const; 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Copy the local udp address into |address| and return a network error code. 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // (similar to getsockname) 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int GetLocalAddress(IPEndPoint* address) const; 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // IO: 515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Multiple outstanding read requests are not supported. 525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Full duplex mode (reading and writing at the same time) is supported 535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Read from the socket. 555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Only usable from the client-side of a UDP socket, after the socket 565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // has been connected. 575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int Read(IOBuffer* buf, int buf_len, const CompletionCallback& callback); 585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Write to the socket. 605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Only usable from the client-side of a UDP socket, after the socket 615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // has been connected. 625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int Write(IOBuffer* buf, int buf_len, const CompletionCallback& callback); 635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Read from a socket and receive sender address information. 655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // |buf| is the buffer to read data into. 665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // |buf_len| is the maximum amount of data to read. 675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // |address| is a buffer provided by the caller for receiving the sender 685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // address information about the received data. This buffer must be kept 695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // alive by the caller until the callback is placed. 705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // |address_length| is a ptr to the length of the |address| buffer. This 715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // is an input parameter containing the maximum size |address| can hold 725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // and also an output parameter for the size of |address| upon completion. 73116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch // |callback| is the callback on completion of the RecvFrom. 745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Returns a net error code, or ERR_IO_PENDING if the IO is in progress. 755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // If ERR_IO_PENDING is returned, the caller must keep |buf|, |address|, 765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // and |address_length| alive until the callback is called. 775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int RecvFrom(IOBuffer* buf, 785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int buf_len, 795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) IPEndPoint* address, 805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const CompletionCallback& callback); 815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Send to a socket with a particular destination. 835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // |buf| is the buffer to send 845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // |buf_len| is the number of bytes to send 855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // |address| is the recipient address. 865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // |address_length| is the size of the recipient address 875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // |callback| is the user callback function to call on complete. 885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Returns a net error code, or ERR_IO_PENDING if the IO is in progress. 895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // If ERR_IO_PENDING is returned, the caller must keep |buf| and |address| 905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // alive until the callback is called. 915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int SendTo(IOBuffer* buf, 925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int buf_len, 935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const IPEndPoint& address, 945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const CompletionCallback& callback); 955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Set the receive buffer size (in bytes) for the socket. 97c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch int SetReceiveBufferSize(int32 size); 985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Set the send buffer size (in bytes) for the socket. 100c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch int SetSendBufferSize(int32 size); 1015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Returns true if the socket is already connected or bound. 1035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool is_connected() const { return socket_ != kInvalidSocket; } 1045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const BoundNetLog& NetLog() const { return net_log_; } 1065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Sets corresponding flags in |socket_options_| to allow the socket 1085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // to share the local address to which the socket will be bound with 1095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // other processes. Should be called before Bind(). 1105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void AllowAddressReuse(); 1115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Sets corresponding flags in |socket_options_| to allow sending 1135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // and receiving packets to and from broadcast addresses. Should be 1145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // called before Bind(). 1155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void AllowBroadcast(); 1165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 117c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Join the multicast group. 118c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // |group_address| is the group address to join, could be either 119c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // an IPv4 or IPv6 address. 120c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Return a network error code. 121c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) int JoinGroup(const IPAddressNumber& group_address) const; 122c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 123c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Leave the multicast group. 124c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // |group_address| is the group address to leave, could be either 125c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // an IPv4 or IPv6 address. If the socket hasn't joined the group, 126c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // it will be ignored. 127c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // It's optional to leave the multicast group before destroying 128c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // the socket. It will be done by the OS. 129c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Return a network error code. 130c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) int LeaveGroup(const IPAddressNumber& group_address) const; 131c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 132a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) // Set interface to use for multicast. If |interface_index| set to 0, default 133a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) // interface is used. 134a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) // Should be called before Bind(). 135a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) // Returns a network error code. 136a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) int SetMulticastInterface(uint32 interface_index); 137a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) 138c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Set the time-to-live option for UDP packets sent to the multicast 139c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // group address. The default value of this option is 1. 140c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Cannot be negative or more than 255. 141c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Should be called before Bind(). 142c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Return a network error code. 143c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) int SetMulticastTimeToLive(int time_to_live); 144c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 145c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Set the loopback flag for UDP socket. If this flag is true, the host 146c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // will receive packets sent to the joined group from itself. 147c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // The default value of this option is true. 148c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Should be called before Bind(). 149c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Return a network error code. 150c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // 151c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Note: the behavior of |SetMulticastLoopbackMode| is slightly 152c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // different between Windows and Unix-like systems. The inconsistency only 153c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // happens when there are more than one applications on the same host 154c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // joined to the same multicast group while having different settings on 155c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // multicast loopback mode. On Windows, the applications with loopback off 156c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // will not RECEIVE the loopback packets; while on Unix-like systems, the 157c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // applications with loopback off will not SEND the loopback packets to 158c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // other applications on the same host. See MSDN: http://goo.gl/6vqbj 159c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) int SetMulticastLoopbackMode(bool loopback); 160c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 1618bcbed890bc3ce4d7a057a8f32cab53fa534672eTorne (Richard Coles) // Set the differentiated services flags on outgoing packets. May not 1628bcbed890bc3ce4d7a057a8f32cab53fa534672eTorne (Richard Coles) // do anything on some platforms. 1638bcbed890bc3ce4d7a057a8f32cab53fa534672eTorne (Richard Coles) // Return a network error code. 1648bcbed890bc3ce4d7a057a8f32cab53fa534672eTorne (Richard Coles) int SetDiffServCodePoint(DiffServCodePoint dscp); 1658bcbed890bc3ce4d7a057a8f32cab53fa534672eTorne (Richard Coles) 166a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) // Resets the thread to be used for thread-safety checks. 167a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) void DetachFromThread(); 168a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) 1695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) private: 1705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) enum SocketOptions { 171c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) SOCKET_OPTION_REUSE_ADDRESS = 1 << 0, 172c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) SOCKET_OPTION_BROADCAST = 1 << 1, 173c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) SOCKET_OPTION_MULTICAST_LOOP = 1 << 2 1745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) }; 1755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 17690dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) class ReadWatcher : public base::MessageLoopForIO::Watcher { 1775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 1785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) explicit ReadWatcher(UDPSocketLibevent* socket) : socket_(socket) {} 1795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // MessageLoopForIO::Watcher methods 1815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void OnFileCanReadWithoutBlocking(int /* fd */) OVERRIDE; 1835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void OnFileCanWriteWithoutBlocking(int /* fd */) OVERRIDE {} 1855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) private: 1875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) UDPSocketLibevent* const socket_; 1885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DISALLOW_COPY_AND_ASSIGN(ReadWatcher); 1905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) }; 1915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 19290dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) class WriteWatcher : public base::MessageLoopForIO::Watcher { 1935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 1945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) explicit WriteWatcher(UDPSocketLibevent* socket) : socket_(socket) {} 1955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // MessageLoopForIO::Watcher methods 1975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void OnFileCanReadWithoutBlocking(int /* fd */) OVERRIDE {} 1995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void OnFileCanWriteWithoutBlocking(int /* fd */) OVERRIDE; 2015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) private: 2035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) UDPSocketLibevent* const socket_; 2045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DISALLOW_COPY_AND_ASSIGN(WriteWatcher); 2065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) }; 2075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void DoReadCallback(int rv); 2095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void DoWriteCallback(int rv); 2105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void DidCompleteRead(); 2115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void DidCompleteWrite(); 2125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Handles stats and logging. |result| is the number of bytes transferred, on 2145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // success, or the net error code on failure. On success, LogRead takes in a 2155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // sockaddr and its length, which are mandatory, while LogWrite takes in an 2165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // optional IPEndPoint. 2175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void LogRead(int result, const char* bytes, socklen_t addr_len, 2185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const sockaddr* addr) const; 2195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void LogWrite(int result, const char* bytes, const IPEndPoint* address) const; 2205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Returns the OS error code (or 0 on success). 2225d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) int CreateSocket(int addr_family); 2235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Same as SendTo(), except that address is passed by pointer 2255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // instead of by reference. It is called from Write() with |address| 2265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // set to NULL. 2275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int SendToOrWrite(IOBuffer* buf, 2285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int buf_len, 2295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const IPEndPoint* address, 2305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const CompletionCallback& callback); 2315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int InternalConnect(const IPEndPoint& address); 2335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int InternalRecvFrom(IOBuffer* buf, int buf_len, IPEndPoint* address); 2345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int InternalSendTo(IOBuffer* buf, int buf_len, const IPEndPoint* address); 2355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Applies |socket_options_| to |socket_|. Should be called before 2375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Bind(). 2385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int SetSocketOptions(); 2395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int DoBind(const IPEndPoint& address); 2405d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) // Binds to a random port on |address|. 2415d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) int RandomBind(const IPAddressNumber& address); 2425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int socket_; 244c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) int addr_family_; 2455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Bitwise-or'd combination of SocketOptions. Specifies the set of 2475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // options that should be applied to |socket_| before Bind(). 2485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int socket_options_; 2495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 250a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) // Multicast interface. 251a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) uint32 multicast_interface_; 252a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) 253c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Multicast socket options cached for SetSocketOption. 254c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Cannot be used after Bind(). 255c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) int multicast_time_to_live_; 256c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 2575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // How to do source port binding, used only when UDPSocket is part of 2585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // UDPClientSocket, since UDPServerSocket provides Bind. 2595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DatagramSocket::BindType bind_type_; 2605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // PRNG function for generating port numbers. 2625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) RandIntCallback rand_int_cb_; 2635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // These are mutable since they're just cached copies to make 2655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // GetPeerAddress/GetLocalAddress smarter. 2665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) mutable scoped_ptr<IPEndPoint> local_address_; 2675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) mutable scoped_ptr<IPEndPoint> remote_address_; 2685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // The socket's libevent wrappers 27090dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) base::MessageLoopForIO::FileDescriptorWatcher read_socket_watcher_; 27190dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) base::MessageLoopForIO::FileDescriptorWatcher write_socket_watcher_; 2725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // The corresponding watchers for reads and writes. 2745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ReadWatcher read_watcher_; 2755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) WriteWatcher write_watcher_; 2765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // The buffer used by InternalRead() to retry Read requests 2785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_refptr<IOBuffer> read_buf_; 2795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int read_buf_len_; 2805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) IPEndPoint* recv_from_address_; 2815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // The buffer used by InternalWrite() to retry Write requests 2835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_refptr<IOBuffer> write_buf_; 2845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int write_buf_len_; 2855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_ptr<IPEndPoint> send_to_address_; 2865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // External callback; called when read is complete. 2885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CompletionCallback read_callback_; 2895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // External callback; called when write is complete. 2915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CompletionCallback write_callback_; 2925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) BoundNetLog net_log_; 2945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DISALLOW_COPY_AND_ASSIGN(UDPSocketLibevent); 2965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 2975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace net 2995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif // NET_UDP_UDP_SOCKET_LIBEVENT_H_ 301