File: //usr/local/CyberPanel/lib64/python3.10/site-packages/asyncssh/__pycache__/forward.cpython-310.pyc
o
�h� � @ s� d Z ddlZddlZddlmZ ddlmZmZmZm Z m
Z
m Z ddlm Z m
Z
ddlmZ ddlmZmZ er>dd lmZ e d
ef ZG d d � d ej�ZG d
d� de�ZG dd� de�ZG dd� de�ZdS )z SSH port forwarding handlers� N)�
TracebackType)�
TYPE_CHECKING�Any� Awaitable�Callable�Dict�Optional)�Type�cast)�Self� )�ChannelOpenError�SockAddr)�
SSHConnection.c @ sN e Zd ZdZ d3ded deeeef fdd�Zde fdd �Z
d
ee e d ee d ee
defd
d�Zd4dededefdd�Zd5dd�Zdeddfdd�Zd6dd�Zdefdd�Zd6d d �Zd6d d �Zd ejddfd!d"�Zd#ee ddfd$d%�Z d6d&d'�Z d4ded(ee ddfd)d*�Z defd+d,�Z d6d-d.�Z!d6d/d0�Z"d6d1d2�Z#dS )7� SSHForwarderz&SSH port forwarding connection handlerN�peer�extrac C s< || _ d | _d| _d| _|r|�| � |d u ri }|| _d S )N� F)�_peer�
_transport�_inpbuf�
_eof_received�set_peer�_extra)�selfr r � r �F/usr/local/CyberPanel/lib/python3.10/site-packages/asyncssh/forward.py�__init__, s
zSSHForwarder.__init__�returnc � s �| S �Nr �r r r r �
__aenter__; s �zSSHForwarder.__aenter__� _exc_type�
_exc_value�
_tracebackc � s �| � � dS )NF��close)r r" r# r$ r r r � __aexit__> s �zSSHForwarder.__aexit__�name�defaultc C s | j �||�S )aS Get additional information about the forwarder
This method returns extra information about the forwarder.
Currently, the only information available is the value
``interface`` for TUN/TAP forwarders, returning the name of the
local TUN/TAP network interface created for this forwarder.
)r �get)r r( r) r r r �get_extra_infoD s
zSSHForwarder.get_extra_infoc C s
|| _ dS )z,Set the peer forwarder to exchange data withN)r �r r r r r r P s
zSSHForwarder.set_peer�datac C s | j dusJ �| j �|� dS )zWrite data to the transportN)r �write)r r- r r r r. U s zSSHForwarder.writec C s4 | j dusJ �z| j �� W dS ty Y dS w )z"Write end of file to the transportN)r � write_eof�OSErrorr r r r r/ [ s �zSSHForwarder.write_eofc C s | j S )z3Return whether end of file has been received or not)r r r r r �was_eof_receivede s z SSHForwarder.was_eof_receivedc C � | j dusJ �| j �� dS )z Pause reading from the transportN)r �
pause_readingr r r r r3 j � zSSHForwarder.pause_readingc C r2 )z Resume reading on the transportN)r �resume_readingr r r r r5 p r4 zSSHForwarder.resume_reading� transportc C sV t td |�| _t tj|�d��}|r'|jtjtjhv r)|�tj tj
d� dS dS dS )� Handle a newly opened connectionzasyncio.Transport�socketr N) r
r r r8 r+ �family�AF_INET�AF_INET6�
setsockopt� IPPROTO_TCP� TCP_NODELAY)r r6 �sockr r r �connection_madev s
�z SSHForwarder.connection_made�excc C s | � � dS )z#Handle an incoming connection closeNr% )r rA r r r �connection_lost� s z SSHForwarder.connection_lostc C s dS )zHandle session startNr r r r r �session_started� s z SSHForwarder.session_started�datatypec C s( | j r | j �|� dS | j|7 _dS )z'Handle incoming data from the transportN)r r. r )r r- rD r r r �
data_received� s zSSHForwarder.data_receivedc C s& d| _ | jr| j�� | j�� S dS )z1Handle an incoming end of file from the transportT)r r r/ r1 r r r r � eof_received� s
zSSHForwarder.eof_receivedc C � | j r
| j �� dS dS )z-Pause writing by asking peer to pause readingN)r r3 r r r r �
pause_writing� � �zSSHForwarder.pause_writingc C rG )z/Resume writing by asking peer to resume readingN)r r5 r r r r �resume_writing� rI zSSHForwarder.resume_writingc C s8 | j r | j �� d| _ | jr| j}d| _|�� dS dS )zClose this port forwarderN)r r& r r, r r r r& � s
�zSSHForwarder.close)NNr )r r r N)r N)$�__name__�
__module__� __qualname__�__doc__r r �strr r r r! r �
BaseExceptionr �boolr' r+ r �bytesr. r/ r1 r3 r5 �asyncio�
BaseTransportr@ � ExceptionrB rC �intrE rF rH rJ r& r r r r r ) sF �
���
�
��
�
r c sN e Zd ZdZdddef� fdd� Zdedd fd
d �Zdedd fd d
�Z� Z S )�SSHLocalForwarderz#Local forwarding connection handler�connr �coroc s t � �� || _|| _d S r )�superr �_conn�_coro)r rX rY �� __class__r r r � s
zSSHLocalForwarder.__init__�argsr Nc
� s� �dt f� fdd� }z� j|g|�R � I dH W n ty0 } z � �|� W Y d}~dS d}~ww � jdus8J �� jrE� j�� j� d� _� jrO� j�� dS dS )zBegin local forwardingr c s t � �S )zReturn an SSH forwarder)r r r r r �session_factory� s z3SSHLocalForwarder._forward.<locals>.session_factoryNr ) r r\ r
rB r r r. r r/ )r r_ r` rA r r r �_forward� s �
���zSSHLocalForwarder._forwardc G s | j �| j|� � dS )z&Start a task to begin local forwardingN)r[ � create_taskra )r r_ r r r �forward� s zSSHLocalForwarder.forward)
rK rL rM rN �SSHForwarderCoror �objectra rc �
__classcell__r r r] r rW � s
rW c �, e Zd ZdZdejddf� fdd� Z� ZS )�SSHLocalPortForwarderz,Local TCP port forwarding connection handlerr6 r Nc s@ t � �|� tt|�d��}|r|dd� \}}| �||� dS )r7 �peernameN� )rZ r@ r
r r+ rc )r r6 ri � orig_host� orig_portr] r r r@ � s
z%SSHLocalPortForwarder.connection_made�rK rL rM rN rS rT r@ rf r r r] r rh � � rh c rg )�SSHLocalPathForwarderz6Local UNIX domain socket forwarding connection handlerr6 r Nc s t � �|� | �� dS )r7 N)rZ r@ rc )r r6 r] r r r@ � s z%SSHLocalPathForwarder.connection_maderm r r r] r ro � rn ro )rN rS r8 �typesr �typingr r r r r r r r
�typing_extensionsr �miscr
r �
connectionr rd � BaseProtocolr rW rh ro r r r r �<module> s %