NP , 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
e Z d e D Z e ee! e"d< G d 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< dZ1e e! e"d < ej2 d g d Z3 G d d e3 Z4 e4e#j5 dd Z6 e4e#j7 dd Z8 G d! d"e9 Z: G d# d$e9 Z;ej< Z=e e> e"d%< d& e?d' D Z@e ee% e"d(< d)e%d*eAd+dfd,ZBereBZCn dd-lDmEZE eEZCn
# eF$ r eBZCY nw xY w e%g d. ZGe e% e"d/< ejH d0 ZIe ee> e"d1< ejH d2 ZJe ee> e"d3< dBd5ee> d6eKd+ee!eKf fd7ZL dCd9e!d6eKd:eKd+e>fd;ZM G d< d=e ZN G d> d? ZO G d@ dA ZPdS )Dz%WebSocket protocol versions 13 and 8. N)IntEnum)Struct) AnyCallableListOptionalPatternSetTupleUnioncast ) BaseProtocol)
NO_EXTENSIONS) DataQueue)Final) 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 x/builddir/build/BUILD/imunify360-venv-2.6.2/opt/imunify360/venv/lib64/python3.11/site-packages/aiohttp/http_websocket.pyr r sS
BJNLO NOOKKKr/ r c , h | ]}t | S r. )int).0is r0 r5 0 s 'D'D'D1A'D'D'Dr/ ALLOWED_CLOSE_CODESc R e Zd ZdZdZdZdZdZdZdZ dZ
d Z eZ eZ
eZeZeZe Ze
Ze Zd
S ) r r r
i i N)r r r CONTINUATIONTEXTBINARYPINGPONGCLOSECLOSINGCLOSEDERRORtextbinarypingpongcloseclosingclosederrorr. r/ r0 r r 3 se L D
F D D E G
F E D
F D D EG
F EEEr/ r s$ 258EAFA5-E914-47DA-95CA-C5AB0DC85B11r z!Hz!Qz!BBz!BBHz!BBQi @ MSG_SIZE
DEFAULT_LIMIT_WSMessageBase)typedataextrac @ e Zd Zej ddeegef defdZdS )r )loadsrV returnc " || j S )zAReturn parsed JSON data.
.. versionadded:: 0.22
)rS )selfrV s r0 jsonzWSMessage.json] s
uTY r/ N)r r r rZ rV r r r. r/ r0 r r \ sP 48J XseSj 1 3 r/ r c < e Zd ZdZdededdf fd ZdefdZ xZS )r z WebSocket protocol parser error.codemessagerW Nc Z || _ t || d S N)r\ super__init__)rY r\ r] __class__s r0 ra zWebSocketError.__init__l s*
w'''''r/ c B t t | j d S )Nr )r
strargsrY s r0 __str__zWebSocketError.__str__p s C1 &&&r/ ) r r r __doc__r2 rd ra rg
__classcell__)rb s @r0 r r i ss **(S (3 (4 ( ( ( ( ( (' ' ' ' ' ' ' ' 'r/ r c e Zd ZdZdS )WSHandshakeErrorz#WebSocket protocol handshake error.N)r r r rh r. r/ r0 rk rk t s ----r/ rk native_byteorderc ` g | ]*t fd t d D +S )c 3 " K | ] }|z V
d S r_ r. )r3 abs r0 z.| s' 'B'B!A'B'B'B'B'B'Br/ r< )bytesrange)r3 rp s @r0
rt | s< !W!W!Wq%'B'B'B'BuSzz'B'B'B"B"B!W!W!Wr/ r<
_XOR_TABLEmaskrS rW c t |t s
J | t | dk s
J | |rd | 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 0 K | ]}t | V d S r_ )ru )r3 ns r0 rq z)_websocket_mask_python.. s& 22jm222222r/ Nr r8 )
isinstance bytearraylen translate)rv rS ro rp cds r0 _websocket_mask_pythonr s dI & &,,,, & t99>>>4>>> -22T222
1a1I''**SSqS !$Q$Z))!,,QTT
!$Q$Z))!,,QTT
!$Q$Z))!,,QTT
- -r/ )_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 rx r9 T zInvalid window sizer8 z#Extension for deflate not supported)r finditergroupr matchr2 rk )r r compress
notakeoverextdefextr s r0 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 r/ r r server_notakeoverc | dk s| dk rt d dg}|s| d | dk r%| dt | z |r| d d | S ) Nr9 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; )
ValueErrorappendrd join)r r r
enabledexts r0
ws_ext_genr s
!||x"}} T
''J 42333"}}3c(mmCDDD 86777 99Z r/ c e Zd ZdZdZdZdZdS )
WSParserStater r8 r{ rx N)r r r READ_HEADERREAD_PAYLOAD_LENGTHREAD_PAYLOAD_MASK READ_PAYLOADr. r/ r0 r r s$ KLLLr/ 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 rW Nc 8 || _ || _ d | _ t | _ t
j | _ d | _ d| _ d | _
t | _ d| _ d| _
d | _ d| _ d| _ d | _ d | _ || _ d S )NFr/ 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)rY r r r s r0 ra zWebSocketReader.__init__ s
)-1 !
#/ &* ,0 'kk
,0 $%!+/#'!r/ c 8 | j d S r_ )r feed_eofrf s r0 r zWebSocketReader.feed_eof s
r/ rS c | j rd|fS | | S # t $ r,}|| _ | j | Y d }~dS d }~ww xY w)NT)Tr/ )r
_feed_data Exceptionr
set_exception)rY rS excs r0 feed_datazWebSocketReader.feed_data sv 9 : ??4(( ( DI J $ $S ) ) ) 99999 s "
A!AAc
: | | D ]\ }}}}|r,| j s%t j t j | _ |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 h|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 t< | j | j | j }
| j j rTt | j j } t t j d | j | z | j ntC | 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 )N)wbitsr8 r i zInvalid close code: utf-8zInvalid UTF-8 text messagezInvalid close frame: zUnexpected opcode=z Message size {} exceeds limit {}z r? r r= r extendr r( formatr
decompressunconsumed_tailrr clear)
rY rS finopcodepayload
compressed
close_code
close_messager msgpayload_mergedleftrF s
r0 r zWebSocketReader._feed_data& s 040@0@0F0F v v ,C*
P$"5
P&*&8&O&O&O# ((w<<1$$!272A2;!?!?!BJ!D((Z?R-R-R , ' 6 ?: ? ? #(/ (:(:7(C(C
- # # # , ' 46R "## $IOZ OOCC <(#2JJJfJJwJJ
$IOQ;;C
$$S!,,,,9>))
$$ ingr::CLL 9>))
$$ ingr::CLL
y~y/? @@@L($ .0OV0O0O
G !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 " >
,,-ABBB)-)<)G)G 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
D"C==DAT++
U5UUbufc 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.r/ r Tr8 r r rx 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} )rY r frames start_pos
buf_lengthrS
first_byte second_byter rsv1rsv2rsv3r has_masklengthr chunk_lens r0 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"/" sU !
"
)
$
!%r/ r] r c K | j r |t j z st d d}|s| j r|dk r|r"t j t
j | }n9| j s+t j t
j | j | _ | j }| | }|| | j
r t
j n t
j z }|
t r
|dd }|dz }t | }| j }|rd}nd}|d k rt# d|z |z ||z } n;|d
k rt% d|z |z d |z | } nt' d|z |z d |z | } |r| dd }
|
d
d }
t- | }t/ |
| | | |
z |z | xj t | t |
z t | z z
c_ nt | t4 k r+| | | | n| | |z | xj t | t | z z
c_ | j | j k r(d| _ | j d{V dS dS )>!Z00D==E**D ((G D' * * * KK
/ 0 0 0 Vs4yy!83w< !T^%>%>%@%@ !&'JKK K T"""""r/ r/ c K t |t r| d }| |t j d{V dS )zSend pong message.r N)r| rd encoder r rA rY r] s r0 rI zWebSocketWriter.pong Y gs # # . nnW--G w 77777777777r/ c K t |t r| d }| |t j d{V dS )zSend ping message.r N)r| rd r r r r@ r s r0 rH zWebSocketWriter.ping r r/ rG c K t |t r| d }|r)| |t j | d{V dS | |t j | d{V dS )r r N)r| rd r r r r? r> )rY r] rG r s r0 sendzWebSocketWriter.send s gs # # . nnW--G F""7I,r9 s` + + +
R R R R R R R R R R R R R R R R R R R R R R ' ' ' ' ' ' " " " " " "
'
(E'D 'D'D'D U3s8_ D D D 0 ?e > > > fTll& fTll& F4LL' F5MM F6NN F6NN &,,# %* !
uSz ! ! ! ('(8:S:S:STT Ii .d;; Yy 0$== ' ' ' ' 'Y ' ' '. . . . .y . . . #} %* , , , "X!WEERUJJ!W!W!W
E$u+ W W W- -i -D - - - -0 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 { { { { { { { {| @! @! @! @! @! @! @! @! @! @!s G G('G(