Current File : //lib/python3.9/site-packages/cockpit/__pycache__/router.cpython-39.pyc
a
��j�% � @ s� d dl Z d dlZd dlZd dlmZmZmZ ddlmZm Z ddl
m Z m Z m
Z
e�e�ZG dd� d�ZG dd � d �ZG d
d � d e �ZG d d
� d
�ZG dd� de
�ZdS )� N)�Dict�List�Optional� )�
JsonObject� JsonValue)�CockpitProblem�CockpitProtocolError�CockpitProtocolServerc @ s. e Zd ZdZdd� Zdd� Zdd�dd �ZdS )
�ExecutionQueueaw Temporarily delay calls to a given set of class methods.
Functions by replacing the named function at the instance __dict__
level, effectively providing an override for exactly one instance
of `method`'s object.
Queues the invocations. Run them later with .run(), which also reverses
the redirection by deleting the named methods from the instance.
c C s* t �� | _|| _| jD ]}| �|� qd S �N)� collections�deque�queue�methods�_wrap)�selfr �method� r �2/usr/lib/python3.9/site-packages/cockpit/router.py�__init__ s
zExecutionQueue.__init__c s t � j� jj� �fdd�� d S )Nc s �j �� | f�S r )r �append)�args�r r r r �<lambda>% � z&ExecutionQueue._wrap.<locals>.<lambda>)�setattr�__self__�__func__�__name__)r r r r r r ! s zExecutionQueue._wrapN��returnc C sJ t �dt| j�� | jD ]\}}||� q| jD ]}t|j|jj� q0d S )Nz.ExecutionQueue: Running %d queued method calls) �logger�debug�lenr r �delattrr r r )r r r r r r �run' s
zExecutionQueue.run)r �
__module__� __qualname__�__doc__r r r& r r r r r s r c @ s� e Zd ZU ded< dZee ed< dd�dd�Zdd�d d
�Zdd�d d �Z dd�d
d�Z
e e e dd�dd�Z
e edd�dd�Zdde dd�dd�Ze edd�dd�Ze e dedd �d d �Zd"dedd �d d!�ZdS )#�Endpoint�Router�routerN� _Endpoint__endpoint_frozen_queue�r, c C s |� | � || _d S r )� add_endpointr, �r r, r r r r 4 s
zEndpoint.__init__r c C s4 | j d u sJ �t�d| � t| j| j| jh�| _ d S )NzFreezing endpoint %s)r- r" r# r �do_channel_control�do_channel_data�do_kill�r r r r �freeze_endpoint8 s zEndpoint.freeze_endpointc C s. | j d usJ �t�d| � | j �� d | _ d S )NzThawing endpoint %s)r- r" r# r& r4 r r r �
thaw_endpoint= s
zEndpoint.thaw_endpointc C s t �d S r ��NotImplementedErrorr4 r r r �do_closeD s zEndpoint.do_close��channel�command�messager! c C s t �d S r r7 )r r; r< r= r r r r1 G s zEndpoint.do_channel_control�r; �datar! c C s t �d S r r7 �r r; r? r r r r2 J s zEndpoint.do_channel_data�
str | None��host�groupr= r! c C s t �d S r r7 )r rC rD r= r r r r3 M s zEndpoint.do_killc C s | j �||� d S r )r, Zwrite_channel_datar@ r r r �send_channel_dataQ s zEndpoint.send_channel_data�JsonObject | None)r; r< �msg�kwargsr! c K sD | j j|f||d�|�� |dkr@| j j| �|� | j �|� d S )N)r; r< �close)r, �
write_control� endpoints�remove� drop_channel)r r; r< rG rH r r r �send_channel_controlT s z Endpoint.send_channel_control)rG rH r! c K s | j j| |fi |�� d S r )r, �shutdown_endpoint)r rG rH r r r rO \ s zEndpoint.shutdown_endpoint)N)r r' r( �__annotations__r- r r r r5 r6 r9 �strr r1 �bytesr2 r3 rE r rN rO r r r r r* 0 s
� r* c @ s e Zd ZdS )� RoutingErrorN)r r' r( r r r r rS ` s rS c @ sF e Zd ZU ded< dd�dd�Zeee d�dd�Zd d
�d d �Z d S )
� RoutingRuler+ r, r. c C s
|| _ d S r r. r0 r r r r g s zRoutingRule.__init__��optionsr! c C s t �dS )a] Check if a routing rule applies to a given 'open' message.
This should inspect the options dictionary and do one of the following three things:
- return an Endpoint to handle this channel
- raise a RoutingError to indicate that the open should be rejected
- return None to let the next rule run
Nr7 )r rV r r r �
apply_rulej s zRoutingRule.apply_ruleNr c C s t �d S r r7 r4 r r r �shutdownu s zRoutingRule.shutdown)
r r' r( rP r r r r* rW rX r r r r rT d s
rT c @ s* e Zd ZU ee ed< eeef ed< ded< e j
ed< dZ e ed< ee d�d d
�Z
ed �d d
�Zeed�dd�Zedd�dd�Zedd�dd�Zd.ededd�dd�Zd d edd �d d �Zeeedd �d!d"�Zeedd#�d$d%�Ze d �d&d'�ZdZee j ed(< ee dd)�d*d+�Z dd �d,d-�Z dS )/r+ �
routing_rules�
open_channelszdict[Endpoint, set[str]]rK � no_endpointsF�_eof)rY c C s: |D ]
}| |_ q|| _i | _i | _t�� | _| j�� d S r )r, rY rZ rK �asyncio�Eventr[ �set)r rY �ruler r r r � s
zRouter.__init__r c C s i S )z7Used by the 'info' channel. Gets overridden in Bridge.r r4 r r r �info� s z Router.inforU c C sT | j D ]6}t�d|� |�|�}|d urt�d|� | S qt�d� td��d S )Nz applying rule %sz resulting endpoint is %sz No rules matchedz
not-supported)rY r" r# rW rS )r rV r` �endpointr r r � check_rules� s
zRouter.check_rulesN)r; r! c C sD z | j �|� t�d|� W n" ty> t�d|| j � Y n0 d S )Nzrouter dropped channel %sz.trying to drop non-existent channel %s from %s)rZ �popr" r# �KeyError�error)r r; r r r rM � s
zRouter.drop_channel)rb r! c C s t � | j|< | j�� d S r )r_ rK r[ �clear)r rb r r r r/ � s zRouter.add_endpointrF )rb rG rH r! c K s� | j �|�}t�d|||� |D ]&}| j|fd|d�|�� | �|� q | j sX| j�� | jr�t�d| j � | j s�| j r�t�d� | j �
� d S )Nz'shutdown_endpoint(%s, %s) will close %srI �r< r; � endpoints remaining: %rz close transport) rK rd r" r# rJ rM r[ r_ r\ Z transportrI )r rb rG rH Zchannelsr; r r r rO � s
zRouter.shutdown_endpointrA rB c C s: t | j�}t�d||t|�� |D ]}|�|||� q"d S )Nz+do_kill(%s, %s). Considering %d endpoints.)r_ rK r" r# r$ r3 )r rC rD r= rK rb r r r r3 � s
zRouter.do_killr: c
C s� |dkr�|| j v rtd��z"t�d||�d�� | �|�}W n: tyv } z"| j|�� d|d� W Y d }~d S d }~0 0 || j |< | j | �
|� n$z| j | }W n t y� Y d S 0 |� |||� d S )N�openzchannel is already openz5Trying to find endpoint for new channel %s payload=%sZpayloadrI rh )
rZ r r" r# �getrc rS rJ Z get_attrsrK �addre r1 )r r; r< r= rb �excr r r �channel_control_received� s
z Router.channel_control_receivedr> c C s4 z| j | }W n ty" Y d S 0 |�||� d S r )rZ re r2 )r r; r? rb r r r �channel_data_received� s
z Router.channel_data_receivedc C sF t �d| � t| j�}|D ] }|�� qd| _t �d| j� t| j�S )Nzeof_received(%r)Tri )r" r# r_ rK r9 r\ �bool)r rK rb r r r � eof_received� s
zRouter.eof_received�_communication_done)rm r! c C s> | j s| �� | jd ur:|d u r.| j�d � n | j�|� d S r )r\ rq rr Z
set_resultZ
set_exception)r rm r r r � do_closed� s
zRouter.do_closedc � s| | j du sJ �t�� �� | _ zBz| j I dH W n ttfyD Y n0 W d| _ | j�� I dH nd| _ | j�� I dH 0 dS )zNWait until communication is complete on the router and all endpoints are done.N)rr r] Zget_running_loopZ
create_future�BrokenPipeError�ConnectionResetErrorr[ �waitr4 r r r � communicate� s �zRouter.communicate)N) r r' r( r rT rP r rQ r* r] r^ r\ rp r r ra rc rM r/ r rO r3 rn rR ro rq rr r ZFuture� Exceptionrs rw r r r r r+ y s$
r+ )r] r
Zlogging�typingr r r Zjsonutilr r Zprotocolr r r
Z getLoggerr r" r r* rS rT r+ r r r r �<module> s
0
Mini Shell