U
|
[ ±di[ ã @ s° d Z ddlZddlZddlmZmZ ddlmZ ddlm Z m
|
Z
|
m Z m Z m Z mZmZ erdddlmZ dd ed
|
D ZG d d d ZG d d dZdZdZdZdZdZeZdZ dZ dZ dZ dZ dZ!dZ"G dd deZ#G dd d eZ$e$j%e$j&e$j'fZ(d Z)d Z*d Z+d Z,d Z-d!Z.d"Z/d"Z0e 1d#d¡Z2G d$d% d%e3Z4G d&d' d'e Z5G d(d) d)e Z6G d*d+ d+e Z7e8e9e8d,d-d.Z:G d/d0 d0Z;G d1d2 d2Z<G d3d4 d4Z=G d5d6 d6Z>dS )7zY
|
wsproto/frame_protocol
|
~~~~~~~~~~~~~~~~~~~~~~
|
|
WebSocket frame protocol implementation.
|
é N)ÚgetincrementaldecoderÚIncrementalDecoder)ÚIntEnum)Ú GeneratorÚListÚ
|
NamedTupleÚOptionalÚTupleÚ TYPE_CHECKINGÚUnioné )Ú Extensionc s& g | ] t fd dtdD qS )c 3 s | ]}| A V qd S ©N© )Ú.0Úa©Úbr úMd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\wsproto/frame_protocol.pyÚ <genexpr> s z<listcomp>.<genexpr>é )ÚbytesÚrange)r r r r Ú
|
<listcomp> s r r c @ s, e Zd ZeddddZeedddZdS )ÚXorMaskerSimpleN)Ú masking_keyÚreturnc C s
|
|| _ d S r )Ú _masking_key)Úselfr r r r Ú__init__ s zXorMaskerSimple.__init__©Údatar c C sÒ |rÎt |}dd | jD \}}}}|d d d
|¡|d d d
< |dd d
|¡|dd d
< |dd d
|¡|dd d
< |dd d
|¡|dd d
< t|d }| j|d
| jd |
| _t|S |S )Nc s s | ]}t | V qd S r )Ú
|
_XOR_TABLE)r Únr r r r s z*XorMaskerSimple.process.<locals>.<genexpr>é r é é )Ú bytearrayr Ú translateÚlenr )r r! Z
|
data_arrayr r ÚcÚdZ key_rotationr r r Úprocess s ÿzXorMaskerSimple.process)Ú__name__Ú
|
__module__Ú __qualname__r r r, r r r r r s r c @ s e Zd ZeedddZdS )Ú XorMaskerNullr c C s |S r r ©r r! r r r r, . s zXorMaskerNull.processN)r- r. r/ r r, r r r r r0 - s r0 é~ é é} iÿÿ l ÿÿÿÿ é é@ é é é c @ s6 e Zd ZdZdZdZdZdZdZdZ e
|
dd d
|
Z d S ) ÚOpcodez7
|
RFC 6455, Section 5.2 - Base Framing Protocol
|
r r r% é é é
|
©r c C s t | d@ S )Nr; )Úbool©r r r r Ú iscontrol_ s zOpcode.iscontrolN) r- r. r/ Ú__doc__Ú CONTINUATIONÚTEXTÚBINARYÚCLOSEÚPINGÚPONGr? rA r r r r r: H s r: c @ sH e Zd ZdZdZdZdZdZdZdZ dZ
|
d Z d
|
Z d Z d Zd ZdZdZdS )Ú CloseReasonz8
|
RFC 6455, Section 7.4.1 - Defined Status Codes
|
éè ié iê ië ií iî iï ið iñ iò ió iô iõ i÷ N)r- r. r/ rB ÚNORMAL_CLOSUREZ
|
GOING_AWAYÚPROTOCOL_ERRORZUNSUPPORTED_DATAÚNO_STATUS_RCVDÚABNORMAL_CLOSUREÚINVALID_FRAME_PAYLOAD_DATAZPOLICY_VIOLATIONZMESSAGE_TOO_BIGZ MANDATORY_EXTÚINTERNAL_ERRORZSERVICE_RESTARTZTRY_AGAIN_LATERÚTLS_HANDSHAKE_FAILEDr r r r rI c s rI rJ i· i¸ i i i z!Ic s, e Zd Zejfeedd fdd Z ZS )Ú ParseFailedN)ÚmsgÚcoder c s t |¡ || _d S r )Úsuperr rT )r rS rT ©Ú __class__r r r Ñ s zParseFailed.__init__)r- r. r/ rI rL Ústrr Ú __classcell__r r rV r rR Ð s ÿ þrR c @ s& e Zd ZU eed< eed< eed< dS )ÚRsvBitsÚrsv1Úrsv2Úrsv3N)r- r. r/ r? Ú__annotations__r r r r rZ Ø s
|
rZ c @ s: e Zd ZU eed< eed< eed< eed< ee ed< dS )ÚHeaderÚfinÚrsvÚopcodeÚ payload_lenr N)
|
r- r. r/ r? r^ rZ r: Úintr r r r r r r_ Þ s
|
|
r_ c @ s@ e Zd ZU eed< eeeee ef f ed< e
|
ed< e
|
ed< dS )ÚFramerb ÚpayloadÚframe_finishedÚmessage_finishedN) r- r. r/ r: r^ r r rX r rd r? r r r r re æ s
|
re )r! Únbytesr c C s4 t | |kr| S | d |
} | jddd d¡} | S )Núutf-8Úignore)Úerrors)r) ÚdecodeÚencode)r! ri r r r Ú_truncate_utf8í s
|
ro c @ s e Zd Zdee ddddZeddddZeedd d
|
Zeee dd d Z dd ddZ
|
dd ddZ ed ddZ dS )ÚBufferN)Ú initial_bytesr c C s t | _d| _|r | |¡ d S ©Nr )r' ÚbufferÚ
|
bytes_usedÚfeed)r rq r r r r þ s zBuffer.__init__)Ú new_bytesr c C s | j |7 _ d S r )rs )r rv r r r ru s z Buffer.feed)ri r c C s6 |s
|
t S | j| j| j|
}| jt|7 _|S r )r' rs rt r) )r ri r! r r r Úconsume_at_most s
|
zBuffer.consume_at_mostc C s" t | j| j |k rd S | |¡S r )r) rs rt rw )r ri r r r Úconsume_exactly s zBuffer.consume_exactlyr> c C s | j d | j
= d| _d S rr )rs rt r@ r r r Úcommit s z Buffer.commitc C s
|
d| _ d S rr )rt r@ r r r Úrollback s zBuffer.rollbackc C s
|
t | jS r )r) rs r@ r r r Ú__len__ s zBuffer.__len__)N) r- r. r/ r r r ru rd rw rx ry rz r{ r r r r rp ý s rp c @ s* e Zd ZddddZeedddZdS )ÚMessageDecoderNr> c C s d | _ d | _d S r )rb Údecoderr@ r r r r " s zMessageDecoder.__init__©Úframer c
|
C s
|
|j ¡ rt| j d kr6|j tjkr,td|j | _ n|j tjk rPtd|j |j tjkrhtd | _|j or|j
|
}| jd kr|j }n\t |j t tfstz| j |j |¡}W n2 tk
|
rà } ztt|tjW 5 d }~X Y nX t| j ||j |}|rd | _ d | _|S )Nzunexpected CONTINUATIONz expected CONTINUATION, got %rrj )rb rA ÚAssertionErrorr: rC rR rD r r} rg rh rf Ú
|
isinstancer r' rm ÚUnicodeDecodeErrorrX rI rO re )r r Úfinishedr! Úexcr r r Ú process_frame& s,
|
|
|
"z MessageDecoder.process_frame)r- r. r/ r re r
r r r r r| ! s r| c @ s e Zd Zdeeed ddddZeddddZee d d
|
d Z
|
ed d d Z e e ee dddZe ee ddddZdS )Ú FrameDecoderNr ©ÚclientÚ
|
extensionsr c C s: || _ |p g | _t | _d | _d | _d | _d| _d| _d S rr ) r r rp rs ÚheaderÚeffective_opcodeÚmaskerÚpayload_requiredÚpayload_consumed©r r r r r r r H s
|
zFrameDecoder.__init__r c C s | j |¡ d S r )rs ru r1 r r r Ú receive_bytesV s zFrameDecoder.receive_bytesr> c C st | j s| ¡ sd S | j d k s t| jd k s.t| jd k s<tt| j| jk rPd S | j j| j }| j
|
|¡}|s~| j jdkr~d S | j ¡ | j t|7 _ | j | j jk}| j |¡}| j D ](}| | |¡}t|trÞtd||}qº|r:t }| j D ]:}| | | j j¡}t|tr td||d k rö||7 }qö||7 }t| j||| j j}|rhd | _ d | _d | _ntj| _|S )Nr úerror in extension)r Ú parse_headerr r r r) rs r rc r rw ry r, r Zframe_inbound_payload_datar rI rR r' Zframe_inbound_completer` re r: rC ) r Zpayload_remainingrf r Ú extensionZpayload_ÚfinalÚresultr r r r Úprocess_bufferY sJ
|
|
|
|
|
|
|
zFrameDecoder.process_bufferc C s¶ | j d¡}|d kr"| j ¡ dS t|d t@ }tt|d t@ t|d t@ t|d t@ }|d t @ }z t
|
|}W n$ t k
|
r t d|dY nX | ¡ r´|s´t dt|d t@ }|d t@ }| ||¡}|d krò| j ¡ dS | |||¡ |r| jrt d|s,| js,t d |rb| j d
|
¡}|d krV| j ¡ dS t|| _nt | _| j ¡ t||||d | _| jj| _| jj ¡ r¦|| _nd| _d| _ d S ) Nr% Fr zInvalid opcode z#xz)Invalid attempt to fragment control framer z'client received unexpected masked framez)server received unexpected unmasked framer$ T) rs rx rz r? ÚFIN_MASKrZ Ú RSV1_MASKÚ RSV2_MASKÚ RSV3_MASKÚ OPCODE_MASKr: Ú
|
ValueErrorrR rA Ú MASK_MASKÚPAYLOAD_LEN_MASKÚ parse_extended_payload_lengthÚextension_processingr r r r0 ry r_ r rb r r r ) r r! r` ra rb Zhas_maskZpayload_len_shortrc r r r r r sV
|
ý
|
|
|
|
|
zFrameDecoder.parse_header)rb rc r c C sª | ¡ r|tkrtd|tkrX| j d¡}|d kr8d S t d|¡\}|tkr¦tdnN|tkr¦| j d¡}|d krxd S t d|¡\}|t krtd|d? r¦td |S )
|
Nz$Control frame with payload len > 125r% ú!Hz6Payload length used 2 bytes when 1 would have sufficedr; ú!Qz6Payload length used 8 bytes when 2 would have sufficedé? z'8-byte payload length with non-zero MSB)
|
rA ÚMAX_PAYLOAD_NORMALrR ÚPAYLOAD_LENGTH_TWO_BYTErs rx ÚstructÚunpackÚPAYLOAD_LENGTH_EIGHT_BYTEÚMAX_PAYLOAD_TWO_BYTE)r rb rc r! r r r r À s. ÿ ÿz*FrameDecoder.parse_extended_payload_length)rb ra rc r c C s dddg}| j D ]F}| | |||¡}t|tr8td|t|D ]\}}|r@d||< q@qt||D ]\} }
|
|
|
rb| sbtdqbd S )NFr Tz Reserved bit set unexpectedly)r Zframe_inbound_headerr rI rR Ú enumerateÚzip) r rb ra rc Zrsv_usedr r ÚbitÚusedÚexpectedÚfoundr r r r Þ s
|
|
|
|
z!FrameDecoder.extension_processing)N)r- r. r/ r? r r r r r re r r r: rd r rZ r r r r r r G s$ ÿ
|
þ 26 þ þr c @ sþ e Zd Zeed ddddZeedddZee e ddf d d
|
d Z
|
e dd d dZ eeddf d ddZ d$e e e e e dddZd%e e dddZd&e e dddZd'ee eef ee dd d Zeeeed d d Zd(ee ee d!d"d#ZdS ))Ú FrameProtocolr Nr c C sB || _ dd |D | _t| j | j| _t | _| ¡ | _d | _d S )Nc S s g | ]}| ¡ r|qS r ©Zenabled©r Úextr r r r ñ s z*FrameProtocol.__init__.<locals>.<listcomp>) r r r Ú_frame_decoderr| Ú_message_decoderÚ_parse_more_genÚ _parse_moreÚ_outbound_opcoder r r r r ï s
|
zFrameProtocol.__init__r~ c
|
C s$ |j }t|ttfst|s(tjdf}nèt|dkr>tdnÒt
|
d|d d
¡\}|t k sd|t krltdz t|}W n t k
|
r Y nX |tkrtdt|ts¸|tkr¸tdz|dd
d ¡}W n8 tk
|
r } ztd
|
t| tjW 5 d }~X Y nX ||f}t|j||j|jS ) NÚ r zCLOSE with 1 byte payloadr¡ r% zCLOSE with invalid codez#remote CLOSE with local-only reasonz CLOSE with unknown reserved coderj z Error decoding CLOSE reason: )rf r r r' r rI rM r) rR r¦ r§ ÚMIN_CLOSE_REASONÚMAX_CLOSE_REASONr ÚLOCAL_ONLY_CLOSE_REASONSÚMAX_PROTOCOL_CLOSE_REASONrm r rX rO re rb rg rh )r r r! rT Úreasonr r r r Ú_process_closeú s4
|
|
þz FrameProtocol._process_closer> c c sj dd | j D | _ d}|sf| j ¡ }|d k r^|j ¡ sD| j |¡}n|jtjkr^| |¡}d}|V qd S )Nc S s g | ]}| ¡ r|qS r r± r² r r r r ! s z1FrameProtocol._parse_more_gen.<locals>.<listcomp>FT)
|
r r´ r rb rA rµ r
r: rF r¿ )r Úclosedr r r r r¶ s
|
|
|
z FrameProtocol._parse_more_genr c C s | j |¡ d S r )r´ r r1 r r r r 0 s zFrameProtocol.receive_bytesc c s$ | j D ]}|d kr q q|V qd S r )r· )r Úeventr r r Úreceived_frames3 s
|
z FrameProtocol.received_frames)rT r¾ r c C s t }|tjkrd }|d kr(|r(td|tkr6tj}|d k rr|t t d|¡7 }|d k rr|t| d¡t
|
d 7 }| t j |¡S )Nz&cannot specify a reason without a coder¡ rj r% )r' rI rM Ú TypeErrorr¼ rK r¦ Úpackro rn r¤ Ú_serialize_framer: rF )r rT r¾ rf r r r Úclose: s
|
ÿzFrameProtocol.closeó )rf r c C s | tj|¡S r )rÅ r: rG ©r rf r r r ÚpingK s zFrameProtocol.pingc C s | tj|¡S r )rÅ r: rH rÈ r r r ÚpongN s zFrameProtocol.pongT)rf r` r c C s t |tttfrtj}n$t |tr4tj}| d¡}nt d| j
|
d krN|| _
|
n| j
|
|k rbt dntj }|rrd | _
|
| |||¡S )Nrj zMust provide bytes or textz!Data type mismatch inside message)r r r' Ú
|
memoryviewr: rE rX rD rn r r¸ rà rC rÅ )r rf r` rb r r r Ú send_dataQ s
|
|
|
|
zFrameProtocol.send_data)r` ra rb r c C sJ t |d> }t |jd> t |jd> t |jd> }t |}||B |B S )Né é é r$ )rd r[ r\ r] )r r` ra rb Zfin_bitsZrsv_bitsZ opcode_bitsr r r Ú_make_fin_rsv_opcodeh s *z"FrameProtocol._make_fin_rsv_opcode)rb rf r` r c C s t ddd}t| jD ]}| | ||||¡\}}q| |||¡}t|}d}|tkr^|} d }
|
n |tkrpt} |}
|
n t } |}
|
d}| j
|
r| dO } t || g} |
|
d k rÜ| ¡ r®t d|rÈ| t t d|
|
¡7 } n| t t d|
|
¡7 } | j
|
rt d¡} t| } | | | |¡ S | | S )NFTr5 z"payload too long for control framer¢ r¡ r$ )rZ Úreversedr Zframe_outboundrÐ r) r¤ r© r¥ r¨ r r' rA r r¦ rÄ ÚosÚurandomr r, )r rb rf r` ra r Zfin_rsv_opcodeZpayload_lengthZ quad_payloadZ first_payloadZsecond_payloadr r r r r r rÅ o s:
|
z FrameProtocol._serialize_frame)NN)rÇ )rÇ )rÇ T)rÇ T)r- r. r/ r? r r re r¿ r r r¶ r r r rd rX rÆ rÉ rÊ r r' rÌ rZ r: rÐ rÅ r r r r r° î s0 ! ÿ þ ÿ þr° )?rB rÒ r¦ Úcodecsr r Úenumr Útypingr r r r r r
|
r r r r r" r r0 r¥ r¨ r¤ r© ZMAX_PAYLOAD_EIGHT_BYTEZMAX_FRAME_PAYLOADr r r r r r r r: rI rM rN rQ r¼ rº ZMIN_PROTOCOL_CLOSE_REASONr½ ZMIN_LIBRARY_CLOSE_REASONZMAX_LIBRARY_CLOSE_REASONZMIN_PRIVATE_CLOSE_REASONZMAX_PRIVATE_CLOSE_REASONr» rÄ Z NULL_MASKÚ ExceptionrR rZ r_ re r rd ro rp r| r r° r r r r Ú<module> s` $ Yý $& (
|
