Current File : //usr/lib/python3.9/site-packages/cockpit/__pycache__/remote.cpython-39.pyc
a
��h�# � @ 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� )�
JsonObject� JsonValue�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
d
d
e 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||
|i d�|�W Y d }~n�d }~0 tj�yP } ztt�d||� t�|jd�}d|v �r$| j
d u�r$|j d
k�r d|d< nd|d< td|j �p2d|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: %sz not-providedr z not-triedZdeniedzauthentication-failedz)unknown failure connecting to host %s: %szinternal-error)rA zcockpit-bridge)!r7 r% r& r8 r r9 �
rpartition�isdigitr4 r r �updateZconnect�OSError�socketZgaierrorr
r2 r ZSshHostKeyErrorr �
isinstanceZSshChangedHostKeyError�typeZSshAuthenticationError�dict�fromkeys�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_transportz
str | None)r r) rA r c s0 || j kr| �� n|d u r,t� �|||� d S �N)r �close�super�do_kill)r r r) rA �� __class__r r rV � s
zSshPeer.do_kill)rA r c C s: t |d��d�r6t |d�}| jd|| jp*dd� d | _d S )NZ challengezplain1:�cookieZ authorize� )ZcommandrY Zresponse)r
�
startswithZ
write_controlr )r rA rY r r r � do_authorize� s
zSshPeer.do_authorizec C s
d | _ d S rS 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) rU r r r8 r
r r9 r �Sessionr7 r Zstart_in_background)r r^ r r8 r? r9 Zinit_superuserra rW r r r � s
zSshPeer.__init__)r- r. r/ r7 r r rb r3 r2 r5 rR r rV r\ r] r r �
__classcell__r r rW r r6 H s
Gr6 c sb e Zd ZU eeeee ee f ef ed< � fdd�Z e
ee d�dd�Z dd�d d
�Z � Z
S ) �HostRoutingRule�remotesc s t � �|� i | _d S rS )rU r re )r r^ rW 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 rS )re � __delitem__r �r r r r �<lambda>� � z,HostRoutingRule.apply_rule.<locals>.<lambda>)
r^ r` r
�getpassZgetuserrC rH r2 r% r& re r6 Zadd_done_callback)r r? r r8 Zuser_from_host�_Znonce�peerr rh r �
apply_rule� s4
zHostRoutingRule.apply_ruleNr: c C s t | j�� �D ] }|�� qd S rS )�setre �valuesrT )r rm r r r �shutdown� s zHostRoutingRule.shutdown)r- r. r/ r r r2 r r r3 r r rn rq rc r r rW r rd � s
" (rd ) rk Zloggingr0 rG �typingr r r r Zcockpit._vendorr Zjsonutilr r r
r rm r r
r^ r r Z getLoggerr- r% ZAskpassHandlerr r6 rd r r r r �<module> s
'p
Mini Shell