Current File : //usr/lib/python3.9/site-packages/cockpit/__pycache__/peer.cpython-39.pyc
a
��j�/ � @ s� d dl Z d dlZd dlZd dlmZmZmZ ddlmZm Z ddl
m Z ddl m
Z
mZmZ 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)�Any�Callable�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 sj e Zd ZU ded< ded< ed�� fdd� Zdd �d
d �Zee ee e e
j d �d
d�Z d7de
ed�dd�Zd8de
dd�dd�Zeg df dd�dd�Zdd �dd�Zedd�dd �Zeedd �d d �Zed �d d!�Zd"dd#�d$d%�Zdd �d&d'�Zeeedd(�d)d*�Zeedd+�d,d-�Z eeedd(�d.d/�Z eedd+�d0d1�Z ddedd2�d3d4�Z dd �d5d6�Z � Z!S )9�Peerzlist[Callable[[], None]]�done_callbacksz!asyncio.Future[JsonObject] | None� 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__N��returnc � s t �d S r )�NotImplementedError�r r r r �do_connect_transport+ s zPeer.do_connect_transport)�argv�env�kwargsr( c � s@ t �� }tdd� |D ��}t|| |fdttjfi |��i|��S )Nc s s | ]}|� d d�V qdS )�=r N)�split)�.0�er r r � <genexpr>1 � z Peer.spawn.<locals>.<genexpr>r- )r# r$ �dictr �os�environ)r r, r- r. ZloopZuser_envr r r �spawn. s z
Peer.spawnz
str | None)� init_hostr. r( c � s� �j dusJ �ddd�� �fdd� }t���� �� � �|� zNz�j I dH }W n" ttfyp � 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.
Nzasyncio.Task[None]��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_doneE s z&Peer.start.<locals>._connect_task_donez$ sending init message back, host %s�initr )�command�version�host)N)
r r# � create_taskr+ �add_done_callbackr �BrokenPipeError�done�cancel�logger�debug�
write_controlZ
thaw_endpoint)r r9 r. rB Z init_messager rA r �start4 s(
�
z
Peer.startc s: dd d�� fdd� }t �| j|fi |���� � �|� d S )Nzasyncio.Task[JsonObject]r: 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_donev s
z2Peer.start_in_background.<locals>._start_task_done)r# rG rO rH )r r9 r. rR r rQ r �start_in_backgroundu s zPeer.start_in_background)�callbackr( c C s | j �|� d S r )r �append)r rT r r r rH � 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 rW r r r � do_authorize� s zPeer.do_authorize)rD rW r( c C sb |dkr*| j d ur*t�d� | j �|� n4|dkr>| �|� n |dkrP| �� ntd|� ���d S )NrC z:Got init message with active init_future. Setting result.Z authorizezsuperuser-init-donez$Received unexpected control message )r rL rM Z
set_resultrX rV r )r rD rW r r r �transport_control_received� s
z Peer.transport_control_receivedc C s t �d| jj� dS )Nz Peer %s received unexpected EOFT)rL rM r& r r* r r r � eof_received� s zPeer.eof_receivedzException | None)r@ 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[ rW zinternal-error�[z] )rL rM r& r �typeZshutdown_endpoint�
isinstancer r r �attrsr rJ Z
set_exceptionrK r )r r@ rT 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) r_ Z transportr rL rM r& r Zget_returncoder r? r )r �
returncoder r r �process_exited� s
zPeer.process_exited)�channelrD rW r( c C s$ | j d urtd��| �|||� d S )Nz7Received unexpected channel control message before init)r r Zsend_channel_control�r rd rD rW r r r �channel_control_received� s
z Peer.channel_control_received)rd �datar( c C s" | j d urtd��| �||� d S )Nz,Received unexpected channel data before init)r r Zsend_channel_data�r rd rg r r r �channel_data_received� s
zPeer.channel_data_receivedc C s | j d u sJ �| �|� d S r �r rN re r r r �do_channel_control� s zPeer.do_channel_controlc C s | j d u sJ �| �||� d S r )r Zwrite_channel_datarh r r r �do_channel_data� s zPeer.do_channel_data)rF �grouprW r( c C s | j d u sJ �| �|� d S r rj )r rF rm rW 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 �__annotations__r
r r+ r �strr r# Z Transportr8 r r rO rS r rH rV rX rY �boolrZ ra rc rf �bytesri rk rl rn ro �
__classcell__r r r% r r s(
A
r c s\ e Zd ZU eed< eed< ee ed< ee ed< eed�� fdd� Zdd �d
d �Z � Z
S ) �ConfiguredPeer�config�method�argsr- �r! rv c s. || _ |j| _|j| _|j| _t� �|� d S r )rv rw r8 rx r7 r- r" r �r r! rv r% r r r � s
zConfiguredPeer.__init__Nr' c � s( | j d usJ �| �| j | j�I d H d S r )rx r8 r- r* r r r r+ � s z#ConfiguredPeer.do_connect_transport) r r r r rp rq r r
r r+ rt r r r% r ru � s
ru c sj e Zd ZU eed< eed< ded< eed�� fdd� Zedd�d d
�Zd d �d
d�Z d d �dd�Z
� Z S )�PeerRoutingRulerv �matchz Peer | None�peerry c s$ t � �|� || _|j| _d | _d S r )r" r rv r| r} rz 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))r9 ) r| �itemsrL rM r} ru r! rv rH � peer_closedr9 rS )r r �key�valuer r r �
apply_rule� s
zPeerRoutingRule.apply_ruleNr' c 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 rp r r
r r� r� r� rt r r r% r r{ � s
r{ c @ sL e Zd ZU g Zded< edd�dd�Zee dd�d d
�Z dd �d d
�Z
dS )�PeersRoutingRulezlist[PeerRoutingRule]�ruleszEndpoint | Noner~ 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)rL rM �lenr� rv �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 )NrF )
r� Z
privilegedr| �listrv �remover{ r! rU r� )r r� Z old_rulesrv r� r r r � set_configs' s
z PeersRoutingRule.set_configsr' c C s | j D ] }|�� qd S r )r� r� )r r� r r r r� ? s
zPeersRoutingRule.shutdown) r r r r� rp r r� r r r� r� r r r r r� s
r� )!r# Zloggingr6 �typingr r r Zjsonutilr r Zpackagesr Zprotocolr r
r r! r r
r Z
transportsr r Z getLoggerr rL r r> r r ru r{ r� r r r r �<module> s
D'
Mini Shell