Q, U d Z ddlZddlZddlZddlZddlZddlZddlZddlm Z ddl
m Z ddl m
Z
mZmZmZmZmZmZmZmZmZmZ ddlmZ ddlmZm Z dd l m Z dd
l m Z d Z! G d d
e Z"d e"D Z#eee$ e%d< dZ& G d de Z'dZ(ee) e%d< e d j* Z+ e d j* Z, e d j- Z. e d j/ Z0 e d j/ Z1 e d j/ Z2 e d j/ Z3dZ4ee$ e%d< d Z5ee$ e%d < G d d e Z6 e6e'j7 dd Z8 e6e'j9 dd Z: G d d!e; Z< G d" d#e; Z=ej> Z?ee@ e%d$< ejA d%ee) fd& ZBd'e)d(eCd%dfd)ZDe reDZEn dd*lFmGZG eGZEn
# eH$ r eDZEY nw xY w e)g d+ ZIee) e%d,< ejJ d- ZKeee@ e%d.< ejJ d/ ZLeee@ e%d0< d?d2ee@ d3eMd%ee$eMf fd4ZN d@d6e$d3eMd7eMd%e@fd8ZO G d9 d:e ZP G d; d< ZQ G d= d> ZRdS )Az%WebSocket protocol versions 13 and 8. N)IntEnum)Struct) AnyCallableFinalList
NamedTupleOptionalPatternSetTupleUnioncast ) BaseProtocol)ZLibCompressorZLibDecompressor)
NO_EXTENSIONS) DataQueue) WS_CLOSED_MESSAGEWS_CLOSING_MESSAGEWS_KEYWebSocketReaderWebSocketWriter WSMessageWebSocketError WSMsgType WSCloseCodec B 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 N)__name__
__module__ __qualname__OK
GOING_AWAYPROTOCOL_ERRORUNSUPPORTED_DATAABNORMAL_CLOSURE INVALID_TEXTPOLICY_VIOLATIONMESSAGE_TOO_BIGMANDATORY_EXTENSIONINTERNAL_ERRORSERVICE_RESTARTTRY_AGAIN_LATER BAD_GATEWAY j/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib64/python3.11/site-packages/aiohttp/http_websocket.pyr r , sS
BJNLO NOOKKKr2 r c , h | ]}t | S r1 )int).0is r3 r8 < s 'D'D'D1A'D'D'Dr2 ALLOWED_CLOSE_CODESi c R e Zd ZdZdZdZdZdZdZdZ dZ
d Z eZ eZ
eZeZeZe Ze
Ze Zd
S ) r r r
i i N)r! r" r# CONTINUATIONTEXTBINARYPINGPONGCLOSECLOSINGCLOSEDERRORtextbinarypingpongcloseclosingclosederrorr1 r2 r3 r r H se L D
F D D E G
F E D
F D D EG
F EEEr2 r s$ 258EAFA5-E914-47DA-95CA-C5AB0DC85B11r z!Hz!Qz!BBz!BBHz!BBQi @ MSG_SIZE
DEFAULT_LIMITc l e Zd ZU eed< eed< ee ed< ej dde
egef defdZdS ) r typedataextra)loadsrX returnc " || j S )zAReturn parsed JSON data.
.. versionadded:: 0.22
)rV )selfrX s r3 jsonzWSMessage.jsont s
uTY r2 N) r! r" r# r __annotations__r r
strr\ rX r r1 r2 r3 r r n st
OOO
III C=48J XseSj 1 3 r2 r c < e Zd ZdZdededdf fd ZdefdZ xZS )r z WebSocket protocol parser error.codemessagerY Nc Z || _ t || d S N)r` super__init__)r[ r` ra __class__s r3 re zWebSocketError.__init__ s*
w'''''r2 c B t t | j d S )Nr )r r^ argsr[ s r3 __str__zWebSocketError.__str__ s C1 &&&r2 ) r! r" r# __doc__r5 r^ re rj
__classcell__)rf s @r3 r r ss **(S (3 (4 ( ( ( ( ( (' ' ' ' ' ' ' ' 'r2 r c e Zd ZdZdS )WSHandshakeErrorz#WebSocket protocol handshake error.N)r! r" r# rk r1 r2 r3 rn rn s ----r2 rn native_byteorderrY c 4 d t d D S )Nc ` g | ]*t fd t d D +S )c 3 " K | ] }|z V
d S rc r1 )r6 abs r3 z(_xor_table... s' ,,A!a%,,,,,,r2 r? )bytesrange)r6 rt s @r3
z _xor_table.. s< A A AE,,,,s,,, , , A A Ar2 r? )rw r1 r2 r3
_xor_tablery s A AeCjj A A AAr2 maskrV c t |t s
J | t | dk s
J | |rt fd| D \ }}}}|ddd | |ddd< |ddd | |ddd< |ddd | |ddd< |ddd | |ddd< dS 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 3 ( K | ] }| V
d S rc r1 )r6 n
_XOR_TABLEs r3 ru z)_websocket_mask_python.. s' 22jm222222r2 Nr r; )
isinstance bytearraylenry translate)rz rV rs rt cdr s @r3 _websocket_mask_pythonr s dI & &,,,, & t99>>>4>>> - \\
2222T222
1a1I''**SSqS !$Q$Z))!,,QTT
!$Q$Z))!,,QTT
!$Q$Z))!,,QTT
- -r2 )_websocket_mask_cython)r r r _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_SPLITFextstrisserverc | sdS d}d}t | D ]0}| d }|sd} nt | }|rd}|r_| d r1t | d }|dk s|dk rd}| d rd} n| d r=t | d }|dk s|dk rt
d
| d rd} n)|s%t
d | d z 2||fS )
N)r Fr Fr r| r< T zInvalid window sizer; z#Extension for deflate not supported)r finditergroupr matchr5 rn )r r compress
notakeoverextdefextr s r3 ws_ext_parser s xHJ ((00 (Y (Y1 H E (( ! YH
;;q>> ! "5;;q>> 2 2H
"}}1 #$ ;;q>> &!%J;;q>> F "5;;q>> 2 2H
"}}1 ./D E EE;;q>> &!%J Y"#H399UV<<#WXX X Y Z r2 r r server_notakeoverc | dk s| dk rt d dg}|s| d | dk r%| dt | z |r| d d | S ) Nr< r zCCompress wbits must between 9 and 15, zlib does not support wbits=8zpermessage-deflateclient_max_window_bitszserver_max_window_bits=server_no_context_takeoverz; )
ValueErrorappendr^ join)r r r
enabledexts r3
ws_ext_genr s
!||x"}} T
''J 42333"}}3c(mmCDDD 86777 99Z r2 c e Zd ZdZdZdZdZdS )
WSParserStater r; r r| N)r! r" r# READ_HEADERREAD_PAYLOAD_LENGTHREAD_PAYLOAD_MASK READ_PAYLOADr1 r2 r3 r r s$ KLLLr2 r c
e Zd Z ddee dededdfdZddZd e de
ee f fd
Z d e de
ee f fd Z d e de
e
eee eee f fd
ZdS )r Tqueue max_msg_sizer rY Nc 8 || _ || _ d | _ t | _ t
j | _ d | _ d| _ d | _
t | _ d| _ d| _
d | _ d| _ d| _ d | _ d | _ || _ d S )NFr2 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)r[ r r r s r3 re zWebSocketReader.__init__ s
)-1 !
#/ &* ,0 'kk
,0 $%!+/:>!r2 c 8 | j d S rc )r feed_eofri s r3 r zWebSocketReader.feed_eof2 s
r2 rV c | j rd|fS | | S # t $ r,}|| _ | j | Y d }~dS d }~ww xY w)NT)Tr2 )r
_feed_data Exceptionr
set_exception)r[ rV excs r3 feed_datazWebSocketReader.feed_data5 sv 9 : ??4(( ( DI J $ $S ) ) ) 99999 s "
A!AAc
, | | D ]|\ }}}}|r | j st d | _ |t j k r!t | dk rt
|d d d }|dk r&|t vr t t j
d| |dd d }n-# t $ r }t t j
d |d }~ww xY wt t j || } n@|r#t t j
d | d
| d
| t t j dd } | j | d X|t j k rC| j t t j |d t | |t j k rC| j t t j |d t | |t j t j fvr$| j t t j
d | |s|t j k r|| _ | j | | j rbt | j | j k rEt t j d
t | j | j | j r=|t j k r-t t j
d | |t j k r| j J | j }d | _ | j | | j rbt | j | j k rEt t j d
t | j | j |r| j J | j t8 | j | j | j }
| j j rTt | j j } t t j d | j | z | j nt? | j }
| j |t j k r |
d } | j t t j | d t | # t $ r }t t j
d |d }~ww xY w| j t t j |
d t |
~dS )NT)suppress_deflate_headerr; r i zInvalid close code: utf-8zInvalid UTF-8 text messagezInvalid close frame: zUnexpected opcode=z Message size {} exceeds limit {}z))
$$ ingr::CLL 9>))
$$ ingr::CLL
y~y/? @@@L($ .0OV0O0O
H !777'- M((111) c$-.@.@DDV.V.V , ' 7 > E E #DM 2 2D4F } !Y%;;;"0 + :!77=vf~~# # !777 #| 7 7 7!% '+ M((111) c$-.@.@DDV.V.V , ' 7 > E E #DM 2 2D4F " > # 2 > > >
,,-ABBB)-)<)L)L M4+=* * .> #&t':'J#K#KD"0 + ; O V V$($6$=t?Q!" !"# # */t})=)=M''))) //'#1#8#8#A#AD J 0 0 )).$ C CSYY 2 ' ' '"0 + 8:V# # #& ''
,, %i&6 K K / /
zs1 * C
C2C--C2 AT$$
U.U Ubufc g }| j r| j |z dc}| _ d}t | } | j t j k rF||z
dk r:|||dz }|dz
}|\ }}|dz dz }|dz dz } |dz dz }
|d z dz } |d
z } |
s | s | r!| j st
t j d | dk r |dk rt
t j d |dz dz }
|d
z }| dk r |dk rt
t j d | j s| j
| rdnd| _
n | rt
t j d t | | _ | | _ t |
| _
|| _ t j | _ nn\| j t j k r| j }|dk r^||z
dk rS|||dz }|dz
}t! | d }|| _ | j
r t j n t j | _ nn|dk r^||z
dk rS|||dz }|dz
}t) | d }|| _ | j
r t j n t j | _ n-nx|| _ | j
r t j n t j | _ | j t j k r4||z
d k r)|||d z | _ |d z
}t j | _ nn| j t j k r| j }| j }||z
}||k r*||z
| _ | ||d |}n,d| _ | ||||z ||z }| j dk rr| j
r | j J t1 | j | | | j | j || j
f t5 | _ t j | _ nn||d | _ |S )z&Return the next frame from the socket.r2 r Tr; r r r| r z*Received frame with non-zero reserved bitsz!Received fragmented control frame } z5Control frame payload cannot be larger than 125 bytesNF~ r> )r r r r r r r r r& r r boolr r r r UNPACK_LEN2r r r UNPACK_LEN3r r r _websocket_maskr r )r[ r frames start_pos
buf_lengthrV
first_byte second_byter rsv1rsv2rsv3r has_masklengthr chunk_lens r3 r zWebSocketReader.parse_frame s : 4 "j3 .OC XX
J {m 777 )Q.. y9q= 89D NI.2+J %?a/C &!Oq0D &!Oq0D &!Oq0D '# -F t T^ , ' 6 H
||q , ' 6 ?
!,q 0A 5H (4 /F || , ' 6 V $*:*B37+B44U(( , ' 6 H
'+3iiDO)/D&%)(^^DN06D-"/"CDKK {m ??? 2S==!I-22 "9y1}#< =!Q !,T!2!21!5/5, $~ 8
77 * 7 K {m === )Q..'*9y1}+D'ED$ NI"/" N@ r rR r l r| big) r r rE ConnectionResetErrorr _make_compress_objr flushr zlib Z_FULL_FLUSH Z_SYNC_FLUSHendswithr r r PACK_LEN1 PACK_LEN2 PACK_LEN3r to_bytesr r _writer rQ r r
_drain_helper) r[ ra r r rsv compressobj
msg_lengthr mask_bitheadermask_intrz s r3 _send_framezWebSocketWriter._send_frameo s = L&9?": L&'JKK K
6A::
0 " 5 5h ? ? ( O(,(?(?
(N(ND% " / ' 0 0 9 9999999G
{((%)_K!!$:K
G 455
'!#2#,*C\\
= HHH tczF 2J4IJJFF
7
#
# tczF 2C(NJOOFF tczF 2C(NJOOF :~~a44H$$Q..D ((G D' * * * KK
/ 0 0 0 Vs4yy!8:!E E H$$ F### G$$$$ FW ,--- Vz!9 9
t{ * * !D ---// / / / / / / / / / + *r2 c F t t j | t S )N)levelwbitsmax_sync_chunk_size)r r Z_BEST_SPEED WEBSOCKET_MAX_SYNC_CHUNK_SIZE)r[ r s r3 r z"WebSocketWriter._make_compress_obj s' #) =
r2 rV c | j | j rt d | j | d S )Nr )r
is_closingr write)r[ rV s r3 r zWebSocketWriter._write sH > !T^%>%>%@%@ !&'JKK K T"""""r2 r2 c K t |t r| d }| |t j d{V dS )zSend pong message.r N)r r^ encoder r rD r[ ra s r3 rL zWebSocketWriter.pong Y gs # # . nnW--G w 77777777777r2 c K t |t r| d }| |t j d{V dS )zSend ping message.r N)r r^ r# r r rC r$ s r3 rK zWebSocketWriter.ping r% r2 rJ c K t |t r| d }|r)| |t j | d{V dS | |t j | d{V dS )r r N)r r^ r# r r rB rA )r[ ra rJ r s r3 sendzWebSocketWriter.send s gs # # . nnW--G F""7I,
#5 #T # # # #
8 8%s
"3 8d 8 8 8 8 8 8%s
"3 8d 8 8 8 8 "& F FsEz" F F 3- F
F F F F ! ! !U5#:5F !QU ! ! ! ! ! !r2 r )F)r FF)Srk r, functoolsr\ r resysr enumr structr typingr r r r r r
r r r
r r
base_protocolr compression_utilsr r helpersr streamsr __all__r r9 r5 r] r r r rv unpack_fromr r unpackr packr r
r r* rQ rS r rG r rF r r r rn byteorderro r^ lru_cachery r r r
_websocketr ImportErrorr compiler r r r r r r r r1 r2 r3 rA s + + +
( ' ' ' ' ' ? ? ? ? ? ? ? ? " " " " " "
'
(E'D 'D'D'D U3s8_ D D D !) 0 ?e > > > fTll& fTll& F4LL' F5MM F6NN F6NN &,,# %* !
uSz ! ! !
Ii .d;; Yy 0$== ' ' ' ' 'Y ' ' '. . . . .y . . . #} %* , , , BDK B B B B- -i -D - - - -2 1,OO16666660 1 1 101 &+U+C+C+C%D%D eEl D D D #-"*/# #
E'#, )3
3P(Q(Q % % Q Q Q0 0 # 0 $ 0 5dCS 0 0 0 0 h KP! !!"&!CG!! ! ! !, G | | | | | | | |~ L! L! L! L! L! L! L! L! L! L!s =G GG