File: //usr/lib/python3.9/site-packages/cockpit/__pycache__/router.cpython-39.pyc
a
�/�h�' � @ 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� 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>1 � z&ExecutionQueue._wrap.<locals>.<lambda>)�setattr�__self__�__func__�__name__)r r r r r r - s zExecutionQueue._wrapc 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 �run3 s
zExecutionQueue.runN)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 � Zd
d � Z 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 @ s
zEndpoint.__init__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_endpointD s zEndpoint.freeze_endpointc C s. | j d usJ �t�d| � | j �� d | _ d S )NzThawing endpoint %s)r+ r r! r$ r2 r r r �
thaw_endpointI s
zEndpoint.thaw_endpointc C s t �d S r ��NotImplementedErrorr2 r r r �do_closeP s zEndpoint.do_close��channel�command�message�returnc C s t �d S r r5 )r r9 r: r; r r r r/ S s zEndpoint.do_channel_control�r9 �datar< c C s t �d S r r5 �r r9 r> r r r r0 V s zEndpoint.do_channel_data�
str | None��host�groupr; r< c C s t �d S r r5 )r rB rC r; r r r r1 Y s zEndpoint.do_killc C s | j �||� d S r )r* Zwrite_channel_datar? r r r �send_channel_data] s zEndpoint.send_channel_data�JsonObject | None)r9 r: �msg�kwargsr< c K sD | j j|f||d�|�� |dkr@| j j| �|� | j �|� d S )N)r9 r: �close)r* �
write_control� endpoints�remove� drop_channel)r r9 r: rF rG r r r �send_channel_control` s z Endpoint.send_channel_control)rF rG r< c K s | j j| |fi |�� d S r )r* �shutdown_endpoint)r rF rG r r r rN h s zEndpoint.shutdown_endpoint)N)r r% r& �__annotations__r+ r r r r3 r4 r7 �strr r/ �bytesr0 r1 rD r rM rN r r r r r( < s
� r( c @ s e Zd ZdS )� RoutingErrorN)r r% r& r r r r rR l s rR c @ s@ e Zd ZU ded< dd�dd�Zeee d�dd�Zd d
� Z d S ) � RoutingRuler) r* r, c C s
|| _ d S r r, r. r r r r s 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
Nr5 )r rU r r r �
apply_rulev s zRoutingRule.apply_rulec C s t �d S r r5 r2 r r r �shutdown� s zRoutingRule.shutdownN)
r r% r& rO r r r r( rV rW r r r r rS p s
rS c @ s e Zd ZU ee ed< eeef ed< ded< e j
ed< dZ e ed< ee d�d d
�Z
eed �d d
�Zedd�dd�Zedd�dd�Zd,ededd�dd�Zddedd�dd �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]]rJ � no_endpointsF�_eof)rX c C s: |D ]
}| |_ q|| _i | _i | _t�� | _| j�� d S r )r* rX rY rJ �asyncio�EventrZ �set)r rX �ruler r r r � s
zRouter.__init__rT 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)rX r r! rV rR )r rU r_ �endpointr r r � check_rules� s
zRouter.check_rulesN)r9 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)rY �popr r! �KeyError�error)r r9 r r r rL � s
zRouter.drop_channel)r` r< c C s t � | j|< | j�� d S r )r^ rJ rZ �clear)r r` r r r r- � s zRouter.add_endpointrE )r` rF rG 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 %srH �r: r9 � endpoints remaining: %rz close transport) rJ rb r r! rI rL rZ r^ r[ Z transportrH )r r` rF rG Zchannelsr9 r r r rN � s
zRouter.shutdown_endpointr@ rA c C s: t | j�}t�d||t|�� |D ]}|�|||� q"d S )Nz+do_kill(%s, %s). Considering %d endpoints.)r^ rJ r r! r" r1 )r rB rC r; rJ r` r r r r1 � s
zRouter.do_killr8 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=%sZpayloadrH rf )
rY r r r! �getra rR rI Z get_attrsrJ �addrc r/ )r r9 r: r; r` �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 )rY rc r0 )r r9 r> r` r r r �channel_data_received� s
z Router.channel_data_received)r< c C sF t �d| � t| j�}|D ] }|�� qd| _t �d| j� t| j�S )Nzeof_received(%r)Trg )r r! r^ rJ r7 r[ �bool)r rJ r` r r r � eof_received� s
zRouter.eof_received�_communication_done)rk r< c C s> | j s| �� | jd ur:|d u r.| j�d � n | j�|� d S r )r[ ro rp Z
set_resultZ
set_exception)r rk 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)rp r\ Zget_running_loopZ
create_future�BrokenPipeError�ConnectionResetErrorrZ �waitr2 r r r � communicate� s �zRouter.communicate)N) r r% r& r rS rO r rP r( r\ r] r[ rn r r ra rL r- r rN r1 rl rQ rm ro rp r ZFuture� Exceptionrq ru r r r r r) � s"
r) )r\ r
Zlogging�typingr r r Zjsonutilr r Zprotocolr r r
Z getLoggerr r r r( rR rS r) r r r r �<module> s
0