U ¹êW[À6ã@sdZddlZddlZddlZddlZddlZddlmZddlm Z m Z m Z m Z ddl mZmZddlmZmZddlmZddlmZdd lmZdd lmZdd lmZmZeee je je jƒGd d „d ej ƒƒZ!Gdd„dƒZ"ee j#ƒGdd„de"e!ƒƒZ$dS)z UDP support for IOCP reactor éN)Ú implementer)ÚdeferÚaddressÚerrorÚ interfaces)Ú isIPAddressÚ isIPv6Address)ÚlogÚfailure)ÚERROR_IO_PENDING)ÚERROR_CONNECTION_REFUSED)ÚERROR_PORT_UNREACHABLE)ÚIReadWriteHandle)Ú iocpsupportÚabstractc@sâeZdZdZejZejZdZ dZ d4dd„Z dd „Z d d „Z d d „Zdd„Zdd„Zdd„Zdd„Zdd„Zdd„Zdd„Zd5dd„Zdd„Zd d!„Zd"d#„Zd$d%„Zd&d'„Zd6d(d)„Zd*d+„Zd,d-„Zd.d/„Zd0d1„Z d2d3„Z!dS)7Ú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Úé cCsx||_||_||_||_| ¡d|_| ¡tj  ||¡t   |j |j ¡}t  | ¡¡}t|ƒ|_tt d¡ƒ|_dS)z> Initialize with a numeric port to listen on. NÚi)ÚportÚprotocolZreadBufferSizeÚ interfaceÚ setLogStrÚ_connectedAddrÚ_setAddressFamilyrÚ FileHandleÚ__init__ÚsocketÚ addressFamilyÚ socketTypeÚ_iocpZ maxAddrLenÚfilenoÚ bytearrayÚ addressBufferÚstructZcalcsizeÚaddressLengthBuffer)ÚselfrÚprotorÚ maxPacketSizeÚreactorÚsktZaddrLen©r+úB/usr/lib/python3/dist-packages/twisted/internet/iocpreactor/udp.pyr*s z Port.__init__cCs@t|jƒrtj|_n(t|jƒr(tj|_n|jrz<%s not connected>)Ú_realPortNumberrÚ __class__r0r+r+r,Ú__repr__Ms   ÿz Port.__repr__cCs|jS)z) Return a socket object. )rr0r+r+r,Ú getHandleUszPort.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Ú_connectToProtocolr0r+r+r,ÚstartListening\szPort.startListeningcCs|j |j|j¡S©N)r)Ú createSocketrrr0r+r+r,r9gszPort.createSocketc Csšz| ¡}| |j|jf¡Wn6tjk rT}zt |j|j|¡‚W5d}~XYnX| ¡d|_t   d|  |j ¡|jf¡d|_ ||_|jj|_dS)Néz%s starting on %sT)r9ZbindrrrrZCannotListenErrorÚ getsocknamer1r ÚmsgÚ _getLogPrefixrÚ connectedr!Ú getFileHandle)r&r*Úler+r+r,r5ks$ ÿzPort._bindSocketcCs$|j |¡| ¡|j |¡dSr8)rZmakeConnectionZ startReadingr)ZaddActiveHandler0r+r+r,r6~s zPort._connectToProtocolcCs |jr| |||¡| ¡dSr8)ZreadingÚ handleReadÚdoRead©r&ÚrcÚdataÚevtr+r+r,ÚcbRead„sz Port.cbReadcCsŠ|tjtjttfkr&|jr†|j ¡n`|rHt  dtj   |d¡|f¡n>z(|j  t |jd|…ƒt |j¡¡Wnt ¡YnXdS)Nzerror in recvfrom -- %s (%s)z unknown error)ÚerrnoÚWSAECONNREFUSEDÚ WSAECONNRESETr r rrÚconnectionRefusedr r<Z errorcodeÚgetZdatagramReceivedÚbytesÚbuffr Z makesockaddrÚ addr_buffÚerrrCr+r+r,rAŠs  ÿ ÿ ÿzPort.handleReadcCsnt |j|¡}|jd|_}|j|_}|j|_}t  |  ¡||||¡\}}|rj|t krj|  |||¡dS©Nr) r ZEventrGZ _readBuffersrNr#rOr%Ú addr_len_buffZrecvfromr?r rA)r&rFrNrOrRrDrEr+r+r,rBšs   ÿ z Port.doReadc Csæ|jr®|d|jfkst‚z|j |¡WStjk r¨}zb|jd}|tjkr`| |¡WY¢>S|tj krvt  d¡‚n"|tj tj t tfkr–|j ¡n‚W5d}~XYnXn4|dksºt‚t|dƒsît|dƒsî|ddkrît |dd¡‚t|dƒr|jtjkrt |dd¡‚t|dƒrF|jtjkrFt |dd¡‚z|j ||¡WStjk rà}zh|jd}|tjkr–| ||¡WY¢@S|tj kr®t  d¡‚n"|tj tj t tfkrÎWY¢ dS‚W5d}~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ÚsendrÚargsrHZWSAEINTRÚwriteZ WSAEMSGSIZEZMessageLengthErrorrIrJr r rrKrrr/rr-r.Zsendto)r&ZdatagramÚaddrZseZnor+r+r,rV§s`     ÿ   ÿþÿÿ     ÿ z Port.writecCs| d |¡|¡dS)Nó)rVÚjoin)r&ÚseqrWr+r+r,Ú writeSequenceÜszPort.writeSequencecCsH|jrtdƒ‚t|ƒs*t|ƒs*t |d¡‚||f|_|j ||f¡dS)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Ú RuntimeErrorrrrr/rÚconnect)r&Zhostrr+r+r,r]àsÿÿ z Port.connectcCs.| ¡|j |¡|jr*|j d|j¡dSrQ)Z stopReadingr)ZremoveActiveHandler>Z callLaterÚconnectionLostr0r+r+r,Ú_loseConnectionïs zPort._loseConnectioncCs&|jrt ¡}|_nd}| ¡|Sr8)r>rZDeferredÚdr_)r&Úresultr+r+r,Ú stopListeningös zPort.stopListeningcCstjdtdd| ¡dS)Nz-Please use stopListening() to disconnect porté)Ú stacklevel)ÚwarningsÚwarnÚDeprecationWarningrbr0r+r+r,ÚloseConnectionÿs ÿzPort.loseConnectioncCs^t d|j¡d|_tj ||¡|j ¡|j  ¡|`|` t |dƒrZ|j   d¡|` dS)z& Cleans up my socket. z(UDP Port %s Closed)Nr`)r r<r1rrr^rZdoStoprÚcloser?Úhasattrr`Úcallback)r&Úreasonr+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)r=rÚlogstr)r&Ú logPrefixr+r+r,rs zPort.setLogStrcCs|jS)zK Returns the name of my class, to prefix log entries with. )rmr0r+r+r,rnszPort.logPrefixcCsJ|j ¡}|jtjkr$tjd|žŽS|jtjkrFtjd|dd…žŽSdS)z¬ Return the local address of the UDP connection @returns: the local address of the UDP connection @rtype: L{IPv4Address} or L{IPv6Address} ÚUDPNrc)ro)ro)rr;rr.rZ IPv4Addressr-Z IPv6Address©r&rWr+r+r,ÚgetHost$s    z Port.getHostcCs|j tjtj|¡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&Úenabledr+r+r,ÚsetBroadcastAllowed2s ÿzPort.setBroadcastAllowedcCst |j tjtj¡¡S)z“ Checks if broadcast is currently allowed on this port. @return: Whether this port may broadcast. @rtype: L{bool} )ÚoperatorÚtruthrÚ getsockoptrsrtr0r+r+r,ÚgetBroadcastAllowed=sÿzPort.getBroadcastAllowed)rrN)N)N)"Ú__name__Ú __module__Ú __qualname__Ú__doc__rr.rZ SOCK_DGRAMrZdynamicReadBuffersr1rrr3r4r7r9r5r6rGrArBrVr[r]r_rbrhr^rrnrqrvrzr+r+r+r,rs<ÿ    5   rc@sleZdZdZdd„Zdd„Zdd„Zdd „Zd d „Zd d „Z dd„Z ddd„Z dd„Z dd„Z ddd„ZdS)ÚMulticastMixinz, Implement multicast functionality. cCs$|j tjtj¡}t t d|¡¡S)Nz@i)rryÚ IPPROTO_IPÚIP_MULTICAST_IFZ inet_ntoar$Úpack)r&rr+r+r,ÚgetOutgoingInterfaceOsz#MulticastMixin.getOutgoingInterfacecCs|j |¡ |j¡S)z. Returns Deferred of success. )r)ÚresolveÚ addCallbackÚ _setInterfacerpr+r+r,ÚsetOutgoingInterfaceTsz#MulticastMixin.setOutgoingInterfacecCs"t |¡}|j tjtj|¡dS)Nr:)rÚ inet_atonrrr€r)r&rWrr+r+r,r†[s zMulticastMixin._setInterfacecCs|j tjtj¡Sr8)rryr€ÚIP_MULTICAST_LOOPr0r+r+r,ÚgetLoopbackModeas ÿzMulticastMixin.getLoopbackModecCs*t dt |¡¡}|j tjtj|¡dS)NÚb)r$r‚rwrxrrrr€r‰)r&Úmoder+r+r,ÚsetLoopbackModefsÿzMulticastMixin.setLoopbackModecCs|j tjtj¡Sr8)rryr€ÚIP_MULTICAST_TTLr0r+r+r,ÚgetTTLls ÿzMulticastMixin.getTTLcCs$t d|¡}|j tjtj|¡dS)NÚB)r$r‚rrrr€rŽ)r&Zttlr+r+r,ÚsetTTLqs zMulticastMixin.setTTLrcCs|j |¡ |j|d¡S)zF Join a multicast group. Returns Deferred of success. r:©r)r„r…Ú _joinAddr1©r&rWrr+r+r,Ú joinGroupvsÿzMulticastMixin.joinGroupcCs|j |¡ |j||¡Sr8)r)r„r…Ú _joinAddr2)r&rWrrYr+r+r,r“~sÿzMulticastMixin._joinAddr1c Csˆt |¡}t |¡}|r tj}ntj}z|j tj|||¡WnBtjk r‚}z"t tj ||f|j žŽ¡WY¢Sd}~XYnXdSr8) rrˆZIP_ADD_MEMBERSHIPZIP_DROP_MEMBERSHIPrrr€rr ZFailureZMulticastJoinErrorrU)r&rrWrYÚcmdÚer+r+r,r–ƒs  ÿzMulticastMixin._joinAddr2cCs|j |¡ |j|d¡S)zD Leave multicast group, return Deferred of success. rr’r”r+r+r,Ú leaveGroup‘sÿzMulticastMixin.leaveGroupN)r)r)r{r|r}r~rƒr‡r†rŠrrr‘r•r“r–r™r+r+r+r,rIs rc@s"eZdZdZd dd„Zdd „ZdS) Ú MulticastPortz. UDP Port that supports multicasting. rrNFcCst ||||||¡||_dSr8)rrÚlistenMultiple)r&rr'rr(r)r›r+r+r,r szMulticastPort.__init__cCsBt |¡}|jr>| tjtjd¡ttdƒr>| tjtjd¡|S)Nr:Ú SO_REUSEPORT) rr9r›rrrrsZ SO_REUSEADDRrjrœ)r&r*r+r+r,r9¦s   zMulticastPort.createSocket)rrNF)r{r|r}r~rr9r+r+r+r,rššs ÿ rš)%r~rrwr$rerHZzope.interfacerZtwisted.internetrrrrZtwisted.internet.abstractrrZtwisted.pythonr r Z"twisted.internet.iocpreactor.constr r r Z'twisted.internet.iocpreactor.interfacesrZtwisted.internet.iocpreactorrr rZIListeningPortZ IUDPTransportZ ISystemHandlerrrZIMulticastTransportršr+r+r+r,Ús((     ÿ1Q