o >h1@sdZddlmZddlZddlZddlZddlZddlmZm Z ddl m Z ddl m Z mZmZmZddlmZddlmZmZdd lmZmZdd lmZmZmZdd lmZdd l m!Z!dd l"m#Z#erlddl$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<  dCdDddZ ddZ dEddZddZddZdFdd Zd!d"Zd#d$Zd%d&Zd'd(Zd)d*ZdGd+d,Zd-d.Zd/d0Zd1d2Zd3d4Zd5d6ZdGd7d8Zd9d:Zd;d<Zd=d>Z d?d@Z!dAdBZ"dS)HPortz 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 portintprotor interfacestr maxPacketSizeIOCPReactor | NonereturnNonecCsx||_||_||_||_|d|_|tj ||t |j |j }t |}t||_ttd|_dS)z> Initialize with a numeric port to listen on. Ni)rprotocolreadBufferSizer setLogStr_connectedAddr_setAddressFamilyr FileHandle__init__socket addressFamily socketType_iocp maxAddrLenfileno bytearray addressBufferstructcalcsizeaddressLengthBuffer)selfrrrr rsktaddrLenr:/var/www/vedio/testing/chatpythonscript.ninositsolution.com/env/lib/python3.10/site-packages/twisted/internet/iocpreactor/udp.pyr+:s  z Port.__init__cCsDt|jr tj|_dSt|jrtj|_dS|jr t|jddS)z8 Resolve address family for the socket. znot an IPv4 or IPv6 addressN) r rr,AF_INET6r-r AF_INETrInvalidAddressErrorr7r:r:r;r)Us    zPort._setAddressFamilycCs2|jdurd|jjd|jdSd|jjdS)Nz not connected>)rr% __class__r?r:r:r;__repr__bs z Port.__repr__cC|jS)z) Return a socket object. )r,r?r:r:r; getHandlehzPort.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?r:r:r;startListeningns zPort.startListening socket.socketcCs|j|j|jSN)r createSocketr-r.r?r:r:r;rLxszPort.createSocketc Csz|}||j|jfWnty$}z t|j|j|d}~ww|d|_t d| |j |jfd|_ ||_|jj|_dS)Nz%s starting on %sT)rLbindrrOSErrorrCannotListenError getsocknamerrmsg _getLogPrefixr% connectedr,r1 getFileHandle)r7r8ler:r:r;rG{s zPort._bindSocketcCs$|j|||j|dSrK)r%makeConnection startReadingraddActiveHandler?r:r:r;rHs zPort._connectToProtocolcCs$|jr|||||dSdSrK)reading handleReaddoReadr7rcdataevtr:r:r;cbReads z Port.cbReadcCs|tjtjttfvr|jr|jdSdS|r(t dtj |d|fdSz|j t |jd|t|jWdStyKtYdSw)Nzerror in recvfrom -- %s (%s)z unknown error)errnoWSAECONNREFUSED WSAECONNRESETrrr(r%connectionRefusedrrR errorcodegetdatagramReceivedbytesbuffr/ makesockaddr addr_buff BaseExceptionerrr]r:r:r;r[s, zPort.handleReadcCs~t|j|}|jd|_}|j|_}|j|_}t | ||||\}}|r;|t kr=|j d|j|||dSdSdSNr)r/Eventra _readBuffersrjr3rlr6 addr_len_buffrecvfromrUrr callLater)r7r`rjrlrrr^r_r:r:r;r\s   z Port.doReadc Cs|jrW|d|jfvs Jz|j|WStyV}z7|jd}|tjkr0||WYd}~S|tjkr:t d|tj tj t tfvrJ|jnWYd}~dSd}~ww|dks]Jt|dswt|dsw|ddkrwt |ddt|dr|jtjkrt |ddt|dr|jtjkrt |ddz|j||WSty}z2|jd}|tjkr|||WYd}~S|tjkrt d|tj tj t tfvrWYd}~dSd}~ww)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(r,sendrOargsrbWSAEINTRwrite WSAEMSGSIZErMessageLengthErrorrcrdrrr%rer r r>r-r<r=sendto)r7datagramaddrsenor:r:r;rxsp              z Port.writecCs|d||dS)N)rxjoin)r7seqr}r:r:r; writeSequenceszPort.writeSequencecCsH|jrtdt|st|st|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)r7hostrr:r:r;rs  z Port.connectcCs2||j||jr|jd|jdSdSro) stopReadingrremoveActiveHandlerTrtconnectionLostr?r:r:r;_loseConnections  zPort._loseConnectioncCs&|jr t}|_nd}||SrK)rTrDeferreddr)r7resultr:r:r; stopListenings zPort.stopListeningcCstjdtdd|dS)Nz-Please use stopListening() to disconnect port) stacklevel)warningswarnDeprecationWarningrr?r:r:r;loseConnection!s  zPort.loseConnectioncCsbtd|jd|_tj|||j|j |`|` t |dr/|j d|` dSdS)z& Cleans up my socket. z(UDP Port %s Closed)Nr)rrRrr r*rr%doStopr,closerUhasattrrcallback)r7reasonr:r:r;r)s    zPort.connectionLostcCs||j}d||_dS)zP Initialize the C{logstr} attribute to be used by C{logPrefix}. z%s (UDP)N)rSr%logstr)r7 logPrefixr:r:r;r'8s zPort.setLogStrcCrD)zK Returns the name of my class, to prefix log entries with. )rr?r:r:r;r?rFzPort.logPrefixcCsR|j}|jtjkrtjdg|RS|jtjkr'tjdg|ddRSdS)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,rQr-r=r IPv4Addressr< IPv6Address)r7r}r:r:r;getHostEs   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)r7enabledr:r:r;setBroadcastAllowedRszPort.setBroadcastAllowedcCst|jtjtjS)z Checks if broadcast is currently allowed on this port. @return: Whether this port may broadcast. @rtype: L{bool} )boolr, getsockoptrrr?r:r:r;getBroadcastAllowed[szPort.getBroadcastAllowed)rrN) rrrrrrr rrr!r"r#)r"rr"rJrK)#__name__ __module__ __qualname____doc____annotations__r,r=r- SOCK_DGRAMr.dynamicReadBuffersrr+r)rCrErIrLrGrHrar[r\rxrrrrrrr'rrrrr:r:r:r;r#sB      A  rc@s.eZdZdZ    ddddZdddZdS) MulticastPortz. UDP Port that supports multicasting. rrNFrrrrrrr rr!listenMultiplerr"r#cCst||||||||_dSrK)rr+r)r7rrrr rrr:r:r;r+ks zMulticastPort.__init__rJcCsBt|}|jr|tjtjdttdr|tjtjd|S)NrM SO_REUSEPORT) rrLrrr,r SO_REUSEADDRrr)r7r8r:r:r;rLws  zMulticastPort.createSocket)rrNF)rrrrrrr rrr!rrr"r#r)rrrrr+rLr:r:r:r;res  r)-r __future__rrbr,r4rtypingrrzope.interfacertwisted.internetrrrr twisted.internet._multicastr twisted.internet.abstractr r twisted.internet.iocpreactorr rr/"twisted.internet.iocpreactor.constrrr'twisted.internet.iocpreactor.interfacesrtwisted.internet.protocolrtwisted.pythonr$twisted.internet.iocpreactor.reactorrIListeningPort IUDPTransport ISystemHandler*rIMulticastTransportrr:r:r:r;s:       >