File: //usr/lib/python3.9/site-packages/cockpit/__pycache__/peer.cpython-39.pyc
a
�/�h�1 � @ s� d dl Z d dlZd dlZd dlmZmZmZmZ ddlm Z m
Z
ddl m Z ddl
mZmZmZ ddlmZmZmZ ddlmZmZ e�e�ZG d d
� d
e�ZG d d � d e �Z G d
d� deee�Z G dd� de �Z G dd� de�Z G dd� de�Z!dS )� N)�Callable�List�Optional�Sequence� )�
JsonObject� JsonValue)� BridgeConfig)�CockpitProblem�CockpitProtocol�CockpitProtocolError)�Endpoint�Router� RoutingRule)�SubprocessProtocol�SubprocessTransportc @ s e Zd ZdS )� PeerErrorN)�__name__�
__module__� __qualname__� r r �0/usr/lib/python3.9/site-packages/cockpit/peer.pyr s r c @ s e Zd Zed�dd�ZdS )�
PeerExited�� exit_codec C s
|| _ d S �Nr )�selfr r r r �__init__% s zPeerExited.__init__N)r r r �intr r r r r r $ s r c s� e Zd ZU eeg df ed< eej ed< e d�� fdd� Z
dd�dd �Z e e
e e
ejd
�d d �Zd4ee
eed
�dd�Zd5ee
edd
�dd�Zeg df dd�dd�Zdd�dd�Zedd�dd�Ze
edd�dd �Zed�d d �Zee dd �d d!�Zdd�d"d#�Z e
e
edd$�d%d&�Z e
e dd'�d(d)�Z e
e
edd$�d*d+�Z e
e dd'�d,d-�Z!d.d.edd/�d0d1�Z"dd�d2d3�Z#� Z$S )6�PeerN�done_callbacks� init_future)�routerc s, t � �|� | �� t�� �� | _g | _d S r )�superr Zfreeze_endpoint�asyncio�get_running_loopZ
create_futurer! r )r r" �� __class__r r r - s z
Peer.__init__��returnc � s t �d S r )�NotImplementedError�r r r r �do_connect_transport7 s zPeer.do_connect_transport)�argv�envr) c � s@ t �� }tdd� |D ��}t|| |fdttjfi |��i|��S )Nc s s | ]}|� d d�V qdS )�=r N)�split)�.0�er r r � <genexpr>= � z Peer.spawn.<locals>.<genexpr>r. )r$ r% �dictr �os�environ)r r- r. �kwargsZloopZuser_envr r r �spawn: s z
Peer.spawn)� init_hostr8 r) c � s� �j dusJ �tjdd�� �fdd� }t���� �� � �|� zNz�j I dH }W n" ttfyr � I dH � Y n0 W d�_ � �� s�� � � nd�_ � �� s�� � � 0 |dur�t
� d|� �j d dd|d�|�� ��
� |S )
a� Request that the Peer is started and connected to the router.
Creates the transport, connects it to the protocol, and participates in
exchanging of init messages. If anything goes wrong, the connection
will be closed and an exception will be raised.
The Peer starts out in a frozen state (ie: attempts to send messages to
it will initially be queued). If init_host is not None then an init
message is sent with the given 'host' field, plus any extra kwargs, and
the queue is thawed. Otherwise, the caller is responsible for sending
the init message and thawing the peer.
In any case, the return value is the init message from the peer.
N��taskr) c
s^ | � u s J �z | � � W n@ tjy, Y n. tyX } z��|� W Y d }~n
d }~0 0 d S r )�resultr$ �CancelledError� Exception�close)r< �exc�Z connect_taskr r r �_connect_task_doneQ s z&Peer.start.<locals>._connect_task_donez$ sending init message back, host %s�initr )�command�version�host)N)r! r$ �Task� create_taskr, �add_done_callbackr �BrokenPipeError�done�cancel�logger�debug�
write_controlZ
thaw_endpoint)r r: r8 rC Z init_messager rB r �start@ s(
�
z
Peer.startc s< t jd d�� fdd� }t �| j|fi |���� � �|� d S )Nr; c s: | � u s J �z | � � W n ttttjfy4 Y n0 d S r )r= �OSErrorr r
r$ r> )r< �Z
start_taskr r �_start_task_done� s
z2Peer.start_in_background.<locals>._start_task_done)r$ rH rI rQ rJ )r r: r8 rT r rS r �start_in_background� s zPeer.start_in_background)�callbackr) c C s | j �|� d S r )r �append)r rV r r r rJ � s zPeer.add_done_callbackc C s d S r r r+ r r r �do_superuser_init_done� s zPeer.do_superuser_init_done)�messager) c C s d S r r )r rY r r r � do_authorize� s zPeer.do_authorize)rE rY r) c C sb |dkr*| j d ur*t�d� | j �|� n4|dkr>| �|� n |dkrP| �� ntd|� ���d S )NrD z:Got init message with active init_future. Setting result.Z authorizezsuperuser-init-donez$Received unexpected control message )r! rN rO Z
set_resultrZ rX r )r rE rY r r r �transport_control_received� s
z Peer.transport_control_receivedc C s t �d| jj� dS )Nz Peer %s received unexpected EOFT)rN rO r' r r+ r r r � eof_received� s zPeer.eof_received)rA r) c C s� t �d| jjt|�|� |d u r.| jdd� npt|t�rh|jdkrP| jdd� q�| jdd|j� �d� n6t|t �r�| �|j
� n | jdd |jj� d
|��d� | j d ur�| j � � s�|d ur�| j �
|� n
| j �� | jD ]
}|� q�d S ) Nz Peer %s connection lost %s %sZ
terminated)�problem� z
no-cockpitzPeer exited with status )r] rY zinternal-error�[z] )rN rO r' r �typeZshutdown_endpoint�
isinstancer r r
�attrsr! rL Z
set_exceptionrM r )r rA rV r r r � do_closed� s$
�
zPeer.do_closedc C sR t | jt�sJ �t�d| jj| j�� � | j�� }t |t�s@J �| � t
|�� d S )NzPeer %s exited, status %d) ra Z transportr rN rO r' r Zget_returncoder r@ r )r �
returncoder r r �process_exited� s
zPeer.process_exited)�channelrE rY r) c C s$ | j d urtd��| �|||� d S )Nz7Received unexpected channel control message before init)r! r Zsend_channel_control�r rf rE rY r r r �channel_control_received� s
z Peer.channel_control_received)rf �datar) c C s" | j d urtd��| �||� d S )Nz,Received unexpected channel data before init)r! r Zsend_channel_data�r rf ri r r r �channel_data_received� s
zPeer.channel_data_receivedc C s | j d u sJ �| �|� d S r �r! rP rg r r r �do_channel_control� s zPeer.do_channel_controlc C s | j d u sJ �| �||� d S r )r! Zwrite_channel_datarj r r r �do_channel_data� s zPeer.do_channel_dataz
str | None)rG �grouprY r) c C s | j d u sJ �| �|� d S r rl )r rG ro rY r r r �do_kill� s z Peer.do_killc C s | � � d S r )r@ r+ r r r �do_close� s z
Peer.do_close)N)N)%r r r r r �__annotations__r r$ ZFuturer r r, r �strZ Transportr9 r r rQ rU rJ rX rZ r[ �boolr\ r? rc re rh �bytesrk rm rn rp rq �
__classcell__r r r& r r ) s(
A
r c sT e Zd ZU eed< ee ed< ee ed< eed�� fdd� Zdd�d d
�Z � Z
S ) �ConfiguredPeer�config�argsr. �r" rx c s& || _ |j| _|j| _t� �|� d S r )rx r9 ry r7 r. r# r �r r" rx r& r r r � s zConfiguredPeer.__init__Nr( c � s | � | j| j�I d H d S r )r9 ry r. r+ r r r r, � s z#ConfiguredPeer.do_connect_transport) r r r r rr r rs r r r, rv r r r& r rw � s
rw c sf e Zd ZU eed< eed< ee ed< eed�� fdd� Z eee d�dd �Z
d
d � Z d d
� Z � Z
S )�PeerRoutingRulerx �match�peerrz c s$ t � �|� || _|j| _d | _d S r )r# r rx r} r~ r{ r& r r r s zPeerRoutingRule.__init__��optionsr) c C s� | j �� D ]P\}}||vr,t�d|� d S |d ur
|| |kr
t�d||| |� d S q
| jd u r�t| j| j�| _| j�| j � | jj
s�J �| jj | jj
d� | jS )Nz+ rejecting because key %s is missingz; rejecting because key %s has wrong value %s (vs %s))r: ) r} �itemsrN rO r~ rw r" rx rJ � peer_closedr: rU )r r� �key�valuer r r �
apply_rule s
zPeerRoutingRule.apply_rulec C s
d | _ d S r )r~ r+ r r r r� s zPeerRoutingRule.peer_closedc C s | j d ur| j �� d S r )r~ r@ r+ r r r �shutdown s
zPeerRoutingRule.shutdown)r r r r rr r r r r r r� r� r� rv r r r& r r| � s
r| c @ sN e Zd ZU g Zee ed< eee d�dd�Z
e e dd�dd�Z
d d
� ZdS ) �PeersRoutingRule�rulesr c C s` t �dt| j�� | jD ]8}t �d|jj� |�|�}|d urt �d� | S qt �d� d S )Nz considering %d rulesz considering %sz selectedz no peer rules matched)rN rO �lenr� rx �namer� )r r� �ruleZendpointr r r r� % s
zPeersRoutingRule.apply_ruleN)�bridge_configsr) c C s| | j }g | _ |D ]T}|jsd|jv r&qt|�D ] }|j|kr.|�|� qXq.t| j|�}| j �|� q|D ] }|� � qjd S )NrG )
r� Z
privilegedr} �listrx �remover| r" rW r� )r r� Z old_rulesrx r� r r r � set_configs0 s
z PeersRoutingRule.set_configsc C s | j D ] }|�� qd S r )r� r� )r r� r r r r� H s
zPeersRoutingRule.shutdown)r r r r� r r| rr r r r
r� r r r� r� r r r r r� " s
r� )"r$ Zloggingr6 �typingr r r r Zjsonutilr r Zpackagesr Zprotocolr
r r r" r
r r Z
transportsr r Z getLoggerr rN r r? r r rw r| r� r r r r �<module> s
D'