File manager - Edit - /opt/saltstack/salt/lib/python3.10/site-packages/aiohttp/_websocket/__pycache__/reader_py.cpython-310.pyc
Back
o ;jiN � @ sB U d Z ddlZddlZddlmZ ddlmZ ddlmZ ddl m Z ddl m Z m Z dd lmZ d d l mZmZmZ d d lmZmZmZmZmZmZ d d� eD �Zeee e d< d Z dZ dZ dZ!ej"Z#ej$Z%dZ&ej'j(Z)ej$j(Z*ej"j(Z+ej,j(Z-ej.j(Z/ej0j(Z1dZ2dZ3dZ4dZ5d Z6e7j8Z9e Z:G dd� d�Z;G dd� d�Z<dS )z0Reader for WebSocket protocol versions 13 and 8.� N)�deque)�Final� )� BaseProtocol)�ZLibDecompressor)� _EXC_SENTINEL� set_exception)� EofStream� )�UNPACK_CLOSE_CODE� UNPACK_LEN3�websocket_mask)�WS_DEFLATE_TRAILING�WebSocketError� WSCloseCode� WSMessage�WSMessageTextBytes� WSMsgTypec C s h | ]}t |��qS � )�int)�.0�ir r �P/opt/saltstack/salt/lib/python3.10/site-packages/aiohttp/_websocket/reader_py.py� <setcomp> s r �ALLOWED_CLOSE_CODES� � ���)T� )Fr c @ s� e Zd ZdZdededejddfdd�Zde fd d �Z de dB fd d �Z e fd e dej ddfdd�Zd dd�Zd dd�Z d dd�ZdeeB fdd �ZdeeB fd d �ZdS )!�WebSocketDataQueuezmWebSocketDataQueue resumes and pauses an underlying stream. It is a destination for WebSocket data. �protocol�limit�loop�returnNc C sN d| _ || _|d | _|| _d| _d | _d | _t� | _| jj | _ | jj | _ d S )Nr r F) �_size� _protocol�_limit�_loop�_eof�_waiter� _exceptionr �_buffer�popleft� _get_buffer�append� _put_buffer)�selfr r! r" r r r �__init__= s zWebSocketDataQueue.__init__c C � | j S �N)r( �r0 r r r �is_eofK � zWebSocketDataQueue.is_eofc C r2 r3 )r* r4 r r r � exceptionN r6 z WebSocketDataQueue.exception�exc� exc_causec C s4 d| _ || _| j }d urd | _t|||� d S d S �NT)r( r* r) r )r0 r8 r9 �waiterr r r r Q s �z WebSocketDataQueue.set_exceptionc C s2 | j }d u r d S d | _ |�� s|�d � d S d S r3 )r) ZdoneZ set_result)r0 r; r r r �_release_waiter\ s �z"WebSocketDataQueue._release_waiterc C s d| _ | �� d | _d S r: )r( r< r* r4 r r r �feed_eofc s zWebSocketDataQueue.feed_eof�data� WSMessage | WSMessageTextBytes�size� cython_intc C sN | j |7 _ | �||f� | �� | j | jkr#| jjs%| j�� d S d S d S r3 )r$ r/ r< r& r% �_reading_pausedZ pause_reading�r0 r> r@ r r r � feed_datah s �z WebSocketDataQueue.feed_datac � sb �| j s-| js-| jr J �| j�� | _z | jI d H W | �� S tjtjfy, d | _� w | �� S r3 ) r+ r( r) r' Z create_future�asyncioZCancelledError� TimeoutError�_read_from_bufferr4 r r r �readq s � ��zWebSocketDataQueue.readc C sV | j r!| �� \}}| j|8 _| j| jk r | jjr | j�� |S | jd ur)| j�t�r3 ) r+ r- r$ r& r% rB Zresume_readingr* r rC r r r rG | s z$WebSocketDataQueue._read_from_buffer�r# N)r> r? r@ rA r# N)�__name__� __module__� __qualname__�__doc__r r rE ZAbstractEventLoopr1 �boolr5 � BaseExceptionr7 r �builtinsr r< r= rD r r rH rG r r r r r 7 s@ ��� ���� � �� � r c @ s� e Zd Z ddededededdf dd �Zdd d �Zd ee B e B de eef fd d�Z dedee B dee B dee B ddf dd�Zd eddfdd�ZdS )�WebSocketReaderT�queue� max_msg_size�compress� decode_textr# Nc C sx || _ || _|| _d | _t� | _t| _t| _ d| _ t| _ g | _ d| _ d| _d| _d | _d| _d| _t| _d | _|| _d S )NFr r )rR � _max_msg_size� _decode_text�_exc� bytearray�_partial� READ_HEADER�_state�OP_CODE_NOT_SET�_opcode� _frame_fin� _frame_opcode�_payload_fragments�_frame_payload_len�_tail� _has_mask� _frame_mask�_payload_bytes_to_read�_payload_len_flag�COMPRESSED_NOT_SET� _compressed�_decompressobj� _compress)r0 rR rS rT rU r r r r1 � s& zWebSocketReader.__init__c C s | j �� d S r3 )rR r= r4 r r r r= � s zWebSocketReader.feed_eofr> c C sr t |�tur t|�}| jd urd|fS z| �|� W tS ty8 } z|| _t| j|� tW Y d }~S d }~ww r: ) �type�bytesrX � _feed_data� Exceptionr rR �EMPTY_FRAME_ERROR� EMPTY_FRAME)r0 r> r8 r r r rD � s � ��zWebSocketReader.feed_data�fin�opcode�payload� compressedc C s0 |t tthv r�|tkr| jtkrttjd��|s(|tkr || _| j|7 _d S t | j�}|tkr8| j}t| _n |rCttjd|����|rP| j| }| j� � n|}|r�| j s]t dd�| _ | j � |t | jrk| jd n| j�}| jr�t|�| jkr�ttjd| j� ���n t|�tu r�|}nt|�}|t kr�| jr�z|�d�}W n ty� } zttjd�| �d } ~ ww | j�ttt |d f�t|�� d S | j�tt t |d f�t|�� d S | j�ttt |d f�t|�� d S |t k�r]t|�d k�r9t |d d � �d } | d k �r| t!v�rttjd | � ���z |d d � �d�} W n t�y. } zttjd�| �d } ~ ww ttt"j#| | f�} n|�rKttjd|� d|� d|����ttt"j#d d f�} | j�| d � d S |t$k�rvttt"j%|d f�} | j�| t|�� d S |t&k�r�ttt"j'|d f�} | j�| t|�� d S ttjd|����)Nz*Continuation frame for non started messagez8The opcode in non-fin frame is expected to be zero, got T)Zsuppress_deflate_headerr z(Decompressed message exceeds size limit zutf-8zInvalid UTF-8 text message� r r i� zInvalid close code: zInvalid close frame: � �Unexpected opcode=)(� OP_CODE_TEXT�OP_CODE_BINARY�OP_CODE_CONTINUATIONr^ r] r r �PROTOCOL_ERRORrZ rN �clearrj r Zdecompress_syncr rV �len�MESSAGE_TOO_BIGrl rm rW �decode�UnicodeDecodeErrorZ INVALID_TEXTrR rD � TUPLE_NEWr �WS_MSG_TYPE_TEXTr �WS_MSG_TYPE_BINARY� OP_CODE_CLOSEr r r �CLOSE� OP_CODE_PING�PING� OP_CODE_PONG�PONG) r0 rr rs rt ru Z has_partialZassembled_payloadZpayload_merged�textr8 Z close_codeZ close_message�msgr r r � _handle_frame� s� � �� �� �� ���� � ��� ������ �z WebSocketReader._handle_framec C s@ | j r | j | d}| _ d}t|�}|} | jtkr�|| dk r"�n�|| }||d }|d7 }|d? d@ }|d? d@ }|d? d@ } |d ? d@ } |d @ } | sU| sU|r[| js[ttjd ��| tt t t t t hvrnttjd | ����| dkr||dkr|ttjd ��|d? d@ } |d@ } | dkr�| dkr�ttjd��| js�| jtkr�|r�tnt| _n|r�ttjd ��t|�| _| | _t| �| _| | _t| _| jtk�r8| j}|dkr�|| dk rԐn9|| }||d }|d7 }|d> |B | _n |dk�r|| dk r��nt||�d | _|d7 }n|| _| j�r/| jt t thv �r/| jt| j� }|| jk�r/ttj d|� d| j� ���| j�r5t nt | _| jt k�rV|| d k �rFn�|||d � | _ |d 7 }t | _| jt k�r || }| j|k�rp|}| j|8 _n|| j }d| _| j }| j || 7 _ |}|}| jdk�r�| j!�"|||� � nt|�r�| j!�"|||� � | j�r�| j du�s�J �t#d�$| j!��}t%| j |� |}nd�$| j!�}| j!�&� n,| j�r�| j du�s�J �|||� }t'|�t#u�r�t#|�}t%| j |� |}n|||� }| �(| j| j|| j� d| _ t| _q||k �r|||� | _ dS d| _ dS )z&Return the next frame from the socket.r r Tr r � � � r � z*Received frame with non-zero reserved bitsrx z!Received fragmented control frame� �} z5Control frame payload cannot be larger than 125 bytes�~ � z Message size z exceeds limit N))rc r~ r\ r[ rk r r r| r{ ry rz r� r� r� r_ ri rh �COMPRESSED_TRUE�COMPRESSED_FALSErN r` rd rg �READ_PAYLOAD_LENGTHrf r rV rZ r �READ_PAYLOAD_MASK� READ_PAYLOADre rb ra r. rY �joinr r} rl r� )r0 r> Z start_posZdata_lenZ data_cstrZ first_byteZ second_byterr Zrsv1Zrsv2Zrsv3rs Zhas_mask�lengthZlen_flagZprojected_sizeZ chunk_lenZ f_end_posZ had_fragmentsZ f_start_posZpayload_bytearrayrt r r r rn F s ���� �� � �� � �� &/zWebSocketReader._feed_data)TTrI )rJ rK rL r r rN r1 r= rm rY � memoryview�tuplerD rA r� rn r r r r rQ � s: ����� � "���� � rQ )=rM rE rP � collectionsr �typingr Z base_protocolr Zcompression_utilsr Zhelpersr r Zstreamsr r r r Zmodelsr r r r r r r �setr �__annotations__r[ r� r� r� ZBINARYr� ZTEXTr� r] Z CONTINUATION�valuer{ ry rz r� r� r� r� r� r� rp rq rh r� r� r� �__new__r� rA r rQ r r r r �<module> sD Q
| ver. 1.4 |
Github
|
.
| PHP 8.2.30 | Generation time: 0 |
proxy
|
phpinfo
|
Settings