tf7zdZddlZddlZddlZddlZddlmZddlmZddl m Z m Z m Z m Z ddlmZmZddlmZmZddlmZmZmZdd lmZdd lmZmZeee je je j Gd d ej!Z"Gd dZ#ee j$Gdde#e"Z%dS)z UDP support for IOCP reactor N)Optional) implementer)addressdefererror interfaces) isIPAddress isIPv6Address)abstract iocpsupport)ERROR_CONNECTION_REFUSEDERROR_IO_PENDINGERROR_PORT_UNREACHABLE)IReadWriteHandle)failurelogceZdZUdZejZejZdZ dZ e e e d<ddZdZd efd Zd Zd Zd ZdZdZdZdZdZd dZdZdZdZdZdZd dZ dZ!dZ"dZ#dZ$dZ%dS)!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. FN_realPortNumber c||_||_||_||_|d|_|tj ||tj |j |j }tj|}t!||_t!t%jd|_dS)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 maxPacketSizereactorsktaddrLens _/var/www/surfInsights/venv3-11/lib/python3.11/site-packages/twisted/internet/iocpreactor/udp.pyr"z Port.__init__2s  +" "    $$T7333mD.@@"3::<<00&w//#,V_S-A-A#B#B   ct|jrtj|_dSt |jrtj|_dS|jrtj|jddS)z8 Resolve address family for the socket. znot an IPv4 or IPv6 addressN) r rr#AF_INET6r$r AF_INETrInvalidAddressErrorr.s r4r zPort._setAddressFamilyFsy  ( ( !'D     ( ( !'D    ^ + =   r5returnc`|jd|jjd|jdSd|jjdS)Nz not connected>)rr __class__r:s r4__repr__z Port.__repr__SsB   +Kt}.KKD4HKKK K?t}.??? ?r5c|jS)z) Return a socket object. )r#r:s r4 getHandlezPort.getHandleY {r5cV||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:s r4startListeningzPort.startListening_s.  !!!!!r5cL|j|j|jSN)r1 createSocketr$r%r:s r4rJzPort.createSocketis|(();T_MMMr5c |}||j|jfn2#t$r%}t j|j|j|d}~wwxYw|d|_tj | |j d|jd|_ ||_|jj|_dS)Nz starting on T)rJbindrrOSErrorrCannotListenError getsocknamerrmsg _getLogPrefixr connectedr#r( getFileHandle)r.r2les r4rEzPort._bindSocketls I##%%C HHdndi0 1 1 1 1 I I I)$.$)RHH H I #003 !!$-0000$2F2F H    ![/s58 A' A""A'c|j|||j|dSrI)rmakeConnection startReadingr1addActiveHandler:s r4rFzPort._connectToProtocolsF $$T***  $$T*****r5cn|jr-|||||dSdSrI)reading handleReaddoReadr.rcdataevts r4cbReadz Port.cbReads= <  OOBc * * * KKMMMMM  r5c|tjtjttfvr$|jr|jdSdS|r;tj dtj |dd|ddS |j t|jd|tj|jdS#t$$rtjYdSwxYw)Nzerror in recvfrom -- z unknown errorz ())errnoWSAECONNREFUSED WSAECONNRESETr rrrconnectionRefusedrrQ errorcodegetdatagramReceivedbytesbuffr& makesockaddr addr_buff BaseExceptionerrr^s r4r\zPort.handleReads!  !   $ "    " 2 //11111 2 2   GG?&&r?;;;;RRRA        ..#(5D5/**E,>s},M,M!     s A CC87C8c\tj|j|}|jdx|_}|jx|_}|jx|_}tj | ||||\}}|r0|tkr'|j d|j|||dSdSdSNr)r&Eventrb _readBuffersrmr*ror- addr_len_buffrecvfromrTrr1 callLater)r.rarmrorvr_r`s r4r]z Port.doReadsk$+t,,+A..4$($66 ,0,DDM>    $ =#  D  B"((( L " "1dk2tS A A A A A  B B((r5c|jr|d|jfvsJ |j|S#t$r}|jd}|t jkr||cYd}~S|t jkrtj d|t j t j ttfvr|jnYd}~dSd}~wwxYw|dksJt#|dsz0write() only accepts IP addresses, not hostnamesz*IPv6 port write() called with IPv4 addressz*IPv4 port write() called with IPv6 address)rr#sendrNargsreWSAEINTRwrite WSAEMSGSIZErMessageLengthErrorrfrgr rrrhr r r9r$r7r8sendto)r.datagramaddrsenos r4r}z Port.writes  8 D$"566666 {''111   WQZ''::h////////5,,,23EFFF)',*  M335555655555  4<<<<Q(( %d1g.. G},,/GO47## (:fo(M(M/GIT!W%% $*<*N*N/GI {))(D999   WQZ''::h555555555,,,23EFFF)',* FFFFF# sL0 C!1C+C!1A%CC!G## I;-2I6I;%A I65I66I;cX|d||dS)Nr5)r}join)r.seqrs r4 writeSequencezPort.writeSequences& 388C==$'''''r5c|jrtdt|s$t|st j|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 rr9r#connect)r.hostrs r4rz Port.connects   4 4   Rt)<)< R+D2PQQ Q#Tl T4L)))))r5c||j||jr"|jd|jdSdSrs) stopReadingr1removeActiveHandlerSrxconnectionLostr:s r4_loseConnectionzPort._loseConnections_  ''--- > ; L " "1d&9 : : : : : ; ;r5cv|jrtjx}|_nd}||SrI)rSrDeferreddr)r.results r4 stopListeningzPort.stopListening s> > #n.. .FTVVF  r5cftjdtd|dS)Nz-Please use stopListening() to disconnect port) stacklevel)warningswarnDeprecationWarningrr:s r4loseConnectionzPort.loseConnections> ;     r5cTtjd|jzd|_tj|||j|j |`|` t|dr|j d|` dSdS)z& Cleans up my socket. z(UDP Port %s Closed)Nr)rrQrr r!rrdoStopr#closerThasattrrcallback)r.reasons r4rzPort.connectionLosts &)==>>>#**4888   K   4    FOOD ! ! !  r5cN||j}d|z|_dS)zP Initialize the C{logstr} attribute to be used by C{logPrefix}. z%s (UDP)N)rRrlogstr)r. logPrefixs r4rzPort.setLogStr)s)&&t}55  9, r5c|jS)zK Returns the name of my class, to prefix log entries with. )rr:s r4rzPort.logPrefix0rCr5c|j}|jtjkrt jdg|RS|jtjkrt jdg|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#rPr$r8r IPv4Addressr7 IPv6Addressr.rs r4getHostz Port.getHost6sy{&&((   / /&u4t444 4  6? 2 2&u:RaR::: :3 2r5cf|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)r.enableds r4setBroadcastAllowedzPort.setBroadcastAllowedCs* v0&2EwOOOOOr5czt|jtjtjS)z Checks if broadcast is currently allowed on this port. @return: Whether this port may broadcast. @rtype: L{bool} )boolr# getsockoptrrr:s r4getBroadcastAllowedzPort.getBroadcastAllowedLs*DK**6+QRRSSSr5)rrNrI)&__name__ __module__ __qualname____doc__r#r8r$ SOCK_DGRAMr%dynamicReadBuffersrrint__annotations__r"r strr@rBrGrJrErFrbr\r]r}rrrrrrrrrrrr5r4rrsNM"J&*OXc])))CCCC(   @#@@@@  """NNN000(+++  ,BBB$????B((( * * *;;;     --- ; ; ;PPPTTTTTr5rcXeZdZdZdZdZdZdZdZdZ dZ dd Z d Z d Z dd ZdS)MulticastMixinz, Implement multicast functionality. c|jtjtj}tjt jd|S)Nz@i)r#r IPPROTO_IPIP_MULTICAST_IF inet_ntoar+pack)r.rs r4getOutgoingInterfacez#MulticastMixin.getOutgoingInterface[s< K " "6#4f6L M M D! 4 4555r5cf|j||jS)z. Returns Deferred of success. )r1resolve addCallback _setInterfacers r4setOutgoingInterfacez#MulticastMixin.setOutgoingInterface_s+|##D))55d6HIIIr5ctj|}|jtjtj|dS)NrL)r# inet_atonrrr)r.rrs r4rzMulticastMixin._setInterfacees7  T " " v0&2H!LLLqr5c`|jtjtjSrI)r#rrIP_MULTICAST_LOOPr:s r4getLoopbackModezMulticastMixin.getLoopbackModejs {%%f&79QRRRr5ctjdt|}|jtjtj|dS)Nb)r+rrr#rrr)r.modes r4setLoopbackModezMulticastMixin.setLoopbackModems?{3T ++ v0&2JDQQQQQr5c`|jtjtjSrI)r#rrIP_MULTICAST_TTLr:s r4getTTLzMulticastMixin.getTTLqs {%%f&79PQQQr5ctjd|}|jtjtj|dS)NB)r+rr#rrr)r.ttls r4setTTLzMulticastMixin.setTTLts9k#s## v0&2I3OOOOOr5rcj|j||j|dS)zF Join a multicast group. Returns Deferred of success. rLr1rr _joinAddr1r.rrs r4 joinGroupzMulticastMixin.joinGroupx/|##D))55doyRSTTTr5cj|j||j||SrI)r1rr _joinAddr2)r.rrrs r4rzMulticastMixin._joinAddr1~s-|##I..::4?DRVWWWr5cbtj|}tj|}|r tj}n tj} |jtj|||zdS#t $r4}tjtj ||g|j RcYd}~Sd}~wwxYwrI) r#rIP_ADD_MEMBERSHIPIP_DROP_MEMBERSHIPrrrNrFailurerMulticastJoinErrorr{)r.rrrcmdes r4rzMulticastMixin._joinAddr2s%%$Y//  ,*CC+C W K " "6#4c4);K L L L L L W W W?5#;D)#Uaf#U#U#UVV V V V V V V Ws)A00 B.:)B)#B.)B.cj|j||j|dS)zD Leave multicast group, return Deferred of success. rrrs r4 leaveGroupzMulticastMixin.leaveGrouprr5N)r)rrrrrrrrrrrrrrrrr5r4rrVs666JJJ  SSSRRRRRRPPPUUUU XXX W W WUUUUUUr5rc(eZdZdZ ddZdZdS) MulticastPortz. UDP Port that supports multicasting. rrNFcRt||||||||_dSrI)rr"listenMultiple)r.rr/rr0r1rs r4r"zMulticastPort.__init__s. dD%M7KKK,r5ct|}|jrk|tjtjdttdr+|tjtjd|S)NrL SO_REUSEPORT) rrJrrr#r SO_REUSEADDRrr)r.r2s r4rJzMulticastPort.createSocketsq%%   J NN6,f.A1 E E Ev~.. Jv0&2EqIII r5)rrNF)rrrrr"rJrr5r4rrsO - - - -r5r)&rrer#r+rtypingrzope.interfacertwisted.internetrrrrtwisted.internet.abstractr r twisted.internet.iocpreactorr r r&"twisted.internet.iocpreactor.constr rr'twisted.internet.iocpreactor.interfacesrtwisted.pythonrrIListeningPort IUDPTransport ISystemHandler!rrIMulticastTransportrrr5r4rs &&&&&&>>>>>>>>>>>>@@@@@@@@GGGGGGGG EDDDDD''''''''   qTqTqTqTqT8 qTqT  qTh ;U;U;U;U;U;U;U;U| Z +,,ND-,r5