File: //opt/saltstack/salt/lib/python3.10/site-packages/aiohttp/__pycache__/web_ws.cpython-310.pyc
o
;j?k � @ sF U d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZmZ d dl m
Z
m Z m Z m
Z
mZmZ d dlZd dlmZ ddlmZ ddlmZ ddlmZ dd lmZ dd
lm Z m Z m Z m Z dd l m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z- dd l.m/Z/ dd
l0m1Z1 ddl2m3Z3 ddl4m5Z5m6Z6m7Z7 ddl8m9Z9m:Z: ddl;m<Z< ddl=m>Z> ej?dkr�d dl m@Z@ nd dlAm@Z@ ej?dkr�d dlmBZB nd dl mCZC eCeDeEddf ZBej?dkr�d dl ZFd dl mGZG n
d dlFZFd dlAmGZG d ZHd ZIe eJ eKd < e@d eLd e
d d!�ZMejNd d d d"�G d#d$� d$��ZOG d%d&� d&e>e eM �ZPdS )'� N)�Callable�Iterable)�Any�Final�Generic�Literal�cast�overload)� CIMultiDict� )�hdrs)�WebSocketDataQueue)�AbstractStreamWriter)�WSMessageTypeError)�DEFAULT_CHUNK_SIZE�calculate_timeout_when�
set_exception�
set_result)
�WS_CLOSED_MESSAGE�WS_CLOSING_MESSAGE�WS_KEY�WebSocketError�WebSocketReader�WebSocketWriter� WSCloseCode� WSMessage�WSMessageDecodeText�WSMessageNoDecodeText� WSMsgType�
ws_ext_gen� ws_ext_parse)�_INTERNAL_RECEIVE_TYPES)� ws_logger)� EofStream)�JSONBytesEncoder� JSONDecoder� JSONEncoder)�HTTPBadRequest�
HTTPException)� BaseRequest)�StreamResponse)� �
)�TypeVar�r+ � )�Buffer)�Unionzmemoryview[int]zmemoryview[bytes])r+ � )�Self)�WebSocketResponse�WebSocketReadyr � �THRESHOLD_CONNLOST_ACCESS� _DecodeTextT)�bound� covariant�default)Z auto_attribs�frozen�slotsc @ s0 e Zd ZU eed< edB ed< defdd�ZdS )r5 �okN�protocol�returnc C � | j S �N)r> ��self� rE �B/opt/saltstack/salt/lib/python3.10/site-packages/aiohttp/web_ws.py�__bool__R � zWebSocketReady.__bool__)�__name__�
__module__� __qualname__�bool�__annotations__�strrG rE rE rE rF r5 M s
r5 c s� e Zd ZU dZeed< dZedB ed< dZe dB ed< dZ
e dB ed< dZ eed< dZ
eed< d Zeed
< dZedB ed < dZejdB ed < dZeed
< dZejd dB ed< dZedB ed< dZeed< dZej dB ed< dZ ej dB ed< dZ ej d dB ed< dZ eed< dZ!ej"dB ed< dddddddde#dd�
d ed edB d ed ed edB d!e$e d"ed#ed$ed%ed&df� fd'd(�Z%d�d)d*�Z&d�d+d,�Z'd�d-d.�Z(d�d/d0�Z)d�d1d2�Z*d�d3d4�Z+d�d7d8�Z,d�d9d:�Z-d;ed&dfd<d=�Z.d�d>d?�Z/d@e0d&e1f� fdAdB� Z2d@e0d&e3dCedB eef fdDdE�Z4d@e0d&e3edB e f fdFdG�Z5d@e0dHedB dIe d&dfdJdK�Z6d@e0d&e7fdLdM�Z8e9d&efdNdO��Z:e9d&efdPdQ��Z;e9d&edB fdRdS��Z<e9d&edB fdTdU��Z=e9d&eeB fdVdW��Z>d�dXedYe?d&e?fdZd[�Z@d&edB fd\d]�ZAd�d_eBd&dfd`da�ZCd�d_eBd&dfdbdc�ZD d�d_eBddeEd"edB d&dfdedf�ZFd�dged"edB d&dfdhdi�ZGd�dgeBd"edB d&dfdjdk�ZH d�eIjJdl�dge?d"edB dmeKd&dfdndo�ZL d�dge?d"edB dmeMd&dfdpdq�ZNd�drds�ZOePjQd^ddt�dued_eBdved&efdwdx�ZRduePd&dfdydz�ZSduePd&dfd{d|�ZTd�d}d~�ZUeV d�dd�d edB d&eWfd�d���ZXeV d�dd�d edB d&eYfd�d���ZXeV d�dd�d edB d&eWeYB fd�d���ZX d�d edB d&eWeYB fd�d��ZXeVdd��dd�d edB d&efd�d���ZZeVdd��dd�d edB d&eBfd�d���ZZeVdd��dd�d edB d&eeBB fd�d���ZZdd��d edB d&eeBB fd�d��ZZdd��d edB d&eBfd�d��Z[eVd�dd��dd�d�e\d edB d&e?fd�d���Z]eVd�dd��dd�d�e^eBge?f d edB d&e?fd�d���Z]eVd�dd��dd�d�e\e^eBge?f B d edB d&e?fd�d���Z]eIj_dd��d�e\e^eBge?f B d edB d&e?fd�d��Z]dge`d&dfd�d��Zad&ebfd�d��ZceVdd�d&eWfd�d���ZdeVdd�d&eYfd�d���ZdeVdd�d&eWeYB fd�d���Zdd&eWeYB fd�d��Zdd;ed&dfd�d��Ze� ZfS )�r4 F�
_length_checkN� _ws_protocol�_writer�_reader�_closed�_closingr �
_conn_lost� _close_code�_loop�_waiting� _close_wait�
_exceptiong �_heartbeat_when�
_heartbeat_cb�_pong_response_cb�
_ping_task�_need_heartbeat_reset�_heartbeat_reset_handleg $@TrE i @ )
�timeout�receive_timeout� autoclose�autoping� heartbeat� protocols�compress� max_msg_size� writer_limit� decode_textra rb rc rd re rf rg rh ri rj r@ c
sl t � jdd� || _|| _|| _|| _|| _|| _|d ur"|d | _|| _ || _
| | _ |
| _ d| _
d | _d S )N�e )�statusg @F)�super�__init__�
_protocols�_timeout�_receive_timeout�
_autoclose� _autoping�
_heartbeat�_pong_heartbeat� _compress�
_max_msg_size�
_writer_limit� _decode_textr_ r` ) rD ra rb rc rd re rf rg rh ri rj �� __class__rE rF rn k s
zWebSocketResponse.__init__c C sd | � � | jd ur| j�� d | _d| _| jd ur!| j�� d | _| jd ur0| j�� d | _d S d S �NF)�_cancel_pong_response_cbr` �cancelr_ r\ r^ rC rE rE rF �_cancel_heartbeat� s
�z#WebSocketResponse._cancel_heartbeatc C s" | j d ur| j �� d | _ d S d S rB )r] r~ rC rE rE rF r} � s
�z*WebSocketResponse._cancel_pong_response_cbc C s> | j d u s| jr
d S | j}|d usJ �d| _|�| j�| _d S �NT)rt r_ rW Z call_soon�_flush_heartbeat_resetr` )rD �looprE rE rF �_on_data_received� s z#WebSocketResponse._on_data_receivedc C s" d | _ | jsd S | �� d| _d S r| )r` r_ �_reset_heartbeatrC rE rE rF r� � s
z(WebSocketResponse._flush_heartbeat_resetc C s� | j d u rd S | �� | j}|d ur|jjnd}| j}|d us!J �|�� }t|| j |�}|| _| j d u r>|�
|| j �| _ d S d S )Nr6 ) rt r} �_req� _protocol�_timeout_ceil_thresholdrW �timer r[ r\ �call_at�_send_heartbeat)rD �req�timeout_ceil_thresholdr� �now�whenrE rE rF r� � s
�
�z"WebSocketResponse._reset_heartbeatc C s� d | _ | jrd S | j}|d ur| jd usJ �|�� }|| jk r*|�| j| j�| _ d S | j}|d ur5|j j
nd}t || j |�}| �
� |�|| j�| _| j�dtj�}tjdkr`tj||dd�}n|�|�}|�� st|| _|�| j� d S | �|� d S )Nr6 � r. T)r� Z eager_start) r\ r_ rW rQ r� r[ r� r� r� r� r� r ru r} �_pong_not_receivedr] �
send_framer �PING�sys� version_info�asyncio�TaskZ create_taskZdoner^ Zadd_done_callback�_ping_task_done)rD r� r� r� r� r� �coroZ ping_taskrE rE rF r� � s2
��
z!WebSocketResponse._send_heartbeat�task�asyncio.Task[None]c C s( |� � s|�� }r| �|� d| _dS )z*Callback for when the ping task completes.N)Z cancelled� exception�_handle_ping_pong_exceptionr^ )rD r� �excrE rE rF r� � s
z!WebSocketResponse._ping_task_donec C s< | j d ur| j jd ur | �t�d| j� d��� d S d S d S )NzNo PONG received after z seconds)r� � transportr� r� � TimeoutErrorru rC rE rE rF r� � s ���z$WebSocketResponse._pong_not_receivedr� c C sb | j rdS | �� | �tj� || _| jr+| js-| jdur/| j� t
t j |d�d� dS dS dS dS )z5Handle exceptions raised during ping/pong processing.Nr )
rS � _set_closed�_set_code_close_transportr �ABNORMAL_CLOSURErZ rX rT rR � feed_datar r �ERROR�rD r� rE rE rF r� � s �z-WebSocketResponse._handle_ping_pong_exceptionc C s d| _ | �� dS )zdSet the connection to closed.
Cancel any heartbeat timers and set the closed flag.
TN)rS r rC rE rE rF r� s z WebSocketResponse._set_closed�requestc � s^ �| j d ur | j S | �|�\}}t� �|�I d H }|d us J �| �|||� |�� I d H |S rB )�_payload_writer�
_pre_startrm �prepare� _post_start�drain)rD r� r? �writerZpayload_writerrz rE rF r�
s �
zWebSocketResponse.preparezCIMultiDict[str]c C s� |j }d|�tjd��� �� kr td|�tj�� d�d��|jjs-td|�tj �� �d��d }tj
|v rYdd� |tj
� d �D �}|D ] }|| j v rN|} n qCt
�d
|j|| j � |�tjd�}|d vrltd |� �d��|�tj�}z|r~tt�|��d
kr�td|��d��W n tjy� td|��d�d �w t�t�|�� t �� � �� � }t tjdtj dtj |i�} d}
d} | j r�|�tj!�} t"| dd�\} }
| r�t#| d|
d�}
|
| tj!< |r�|| tj
< | || |
fS )NZ websocket� zNo WebSocket UPGRADE hdr: z$
Can "Upgrade" only to "WebSocket".)�textzNo CONNECTION upgrade hdr: c S s g | ]}t |�� ��qS rE )rN �strip)�.0�protorE rE rF �
<listcomp>* s
��z0WebSocketResponse._handshake.<locals>.<listcomp>�,u< %s: Client protocols %r don’t overlap server-known ones %r)Z13�8�7zUnsupported version: � zHandshake error: �upgradeFr T)�isserver)rg r� Zserver_notakeover)$�headers�getr ZUPGRADE�lowerr� r' Z_messager� Z
CONNECTIONZSEC_WEBSOCKET_PROTOCOL�splitro r" �warningZremoteZSEC_WEBSOCKET_VERSIONZSEC_WEBSOCKET_KEY�len�base64� b64decode�binascii�Error� b64encode�hashlib�sha1�encoder �digest�decoder
ZSEC_WEBSOCKET_ACCEPTrv ZSEC_WEBSOCKET_EXTENSIONSr r )rD r� r� r? Z
req_protocolsr� �version�keyZ
accept_valZresponse_headers�
notakeoverrg �
extensionsZ
enabledextrE rE rF �
_handshake s� ��
�
�� ������ �
�z WebSocketResponse._handshakec C st |j | _ | �|�\}}}}| �d� | j�|� | �� || _|jj}|d u r+t d��t
|j|||| j d�}||fS )Nrk zConnection lost)rg r� �limit) rW r� Z
set_statusr� �updateZ force_closerv r� r� �ConnectionResetErrorr rx )rD r� r� r? rg r� r� r� rE rE rF r� j s"
�z WebSocketResponse._pre_startr? r� c C s� || _ || _| �� | j}|d usJ �t|jt|d�| _t| j| j t
| j �| j d�}| j
d u r1d n| j}|jj||d� |j�d� d S )N)r� )rg rj )Zdata_received_cbF)rP rQ r� rW r
r� r rR r rw rL rv ry rt r� r? Z
set_parserZ
keep_alive)rD r� r? r� r� �parser�cbrE rE rF r� � s" ��z WebSocketResponse._post_startc C sP | j d ur td��z | �|�\}}}}W n ty" tdd � Y S w td|�S )NzAlready startedFT)rQ � RuntimeErrorr� r( r5 )rD r� �_r? rE rE rF � can_prepare� s
�
z WebSocketResponse.can_preparec C s
| j d uS rB )rQ rC rE rE rF �prepared� s
zWebSocketResponse.preparedc C rA rB )rS rC rE rE rF �closed� � zWebSocketResponse.closedc C rA rB )rV rC rE rE rF �
close_code� r� z WebSocketResponse.close_codec C rA rB )rP rC rE rE rF � ws_protocol� r� z WebSocketResponse.ws_protocolc C rA rB )rv rC rE rE rF rg � r� zWebSocketResponse.compress�namer; c C s0 | j }|du r |S |j}|du r|S |�||�S )z}Get optional transport information.
If no value associated with ``name`` is found, ``default`` is returned.
N)rQ r� �get_extra_info)rD r� r; r� r� rE rE rF r� � s z WebSocketResponse.get_extra_infoc C rA rB )rZ rC rE rE rF r� � rH zWebSocketResponse.exceptionr� �messagec � �. �| j d u r
td��| j �|tj�I d H d S �N�Call .prepare() first)rQ r� r� r r� �rD r� rE rE rF �ping� s �
zWebSocketResponse.pingc � r� r� )rQ r� r� r �PONGr� rE rE rF �pong� s �
zWebSocketResponse.pong�opcodec � s. �| j du r
td��| j �|||�I dH dS )z Send a frame over the websocket.Nr� )rQ r� r� )rD r� r� rg rE rE rF r� � s �
z WebSocketResponse.send_frame�datac � sR �| j d u r
td��t|t�stdt|� ��| j j|�d�tj |d�I d H d S )Nr� z data argument must be str (%r)zutf-8�rg )
rQ r� �
isinstancerN � TypeError�typer� r� r �TEXT�rD r� rg rE rE rF �send_str� s �
�zWebSocketResponse.send_strc � sR �| j d u r
td��t|tttf�stdt|� ��| j j|t j
|d�I d H d S )Nr� z#data argument must be byte-ish (%r)r� ) rQ r� r� �bytes� bytearray�
memoryviewr� r� r� r �BINARYr� rE rE rF �
send_bytes� s �
z WebSocketResponse.send_bytes)�dumpsr� c � s �| j ||�|d�I d H d S )Nr� )r� �rD r� rg r� rE rE rF � send_json� s � zWebSocketResponse.send_jsonc � s �| j ||�|d�I dH dS )z�Send JSON data using a bytes-returning encoder as a binary frame.
Use this when your JSON encoder (like orjson) returns bytes
instead of str, avoiding the encode/decode overhead.
r� N)r� r� rE rE rF �send_json_bytes� s � z!WebSocketResponse.send_json_bytesc � s6 �| j rd S | jd u rtd��| �� I d H d| _ d S )Nz Response has not been startedT)Z _eof_sentr� r� �closerC rE rE rF � write_eof s �
zWebSocketResponse.write_eof)�coder� r� r� r� c
� s� �| j du r
td��| jrdS | �� z | j �||�I dH | j}|dus'J �|r0|�� I dH W n- tjtj fyC | �
t j � � t
y^ } z|| _| �
t j � W Y d}~dS d}~ww | j}|dushJ �| jr�| jdusrJ �| jdu syJ �| j�� | _|�td� | jI dH | jr�| �� dS z;t�| j�4 I dH �# |�� I dH }|j t j u r�| �
|j � W d �I dH W dS q�1 I dH s�w Y W dS tjy� | �
t j � � t
y� } z|| _| �
t j � W Y d}~dS d}~ww )zClose websocket connection.Nr� FTr ) rQ r� rS r� r� r� r� r� ZCancelledErrorr� r� r r� � ExceptionrZ rR rX rW rY Z
create_futurer� r rT �_close_transport�
async_timeoutra rp �readr� r �CLOSEr� )rD r� r� r� r� r� �reader�msgrE rE rF r� sh �
� �� � � ��zWebSocketResponse.closec C s d| _ || _| �� dS )z6Set the close code and mark the connection as closing.TN)rT rV r �rD r� rE rE rF � _set_closing@ s z WebSocketResponse._set_closingc C s || _ | �� dS )z+Set the close code and close the transport.N)rV r� r rE rE rF r� F s z+WebSocketResponse._set_code_close_transportc C s. | j dur| j jdur| j j�� dS dS dS )zClose the transport.N)r� r� r� rC rE rE rF r� K s �z"WebSocketResponse._close_transportrD z WebSocketResponse[Literal[True]]c � � �d S rB rE �rD ra rE rE rF �receiveP � �zWebSocketResponse.receivez!WebSocketResponse[Literal[False]]c � r rB rE r rE rE rF r U r z WebSocketResponse[_DecodeText]c � r rB rE r rE rE rF r Z r c
� s� �| j d u r
td��|p| j} | jrtd��| jr,| jd7 _| jtkr*td��tS | jr1t S zTd| _z@|r`t
� |�4 I d H � | j � � I d H }W d �I d H n 1 I d H sZw Y n| j � � I d H }W d| _| j
rut| j
d � nd| _| j
r�t| j
d � w w W nq tjy� � ty� tj| _| �� I d H ttjd d � Y S ty� } z|j| _| j|jd�I d H ttj|d �W Y d }~S d }~w t y� } z || _ | � tj � | �� I d H ttj|d �W Y d }~S d }~ww |j t!vr�|S |j tj"u �r | � |j#� | j�s | j$�r | jdd�I d H |S |j tj%u �r-| � tj� |S |j tj&u �rB| j'�rB| �(|j#�I d H q|j tj)u �rN| j'�rNq|S ) Nr� Tz+Concurrent call to receive() is not allowedr z WebSocket connection is closed.F)r� )r� )*rR r� rq rX rS rU r7 r rT r r� ra r� rY r r� r� r# r �OKrV r� r r �CLOSEDr r� r� r� rZ r r� r� r! r� r� rr �CLOSINGr� rs r� r� )rD ra rb r r� rE rE rF r _ s� �
(��� ���� � ��
� ��ra c � r rB rE r rE rE rF � receive_str� r z WebSocketResponse.receive_strc � r rB rE r rE rE rF r � r c � r rB rE r rE rE rF r � r c � sD �| � |�I dH }|jtjur td|j� d|j�d���tt|j�S )zrReceive TEXT message.
Returns str when decode_text=True (default), bytes when decode_text=False.
N�Received message �:z is not WSMsgType.TEXT)r r� r r� r r� r rN �rD ra r rE rE rF r � s � � c � sD �| � |�I d H }|jtjur td|j� d|j�d���tt|j�S )Nr r
z is not WSMsgType.BINARY)r r� r r� r r� r r� r rE rE rF �
receive_bytes� s � � z WebSocketResponse.receive_bytes.)�loadsra r c � r rB rE �rD r ra rE rE rF � receive_json� � �z WebSocketResponse.receive_jsonc � r rB rE r rE rE rF r � r c � r rB rE r rE rE rF r � r c � s �| j |d�I d H }||�S )Nr
)r )rD r ra r� rE rE rF r � s �c � s
�t d��)Nz"Cannot call .write() for websocket)r� )rD r� rE rE rF �write� s �zWebSocketResponse.writec C s | S rB rE rC rE rE rF � __aiter__� s zWebSocketResponse.__aiter__c � r rB rE rC rE rE rF � __anext__� r zWebSocketResponse.__anext__c � r rB rE rC rE rE rF r � r c � r rB rE rC rE rE rF r � r c � s. �| � � I d H }|jtjtjtjfv rt�|S rB )r r� r r� r r �StopAsyncIteration)rD r rE rE rF r s
�c C s, d| _ | �� | jd urt| j|� d S d S r� )rT r rR r r� rE rE rF �_cancel s
�zWebSocketResponse._cancel)r@ N)r� r� r@ NrB )r� )grI rJ rK rO rL rM rP rN rQ r rR r
rS rT rU �intrV rW r� ZAbstractEventLooprX rY ZFuturerZ �
BaseExceptionr[ �floatr\ Z TimerHandler] r^ r� r_ r` �Handler r rn r r} r� r� r� r� r� r� r� r� r) r r� �tupler� r� r� r5 r� �propertyr� r� r� r� rg r r� r� r� r� r� r r� r� r� �jsonr� r& r� r$ r� r� r r r� r r� r� r r r r r r r% r r r r0 r r3 r r r �
__classcell__rE rE rz rF r4 V s
������� �
� � �
�
)
�
� T���
�
����
�
�����
� ����
�
����
�7
��� ���� ���� ���
�K��� ���� ���� � ���� ��� �� ����� �
���
� � �� �� �r4 )Qr� r� r� r� r r� �collections.abcr r �typingr r r r r r �attrZ multidictr
r� r Z_websocket.readerr
�abcr Zclient_exceptionsr Zhelpersr r r r �httpr r r r r r r r r r r r r Zhttp_websocketr! �logr" Zstreamsr# Ztypedefsr$ r% r& Zweb_exceptionsr' r( Z web_requestr) Z web_responser* r� r- Ztyping_extensionsr0 r1 r� r� r� r3 �__all__r7 r rM rL r8 �sr5 r4 rE rE rE rF �<module> sR
<