U ~Hj1@sdZddlmZddlZddlZddlZddlZddlmZm Z ddl m Z ddl m Z mZmZmZddlmZddlmZmZdd lmZmZdd lmZmZmZdd lmZdd l m!Z!dd l"m#Z#erddl$m%Z%e eej&ej'ej(Gdddej)Z*e ej+Gdddee*Z,dS)z UDP support for IOCP reactor ) annotationsN) TYPE_CHECKINGOptional) implementer)addressdefererror interfaces)MulticastMixin) isIPAddress isIPv6Address)abstract iocpsupport)ERROR_CONNECTION_REFUSEDERROR_IO_PENDINGERROR_PORT_UNREACHABLE)IReadWriteHandle)AbstractDatagramProtocol)log) IOCPReactorc@seZdZUdZded<ejZejZ dZ dZ ded<d@d d d d d ddddZ ddZ d dddZddZddZddddZddZd d!Zd"d#Zd$d%Zd&d'ZdAd(d)Zd*d+Zd,d-Zd.d/Zd0d1Zd2d3ZdBd4d5Zd6d7Zd8d9Zd:d;Z dd?Z"dS)CPortz UDP port, listening for packets. @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. rreactorFNz Optional[int]_realPortNumber intrstrIOCPReactor | NoneNone)portproto interface maxPacketSizerreturncCsx||_||_||_||_|d|_|tj ||t |j |j }t |}t||_ttd|_dS)z> Initialize with a numeric port to listen on. Ni)rprotocolZreadBufferSizer! setLogStr_connectedAddr_setAddressFamilyr FileHandle__init__socket addressFamily socketType_iocpZ maxAddrLenfileno bytearray addressBufferstructcalcsizeaddressLengthBuffer)selfrr r!r"rsktZaddrLenr7D/tmp/pip-unpacked-wheel-ewod1tgb/twisted/internet/iocpreactor/udp.pyr*:s  z Port.__init__cCs@t|jrtj|_n(t|jr(tj|_n|jrz not connected>)rr% __class__r<r7r7r8__repr__bs z Port.__repr__cCs|jS)z) Return a socket object. )r+r<r7r7r8 getHandlehszPort.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_connectToProtocolr<r7r7r8startListeningnszPort.startListening socket.socketcCs|j|j|jSN)r createSocketr,r-r<r7r7r8rHxszPort.createSocketc Csz|}||j|jfWn4tk rR}zt|j|j|W5d}~XYnX|d|_t d| |j |jfd|_ ||_|jj|_dS)Nz%s starting on %sT)rHbindr!rOSErrorrZCannotListenError getsocknamerrmsg _getLogPrefixr% connectedr+r/ getFileHandle)r5r6ler7r7r8rC{s$zPort._bindSocketcCs$|j|||j|dSrG)r%ZmakeConnectionZ startReadingrZaddActiveHandler<r7r7r8rDs zPort._connectToProtocolcCs |jr|||||dSrG)reading handleReaddoReadr5rcdataevtr7r7r8cbReadsz Port.cbReadcCs|tjtjttfkr&|jr|jnh|rHt dtj |d|fnFz(|j t |jd|t|jWntk rtYnXdS)Nzerror in recvfrom -- %s (%s)z unknown error)errnoWSAECONNREFUSED WSAECONNRESETrrr'r%connectionRefusedrrM errorcodegetZdatagramReceivedbytesbuffr.Z makesockaddr addr_buff BaseExceptionerrrUr7r7r8rSs*  zPort.handleReadcCsvt|j|}|jd|_}|j|_}|j|_}t | ||||\}}|rr|t krr|j d|j|||dSNr)r.EventrYZ _readBuffersrar1rbr4 addr_len_buffrecvfromrPrr callLater)r5rXrarbrgrVrWr7r7r8rTs   z Port.doReadc Cs|jr|d|jfkstz|j|WStk r}zb|jd}|tjkr^||WY>S|tj krtt dn"|tj tj ttfkr|jnW5d}~XYnXn2|dkstt|dst|ds|ddkrt |ddt|dr|jtjkrt |ddt|drD|jtjkrDt |ddz|j||WStk r}zh|jd}|tjkr|||WY@S|tj krt dn"|tj tj ttfkrWY dSW5d}~XYnXdS)z~ Write a datagram. @param addr: should be a tuple (ip, port), can be None in connected mode. Nrzmessage too longz z0write() only accepts IP addresses, not hostnamesz*IPv6 port write() called with IPv4 addressz*IPv4 port write() called with IPv6 address)r'AssertionErrorr+sendrKargsrZZWSAEINTRwriteZ WSAEMSGSIZErZMessageLengthErrorr[r\rrr%r]r r r;r,r9r:sendto)r5Zdatagramaddrsenor7r7r8rmsn              z Port.writecCs|d||dS)N)rmjoin)r5seqror7r7r8 writeSequenceszPort.writeSequencecCsH|jrtdt|s*t|s*t|d||f|_|j||fdS)z- 'Connect' to remote server. z\already connected, reconnecting is not currently supported (talk to itamar if you want this)znot an IPv4 or IPv6 address.N)r' RuntimeErrorr r rr;r+connect)r5hostrr7r7r8rws  z Port.connectcCs.||j||jr*|jd|jdSre)Z stopReadingrZremoveActiveHandlerOriconnectionLostr<r7r7r8_loseConnections zPort._loseConnectioncCs&|jrt}|_nd}||SrG)rOrZDeferreddrz)r5resultr7r7r8 stopListenings zPort.stopListeningcCstjdtdd|dS)Nz-Please use stopListening() to disconnect port) stacklevel)warningswarnDeprecationWarningr}r<r7r7r8loseConnection!s zPort.loseConnectioncCs^td|jd|_tj|||j|j |`|` t |drZ|j d|` dS)z& Cleans up my socket. z(UDP Port %s Closed)Nr{)rrMrr r)ryr%ZdoStopr+closerPhasattrr{callback)r5reasonr7r7r8ry)s    zPort.connectionLostcCs||j}d||_dS)zP Initialize the C{logstr} attribute to be used by C{logPrefix}. z%s (UDP)N)rNr%logstr)r5 logPrefixr7r7r8r&8s zPort.setLogStrcCs|jS)zK Returns the name of my class, to prefix log entries with. )rr<r7r7r8r?szPort.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} UDPNr~)r)r)r+rLr,r:r IPv4Addressr9 IPv6Address)r5ror7r7r8getHostEs    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)r+ setsockopt SOL_SOCKET SO_BROADCAST)r5Zenabledr7r7r8setBroadcastAllowedRszPort.setBroadcastAllowedcCst|jtjtjS)z Checks if broadcast is currently allowed on this port. @return: Whether this port may broadcast. @rtype: L{bool} )boolr+ getsockoptrrr<r7r7r8getBroadcastAllowed[szPort.getBroadcastAllowed)rrN)N)N)#__name__ __module__ __qualname____doc____annotations__r+r:r, SOCK_DGRAMr-ZdynamicReadBuffersrr*r(rArBrErHrCrDrYrSrTrmrurwrzr}rryr&rrrrr7r7r7r8r#s@     A   rc @s:eZdZdZdddddd d d d d dZddddZdS) MulticastPortz. UDP Port that supports multicasting. rrNFrrrrrr)rr r!r"rlistenMultipler#cCst||||||||_dSrG)rr*r)r5rr r!r"rrr7r7r8r*ks zMulticastPort.__init__rFr=cCsBt|}|jr>|tjtjdttdr>|tjtjd|S)NrI SO_REUSEPORT) rrHrrr+r SO_REUSEADDRrr)r5r6r7r7r8rHws   zMulticastPort.createSocket)rrNF)rrrrr*rHr7r7r7r8res r)-r __future__rrZr+r2rtypingrrZzope.interfacerZtwisted.internetrrrr Ztwisted.internet._multicastr Ztwisted.internet.abstractr r Ztwisted.internet.iocpreactorr rr.Z"twisted.internet.iocpreactor.constrrrZ'twisted.internet.iocpreactor.interfacesrZtwisted.internet.protocolrZtwisted.pythonrZ$twisted.internet.iocpreactor.reactorrZIListeningPortZ IUDPTransportZ ISystemHandler)rZIMulticastTransportrr7r7r7r8s8       >