>Mc@s$dZddlZddlZddlZddlZddlmZmZddlm Z m Z m Z m Z m Z ddlmZddlmZmZddlmZmZmZmZddlmZmZdd lmZdd lm Z dd lm!Z!dd lm"Z"dd lm#Z#ddlm$Z$yddl%m&Z&m'Z'Wn!e(k rodZ'Z&dZ*n Xe j+fZ*iej,e#6ej-e$6Z.de/fdYZ0dej1efdYZ2de2fdYZ3de2fdYZ4defdYZdefdYZ5dS(s TCP support for IOCP reactor iN(t implementstdirectlyProvides(t interfacesterrortaddresstmaintdefer(t isIPAddress(t _SocketClosert Connector(tlogtfailuretreflecttutil(t iocpsupporttabstract(tIReadWriteHandle(tERROR_IO_PENDING(tSO_UPDATE_CONNECT_CONTEXT(tSO_UPDATE_ACCEPT_CONTEXT(tERROR_CONNECTION_REFUSED(tERROR_NETWORK_UNREACHABLE(tTLSMemoryBIOFactorytTLSMemoryBIOProtocolt _BypassTLScBs>eZdZdZdZdZdZddZRS(s0 L{_BypassTLS} is used as the transport object for the TLS protocol object used to implement C{startTLS}. Its methods skip any TLS logic which C{startTLS} enables. @ivar _connection: A L{Connection} which TLS has been started on which will be proxied to by this object. Any method which has its behavior altered after C{startTLS} will be skipped in favor of the base class's implementation. This allows the TLS protocol object to have direct access to the transport, necessary to actually implement TLS. cCs ||_dS(N(t _connection(tselft connection((sD/usr/lib/python2.7/dist-packages/twisted/internet/iocpreactor/tcp.pyt__init__4scCst|j|S(N(tgetattrR(Rtname((sD/usr/lib/python2.7/dist-packages/twisted/internet/iocpreactor/tcp.pyt __getattr__8scCstjj|j|S(N(Rt FileHandletwriteR(Rtdata((sD/usr/lib/python2.7/dist-packages/twisted/internet/iocpreactor/tcp.pyR!<scCstjj|j|S(N(RR t writeSequenceR(Rtiovec((sD/usr/lib/python2.7/dist-packages/twisted/internet/iocpreactor/tcp.pyR#@scCstjj|j|S(N(RR tloseConnectionR(Rtreason((sD/usr/lib/python2.7/dist-packages/twisted/internet/iocpreactor/tcp.pyR%DsN( t__name__t __module__t__doc__RRR!R#tNoneR%(((sD/usr/lib/python2.7/dist-packages/twisted/internet/iocpreactor/tcp.pyR(s      t ConnectioncBseZdZeeejejee Z ddZ dZ dZdZdZdZdZdZd Zd Zd Zd Zd Zedk redZndZdZddZRS(s @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. @ivar _tlsClientDefault: A flag which must be set by a subclass. If set to C{True}, L{startTLS} will default to initiating SSL as a client. If set to C{False}, L{startTLS} will default to initiating SSL as a server. cCs5tjj||||_|j|_||_dS(N(RR Rtsockettfilenot getFileHandletprotocol(Rtsocktprototreactor((sD/usr/lib/python2.7/dist-packages/twisted/internet/iocpreactor/tcp.pyRYs  cCs|jS(N(R,(R((sD/usr/lib/python2.7/dist-packages/twisted/internet/iocpreactor/tcp.pyt getHandle`scCs|jjt|dS(N(R/t dataReceivedtstr(Rtrbuffer((sD/usr/lib/python2.7/dist-packages/twisted/internet/iocpreactor/tcp.pyR4dscCstj|j||S(N(t_iocptrecvR.(Rtbufflisttevt((sD/usr/lib/python2.7/dist-packages/twisted/internet/iocpreactor/tcp.pytreadFromHandleiscCs(tj|jt|d|j|S(s Send C{buff} to current file handle using C{_iocp.send}. The buffer sent is limited to a size of C{self.SEND_LIMIT}. i(R7tsendR.tbuffert SEND_LIMIT(RtbuffR:((sD/usr/lib/python2.7/dist-packages/twisted/internet/iocpreactor/tcp.pyt writeToHandlemscCsyt|j|jdWntjk r3nXtj|jd}|ry|jWqt j }t j |j |qXndS(Ni(RR,t_socketShutdownMethodRRtIHalfCloseableProtocolR/R*twriteConnectionLostR tFailureR terrtconnectionLost(Rtptf((sD/usr/lib/python2.7/dist-packages/twisted/internet/iocpreactor/tcp.pyt_closeWriteConnectionvs  cCsdtj|jd}|rSy|jWq`tj|jtj q`Xn |j|dS(N( RRBR/R*treadConnectionLostR RERFR RD(RR&RG((sD/usr/lib/python2.7/dist-packages/twisted/internet/iocpreactor/tcp.pyRJs cCsItjj|||j|j}|`|`|`|j|dS(N(RR RFt _closeSocketR/R,R.(RR&R/((sD/usr/lib/python2.7/dist-packages/twisted/internet/iocpreactor/tcp.pyRFs  cCs|jS(sN Return the prefix to log with when I own the logging thread. (tlogstr(R((sD/usr/lib/python2.7/dist-packages/twisted/internet/iocpreactor/tcp.pyt logPrefixscCs"tj|jjtjtjS(N(toperatorttruthR,t getsockoptt IPPROTO_TCPt TCP_NODELAY(R((sD/usr/lib/python2.7/dist-packages/twisted/internet/iocpreactor/tcp.pyt getTcpNoDelayscCs |jjtjtj|dS(N(R,t setsockoptRQRR(Rtenabled((sD/usr/lib/python2.7/dist-packages/twisted/internet/iocpreactor/tcp.pyt setTcpNoDelayscCs"tj|jjtjtjS(N(RNROR,RPt SOL_SOCKETt SO_KEEPALIVE(R((sD/usr/lib/python2.7/dist-packages/twisted/internet/iocpreactor/tcp.pytgetTcpKeepAlivescCs |jjtjtj|dS(N(R,RTRWRX(RRU((sD/usr/lib/python2.7/dist-packages/twisted/internet/iocpreactor/tcp.pytsetTcpKeepAlivescCs|r|j}n |j }t||d}t||jt}||_|j|_|j|_t|t j t |_ |jj t|dS(s9 @see: L{ITLSTransport.startTLS} N(t_tlsClientDefaultRR*RR/tFalseR3tgetPeerCertificateRRt ISSLTransporttTruet_tlstmakeConnectionR(RtcontextFactorytnormaltclientt tlsFactoryt tlsProtocol((sD/usr/lib/python2.7/dist-packages/twisted/internet/iocpreactor/tcp.pytstartTLSs      cCs3|jr|jj|ntjj||dS(s 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{ITCPTransport.write} N(R`R/R!RR (RR"((sD/usr/lib/python2.7/dist-packages/twisted/internet/iocpreactor/tcp.pyR!s cCs3|jr|jj|ntjj||dS(s 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{ITCPTransport.writeSequence} N(R`R/R#RR (RR$((sD/usr/lib/python2.7/dist-packages/twisted/internet/iocpreactor/tcp.pyR#s cCsF|jr/|jrB|j rB|jjqBntjj||dS(s Close the underlying handle or, if TLS has been started, first shut it down. @see: L{ITCPTransport.loseConnection} N(R`t connectedt disconnectingR/R%RR (RR&((sD/usr/lib/python2.7/dist-packages/twisted/internet/iocpreactor/tcp.pyR%s N(R'R(R)RRRt ITCPTransportt ISystemHandlet_extraInterfacesR\R`R*RR3R4R;R@RIRJRFRMRSRVRYRZRR_RgR!R#R%(((sD/usr/lib/python2.7/dist-packages/twisted/internet/iocpreactor/tcp.pyR+Is,                tClientcBseZejZejZeZdZ dZ dZ dZ dZ dZdZdZdZd Zd ZRS( c Cs2||_||f|_||_|dkr6d}nyy|j|j|j}Wn0tjk r}tj |d|dnwXy|j |Wn0tjk r}tj |d|dnX||_t j ||d||j d|jWn,tj k r-}|j d|j|nXdS(Ntii(Rni(t connectortaddrR2R*t createSockett addressFamilyt socketTypeR,RtConnectBindErrortbindR+Rt callLatertresolveAddresstfailIfNotConnected( Rthosttportt bindAddressRoR2tskttseRE((sD/usr/lib/python2.7/dist-packages/twisted/internet/iocpreactor/tcp.pyRs&     cCs]t|jdr*|j|jdn/|jj|jd}|j|j|jdS(Ni(RRpt_setRealAddressR2tresolvet addCallbacksRx(Rtd((sD/usr/lib/python2.7/dist-packages/twisted/internet/iocpreactor/tcp.pyRwscCs$||jdf|_|jdS(Ni(Rpt realAddresst doConnect(RR((sD/usr/lib/python2.7/dist-packages/twisted/internet/iocpreactor/tcp.pyR~&scCs|js"|js"t|d r&dSy|jWntk rGn X|`|`|jj||j j t j ||` dS(NRo( Rht disconnectedthasattrRKtAttributeErrorR,R.R2tremoveActiveHandleRotconnectionFailedR RD(RRE((sD/usr/lib/python2.7/dist-packages/twisted/internet/iocpreactor/tcp.pyRx+s  cCs|jtjdS(s* Stop attempt to connect. N(RxRt UserError(R((sD/usr/lib/python2.7/dist-packages/twisted/internet/iocpreactor/tcp.pytstopConnecting<scCs|rFtj||}|jtj|tjj|dfn|jjtj t t j d|jj |jj|j|_t|_|jjjd|_|jj||jdS(Ns Unknown errortIs,client(tconnectExErrorstgetRxRtgetConnectErrorterrnot errorcodeR,RTRWRtstructtpackR-Rot buildProtocoltgetPeerR/R_Rht __class__R'RLRat startReading(RtrctbytesR:((sD/usr/lib/python2.7/dist-packages/twisted/internet/iocpreactor/tcp.pyt cbConnectCs cCst|dsdStjs"t|jj|tj|j|}tj|j j |j |}|r|t kr|j|dd|ndS(NRoi( RR7thave_connectextAssertionErrorR2taddActiveHandletEventRtconnectR,R-RR(RR:R((sD/usr/lib/python2.7/dist-packages/twisted/internet/iocpreactor/tcp.pyRSs!cCstjd|jjS(si Returns an IPv4Address. This indicates the address from which I am connecting. tTCP(Rt IPv4AddressR,t getsockname(R((sD/usr/lib/python2.7/dist-packages/twisted/internet/iocpreactor/tcp.pytgetHostascCstjd|jS(se Returns an IPv4Address. This indicates the address that I am connected to. R(RRR(R((sD/usr/lib/python2.7/dist-packages/twisted/internet/iocpreactor/tcp.pyRjscCs&d|j|jtj|f}|S(Ns<%s to %s at %x>(RRpR t unsignedID(Rts((sD/usr/lib/python2.7/dist-packages/twisted/internet/iocpreactor/tcp.pyt__repr__sscCsI|js%|jtjd|n tj|||jj|dS(Ntstring(RhRxRt ConnectErrorR+RFRo(RR&((sD/usr/lib/python2.7/dist-packages/twisted/internet/iocpreactor/tcp.pyRFys (R'R(R,tAF_INETRrt SOCK_STREAMRsR_R[RRwR~RxRRRRRRRF(((sD/usr/lib/python2.7/dist-packages/twisted/internet/iocpreactor/tcp.pyRms          tServercBs8eZdZeZdZdZdZdZRS(s Serverside socket-stream connection class. I am a serverside network connection transport; a socket which came from an accept() on a server. cCstj||||||_||_||_d|jjj||jjf|_ d|jjj|j|jj f|_ t |_ |jdS(s  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. s%s,%s,%ss<%s #%s on %s>N(R+Rt serverAddrt clientAddrt sessionnoR/RR'RyRLRztrepstrR_RhR(RR0R/RRRR2((sD/usr/lib/python2.7/dist-packages/twisted/internet/iocpreactor/tcp.pyRs    cCs|jS(s= A string representation of this connection. (R(R((sD/usr/lib/python2.7/dist-packages/twisted/internet/iocpreactor/tcp.pyRscCs|jS(sW Returns an IPv4Address. This indicates the server's address. (R(R((sD/usr/lib/python2.7/dist-packages/twisted/internet/iocpreactor/tcp.pyRscCs|jS(sW Returns an IPv4Address. This indicates the client's address. (R(R((sD/usr/lib/python2.7/dist-packages/twisted/internet/iocpreactor/tcp.pyRs( R'R(R)R\R[RRRR(((sD/usr/lib/python2.7/dist-packages/twisted/internet/iocpreactor/tcp.pyRs    R cBseZdZRS(cCs"t|j|j|j||jS(N(RmRyRzR{R2(R((sD/usr/lib/python2.7/dist-packages/twisted/internet/iocpreactor/tcp.pyt_makeTransports(R'R(R(((sD/usr/lib/python2.7/dist-packages/twisted/internet/iocpreactor/tcp.pyR stPortcBseZeejeZeZeZe j Z e j Z dZdZddddZdZdZejejdZeZdZdZd Zd Zd Zd Zd Z RS(ii2RncCs1||_||_||_||_||_dS(N(Rztfactorytbacklogt interfaceR2(RRzRRRR2((sD/usr/lib/python2.7/dist-packages/twisted/internet/iocpreactor/tcp.pyRs     cCsG|jdk r,d|j|jj|jfSd|j|jjfSdS(Ns<%s of %s on %s>s<%s of %s (not listening)>(t_realPortNumberR*RR(R((sD/usr/lib/python2.7/dist-packages/twisted/internet/iocpreactor/tcp.pyRs     cCsy8|jj|j|j}|j|j|jfWn1tjk rk}tj |j|j|fnXt j |j |_ |jd|_tjd|jj|jf|jj|j|jt|_t|_|jj|||_|jj |_|jdS(Nis%s starting on %s(R2RqRrRsRuRRzR,RtCannotListenErrorR7t maxAddrLenR-taddrLenRRR tmsgRRtdoStarttlistenRR_RhR\RRR.tdoAccept(RR|tle((sD/usr/lib/python2.7/dist-packages/twisted/internet/iocpreactor/tcp.pytstartListenings$     cCsEt|_|jrAtj|_|jjd|j||jSdS(s 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. iN( R_RiRhRtDeferredtdeferredR2RvRF(RtconnDone((sD/usr/lib/python2.7/dist-packages/twisted/internet/iocpreactor/tcp.pyR%s   cCstjd|jfdS(s. Log message for closing port s(TCP Port %s Closed)N(R RR(R((sD/usr/lib/python2.7/dist-packages/twisted/internet/iocpreactor/tcp.pyt_logConnectionLostMsgscCs|jd|_d}t|dr:|j}|`nt|_|jj|t |_ |j |` |` y|jjWn5t |_|dk r|jtjqn&Xt |_|dk r|jdndS(s' Cleans up the socket. RN(RR*RRRR_RR2RR\RhRKR,R.RtdoStopRiterrbackR RDtcallback(RR&R((sD/usr/lib/python2.7/dist-packages/twisted/internet/iocpreactor/tcp.pyRFs,           cCstj|jjS(sK Returns the name of my class, to prefix log entries with. (R tqualRR(R((sD/usr/lib/python2.7/dist-packages/twisted/internet/iocpreactor/tcp.pyRM8scCstjd|jjS(sW Returns an IPv4Address. This indicates the server's address. R(RRR,R(R((sD/usr/lib/python2.7/dist-packages/twisted/internet/iocpreactor/tcp.pyR?scCs3|j|||jp|js/|jndS(N(t handleAcceptRiRR(RRRR:((sD/usr/lib/python2.7/dist-packages/twisted/internet/iocpreactor/tcp.pytcbAcceptHsc Cso|js|jrtS|rFtjdtjj|d|ftS|jj t j t t jd|j jtj|jj|j\}}}||jkst|jjtjd|d|d}|dkr|jjnr|j}|d|_t|j|tjd|d|dtjd|d|d||j}|j|t SdS(Ns*Could not accept new connection -- %s (%s)s unknown errorRRii(!RiRR\R RRRRtnewsktRTR,RWRRRR-R7tget_accept_addrsR?RrRRRRt_ServerFactoryIPv4AddressR*tcloseRRRR2RaR_( RRR:tfamilytlAddrtrAddrR/Rt transport((sD/usr/lib/python2.7/dist-packages/twisted/internet/iocpreactor/tcp.pyRNs.        cCstj|j|}tjd|jd|_}|jj|j|j |_ }tj |j j |j ||}|r|tkr|j||ndS(Nii(R7RRtAllocateReadBufferRR?R2RqRrRsRtacceptR,R-RR(RR:R?RR((sD/usr/lib/python2.7/dist-packages/twisted/internet/iocpreactor/tcp.pyRns!'N(!R'R(RRtIListeningPortR\RhRRiR,RRrRRsRR*RRRRR RDRtCONNECTION_DONER%t stopListeningRRFRMRRRR(((sD/usr/lib/python2.7/dist-packages/twisted/internet/iocpreactor/tcp.pyRs(        ((6R)R,RNRRtzope.interfaceRRttwisted.internetRRRRRttwisted.internet.abstractRttwisted.internet.tcpRR t TCPConnectorttwisted.pythonR R R R ttwisted.internet.iocpreactorRR7Rt'twisted.internet.iocpreactor.interfacesRt"twisted.internet.iocpreactor.constRRRRRttwisted.protocols.tlsRRt ImportErrorR*Rlt ITLSTransporttWSAECONNREFUSEDtWSAENETUNREACHRtobjectRR R+RmRR(((sD/usr/lib/python2.7/dist-packages/twisted/internet/iocpreactor/tcp.pyts80("      !9