o i @sdZddlZddlmZmZddlmZmZddlm Z m Z Gdddej Z Gdd d ej ZGd d d eZGd d d eZGdddej ZddZGdddejZddZeZddZeZddZddZdS)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 N)channelcommon)protocolreactor)HostnameEndpointconnectProtocolc@seZdZddZddZdS)SSHListenForwardingFactorycCs||_||_||_dSN)connhostportklass)self connectionr r rbC:\Users\psant\Downloads\PAF backend\school\venv\Lib\site-packages\twisted/conch/ssh/forwarding.py__init__s z#SSHListenForwardingFactory.__init__cCsF|j|jd}t|}||_|j|jf}t|j|}|j|||S)N)r ) r r SSHForwardingClientclienthostportpackOpen_direct_tcpipr Z openChannel)r addrrrZ addrTupleZchannelOpenDatarrr buildProtocols  z(SSHListenForwardingFactory.buildProtocolN)__name__ __module__ __qualname__rrrrrrrs rc@s4eZdZddZddZddZddZd d Zd S) SSHListenForwardingChannelcCsH|jjd|jdt|jjdkr|jjdd}||d|j_dS)Nzopened forwarding channel {id}id)_loginforlenrbufwrite)r specificDatabrrr channelOpen$s   z&SSHListenForwardingChannel.channelOpencCs |dSr )closedr reasonrrr openFailed+s z%SSHListenForwardingChannel.openFailedcCs|jj|dSr )r transportr%r datarrr dataReceived.sz'SSHListenForwardingChannel.dataReceivedcCs|jjdSr )rr-loseConnectionr rrr eofReceived1sz&SSHListenForwardingChannel.eofReceivedcCs4t|dr|jjd|jd|jj|`dSdS)Nrz%closing local forwarding channel {id}r)hasattrr!r"rrr-r1r2rrrr)4s  z!SSHListenForwardingChannel.closedN)rrrr(r,r0r3r)rrrrr#s  rc@eZdZdZdS) SSHListenClientForwardingChannels direct-tcpipNrrrnamerrrrr6;r6c@r5) SSHListenServerForwardingChannelsforwarded-tcpipNr7rrrrr:?r9r:c@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} cOs0tjj|g|Ri|||_d|_d|_dS)Nr )r SSHChannelrr r clientBuf)r r argskwrrrr]s z$SSHConnectForwardingChannel.__init__cCs`|jjd|jd|jddt|j|jd|jd}t|t|}||j|j ||_ dS), See: L{channel.SSHChannel} zconnecting to {host}:{port}rrrrN) r!r"r r_reactorrrZ addCallbacks _setClient_closeZ_channelOpenDeferred)r r&epdrrrr(cs z'SSHConnectForwardingChannel.channelOpencCsr||_|jjd|jd|jdd|jr |jj|jd|_|jjddr3||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 {host}:{port}rrrANr )rr!r"r r=r-r%r$)r rrrrrCos z&SSHConnectForwardingChannel._setClientcCs,|jjd|jd|jd|d|dS)z Called when failed to connect to the forwarding destination. @param reason: Reason why connection failed. @type reason: L{twisted.python.failure.Failure} z,failed to connect to {host}:{port}: {reason}rr)rrr+N)r!errorr r1r*rrrrDs z"SSHConnectForwardingChannel._closecCs*|jr |jj|dS|j|7_dS)r@N)rr-r%r=r.rrrr0sz(SSHConnectForwardingChannel.dataReceivedcCs@|jr|jjd|jd|jjr||jj|`dSdS)r@z%closed remote forwarding channel {id}rN)rr!r"rrr1r-r2rrrr)s z"SSHConnectForwardingChannel.closedN) rrr__doc__rrBrr(rCrDr0r)rrrrr;Cs  r;cCst|\}}t||||dS)N) remoteWindowremoteMaxPacketavatar)unpackOpen_direct_tcpipr;)rIrJr/rKZremoteHPZorigHPrrropenConnectForwardingClients rMc@s$eZdZddZddZddZdS)rcCs||_d|_dS)N)rr$)r rrrrrs zSSHForwardingClient.__init__cCs(|jr |j|7_dS|j|dSr )r$rr%r.rrrr0sz SSHForwardingClient.dataReceivedcCs|jr |jd|_dSdSr )rr1r*rrrconnectionLosts  z"SSHForwardingClient.connectionLostN)rrrrr0rOrrrrrs rcCsl|\}}|\}}t|tr|d}t|tr|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>LN) isinstancestrencoderNSstructpack) destinationsourceconnHostconnPortorigHostorigPortr origrrrrs     rcCst|\}}t|tr|d}ttd|ddd}t|dd\}}t|tr4|d}ttd|ddd}||f||ffS)z#Unpack the data to a usable format.rPrQNrrZgetNSrRbytesdecodeintrVunpack)r/rZrestr[r\r]rrrrLs    rLcCs|\}}t|td|S)zv Pack the data for tcpip forwarding. @param peer: A tuple of the (host, port) . @type peer: L{tuple} rQN)rrUrVrW)ZpeerrrrrrpackGlobal_tcpip_forwardsrfcCsFt|\}}t|tr|d}ttd|ddd}||fS)NrPrQr_rr`)r/rrerrrrunpackGlobal_tcpip_forwards   rg)rHrVZtwisted.conch.sshrrZtwisted.internetrrZtwisted.internet.endpointsrrFactoryrr<rr6r:r;rMProtocolrrZpackOpen_forwarded_tcpiprLZunpackOpen_forwarded_tcpiprfrgrrrrs&c