File: //lib/python3/dist-packages/twisted/protocols/__pycache__/portforward.cpython-310.pyc
o
�bC � @ sv d Z ddlmZ ddlmZ G dd� dej�ZG dd� de�ZG dd � d ej�Z G d
d � d e�Z
G d d
� d
ej �Z dS )z
A simple port forwarder.
� )�protocol)�logc @ s, e Zd ZdZdZdd� Zdd� Zdd� ZdS ) �ProxyTNc C �
|| _ d S �N)�peer)�selfr � r �?/usr/lib/python3/dist-packages/twisted/protocols/portforward.py�setPeer �
z
Proxy.setPeerc C s> | j d ur| j j�� d | _ d S | jr t�d|� �� d S d S )NzUnable to connect to peer: )r � transport�loseConnection�noisyr �msg)r �reasonr r r
�connectionLost s
�zProxy.connectionLostc C s | j j�|� d S r )r r
�write)r �datar r r
� dataReceived s zProxy.dataReceived)�__name__�
__module__� __qualname__r r r r r r r r r
r
s r c @ s e Zd Zdd� ZdS )� ProxyClientc C s@ | j �| � | j�| j jd� | j j�| jd� | j j�� d S )NT)r r r
�registerProducer�resumeProducing)r r r r
�connectionMade! s zProxyClient.connectionMadeN)r r r r r r r r
r s r c @ s( e Zd ZeZdd� Zdd� Zdd� ZdS )�ProxyClientFactoryc C r r )�server)r r r r r
� setServer3 r z ProxyClientFactory.setServerc O s* t jj| g|�R i |��}|�| j� |S r )r �
ClientFactory�
buildProtocolr r )r �args�kw�protr r r
r! 6 s z ProxyClientFactory.buildProtocolc C s | j j�� d S r )r r
r )r � connectorr r r r
�clientConnectionFailed; s z)ProxyClientFactory.clientConnectionFailedN)r r r r r r r! r& r r r r
r / s
r c @ s e Zd ZeZdZdd� ZdS )� ProxyServerNc C sT | j �� | �� }|�| � | jd u r ddlm} || _| j�| jj| jj |� d S )Nr )�reactor)
r
�pauseProducing�clientProtocolFactoryr r( �twisted.internet�
connectTCP�factory�host�port)r �clientr( r r r
r D s
zProxyServer.connectionMade)r r r r r* r( r r r r r
r' ? s r' c @ s e Zd ZdZeZdd� ZdS )� ProxyFactoryz%
Factory for port forwarder.
c C s || _ || _d S r )r. r/ )r r. r/ r r r
�__init__Z s
zProxyFactory.__init__N)r r r �__doc__r' r r2 r r r r
r1 S s r1 N)
r3 r+ r �twisted.pythonr �Protocolr r r r r' �Factoryr1 r r r r
�<module> s