Edit File: http_websocket.cpython-37.pyc
B �k�ac � @ s U d Z ddlZddlZddlZddlZddlZddlZddlZddlm Z ddl m Z ddl m Z mZmZmZmZmZmZmZmZ ddlmZ ddlmZ dd lmZ dd l m Z d Z G d d � d e �Z dd� e D �Z e ee! e"d< G dd� de �Z#dZ$e e% e"d< e d�j&Z'e d�j&Z(e d�j)Z*e d�j+Z,e d�j+Z-e d�j+Z.e d�j+Z/dZ0e e! e"d< d Z1e e! e"d < e�2d d d d!g�Z3G d"d#� d#e3�Z4e4e#j5dd�Z6e4e#j7dd�Z8G d$d%� d%e9�Z:G d&d'� d'e9�Z;ej<Z=e e> e"d(< d)d*� e?d+�D �Z@e ee% e"d,< e%eAdd-�d.d/�ZBe�reBZCn0ydd0lDmEZE eEZCW n eFk �rH eBZCY nX e%ddd1d1g�ZGe e% e"d2< e�Hd3�ZIe ee> e"d4< e�Hd5�ZJe ee> e"d6< dEee> eKee!eKf d8�d9d:�ZLdFe!eKeKe>d<�d=d>�ZMG d?d@� d@e �ZNG dAdB� dB�ZOG dCdD� dD�ZPdS )Gz%WebSocket protocol versions 13 and 8.� N)�IntEnum)�Struct) �Any�Callable�List�Optional�Pattern�Set�Tuple�Union�cast� )� BaseProtocol)� NO_EXTENSIONS)� DataQueue)�Final) �WS_CLOSED_MESSAGE�WS_CLOSING_MESSAGE�WS_KEY�WebSocketReader�WebSocketWriter� WSMessage�WebSocketError� WSMsgType� WSCloseCodec @ s@ e Zd ZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdS )r i� i� i� i� i� i� i� i� i� i� i� i� i� N)�__name__� __module__� __qualname__ZOKZ GOING_AWAY�PROTOCOL_ERRORZUNSUPPORTED_DATAZABNORMAL_CLOSURE� INVALID_TEXTZPOLICY_VIOLATION�MESSAGE_TOO_BIGZMANDATORY_EXTENSIONZINTERNAL_ERRORZSERVICE_RESTARTZTRY_AGAIN_LATERZ BAD_GATEWAY� r! r! �I/opt/alt/python37/lib64/python3.7/site-packages/aiohttp/http_websocket.pyr s r c C s h | ] }t |��qS r! )�int)�.0�ir! r! r" � <setcomp>0 s r& �ALLOWED_CLOSE_CODESc @ sP e Zd ZdZdZdZdZdZdZdZ dZ d Z eZ eZ eZeZeZe Ze Ze Zd S ) r r r � � � � � i i N)r r r � CONTINUATION�TEXT�BINARY�PING�PONG�CLOSE�CLOSING�CLOSED�ERROR�text�binary�ping�pong�close�closing�closed�errorr! r! r! r" r 3 s" r s$ 258EAFA5-E914-47DA-95CA-C5AB0DC85B11r z!Hz!Qz!BBz!BBHz!BBQi @ �MSG_SIZEi � DEFAULT_LIMIT�_WSMessageBase�type�data�extrac @ s. e Zd Zejd�eegef ed�dd�ZdS )r )�loads)rD �returnc C s || j �S )zAReturn parsed JSON data. .. versionadded:: 0.22 )rB )�selfrD r! r! r" �json] s zWSMessage.jsonN)r r r rG rD r r r! r! r! r" r \ s r c s8 e Zd ZdZeedd�� fdd� Zed�dd�Z� ZS ) r z WebSocket protocol parser error.N)�code�messagerE c s || _ t� �||� d S )N)rH �super�__init__)rF rH rI )� __class__r! r" rK l s zWebSocketError.__init__)rE c C s t t| jd �S )Nr )r �str�args)rF r! r! r" �__str__p s zWebSocketError.__str__) r r r �__doc__r# rM rK rO � __classcell__r! r! )rL r" r i s r c @ s e Zd ZdZdS )�WSHandshakeErrorz#WebSocket protocol handshake error.N)r r r rP r! r! r! r" rR t s rR �native_byteorderc s& g | ] � t � fd d�td�D ���qS )c 3 s | ]}|� A V qd S )Nr! )r$ �a)�br! r" � <genexpr>| s z<listcomp>.<genexpr>r, )�bytes�range)r$ r! )rU r" � <listcomp>| s rY r, � _XOR_TABLE)�maskrB rE c C s� t |t�st|��t| �dks&t| ��|r�dd� | D �\}}}}|ddd� �|�|ddd�< |ddd� �|�|ddd�< |ddd� �|�|ddd�< |ddd� �|�|ddd�< dS )aw Websocket masking function. `mask` is a `bytes` object of length 4; `data` is a `bytearray` object of any length. The contents of `data` are masked with `mask`, as specified in section 5.3 of RFC 6455. Note that this function mutates the `data` argument. This pure-python implementation may be replaced by an optimized version when available. � c s s | ]}t | V qd S )N)rZ )r$ �nr! r! r" rV � s z)_websocket_mask_python.<locals>.<genexpr>Nr r( � )� isinstance� bytearray�AssertionError�len� translate)r[ rB rT rU �c�dr! r! r" �_websocket_mask_python s rf )�_websocket_mask_cython� �_WS_DEFLATE_TRAILINGz�^(?:;\s*(?:(server_no_context_takeover)|(client_no_context_takeover)|(server_max_window_bits(?:=(\d+))?)|(client_max_window_bits(?:=(\d+))?)))*$� _WS_EXT_REzpermessage-deflate([^,]+)?�_WS_EXT_RE_SPLITF)�extstr�isserverrE c C s� | sdS d}d}x�t �| �D ]�}|�d�}|s4d}P t�|�}|r�d}|r�|�d�rxt|�d��}|dksr|dk rxd}q |�d�r�d}P q�|�d �r�t|�d ��}|dks�|dk r�td ��|�d �r�d}P q |s td |�d� ��q W ||fS ) N)r Fr Fr � r\ r) T� zInvalid window sizer( z#Extension for deflate not supported)rk �finditer�grouprj �matchr# rR )rl rm �compress� notakeover�extZdefextrr r! r! r" � ws_ext_parse� s> rv rn )rs rm �server_notakeoverrE c C s^ | dk s| dkrt d��dg}|s,|�d� | dk rF|�dt| � � |rT|�d� d�|�S ) Nr) rn zCCompress wbits must between 9 and 15, zlib does not support wbits=8zpermessage-deflateZclient_max_window_bitszserver_max_window_bits=Zserver_no_context_takeoverz; )� ValueError�appendrM �join)rs rm rw Z enabledextr! r! r" � ws_ext_gen� s r{ c @ s e Zd ZdZdZdZdZdS )� WSParserStater r( r^ r\ N)r r r � READ_HEADER�READ_PAYLOAD_LENGTH�READ_PAYLOAD_MASK� READ_PAYLOADr! r! r! r" r| � s r| c @ s� e Zd Zdee eedd�dd�Zdd�dd�Ze e ee f d �d d �Z e e ee f d �d d �Z e e e eee eee f d�dd�ZdS )r TN)�queue� max_msg_sizers rE c C sp || _ || _d | _t� | _tj| _d | _d| _ d | _ t� | _ d| _ d| _ d | _d| _d| _d | _d | _|| _d S )NF� r )r� � _max_msg_size�_excr` �_partialr| r} �_state�_opcode� _frame_fin� _frame_opcode�_frame_payload�_tail� _has_mask� _frame_mask�_payload_length�_payload_length_flag� _compressed�_decompressobj� _compress)rF r� r� rs r! r! r" rK s" zWebSocketReader.__init__)rE c C s | j �� d S )N)r� �feed_eof)rF r! r! r" r� s zWebSocketReader.feed_eof)rB rE c C sR | j rd|fS y | �|�S tk rL } z|| _ | j�|� dS d }~X Y nX d S )NT)Tr� )r� � _feed_data� Exceptionr� � set_exception)rF rB �excr! r! r" � feed_data s zWebSocketReader.feed_datac C s� �x�| � |�D �]x\}}}}|r8| js8tjtj d�| _|tjk�r"t|�dkr�t|d d� �d }|dk r�|t kr�t t j d|� ���y|dd � � d�}W n0 tk r� } zt t jd�|�W d d }~X Y nX ttj||�} n4|�rt t j d|� d |� d |����nttjdd �} | j�| d� q|tjk�rL| j�ttj|d �t|�� q|tjk�rv| j�ttj|d �t|�� q|tjtjfk�r�| jd k�r�t t j d |����q|�s|tjk�r�|| _| j�|� | j�r�t| j�| jk�r�t t j d � t| j�| j���q| j�r*|tjk�r*t t j d � |���|tjk�rR| jd k �sFt �| j}d | _| j�|� | j�r�t| j�| jk�r�t t j d � t| j�| j���|�r�| j�t � | j� | j| j�} | jj!�r�t| jj!�} t t j d� | j| | j���n t"| j�} | j�#� |tjk�rly*| � d�} | j�ttj| d �t| �� W n2 tk �rh } zt t jd�|�W d d }~X Y nX q| j�ttj| d �t| �� qW dS )N)�wbitsr( r i� zInvalid close code: zutf-8zInvalid UTF-8 text messagezInvalid close frame: � � zUnexpected opcode=z Message size {} exceeds limit {}z<The opcode in non-fin frame is expected to be zero, got {!r}z-Decompressed message size {} exceeds limit {})Fr� )$� parse_framer� �zlib� decompressobj� MAX_WBITSr r2 rb �UNPACK_CLOSE_CODEr' r r r �decode�UnicodeDecodeErrorr r r� r� r0 r1 r. r/ r� r- r� �extendr� r �formatra ri � decompress�unconsumed_tailrW �clear) rF rB �fin�opcode�payload� compressedZ close_codeZ close_messager� �msgZpayload_merged�leftr6 r! r! r" r� & s� zWebSocketReader._feed_data)�bufrE c C s� g }| j r | j | d }| _ d}t|�}�xD| jtjk�rr|| dk�rp|||d � }|d7 }|\}}|d? d@ }|d? d@ } |d? d@ } |d? d@ } |d @ } | s�| s�| r�| js�ttjd ��| dkr�|dkr�ttjd ��|d? d@ } |d @ }| dk�r|d k�rttjd��| j �s"| j dk�r4| �r,dnd| _ n| �rFttjd ��t |�| _ | | _ t | �| _ || _tj| _nP | jtjk�r^| j}|dk�r�|| dk�r�|||d � }|d7 }t|�d }|| _| j �r�tjntj| _nP nz|dk�rB|| dk�r>|||d � }|d7 }t|�d }|| _| j �r4tjntj| _nP n || _| j �rVtjntj| _| jtjk�r�|| dk�r�|||d � | _|d7 }tj| _nP | jtjkr,| j}| j}|| }||k�r�|| | _|�||d� � |}n$d| _|�|||| � � || }| jdk�rj| j �r@| jdk �s4t�t| j|� |�| j | j || j f� t� | _tj| _q,P q,W ||d� | _ |S )z&Return the next frame from the socket.r� r r( � r ro � r\ rn z*Received frame with non-zero reserved bitsz!Received fragmented control frame� �} z5Control frame payload cannot be larger than 125 bytesNTF�~ r+ ) r� rb r� r| r} r� r r r r� r� �boolr� r� r� r~ � UNPACK_LEN2r� r r� � UNPACK_LEN3r� r� r� ra �_websocket_maskry r` )rF r� �framesZ start_posZ buf_lengthrB Z first_byteZ second_byter� Zrsv1Zrsv2Zrsv3r� Zhas_mask�lengthr� Z chunk_lenr! r! r" r� � s� zWebSocketReader.parse_frame)T)r r r r r r# r� rK r� rW r r� r� r r r` r� r! r! r! r" r � s |r c @ s� e Zd Zdee�� ddd�eeje e e e e dd�dd�Z de e ee dd�d d �Ze dd �d d �Zd e dd�dd�Zd e dd�dd�Zd eee f e ee dd�dd�Zd e e dd�dd�ZdS ) r Fr )�use_mask�limit�randomrs rt N)�protocol� transportr� r� r� rs rt rE c C sB || _ || _|| _|j| _|| _|| _d| _|| _d| _d | _ d S )NFr ) r� r� r� � randrangers rt �_closing�_limit� _output_size� _compressobj)rF r� r� r� r� r� rs rt r! r! r" rK > s zWebSocketWriter.__init__)rI r� rs rE c � s� | j r|tj@ std��d}|s&| jr�|dk r�|rFtjtj| d�}n"| jsbtjtj| j d�| _| j}|�|�}||� | j r�tj ntj � }|� t�r�|dd� }|dB }t|�}| j}|r�d}nd}|d k r�td|B |B ||B �} n<|d k �r td|B |B d |B |�} ntd|B |B d |B |�} |�r�| �dd �} | �d d�} t|�}t| |� | �| | | � | jt| �t| � t|� 7 _nLt|�tk�r�| �| � | �|� n| �| | � | jt| �t|� 7 _| j| jk�r�d| _| j � � I dH dS )z<Send a frame over the websocket with message as its payload.z!Cannot write to closing transportr r+ )�levelr� N����@ � r� i r� l �� r\ �big) r� r r2 �ConnectionResetErrorrs r� � compressobj� Z_BEST_SPEEDr� �flushrt � Z_FULL_FLUSH� Z_SYNC_FLUSH�endswithri rb r� � PACK_LEN1� PACK_LEN2� PACK_LEN3r� �to_bytesr` r� �_writer� r> r� r� Z _drain_helper) rF rI r� rs Zrsvr� Z msg_lengthr� Zmask_bit�headerr[ r! r! r" � _send_frameT sR $ zWebSocketWriter._send_frame)rB rE c C s, | j d ks| j �� r td��| j �|� d S )Nz!Cannot write to closing transport)r� Z is_closingr� �write)rF rB r! r! r" r� � s zWebSocketWriter._writer� )rI rE c � s, t |t�r|�d�}| �|tj�I dH dS )zSend pong message.zutf-8N)r_ rM �encoder� r r1 )rF rI r! r! r" r9 � s zWebSocketWriter.pongc � s, t |t�r|�d�}| �|tj�I dH dS )zSend ping message.zutf-8N)r_ rM r� r� r r0 )rF rI r! r! r" r8 � s zWebSocketWriter.ping)rI r7 rs rE c � sJ t |t�r|�d�}|r0| �|tj|�I dH n| �|tj|�I dH dS )z<Send a frame over the websocket with message as its payload.zutf-8N)r_ rM r� r� r r/ r. )rF rI r7 rs r! r! r" �send� s zWebSocketWriter.send�� )rH rI rE c � sD t |t�r|�d�}z"| jt|�| tjd�I dH W dd| _X dS )z<Close the websocket, sending the specified code and message.zutf-8)r� NT)r_ rM r� r� �PACK_CLOSE_CODEr r2 r� )rF rH rI r! r! r" r: � s zWebSocketWriter.close)N)r� )r� )FN)r� r� )r r r r? r� �Randomr �asyncioZ Transportr� r# r rK rW r r� r� r9 r8 r rM r� r: r! r! r! r" r = s ? r )F)rn FF)QrP r� � collectionsrG r� �re�sysr� �enumr �structr �typingr r r r r r r r r Z base_protocolr Zhelpersr Zstreamsr Ztypedefsr �__all__r r' r# �__annotations__r r rW � unpack_fromr� r� �unpackr� �packr� r� r� r� r> r? � namedtupler@ r r4 r r3 r r� r rR � byteorderrS rM rX rZ r` rf r� Z _websocketrg � ImportErrorri �compilerj rk r� rv r{ r| r r r! r! r! r" �<module> sp , " 4 @