관리-도구
편집 파일: router.cpython-39.pyc
a ��g�' � @ 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