관리-도구
편집 파일: interaction_agent.cpython-39.pyc
a ��g�= � @ sR d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl m Z m Z m Z m Z mZ ddlmZ e�e�Ze�d�ZdZde�d�d d �ZG d d � d e�Zz ejZW n, ey� d eje e e d d�dd�ZY n0 ej d�dd�Z G dd� d�Z G dd� de �Z G dd� d�Z!e"e"d�dd �Z#ej$e e e"ddf d �d d ��Z%dS )!� N)�Any�Callable�ClassVar� Generator�Sequence� )�interaction_clients ferny ([^ ]*) z ferny {(command, args)!r} zW import sys def command(command, *args): sys.stderr.write(fz%) sys.stderr.flush() z9 def end(): command('ferny.end') )�command�endc @ s e Zd ZdS )�InteractionErrorN)�__name__� __module__� __qualname__� r r �K/usr/lib/python3.9/site-packages/cockpit/_vendor/ferny/interaction_agent.pyr 3 s r z"tuple[bytes, list[int], int, None])�sock�bufsize�maxfds�flags�returnc C s� t � d�}| �|t�||j ��\}}}}|D ]B\}} } |tjkr.| tjkr.|�| d t| �t| �|j � � q.|t |�||fS )N�i) �arrayZrecvmsg�socketZCMSG_LEN�itemsizeZ SOL_SOCKETZ SCM_RIGHTSZ frombytes�len�list) r r r r �fds�msgZancdata�addrZ cmsg_levelZ cmsg_typeZ cmsg_datar r r �recv_fds< s &r �r c C s* z t �� W S ty$ t �� Y S 0 d S �N)�asyncio�get_running_loop�AttributeErrorZget_event_loopr r r r r# G s r# c @ s4 e Zd ZU eee ed< eddedd�dd�ZdS )�InteractionHandler�commands�tuple[object, ...]� list[int]N�r �argsr �stderrr c � s t �d S r! )�NotImplementedError��selfr r* r r+ r r r � run_commandR s z InteractionHandler.run_command)r r r r r �str�__annotations__r/ r r r r r% O s r% c @ s� e Zd ZU dZeee ed< eeedd�dd�Zeeeeee d�dd �Z ed d ed d �dd�Z d d ed d�dd�Z ed d ed d �dd�Z d S )�AskpassHandler)� ferny.askpassr& z str | None)�messages�prompt�hintr c � s dS )a+ Prompt the user for an authentication or confirmation interaction. 'messages' is data that was sent to stderr before the interaction was requested. 'prompt' is the interaction prompt. The expected response type depends on hint: - "confirm": ask for permission, returning "yes" if accepted - example: authorizing agent operation - "none": show a request without need for a response - example: please touch your authentication token - otherwise: return a password or other form of text token - examples: enter password, unlock private key In any case, the function should properly handle cancellation. For the "none" case, this will be the normal way to dismiss the dialog. Nr )r. r4 r5 r6 r r r � do_askpassY s zAskpassHandler.do_askpass)�reason�host� algorithm�key� fingerprintr c � s dS )a� Prompt the user for a decision regarding acceptance of a host key. The "reason" will be either "HOSTNAME" or "ADDRESS" (if `CheckHostIP` is enabled). The host, algorithm, and key parameters are the values in the form that they would appear one a single line in the known hosts file. The fingerprint is the key fingerprint in the format that ssh would normally present it to the user. In case the host key should be accepted, this function needs to return True. Returning False means that ssh implements its default logic. To interrupt the connection, raise an exception. Fr )r. r8 r9 r: r; r< r r r � do_hostkeyo s zAskpassHandler.do_hostkeyr' r( Nr) c � s dS )z�Handle a custom command. The command name, its arguments, the passed fds, and the stderr leading up to the command invocation are all provided. See doc/interaction-protocol.md Nr r- r r r �do_custom_command s z AskpassHandler.do_custom_command)r* r r+ r c � s� t �d|||� zh|\}}t|t�s(J �tdd� |D ��s>J �t|t�sLJ �tdd� |�� D ��sfJ �t|�dksvJ �W n< tt t fy� } z t � d|||� W Y d }~d S d }~0 0 t |� d�d����}t |� d�d����}�z`t� } z t�� } W n t�y tj�� } Y n0 | d u�s"J �| �|| j� t|�dk�r�|d } |�d d �} t �d || | � | �|| | �I d H } t �d | � | d u�r4t| |d� td|d� n�t|�dk�r(|\}}}}}}|dv �rt �d|||||� | �|||||�I d H �rt||||d� n t �d|� td|d� n t � d|� W | �|� n | �|� 0 W d � n1 �sd0 Y W d � n1 �s�0 Y d S )Nz _askpass_command(%s, %s, %s)c s s | ]}t |t�V qd S r! �� isinstancer0 )�.0�argr r r � <genexpr>� � z2AskpassHandler._askpass_command.<locals>.<genexpr>c s s& | ] \}}t |t�o t |t�V qd S r! r? )rA r; �valr r r rC � rD � z4Invalid arguments to askpass interaction: %s, %s: %sr �wr ZSSH_ASKPASS_PROMPT� zdo_askpass(%r, %r, %r)zdo_askpass answer %r)�file� )ZADDRESSZHOSTNAMEz do_hostkey(%r, %r, %r, %r, %r)z$ignoring KnownHostsCommand reason %rz?Incorrect number of command-line arguments to ferny-askpass: %s)�logger�debugr@ r �all�dict�itemsr � ValueError� TypeError�AssertionError�error�open�popr# r" Z current_taskr$ �Task� add_reader�cancel�getr7 �printr= � remove_reader)r. r* r r+ �argv�env�exc�status�stdout�loop�taskr5 r6 ZanswerZargv0r8 r9 r: r; r< r r r �_askpass_command� sN ( z AskpassHandler._askpass_commandc � sJ t �d||||� |dkr0| �|||�I d H n| �||||�I d H d S )Nzrun_command(%s, %s, %s, %s)r3 )rK rL rc r> r- r r r r/ � s zAskpassHandler.run_command)r r r r& r r r0 r1 r7 �boolr= r> rc r/ r r r r r2 V s � 3r2 c @ s e Zd ZU ded< ejed< ded< eed< ejed< ejed< d ed < d Zd ed < dZ e ed< d d�dd�Z dd d�dd�Z e e dd d�dd�Ze dd d�d d �Zd d�d d �Zd-ee d d!d d"�d#d$�Zed�d%d&�Zd d�d'd(�Zd d�d)d*�Zd d�d+d,�Zd S ).�InteractionAgentz dict[str, InteractionHandler]� _handlers�_loopzset[asyncio.Task]�_tasks�_buffer�_ours�_theirszasyncio.Future[str]�_completion_futureNzNone | str | Exception�_pending_resultF�_endr c C s� t �d| � | jd u s | jr(t �d� nX| j�� r>t �d� nBt| jt�rht �d| j� | j�| j� nt �d� | j� | j� d S )Nz_consider_completion(%r)z but not ready yetz but already completez- submitting stderr (%r) to completion_futurez0 submitting exception (%r) to completion_future) rK rL rm rh rl Zdoner@ r0 Z set_resultZ set_exception�r. r r r �_consider_completion� s z%InteractionAgent._consider_completionzstr | Exception)�resultr c C s� t �d| |� | jd u r || _| j�� dkrLt �d| j� | j�| j�� � | jD ]}t �d|� |�� qRt �d� | j � � | j� � | � � d S )Nz_result(%r, %r)���z remove_reader(%r)z cancel(%r)z closing sockets) rK rL rm rj �filenorg r[ rh rX rk �closerp )r. rq rb r r r �_result� s zInteractionAgent._resultr( )r+ � command_blobr r c s< t �d|||� z2t�|�� �\}}t|t�r8t|t�s@td��W n< t t t tfy~ } z t � d||� W Y d }~d S d }~0 0 |dkr�d�_ ���jjdd�� d S z�j| � W n ty� t � d|� Y d S 0 t|���j�� �||�|�� ���tjd d �� ���fd d � }��|� �j��� g |d d �< d S ) Nz_invoke_command(%r, %r, %r)zInvalid argument typesz&Received invalid ferny command: %s: %sz ferny.endT�replace��errorsz$Received unhandled ferny command: %s)�completed_taskr c s� | �u s J ��r t ���� � q �j��� z��� t�d� � W nZ tj yd t�d� � Y n< t y� } z$t�d� |� �� |� W Y d }~n d }~0 0 �� � d S )Nz%r completed cleanlyz%r was cancelledz %r raised %r) �osrt rU rh �removerq rK rL r" ZCancelledError� Exceptionru rp )rz r^ ��handlerr. rb Ztask_fdsr r � bottom_half s z5InteractionAgent._invoke_command.<locals>.bottom_half)rK rL �astZ literal_eval�decoder@ r0 �tuplerQ �UnicodeDecodeError� SyntaxErrorrP rS rn ru ri rf �KeyErrorr rg Z create_taskr/ r" rV �add_done_callbackrh �add)r. r+ rv r r r* r^ r� r r~ r �_invoke_command� s. z InteractionAgent._invoke_command)�datar r c C s t �d||� |dkr.| �| jjdd�� d S | j�|� t�| j�}t|� � �| _t |�dkr�| � |d |d g � |dd � }qT|r�| j� d�s�J | j��| jd d � }td�| _t |� d�d ��}|�� }W d � n1 s�0 Y | � |||� d S ) Nz_got_data(%r, %r)rD rw rx r r rF � rr �rb)rK rL ru ri r� �extend� COMMAND_RE�split� bytearrayrU r r� �endswithrT �read)r. r� r �chunksr+ Zcommand_channelr r r r � _got_data) s" &zInteractionAgent._got_datac C s� z�z t | jddtjd�\}}}}W nV tyL Y W |rHt�|�� � q4d S tyx } z| � |� W Y d }~nd }~0 0 | � ||� W |r�t�|�� � q�n|r�t�|�� � q�0 d S )N� � )r ) r rj r � MSG_DONTWAIT�BlockingIOErrorr{ rt rU �OSErrorru r� )r. r� r Z_flagsZ_addrr^ r r r � _read_readyB s � �z InteractionAgent._read_readyz asyncio.AbstractEventLoop | Nonez,Callable[[asyncio.Future[str]], None] | None)�handlersra � done_callbackr c C s~ |pt � | _| j�� | _t� | _i | _|D ]}|jD ]}|| j|< q4q*|d urZ| j�|� t � t j t j �\| _ | _t� | _d S r! )r# rg Z create_futurerl �setrh rf r& r� r Z socketpairZAF_UNIXZ SOCK_STREAMrk rj r� ri )r. r� ra r� r r r r r �__init__O s zInteractionAgent.__init__c C s | j �� S r! )rk rs ro r r r rs d s zInteractionAgent.filenoc C sf t �d| � | j�� dkr@t �d| j� | j�| j�� | j� n t �d� t �d| j� | j�� d S )Nz start(%r)rr z add_reader(%r)z# ...but agent is already finished.z close(%r)) rK rL rj rs rg rW r� rk rt ro r r r �startg s zInteractionAgent.startc C s� t �d| � zz| j�� dkr�t �d� t�t��D | j�dtj �}t �dt |�� |sXqf| j � |� q2W d � n1 sz0 Y W n. t y� } z| �|� W Y d }~n d }~0 0 | �| j jdd�� d S )Nzforce_completion(%r)rr z- draining pending stderr data (non-blocking)r� z got %d bytesrw rx )rK rL rj rs � contextlib�suppressr� Zrecvr r� r ri r� r� ru r� )r. r� r^ r r r �force_completionr s 0 z!InteractionAgent.force_completionc � s� t �d| � z>| �� t�| j�I d H }t �d| |� W t �d� | �� nt �d� | �� 0 | js|t �d� t|� � ��d S )Nz_communicate(%r)z$_communicate(%r) stderr result is %rz,_communicate finished. Ensuring completion.z<_communicate never saw ferny.end. raising InteractionError.) rK rL r� r" Zshieldrl r� rn r �strip)r. r+ r r r � communicate� s � z InteractionAgent.communicate)NN)r r r r1 r" �AbstractEventLoopr� r rm rn rd rp ru �bytesr� r� r� r r% r� �intrs r� r� r� r r r r re � s4 0 �� re )�tmpdirr c C sj t j�| d�}t �|t jt jB t jB t jB t jB d�}z"t � |t � t j �� W t �|� n t �|� 0 |S )Nz ferny-askpassi� )r{ �path�joinrT �O_CREAT�O_WRONLY� O_CLOEXEC�O_EXCL� O_NOFOLLOW�write� __loader__�get_datar �__file__rt )r� Z askpass_path�fdr r r �write_askpass_to_tmpdir� s (r� )�kwargsr c k s> t jf i | ���}t|�V W d � n1 s00 Y d S r! )�tempfileZTemporaryDirectoryr� )r� Z directoryr r r �temporary_askpass� s r� )r )&r r� r" r� Zloggingr{ �rer r� �typingr r r r r rH r Z getLoggerr rK �compiler� ZCOMMAND_TEMPLATEZBEIBOOT_GADGETSr} r r r$ r� r� r# r% r2 re r0 r� �contextmanagerr� r r r r �<module> sF �� � � o T