File manager - Edit - /opt/saltstack/salt/lib/python3.10/site-packages/aiohttp/_websocket/__pycache__/writer.cpython-310.pyc
Back
o ;j�+ � @ s� U d Z ddlZddlZddlZddlmZ ddlmZmZm Z ddl m Z ddl m Z ddlmZmZ dd lmZ d d lmZmZmZmZmZmZmZmZ d d lm Z m Z d Z ee e d< dZ!G dd� d�Z"dS )z%WebSocket protocol versions 13 and 8.� N)�partial)�Final�Optional�Set� )� BaseProtocol)�ClientConnectionResetError)� ZLibBackend�ZLibCompressor)�DEFAULT_CHUNK_SIZE� )�MASK_LEN�MSG_SIZE�PACK_CLOSE_CODE� PACK_LEN1� PACK_LEN2� PACK_LEN3� PACK_RANDBITS�websocket_mask)�WS_DEFLATE_TRAILING� WSMsgType� �WS_CONTROL_FRAME_OPCODEi @ c @ s� e Zd ZdZdee�� ddd�dedej de de d ejd e d e d d fdd�Z d"de de d e d B d d fdd�Zde de de d d fdd�Zd e d B d efdd�Zde de d e d B d d fdd�Zde de d e d B d d fdd �Zd#d e de eB d d fd d!�Zd S )$�WebSocketWritera3 WebSocket writer. The writer is responsible for sending messages to the client. It is created by the protocol when a connection is established. The writer should avoid implementing any application logic and should only be concerned with the low-level details of the WebSocket protocol. Fr )�use_mask�limit�random�compress� notakeover�protocol� transportr r r r r �returnNc C sZ || _ || _|| _t|jd�| _|| _|| _d| _|| _ d| _ d| _ t � � | _t� | _dS )z Initialize a WebSocket writer.� Fr N)r r r r � getrandbits�get_random_bitsr r �_closing�_limit� _output_size� _compressobj�asyncio�Lock� _send_lock�set�_background_tasks)�selfr r r r r r r � r/ �M/opt/saltstack/salt/lib/python3.10/site-packages/aiohttp/_websocket/writer.py�__init__0 s zWebSocketWriter.__init__�message�opcodec � s. �| j r |tj@ s td��|s| jr|tkr | �||d� n^t|�tkrI| j 4 I dH � | � |||� W d �I dH n 1 I dH sCw Y n3t � � }| � |||�}tjdkrbt j||dd�}n|�|�}| j�|� |�| jj� t �|�I dH | j| jkr�d| _| jjr�| j�� I dH dS dS dS )z<Send a frame over the websocket with message as its payload.�!Cannot write to closing transportr N)� � T)�loopZ eager_start) r% r �CLOSEr r r �_write_websocket_frame�len� WEBSOCKET_MAX_SYNC_CHUNK_SIZEr+ �_send_compressed_frame_syncr) Zget_running_loop�#_send_compressed_frame_async_locked�sys� version_infoZTaskZ create_taskr- �addZadd_done_callback�discardZshieldr' r& r Z_pausedZ _drain_helper)r. r2 r3 r r7 �coroZ send_taskr/ r/ r0 � send_frameI s0 � (�� �zWebSocketWriter.send_frame�rsvc C s t |�}| j}|r dnd}d|B |B }|dk r!t|||B �}d} n|dk r0t|d|B |�}d} n t|d|B |�}d} | j�� rCtd ��|rft| � � �} t |�} t | | � | j� || | � | j t7 _ n|tkrw| j� |� | j� |� n| j� || � | j | | 7 _ d S ) a Write a websocket frame to the transport. This method handles frame header construction, masking, and writing to transport. It does not handle compression or flow control - those are the responsibility of the caller. � r �~ r i � � � r4 N)r: r r r r r Z is_closingr r r$ � bytearrayr �writer' r r ) r. r2 r3 rD Z msg_lengthr Zmask_bitZ first_byte�headerZ header_len�maskZ message_arrr/ r/ r0 r9 { s2 z&WebSocketWriter._write_websocket_framec C s8 |r t tj| td�S | jst tj| j td�| _| jS )zBGet or create a compressor object for the given compression level.)�levelZwbitsZmax_sync_chunk_size)r r � Z_BEST_SPEEDr; r( r )r. r r/ r/ r0 �_get_compressor� s ��z WebSocketWriter._get_compressorc C s@ | � |�}| �|�|�|�| jrtjntj� �t �|d� dS )z� Synchronous send for small compressed frames. This is used for small compressed payloads that compress synchronously in the event loop. Since there are no await points, this is inherently cancellation-safe. �@ N) rP r9 Z compress_sync�flushr r � Z_FULL_FLUSH� Z_SYNC_FLUSH� removesuffixr �r. r2 r3 r � compressobjr/ r/ r0 r< � s ����z+WebSocketWriter._send_compressed_frame_syncc � s� �| j 4 I dH �- | �|�}| �|�|�I dH |�| jr tjntj� � t �|d� W d �I dH dS 1 I dH s;w Y dS )a� Async send for large compressed frames with lock. Acquires the lock and compresses large payloads asynchronously in the executor. The lock is held for the entire operation to ensure the compressor state is not corrupted by concurrent sends. MUST be run shielded from cancellation. If cancelled after compression but before sending, the compressor state would be advanced but data not sent, corrupting subsequent frames. NrQ ) r+ rP r9 r rR r r rS rT rU r rV r/ r/ r0 r= � s � ����.�z3WebSocketWriter._send_compressed_frame_async_locked�� � �codec � sJ �t |t�r |�d�}z| jt|�| tjd�I dH W d| _dS d| _w )z<Close the websocket, sending the specified code and message.zutf-8)r3 NT)� isinstance�str�encoderC r r r8 r% )r. rZ r2 r/ r/ r0 �close� s � �zWebSocketWriter.close)N)rX rY )�__name__� __module__� __qualname__�__doc__r r �Randomr r) Z Transport�bool�intr1 �bytesrC r9 r rP r< r= r\ r^ r/ r/ r/ r0 r ' sn ������� � ����� �22��� � ��� � "r )#rb r) r r> � functoolsr �typingr r r Z base_protocolr Zclient_exceptionsr Zcompression_utilsr r Zhelpersr r r r r r r r r Zmodelsr r r re �__annotations__r; r r/ r/ r/ r0 �<module> s (
| ver. 1.4 |
Github
|
.
| PHP 8.2.30 | Generation time: 0 |
proxy
|
phpinfo
|
Settings