U AcjkOã@sòdZddlmZddlZddlZddlZddlmZmZm 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/m0Z0m1Z1ddl2m3Z3m4Z4m5Z5zddl6m7Z8Wne9k r:dZ:YnXe8Z:erRddl;miZ?e e&ej@ejAƒGdd„dejBe1e,ƒƒZCe:dk rže eCejDƒGdd„de-e.eCƒZEGdd„deCƒZFGdd„de+ƒZ*e ejGƒGdd„de1eƒƒZHdS)z TCP support for IOCP reactor é)Ú annotationsN)Ú TYPE_CHECKINGÚOptionalÚUnion)ÚclassImplementsÚ implementer)ÚaddressÚdeferÚerrorÚ interfacesÚmain)Ú _LogOwnerÚ isIPv6Address)Ú IPv4AddressÚ IPv6Address)Ú IProtocol)ÚabstractÚ iocpsupport)ÚERROR_CONNECTION_REFUSEDÚERROR_IO_PENDINGÚERROR_NETWORK_UNREACHABLEÚSO_UPDATE_ACCEPT_CONTEXTÚSO_UPDATE_CONNECT_CONTEXT)ÚIReadWriteHandle)ÚProtocol)Ú ConnectorÚ_AbortingMixinÚ_BaseBaseClientÚ_BaseTCPClientÚ _getsocknameÚ _resolveIPv6Ú _SocketCloser)ÚfailureÚlogÚreflect)ÚstartTLS)Ú IOCPReactorc@sÊeZdZdZdZd/dd„Zdd„Zdd „Zd d „Zd d „Z dd„Z dd„Z dd„Z dd„Z dd„Zdd„Zdd„Zdd„Zedk rŒd0dd „Zd!d"„Zd#d$„Zd1d%d&„Zd'd(„Zd)d*„Zd+d,„Zd-d.„ZdS)2Ú 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|_||_dS©N)rÚ FileHandleÚ__init__ÚsocketÚfilenoÚ getFileHandleÚprotocol)ÚselfÚsockÚprotoÚreactor©r3úD/tmp/pip-unpacked-wheel-nrvc_22y/twisted/internet/iocpreactor/tcp.pyr*FszConnection.__init__cCs|jSr()r+©r/r3r3r4Ú getHandleLszConnection.getHandlecCs@t|tƒr n$t|tƒr t|ƒ}ntdt|ƒƒ‚|j |¡dS)z` @param rbuffer: Data received. @type rbuffer: L{bytes} or L{bytearray} z%data must be bytes or bytearray, not N)Ú isinstanceÚbytesÚ bytearrayÚ TypeErrorÚtyper.Ú dataReceived)r/Zrbufferr3r3r4r<Os    zConnection.dataReceivedcCst | ¡||¡Sr()Ú_iocpÚrecvr-)r/ZbufflistÚevtr3r3r4ÚreadFromHandle^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-Z SEND_LIMITÚtobytes)r/Úbuffr?Z writeViewr3r3r4Ú writeToHandleas ÿzConnection.writeToHandlecCsxz|j d¡Wntk r$YnXt |jd¡}|rtz | ¡Wn.tk rrt  ¡}t   ¡|  |¡YnXdS)Né) r+ÚshutdownÚOSErrorr ÚIHalfCloseableProtocolr.ZwriteConnectionLostÚ BaseExceptionr"ÚFailurer#ÚerrÚconnectionLost)r/ÚpÚfr3r3r4Ú_closeWriteConnectionks z Connection._closeWriteConnectioncCsZt |jd¡}|rLz | ¡WqVtk rHt ¡| t  ¡¡YqVXn | |¡dSr() r rIr.ÚreadConnectionLostrJr#rLrMr"rK)r/ÚreasonrNr3r3r4rQys zConnection.readConnectionLostcCsX|jr dStj ||¡|dkp,| tj¡ }| |¡|j}|`|` |` | |¡dSr() Ú disconnectedrr)rMÚcheckr ZConnectionAbortedÚ _closeSocketr.r+r-)r/rRZisCleanr.r3r3r4rM„s zConnection.connectionLostcCs|jS)zN Return the prefix to log with when I own the logging thread. )Úlogstrr5r3r3r4Ú logPrefixszConnection.logPrefixcCst|j tjtj¡ƒSr()Úboolr+Ú getsockoptÚ IPPROTO_TCPÚ TCP_NODELAYr5r3r3r4Ú getTcpNoDelay–szConnection.getTcpNoDelaycCs|j tjtj|¡dSr()r+Ú setsockoptrZr[©r/Zenabledr3r3r4Ú setTcpNoDelay™szConnection.setTcpNoDelaycCst|j tjtj¡ƒSr()rXr+rYÚ SOL_SOCKETÚ SO_KEEPALIVEr5r3r3r4ÚgetTcpKeepAliveœszConnection.getTcpKeepAlivecCs|j tjtj|¡dSr()r+r]r`rar^r3r3r4ÚsetTcpKeepAliveŸszConnection.setTcpKeepAliveTcCst|||tjƒdS)z9 @see: L{ITLSTransport.startTLS} N)Ú _startTLSrr))r/ZcontextFactoryÚnormalr3r3r4r%¤szConnection.startTLScCs0|jr dS|jr|j |¡ntj ||¡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)rSÚTLSr.Úwriterr))r/Údatar3r3r4rgªs zConnection.writecCs0|jr dS|jr|j |¡ntj ||¡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)rSrfr.Ú writeSequencerr))r/Ziovecr3r3r4ri¹s zConnection.writeSequencecCs0|jr|jr,|js,|j ¡ntj ||¡dS)z® Close the underlying handle or, if TLS has been started, first shut it down. @see: L{twisted.internet.interfaces.ITransport.loseConnection} N)rfÚ connectedÚ disconnectingr.ÚloseConnectionrr))r/rRr3r3r4rlÈs  zConnection.loseConnectioncCs*|jr|j ||¡ntj |||¡dS)zc Register a producer. If TLS is enabled, the TLS connection handles this. N)rfr.ÚregisterProducerrr))r/ZproducerZ streamingr3r3r4rmÕszConnection.registerProducercCs"|jr|j ¡n tj |¡dS)ze Unregister a producer. If TLS is enabled, the TLS connection handles this. N)rfr.ÚunregisterProducerrr)r5r3r3r4rnäs zConnection.unregisterProducercCsdSr(r3r5r3r3r4ÚgetHostïszConnection.getHostcCsdSr(r3r5r3r3r4ÚgetPeerószConnection.getPeer)N)T)N)Ú__name__Ú __module__Ú __qualname__Ú__doc__rfr*r6r<r@rErPrQrMrWr\r_rbrcrdr%rgrirlrmrnrorpr3r3r3r4r'<s0       r'c@sTeZdZdZejZejZdZ e Z dd„Z dd„Z dd„Zd d „Zd d „Zd d„ZdS)Ú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*|dkr d}||_t ||||||¡dS)N)Úr)r2rr*)r/ÚhostÚportÚ bindAddressÚ connectorr2r3r3r4r* szClient.__init__cCs|j |j|j¡S)zd Create a socket registered with the IOCP reactor. @see: L{_BaseTCPClient} )r2Ú createSocketÚ addressFamilyÚ socketTyper5r3r3r4ÚcreateInternetSocketszClient.createInternetSocketcCs |`|`dS)z– Clean up potentially circular references to the socket and to its C{getFileHandle} method. @see: L{_BaseBaseClient} N)r+r-r5r3r3r4Ú_collectSocketDetailsszClient._collectSocketDetailscCs|j |¡dS)z^ Remove the active handle from the reactor. @see: L{_BaseBaseClient} N)r2ÚremoveActiveHandler5r3r3r4Ú_stopReadingAndWriting!szClient._stopReadingAndWritingc Cs²|r0t ||¡}| t |tj |d¡f¡¡n~|j tj t t   d|j  ¡¡¡|j | ¡¡|_d|_| |j¡}|d|_|jdkrštƒ|_| ¡n|j |¡| ¡dS)Nz Unknown errorÚPTz,client)ÚconnectExErrorsÚgetZfailIfNotConnectedr ZgetConnectErrorÚerrnoÚ errorcoder+r]r`rÚstructÚpackr,rzÚ buildProtocolrpr.rjÚ _getLogPrefixrVrrlÚmakeConnectionÚ startReading)r/Úrcrhr?rWr3r3r4Ú cbConnect)s& ÿý     zClient.cbConnectcCsft|dƒsdStjst‚|j |¡t |j|¡}t |j   ¡|j |¡}|rb|t krb| |d|¡dS)Nrzr) Úhasattrr=Zhave_connectexÚAssertionErrorr2ÚaddActiveHandleÚEventrŽÚconnectr+r,Z realAddressr)r/r?rr3r3r4Ú doConnectGs    zClient.doConnectN)rqrrrsrtr+ÚAF_INETr|Ú SOCK_STREAMr}Ú_tlsClientDefaultr'Z_commonConnectionr*r~rrrŽr”r3r3r3r4ruüs ruc@sJeZdZdZdZdddddddœd d „Zd d œd d„Zdd„Zdd„ZdS)Ú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. Fz socket.socketrzUnion[IPv4Address, IPv6Address]Úintr&)r0r.Ú clientAddrÚ serverAddrÚ sessionnor2cCsvt ||||¡||_||_||_| |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*r›ršrœrŠr.rwrVÚformatÚ __class__rqrxÚrepstrrjrŒ)r/r0r.ršr›rœr2rWr3r3r4r*cs ýzServer.__init__Ústr©ÚreturncCs|jS)z= A string representation of this connection. )r r5r3r3r4Ú__repr__szServer.__repr__cCs|jS)zW Returns an IPv4Address. This indicates the server's address. )r›r5r3r3r4ro‡szServer.getHostcCs|jS)zW Returns an IPv4Address. This indicates the client's address. )ršr5r3r3r4rpszServer.getPeerN) rqrrrsrtr—r*r¤rorpr3r3r3r4r˜Us  r˜c@seZdZdd„ZdS)rcCst|j|j|j||jƒSr()rurwrxryr2r5r3r3r4Ú_makeTransport™szConnector._makeTransportN)rqrrrsr¥r3r3r3r4r˜src@s°eZdZUdZdZdZejZej Z e j Z dZdZded<dZd!d d „Zd d œd d„Zdd„Ze ej¡fdd„ZeZdd„Zdd„Zdd„Zdd„Zdd„Zdd„Z dd „Z!dS)"ÚPortFrNz Optional[int]Ú_realPortNumberÚTCPé2rvcCs:||_||_||_||_||_t|ƒr6tj|_t j |_ dSr() rxÚfactoryÚbacklogÚ interfacer2rr+ÚAF_INET6r|rrÚ _addressType)r/rxrªr«r¬r2r3r3r4r*±sz Port.__init__r¡r¢cCs6|jdk r d |j|jj|j¡Sd |j|jj¡SdS)Nz<{} of {} on {}>z<{} of {} (not listening)>)r§ržrŸrªr5r3r3r4r¤»s ýþz Port.__repr__c CszH|j |j|j¡}|jtjkr0t|j|jƒ}n |j|jf}|  |¡Wn4t k r|}zt   |j|j|¡‚W5d}~XYnXt  | ¡¡|_| ¡d|_t d| |j¡|jf¡|j ¡| |j¡d|_d|_|j |¡||_|jj|_| ¡dS)NrFz%s starting on %sTF)r2r{r|r}r+r­r r¬rxÚbindrHr ZCannotListenErrorr=Z maxAddrLenr,ÚaddrLenÚ getsocknamer§r#ÚmsgrŠrªZdoStartÚlistenr«rjrSr‘r-ÚdoAccept)r/ZsktÚaddrÚler3r3r4ÚstartListeningÈs.  $ÿÿ    zPort.startListeningcCs2d|_|jr.t ¡|_|j d|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)rkrjr ZDeferredÚdeferredr2Z callLaterrM)r/ZconnDoner3r3r4rlès  zPort.loseConnectioncCs t d|j›d|j›d¡dS)z. Log message for closing port ú(z Port z Closed)N)r#r²Ú_typer§r5r3r3r4Ú_logConnectionLostMsgøszPort._logConnectionLostMsgcCs°| ¡d|_d}t|dƒr&|j}|`d|_|j |¡d|_| d¡|` |` z|j   ¡Wn4t k r’d|_|dk rŒ| t ¡¡n‚YnXd|_|dk r¬| d¡dS)z' Cleans up the socket. Nr¸TF)r»r§rr¸rSr2r€rjrUr+r-rªZdoStoprJrkZerrbackr"rKÚcallback)r/rRÚdr3r3r4rMþs,   zPort.connectionLostcCst |jj¡S)zK Returns the name of my class, to prefix log entries with. )r$ZqualrªrŸr5r3r3r4rWszPort.logPrefixcCs|jdt|jƒžŽS)zf Returns an IPv4Address or IPv6Address. This indicates the server's address. r¨)r¨)r®rr+r5r3r3r4ro#sz Port.getHostcCs$| ||¡|js |js | ¡dSr()Ú handleAcceptrkrSr´)r/rrhr?r3r3r4ÚcbAccept+s  z Port.cbAcceptc Cs`|js |jrdS|r4t dtj |d¡|f¡dS|j t j t t   d|j  ¡¡¡t |j ¡|j¡\}}}||jks|t‚d|dkr²t|d d¡dƒ}|d|dd|f}d|dkrèt|d d¡dƒ}|d|dd|f}|j |jd |žŽ¡}|dkr|j ¡nD|j}|d|_t|j||jd |žŽ|jd |žŽ||jƒ} | | ¡d SdS) NFz*Could not accept new connection -- %s (%s)z unknown errorr‚ú%rrFr¨T)r¨)r¨)r¨)rkrSr#r²r…r†r„Únewsktr]r+r`rr‡rˆr,r=Zget_accept_addrsrDr|rr™Úsplitrªr‰r®Úcloserœr˜r2r‹) r/rr?ÚfamilyZlAddrZrAddrZscoper.ÚsÚ transportr3r3r4r¾0sJ ÿÿý       ú zPort.handleAcceptcCstt |j|¡}td|jdƒ|_}|j |j|j ¡|_ }t  |j   ¡|  ¡||¡}|rp|tkrp| ||¡dS)Néé)r=r’r¿r9r°rDr2r{r|r}rÁÚacceptr+r,rr¾)r/r?rDrÁrr3r3r4r´csÿ  z Port.doAccept)r©rvN)"rqrrrsrjrSrkr+r•r|r–r}rrr®rœr§Ú__annotations__rºr*r¤r·r"rKr ZCONNECTION_DONErlÚ stopListeningr»rMrWror¿r¾r´r3r3r3r4r¦s*     3r¦)IrtÚ __future__rr…r+r‡ÚtypingrrrZzope.interfacerrZtwisted.internetrr r r r Ztwisted.internet.abstractr rZtwisted.internet.addressrrZtwisted.internet.interfacesrZtwisted.internet.iocpreactorrrr=Z"twisted.internet.iocpreactor.constrrrrrZ'twisted.internet.iocpreactor.interfacesrZtwisted.internet.protocolrZtwisted.internet.tcprZ TCPConnectorrrrrr r!Ztwisted.pythonr"r#r$Ztwisted.internet._newtlsr%Z __startTLSÚ ImportErrorrdZ$twisted.internet.iocpreactor.reactorr&ZWSAECONNREFUSEDZWSAENETUNREACHrƒZ ITCPTransportZ ISystemHandler)r'Z ITLSTransportrur˜ZIListeningPortr¦r3r3r3r4ÚsL    $   þ<  YC