i+UdZddlZddlZddlZddlmZddlmZmZm Z m Z ddl m Z ddl mZddlmZmZd d lmZmZmZmZmZmZmZmZd d lmZmZd Zeee d <dZ!eee d<dZ"GddZ#y)z%WebSocket protocol versions 13 and 8.N)partial)FinalOptionalSetUnion) BaseProtocol)ClientConnectionResetError) ZLibBackendZLibCompressor)MASK_LENMSG_SIZEPACK_CLOSE_CODE PACK_LEN1 PACK_LEN2 PACK_LEN3 PACK_RANDBITSwebsocket_mask)WS_DEFLATE_TRAILING WSMsgType DEFAULT_LIMITWS_CONTROL_FRAME_OPCODEi@c8eZdZdZdeej ddddedejde de d ej d e d e d d fdZ dde de d ee d d fdZde de de d d fdZd ee d efdZde de d ee d d fdZde de d ee d d fdZdde dee efd d fdZy )WebSocketWritera3WebSocket writer. The writer is responsible for sending messages to the client. It is created by the protocol when a connection is established. The writer should avoid implementing any application logic and should only be concerned with the low-level details of the WebSocket protocol. Fr)use_masklimitrandomcompress notakeoverprotocol transportrrr r!r"returnNc||_||_||_t|jd|_||_||_d|_||_ d|_ d|_ tj|_t|_y)zInitialize a WebSocket writer. FrN)r#r$rr getrandbitsget_random_bitsr!r"_closing_limit _output_size _compressobjasyncioLock _send_lockset_background_tasks)selfr#r$rrr r!r"s T/var/www/html/binance/venv/lib/python3.12/site-packages/aiohttp/_websocket/writer.py__init__zWebSocketWriter.__init__1sq! "  &v'9'92>  $  6:!,,.:=%messageopcodecK|jr|tjzs td|s |jr |t k\r|j ||dn t|tkr9|j4d{|j|||dddd{ntj}|j|||}tjdk\rtj ||d}n|j#|}|j$j'||j)|j$j*tj,|d{|j.|j0kDrAd|_|j2j4r#|j2j7d{yyy7E7%#1d{7swYsxYw7z7$w)z(>(F(F G..+ + +   t{{ * !D }}$$mm11333% += L L L L L* ,4shA9G;G<G?G GGCG G!AG8G9GGG G GGGrsvct|}|j}|rdnd}d|z|z}|dkrt|||z}d} n*|dkrt|d|z|}d} nt |d|z|}d} |j j r td |rkt|j} t|}t| ||j j|| z|z|xjtz c_ n^|tkDr7|j j||j j|n|j j||z|xj| |zz c_ y ) a Write a websocket frame to the transport. This method handles frame header construction, masking, and writing to transport. It does not handle compression or flow control - those are the responsibility of the caller. r~rr r:N)rArrrrr$ is_closingr rr) bytearrayrwriter,rr) r3r7r8rS msg_lengthrmask_bit first_byteheader header_lenmasks r4r@z&WebSocketWriter._write_websocket_frame|sP\ ==#4CZ&(  z:+@AFJ % z3>:FFJz3>:FFJ >> $ $ &,-PQ Q  !5!5!78D(G 4 ) NN $!8 9    )  ( " NN  ( NN  ) NN '!1 2 Z*44r6c|r!ttj| tS|js0ttj|j t|_|jS)zBGet or create a compressor object for the given compression level.)levelwbitsmax_sync_chunk_size)r r Z_BEST_SPEEDrBr-r!)r3r!s r4_get_compressorzWebSocketWriter._get_compressors^ !!..i$A    .!..}}n$A!D     r6c|j|}|j|j||j|jrt j nt jzjt|dy)z Synchronous send for small compressed frames. This is used for small compressed payloads that compress synchronously in the event loop. Since there are no await points, this is inherently cancellation-safe. @N) rhr@ compress_syncflushr"r Z_FULL_FLUSH Z_SYNC_FLUSH removesuffixrr3r7r8r! compressobjs r4rCz+WebSocketWriter._send_compressed_frame_syncss**84  ##))'2## ,,$11 l./   r6cK|j4d{|j|}|j|j|d{|j |j rt jnt jzjt|ddddd{y77l7 #1d{7swYyxYww)a Async send for large compressed frames with lock. Acquires the lock and compresses large payloads asynchronously in the executor. The lock is held for the entire operation to ensure the compressor state is not corrupted by concurrent sends. MUST be run shielded from cancellation. If cancelled after compression but before sending, the compressor state would be advanced but data not sent, corrupting subsequent frames. Nrj) r0rhr@r!rlr"r rmrnrorrps r4rEz3WebSocketWriter._send_compressed_frame_async_lockeds??  ..x8K  ' '%..w77!''??$00(55 ,23    8    sWCB3C1B9B5 AB9" C-B7.C5B97C9C ?C C CcodecKt|tr|jd} |jt ||zt j d{d|_y7 #d|_wxYww)zrsj+ ..(:;   3! uSz!'(s'!*^!^!r6