File: //proc/self/root/usr/lib/python3.9/site-packages/cockpit/__pycache__/router.cpython-39.pyc
a
���e� � @ s� d dl Z d dlZd dlmZmZmZ ddlmZmZ ddl 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� )� JsonDocument�
JsonObject)�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>0 � 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 �run2 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 e
e
e dd �d
d�Z e
e
dd�dd�Zee
ee
dd�dd�Ze
e
dd�dd�Ze
edd�dd�Zedd�dd �Zedd�d d �ZdS ) �Endpoint�Router�routerN� _Endpoint__endpoint_frozen_queue�r) c C s
|| _ d S r r+ �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_endpointB s zEndpoint.freeze_endpointc C s. | j d usJ �t�d| � | j �� d | _ d S )NzThawing endpoint %s)r* r r r# r0 r r r �
thaw_endpointG s
zEndpoint.thaw_endpoint��channel�command�message�returnc C s t �d S r ��NotImplementedError)r r4 r5 r6 r r r r- N s zEndpoint.do_channel_control�r4 �datar7 c C s t �d S r r8 �r r4 r; r r r r. Q s zEndpoint.do_channel_data��host�groupr7 c C s t �d S r r8 )r r> r? r r r r/ T s zEndpoint.do_killc C s | j �||� d S r )r) Zwrite_channel_datar< r r r �send_channel_dataX s zEndpoint.send_channel_data)r4 �kwargsr7 c K s | j j|fi |�� d S r )r) Z
write_message)r r4 rA r r r �send_channel_message[ s z Endpoint.send_channel_message)rA r7 c K s0 | j jf ||d�|�� |dkr,| j �|� d S )N)r4 r5 �close)r) �
write_control� drop_channel)r r4 r5 rA r r r �send_channel_control^ s z Endpoint.send_channel_controlc K s | j j| fi |�� d S r )r) �shutdown_endpoint)r rA r r r rG c s zEndpoint.shutdown_endpoint)r r$ r% �__annotations__r* r r
r r1 r2 �strr r- �bytesr. r/ r@ r rB rF rG r r r r r' ; s
r' c @ s e Zd Zdd� ZdS )� RoutingErrorc K s || _ || _d S r )�problemrA )r rL rA r r r r h s zRoutingError.__init__N)r r$ r% r r r r r rK g s rK 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 p s zRoutingRule.__init__��optionsr7 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
Nr8 )r rO r r r �
apply_rules s zRoutingRule.apply_rulec C s t �d S r r8 r0 r r r �shutdown~ s zRoutingRule.shutdownN)
r r$ r% rH r r r r' rP rQ r r r r rM m s
rM c @ s� e Zd ZU ee ed< eeef ed< dZ e
ed< ee d�dd�Z e ed�d d
�Z
ed d �d
d�Zed d�dd�Zee ee d d�dd�Zeee d d�dd�Zeed d�dd�Ze
d�d d �Zee d d �d d �Zd S )!r( �
routing_rules�
open_channelsF�_eof)rR c C s |D ]
}| |_ q|| _i | _d S r )r) rR rS )r rR �ruler r r r � s zRouter.__init__rN 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)rR r r rP rK )r rO rU �endpointr r r � check_rules� s
zRouter.check_rulesN)r4 r7 c C s` z | j �|� t�d|� W n" ty> t�d|| j � Y n0 | j s\| jr\| jr\| j�� d S )Nzrouter dropped channel %sz.trying to drop non-existent channel %s from %s) rS �popr r �KeyError�errorrT Z transportrC )r r4 r r r rE � s zRouter.drop_channel)rV r7 c sV � fdd�| j �� D �}t�d� ||� |D ]$}| jf d|d�|�� | �|� q,d S )Nc s h | ]\}}|� kr|�qS r r )�.0�key�value�rV r r � <setcomp>� r z+Router.shutdown_endpoint.<locals>.<setcomp>z'shutdown_endpoint(%s, %s) will close %srC �r5 r4 )rS �itemsr r rD rE )r rV rA Zchannelsr4 r r^ r rG � s
zRouter.shutdown_endpointr= c C s< t | j�� �}t�d||t|�� |D ]}|�||� q&d S )Nz+do_kill(%s, %s). Considering %d endpoints.)�setrS �valuesr r r! r/ )r r> r? Z endpointsrV r r r r/ � s zRouter.do_killr3 c
C s� |dkr�|| j v rtd��z"t�d||�d�� | �|�}W nB ty~ } z*| jf d||jd�|j �� W Y d }~d S d }~0 0 || 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=%sZpayloadrC )r5 r4 rL ) rS r r r �getrW rK rD rL rA rY r- )r r4 r5 r6 rV �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 )rS rY r. )r r4 r; rV r r r �channel_data_received� s
z Router.channel_data_received)r7 c C s< d| _ t| j�� �D ] \}}|�|dd|d�� qt| j�S )NTrC r` )rT �listrS ra r- �bool)r r4 rV r r r � eof_received� s zRouter.eof_received)rf r7 c C s | j D ] }|�� qd S r )rR rQ )r rf rU r r r � do_closed� s
zRouter.do_closed)r r$ r% r rM rH r rI r' rT rj r r rW rE rG r r/ rg rJ rh rk � Exceptionrl r r r r r( � s
r( )r Zlogging�typingr r r Zjsonutilr r Zprotocolr r Z getLoggerr r r
r' rm rK rM r( r r r r �<module> s
,