File: //usr/local/CyberCP/lib/python3.10/site-packages/asyncssh/__pycache__/socks.cpython-310.pyc
o
�h� � @ s� d Z ddlmZ ddlmZmZmZ ddlmZm Z ddl
m Z er(ddl m
Z
eeegdf Zd Zd
ZdZd ZdZdZdZd Zd Zededdddddf�Zeeedf�Zed ed
iZG dd� de �Z dS )zSOCKS forwarding support� )�
ip_address)�
TYPE_CHECKING�Callable�Optional� )�SSHForwarderCoro�SSHLocalForwarder)�DataType)�
SSHConnectionN� � �Z � � c s e Zd ZdZdddef� fdd� Zd'd d
�Zd'd d �Zd'd
d�Zde ddfdd�Z
de ddfdd�Z de ddfdd�Z de ddfdd�Z
de ddfdd�Zde ddfdd�Zde ddfd d �Zde ddfd d �Zde ddfd d!�Zde ddfd"d#�Zd(de d$eddf� fd%d&�
Z� ZS ))�SSHSOCKSForwarderz0SOCKS dynamic port forwarding connection handler�connr
�coroc s8 t � �||� d| _d| _| j| _d| _d| _d| _d S )N� � r � ) �super�__init__�_inpbuf�
_bytes_needed�
_recv_version�
_recv_handler� _addrtype�_host�_port)�selfr r �� __class__� �A/usr/local/CyberCP/lib/python3.10/site-packages/asyncssh/socks.pyr = s
zSSHSOCKSForwarder.__init__�returnNc C sD | j dusJ �d| _| j �d�dd� \}}| �| j| j||� dS )z,Send request to open a new tunnel connectionN�peernamer )�
_transportr �get_extra_info�forwardr r )r � orig_host� orig_portr" r" r# �_connectG s zSSHSOCKSForwarder._connectc C s | j dusJ �| j �t� dS )z Send SOCKS4 success responseN)r& �write�SOCKS4_OK_RESPONSE)r r" r" r# �_send_socks4_okQ s z!SSHSOCKSForwarder._send_socks4_okc C s@ | j dusJ �t| j d }| j �tt| jf� |d � dS )z Send SOCKS5 success responseNr � )r& �_socks5_addr_lenr r, �SOCKS5_OK_RESPONSE_HDR�bytes)r �addrlenr" r" r# �_send_socks5_okX s
�
�z!SSHSOCKSForwarder._send_socks5_ok�datac C sd |d t kr|d tkrd| _| j| _dS | �� dS |d tkr,|d | _| j| _dS | �� dS )zParse SOCKS versionr r � N)�SOCKS4�
SOCKS_CONNECTr �_recv_socks4_addrr �close�SOCKS5�_recv_socks5_authlist�r r5 r" r" r# r c s
z SSHSOCKSForwarder._recv_versionc C sZ |d d> |d | _ |dd� dks|d dkr$tt|dd� ��| _d| _| j| _dS ) z Parse SOCKSv4 address and portr � r r r s N���)r �strr r r �_recv_socks4_userr r= r" r" r# r9 r s
z#SSHSOCKSForwarder._recv_socks4_addrc C s, | j r
| �� | �� dS d| _| j| _dS )zParse SOCKSv4 usernamer? N)r r. r+ r �_recv_socks4_hostnamer r= r" r" r# rA ~ s
z#SSHSOCKSForwarder._recv_socks4_userc C sB z|� d�| _W n ty | �� Y dS w | �� | �� dS )zParse SOCKSv4 hostname�utf-8N)�decoder �UnicodeDecodeErrorr: r. r+ r= r" r" r# rB � s � z'SSHSOCKSForwarder._recv_socks4_hostnamec C sH | j dusJ �t|v r | j �tttf�� d| _| j| _dS | �� dS )z,Parse SOCKSv5 list of authentication methodsNr ) r& �SOCKS5_AUTH_NONEr, r2 r; r �_recv_socks5_commandr r: r= r" r" r# r< � s z'SSHSOCKSForwarder._recv_socks5_authlistc C s� |d t krA|d tkrA|d dkrA|d tkr$d| _| j| _t| _dS t� |d �}|r;|| _| j
| _|d | _dS | � � dS | � � dS )zParse SOCKSv5 commandr r r r N) r; r8 �SOCKS5_ADDR_HOSTNAMEr �_recv_socks5_hostlenr �SOCKS5_ADDR_IPV4r r0 �get�_recv_socks5_addrr: )r r5 r3 r" r" r# rG � s $
z&SSHSOCKSForwarder._recv_socks5_commandc C s t t|��| _d| _| j| _dS )zParse SOCKSv5 addressr N)r@ r r r �_recv_socks5_portr r= r" r" r# rL � s z#SSHSOCKSForwarder._recv_socks5_addrc C s |d | _ | j| _dS )zParse SOCKSv5 host lengthr N)r �_recv_socks5_hostr r= r" r" r# rI � s
z&SSHSOCKSForwarder._recv_socks5_hostlenc C s@ z|� d�| _W n ty | �� Y dS w d| _| j| _dS )zParse SOCKSv5 hostrC Nr )rD r rE r: r rM r r= r" r" r# rN � s � z#SSHSOCKSForwarder._recv_socks5_hostc C s* |d d> |d | _ | �� | �� dS )zParse SOCKSv5 portr r> r N)r r4 r+ r= r" r" r# rM � s z#SSHSOCKSForwarder._recv_socks5_port�datatypec s� | j rl| j|7 _| j rf| jdk rB| j�d�}|dkr3| jd|� }| j|d d� | _| � |� n0t| j�dkr@| �� dS dS t| j�| jkra| jd| j� }| j| jd� | _| � |� ndS | j s
| j}d| _|rwt� �||� dS dS )z*Handle incoming data from the SOCKS clientr r/ Nr � r )r r r �find�lenr: r �
data_received)r r5 rO �idxr r" r# rS � s0
��z SSHSOCKSForwarder.data_received)r$ N)N)�__name__�
__module__� __qualname__�__doc__r r r+ r. r4 r2 r r9 rA rB r< rG rL rI rN rM r rS �
__classcell__r" r" r r# r : s"
$r ) rX � ipaddressr �typingr r r r( r r �sessionr �
connectionr
r2 � _RecvHandlerr7 r; r8 � SOCKS4_OK� SOCKS5_OKrF rJ rH �SOCKS5_ADDR_IPV6r- r1 r0 r r" r" r" r# �<module> s*