tniojddlmZddlZddlZddlZddlmZddlmZm Z m Z m Z m Z m Z ddlmZddlmZmZmZmZmZmZmZmZmZmZddlmZmZdd lmZdd l m!Z!m"Z"m#Z#m$Z$gd Z%eezezZ& Gd d ej'Z(e(j)Z)e(j*Z*Gddej'Z+e+j,Z,e+j-Z-e+j.Z.e+j/Z/dZ0 GddZ1dS)) annotationsN) Generator)ConnectionClosedConnectionClosedErrorConnectionClosedOK InvalidState PayloadTooBig ProtocolError) Extension) OK_CLOSE_CODES OP_BINARYOP_CLOSEOP_CONTOP_PINGOP_PONGOP_TEXTClose CloseCodeFrame)RequestResponse) StreamReader) BytesLike LoggerLikeOrigin Subprotocol)ProtocolSideStateSEND_EOFc.eZdZdZed\ZZdS)rz6A WebSocket connection is either a server or a client.N)__name__ __module__ __qualname____doc__rangeSERVERCLIENTP/var/www/html/gpu-tools/venv/lib/python3.11/site-packages/websockets/protocol.pyrr-s%@@U1XXNFFFFr,rc2eZdZdZed\ZZZZdS)r z6A WebSocket connection is in one of these four states.N) r$r%r&r'r( CONNECTINGOPENCLOSINGCLOSEDr+r,r-r r 7s*@@(-a%Jgvvvr,r r,c6eZdZdZedddd:dZed;dZejddZ ed?dZ d@dZ dAdZ dBdZdCdBd!ZdCdBd"ZdDdEd(ZdFd)ZdFd*ZdGdHd,ZdId.ZdJd0ZdKd1ZdLd3ZdLd4ZdMd7ZdMd8ZdAd9ZdS)Nra Sans-I/O implementation of a WebSocket connection. Args: side: :attr:`~Side.CLIENT` or :attr:`~Side.SERVER`. state: Initial state of the WebSocket connection. max_size: Maximum size of incoming messages in bytes. :obj:`None` disables the limit. You may pass a ``(max_message_size, max_fragment_size)`` tuple to set different limits for messages and fragments when you expect long messages sent in short fragments. logger: Logger for this connection; depending on ``side``, defaults to ``logging.getLogger("websockets.client")`` or ``logging.getLogger("websockets.server")``; see the :doc:`logging guide <../../topics/logging>` for details. iN)statemax_sizeloggersiderr5r r6*tuple[int | None, int | None] | int | Noner7LoggerLike | NonereturnNonectj|_ |.tjd|j}||_ |tj |_ ||_ ||_ t|ts||dc|_|_n|\|_|_d|_d|_d|_g|_d|_d|_d|_d|_d|_ d|_t7|_g|_g|_||_ tC|j d|_"dS)Nz websockets.F)#uuiduuid4idlogging getLoggernamelowerr7 isEnabledForDEBUGdebugr8r5 isinstanceintmax_message_sizemax_fragment_size current_sizeexpect_continuation_frameorigin extensions subprotocol close_rcvd close_sentclose_rcvd_then_sent handshake_exceof_sentrreadereventswritesparseparsernext parser_exc)selfr8r5r6r7s r-__init__zProtocol.__init__YsU"Z\\B >&'HTY__5F5F'H'HIIF"( )((77    h $ $ E(8
s  ) A ?@@ @ ;d " "I 0E0E0G0GIIJJ J-0& gtS1122222r,c |jrtd|jtur.t d|jj| |_|tt||dS)a Send a `Binary frame`_. .. _Binary frame: https://datatracker.ietf.org/doc/html/rfc6455#section-5.6 Parameters: data: payload containing arbitrary binary data. fin: FIN bit; set it to :obj:`False` if this is the first frame of a fragmented message. Raises: ProtocolError: If a fragmented message is in progress. rrN) rMr r`r1r r5rCrDrrrrs r- send_binaryzProtocol.send_binaryUs  ) A ?@@ @ ;d " "I 0E0E0G0GIIJJ J-0& is3344444r,rjrfCloseCode | int | Nonerkstrc|jtur.td|jj|2|dkrt dttj d}d}n$t||}| }| tt||jJ||_t |_dS)a_ Send a `Close frame`_. .. _Close frame: https://datatracker.ietf.org/doc/html/rfc6455#section-5.5.1 Parameters: code: close code. reason: close reason. Raises: ProtocolError: If the code isn't valid or if a reason is provided without a code. rNrjz#cannot send a reason without a coder,)r`r1r r5rCrDr rrNO_STATUS_RCVD serializerrrrQrRr2r]rfrkcloserrs r- send_closezProtocol.send_closels$ ;d " "I 0E0E0G0GIIJJ J <||#$IJJJ)2B77EDD$''E??$$D h--...&&& r,c|jtur<|jtur.td|jj|tt|dS)z Send a `Ping frame`_. .. _Ping frame: https://datatracker.ietf.org/doc/html/rfc6455#section-5.5.2 Parameters: data: payload containing arbitrary binary data. rN) r`r1r2r r5rCrDrrrrvs r- send_pingzProtocol.send_pingf ;d " "t{''A'AI 0E0E0G0GIIJJ J gt,,-----r,c|jtur<|jtur.td|jj|tt|dS)z Send a `Pong frame`_. .. _Pong frame: https://datatracker.ietf.org/doc/html/rfc6455#section-5.5.3 Parameters: data: payload containing arbitrary binary data. rN) r`r1r2r r5rCrDrrrrvs r- send_pongzProtocol.send_pongrr,CloseCode | intc|jtur}|tjkrmt ||}|}|tt|||_ |j d|_ t|_|j tur|js|||_t'|jdS)a? `Fail the WebSocket connection`_. .. _Fail the WebSocket connection: https://datatracker.ietf.org/doc/html/rfc6455#section-7.1.7 Parameters: code: close code reason: close reason Raises: ProtocolError: If the code isn't valid. NT)r5r1rrerrrrrrRrQrSr2r8r)rUsend_eofdiscardrZr[rs r-failz Protocol.fails& :  y111dF++((h 5 5666"' ?.04D-$ 9  t}  MMOOO llnn  T[r, list[Event]c$|jgc}|_|S)a* Fetch events generated from data received from the network. Call this method immediately after any of the ``receive_*()`` methods. Process resulting events, likely by passing them to the application. Returns: Events read from the connection. )rW)r]rWs r-events_receivedzProtocol.events_receiveds#k2  r, list[bytes]c$|jgc}|_|S)a Obtain data to send to the network. Call this method immediately after any of the ``receive_*()``, ``send_*()``, or :meth:`fail` methods. Write resulting data to the connection. The empty bytestring :data:`~websockets.protocol.SEND_EOF` signals the end of the data stream. When you receive it, half-close the TCP connection. Returns: Data to write to the connection. )rX)r]rXs r- data_to_sendzProtocol.data_to_sends"#k2  r,c|jturdS|jturdS|jturdS|jtusJ|jS)a Tell if the TCP connection is expected to close soon. Call this method immediately after any of the ``receive_*()``, ``send_close()``, or :meth:`fail` methods. If it returns :obj:`True`, schedule closing the TCP connection after a short timeout if the other side hasn't already closed it. Returns: Whether the TCP connection is expected to close soon. FT)r5r1r2r3r0rUras r-close_expectedzProtocol.close_expectedsVD :  5 : 4 :  5zZ''''}r,Generator[None]c#jK |jEd{Vr0|jr|jdt dd}|j|j|j}n|j|jz }|j||j}nt||j}tj |jj |j tu||jEd{V}|jr|jd|||#t $r?}|t$jt)|||_Yd}~nKd}~wt$r?}|t$jt)|||_Yd}~nd}~wt.$r@}|t$j|jd|j||_Yd}~nd}~wt6$rX}||j|t$jt)|||_Yd}~n\d}~wt<$rL}|jdd |t$j ||_Yd}~nd}~wwxYwdVtCd ) a- Parse incoming data into frames. :meth:`receive_data` and :meth:`receive_eof` run this generator coroutine until it needs more data or reaches EOF. :meth:`parse` never raises an exception. Instead, it sets the :attr:`parser_exc` and yields control. TN< EOFzunexpected end of stream)maskr6rOz< %sz at position z parser failed)exc_infoz"parse() shouldn't step after error)"rVat_eofrGr7EOFErrorrJrLrKminrrY read_exactr8r)rO recv_framer rrPROTOCOL_ERRORrr\reUnicodeDecodeError INVALID_DATArkstartr set_current_sizeMESSAGE_TOO_BIG ExceptionerrorINTERNAL_ERRORAssertionError)r]r6framerps r-rYzProtocol.parse2s= "% '#{1133333333?z3 ))'222 ##=>>>(4(0#'#8#'#84;L#L)5'#'#9#&x1G#H#H $);K*f,%# $$$:5K%%fe444&&&K% 'N " " " IIi.C 9 9 9!DOOOOOO " " " IIi0#c(( ; ; ;!DOOOOOO! " " " IIi,.U.U#).U.U V V V!DOOOOOO " " "  !2 3 3 3 IIi/S : : :!DOOOOOO " " " K  o  = = = IIi. / / /!DOOOOOO  " ABBBsJDD J4E J$4F J+6G&& J3AI JAJJc#K|jtup |jtu|jksJ|jEd{Vs8|j|jEd{V8|jr|j d|jtur"|jtur| t|_dVtd)z Discard incoming data. This coroutine replaces :meth:`parse`: - after receiving a close frame, during a normal closure (1.4); - after sending a close frame, during an abnormal closure (7.1.7). Nrz"discard() shouldn't step after EOF)r8r)r5r0rUrVrrrGr7r*rr3rras r-rzProtocol.discards V#?tzZ'?T]SSSS"k0022222222 " K   ! ! !#k0022222222 " : ' K  g & & & 9  4:Z#?#? MMOOO  ABBBr,rrct|jtus|jtur8|jt d|jst |j|_n|jturJ|jt d|jr d|_n|xjt |jz c_nq|jtur1tt|j}| |n2|jturn"|jturtj|j|_|jt$ur|jJd|_|jt d|jt*urL| tt|j|j|_d|_t$|_|jt.ur|||_t7|jnt9d|jd|j|dS) z- Process an incoming frame. NrrFzincomplete fragmented messageTzunexpected opcode: 02x)opcoderrrLr r|lenrrrrrrrrrrYrQr5r2rRrSr1r8r)rrrZr[rrWappend)r]r pong_frames r-rzProtocol.recv_frames <7 " "eli&?&? ,#$CDDD9 4$' OO! \W $ $ (#$CDDDy 5$(!!!!S__4!!! \W $ $w 33J OOJ ' ' ' ' \W $ $  \X % %$k%*55DOzW$$222,1) ,#$CDDDzT!! h ; ;<<<"&/,0)$ yF"" ,,..DK     !!Iu|!I!I!IJJ J 5!!!!!r,c|jr|jd||j||jt u|jdS)Nz> %s)rrO)rGr7rXrrr8r*rO)r]rs r-rzProtocol.send_framesm : - K  fe , , ,  OOY&(?        r,c|jrJd|_|jr|jd|jt dS)NTz> EOF)rUrGr7rXrr!ras r-rzProtocol.send_eofsS=    : ' K  g & & & 8$$$$$r,) r8rr5r r6r9r7r:r;r<)r;r )r5r r;r<)r;rc)r;rh)r;r)rrrsr;r<)r;r<)rrrr|r}r;r<)T)Nrj)rfrrkrr;r<)rrrr;r<)rj)rfrrkrr;r<)r;r)r;r)r;r})r;r)rrr;r<)r$r%r&r'r1r^propertyr5setterrgrlrqrwr{rrrrrrrrrrrYrrrrr+r,r-rrGsn*?D$( E1E1E1E1E1E1N   X  \\ (((X($***X*$!!!X!J *3333033333.55555."""""H.... .... -----b     ())))ZMCMCMCMC^CCCCcollections.abcr exceptionsrrrr r r rOr framesr rrrrrrrrrhttp11rrstreamsrtypingrrrr__all__EventIntEnumrr)r*r r0r1r2r3r!rr+r,r-rsb""""""  %%%%%%"!!!!!                        &%%%%%%%!!!!!!>>>>>>>>>>>>    (U"?4<  11111DL111    z -  Ey %y %y %y %y %y %y %y %y %y %r,