File: //lib/python3.9/site-packages/cockpit/__pycache__/protocol.cpython-39.pyc
a
���e~* � @ s� d dl Z d dlZd dlZd dlZd dlmZmZmZ d dlm Z ddl
m Z m Z m
Z
mZ e�e�ZG dd� de�ZG dd � d e�ZG d
d � d e j�ZG d d
� d
e�ZdS )� N)�ClassVar�Dict�Optional)�systemd_ctypes� )� JsonError�
JsonObject�get_str� typecheckedc s& e Zd ZdZed�� fdd� Z� ZS )�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.
)�problemc s& t � �|�d�p|� || _|| _d S )N�message)�super�__init__�getr �kwargs)�selfr r �� __class__� �4/usr/lib/python3.9/site-packages/cockpit/protocol.pyr + s zCockpitProblem.__init__)�__name__�
__module__� __qualname__�__doc__�strr �
__classcell__r r r r r s r c s e Zd Zd� fdd� Z� ZS )�CockpitProtocolError�protocol-errorc s t � j||d� d S )N)r
)r r )r r
r r r r r 2 s z CockpitProtocolError.__init__)r )r r r r r r r r r r 1 s r c @ sF e Zd ZU dZejdd�Zeej e d< dZ
e e j
e d< dZdZee d < dZe e j e d
< dd �d d
�Ze e dd�dd�Zeedd�dd�Zeeedd�dd�Zeedd�dd�Zedd�dd �Z ed �d d �Z d d!� Z d"d#� Z d$d%� Z d4e e dd�d&d'�Z!d(d)� Z"d*d+� Z#d,d-� Z$d.d/� Z%e e d �d0d1�Z&dd �d2d3�Z'dS )5�CockpitProtocolz�A naive implementation of the Cockpit frame protocol
We need to use this because Python's SelectorEventLoop doesn't supported
buffered protocols.
� )�indent� json_encoderN� transport� F�_closed�_communication_done)�returnc C s d S �Nr �r r r r �do_readyB s zCockpitProtocol.do_ready)�excr' c C s d S r( r �r r+ r r r � do_closedE s zCockpitProtocol.do_closed)�commandr
r' c C s t �d S r( ��NotImplementedError)r r. r
r r r �transport_control_receivedH s z*CockpitProtocol.transport_control_received)�channelr. r
r' c C s t �d S r( r/ )r r2 r. r
r r r �channel_control_receivedK s z(CockpitProtocol.channel_control_received)r2 �datar' c C s t �d S r( r/ )r r2 r4 r r r �channel_data_receivedN 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�lenr5 �control_received)r r6 �header�_r4 r2 r r r �frame_receivedQ s
z CockpitProtocol.frame_received)r4 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 )Nr. r2 zchannel control received %sz transport control received %szcontrol message: ) r
�json�loads�dictr r; r< r3 r1 ZJSONDecodeErrorr r )r r4 r
r. r2 r+ r r r r> \ s
z CockpitProtocol.control_receivedc
C s |sdS t |�}t |dd� �}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.
r N�
r7 zsize line is too longz frame size is not an integerr )�bytes�index�
ValueErrorr= r �intrA )r �viewr? �newliner+ �length�start�endr r r �consume_one_framel s( z!CockpitProtocol.consume_one_framec C s6 t �d|� || _| �� | jr2t �d� |�� d S )Nzconnection_made(%s)z; but the protocol already was closed, so closing transport)r; r< r# r* r% �close)r r# r r r �connection_made� s
z CockpitProtocol.connection_madec C s, t �d� | jd usJ �d | _| �|� d S )N�connection_lost)r; r< r# rP r, r r r rR � s
z CockpitProtocol.connection_lostc C sZ | j r
d S d| _ | jr | j�� | �|� | jd urV|d u rJ| j�d � n | j�|� d S )NT)r% r# rP r- r& �
set_resultZ
set_exceptionr, r r r rP � s
zCockpitProtocol.closec C sb t |d �t |� }|� d|� d��d�}| jdurTt�d| j� | j�|| � n
t�d� dS )z0Send a given payload (bytes) on channel (string)�
r8 Nzwriting to transport %sz cannot write to closed transport)r= �encoder# r; r< �write)r r2 ZpayloadZ frame_lengthr? r r r �write_channel_data� s
z"CockpitProtocol.write_channel_datac K sd t |�D ]&}d|v r|| ||�dd�< ||= qt�d||� tj�|�d }| �||�d�� dS )z�Format kwargs as a JSON blob and send as a message
Any kwargs with '_' in their names will be converted to '-'
r@ �-zsending message %s %srT zutf-8N)�list�replacer; r< r r" rU rW )r Z_channelr �nameZprettyr r r �
write_message� s z CockpitProtocol.write_messagec K s | j di |�� d S )N� )r] )r\ )r r r r r �
write_control� s z CockpitProtocol.write_controlc
C sr z>| j |7 _ | �| j �}|dkr*W d S | j |d � | _ qW n. tyl } z| �|� W Y d }~n
d }~0 0 d S )Nr )�bufferrO r rP )r r4 �resultr+ r r r �
data_received� s z CockpitProtocol.data_receivedc C s dS )NFr r) r r r � eof_received� s z CockpitProtocol.eof_receivedc � s2 | j du sJ �t�� �� | _ | j I dH d| _ dS )z6Wait until communication is complete on this protocol.N)r& �asyncio�get_running_loop�
create_futurer) r r r � communicate� s zCockpitProtocol.communicate)N)(r r r r r Z JSONEncoderr" r rB �__annotations__r# r rc Z Transportr_ r% �boolr& �Futurer* � Exceptionr- r r r1 r3 rF r5 rA r> rO rQ rR rP rW r\ r^ ra rb rf r r r r r 6 s.
,
r c @ s� e Zd ZU dZee ed< dZeeee j
f ed< dd� Z dd� Z ee ee dd�d d
�Z
d d � Zd
d� Zdeee eed�dd�Zedd�dd�ZdS )�CockpitProtocolServerN� init_host�authorizationsc C s t �d S r( r/ r) r r r � do_send_init� s z"CockpitProtocolServer.do_send_initc C s d S r( r )r r
r r r �do_init� s z CockpitProtocolServer.do_init)�host�groupr' c C s t �d S r( r/ )r rp rq r r r �do_kill� s z CockpitProtocolServer.do_killc
C s |dkr�z t |d �dkr$tdd��W n^ tyV } ztdd�|�W Y d }~n8d }~0 ty� } ztdd�|�W Y d }~n
d }~0 0 z|d | _W n0 ty� } ztd d�|�W Y d }~n
d }~0 0 | �|� nH|d
kr�| �|�d�|�d �� n&|d k�r
| �|� ntd
|� d���d S )N�init�versionr zincorrect version numberr zversion field is missingzversion field is not an intrp zmissing host field�killrq � authorizezunexpected control message z received) rI r �KeyErrorrH rl ro rr r � do_authorize)r r. r
r+ r r r r1 � s$ ""
z0CockpitProtocolServer.transport_control_receivedc C s | � � d S r( )rn r) r r r r* s z CockpitProtocolServer.do_ready)� challenge�timeoutr r' c � s~ | j d u ri | _ tt�� �}t�� �� }zB|| j |< | jf d||d�|�� t�||�I d H W | j � |� S | j � |� 0 d S )Nrv )r. ry �cookie)
rm r �uuidZuuid4rc rd re r^ �wait_for�pop)r ry rz r r{ Zfuturer r r �request_authorization s
�z+CockpitProtocolServer.request_authorization)r
r' c C sf |� d�}|� d�}t|t�r(t|t�s0td��| jd u sD|| jvrRt�d� d S | j| �|� d S )Nr{ �responsezinvalid authorize responsez no matching authorize request)r �
isinstancer r rm r; ZwarningrS )r r
r{ r� r r r rx s
z"CockpitProtocolServer.do_authorize)N)r r r rl r r rg rm r rc ri rn ro rr r1 r* rI �objectr r rx r r r r rk � s
rk )rc rB Zloggingr| �typingr r r Zcockpit._vendorr Zjsonutilr r r r
Z getLoggerr r; rj r r �Protocolr rk r r r r �<module> s
4