U g1@sUddlmZddlZddlZddlZddlZddlZddlZddlm Z m Z m Z ddl m Z mZzddlmZWn ek rddlmZYnXddd d d d d ddddg ZGdddejZejZejZejZejZejZej Z!eeefZ"eee!fZ#GdddejZ$e$j%de$j&de$j'de$j(de$j)de$j*de$j+de$j,de$j-de$j.de$j/de$j0d e$j1d!e$j2d"e$j3d#iZ4d$e5d%<e$j%e$j&e$j'e$j(e$j+e$j,e$j-e$j.e$j/e$j0e$j1e$j2h Z6e$j%e$j&e$j)hZ7e8e9e:fZ;ejdd(l?m@Z@dS))) annotationsN)Callable GeneratorSequence) PayloadTooBig ProtocolError) apply_maskOpcodeOP_CONTOP_TEXT OP_BINARYOP_CLOSEOP_PINGOP_PONG DATA_OPCODES CTRL_OPCODESFrameClosec@s$eZdZdZd\ZZZd\ZZZ dS)r z#Opcode values for WebSocket frames.)rr) N) __name__ __module__ __qualname____doc__CONTTEXTBINARYCLOSEPINGPONGr#r#5/tmp/pip-unpacked-wheel-dx_q7dq3/websockets/frames.pyr #s c@sLeZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdS) CloseCodez-Close code values for WebSocket close frames.iiiiiiiiiiiiiiiN)rrrrNORMAL_CLOSURE GOING_AWAYPROTOCOL_ERRORUNSUPPORTED_DATANO_STATUS_RCVDABNORMAL_CLOSURE INVALID_DATAPOLICY_VIOLATIONMESSAGE_TOO_BIGMANDATORY_EXTENSIONINTERNAL_ERRORSERVICE_RESTARTTRY_AGAIN_LATER BAD_GATEWAY TLS_HANDSHAKEr#r#r#r$r%5s r%OKz going awayzprotocol errorzunsupported datazno status received [internal]zabnormal closure [internal]zinvalid frame payload datazpolicy violationzmessage too bigzmandatory extensionzinternal errorzservice restartztry again laterz bad gatewayz TLS handshake failure [internal]zdict[int, str]CLOSE_CODE_EXPLANATIONSc@seZdZUdZded<ded<dZded<d Zded <d Zded <d Zded <e e j d dZ ddddZedddddddddddZddddddddZd dd!d"ZdS)#raN WebSocket frame. Attributes: opcode: Opcode. data: Payload data. fin: FIN bit. rsv1: RSV1 bit. rsv2: RSV2 bit. rsv3: RSV3 bit. Only these fields are needed. The MASK bit, payload length and masking-key are handled on the fly when parsing and serializing frames. r opcodebytesdataTboolfinFrsv1rsv2rsv3ZWEBSOCKETS_MAX_LOG_SIZEZ75strreturnc Csd}t|jdt|jdkr"dnd}|jr4dnd}|jtkrTt|j}n6|jtkr|j}t||jdkr|jddd}d |dd |d || dg}d d d |D}n|jt krt t |j}n|jrzt|j}d}Wnttfk r|j}t||jdkrf|jddd}d |dd |d || dg}d dd |D}d}YnXnd}t||jkr|jdd}|dd |d|| d}d td|||g}|jjd |d|dS)zE Return a human-readable representation of a frame. Nz byters continuedrs css|]}|dVqdSZ02xNr#.0byter#r#r$ sz Frame.__str__..textcss|]}|dVqdSrHr#rIr#r#r$rLsbinaryz''z...z, z [])lenr9r;r7r reprdecoder MAX_LOG_SIZEjoinrr?rparseUnicodeDecodeErrorAttributeErrorfiltername)selfZcodinglengthZ non_finalr9rNZcutmetadatar#r#r$__str__s<&  & & "z Frame.__str__N)max_size extensionsz-Callable[[int], Generator[None, None, bytes]]z int | Nonez%Sequence[extensions.Extension] | NonezGenerator[None, None, Frame]) read_exactmaskr^r_rAc cs|dEdH}td|\}}|d@r*dnd}|d@r:dnd} |d@rJdnd} |d @rZdnd} zt|d @} Wn,tk r} ztd | W5d} ~ XYnX|d@rdnd|krtd |d @}|dkr|dEdH}td|\}n&|d kr |dEdH}td|\}|dk r6||kr6td|d|d|rJ|dEdH}||EdH}|rht||}|| ||| | | }|dkrg}t|D]}|j||d}q| |S)a& Parse a WebSocket frame. This is a generator-based coroutine. Args: read_exact: Generator-based coroutine that reads the requested bytes or raises an exception if there isn't enough data. mask: Whether the frame should be masked i.e. whether the read happens on the server side. max_size: Maximum payload size in bytes. extensions: List of extensions, applied in reverse order. Raises: EOFError: If the connection is closed without a full WebSocket frame. UnicodeDecodeError: If the frame contains invalid UTF-8. PayloadTooBig: If the frame's payload size exceeds ``max_size``. ProtocolError: If the frame contains incorrect values. rN!BBTF@ zinvalid opcodezincorrect masking~!Hrz!Qzover size limit (z > z bytes))r^) structunpackr ValueErrorrrr reversedrRcheck)clsr`rar^r_r9head1head2r;r<r=r>r7excr[ mask_bytesframe extensionr#r#r$rUsB    z Frame.parser_)rar_rAc Cs(||dkrg}|D]}||}qt}|jr:dnd|jrFdndB|jrTdndB|jrbdndB|jB}|rvdnd}t |j }|dkr| t d|||Bn>|d kr| t d ||dB|n| t d ||d B||rtd }| ||rt|j |} n|j } | | |S)aH Serialize a WebSocket frame. Args: mask: Whether the frame should be masked i.e. whether the write happens on the client side. extensions: List of extensions, applied in order. Raises: ProtocolError: If the frame contains incorrect values. Nrcrrdrerfrirbiz!BBHz!BBQrhrk)rpencodeioBytesIOr;r<r=r>r7rPr9writerlpacksecretsZ token_bytesr getvalue) rZrar_rwoutputrrrsr[rur9r#r#r$ serializes>          zFrame.serializeNonecCsL|js|js|jrtd|jtkrHt|jdkr:td|jsHtddS)z Check that reserved bits and opcode have acceptable values. Raises: ProtocolError: If a reserved bit or the opcode is invalid. zreserved bits must be 0}zcontrol frame too longzfragmented control frameN) r<r=r>rr7rrPr9r;rZr#r#r$rpNs z Frame.check)rrrr__annotations__r;r<r=r>intosenvirongetrSr] classmethodrUrrpr#r#r#r$rzs     2N;c@s`eZdZUdZded<ded<ddddZed dd d d Zd dd dZddddZ dS)rz Code and reason for WebSocket close frames. Attributes: code: Close code. reason: Close reason. rcoder?reasonr@cCszd|jkrdkr nnd}n.d|jkr6dkr@nnd}nt|jd}|jd|d}|jrv|d |j}|S) zU Return a human-readable representation of a close code and reason. iZ registeredz private useunknownz ()rG)rr6rr)rZZ explanationresultr#r#r$r]nsz Close.__str__r8)r9rAcCslt|dkrHtd|dd\}|dd}|||}||St|dkr`|tjdStddS)z Parse the payload of a close frame. Args: data: Payload of the close frame. Raises: ProtocolError: If data is ill-formed. UnicodeDecodeError: If the reason isn't valid UTF-8. rrjNrrBzclose frame too short)rPrlrmrRrpr%r*r)rqr9rrcloser#r#r$rUs    z Close.parsecCs |td|j|jS)z: Serialize the payload of a close frame. rj)rprlr}rrryrr#r#r$rszClose.serializercCs.|jtks*d|jkr dks*ntddS)z Check that the close code has a valid value for a close frame. Raises: ProtocolError: If the close code is invalid. rrzinvalid status codeN)rEXTERNAL_CLOSE_CODESrrr#r#r$rps"z Close.checkN) rrrrrr]rrUrrpr#r#r#r$r`s  rx)A __future__rZ dataclassesenumrzrr~rltypingrrr exceptionsrrZspeedupsr ImportErrorutils__all__IntEnumr rr rr rr r rr!rr"rrrr%r&r'r(r)r*r+r,r-r.r/r0r1r2r3r4r6rrZOK_CLOSE_CODESr8 bytearray memoryviewZ BytesLikeZ dataclassrrrBr_r#r#r#r$s     fL