Current File : //usr/lib/python3.9/site-packages/cockpit/_vendor/ferny/__pycache__/session.cpython-39.pyc
a
��h� � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl m
Z
m Z ddl m
Z
ddlmZmZmZmZ e�d�jZe�e�ZdZe�� deeed�dd ��Z G d
d � d �Z G d d
� d
e e�Z dS )� N)�Mapping�Sequence� )�
ssh_errors)�InteractionAgent�InteractionError�InteractionHandler�write_askpass_to_tmpdir�x)�feature�teststr�returnc C sF z*t jdd| � d|� �ddgt jd� W dS t jy@ Y dS 0 d S ) N�ssh�-o� z-GZ nonexisting)�stderrTF)�
subprocessZ check_output�DEVNULLZCalledProcessError)r r � r �A/usr/lib/python3.9/site-packages/cockpit/_vendor/ferny/session.py� has_feature% s
$r c @ sD e Zd Zee ee d�dd�Zeeef eeef d�dd�ZdS )�SubprocessContext��argsr
c C s |S )a� Return the args required to launch a process in the given context.
For example, this might return a vector with
["sudo"]
or
["flatpak-spawn", "--host"]
prepended.
It is also possible that more substantial changes may be performed.
This function is not permitted to modify its argument, although it may
(optionally) return it unmodified, if no changes are required.
r ��selfr r r r �wrap_subprocess_args/ s z&SubprocessContext.wrap_subprocess_args)�envr
c C s |S )aj Return the envp required to launch a process in the given context.
For example, this might set the "SUDO_ASKPASS" environment variable, if
needed.
As with wrap_subprocess_args(), this function is not permitted to
modify its argument, although it may (optionally) return it unmodified
if no changes are required.
r )r r r r r �wrap_subprocess_env? s
z%SubprocessContext.wrap_subprocess_envN)�__name__�
__module__� __qualname__r �strr r r r r r r r . s r c @ s� e Zd ZU dZded< dZded< dZded< deedddd dd
d dd �
d
d�Z ed�dd�Z
dd�dd�Z dd�dd�Z dd�dd�Z
ee ee d�dd�ZdS ) �SessionNz"tempfile.TemporaryDirectory | None� _controldirz
str | None� _controlsockz!asyncio.subprocess.Process | None�_processFzMapping[str, str] | Nonez
int | NonezInteractionHandler | None)
� destination�handle_host_key�
configfile�
identity_file�
login_name�options�pkcs11�port�interaction_responderr
c
� sl t j�t j�dd�d�}
t j|
dd� tj|
d�| _| jj � d�| _
t | jj �} t t j�} | | d< d | d
< d | d < d
dd| j
dddd| � �g}
|d ur�|
�
d|� �� |d ur�|
�
d|� �� |d ur�|D ] }|
�
d|� d|| � �� q�|d u�r|
�
d|� �� |d u�r |
�
d|� �� |d u�r8|
�
d|� �� |�rbtd��rb|
�dd| � d�dd g� t| d u�rt| gng �}tjd g|
�|�R | dtjjtjj|d d � d ��I d H }z,|�� I d H t j�| j
��s�J �|| _W n� t�y" } z*|�� I d H t�t |��d �W Y d }~nNd }~0 t �yf z |� � W n t �yP Y n0 |�� I d H � Y n0 d S )!NZXDG_RUNTIME_DIRz/runZfernyT)�exist_ok)�dirz/socketZ SSH_ASKPASS�forceZSSH_ASKPASS_REQUIRE�-ZDISPLAYz-Mz-N�-Sr zPermitLocalCommand=yesz
LocalCommand=z-Fz-ir z-Iz-pz-lZKnownHostsCommandzKnownHostsCommand=z %I %H %t %K %fzStrictHostKeyChecking=yesz /usr/bin/sshc S s t ttj�S �N)�prctl�PR_SET_PDEATHSIG�signal�SIGKILLr r r r �<lambda>� � z!Session.connect.<locals>.<lambda>)r Zstart_new_session�stdin�stdoutr Z
preexec_fn) �os�path�join�environ�get�makedirs�tempfileZTemporaryDirectoryr$ �namer% r �dict�appendr �extendr �asyncioZcreate_subprocess_execr r Z communicate�existsr& r �waitr Z get_exception_for_ssh_stderrr" �
BaseException�kill�ProcessLookupError)r r' r( r) r* r+ r, r- r. r/ ZrundirZ askpass_pathr r �keyZagentZprocess�excr r r �connectT sl
�
��
$ zSession.connect)r
c C s
| j d uS r5 )r& �r r r r � is_connected� s zSession.is_connectedc � s" | j d usJ �| j �� I d H d S r5 )r& rK rR r r r rK � s z Session.waitc C s | j d usJ �| j �� d S r5 )r& Z terminaterR r r r �exit� s z Session.exitc � s | � � | �� I d H d S r5 )rT rK rR r r r �
disconnect� s zSession.disconnectr c C s* | j d usJ �dd| j dgttj|��R S )Nr r4 � )r% �map�shlex�quoter r r r r � s z Session.wrap_subprocess_args)FNNNNNNN)r r r! r$ �__annotations__r% r& r" �boolrQ rS rK rT rU r r r r r r r# L s6
�� Zr# )r
) rI Zctypes� functoolsZloggingr> rX r8 r rD �typingr r rV r Zinteraction_agentr r r r ZCDLLr6 Z getLoggerr �loggerr7 � lru_cacher" r[ r r r# r r r r �<module> s$
Mini Shell