o i1@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)rprotocolZreadBufferSizer setLogStr_connectedAddr_setAddressFamilyr FileHandle__init__socket addressFamily socketType_iocpZ maxAddrLenfileno bytearray addressBufferstructcalcsizeaddressLengthBuffer)selfrrrr rsktZaddrLenr7fC:\Users\psant\Downloads\PAF backend\school\venv\Lib\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_INETrInvalidAddressErrorr5r7r7r8r(Us    zPort._setAddressFamilycCs2|jdurd|jjd|jdSd|jjdS)Nz not connected>)rr% __class__r<r7r7r8__repr__bs z Port.__repr__cC|jS)z) Return a socket object. N)r+r<r7r7r8 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<r7r7r8startListeningns zPort.startListening socket.socketcCs|j|j|jSN)r createSocketr,r-r<r7r7r8rIxszPort.createSocketc Csz|}||j|jfWnty$}z t|j|j|d}~ww|d|_t d| |j |jfd|_ ||_|jj|_dS)Nz%s starting on %sT)rIbindrrOSErrorrZCannotListenError getsocknamerrmsg _getLogPrefixr% connectedr+r/ getFileHandle)r5r6ler7r7r8rD{s zPort._bindSocketcCs$|j|||j|dSrH)r%ZmakeConnectionZ startReadingrZaddActiveHandler<r7r7r8rEs zPort._connectToProtocolcCs$|jr|||||dSdSrH)reading handleReaddoReadr5rcdataevtr7r7r8cbReads 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%connectionRefusedrrN errorcodegetZdatagramReceivedbytesbuffr.Z makesockaddr addr_buff BaseExceptionerrrVr7r7r8rTs, zPort.handleReadcCs~t|j|}|jd|_}|j|_}|j|_}t | ||||\}}|r;|t kr=|j d|j|||dSdSdSNr)r.EventrZZ _readBuffersrbr1rcr4 addr_len_buffrecvfromrQrr callLater)r5rYrbrcrhrWrXr7r7r8rUs   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+sendrLargsr[ZWSAEINTRwriteZ WSAEMSGSIZErZMessageLengthErrorr\r]rrr%r^r r r;r,r9r:sendto)r5Zdatagramaddrsenor7r7r8rmsr              z Port.writecCs|d||dS)N)rmjoin)r5seqror7r7r8 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)r5hostrr7r7r8rws  z Port.connectcCs2||j||jr|jd|jdSdSrf)Z stopReadingrZremoveActiveHandlerPrjconnectionLostr<r7r7r8_loseConnections  zPort._loseConnectioncCs&|jr t}|_nd}||SrH)rPrZDeferreddrz)r5resultr7r7r8 stopListenings zPort.stopListeningcCstjdtdd|dS)Nz-Please use stopListening() to disconnect port) stacklevel)warningswarnDeprecationWarningr}r<r7r7r8loseConnection!s  zPort.loseConnectioncCsbtd|jd|_tj|||j|j |`|` t |dr/|j d|` dSdS)z& Cleans up my socket. z(UDP Port %s Closed)Nr{)rrNrr r)ryr%ZdoStopr+closerQhasattrr{callback)r5reasonr7r7r8ry)s    zPort.connectionLostcCs||j}d||_dS)zP Initialize the C{logstr} attribute to be used by C{logPrefix}. z%s (UDP)N)rOr%logstr)r5 logPrefixr7r7r8r&8s zPort.setLogStrcCrA)zK Returns the name of my class, to prefix log entries with. N)rr<r7r7r8r?rCzPort.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} ZUDPNr~)r+rMr,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)r5enabledr7r7r8setBroadcastAllowedRszPort.setBroadcastAllowedcCst|jtjtjS)z Checks if broadcast is currently allowed on this port. @return: Whether this port may broadcast. @rtype: L{bool} N)boolr+ getsockoptrrr<r7r7r8getBroadcastAllowed[szPort.getBroadcastAllowed)rrN) rrrrrrr rrr!r"r#)r"rr"rGrH)#__name__ __module__ __qualname____doc____annotations__r+r:r, SOCK_DGRAMr-ZdynamicReadBuffersrr*r(r@rBrFrIrDrErZrTrUrmrurwrzr}rryr&rrrrr7r7r7r8r#sB      A  rc@s.eZdZdZ    ddddZdddZdS) MulticastPortz. UDP Port that supports multicasting. rrNFrrrrrrr rr!listenMultiplerr"r#cCst||||||||_dSrH)rr*r)r5rrrr rrr7r7r8r*ks zMulticastPort.__init__rGcCsBt|}|jr|tjtjdttdr|tjtjd|S)NrJ SO_REUSEPORT) rrIrrr+r SO_REUSEADDRrr)r5r6r7r7r8rIws  zMulticastPort.createSocket)rrNF)rrrrrrr rrr!rrr"r#r)rrrrr*rIr7r7r7r8res  r)-r __future__rr[r+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)rZIMulticastTransportrr7r7r7r8s:       >