\eM`dZddlZddlZddlZddlmZddlmZmZddl m Z m Z m Z m Z mZddlmZmZddlmZmZddlmZmZmZmZmZdd lmZdd lmZdd l m!Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(dd l)m*Z*m+Z+m,Z, dd l-m.Z/e/Z0n #e1$rdZ0YnwxYweej2eej3iZ4eee j5e j6Gddej7e(e#Z8e0ee8e j9Gdde$e%e8Z:Gdde8Z;Gdde"Z!ee j<Gdde(eZ=dS)z TCP support for IOCP reactor N)Optional)classImplements implementer)addressdefererror interfacesmain) _LogOwner isIPv6Address)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)startTLSceZdZdZdZddZdZdZdZdZ d Z d Z d Z d Z d ZdZdZdZeddZdZdZddZdZdZdZdZdS) 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. FNcztj||||_|j|_||_dSN)r FileHandle__init__socketfileno getFileHandleprotocol)selfsockprotoreactors @lib/python3.11/site-packages/twisted/internet/iocpreactor/tcp.pyr&zConnection.__init__?s7$$T7333 ![ c|jSr$)r'r+s r/ getHandlezConnection.getHandleEs {r0ct|trnDt|trt|}ntdt |z|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+rbuffers r/r:zConnection.dataReceivedHsp gu % % U   + + UGnnGGCd7mmSTT T ""7+++++r0cRtj|||Sr$)_iocprecvr))r+bufflistevts r/readFromHandlezConnection.readFromHandleWs"z$,,..#>>>r0ct|}tj||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+buffr@ writeViews r/ writeToHandlezConnection.writeToHandleZsP t$$ z    )A,?"@"H"H"J"JC   r0c\ |jdn#t$rYnwxYwtj|jd}|rc |dS#t$r?tj }tj | |YdSwxYwdS)N) r'shutdownOSErrorr IHalfCloseableProtocolr*writeConnectionLost BaseExceptionrFailurererrconnectionLost)r+pfs r/_closeWriteConnectionz Connection._closeWriteConnectionds  K  # # # #    D   -dmT B B  ' '%%'''''  ' ' 'O%% ##A&&&&&& ' ' 's ** A AB)(B)c*tj|jd}|ra |dS#t$r=t j|tj YdSwxYw||dSr$) r rNr*readConnectionLostrPrrRrSrrQ)r+reasonrTs r/rXzConnection.readConnectionLostrs  -dmT B B  ( 7$$&&&&&  7 7 7 ##GO$5$5666666 7    ' ' ' ' 's4AA;:A;c|jrdStj|||dup|t j }|||j}|`|` |` ||dSr$) disconnectedr r%rScheckrConnectionAborted _closeSocketr*r'r))r+rYisCleanr*s r/rSzConnection.connectionLost}s    F**4888D.M U5L(M(M$M '"""= M K  '''''r0c|jS)zN Return the prefix to log with when I own the logging thread. )logstrr2s r/ logPrefixzConnection.logPrefix {r0czt|jtjtjSr$)boolr' getsockopt IPPROTO_TCP TCP_NODELAYr2s r/ getTcpNoDelayzConnection.getTcpNoDelays(DK**6+=v?QRRSSSr0cf|jtjtj|dSr$)r' setsockoptrgrhr+enableds r/ setTcpNoDelayzConnection.setTcpNoDelays( v163EwOOOOOr0czt|jtjtjSr$)rer'rf SOL_SOCKET SO_KEEPALIVEr2s r/getTcpKeepAlivezConnection.getTcpKeepAlives(DK**6+QRRSSSr0cf|jtjtj|dSr$)r'rkrprqrls r/setTcpKeepAlivezConnection.setTcpKeepAlives( v0&2EwOOOOOr0Tc>t|||tjdS)z9 @see: L{ITLSTransport.startTLS} N) _startTLSr r%)r+contextFactorynormals r/r zConnection.startTLSs! dNFH4G H H H H Hr0c|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)r[TLSr*writer r%)r+datas r/r{zConnection.writesY    F 8 2 M   % % % % %   % %dD 1 1 1 1 1r0c|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.writeSequence} N)r[rzr* writeSequencer r%)r+iovecs r/r~zConnection.writeSequencesY    F 8 ; M ' ' . . . . .   - -dE : : : : :r0c|jr-|jr"|js|jdSdSdSt j||dS)z Close the underlying handle or, if TLS has been started, first shut it down. @see: L{twisted.internet.interfaces.ITransport.loseConnection} N)rz connected disconnectingr*loseConnectionr r%)r+rYs r/rzConnection.loseConnectionsq 8 =~ /d&8 / ,,..... / / / /   . .tV < < < < +>r?+S+S&TUU      K " "!) C!3!3!5!566    !N88HHDM!DN**4=99I#i/DK}$!)  ##%%%%% ,,T222!!#####r0cbt|dsdStjsJ|j|tj|j|}tj|j |j |}|r$|tkr||d|dSdSdS)Nrr) hasattrr=have_connectexr.addActiveHandleEventrconnectr'r( realAddressr)r+r@rs r/ doConnectzClient.doConnect@st[))  F#### $$T***k$.$// ]4;--//1A3 G G  '"((( NN2q# & & & & & ' '((r0N)rrrrr'AF_INETr SOCK_STREAMr_tlsClientDefaultr"_commonConnectionr&rrrrrrr0r/rrs NM#J"SSSNNN,,,...$$$< ' ' ' ' 'r0rc4eZdZdZdZdZdefdZdZdZ dS) 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. Fczt||||||_||_||_||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& serverAddr clientAddr sessionnorr*rraformat __class__rrrepstrrr)r+r,r*rrrr.rbs r/r&zServer.__init__\s D$':::$$"&&t}55 "GGYGG1EGG +22 M # , N O     r0returnc|jS)z= A string representation of this connection. )rr2s r/__repr__zServer.__repr__rrcr0c|jS)zW Returns an IPv4Address. This indicates the server's address. )rr2s r/rzServer.getHostx r0c|jS)zW Returns an IPv4Address. This indicates the client's address. )rr2s r/rzServer.getPeerrr0N) rrrrrr&strrrrrr0r/rrNsn  ,# r0rceZdZdZdS)rcPt|j|j|j||jSr$)rrrrr.r2s r/_makeTransportzConnector._makeTransports!diD,z<{} of {} (not listening)>)rrrrr2s r/rz Port.__repr__s]   +%,, &$  066 & r0cl |j|j|j}|jtjkrt |j|j}n|j|jf}| |n2#t$r%}tj |j|j|d}~wwxYwtj||_|d|_t'j||jd|j|j||jd|_d|_|j|||_|jj|_|dS)NrKz starting on TF)r.rrrr'rrrrbindrMrCannotListenErrorr= maxAddrLenr(addrLen getsocknamerrmsgrrdoStartlistenrrr[rr)doAccept)r+sktaddrles r/startListeningzPort.startListenings I,++D,>PPC!V_44#DNDI>> 2 HHTNNNN I I I)$.$)RHH H I' 55  #003 !!$,////1E1E G    4<   ! $$T*** ![/ sA8A;; B* B%%B*cd|_|jr@tj|_|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)rrrDeferreddeferredr. callLaterrS)r+connDones r/rzPort.loseConnectionsT" > !!N,,DM L " "1d&98 D D D=  ! !r0cPtjd|jd|jddS)z. Log message for closing port (z Port z Closed)N)rr_typerr2s r/_logConnectionLostMsgzPort._logConnectionLostMsgs3 DDJDDd&:DDDEEEEEr0c|d|_d}t|dr |j}|`d|_|j|d|_|d|` |` |j d|_ ||ddSdS#t$r4d|_ |)|t#jYdSwxYw)z' Cleans up the socket. NrTF)rrrrr[r.rrr^r'r)rdoStoprcallbackrPerrbackrrQ)r+rYds r/rSzPort.connectionLosts ""$$$#  4 $ $  A   ''--- $ K   ! L   ! ! !"'D } 4     }   !&D } '/++,,,,,,  s9B449C20C2c>tj|jjS)zK Returns the name of my class, to prefix log entries with. )rqualrrr2s r/rbzPort.logPrefixs|DL2333r0cB|jdgt|jRS)zf Returns an IPv4Address or IPv6Address. This indicates the server's address. r)rrr'r2s r/rz Port.getHosts) !t Cdk)B)BCCCCr0c~||||js|js|dSdSdSr$) handleAcceptrr[r)r+rr|r@s r/cbAcceptz Port.cbAcceptsR "c"""" d&7  MMOOOOO    r0c 2|js|jrdS|r;tjdtj|dd|ddS|jtj ttj d|j tj|j|j\}}}||jksJd|dvr@t'|ddd }|d|d d|f}d|dvr@t'|ddd }|d|d d|f}|j|jd g|R}||jn\|j}|d z|_t5|j||jd g|R|jd g|R||j} || d S) NFz#Could not accept new connection -- z unknown errorz ()r%rrKrT)rr[rrrrrnewsktrkr'rprrrr(r=get_accept_addrsrGrintsplitrrrcloserrr.r) r+rr@familylAddrrAddrscoper*s transports r/rzPort.handleAccept!s;   !2 5 +  GG?&&r?;;;;RRRA   5 J ! !!( C!3!3!5!566    $)#9#*:K:K:M:Msx#X#X FE5T/////eAhE!HNN3//233q58Q6eAhE!HNN3//233q58Q6|112C$2CE2RE2R2R2RSSH   """"N!"Q"J%D%e4e444%D%e4e444L  '' 2224r0ctj|j|}td|jdzzx|_}|j|j|j x|_ }tj |j | ||}|r#|tkr|||dSdSdS)N)r=rrr7rrGr.rrrracceptr'r(rr)r+r@rGrrs r/rz Port.doAcceptTsk$-..$A):$;<<<4"l77      V\$+,,.. s K K  '"(((   b# & & & & & ' '((r0)rrN)%rrrrr[rr'rrrrr IPv4Addressrrrrr __annotations__rr&rrrrrQr CONNECTION_DONEr stopListeningrrSrbrrrrrr0r/rrs<ILMNM#J&LI&*OXc]))) E4444 #    @'6god6J&K&K ! ! ! !#MFFF !!!>444 DDD 111f ' ' ' ' 'r0r)>rrr'rtypingrzope.interfacerrtwisted.internetrrrr r twisted.internet.abstractr r twisted.internet.iocpreactorr rr="twisted.internet.iocpreactor.constrrrrr'twisted.internet.iocpreactor.interfacesrtwisted.internet.protocolrtwisted.internet.tcpr TCPConnectorrrrrrrtwisted.pythonrrrtwisted.internet._newtlsr __startTLSrv ImportErrorWSAECONNREFUSEDWSAENETUNREACHr ITCPTransport ISystemHandler%r" ITLSTransportrrIListeningPortrrr0r/r/s^ 77777777DDDDDDDDDDDDDD>>>>>>>>GGGGGGGGEDDDDD......1000000000??????IIIIIe3u3   z79QRRx x x x x $m^x x SRx v OJ 8999V'V'V'V'V'_njV'V'V'r88888Z888vRRRRR RRR  Z &''Q'Q'Q'Q'Q'=)Q'Q'('Q'Q'Q's2A;;BB