File: //usr/lib/python3.9/site-packages/cockpit/__pycache__/remote.cpython-39.pyc
a
���e�# � @ s� d dl Z d dlZd dlZd dlZd dlmZmZmZmZ d dl m
Z
ddl m Z m
Z
mZmZ ddlmZmZ ddlmZmZ e�e�ZG dd � d e
j�ZG d
d � d e�ZG d d
� d
e�Z dS )� N)�Dict�List�Optional�Tuple)�ferny� )� JsonDocument�
JsonObject�get_str�get_str_or_none)�Peer� PeerError)�Router� RoutingRulec @ s� e Zd ZU e�d�Zee ed< e e
eeeeef ed< ee ed< e ed< ee d�dd�Z eeeeee
d �d
d �Zeeeee d �d
d�ZdS )�PasswordResponderz!Enter passphrase for key '(.*)': �password�
hostkeys_seen�
error_message�password_attempts�r c C s || _ g | _d | _d| _d S )Nr )r r r r )�selfr � r �2/usr/lib/python3.9/site-packages/cockpit/remote.py�__init__* s zPasswordResponder.__init__)�reason�host� algorithm�key� fingerprint�returnc � s | j �|||||f� dS )NF)r �append)r r r r r r r r r �
do_hostkey1 s z PasswordResponder.do_hostkey)�messages�prompt�hintr c � sx t �d||� tj�|�}|d urH|�d�}t �d|� d|� �| _d S | jd usVJ �| jdksdJ �| jd7 _| jS )NzGot askpass(%s): %sr zCThis is a passphrase request for %s, but we don't do those. Abort.zlocked identity: r ) �logger�debugr �
PASSPHRASE_RE� fullmatch�groupr r r )r r" r# r$ �match�pathr r r �
do_askpass5 s
z PasswordResponder.do_askpassN)�__name__�
__module__� __qualname__�re�compiler' r �str�__annotations__r r �intr �boolr! r, r r r r r ! s
r c s� e Zd ZU dZeej ed< eed< ee ed< ee ed< e ed< dd�dd �Z
ee ee dd
�� fd d � Z e dd
�dd�Z
dd�dd�Zeeee e e dd�� fdd� Z� ZS )�SshPeerN�sessionr �userr �private)r c � s� | j d usJ �t�d| j| j| j� | j�d�\}}}|rP|�� rP|}t|�}n
| j}d }t | j
�}ddi}| j
d ur�|j dd� n|j ddd� z(| j j || j|| j||d �I d H W �n� t
tjf�y
} z0t�d
||� td d t|�d �|�W Y d }~�n�d }~0 tj�y� } z�|j�rT|jd
\} }}
} } |� d|
� d| � �| d�}
ni }
t|tj��rld}n| j�rzd}nd}t�dt|�|| j|j||
� t|f|i d�|
��|�W Y d }~n�d }~0 tj�yX } zvt�d||� dd� |jD �}d|v �r,| j
d u�r,|jd
k�r$d|d< nd|d< td|j�p:d|d�|�W Y d }~nPd }~0 tj �y� } z,t�d ||� td t|�d �|�W Y d }~n
d }~0 0 | j � d g�}| � |g �I d H d S ) Nz1Starting ssh session user=%s, host=%s, private=%s�:ZStrictHostKeyCheckingZyes�1)ZNumberOfPasswordPrompts�no)ZPasswordAuthenticationZ KbdInteractiveAuthentication)Z
login_name�portZhandle_host_key�optionsZinteraction_responderz connecting to host %s failed: %szno-host)�error�messager � )Zhost_keyZhost_fingerprintzinvalid-hostkeyzunknown-hostkeyz unknown-hostzPSshPeer got a %s %s; private %s, seen hostkeys %r; raising %s with extra args %r)r? Zauth_method_resultsz$authentication to host %s failed: %sc S s i | ]
}|d �qS )z not-providedr )�.0�methodr r r �
<dictcomp>� � z0SshPeer.do_connect_transport.<locals>.<dictcomp>r z not-triedZdeniedzauthentication-failedz)unknown failure connecting to host %s: %szinternal-error)r@ zcockpit-bridge) r7 r% r& r8 r r9 �
rpartition�isdigitr4 r r �updateZconnect�OSError�socketZgaierrorr
r2 r Z HostKeyErrorr �
isinstanceZChangedHostKeyError�typeZAuthenticationError�methodsr r ZSshErrorZwrap_subprocess_argsZspawn)r �basename�colonZportstrr r= Z responderr> �excZ_reasonr r r Z
error_argsr? �results�argsr r r �do_connect_transportO sj
�* �,
��(z SshPeer.do_connect_transport)r r) r c s. || j kr| �� n|d u r*t� �d |� d S �N)r �close�super�do_kill)r r r) �� __class__r r rW � s
zSshPeer.do_kill)r@ r c C s: t |d��d�r6t |d�}| jd|| jp*dd� d | _d S )NZ challengezplain1:�cookieZ authorize� )ZcommandrZ Zresponse)r
�
startswithZ
write_controlr )r r@ rZ r r r � do_authorize� s
zSshPeer.do_authorizec C s
d | _ d S rT r )r r r r �do_superuser_init_done� s z SshPeer.do_superuser_init_done)�routerr r8 r> r9 r c sj t � �|� || _|| _t|dd �| _|| _t�� | _ t
|dd �}|dv rPd}nd|i}| j ||d� d S )Nr zinit-superuser)NZnoneF�id)� init_host� superuser) rV r r r8 r
r r9 r �Sessionr7 r Zstart_in_background)r r_ r r8 r> r9 Zinit_superuserrb rX r r r � s
zSshPeer.__init__)r- r. r/ r7 r r rc r3 r2 r5 rS rW r r] r^ r r �
__classcell__r r rX r r6 H s
G r6 c s\ e Zd ZU eeeee ee f ef ed< � fdd�Z e
ee d�dd�Z dd� Z � Z
S ) �HostRoutingRule�remotesc s t � �|� i | _d S rT )rV r rf )r r_ rX r r r � s zHostRoutingRule.__init__)r> r c sN �j d usJ ��j jd us J �t|d�j j�}|�j jkr>d S t|dd �}|t�� krZd }|sv|�d�\}}}|ptd }t|dd �dkr�t|d�}nd }t|t�s�J �|d u s�t|t�s�J �|d u s�t|t�s�J �|||f� t� d|� t� d� � � �j
v�rDt� d � �j
� t �j ||||d ud
�}|� � �fd d �� |�j
� < �j
� S )
Nr r8 �@r7 r9 Zchannelz!Request for channel %s is remote.zkey=%szC%s is not among the existing remotes %s. Opening a new connection.)r9 c s �j �� �S rT )rf � __delitem__r �r r r r �<lambda>� rE z,HostRoutingRule.apply_rule.<locals>.<lambda>)
r_ ra r
�getpassZgetuserrF rK r2 r% r& rf r6 Zadd_done_callback)r r> r r8 Zuser_from_host�_Znonce�peerr ri r �
apply_rule� s4
zHostRoutingRule.apply_rulec C s t | j�� �D ] }|�� qd S rT )�setrf �valuesrU )r rm r r r �shutdown� s zHostRoutingRule.shutdown)r- r. r/ r r r2 r r r3 r r rn rq rd r r rX r re � s
" (re ) rk Zloggingr0 rJ �typingr r r r Zcockpit._vendorr Zjsonutilr r r
r rm r r
r_ r r Z getLoggerr- r% ZInteractionResponderr r6 re r r r r �<module> s
'p