U
'7`
b @ sj d Z ddlZddlZddlZddlZddlZddlZddlZddlm Z ddl
m Z ddl m
Z
mZmZmZmZmZ ddlmZ ddlmZ dd lmZ d
ZG d d d e Zd
d eD ZG dd de Z dZ e dj Z e dj Z e dj!Z"e dj#Z$e dj#Z%e dj#Z&e dj#Z'dZ(dZ)e*dddd gZ+G d d d e+Z,e,e j-ddZ.e,e j/ddZ0G d d d e1Z2G d!d" d"e1Z3ej4Z5d#d$ e6d%D Z7e8e9dd&d'd(Z:ere:Z;n0zdd)l<m=Z= e=Z;W n e>k
r e:Z;Y nX e8ddd*d*gZ?e@d+ZAe@d,ZBd;eeC eDeeEeDf d.d/d0ZFd<eEeDeDeCd2d3d4ZGG d5d6 d6e ZHG d7d8 d8ZIG d9d: d:ZJdS )=z%WebSocket protocol versions 13 and 8. N)IntEnum)Struct)AnyCallableListOptionalTupleUnion ) BaseProtocol)
NO_EXTENSIONS) DataQueue) WS_CLOSED_MESSAGEWS_CLOSING_MESSAGEWS_KEYWebSocketReaderWebSocketWriter WSMessageWebSocketError WSMsgType WSCloseCodec @ s8 e Zd ZdZdZdZdZdZdZdZ dZ
d Z d
Z d Z
d S )
r i i i i i i i i i i N)__name__
__module__ __qualname__ZOKZ
GOING_AWAYPROTOCOL_ERRORZUNSUPPORTED_DATA INVALID_TEXTZPOLICY_VIOLATIONMESSAGE_TOO_BIGZMANDATORY_EXTENSIONZINTERNAL_ERRORZSERVICE_RESTARTZTRY_AGAIN_LATER r r I/opt/alt/python38/lib64/python3.8/site-packages/aiohttp/http_websocket.pyr s r c C s h | ] }t |qS r )int).0ir r r <setcomp>- s r# c @ sP e Zd ZdZdZdZdZdZdZdZ dZ
d Z eZ eZ
eZeZeZe Ze
Ze Zd
S ) r r r
i i N)r r r CONTINUATIONTEXTBINARYPINGPONGCLOSECLOSINGCLOSEDZERRORtextbinarypingpongcloseclosingclosederrorr r r r r 0 s" r s$ 258EAFA5-E914-47DA-95CA-C5AB0DC85B11z!Hz!Qz!BBz!BBHz!BBQi @ _WSMessageBasetypedataZextrac @ s. e Zd Zejdeegef edddZdS )r )loads)r= returnc C s
|| j S )zAReturn parsed JSON data.
.. versionadded:: 0.22
)r< )selfr= r r r jsonZ s zWSMessage.jsonN)r r r r@ r= r r r r r r r Y s r c s8 e Zd ZdZeedd fdd ZedddZ ZS ) r z WebSocket protocol parser error.Ncodemessager> c s || _ t || d S N)rB super__init__r? rB rC __class__r r rF i s zWebSocketError.__init__r> c C s
| j d S )Nr
)argsr? r r r __str__m s zWebSocketError.__str__) r r r __doc__r strrF rM
__classcell__r r rH r r f s r c @ s e Zd ZdZdS )WSHandshakeErrorz#WebSocket protocol handshake error.N)r r r rN r r r r rQ q s rQ c s& g | ] t fd dtdD qS )c 3 s | ]}| A V qd S rD r )r! abr r <genexpr>y s z<listcomp>.<genexpr>r( )bytesrange)r! r rS r
<listcomp>y s rX r( )maskr< r> c C s t |tst|t| dks&t| |rdd | 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 rD )
_XOR_TABLE)r! nr r r rU s z)_websocket_mask_python.<locals>.<genexpr>Nr
r$ )
isinstance bytearrayAssertionErrorlen translate)rY r< rR rT cdr r r _websocket_mask_python| s
re )_websocket_mask_cython z^(?:;\s*(?:(server_no_context_takeover)|(client_no_context_takeover)|(server_max_window_bits(?:=(\d+))?)|(client_max_window_bits(?:=(\d+))?)))*$zpermessage-deflate([^,]+)?F)extstrisserverr> c C s | sdS d}d}t | D ]}|d}|s4d} qt|}|rd}|r|drxt|d}|dksr|dk rxd}q|drd} qq|d rt|d }|dks|dk rtd
|d rd} qq|std |d q||fS )
N)r Fr Fr
rZ r% T zInvalid window sizer$ z#Extension for deflate not supported)_WS_EXT_RE_SPLITfinditergroup
_WS_EXT_REmatchr rQ )rh ri compress
notakeoverZextZdefextrp r r r ws_ext_parse s>
rs rj )rq ri server_notakeoverr> c C s^ | dk s| dkrt ddg}|s,|d | dk rF|dt| |rT|d d|S ) Nr% rj zCCompress wbits must between 9 and 15, zlib does not support wbits=8zpermessage-deflateZclient_max_window_bitszserver_max_window_bits=Zserver_no_context_takeoverz; )
ValueErrorappendrO join)rq ri rt Z
enabledextr r r
ws_ext_gen s
rx c @ s e Zd ZdZdZdZdZdS )
WSParserStater
r$ r] rZ N)r r r READ_HEADERREAD_PAYLOAD_LENGTHREAD_PAYLOAD_MASK READ_PAYLOADr r r r ry s ry c @ s e Zd Zdee eeddddZddddZe e
ee f d d
d Z e e
ee f d d d
Z e e
e
eee eee f dddZdS )r TN)queue max_msg_sizerq r> 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_ _partialry rz _state_opcode
_frame_fin
_frame_opcode_frame_payload_tail _has_mask _frame_mask_payload_length_payload_length_flag _compressed_decompressobj _compress)r? r~ r rq r r r rF s" zWebSocketReader.__init__rJ c C s | j d S rD )r~ feed_eofrL r r r r s zWebSocketReader.feed_eofr< r> c
C sZ | j rd|fS z | |W S tk
rT } z || _ | j| W Y dS d }~X Y nX d S )NT)Tr )r
_feed_data Exceptionr~ Z
set_exception)r? r< excr r r feed_data s zWebSocketReader.feed_datac
C s | |D ]v\}}}}|r4| js4tjtj d| _|tjkr t|dkrt|d d d }|dk r|t krt
t j d| z|dd
d}W n0 tk
r } zt
t jd|W 5 d }~X Y nX ttj||} n2|rt
t j d| d | d |nttjdd
} | j| d q
|tjkrF| jttj|d
t| q
|tjkrp| jttj|d
t| q
|tjtjfkr| jd krt
t j d |q
|s|tjkr|| _| j| | jrt| j| jkrt
t j d t| j| jq
| jr$|tjkr$t
t j d
||tjkrL| jd k s@t | j}d | _| j| | jrt| j| jkrt
t j d t| j| j|r| jt | j | j| j}
| jj!rt| jj!} t
t j d | j| | jn
t"| j}
| j# |tjkrfz*|
d} | jttj| d
t| W n2 tk
rb } zt
t jd|W 5 d }~X Y nX q
| jttj|
d
t|
q
dS )N)wbitsr$ r i zInvalid close code: utf-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 zlibZ
decompressobjZ MAX_WBITSr r. ra UNPACK_CLOSE_CODEALLOWED_CLOSE_CODESr r r decodeUnicodeDecodeErrorr r r~ r r, r- r* r+ r r) r extendr r formatr` _WS_DEFLATE_TRAILING
decompressZunconsumed_tailrV clear)
r? r< finopcodepayloadZ
compressedZ
close_codeZ
close_messager msgZpayload_mergedleftr1 r r r r # s
zWebSocketReader._feed_data)bufr> c C s g }| j r | j | d }| _ d}t|}| jtjkrp|| dkrt|||d }|d7 }|\}}|d? d@ }|d? d@ } |d? d@ }
|d? d@ } |d @ } |
s| s| r| jsttjd
| dkr|dkrttjd |d? d@ }
|d @ }| dkr
|d
kr
ttjd| j s | j
dkr0| r(dnd| _
n| rBttjd
t || _ | | _ t |
| _
|| _tj| _nqt| jtjkr`| j}|dkr|| dkrt|||d }|d7 }t|d }|| _| j
rtjntj| _nqtn||dkrD|| dkrt|||d }|d7 }t|d }|| _| j
r4tjntj| _nqtn || _| j
rXtjntj| _| jtjkr|| dkrt|||d | _|d7 }tj| _nqt| jtjkr(| j}| j}|| }||kr|| | _|||d |}n$d| _||||| || }| jdkrt| j
rD| jdk s8tt| j| || j | j || j
f t | _tj| _q(qtq(||d | _ |S )z&Return the next frame from the socket.r r r$ r
rk rZ rj z*Received frame with non-zero reserved bitsz!Received fragmented control frame } z5Control frame payload cannot be larger than 125 bytesNTF~ r' ) r ra r ry rz r r r r r r boolr r r r{ UNPACK_LEN2r r| r} UNPACK_LEN3r r r r` _websocket_maskrv r_ )r? r ZframesZ start_posZ
buf_lengthr< Z
first_byteZ second_byter Zrsv1Zrsv2Zrsv3r Zhas_maskZlengthr Z chunk_lenr r r r s
zWebSocketReader.parse_frame)T)r r r r
r r r rF r rV r r r r r r_ r r r r r r s |r c
@ s e Zd Zdee dddeeje e
e e
e ddddZ de
e
ee
ddd d
Ze
dd d d
Zd e
ddddZd e
ddddZd eee
f e ee
ddddZd e
e
ddddZdS ) r Fr )use_masklimitrandomrq rr N)protocol transportr r r rq rr r> c C sB || _ || _|| _|j| _|| _|| _d| _|| _d| _d | _ d S )NFr )
r r r randrangerq rr _closing_limit _output_size _compressobj)r? r r r r r rq rr r r r rF ; s zWebSocketWriter.__init__)rC r rq r> c s | j r|tj@ stdd}|s&| jr|dk r|rFtjtj| d}n"| jsbtjtj| j d| _| j}||}|| | j
rtj ntj }|
tr|dd }|dB }t|}| j}|rd}nd}|d k rtd|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|tkr| | | | n| | | | jt| t| 7 _| j| jkrd| _| j I dH dS )<Send a frame over the websocket with message as its payload.!Cannot write to closing transportr r' )levelr N@ r r9 r l rZ Zbig) r r r. ConnectionResetErrorrq r compressobjZ Z_BEST_SPEEDr flushrr Z Z_FULL_FLUSHZ Z_SYNC_FLUSHendswithr ra r PACK_LEN1 PACK_LEN2 PACK_LEN3r to_bytesr_ r _writer MSG_SIZEr r Z
_drain_helper) r? rC r rq Zrsvr Z
msg_lengthr Zmask_bitheaderrY r r r _send_frameQ sX
$
zWebSocketWriter._send_framer c C s, | j d ks| j r td| j | d S )Nr )r Z
is_closingr write)r? r< r r r r s zWebSocketWriter._writer )rC r> c s, t |tr|d}| |tjI dH dS )zSend pong message.r N)r^ rO encoder r r- r? rC r r r r4 s
zWebSocketWriter.pongc s, t |tr|d}| |tjI dH dS )zSend ping message.r N)r^ rO r r r r, r r r r r3 s
zWebSocketWriter.ping)rC r2 rq r> c sJ t |tr|d}|r0| |tj|I dH n| |tj|I dH dS )r r N)r^ rO r r r r+ r* )r? rC r2 rq r r r send s
zWebSocketWriter.sendr rA c sD t |tr|d}z"| jt|| tjdI dH W 5 d| _X dS )z<Close the websocket, sending the specified code and message.r T)r N)r^ rO r r r PACK_CLOSE_CODEr r. rG r r r r5 s
zWebSocketWriter.close)N)r )r )FN)r r )r r r
DEFAULT_LIMITr ZRandomr asyncioZ Transportr r r rF rV r r r r4 r3 r rO r r5 r r r r r : sD A
r )F)rj FF)KrN r collectionsr@ r resysr enumr structr typingr r r r r r Z
base_protocolr Zhelpersr Zstreamsr
__all__r r r r unpack_fromr r unpackr packr r r r r r
namedtupler: r r0 r r/ r r r rQ byteorderZnative_byteorderrW r[ rV r_ re r Z
_websocketrf ImportErrorr compilero rl rO r r rs rx ry r r r r r r <module> s~
4 @