관리-도구
편집 파일: transport.cpython-39.pyc
a ��gB � @ 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