3 ,sY@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||_||_||_dS)N)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|}|jj|||S)N)r) rrSSHForwardingClientclienthostportpackOpen_direct_tcpiprZ openChannel)rZaddrr rZ addrTupleZchannelOpenDatarrr buildProtocols  z(SSHListenForwardingFactory.buildProtocolN)__name__ __module__ __qualname__rrrrrrr sr c@s4eZdZddZddZddZddZd d Zd S) SSHListenForwardingChannelcCsFtjd|jt|jjdkr:|jjdd}|j|d|j_dS)Nzopened forwarding channel %s)rmsgidlenrbufwrite)r specificDatabrrr channelOpen's  z&SSHListenForwardingChannel.channelOpencCs |jdS)N)closed)rreasonrrr openFailed.sz%SSHListenForwardingChannel.openFailedcCs|jjj|dS)N)r transportr%)rdatarrr dataReceived1sz'SSHListenForwardingChannel.dataReceivedcCs|jjjdS)N)rr,loseConnection)rrrr eofReceived4sz&SSHListenForwardingChannel.eofReceivedcCs.t|dr*tjd|j|jjj|`dS)Nrz#closing local forwarding channel %s)hasattrrr!r"rr,r/)rrrrr)7s  z!SSHListenForwardingChannel.closedN)rrrr(r+r.r0r)rrrrr%s rc@seZdZdZdS) SSHListenClientForwardingChannels direct-tcpipN)rrrnamerrrrr2=sr2c@seZdZdZdS) SSHListenServerForwardingChannelsforwarded-tcpipN)rrrr3rrrrr4Asr4c@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__cCsRtjd|jt|j|jd|jd}t|t|}|j|j|j ||_ dS)z, See: L{channel.SSHChannel} zconnecting to %s:%irrN) rr!rr_reactorrrZ addCallbacks _setClient_closeZ_channelOpenDeferred)rr&Zepdrrrr(gs z'SSHConnectForwardingChannel.channelOpencCsd||_tjd|j|jr2|jjj|jd|_|jjddrX|j|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:%iNrr )rrr!rr7r,r%r$)rrrrrr;rsz&SSHConnectForwardingChannel._setClientcCstjd||jdS)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!r/)rr*rrrr<sz"SSHConnectForwardingChannel._closecCs(|jr|jjj|n|j|7_dS)z, See: L{channel.SSHChannel} N)rr,r%r7)rr-rrrr.sz(SSHConnectForwardingChannel.dataReceivedcCs:|jr6tjd|j|jjr&|j|jjj|`dS)z, See: L{channel.SSHChannel} z#closed remote forwarding channel %sN)rrr!r"r r/r,)rrrrr)s  z"SSHConnectForwardingChannel.closedN) rrr__doc__rr:rr(r;r<r.r)rrrrr5Gs   r5cCst|\}}t||||dS)N) remoteWindowremoteMaxPacketavatar)unpackOpen_direct_tcpipr5)r?r@r-rAZremoteHPZorigHPrrropenConnectForwardingClients  rCc@s$eZdZddZddZddZdS)rcCs||_d|_dS)N)r r$)rr rrrrszSSHForwardingClient.__init__cCs&|jr|j|7_n |jj|dS)N)r$r r%)rr-rrrr.sz SSHForwardingClient.dataReceivedcCs|jr|jjd|_dS)N)r r/)rr*rrrconnectionLosts z"SSHForwardingClient.connectionLostN)rrrrr.rErrrrrsrcCsl|\}}|\}}t|tr$|jd}t|tr8|jd}tj|tjd|}tj|tjd|}||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. zutf-8z>L) isinstancer encoder NSstructpack)Z destinationsourceconnHostconnPortorigHostorigPortrZorigrrrrs     rcCstj|\}}tr&t|tr&|jd}ttjd|ddd}tj|dd\}}trpt|trp|jd}ttjd|ddd}||f||ffS)z(Unpack the data to a usable format. zutf-8z>LNr) r getNSr rFbytesdecodeintrIunpack)r-rLrestrMrNrOrrrrBs  rBcCs|\}}tj|tjd|S)zv Pack the data for tcpip forwarding. @param peer: A tuple of the (host, port) . @type peer: L{tuple} z>L)r rHrIrJ)ZpeerrrrrrpackGlobal_tcpip_forwardsrWcCsJtj|\}}tr&t|tr&|jd}ttjd|ddd}||fS)Nzutf-8z>LrPr) r rQr rFrRrSrTrIrU)r-rrVrrrrunpackGlobal_tcpip_forwards  rX)#r>Z __future__rrrIZtwisted.internetrrZtwisted.internet.endpointsrrZtwisted.pythonrZtwisted.python.compatr r Ztwisted.conch.sshr r ZFactoryr r6rr2r4r5rCZProtocolrrZpackOpen_forwarded_tcpiprBZunpackOpen_forwarded_tcpiprWrXrrrr s* _