U ¹êW[¯Nã@sØdZddlZddlZddlZddlZddlmZmZddlm Z m Z m Z m Z m Z ddlmZddlmZmZddlmZmZmZmZmZmZmZddlmZmZmZdd l m!Z!m"Z"dd l#m$Z%m&Z&dd l'm(Z(dd l)m*Z*dd l)m+Z+ddl)m,Z,ddl)m-Z-ddl)m.Z.zddl/m0Z1Wne2k r6dZ1YnXe-ej3e.ej4iZ5ee(e j6e j7ƒGdd„de&j8eeƒƒZ9e1dk r„ee9e j:ƒGdd„deee9ƒZ;Gdd„de9ƒZdS)z TCP support for IOCP reactor éN)Ú implementerÚclassImplements)Ú interfacesÚerrorÚaddressÚmainÚdefer)ÚProtocol)Ú _LogOwnerÚ isIPv6Address)Ú _SocketCloserÚ ConnectorÚ_AbortingMixinÚ_BaseBaseClientÚ_BaseTCPClientÚ _resolveIPv6Ú _getsockname)ÚlogÚfailureÚreflect)Ú_PY3Ú nativeString)Ú iocpsupportÚabstract)ÚIReadWriteHandle)ÚERROR_IO_PENDING)ÚSO_UPDATE_CONNECT_CONTEXT)ÚSO_UPDATE_ACCEPT_CONTEXT)ÚERROR_CONNECTION_REFUSED)ÚERROR_NETWORK_UNREACHABLE)ÚstartTLSc@sºeZdZdZdZd+dd„Zdd„Zdd „Zd d „Zd d „Z dd„Z dd„Z dd„Z dd„Z dd„Zdd„Zdd„Zdd„Zedk rŒd,dd „Zd!d"„Zd#d$„Zd-d%d&„Zd'd(„Zd)d*„ZdS).Ú 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©r-úB/usr/lib/python3/dist-packages/twisted/internet/iocpreactor/tcp.pyr$3szConnection.__init__cCs|jSr")r%©r)r-r-r.Ú getHandle:szConnection.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)Zrbufferr-r-r.r6>s   ÿzConnection.dataReceivedcCst | ¡||¡Sr")Ú_iocpZrecvr')r)ZbufflistÚevtr-r-r.ÚreadFromHandleOszConnection.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)Ú memoryviewr7Úsendr'Z SEND_LIMITÚtobytes)r)Úbuffr8Z writeViewr-r-r.Ú writeToHandleSs  ÿzConnection.writeToHandlecCsrz|j d¡Wntjk r&YnXt |jd¡}|rnz | ¡Wn&t ¡}t   ¡|  |¡YnXdS)Né) r%ZshutdownrrÚIHalfCloseableProtocolr(ZwriteConnectionLostrÚFailurerÚerrÚconnectionLost)r)ÚpÚfr-r-r.Ú_closeWriteConnection]s z Connection._closeWriteConnectioncCsRt |jd¡}|rDz | ¡WqNt ¡| t ¡¡YqNXn | |¡dSr") rr@r(ÚreadConnectionLostrrBrCrrA)r)ÚreasonrDr-r-r.rGls zConnection.readConnectionLostcCsX|jr dStj ||¡|dkp,| tj¡ }| |¡|j}|`|` |` | |¡dSr") Ú disconnectedrr#rCZcheckrZConnectionAbortedÚ _closeSocketr(r%r')r)rHZisCleanr(r-r-r.rCxs ÿ zConnection.connectionLostcCs|jS)zN Return the prefix to log with when I own the logging thread. )Úlogstrr/r-r-r.Ú logPrefix†szConnection.logPrefixcCst |j tjtj¡¡Sr")ÚoperatorÚtruthr%Ú getsockoptÚ IPPROTO_TCPÚ TCP_NODELAYr/r-r-r.Ú getTcpNoDelaysÿzConnection.getTcpNoDelaycCs|j tjtj|¡dSr")r%Ú setsockoptrPrQ©r)Úenabledr-r-r.Ú setTcpNoDelay’szConnection.setTcpNoDelaycCst |j tjtj¡¡Sr")rMrNr%rOÚ SOL_SOCKETÚ SO_KEEPALIVEr/r-r-r.ÚgetTcpKeepAlive–sÿzConnection.getTcpKeepAlivecCs|j tjtj|¡dSr")r%rSrWrXrTr-r-r.ÚsetTcpKeepAlive›szConnection.setTcpKeepAliveTcCst|||tjƒdS)z9 @see: L{ITLSTransport.startTLS} N)Ú _startTLSrr#)r)ZcontextFactoryZnormalr-r-r.r  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)rIÚTLSr(Úwriterr#)r)Údatar-r-r.r]§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)rIr\r(Ú writeSequencerr#)r)Ziovecr-r-r.r_·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)r\Ú connectedÚ disconnectingr(ÚloseConnectionrr#)r)rHr-r-r.rbÇs  zConnection.loseConnectioncCs*|jr|j ||¡ntj |||¡dS)zc Register a producer. If TLS is enabled, the TLS connection handles this. N)r\r(ÚregisterProducerrr#)r)ZproducerZ streamingr-r-r.rcÕszConnection.registerProducercCs"|jr|j ¡n tj |¡dS)ze Unregister a producer. If TLS is enabled, the TLS connection handles this. N)r\r(ÚunregisterProducerrr#r/r-r-r.rdås zConnection.unregisterProducer)N)T)N)Ú__name__Ú __module__Ú __qualname__Ú__doc__r\r$r0r6r9r>rFrGrCrLrRrVrYrZr[r r]r_rbrcrdr-r-r-r.r!(s,     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)r,rr$)r)ÚhostÚportÚ bindAddressÚ connectorr,r-r-r.r$s ÿzClient.__init__cCs|j |j|j¡S)zd Create a socket registered with the IOCP reactor. @see: L{_BaseTCPClient} )r,Ú createSocketÚ addressFamilyÚ socketTyper/r-r-r.ÚcreateInternetSocket szClient.createInternetSocketcCs |`|`dS)z– Clean up potentially circular references to the socket and to its C{getFileHandle} method. @see: L{_BaseBaseClient} N)r%r'r/r-r-r.Ú_collectSocketDetailsszClient._collectSocketDetailscCs|j |¡dS)z^ Remove the active handle from the reactor. @see: L{_BaseBaseClient} N)r,ÚremoveActiveHandler/r-r-r.Ú_stopReadingAndWritingszClient._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ÚgetZfailIfNotConnectedrZgetConnectErrorÚerrnoÚ errorcoder%rSrWrÚstructÚpackr&rnÚ buildProtocolÚgetPeerr(r`Ú _getLogPrefixrKr rbÚmakeConnectionÚ startReading)r)Úrcr^r8rLr-r-r.Ú cbConnect&s&   ÿ þ     zClient.cbConnectcCsft|dƒsdStjst‚|j |¡t |j|¡}t |j   ¡|j |¡}|rb|t krb| |d|¡dS)Nrnr) Úhasattrr7Zhave_connectexÚAssertionErrorr,ÚaddActiveHandleÚEventrƒZconnectr%r&Z realAddressr)r)r8r‚r-r-r.Ú doConnectBs    zClient.doConnectN)rerfrgrhr%ÚAF_INETrpÚ SOCK_STREAMrqÚ_tlsClientDefaultr!Z_commonConnectionr$rrrsrurƒrˆr-r-r-r.riõs    ric@s4eZdZdZdZdd„Zdd„Zdd„Zd d „Zd S) Ú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. FcCspt ||||¡||_||_||_| |j¡}d|||jjf|_d|jj j |j|jj f|_ 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%s,%s,%sz<%s #%s on %s>TN)r!r$Ú serverAddrÚ clientAddrÚ sessionnorr(rkrKÚ __class__rerlÚrepstrr`r)r)r*r(rŽrrr,rLr-r-r.r$`s  ÿzServer.__init__cCs|jS)z= A string representation of this connection. )r‘r/r-r-r.Ú__repr__tszServer.__repr__cCs|jS)zW Returns an IPv4Address. This indicates the server's address. )rr/r-r-r.ÚgetHost{szServer.getHostcCs|jS)zW Returns an IPv4Address. This indicates the client's address. )rŽr/r-r-r.r~„szServer.getPeerN) rerfrgrhr‹r$r’r“r~r-r-r-r.rŒQs   rŒc@seZdZdd„ZdS)r cCst|j|j|j||jƒSr")rirkrlrmr,r/r-r-r.Ú_makeTransportsÿzConnector._makeTransportN)rerfrgr”r-r-r-r.r Žsr c@s eZdZdZdZdZejZej Z e j Z dZdZdZddd„Zd d „Zd d „Ze ej¡fd d„ZeZdd„Zdd„Zdd„Zdd„Zdd„Zdd„Zdd„Z dS)ÚPortFrNÚTCPé2rjcCs:||_||_||_||_||_t|ƒr6tj|_t j |_ dSr") rlÚfactoryÚbacklogÚ interfacer,r r%ÚAF_INET6rprZ IPv6AddressÚ _addressType)r)rlr˜r™ršr,r-r-r.r$ªsz Port.__init__cCs6|jdk r d|j|jj|jfSd|j|jjfSdS)Nz<%s of %s on %s>z<%s of %s (not listening)>)Ú_realPortNumberrr˜r/r-r-r.r’µs þÿz Port.__repr__c CszH|j |j|j¡}|jtjkr0t|j|jƒ}n |j|jf}|  |¡Wn6tj k r~}zt   |j|j|¡‚W5d}~XYnXt   | ¡¡|_| ¡d|_t d| |j¡|jf¡|j ¡| |j¡d|_d|_|j |¡||_|jj|_| ¡dS)Nr?z%s starting on %sTF)r,rorprqr%r›rršrlZbindrZCannotListenErrorr7Z maxAddrLenr&ÚaddrLenZ getsocknamerrÚmsgrr˜ZdoStartZlistenr™r`rIr†r'ÚdoAccept)r)ZsktZaddrÚler-r-r.Ú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)rar`rZDeferredÚdeferredr,Z callLaterrC)r)ZconnDoner-r-r.rbßs  zPort.loseConnectioncCst d|j|jf¡dS)z. Log message for closing port z(%s Port %s Closed)N)rrŸÚ_typerr/r-r-r.Ú_logConnectionLostMsgðszPort._logConnectionLostMsgcCs¨| ¡d|_d}t|dƒr&|j}|`d|_|j |¡d|_| d¡|` |` z|j   ¡Wn,d|_ |dk r„| t ¡¡n‚YnXd|_ |dk r¤| d¡dS)z' Cleans up the socket. Nr£TF)r¥rr„r£rIr,rtr`rJr%r'r˜ZdoStopraZerrbackrrAÚcallback)r)rHÚdr-r-r.rC÷s,   zPort.connectionLostcCst |jj¡S)zK Returns the name of my class, to prefix log entries with. )rZqualr˜rr/r-r-r.rLszPort.logPrefixcCs|jdt|jƒžŽS)zf Returns an IPv4Address or IPv6Address. This indicates the server's address. r–)r–)rœrr%r/r-r-r.r“sz Port.getHostcCs$| ||¡|js |js | ¡dSr")Ú handleAcceptrarIr )r)r‚r^r8r-r-r.ÚcbAccept's  z Port.cbAcceptc CsŽ|js |jrdS|r4t dtj |d¡|f¡dS|j t j t t   d|j  ¡¡¡t |j ¡|j¡\}}}tsšt|dƒ|df}t|dƒ|df}||jks¨t‚d|dkrÞt|d d¡dƒ}|d|dd|f}d|dkrt|d d¡dƒ}|d|dd|f}|j |jd |žŽ¡}|dkrB|j ¡nD|j}|d|_t|j||jd |žŽ|jd |žŽ||jƒ} | | ¡d SdS) NFz*Could not accept new connection -- %s (%s)z unknown errorrvrr?ú%r–T)r–)r–)r–) rarIrrŸryrzrxÚnewsktrSr%rWrr{r|r&r7Zget_accept_addrsr=rrrpr…ÚintÚsplitr˜r}rœÚcloserrŒr,r€) r)r‚r8ZfamilyZlAddrZrAddrZscoper(ÚsZ transportr-r-r.r¨-sP ÿþ ÿ   ÿ     ý zPort.handleAcceptcCstt |j|¡}td|jdƒ|_}|j |j|j ¡|_ }t  |j   ¡|  ¡||¡}|rp|tkrp| ||¡dS)Néé)r7r‡r©r3ržr=r,rorprqr«Zacceptr%r&rr¨)r)r8r=r«r‚r-r-r.r ^s ÿ  z Port.doAccept)r—rjN)!rerfrgr`rIrar%r‰rprŠrqrZ IPv4Addressrœrrr¤r$r’r¢rrArZCONNECTION_DONErbZ stopListeningr¥rCrLr“r©r¨r r-r-r-r.r••s*     1r•)?rhr%rMryr{Zzope.interfacerrZtwisted.internetrrrrrZtwisted.internet.protocolr Ztwisted.internet.abstractr r Ztwisted.internet.tcpr r Z TCPConnectorrrrrrZtwisted.pythonrrrZtwisted.python.compatrrZtwisted.internet.iocpreactorrr7rZ'twisted.internet.iocpreactor.interfacesrZ"twisted.internet.iocpreactor.constrrrrrZtwisted.internet._newtlsr r[Ú ImportErrorZWSAECONNREFUSEDZWSAENETUNREACHrwZ ITCPTransportZ ISystemHandler#r!Z ITLSTransportrirŒZIListeningPortr•r-r-r-r.ÚsH  $       þÿG  \=