Current File : //usr/lib/python3.9/site-packages/cockpit/_vendor/ferny/__pycache__/transport.cpython-39.pyc
a
��hB � @ s� d dl Z d dlZd dlZd dlZd dlmZmZmZmZmZ ddl m
Z
m Z m Z ddl
mZ e�e�Zede jd�ZG dd � d e�ZG d
d � d e je j�ZdS ) � N)�Any�Callable�Iterable�Sequence�TypeVar� )�InteractionAgent�InteractionHandler�get_running_loop)� get_exception_for_ssh_stderr�P)�boundc s8 e Zd ZU eed< eed< eedd�� fdd� Z� ZS )�SubprocessError�
returncode�stderrN)r r �returnc s t � �||� || _|| _d S �N)�super�__init__r r )�selfr r �� __class__� �C/usr/lib/python3.9/site-packages/cockpit/_vendor/ferny/transport.pyr $ s zSubprocessError.__init__)�__name__�
__module__� __qualname__�int�__annotations__�strr �
__classcell__r r r r r s
r c @ s� e Zd ZU eed< ded< eed< ejed< dZeed< dZ d ed
< dZ
d ed < dZ d
ed< dZ ded< dZ
ded< dZded< dZeed< dZeed< edkdeg ef ee dee eedd �d d ��Zejdd �d d!�Zdd"�d#d$�Zd%dd&�d'd(�Zej dd)�d*d+�Z ddd,�d-d.�Z e e dd/�d0d1�Z!e ddd2�d3d4�Z"dd"�d5d6�Z#dd"�d7d8�Z$dd"�d9d:�Z%dlddd,�d;d<�Z&ed"�d=d>�Z'dmee(e(d?�d@dA�Z)ej*dd �dBdC�Z+ejd"�dDdE�Z,ed"�dFdG�Z-dd"�dHdI�Z.dd"�dJdK�Z/dd"�dLdM�Z0ed"�dNdO�Z1e d"�dPdQ�Z2dRd"�dSdT�Z3dnddddU�dVdW�Z4e ddX�dYdZ�Z5e6e dd[�d\d]�Z7dd"�d^d_�Z8e d"�d`da�Z9dd"�dbdc�Z:dd"�ddde�Z;e ddf�dgdh�Z<dd"�didj�Z=dS )o�FernyTransport�_agentz@asyncio.Task[tuple[asyncio.SubprocessTransport, FernyTransport]]�
_exec_task�_is_ssh� _protocolF�_protocol_disconnectedNz"asyncio.SubprocessTransport | None�_subprocess_transportz asyncio.WriteTransport | None�_stdin_transportz asyncio.ReadTransport | None�_stdout_transportzException | None�
_exceptionz
str | None�_stderr_outputz
int | None� _returncode�_transport_disconnected�_closedr Tztype[typing.Self]z asyncio.AbstractEventLoop | Noneztuple[typing.Self, P])�cls�protocol_factory�args�loop�interaction_handlers�is_ssh�kwargsr c s� t �d| ||� |� }| |�� |� _|du r2t� }t||� j�� _|�d� j�� � |� |j
� fdd�g|�R i |���� _ t j
dd�� fdd� }� j �|� � |fS ) a� Connects a FernyTransport to a protocol, using the given command.
This spawns an external command and connects the stdin and stdout of
the command to the protocol returned by the factory.
An instance of ferny.InteractionAgent is created and attached to the
stderr of the spawned process, using the provided handlers. It is the
responsibility of the caller to ensure that:
- a `ferny-askpass` client program is installed somewhere; and
- any relevant command-line arguments or environment variables are
passed correctly to the program to be spawned
This function returns immediately and never raises exceptions, assuming
all preconditions are met.
If spawning the process fails then connection_lost() will be
called with the relevant OSError, even before connection_made() is
called. This is somewhat non-standard behaviour, but is the easiest
way to report these errors without making this function async.
Once the process is successfully executed, connection_made() will be
called and the transport can be used as normal. connection_lost() will
be called if the process exits or another error occurs.
The return value of this function is the transport, but it exists in a
semi-initialized state. You can call .close() on it, but nothing else.
Once .connection_made() is called, you can call all the other
functions.
After you call this function, `.connection_lost()` will be called on
your Protocol, exactly once, no matter what. Until that happens, you
are responsible for holding a reference to the returned transport.
:param args: the full argv of the command to spawn
:param loop: the event loop to use. If none is provided, we use the
one which is (read: must be) currently running.
:param interaction_handlers: the handlers passed to the
InteractionAgent
:param is_ssh: whether we should attempt to interpret stderr as ssh
error messages
:param kwargs: anything else is passed through to `subprocess_exec()`
:returns: the usual `(Transport, Protocol)` pair
zspawn(%r, %r, %r)Nr c s � S r r r �r r r �<lambda>� � z&FernyTransport.spawn.<locals>.<lambda>)�taskr c
s� t �d� | � | � ju s J �z&| �� \}}|� u s6J �t �d� W nP tjyX Y d S ty� } z$t �d|� � �|� W Y d }~d S d }~0 0 � j|u s�J �� j d us�J �� j
d us�J �� j � � d S )Nzexec_completed(%r, %r)z
success.z OSError %r)
�logger�debugr# �result�asyncioZCancelledError�OSError�closer' r( r) r" �start)r9 � transport�me�excr6 r r �exec_completed� s
z,FernyTransport.spawn.<locals>.exec_completed)r: r; r$ r
r �_interaction_completedr" �
setdefault�filenoZ create_taskZsubprocess_execr# r= ZTaskZadd_done_callback) r/ r0 r1 r2 r3 r4 r5 �protocolrD r r6 r �spawn@ s 5( zFernyTransport.spawn)rH r c C s
|| _ d S r �r% �r rH r r r r � s zFernyTransport.__init__)r c C s: t �d| � | j�� s(t �d| j� d S | jd urJ| jsJt �d| j� d S | jd u rbt �d� d S | jrvt �d� d S d| _| jd ur�t �d| j� | j �
| j� n�| j dks�| j r�t �d � | j �
d � nj| j
r�| j d
kr�t �d | j� | j �
t| j�� n8t �d | j | j� | j d u�s J �| j �
t| j | j�� d S )
Nz_consider_disconnect(%r)z exec_task still running %rz transport still connected %rz agent still runningz already disconnectedTz disconnect with exception %rr z clean disconnect� z disconnect with ssh error %rz) disconnect with exit code %r, stderr %r)r: r; r# �doner' r- r+ r& r* r% �connection_lostr, r. r$ r r r6 r r r �_consider_disconnect� s4
z#FernyTransport._consider_disconnectzasyncio.Future[str])�futurer c
C sx t �d| |� z |�� | _t �d| j� W n@ tyj } z(t �d|� d| _| �|� W Y d }~n
d }~0 0 | �� d S )Nz _interaction_completed(%r, %r)z stderr: %rz exception: %r� )r: r; r< r+ � Exceptionr? rO )r rP rC r r r rE � s
z%FernyTransport._interaction_completed)rA r c C s� t �d| |� t|tj�s J �|| _|�d�}t|tj�s>J �|| _|�d�}t|tj �s^J �|| _
|�d�}|d u szJ �t �d| | j � | j � | � d S )Nzconnection_made(%r, %r)r r � z calling connection_made(%r, %r))
r: r; �
isinstancer= ZSubprocessTransportr' Zget_pipe_transportZWriteTransportr( Z
ReadTransportr) r% �connection_made)r rA Zstdin_transportZstdout_transportZstderr_transportr r r rU � s
z FernyTransport.connection_made)rC r c C s0 t �d| |� | jd u r || _d| _| �� d S )Nzconnection_lost(%r, %r)T)r: r; r* r- rO �r rC r r r rN � s
z FernyTransport.connection_lost)�fd�datar c C s0 t �d| |t|�� |dks J �| j�|� d S )Nz pipe_data_received(%r, %r, %r)r )r: r; �lenr% Z
data_received)r rW rX r r r �pipe_data_received� s z!FernyTransport.pipe_data_received)rW rC r c C sb t �d| ||� |dv s J �t|t�r*d }|d ur>| �|� n |dkr^| js^| j�� s^| �� d S )Nz pipe_connection_lost(%r, %r, %r))r r r )r: r; rT �BrokenPipeErrorr? r. r% Z eof_received)r rW rC r r r �pipe_connection_lost s
z#FernyTransport.pipe_connection_lostc C sB t �d| � | jd usJ �| j�� | _t �d| j� | j�� d S )Nzprocess_exited(%r)z ._returncode = %r)r: r; r' �get_returncoder, r" �force_completionr6 r r r �process_exited s
z FernyTransport.process_exitedc C s t �d| � | j�� d S )Nzpause_writing(%r))r: r; r% �
pause_writingr6 r r r r` s z FernyTransport.pause_writingc C s t �d| � | j�� d S )Nzresume_writing(%r))r: r; r% �resume_writingr6 r r r ra s z FernyTransport.resume_writingc C s� t �d| |� d| _| jd u r0t �d|� || _| j�� sNt �d� | j�� | jd ur�t �d� t� t
�� | j� � W d � n1 s�0 Y | j �
� d S )Nz
close(%r, %r)Tz setting exception %rz cancelling _exec_taskz closing _subprocess_transport)r: r; r. r* r# rM �cancelr' �
contextlib�suppress�PermissionErrorr? r" r^ rV r r r r? s
(zFernyTransport.closec C s | j d usJ �| j �� S r )r' �
is_closingr6 r r r rf 0 s zFernyTransport.is_closing)�name�defaultr c C s | j d usJ �| j �||�S r )r' �get_extra_info)r rg rh r r r ri 4 s z FernyTransport.get_extra_infoc C s t |tj�sJ �|| _d S r )rT r= �Protocolr% rK r r r � set_protocol8 s zFernyTransport.set_protocolc C s | j S r rJ r6 r r r � get_protocol< s zFernyTransport.get_protocolc C s� | j d usJ �z | j �� W S ty6 | j j Y S ty~ z | j jj}|�| j j� W Y dS t yx Y Y dS 0 Y n0 d S )NTF)
r) �
is_reading�NotImplementedErrorZ_paused�AttributeErrorZ_loopZ _selectorZget_keyZ_fileno�KeyError)r �selectorr r r rm ? s
zFernyTransport.is_readingc C s | j d usJ �| j �� d S r )r) �
pause_readingr6 r r r rr O s z FernyTransport.pause_readingc C s | j d usJ �| j �� d S r )r) �resume_readingr6 r r r rs S s z FernyTransport.resume_readingc C s4 | j d usJ �| jd us J �| j �� | j�� d S r )r( r' �abort�killr6 r r r rt W s
zFernyTransport.abortc C s | j d usJ �| j �� S r )r( �
can_write_eofr6 r r r rv ] s z FernyTransport.can_write_eofc C s | j d usJ �| j �� S r )r( �get_write_buffer_sizer6 r r r rw a s z$FernyTransport.get_write_buffer_sizeztuple[int, int]c C s | j d usJ �| j �� S r )r( �get_write_buffer_limitsr6 r r r rx e s z&FernyTransport.get_write_buffer_limits)�high�lowr c C s | j d usJ �| j �||�S r )r( �set_write_buffer_limits)r ry rz r r r r{ i s z&FernyTransport.set_write_buffer_limits)rX r c C s | j d usJ �| j �|�S r )r( �write)r rX r r r r| m s zFernyTransport.write)� list_of_datar c C s | j d usJ �| j �|�S r )r( �
writelines)r r} r r r r~ q s zFernyTransport.writelinesc C s | j d usJ �| j �� S r )r( � write_eofr6 r r r r u s zFernyTransport.write_eofc C s | j d usJ �| j �� S r )r' �get_pidr6 r r r r� { s zFernyTransport.get_pidc C s | j S r )r, r6 r r r r] s z FernyTransport.get_returncodec C s | j d usJ �| j �� d S r )r' ru r6 r r r ru � s zFernyTransport.kill)�numberr c C s | j d usJ �| j �|� d S r )r' � send_signal)r r� r r r r� � s zFernyTransport.send_signalc C s | j d usJ �| j �� d S r )r' � terminater6 r r r r� � s zFernyTransport.terminate)Nr T)N)N)NN)>r r r r r �boolr= rj r& r' r( r) r* r+ r, r- r. � classmethodr r r r r r rI r rO rE Z
BaseTransportrU rN r �bytesrZ r\ r_ r` ra r? rf �objectri Z BaseProtocolrk rl rm rr rs rt rv rw rx r{ r| r r~ r r� r] ru r� r� r r r r r! * st
�
�c,
r! )r= rc Zlogging�typingr r r r r Zinteraction_agentr r r
Z
ssh_errorsr Z getLoggerr r: rj r rR r Z TransportZSubprocessProtocolr! r r r r �<module> s
Mini Shell