← Back
Editing: forwarding.cpython-38.pyc
U W[� � @ s d Z ddlmZmZ ddlZddlmZmZ ddlm Z m Z ddl m Z ddl mZmZ ddlmZmZ G d d � d ej�ZG d d � d ej�ZG d d� de�ZG dd� de�ZG dd� dej�Zdd� ZG dd� dej�Z dd� Z e Z dd� Z e Z dd � Z!d d � Z"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 � )�division�absolute_importN)�protocol�reactor)�HostnameEndpoint�connectProtocol)�log)�_PY3�unicode)�common�channelc @ s e Zd Zdd� Zdd� ZdS )�SSHListenForwardingFactoryc C s || _ || _|| _d S �N)�conn�hostport�klass)�selfZ connectionr r � r �>/usr/lib/python3/dist-packages/twisted/conch/ssh/forwarding.py�__init__ s z#SSHListenForwardingFactory.__init__c C sF | j | jd�}t|�}||_|j|jf}t| j|�}| j�||� |S )N)r ) r r �SSHForwardingClient�client�host�port�packOpen_direct_tcpipr Z openChannel)r Zaddrr r Z addrTupleZchannelOpenDatar r r � buildProtocol s z(SSHListenForwardingFactory.buildProtocolN)�__name__� __module__� __qualname__r r r r r r r s r c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd S ) �SSHListenForwardingChannelc C sF t �d| j � t| jj�dkr:| jjdd � }| �|� d| j_d S )Nz opened forwarding channel %s� � )r �msg�id�lenr �buf�write)r � specificData�br r r � channelOpen' s z&SSHListenForwardingChannel.channelOpenc C s | � � d S r )�closed�r �reasonr r r � openFailed. s z%SSHListenForwardingChannel.openFailedc C s | j j�|� d S r )r � transportr& �r �datar r r � dataReceived1 s z'SSHListenForwardingChannel.dataReceivedc C s | j j�� d S r )r r. �loseConnection�r r r r � eofReceived4 s z&SSHListenForwardingChannel.eofReceivedc C s. t | d�r*t�d| j � | jj�� | `d S )Nr z#closing local forwarding channel %s)�hasattrr r" r# r r. r2 r3 r r r r* 7 s z!SSHListenForwardingChannel.closedN)r r r r) r- r1 r4 r* r r r r r % s r c @ s e Zd ZdZdS )� SSHListenClientForwardingChannels direct-tcpipN�r r r �namer r r r r6 = s r6 c @ s e Zd ZdZdS )� SSHListenServerForwardingChannels forwarded-tcpipNr7 r r r r r9 A s r9 c @ sD e Zd ZdZeZdd� Zdd� Zdd� Zdd � 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} c O s* t jj| f|�|� || _d | _d| _d S )Nr! )r � SSHChannelr r r � clientBuf)r r �args�kwr r r r ` s z$SSHConnectForwardingChannel.__init__c C sR t �d| j � t| j| jd | jd �}t|t| ��}|�| j| j � || _ dS )�, See: L{channel.SSHChannel} zconnecting to %s:%ir r N) r r" r r �_reactorr r Z addCallbacks� _setClient�_closeZ_channelOpenDeferred)r r' Zep�dr r r r) g s �z'SSHConnectForwardingChannel.channelOpenc C sd || _ t�d| j � | jr2| j j�| j� d| _| j jdd� rX| �| j jdd� � d| 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! )r r r" r r< r. r& r% )r r r r r rA r s z&SSHConnectForwardingChannel._setClientc C s t �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} zfailed to connect: %sN)r r" r2 r+ r r r rB � s z"SSHConnectForwardingChannel._closec C s( | j r| j j�|� n| j|7 _dS )r? N)r r. r& r< r/ r r r r1 � s z(SSHConnectForwardingChannel.dataReceivedc C s: | j r6t�d| j � | j jr&| �� | j j�� | ` dS )r? z#closed remote forwarding channel %sN)r r r" r# r r2 r. r3 r r r r* � s z"SSHConnectForwardingChannel.closedN) r r r �__doc__r r@ r r) rA rB r1 r* r r r r r: G s r: c C s t |�\}}t|| ||d�S )N)� remoteWindow�remoteMaxPacket�avatar)�unpackOpen_direct_tcpipr: )rE rF r0 rG ZremoteHPZorigHPr r r �openConnectForwardingClient� s �rI c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )r c C s || _ d| _d S )N� )r r% )r r r r r r � s z SSHForwardingClient.__init__c C s& | j r| j |7 _ n | j�|� d S r )r% r r& r/ r r r r1 � s z SSHForwardingClient.dataReceivedc C s | j r| j �� d | _ d S r )r r2 r+ r r r �connectionLost� s z"SSHForwardingClient.connectionLostN)r r r r r1 rK r r r r r � s r c C sl | \}}|\}}t |t�r$|�d�}t |t�r8|�d�}t�|�t�d|� }t�|�t�d|� }|| 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 �NS�struct�pack)Z destination�source�connHost�connPort�origHost�origPortr Zorigr r r r � s r c C s� t �| �\}}tr&t|t�r&|�d�}tt�d|dd� �d �}t �|dd� �\}}trpt|t�rp|�d�}tt�d|dd� �d �}||f||ffS )z(Unpack the data to a usable format. rL rM N� r � r ZgetNSr rN �bytes�decode�intrQ Zunpack)r0 rT �restrU rV rW r r r rH � s rH c C s | \}}t �|�t�d|� S )zv Pack the data for tcpip forwarding. @param peer: A tuple of the (host, port) . @type peer: L{tuple} rM )r rP rQ rR )Zpeerr r r r r �packGlobal_tcpip_forward� s r^ c C sJ t �| �\}}tr&t|t�r&|�d�}tt�d|d d� �d �}||fS )NrL rM rX r rY )r0 r r] r r r r �unpackGlobal_tcpip_forward� s r_ )#rD Z __future__r r rQ Ztwisted.internetr r Ztwisted.internet.endpointsr r Ztwisted.pythonr Ztwisted.python.compatr r Ztwisted.conch.sshr r ZFactoryr r; r r6 r9 r: rI ZProtocolr r ZpackOpen_forwarded_tcpiprH ZunpackOpen_forwarded_tcpipr^ r_ r r r r �<module> s* _
Save File
Cancel