File: //usr/local/CyberPanel/lib64/python3.10/site-packages/asyncssh/__pycache__/x11.cpython-310.pyc
o
�hmB �
@ sd d Z ddlZddlZddlmZ ddlZddlZddlmZm Z m
Z
m Z ddlm Z m
Z
mZmZmZ ddlmZ ddlmZmZ dd lmZmZ dd
lmZ dd lm Z dd l m Z erhdd
l m Z ddl!m"Z" e
e e#gdf Z$dZ%dZ&dZ'dZ(dZ)dZ*dZ+dZ,dZ-dZ.dZ/dZ0dZ1dZ2dZ3de4d ee4e4e5f fd d �Z6d ej7d e4d!e5d ee4 fd"d#�Z8G d$d%� d%e �Z9G d&d'� d'e�Z:G d(d)� d)�Z;G d*d+� d+�Z<d,e
e4 d e4fd-d.�Z=d,e4d e e9 fd/d0�Z>d ej7d,e
e4 d e4d1e4d ee#e#f f
d2d3�Z?d,e
e4 d e4d1e4d4e#d5e#d df d6d7�Z@d ej7de4d,e
e4 d e;fd8d9�ZAd:d;d ej7d,e
e4 d4e#d5e#d e
e< f d<d=�ZBdS )>zX11 forwarding support� N)�Path)�
TYPE_CHECKING�Callable�Dict�Iterable)�
NamedTuple�Optional�Sequence�Set�Tuple� )�OPEN_CONNECT_FAILED)� SSHForwarder�SSHForwarderCoro)� SSHListener�create_tcp_forward_listener)�logger)�ChannelOpenError)�DataType)�
SSHChannel)�SSHServerConnection� � i�� s MIT-MAGIC-COOKIE-1� z-c� g�������?ip �
�@ � localhost�display�returnc C s� z6| � dd�\}}|�d�r|�d�r|dd� }|�d�}|dkr3t||d d� �}|d|� }nd}W n ttfyD td �d�w |||fS )
zParse an X11 display value�:r �[�]����.r NzInvalid X11 display)�rsplit�
startswith�endswith�find�int�
ValueError�UnicodeEncodeError)r �host�dpynum�idx�screen� r0 �B/usr/local/CyberPanel/lib/python3.10/site-packages/asyncssh/x11.py�_parse_displayJ s
�
�
r2 �loopr, �familyc � sH �z| j |d|tjd�I dH }W n tjy g Y S w dd� |D �S )z-Look up IPv4 or IPv6 addresses of a host namer )r4 �typeNc S s g | ]}|d d �qS )� r r0 )�.0�air0 r0 r1 �
<listcomp>h s z _lookup_host.<locals>.<listcomp>)� getaddrinfo�socket� SOCK_STREAM�gaierror)r3 r, r4 �addrinfor0 r0 r1 � _lookup_host^ s �
��r? c @ sH e Zd ZU dZeed< eed< eed< eed< eed< defdd �Zd
S ) �SSHXAuthorityEntryz An entry in an Xauthority filer4 �addrr- �proto�datar c sZ dt dtfdd�� dtdtf� fdd� }d�� | j�|| j�|| j�|| j�|| j�f�S ) z Construct an Xauthority entry�valuer c S s | � dd�S )z.Construct a big-endian 16-bit unsigned integer� �big)�to_bytes)rD r0 r0 r1 �_uint16w s z-SSHXAuthorityEntry.__bytes__.<locals>._uint16rC c s � t | ��| S )z.Construct a binary string with a 16-bit length��len)rC �rH r0 r1 �_string| � z-SSHXAuthorityEntry.__bytes__.<locals>._string� )r) �bytes�joinr4 rA r- rB rC )�selfrL r0 rK r1 � __bytes__t s �z SSHXAuthorityEntry.__bytes__N)�__name__�
__module__� __qualname__�__doc__r) �__annotations__rO rR r0 r0 r0 r1 r@ k s
r@ c s� e Zd ZdZdddef� fdd� Zdedefd d
�Zdedefd d �Z e
d
edefdd��Z e
dedefdd��Z deddfdd�Z
deddfdd�Zdeddfdd�Zd dededdf� fdd �
Z� ZS ) �SSHX11ClientForwarderz!X11 forwarding connection handler�listener�SSHX11ClientListener�peerc sZ t � �|� || _d| _d| _| j| _d| _d| _d| _ d| _
d| _ d| _ d| _
d| _d S )NrN � r )�super�__init__� _listener�_inpbuf�
_bytes_needed� _recv_prefix�
_recv_handler�_endian�_prefix�_auth_proto_len�_auth_data_len� _auth_proto�_auth_proto_pad�
_auth_data�_auth_data_pad)rQ rY r[ �� __class__r0 r1 r^ � s
z SSHX11ClientForwarder.__init__rD r c C s2 | j dkrt|d? |d@ f�S t|d@ |d? f�S )z Encode a 16-bit unsigned integer� B� � )rd rO �rQ rD r0 r0 r1 �_encode_uint16� �
z$SSHX11ClientForwarder._encode_uint16c C s2 | j dkr|d d> |d S |d d> |d S )z Decode a 16-bit unsigned integerrn r ro r )rd rq r0 r0 r1 �_decode_uint16� rs z$SSHX11ClientForwarder._decode_uint16�lengthc C s | d d d S )z8Return length rounded up to the next multiple of 4 bytes� r6 r0 )ru r0 r0 r1 � _padded_len� s z!SSHX11ClientForwarder._padded_lenrC c C s& t | �d }| |rd| d S d S )z%Pad a string to a multiple of 4 bytesr6 � rN rI )rC ru r0 r0 r1 �_pad� s zSSHX11ClientForwarder._padNc C sV |dd� | _ || _| �|dd� �| _| �|dd� �| _| j| _| �| j�| _dS )zParse X11 client prefixNr r ro r ) rd re rt rf rg �_recv_auth_protorc rw ra �rQ rC r0 r0 r1 rb � s z"SSHX11ClientForwarder._recv_prefixc C s: |d| j � | _|| j d� | _| j| _| �| j�| _dS )zExtract X11 auth protocolN)rf rh ri �_recv_auth_datarc rw rg ra r{ r0 r0 r1 rz � s z&SSHX11ClientForwarder._recv_auth_protoc C s� |d| j � | _|| j d� | _z
| j�| j�| _W nE ty_ d}d�tdt|�f�| � d�| � d�| � t|�d d �| �
|�f�}z | � |� | � � W n t
yY Y nw d| _Y nw | j| j | j | j | j | _d| _d| _dS )zExtract X11 auth dataNs Invalid authentication key
rN r � rv r6 )rg rj rk r_ �
validate_auth�KeyErrorrP rO rJ rr ry �write� write_eof�OSErrorr` re rh ri rc ra )rQ rC �reason�responser0 r0 r1 r| � s: �
�
�
���
z%SSHX11ClientForwarder._recv_auth_data�datatypec s� | j r7| j|7 _| j r1t| j�| jkr,| jd| j� }| j| jd� | _| � |� ndS | j s
| j}d| _|rBt� �||� dS dS )z(Handle incoming data from the X11 clientNrN )rc r` rJ ra r] �
data_received)rQ rC r� rl r0 r1 r� � s ��z#SSHX11ClientForwarder.data_received�N)rS rT rU rV r r^ r) rO rr rt � staticmethodrw ry rb rz r| r r� �
__classcell__r0 r0 rl r1 rX � s $ rX c @ s� e Zd ZdZdejdedededef
dd�Zd ed
d d e d
e
eee f fdd�Z d
d d
e fdd�Z
d
efdd�Zded
efdd�ZdS )rZ z8Client listener used to accept forwarded X11 connectionsr3 r, r- �
auth_proto� auth_datac C s� || _ || _|| _|| _|�d�r |j| _|d | f| _n |dv r-|j| _d| f| _n
|j| _|t t
|� f| _i | _ i | _ d S )N�/r )� �unixz/tmp/.X11-unix/X)
�_host�_dpynumrh � _local_authr& �create_unix_connection�
_connect_coro�
_connect_args�create_connection�
X11_BASE_PORTr) � _remote_auth�_channel)rQ r3 r, r- r� r� r0 r0 r1 r^ s
z SSHX11ClientListener.__init__r �chanr �single_connectionr c C s^ t |�\}}}| j|ks| j|krtd��t�t| j��}|| j|< ||f| j |< | j
||fS )z!Attach a channel to this listenerz)Already forwarding to another X11 display) r2 r� r� r* �os�urandomrJ r� r� r� rh )rQ r r� r� r, r- r/ � remote_authr0 r0 r1 �attach s
zSSHX11ClientListener.attachc C s8 z | j �|�}| j|= W n ty Y nw t| j � S )�#Detach a channel from this listener)r� �popr� r �bool)rQ r� r� r0 r0 r1 �detach+ s � zSSHX11ClientListener.detachc
� sX �z| j tg| j�R � I dH \}}W n ty& } zttt|��d�d}~ww t| |�S )z4Forward an incoming connection to the local X serverN)r� r r� r� r r
�strrX )rQ �_r[ �excr0 r0 r1 �forward_connection6 s ����
z'SSHX11ClientListener.forward_connectionr� c C s( | j | \}}|r| j |= | j|= | jS )z7Validate client auth and enforce single connection flag)r� r� r� )rQ r� r� r� r0 r0 r1 r~ C s
z"SSHX11ClientListener.validate_authN)rS rT rU rV �asyncio�AbstractEventLoopr� rO r^ r� r r) r� r� rX r� r~ r0 r0 r0 r1 rZ s �
�
�
�
rZ c @ sJ e Zd ZdZdedefdd�Zddded efd
d �Zddd e fd d
�Z
dS )�SSHX11ServerListenerz/Server listener used to forward X11 connections� tcp_listenerr c C s || _ || _t� | _d S r� )�
_tcp_listener�_display�set� _channels)rQ r� r r0 r0 r1 r^ R s z SSHX11ServerListener.__init__r� r r/ r c C s | j �|� | j� d|� �S )z8Attach a channel to this listener and return its displayr$ )r� �addr� )rQ r� r/ r0 r0 r1 r� W s zSSHX11ServerListener.attachc C s< z| j �|� W n ty Y nw | j s | j�� dS dS )r� TF)r� �remover r� �close)rQ r� r0 r0 r1 r� ^ s �
zSSHX11ServerListener.detachN) rS rT rU rV r r� r^ r) r� r� r� r0 r0 r0 r1 r� O s
r� � auth_pathc C s* | st j�d�} | sttdd��� �} | S )z'Compute the path to the Xauthority file�
XAUTHORITY�~z .Xauthority)r� �environ�getr� r �
expanduser)r� r0 r0 r1 �get_xauth_pathm s
r� c # s� � dt dtf�fdd� � dt f� fdd� �dtf� �fdd� }zGt| d ��7� z�� }W n ty5 Y n w zt||� |� |� |� �V W n tyQ td �d �w q'W d � W d S 1 s_w Y W d S typ Y d S w )
z&Walk the entries in an Xauthority file�nr c s � � | �}t|�| kr
t�|S )zRead exactly n bytes)�readrJ �EOFError)r� rC )� auth_filer0 r1 � _read_bytes| s
z walk_xauth.<locals>._read_bytesc s t �� d�d�S )z Read a 16-bit unsigned integerrE rF )r) �
from_bytesr0 )r� r0 r1 � _read_uint16� rM z walk_xauth.<locals>._read_uint16c s
� �� �S )z
Read a stringr0 r0 )r� r� r0 r1 � _read_string� s
z walk_xauth.<locals>._read_string�rbTzIncomplete Xauthority entryN)r) rO �openr� r@ r* r� )r� r� r4 r0 )r� r� r� r1 �
walk_xauthy s4 �
�
�
��&�
�r� r- c � s* �t |�}|�d�s|dv rt�� }|�d�}g }g }t|�D ]i}|jr*|j|kr*q |jtkrH|s;t | |tj
�I dH }t� tj
|j �}||v }n6|jt
krf|sYt | |tj�I dH }t� tj|j �}||v }n|jtkrt|j |�d�k}n
|jtkr|d}nd}|r�|j|jf S q t�d� tt�t�fS ) z1Look up Xauthority data for the specified displayr� �r� r� r �asciiN�idnaTFz3No xauth entry found for display: using random auth)r� r& r; � gethostname�encoder� r- r4 �XAUTH_FAMILY_IPV4r? �AF_INET� inet_ntoprA �XAUTH_FAMILY_IPV6�AF_INET6�XAUTH_FAMILY_HOSTNAME�XAUTH_FAMILY_WILDrB rC r �debug1�XAUTH_PROTO_COOKIEr� r� �XAUTH_COOKIE_LEN) r3 r� r, r- �
ipv4_addrs�
ipv6_addrs�entryrA �matchr0 r0 r1 � lookup_xauth� s< �
�
r� r� r� c
� sD �|� d�s
|dv rt�� }|�d�}t|��d�}t| �} | t }d}zt�� t� |�j
t kr6t� |� W n t
y@ Y nw tt�D ]}zt|d�}W n ty_ t�t�I dH Y qEw |sgtd��tt||||�}|�t|�� t| �D ]} | j|jks�| j |j ks�| j |j kr�|�t| �� qz|� � t� || � dS )z0Update Xauthority data for the specified displayr� r� r� r� N�xbz!Unable to acquire Xauthority lock) r& r; r� r� r� r� �XAUTH_LOCK_SUFFIX�timer� �stat�st_ctime�XAUTH_LOCK_DEAD�unlink�FileNotFoundError�range�XAUTH_LOCK_TRIESr� �FileExistsErrorr� �sleep�XAUTH_LOCK_DELAYr* r@ r� r� rO r� r4 rA r- r� �replace)
r� r, r- r� r� �
new_auth_path�new_filer� � new_entryr� r0 r0 r1 � update_xauth� sF �
� � �� �r� c � s8 �t |�\}}}t| |||�I dH \}}t| ||||�S )z>Create a listener to accept X11 connections forwarded over SSHN)r2 r� rZ )r3 r r� r, r- r� r� r� r0 r0 r1 �create_x11_client_listener� s �r� �connr c � s� �t tt�D ]H}zt| || jtt| �I dH }W n ty! Y qw t� d|� �}zt|tt |�||�I dH W n t
yG |� � Y dS w t ||� S dS )z5Create a listener to forward X11 connections over SSHNr )
r� �X11_DISPLAY_START�X11_MAX_DISPLAYSr �create_x11_connection�X11_LISTEN_HOSTr� r� r� r� r* r� r� )r� r3 r� r� r� r- r� r r0 r0 r1 �create_x11_server_listener
s, �� � � �r� )CrV r� r� �pathlibr r; r� �typingr r r r r r r r
r � constantsr
�forwardr r rY r r �loggingr �miscr �sessionr �channelr �
connectionr rO � _RecvHandlerr� �XAUTH_FAMILY_DECNETr� r� r� r� r� r� r� r� r� r� r� r� r� r� r) r2 r� r? r@ rX rZ r� r� r� r� r� r� r� r0 r0 r0 r1 �<module> s� �
�
~K )���
�-��
�.��
�
�����