Google
Edit File: protocol.cpython-39.opt-1.pyc
a ��j�# � @ s� d dl Z d dlZd dlZd dlZddlmZmZmZmZm Z m Z m Z m Z e� e�ZG dd� de�ZG dd� de�ZG dd � d e j�ZG d d � d e�ZdS ) � N� )� JsonError� JsonObject� JsonValue� create_object�get_int�get_str�get_str_or_none� typecheckedc sF e Zd ZU dZeed< d ededd�� fdd� Zed�d d �Z � Z S ) �CockpitProblema� A type of exception that carries a problem code and a message. Depending on the scope, this is used to handle shutting down: - an individual channel (sends problem code in the close message) - peer connections (sends problem code in close message for each open channel) - the main stdio interaction with the bridge It is usually thrown in response to some violation of expected protocol when parsing messages, connecting to a peer, or opening a channel. �attrsN�JsonObject | None)�problem�msg�kwargs�returnc s. ||d< t ||�| _t� �t| jd|�� d S )Nr �message)r r �super�__init__r )�selfr r r �� __class__� �4/usr/lib/python3.9/site-packages/cockpit/protocol.pyr s zCockpitProblem.__init__�r c C sD | j d dkr:| jd ur:t| j t�| jj| j| jj�d�S | j S d S )Nr zinternal-error)�cause)r � __cause__�dict� traceback�format_exceptionr � __traceback__�r r r r � get_attrs# s �zCockpitProblem.get_attrs)N) �__name__� __module__� __qualname__�__doc__r �__annotations__�strr r r" � __classcell__r r r r r s r c s& e Zd Zdeed�� fdd� Z� ZS )�CockpitProtocolError�protocol-error)r r c s t � j||d� d S )N)r )r r )r r r r r r r - s z CockpitProtocolError.__init__)r+ )r# r$ r% r( r r) r r r r r* , s r* c @ s8 e Zd ZU dZdZded< dZdZeed< dZ ded < dd �d d �Z d dd�dd�Z e e dd�dd�Ze e e dd�dd�Ze edd�dd�Zedd�dd �Zedd �d d �Zeed �d d!�Zejdd"�d#d$�Zd dd�d%d&�Zd4d dd�d'd(�Ze edd)�d*d+�Zd5d,e dd-�d.d/�Z edd �d0d1�Z ed �d2d3�Z dS )6�CockpitProtocolz�A naive implementation of the Cockpit frame protocol We need to use this because Python's SelectorEventLoop doesn't supported buffered protocols. Nzasyncio.Transport | None� transport� F�_closedzasyncio.Future[None] | None�_communication_doner c C s d S �Nr r! r r r �do_ready< s zCockpitProtocol.do_readyzException | None)�excr c C s d S r1 r �r r3 r r r � do_closed? s zCockpitProtocol.do_closed��commandr r c C s t �d S r1 ��NotImplementedError�r r7 r r r r �transport_control_receivedB s z*CockpitProtocol.transport_control_received)�channelr7 r r c C s t �d S r1 r8 )r r< r7 r r r r �channel_control_receivedE s z(CockpitProtocol.channel_control_received)r< �datar c C s t �d S r1 r8 )r r< r> r r r �channel_data_receivedH s z%CockpitProtocol.channel_data_received)�framer c C sP |� d�\}}}|dkrB|�d�}t�dt|�|� | �||� n | �|� d S )N� r. �asciiz.data received: %d bytes of data for channel %s)� partition�decode�logger�debug�lenr? �control_received)r r@ �header�_r> r< r r r �frame_receivedK s z CockpitProtocol.frame_received)r> r c C s� zft t�|�t�}t|d�}t|dd �}|d urLt�d|� | �|||� nt�d|� | �||� W n: tj t fy� } z t d|���|�W Y d }~n d }~0 0 d S )Nr7 r< zchannel control received %sz transport control received %szcontrol message: ) r �json�loadsr r rE rF r= r; ZJSONDecodeErrorr r* )r r> r r7 r< r3 r r r rH V s z CockpitProtocol.control_receivedc C s� z|� d�}W nR ty` } z:t|�dk rBt|�d W Y d}~S td�|�W Y d}~n d}~0 0 zt|d|� �}W n. ty� } ztd�|�W Y d}~n d}~0 0 |d }|| }|t|�kr�t|�| S | �|||� � |S )z�Consumes a single frame from view. Returns positive if a number of bytes were consumed, or negative if no work can be done because of a given number of bytes missing. rA � Nzsize line is too longz frame size is not an integerr )�index� ValueErrorrG r* �intrK )r r> �newliner3 �length�start�endr r r �consume_one_framef s z!CockpitProtocol.consume_one_frame)r- r c C s6 t �d|� || _| �� | jr2t �d� |�� d S )Nzconnection_made(%s)z; but the protocol already was closed, so closing transport)rE rF r- r2 r/ �close)r r- r r r �connection_made� s z CockpitProtocol.connection_madec C s t �d� d | _| �|� d S )N�connection_lost)rE rF r- rW r4 r r r rY � s z CockpitProtocol.connection_lostc C s. | j r d S d| _ | jr | j�� | �|� d S )NT)r/ r- rW r5 r4 r r r rW � s zCockpitProtocol.close)r< �payloadr c C sb t |d �t |� }|� d|� d��d�}| jdurTt�d| j� | j�|| � n t�d� dS )z0Send a given payload (bytes) on channel (string)� rB Nzwriting to transport %sz cannot write to closed transport)rG �encoder- rE rF �write)r r< rZ Z frame_lengthrI r r r �write_channel_data� s z"CockpitProtocol.write_channel_datar )r r r c K s: t �d||� tjt||�dd�d }| �d|�� � dS )zCWrite a control message. See jsonutil.create_object() for details.z sending control message %r %r� )�indentr[ � N)rE rF rL �dumpsr r^ r\ )r r r Zprettyr r r � write_control� s z CockpitProtocol.write_controlc C sx zD| j |7 _ | j rB| �| j �}|dkr0W d S | j |d � | _ qW n. tyr } z| �|� W Y d }~n d }~0 0 d S )Nr )�bufferrV r* rW )r r> �resultr3 r r r � data_received� s z CockpitProtocol.data_receivedc C s dS )NFr r! r r r � eof_received� s z CockpitProtocol.eof_received)N)N) r# r$ r% r& r- r' rd r/ �boolr0 r2 r5 r( r r; r= �bytesr? rK rH rQ rV �asyncioZ BaseTransportrX rY rW r^ r rc rf rg r r r r r, 1 s( r, c @ s� e Zd ZU dZded< dZded< dZdd�dd �Zedd �d d �Z ddedd �dd�Z e edd�dd�Z dd�dd�Z d e deed�dd�Zd e dee d�dd�Zedd �dd �ZdS ) �CockpitProtocolServerNz str | None� init_hostz,dict[str, asyncio.Future[JsonObject]] | None�authorizationsr r c C s t �d S r1 r8 r! r r r � do_send_init� s z"CockpitProtocolServer.do_send_init)r r c C s d S r1 r )r r r r r �do_init� s z CockpitProtocolServer.do_init)�host�groupr r c C s t �d S r1 r8 )r rp rq r r r r �do_kill� s z CockpitProtocolServer.do_killr6 c C s� |dkr6t |d�dkr td��t|d�| _| �|� nL|dkr^| �t|dd �t|dd �|� n$|dkrr| �|� ntd |� d ���d S ) N�init�versionr zincorrect version numberrp �killrq � authorizezunexpected control message z received)r r* r rl ro rr r � do_authorizer: r r r r; � s z0CockpitProtocolServer.transport_control_receivedc C s | � � d S r1 )rn r! r r r r2 � s z CockpitProtocolServer.do_readyz int | None)� challenge�timeoutr r c � s� | j d u ri | _ t| j�}| jd7 _t�� �� }zB|| j |< | jdd||d�|�� t�||�I d H W | j �|� S | j �|� 0 d S )Nr rv )r7 rx �cookie)N) rm r( � next_auth_idrj Zget_running_loopZ create_futurerc �wait_for�pop)r rx ry r rz Zfuturer r r � request_authorization_object� s �z2CockpitProtocolServer.request_authorization_objectc � s t | j||fi |��I d H d�S )NZresponse)r r~ )r rx ry r r r r �request_authorization� s z+CockpitProtocolServer.request_authorizationc C s@ t |d�}| jd u s || jvr,t�d� d S | j| �|� d S )Nrz z no matching authorize request)r rm rE ZwarningZ set_result)r r rz r r r rw � s z"CockpitProtocolServer.do_authorize)N)N)r# r$ r% rl r' rm r{ rn r ro rr r( r; r2 r r~ r rw r r r r rk � s& �� �� rk )rj rL Zloggingr Zjsonutilr r r r r r r r Z getLoggerr# rE � Exceptionr r* �Protocolr, rk r r r r �<module> s (