U AcjB@svdZddlmZddlZddlZddlmZddlmZddl m Z ddl m Z ddl mZdd lmZed krdd lmZdd lmZdd lmZmZmZmZmZmZmZeeeegZeeeegZeZeZeZ eZ!n0ddlm Z mZm!Z!mZmZm"Z"e e!e"gZegZddl#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)ddl*m+Z+ee)j,e)j-e)j.Gddde&j/Z0ee)j1Gddde e0Z2dS)a Various asynchronous UDP classes. Please do not use this module directly. @var _sockErrReadIgnore: list of symbolic error constants (from the C{errno} module) representing socket errors where the error is temporary and can be ignored. @var _sockErrReadRefuse: list of symbolic error constants (from the C{errno} module) representing socket errors that indicate connection refused. ) annotationsN)Optional) implementer)MulticastMixin)IReactorMulticast)AbstractDatagramProtocol) platformTypewin32)WSAEINPROGRESS)WSAEWOULDBLOCK)WSAECONNREFUSED WSAECONNRESETWSAEINTR WSAEMSGSIZE WSAENETRESETWSAENOPROTOOPT WSAETIMEDOUT)EAGAIN ECONNREFUSEDEINTREMSGSIZE ENOPROTOOPT EWOULDBLOCK)abstractaddressbasedefererror interfaces)logc@seZdZUdZejZded<ejZ ded<dZ dZ ded <dZ d8d d Z eddZddddZddZddZddZddZddZd9ddZd d!Zd"d#Zd$d%Zd&d'Zd(d)Zd:d*d+Zd,d-Zd.d/Zd0d1Zd2d3Z d4d5Z!d6d7Z"dS);Porta UDP port, listening for packets. @ivar maxThroughput: Maximum number of bytes read in one event loop iteration. @ivar addressFamily: L{socket.AF_INET} or L{socket.AF_INET6}, depending on whether this port is listening on an IPv4 address or an IPv6 address. @ivar _realPortNumber: Actual port number being listened on. The value will be L{None} until this L{Port} is listening. @ivar _preexistingSocket: If not L{None}, a L{socket.socket} instance which was created and initialized outside of the reactor and will be used to listen for connections (instead of a new socket being created by this L{Port}). zsocket.AddressFamily addressFamilyzsocket.SocketKind socketTypeiNz Optional[int]_realPortNumber cCs@tj||||_||_||_||_|d|_| dS)a @param port: A port number on which to listen. @type port: L{int} @param proto: A C{DatagramProtocol} instance which will be connected to the given C{port}. @type proto: L{twisted.internet.protocol.DatagramProtocol} @param interface: The local IPv4 or IPv6 address to which to bind; defaults to '', ie all IPv4 addresses. @type interface: L{str} @param maxPacketSize: The maximum packet size to accept. @type maxPacketSize: L{int} @param reactor: A reactor which will notify this C{Port} when its socket is ready for reading or writing. Defaults to L{None}, ie the default global reactor. @type reactor: L{interfaces.IReactorFDSet} N) rBasePort__init__portprotocol maxPacketSize interface setLogStr_connectedAddr_setAddressFamily)selfr(protor+r*reactorr28/tmp/pip-unpacked-wheel-nrvc_22y/twisted/internet/udp.pyr']sz Port.__init__c Cs8t|||j}|d}|d||||d}||_|S)a Create a new L{Port} based on an existing listening I{SOCK_DGRAM} socket. @param reactor: A reactor which will notify this L{Port} when its socket is ready for reading or writing. Defaults to L{None}, ie the default global reactor. @type reactor: L{interfaces.IReactorFDSet} @param fd: An integer file descriptor associated with a listening socket. The socket must be in non-blocking mode. Any additional attributes desired, such as I{FD_CLOEXEC}, must also be set already. @type fd: L{int} @param addressFamily: The address family (sometimes called I{domain}) of the existing socket. For example, L{socket.AF_INET}. @type addressFamily: L{int} @param protocol: A C{DatagramProtocol} instance which will be connected to the C{port}. @type protocol: L{twisted.internet.protocol.DatagramProtocol} @param maxPacketSize: The maximum packet size to accept. @type maxPacketSize: L{int} @return: A new instance of C{cls} wrapping the socket given by C{fd}. @rtype: L{Port} rN)r+r1r*)socketfromfdr" getsockname_preexistingSocket) clsr1fdr!r)r*r(r+r/r2r2r3_fromListeningDescriptor{s  zPort._fromListeningDescriptorstrreturncCs6|jdk r"d|jjd|jdSd|jjdSdS)Nz not connected>)r#r) __class__r/r2r2r3__repr__s z Port.__repr__cCs|jS)z) Return a socket object. )r4rAr2r2r3 getHandleszPort.getHandlecCs||dS)z Create and bind my socket, and begin listening on it. This is called on unserialization, and must be called after creating a server to begin listening on the specified port. N) _bindSocket_connectToProtocolrAr2r2r3startListeningszPort.startListeningc Cs|jdkr`z|}||j|jfWqltk r\}zt|j|j|W5d}~XYqlXn |j}d|_|d|_ t d| |j |j fd|_||_|jj|_dS)aJ Prepare and assign a L{socket.socket} instance to C{self.socket}. Either creates a new SOCK_DGRAM L{socket.socket} bound to C{self.interface} and C{self.port} or takes an existing L{socket.socket} provided via the L{interfaces.IReactorSocket.adoptDatagramPort} interface. Nz%s starting on %s)r7createInternetSocketbindr+r(OSErrorrZCannotListenErrorr6r#rmsg _getLogPrefixr) connectedr4filenor/Zsktler2r2r3rDs" &zPort._bindSocketcCs|j||dSN)r)ZmakeConnectionZ startReadingrAr2r2r3rEs zPort._connectToProtocolc Csd}||jkrz|j|j\}}Wnbtk r}zD|jd}|tkrRWY,dS|tkrt|jrj|j WY dSW5d}~XYqX|t |7}|j tj kr|dd}z|j ||Wqtk rtYqXqdS)z= Called when my socket is ready for reading. rN) maxThroughputr4recvfromr*rJargs_sockErrReadIgnore_sockErrReadRefuser-r)connectionRefusedlenr!AF_INET6ZdatagramReceived BaseExceptionrerr)r/readdataaddrsenor2r2r3doReads(        z Port.doReadc Cs|jr|d|jfkstz|j|WStk r}zR|jd}|tkr\||WY0S|tkrpt dn|t kr|j nW5d}~XYnXn8|dkstt|dst|ds|ddkrt |ddt|ds|ddkr|jtjkrt |ddt|drJ|jtjkrJt |ddz|j||WStk r}zX|jd}|tkr|||WY2S|tkrt dn|t krWY dSW5d}~XYnXdS)az Write a datagram. @type datagram: L{bytes} @param datagram: The datagram to be sent. @type addr: L{tuple} containing L{str} as first element and L{int} as second element, or L{None} @param addr: A tuple of (I{stringified IPv4 or IPv6 address}, I{integer port number}); can be L{None} in connected mode. Nrzmessage too longz z0write() only accepts IP addresses, not hostnamesz7IPv6 port write() called with IPv4 or broadcast addressz*IPv4 port write() called with IPv6 address)r-AssertionErrorr4sendrJrUrwriterrZMessageLengthErrorrr)rXr isIPAddress isIPv6AddressInvalidAddressErrorr!rZAF_INETsendto)r/Zdatagramr_r`rar2r2r3resd                 z Port.writecCs|d||dS)a Write a datagram constructed from an iterable of L{bytes}. @param seq: The data that will make up the complete datagram to be written. @type seq: an iterable of L{bytes} @type addr: L{tuple} containing L{str} as first element and L{int} as second element, or L{None} @param addr: A tuple of (I{stringified IPv4 or IPv6 address}, I{integer port number}); can be L{None} in connected mode. N)rejoin)r/seqr_r2r2r3 writeSequenceBs zPort.writeSequencecCsL|jrtdt|s.t|s.t|d||f|_|j||fdS)z- 'Connect' to remote server. z:already connected, reconnecting is not currently supportednot an IPv4 or IPv6 address.N) r- RuntimeErrorrrfrgrrhr4connect)r/hostr(r2r2r3rqQs  z Port.connectcCs"||jr|jd|jdS)Nr)Z stopReadingrMr1Z callLaterconnectionLostrAr2r2r3_loseConnection^szPort._loseConnectioncCs&|jrt}|_nd}||SrQ)rMrZDeferreddrt)r/resultr2r2r3 stopListeningcs zPort.stopListeningcCstjdtdd|dS)Nz-Please use stopListening() to disconnect portrR) stacklevel)warningswarnDeprecationWarningrwrAr2r2r3loseConnectionks zPort.loseConnectioncCsdtd|jd|_d|_tj|||j|j |` |` t |dr`|j d|` dS)z& Cleans up my socket. z(UDP Port %s Closed)Nru)rrKr#rSrr&rsr)ZdoStopr4closerNhasattrrucallback)r/reasonr2r2r3rsss    zPort.connectionLostcCs||j}d||_dS)zP Initialize the C{logstr} attribute to be used by C{logPrefix}. z%s (UDP)N)rLr)logstr)r/ logPrefixr2r2r3r,s zPort.setLogStrcCsDt|jrtj|_n*t|jr,tj|_n|jr@t |jddS)z8 Resolve address family for the socket. roN) rrgr+r4rZr!rfrirrhrAr2r2r3r.s    zPort._setAddressFamilycCs|jS)z0 Return the prefix to log with. )rrAr2r2r3rszPort.logPrefixcCsJ|j}|jtjkr$tjd|S|jtjkrFtjd|ddSdS)z Return the local address of the UDP connection @returns: the local address of the UDP connection @rtype: L{IPv4Address} or L{IPv6Address} UDPNrR)r)r)r4r6r!rir IPv4AddressrZ IPv6Address)r/r_r2r2r3getHosts    z Port.getHostcCs|jtjtj|dS)z Set whether this port may broadcast. This is disabled by default. @param enabled: Whether the port may broadcast. @type enabled: L{bool} N)r4 setsockopt SOL_SOCKET SO_BROADCAST)r/Zenabledr2r2r3setBroadcastAllowedszPort.setBroadcastAllowedcCst|jtjtjS)z Checks if broadcast is currently allowed on this port. @return: Whether this port may broadcast. @rtype: L{bool} )boolr4 getsockoptrrrAr2r2r3getBroadcastAllowedszPort.getBroadcastAllowed)r$r%N)N)N)#__name__ __module__ __qualname____doc__r4rir!__annotations__ SOCK_DGRAMr"rSr#r7r' classmethodr:rBrCrFrDrErbrernrqrtrwr|rsr,r.rrrrr2r2r2r3r @s8    + # >     r c @s:eZdZdZdddddd d d d d dZddddZdS) MulticastPortz. UDP Port that supports multicasting. r$r%NFintrr;zIReactorMulticast | NonerNone)r(r0r+r*r1listenMultipler=cCst||||||||_dS)zX @see: L{twisted.internet.interfaces.IReactorMulticast.listenMulticast} N)r r'r)r/r(r0r+r*r1rr2r2r3r's zMulticastPort.__init__z socket.socketr<c Csxt|}|jrt|tjtjdttdrtz|tjtjdWn0t k rr}z|j t kr`nW5d}~XYnX|S)z Override L{Port.createInternetSocket} to configure the socket to honor the C{listenMultiple} argument to L{IReactorMulticast.listenMultiple}. rG SO_REUSEPORTN) r rHrrr4r SO_REUSEADDRrrrJerrnorrOr2r2r3rHs   z"MulticastPort.createInternetSocket)r$r%NF)rrrrr'rHr2r2r2r3rsr)3r __future__rr4rytypingrZzope.interfacerZtwisted.internet._multicastrZtwisted.internet.interfacesrZtwisted.internet.protocolrZtwisted.python.runtimerrr r r r rrrrrrrVrWrrrrrZtwisted.internetrrrrrrZtwisted.pythonrZIListeningPortZ IUDPTransportZ ISystemHandler&r ZIMulticastTransportrr2r2r2r3sD         $      |