Vh1dZddlmZddlZddlZddlZddlZddlmZm Z ddl m Z ddl m Z mZmZmZddlmZddlmZmZdd lmZmZdd lmZmZmZdd lmZdd l m!Z!dd l"m#Z#erddl$m%Z%e eejLejNejPGddejRZ*e ejVGddee*Z,y)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) IOCPReactorceZdZUdZded<ej ZejZ dZ dZ ded< d d dZ d Z d!d Zd Zd Zd"d ZdZdZdZdZdZd#dZdZdZdZdZdZd#dZdZdZdZ dZ!dZ"y)$Portz 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]_realPortNumberc||_||_||_||_|j d|_|j tjj||tj|j|j}tj|j}t!||_t!t%j&d|_y)z> Initialize with a numeric port to listen on. Ni)portprotocolreadBufferSize interface setLogStr_connectedAddr_setAddressFamilyr FileHandle__init__socket addressFamily socketType_iocp maxAddrLenfileno bytearray addressBufferstructcalcsizeaddressLengthBuffer)selfrprotor maxPacketSizersktaddrLens P/home/dcms/DCMS/lib/python3.12/site-packages/twisted/internet/iocpreactor/udp.pyr%z Port.__init__:s  +" "  $$T73mmD..@""3::<0&w/#,V__S-A#B ct|jrtj|_yt |jrtj |_y|jr tj|jdy)z8 Resolve address family for the socket. znot an IPv4 or IPv6 addressN) r r r&AF_INET6r'r AF_INETr InvalidAddressErrorr1s r6r#zPort._setAddressFamilyUs]  (!'D   (!'D  ^^++ = r7c|j'd|jjd|jdSd|jjdS)Nz not connected>)rr __class__r<s r6__repr__z Port.__repr__bsO    +t}}../tD4H4H3IK Kt}}../? ?r7c|jS)z) Return a socket object. )r&r<s r6 getHandlezPort.getHandleh{{r7cD|j|jy)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<s r6startListeningzPort.startListeningns  !r7cb|jj|j|jSN)r createSocketr'r(r<s r6rKzPort.createSocketxs#||((););T__MMr7c |j}|j|j|jf|jd|_tj|j|jd|jd|_ ||_|jj|_y#t$r0}t j |j|j|d}~wwxYw)Nz starting on T)rKbindr rOSErrorr CannotListenError getsocknamerrmsg _getLogPrefixr connectedr&r+ getFileHandle)r1r4les r6rFzPort._bindSocket{s I##%C HHdnndii0 1 #03 !!$--0$2F2F H  ![[// I))$..$))RH H Is7B88 C1+C,,C1c|jj||j|jj |yrJ)rmakeConnection startReadingraddActiveHandler<s r6rGzPort._connectToProtocols2 $$T*  $$T*r7cd|jr$|j||||jyyrJ)reading handleReaddoReadr1rcdataevts r6cbReadz Port.cbReads' << OOBc * KKM r7c |tjtjttfvr(|j r|j jyy|r;tjdtjj|dd|dy |j jt|jd|tj |j"y#t$$rtj&YywxYw)Nzerror in recvfrom -- z unknown errorz ())errnoWSAECONNREFUSED WSAECONNRESETrrr"rconnectionRefusedrrR errorcodegetdatagramReceivedbytesbuffr) makesockaddr addr_buff BaseExceptionerrr_s r6r]zPort.handleReads  ! !    $ "   "" //1#  GG??&&r?;RA    ..#((5D/*E,>,>s}},M!   sAC##DDc|tj|j|}|jdx|_}|j x|_}|jx|_}tj|j||||\}}|r4|tk7r*|jjd|j|||yyyNr)r)Eventrc _readBuffersrnr-rpr0 addr_len_buffrecvfromrUrr callLater)r1rbrnrprwr`ras r6r^z Port.doReadskk$++t,++A..4$($6$66 ,0,D,DDM>>    $ =# D "(( LL " "1dkk2tS A )2r7c |jr.|d|jfvsJ |jj|S|dk7sJt#|ds/t%|ds!|ddk7rtj&|ddt#|dr6|j(tj*k(rtj&|ddt%|dr6|j(tj,k(rtj&|dd |jj/||S#t$r}|jd}|t j k(r|j|cYd}~S|t jk(rtjd|t jt jttfvr|jj!nYd}~yd}~wwxYw#t$r}|jd}|t j k(r|j||cYd}~S|t jk(rtjd|t jt jttfvrYd}~yd}~wwxYw)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&sendrOargsrfWSAEINTRwrite WSAEMSGSIZEr MessageLengthErrorrgrhrrrrir r r;r'r9r:sendto)r1datagramaddrsenos r6r~z Port.writesF   D$"5"566 66 {{''11"4< <Q(%d1g.G},//GO47#(:(:foo(M//GIT!W%$*<*<*N//GI {{))(D99E WWQZ'::h//5,,,223EFF))'',*  MM3356 F WWQZ'::h555,,,223EFF))'',* # sOD$G"$ G-2GG%A0GG" J+3I>J$AI>=I>>JcF|jdj||y)Nr7)r~join)r1seqrs r6 writeSequencezPort.writeSequences 388C=$'r7c|jr tdt|s!t|st j |d||f|_|j j||fy)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 r r;r&connect)r1hostrs r6rz Port.connectsf   4 4 t)<++D2PQ Q#Tl T4L)r7c|j|jj||jr'|jj d|j yyrt) stopReadingrremoveActiveHandlerTryconnectionLostr<s r6_loseConnectionzPort._loseConnectionsF  ''- >> LL " "1d&9&9 : r7cz|jrtjx}|_nd}|j |SrJ)rTrDeferreddr)r1results r6 stopListeningzPort.stopListenings2 >>#nn. .FTVF  r7c\tjdtd|jy)Nz-Please use stopListening() to disconnect port) stacklevel)warningswarnDeprecationWarningrr<s r6loseConnectionzPort.loseConnection!s% ;  r7cZtjd|jzd|_tjj |||j j|jj|`|` t|dr|jjd|` yy)z& Cleans up my socket. z(UDP Port %s Closed)Nr)rrRrrr$rrdoStopr&closerUhasattrrcallback)r1reasons r6rzPort.connectionLost)s &)=)==>#**48   K   4  FFOOD ! r7cN|j|j}d|z|_y)zP Initialize the C{logstr} attribute to be used by C{logPrefix}. z%s (UDP)N)rSrlogstr)r1 logPrefixs r6r!zPort.setLogStr8s$&&t}}5  9, r7c|jS)zK Returns the name of my class, to prefix log entries with. )rr<s r6rzPort.logPrefix?rDr7c |jj}|jtjk(rt j dg|S|jtj k(rt jdg|ddSy)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 IPv4Addressr9 IPv6Address)r1rs r6getHostz Port.getHostEsq{{&&(    /&&u4t4 4   6?? 2&&u:Ra: :3r7cv|jjtjtj|y)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)r1enableds r6setBroadcastAllowedzPort.setBroadcastAllowedRs& v00&2E2EwOr7ct|jjtjtjS)z Checks if broadcast is currently allowed on this port. @return: Whether this port may broadcast. @rtype: L{bool} )boolr& getsockoptrrr<s r6getBroadcastAllowedzPort.getBroadcastAllowed[s,DKK**6+<+Q>QRSSr7) N) rintr2rr strr3rrIOCPReactor | NonereturnNone)rrrz socket.socketrJ)#__name__ __module__ __qualname____doc____annotations__r&r:r' SOCK_DGRAMr(dynamicReadBuffersrr%r#rArCrHrKrFrGrcr]r^r~rrrrrrr!rrrrr7r6rr#sNNM""J&*O]) !&* CC(C C  C $ C C6 @  "N0(+  ,B$?B( *;  - ;PTr7rcDeZdZdZ d ddZddZy) MulticastPortz. UDP Port that supports multicasting. NcFtj||||||||_yrJ)rr%listenMultiple)r1rr2r r3rrs r6r%zMulticastPort.__init__ks" dD%M7K,r7c$tj|}|jrn|jtj tj dttdr/|jtj tjd|S)NrM SO_REUSEPORT) rrKrrr&r SO_REUSEADDRrr)r1r4s r6rKzMulticastPort.createSocketwsb%    NN6,,f.A.A1 Ev~.v00&2E2EqI r7)rrNF)rrr2rr rr3rrrrrrrr)rrrrr%rKrr7r6rresc!&*$ - -( - -  - $ - -  -r7r)-r __future__rrfr&r.rtypingrrzope.interfacertwisted.internetrrr r twisted.internet._multicastr twisted.internet.abstractr r twisted.internet.iocpreactorrrr)"twisted.internet.iocpreactor.constrrr'twisted.internet.iocpreactor.interfacesrtwisted.internet.protocolrtwisted.pythonr$twisted.internet.iocpreactor.reactorrIListeningPort IUDPTransport ISystemHandler$rIMulticastTransportrrr7r6rs# *&>>6@G E>@   yT8  yT  yTx  Z + +,ND-r7