U  W[@sdZddlmZmZddlZddlmZmZddlm Z m Z ddl m Z ddl mZmZddlmZmZGd d d ejZGd d d ejZGd ddeZGdddeZGdddejZddZGdddejZddZeZddZeZ ddZ!ddZ"dS)z This module contains the implementation of the TCP forwarding, which allows clients and servers to forward arbitrary TCP data across the connection. Maintainer: Paul Swartz )divisionabsolute_importN)protocolreactor)HostnameEndpointconnectProtocol)log)_PY3unicode)commonchannelc@seZdZddZddZdS)SSHListenForwardingFactorycCs||_||_||_dSN)connhostportklass)selfZ connectionrrr>/usr/lib/python3/dist-packages/twisted/conch/ssh/forwarding.py__init__sz#SSHListenForwardingFactory.__init__cCsF|j|jd}t|}||_|j|jf}t|j|}|j|||S)N)r) rrSSHForwardingClientclienthostportpackOpen_direct_tcpiprZ openChannel)rZaddrr rZ addrTupleZchannelOpenDatarrr buildProtocols  z(SSHListenForwardingFactory.buildProtocolN)__name__ __module__ __qualname__rrrrrrr sr c@s4eZdZddZddZddZddZd d Zd S) SSHListenForwardingChannelcCsFtd|jt|jjdkr:|jjdd}||d|j_dS)Nzopened forwarding channel %s)rmsgidlenrbufwrite)r specificDatabrrr channelOpen's  z&SSHListenForwardingChannel.channelOpencCs |dSr)closedrreasonrrr openFailed.sz%SSHListenForwardingChannel.openFailedcCs|jj|dSr)r transportr&rdatarrr dataReceived1sz'SSHListenForwardingChannel.dataReceivedcCs|jjdSr)rr.loseConnectionrrrr eofReceived4sz&SSHListenForwardingChannel.eofReceivedcCs.t|dr*td|j|jj|`dS)Nrz#closing local forwarding channel %s)hasattrrr"r#rr.r2r3rrrr*7s  z!SSHListenForwardingChannel.closedN)rrrr)r-r1r4r*rrrrr%s rc@seZdZdZdS) SSHListenClientForwardingChannels direct-tcpipNrrrnamerrrrr6=sr6c@seZdZdZdS) SSHListenServerForwardingChannelsforwarded-tcpipNr7rrrrr9Asr9c@sDeZdZdZeZddZddZddZdd Z d d Z d d Z dS)SSHConnectForwardingChannela Channel used for handling server side forwarding request. It acts as a client for the remote forwarding destination. @ivar hostport: C{(host, port)} requested by client as forwarding destination. @type hostport: L{tuple} or a C{sequence} @ivar client: Protocol connected to the forwarding destination. @type client: L{protocol.Protocol} @ivar clientBuf: Data received while forwarding channel is not yet connected. @type clientBuf: L{bytes} @var _reactor: Reactor used for TCP connections. @type _reactor: A reactor. @ivar _channelOpenDeferred: Deferred used in testing to check the result of C{channelOpen}. @type _channelOpenDeferred: L{twisted.internet.defer.Deferred} cOs*tjj|f||||_d|_d|_dS)Nr!)r SSHChannelrrr clientBuf)rrargskwrrrr`sz$SSHConnectForwardingChannel.__init__cCsRtd|jt|j|jd|jd}t|t|}||j|j ||_ dS), See: L{channel.SSHChannel} zconnecting to %s:%irr N) rr"rr_reactorrrZ addCallbacks _setClient_closeZ_channelOpenDeferred)rr'Zepdrrrr)gsz'SSHConnectForwardingChannel.channelOpencCsd||_td|j|jr2|jj|jd|_|jjddrX||jjddd|j_dS)z Called when the connection was established to the forwarding destination. @param client: Client protocol connected to the forwarding destination. @type client: L{protocol.Protocol} zconnected to %s:%iNr r!)rrr"rr<r.r&r%)rrrrrrArsz&SSHConnectForwardingChannel._setClientcCstd||dS)z Called when failed to connect to the forwarding destination. @param reason: Reason why connection failed. @type reason: L{twisted.python.failure.Failure} zfailed to connect: %sN)rr"r2r+rrrrBsz"SSHConnectForwardingChannel._closecCs(|jr|jj|n|j|7_dS)r?N)rr.r&r<r/rrrr1sz(SSHConnectForwardingChannel.dataReceivedcCs:|jr6td|j|jjr&||jj|`dS)r?z#closed remote forwarding channel %sN)rrr"r#r r2r.r3rrrr*s  z"SSHConnectForwardingChannel.closedN) rrr__doc__rr@rr)rArBr1r*rrrrr:Gs   r:cCst|\}}t||||dS)N) remoteWindowremoteMaxPacketavatar)unpackOpen_direct_tcpipr:)rErFr0rGZremoteHPZorigHPrrropenConnectForwardingClients  rIc@s$eZdZddZddZddZdS)rcCs||_d|_dS)N)r r%)rr rrrrszSSHForwardingClient.__init__cCs&|jr|j|7_n |j|dSr)r%r r&r/rrrr1sz SSHForwardingClient.dataReceivedcCs|jr|jd|_dSr)r r2r+rrrconnectionLosts z"SSHForwardingClient.connectionLostN)rrrrr1rKrrrrrsrcCsl|\}}|\}}t|tr$|d}t|tr8|d}t|td|}t|td|}||S)a Pack the data suitable for sending in a CHANNEL_OPEN packet. @type destination: L{tuple} @param destination: A tuple of the (host, port) of the destination host. @type source: L{tuple} @param source: A tuple of the (host, port) of the source host. utf-8>L) isinstancer encoder NSstructpack)Z destinationsourceconnHostconnPortorigHostorigPortrZorigrrrrs     rcCst|\}}tr&t|tr&|d}ttd|ddd}t|dd\}}trpt|trp|d}ttd|ddd}||f||ffS)z(Unpack the data to a usable format. rLrMNr r ZgetNSr rNbytesdecodeintrQZunpack)r0rTrestrUrVrWrrrrHs  rHcCs|\}}t|td|S)zv Pack the data for tcpip forwarding. @param peer: A tuple of the (host, port) . @type peer: L{tuple} rM)r rPrQrR)ZpeerrrrrrpackGlobal_tcpip_forwardsr^cCsJt|\}}tr&t|tr&|d}ttd|ddd}||fS)NrLrMrXrrY)r0rr]rrrrunpackGlobal_tcpip_forwards  r_)#rDZ __future__rrrQZtwisted.internetrrZtwisted.internet.endpointsrrZtwisted.pythonrZtwisted.python.compatr r Ztwisted.conch.sshr r ZFactoryr r;rr6r9r:rIZProtocolrrZpackOpen_forwarded_tcpiprHZunpackOpen_forwarded_tcpipr^r_rrrrs* _