o >hkO@sdZddlmZddlZddlZddlZddlmZmZm Z ddl m Z m Z ddl mZmZmZmZmZddlmZmZddlmZmZdd lmZdd lmZmZdd lm Z m!Z!m"Z"m#Z#m$Z$dd l%m&Z&dd l'm(Z(ddl)m*Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1ddl2m3Z3m4Z4m5Z5zddl6m7Z8Wn e9ydZ:Ynwe8Z:erddl;miZ?e e&ej@ejAGdddejBe1e,ZCe:dure eCejDGddde-e.eCZEGdddeCZFGddde+Z*e ejGGddde1eZHdS)z TCP support for IOCP reactor ) annotationsN) TYPE_CHECKINGOptionalUnion)classImplements implementer)addressdefererror interfacesmain) _LogOwner isIPv6Address) IPv4Address IPv6Address) IProtocol)abstract iocpsupport)ERROR_CONNECTION_REFUSEDERROR_IO_PENDINGERROR_NETWORK_UNREACHABLESO_UPDATE_ACCEPT_CONTEXTSO_UPDATE_CONNECT_CONTEXT)IReadWriteHandle)Protocol) Connector_AbortingMixin_BaseBaseClient_BaseTCPClient _getsockname _resolveIPv6 _SocketCloser)failurelogreflect)startTLS) IOCPReactorc@seZdZdZdZd/ddZddZdd Zd d Zd d Z ddZ ddZ ddZ ddZ ddZddZddZddZedurFd0dd Zd!d"Zd#d$Zd/d%d&Zd'd(Zd)d*Zd+d,Zd-d.ZdS)1 Connectionz @ivar TLS: C{False} to indicate the connection is in normal TCP mode, C{True} to indicate that TLS has been started and that operations must be routed through the L{TLSMemoryBIOProtocol} instance. FNcCs&tj||||_|j|_||_dSN)r FileHandle__init__socketfileno getFileHandleprotocol)selfsockprotoreactorr3/var/www/vedio/testing/chatpythonscript.ninositsolution.com/env/lib/python3.10/site-packages/twisted/internet/iocpreactor/tcp.pyr*Fs zConnection.__init__cC|jSr()r+r/r3r3r4 getHandleLszConnection.getHandlecCs@t|trnt|trt|}ntdt||j|dS)z` @param rbuffer: Data received. @type rbuffer: L{bytes} or L{bytearray} z%data must be bytes or bytearray, not N) isinstancebytes bytearray TypeErrortyper. dataReceived)r/rbufferr3r3r4r=Os   zConnection.dataReceivedcCst|||Sr()_iocprecvr-)r/bufflistevtr3r3r4readFromHandle^szConnection.readFromHandlecCs(t|}t||d|j|S)z Send C{buff} to current file handle using C{_iocp.send}. The buffer sent is limited to a size of C{self.SEND_LIMIT}. r) memoryviewr?sendr- SEND_LIMITtobytes)r/buffrB writeViewr3r3r4 writeToHandleaszConnection.writeToHandlecCsxz|jdWn tyYnwt|jd}|r:z|WdSty9t }t | |YdSwdS)N) r+shutdownOSErrorr IHalfCloseableProtocolr.writeConnectionLost BaseExceptionr"Failurer#errconnectionLost)r/pfr3r3r4_closeWriteConnectionks  z Connection._closeWriteConnectioncCsZt|jd}|r&z|WdSty%t|t YdSw||dSr() r rNr.readConnectionLostrPr#rRrSr"rQ)r/reasonrTr3r3r4rWys zConnection.readConnectionLostcCsX|jrdStj|||dup|tj }|||j}|`|` |` ||dSr() disconnectedrr)rScheckr ConnectionAborted _closeSocketr.r+r-)r/rXisCleanr.r3r3r4rSs zConnection.connectionLostcCr5)zN Return the prefix to log with when I own the logging thread. )logstrr6r3r3r4 logPrefixzConnection.logPrefixcCt|jtjtjSr()boolr+ getsockopt IPPROTO_TCP TCP_NODELAYr6r3r3r4 getTcpNoDelayzConnection.getTcpNoDelaycC|jtjtj|dSr()r+ setsockoptrdrer/enabledr3r3r4 setTcpNoDelayzConnection.setTcpNoDelaycCrar()rbr+rc SOL_SOCKET SO_KEEPALIVEr6r3r3r4getTcpKeepAlivergzConnection.getTcpKeepAlivecCrhr()r+rirnrorjr3r3r4setTcpKeepAlivermzConnection.setTcpKeepAliveTcCst|||tjdS)z9 @see: L{ITLSTransport.startTLS} N) _startTLSrr))r/contextFactorynormalr3r3r4r%szConnection.startTLScC2|jrdS|jr|j|dStj||dS)z Write some data, either directly to the underlying handle or, if TLS has been started, to the L{TLSMemoryBIOProtocol} for it to encrypt and send. @see: L{twisted.internet.interfaces.ITransport.write} N)rYTLSr.writerr))r/datar3r3r4rw zConnection.writecCru)z Write some data, either directly to the underlying handle or, if TLS has been started, to the L{TLSMemoryBIOProtocol} for it to encrypt and send. @see: L{twisted.internet.interfaces.ITransport.writeSequence} N)rYrvr. writeSequencerr))r/iovecr3r3r4rzryzConnection.writeSequencecCs:|jr|jr|js|jdSdSdStj||dS)z Close the underlying handle or, if TLS has been started, first shut it down. @see: L{twisted.internet.interfaces.ITransport.loseConnection} N)rv connected disconnectingr.loseConnectionrr))r/rXr3r3r4r~s  zConnection.loseConnectioncCs,|jr |j||dStj|||dS)zc Register a producer. If TLS is enabled, the TLS connection handles this. N)rvr.registerProducerrr))r/producer streamingr3r3r4rszConnection.registerProducercCs$|jr |jdStj|dS)ze Unregister a producer. If TLS is enabled, the TLS connection handles this. N)rvr.unregisterProducerrr)r6r3r3r4rszConnection.unregisterProducercCdSr(r3r6r3r3r4getHostzConnection.getHostcCrr(r3r6r3r3r4getPeerrzConnection.getPeerr()T)__name__ __module__ __qualname____doc__rvr*r7r=rCrJrVrWrSr_rfrlrprqrrr%rwrzr~rrrrr3r3r3r4r'<s2        r'c@sTeZdZdZejZejZdZ e Z ddZ ddZ ddZd d Zd d Zd dZdS)Clientz @ivar _tlsClientDefault: Always C{True}, indicating that this is a client connection, and by default when TLS is negotiated this class will act as a TLS client. TcCs*|durd}||_t||||||dS)N)r)r2rr*)r/hostport bindAddress connectorr2r3r3r4r* szClient.__init__cCs|j|j|jS)zd Create a socket registered with the IOCP reactor. @see: L{_BaseTCPClient} )r2 createSocket addressFamily socketTyper6r3r3r4createInternetSocketszClient.createInternetSocketcCs |`|`dS)z Clean up potentially circular references to the socket and to its C{getFileHandle} method. @see: L{_BaseBaseClient} N)r+r-r6r3r3r4_collectSocketDetailss zClient._collectSocketDetailscCs|j|dS)z^ Remove the active handle from the reactor. @see: L{_BaseBaseClient} N)r2removeActiveHandler6r3r3r4_stopReadingAndWriting!szClient._stopReadingAndWritingc Cs|rt||}|t|tj|dfdS|jtj t t d|j |j||_d|_||j}|d|_|jdurOt|_|dS|j||dS)Nz Unknown errorPTz,client)connectExErrorsgetfailIfNotConnectedr getConnectErrorerrno errorcoder+rirnrstructpackr,r buildProtocolrr.r| _getLogPrefixr^rr~makeConnection startReading)r/rcrxrBr_r3r3r4 cbConnect)s&       zClient.cbConnectcCsnt|dsdStjs J|j|t|j|}t|j |j |}|r3|t kr5||d|dSdSdS)Nrr) hasattrr?have_connectexr2addActiveHandleEventrconnectr+r, realAddressr)r/rBrr3r3r4 doConnectGs    zClient.doConnectN)rrrrr+AF_INETr SOCK_STREAMr_tlsClientDefaultr'_commonConnectionr*rrrrrr3r3r3r4rs  rc@s8eZdZdZdZdddZdddZddZddZdS)ServeraV Serverside socket-stream connection class. I am a serverside network connection transport; a socket which came from an accept() on a server. @ivar _tlsClientDefault: Always C{False}, indicating that this is a server connection, and by default when TLS is negotiated this class will act as a TLS server. Fr0 socket.socketr.r clientAddrUnion[IPv4Address, IPv6Address] serverAddr sessionnointr2r&cCsvt||||||_||_||_||j}|d|d|jj|_d |jj j |j|jj |_ d|_|dS)a  Server(sock, protocol, client, server, sessionno) Initialize me with a socket, a protocol, a descriptor for my peer (a tuple of host, port describing the other end of the connection), an instance of Port, and a session number. ,z<{} #{} on {}>TN)r'r*rrrrr.rr^format __class__rrrepstrr|r)r/r0r.rrrr2r_r3r3r4r*cs  zServer.__init__returnstrcCr5)z= A string representation of this connection. )rr6r3r3r4__repr__r`zServer.__repr__cCr5)zW Returns an IPv4Address. This indicates the server's address. )rr6r3r3r4rzServer.getHostcCr5)zW Returns an IPv4Address. This indicates the client's address. )rr6r3r3r4rrzServer.getPeerN) r0rr.rrrrrrrr2r&rr) rrrrrr*rrrr3r3r3r4rUs   rc@seZdZddZdS)rcCst|j|j|j||jSr()rrrrr2r6r3r3r4_makeTransportrmzConnector._makeTransportN)rrrrr3r3r3r4rs rc@seZdZUdZdZdZejZej Z e j Z dZdZded<dZd!d d Zd"d dZddZeejfddZeZddZddZddZddZddZddZ dd Z!dS)#PortFrNz Optional[int]_realPortNumberTCP2rcCs>||_||_||_||_||_t|rtj|_t j |_ dSdSr() rfactorybacklog interfacer2rr+AF_INET6rrr _addressType)r/rrrrr2r3r3r4r*s z Port.__init__rrcCs2|jdurd|j|jj|jSd|j|jjS)Nz<{} of {} on {}>z<{} of {} (not listening)>)rrrrr6r3r3r4rs z Port.__repr__c Csz$|j|j|j}|jtjkrt|j|j}n|j|jf}| |Wnt y9}z t |j|j|d}~wwt ||_|d|_td||j|jf|j||jd|_d|_|j|||_|jj|_|dS)NrKz%s starting on %sTF)r2rrrr+rr rrbindrMr CannotListenErrorr? maxAddrLenr,addrLen getsocknamerr#msgrrdoStartlistenrr|rYrr-doAccept)r/sktaddrler3r3r4startListenings2       zPort.startListeningcCs2d|_|jrt|_|jd|j||jSdS)z Stop accepting connections on this port. This will shut down my socket and call self.connectionLost(). It returns a deferred which will fire successfully when the port is actually closed. TrN)r}r|r Deferreddeferredr2 callLaterrS)r/connDoner3r3r4r~s  zPort.loseConnectioncCs td|jd|jddS)z. Log message for closing port (z Port z Closed)N)r#r_typerr6r3r3r4_logConnectionLostMsgs zPort._logConnectionLostMsgcCs|d|_d}t|dr|j}|`d|_|j|d|_|d|` |` z|j Wnt yHd|_|durG|tYdSwd|_|durW|ddSdS)z' Cleans up the socket. NrTF)rrrrrYr2rr|r\r+r-rdoStoprPr}errbackr"rQcallback)r/rXdr3r3r4rSs0    zPort.connectionLostcCst|jjS)zK Returns the name of my class, to prefix log entries with. )r$qualrrr6r3r3r4r_szPort.logPrefixcCs|jdgt|jRS)zf Returns an IPv4Address or IPv6Address. This indicates the server's address. r)rrr+r6r3r3r4r#sz Port.getHostcCs,||||js|js|dSdSdSr() handleAcceptr}rYr)r/rrxrBr3r3r4cbAccept+s   z Port.cbAcceptc Csh|js|jrdS|rtdtj|d|fdS|jt j t t d|j t|j|j\}}}||jks>Jd|dvrYt|ddd}|d|dd|f}d|dvrtt|ddd}|d|dd|f}|j|jdg|R}|dur|jd S|j}|d|_t|j||jdg|R|jdg|R||j} || d S) NFz*Could not accept new connection -- %s (%s)z unknown errorr%rrKrT)r}rYr#rrrrnewsktrir+rnrrrr,r?get_accept_addrsrHrrsplitrrrcloserrr2r) r/rrBfamilylAddrrAddrscoper.s transportr3r3r4r0sL       zPort.handleAcceptcCs|t|j|}td|jd|_}|j|j|j |_ }t |j | ||}|r:|tkr<|||dSdSdS)N)r?rrr:rrHr2rrrracceptr+r,rr)r/rBrHrrr3r3r4rcs  z Port.doAccept)rrNr)"rrrr|rYr}r+rrrrrrrrr__annotations__rr*rrr"rQr CONNECTION_DONEr~ stopListeningrrSr_rrrrr3r3r3r4rs,       3r)Ir __future__rrr+rtypingrrrzope.interfacerrtwisted.internetrr r r r twisted.internet.abstractr rtwisted.internet.addressrrtwisted.internet.interfacesrtwisted.internet.iocpreactorrrr?"twisted.internet.iocpreactor.constrrrrr'twisted.internet.iocpreactor.interfacesrtwisted.internet.protocolrtwisted.internet.tcpr TCPConnectorrrrrr r!twisted.pythonr"r#r$twisted.internet._newtlsr% __startTLS ImportErrorrr$twisted.internet.iocpreactor.reactorr&WSAECONNREFUSEDWSAENETUNREACHr ITCPTransport ISystemHandler)r' ITLSTransportrrIListeningPortrr3r3r3r4sL    $   < YC