o TjA@sUddlmZddlZddlZddlZddlZddlZddlZddlm Z m Z ddl m Z ddl mZmZddl mZzddlmZWneyRddlmZYnwgd ZGd d d ejZejZejZejZejZej Z!ej"Z#eeefZ$ee!e#fZ%Gd d d ejZ&e&j'de&j(de&j)de&j*de&j+de&j,de&j-de&j.de&j/de&j0de&j1de&j2de&j3de&j4de&j5diZ6de7d<e&j'e&j(e&j)e&j*e&j-e&j.e&j/e&j0e&j1e&j2e&j3e&j4h Z8e&j'e&j(e&j+hZ9ej:Gdd d Z;ej:Gd!d"d"Z< # #d-d.d*d+Z=dd,l>m?Z?dS)/) annotationsN) GeneratorSequence)Callable) PayloadTooBig ProtocolError) BytesLike) apply_mask) OpcodeOP_CONTOP_TEXT OP_BINARYOP_CLOSEOP_PINGOP_PONG DATA_OPCODES CTRL_OPCODES CloseCodeFrameClosec@s$eZdZdZd\ZZZd\ZZZ dS)r z#Opcode values for WebSocket frames.)rr) N) __name__ __module__ __qualname____doc__CONTTEXTBINARYCLOSEPINGPONGr%r%T/home/dinkstrade/pdmp-scanner/venv/lib/python3.10/site-packages/websockets/frames.pyr &s r c@sLeZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdS)rz-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&r8s"rOKz 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 eded ediZd)ddZd*ddZedddd+d d!Zdd"d,d$d%Zd-d'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 opcoder dataTboolfinFrsv1rsv2rsv3WEBSOCKETS_MAX_LOG_SIZE75returnstrcCs|j|j}|\}}||krdn|rdnd}t|jdt|jdkr(dnd}|jr1dnd}dtd |||g}|jj d |d |d S) ax Return a human-readable representation of a frame. This function is intended for logging and debugging. It doesn't aim to support round-tripping because payloads can be too long for displaying conveniently. Instead, it shows the beginning and the end. It's robust to incorrect data. It attempts to decode UTF-8 payloads whenever possible, even for binary frames and control frames, because those frequently contain UTF-8 data. It applies the same logic to continuation frames, because we don't know if they continue a text frame or a binary frame. textbinaryz byters continuedz, N z []) DEFAULT_IS_TEXTgetr8 _data_reprlenr9r;joinfiltername)self expect_text data_repris_text data_typelength non_finalmetadatar%r%r&__str__s &z Frame.__str__tuple[str, bool | None]c Cs|js d|j|jfS|jtur(z tt|jdfWStt fy'Ynwt |jd|j krkt |jdd|j d}t |jd|j dd}t ||jtkdo]t ||jd }|rjt||jd d }nt |j}t ||jtk|jd }|rt|jd d }|s|j}t ||j dkr|j dd d}d|dd|d|| dg}ddd|D}t ||j kr|j dd }|dd|d|| d}||fS)z Return a human-readable representation of the payload. Also returns whether the payload is text. The representation is elided to fit ``MAX_LOG_SIZE``. This is a helper for the __str__ method. z''TNr)must_start_clean)must_end_cleanreplace)errors)r^r_rrsrHcss|]}|dVqdS)02xNr%).0byter%r%r& sz#Frame._data_repr..z...)r9rJrKr8rrBrparserUnicodeDecodeErrorrM MAX_LOG_SIZEbytesis_utf8_fragmentr r;reprdecoderN)rQ data_startdata_endrTrSr9rEcutr%r%r&rLsR   &"zFrame._data_reprN)max_size extensions read_exact9Callable[[int], Generator[None, None, bytes | bytearray]]maskrq int | Nonerr%Sequence[extensions.Extension] | NoneGenerator[None, None, Frame]c cs|dEdH}td|\}}|d@rdnd}|d@rdnd} |d@r&dnd} |d @r.dnd} zt|d @} WntyI} ztd | d} ~ ww|d@rPdnd|krXtd |d @}|dkro|dEdH}td|\}n|d kr|dEdH}td|\}|dur||krt|||r|dEdH}||EdH}|rt||}|| ||| | | }|durg}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. PayloadTooBig: If the frame's payload size exceeds ``max_size``. ProtocolError: If the frame contains incorrect values. rN!BBTF@ zinvalid opcodezincorrect masking~!Hrz!Qr[)rq) structunpackr ValueErrorrrr reversedrmcheck)clsrsrurqrrr9head1head2r;r<r=r>r8excrV mask_bytesframe extensionr%r%r&rgsH    z Frame.parserrrjc Cs&||dur g}|D]}||}q t}|jrdnd|jr#dndB|jr*dndB|jr1dndB|jB}|r;dnd}t |j }|dkrS| t d|||Bn|d kre| t d ||dB|n | t d ||d B||r~td }| ||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. Nrzrr{r|r}rryiz!BBHz!BBQrr[)rencodeioBytesIOr;r<r=r>r8rMr9writerpacksecrets token_bytesr getvalue) rQrurrroutputrrrVrr9r%r%r& serializeCs>          zFrame.serializeNonecCsP|js |js |jr td|jtvr$t|jdkrtd|js&tddSdS)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>rr8rrMr9r;rQr%r%r&rs z Frame.checkrArB)rArZ) rsrtrur:rqrvrrrwrArx)rur:rrrwrArjrAr)rrrr__annotations__r;r<r=r>intosenvironrKrir rrrJrYrL classmethodrgrrr%r%r%r&rzs&       HMs      O I