vggddlmZddlZddlZddlZddlmZddlmZddl m Z m Z m Z m Z mZmZddlmZddlmZmZmZmZmZmZmZmZmZmZdd lmZmZdd l m!Z!dd lm"Z"m#Z#m$Z$gd Z%eeeefZ& Gd dej'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)Union)ConnectionClosedConnectionClosedErrorConnectionClosedOK InvalidState PayloadTooBig ProtocolError) Extension) OK_CLOSE_CODES OP_BINARYOP_CLOSEOP_CONTOP_PINGOP_PONGOP_TEXTClose CloseCodeFrame)RequestResponse) StreamReader) LoggerLikeOrigin Subprotocol)ProtocolSideStateSEND_EOFc.eZdZdZed\ZZdS)rz6A WebSocket connection is either a server or a client.N)__name__ __module__ __qualname____doc__rangeSERVERCLIENTP/var/www/pixelcanvas.ch/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 9s*@@(-a%Jgvvvr,r r,c6eZdZdZedddd7dZed8dZejd9dZed:dZ ed;dZ eddZ d?dZd@d?dZd@d?d ZdAdBd%Zd=d&Zd=d'ZdCdDd)ZdEd+ZdFd-ZdGd.ZdHd0ZdHd1ZdId4ZdId5Zd>d6ZdS)Jra6 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. 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 int | Noner7LoggerLike | NonereturnNonecRtj|_ |.tjd|j}||_ |tj |_ ||_ ||_ ||_ d|_d|_d|_g|_d|_d|_d|_d|_d|_ d|_t1|_g|_g|_||_t=|jd|_dS)Nz websockets.F) uuiduuid4idlogging getLoggernamelowerr7 isEnabledForDEBUGdebugr8r5r6cur_sizeexpect_continuation_frameorigin extensions subprotocol close_rcvd close_sentclose_rcvd_then_sent handshake_exceof_sentrreadereventswritesparseparsernext parser_exc)selfr8r5r6r7s r-__init__zProtocol.__init__Ys"Z\\B >&'HTY__5F5F'H'HIIF"( )((77   ! %) */&&* +-/3)-(,15!04  #nn #% #% jjll  T[,0r,c|jS)a State of the WebSocket connection. Defined in 4.1_, 4.2_, 7.1.3_, and 7.1.4_ of :rfc:`6455`. .. _4.1: https://datatracker.ietf.org/doc/html/rfc6455#section-4.1 .. _4.2: https://datatracker.ietf.org/doc/html/rfc6455#section-4.2 .. _7.1.3: https://datatracker.ietf.org/doc/html/rfc6455#section-7.1.3 .. _7.1.4: https://datatracker.ietf.org/doc/html/rfc6455#section-7.1.4 )_staterYs r-r5zProtocol.states {r,cb|jr |jd|j||_dS)Nz= connection is %s)rGr7rCr\)rYr5s r-r5zProtocol.states3 : @ K  2EJ ? ? ? r,c`|jturdS|j tjS|jjS)z WebSocket close code received from the remote endpoint. Defined in 7.1.5_ of :rfc:`6455`. .. _7.1.5: https://datatracker.ietf.org/doc/html/rfc6455#section-7.1.5 :obj:`None` if the connection isn't closed yet. N)r5r3rMrABNORMAL_CLOSUREcoder]s r- close_codezProtocol.close_codes2 :V # #4 _ $- -?' 'r, str | NonecL|jturdS|jdS|jjS)a WebSocket close reason received from the remote endpoint. Defined in 7.1.6_ of :rfc:`6455`. .. _7.1.6: https://datatracker.ietf.org/doc/html/rfc6455#section-7.1.6 :obj:`None` if the connection isn't closed yet. N)r5r3rMreasonr]s r- close_reasonzProtocol.close_reasons. :V # #4 _ $2?) )r,rc |jtus Jd|j5|j.|jjt vr|jjt vrt }nt}||j|j|j}|j |_ |S)a Exception to raise when trying to interact with a closed connection. Don't raise this exception while the connection :attr:`state` is :attr:`~websockets.protocol.State.CLOSING`; wait until it's :attr:`~websockets.protocol.State.CLOSED`. Indeed, the exception includes the close code and reason, which are known only once the connection is closed. Raises: AssertionError: If the connection isn't closed yet. zconnection isn't closed yet) r5r3rMrNrarr rrOrX __cause__)rYexc_typeexcs r- close_exczProtocol.close_excs zV###%B### O '+$66$66)HH,H ( O O  %! !    r,databytescb|j|t|jdS)aH Receive data from the network. After calling this method: - You must call :meth:`data_to_send` and send this data to the network. - You should call :meth:`events_received` and process resulting events. Raises: EOFError: If :meth:`receive_eof` was called earlier. N)rR feed_datarWrVrYrms r- receive_datazProtocol.receive_datas0 d### T[r,c||jjrdS|jt|jdS)a Receive the end of the data stream from the network. After calling this method: - You must call :meth:`data_to_send` and send this data to the network; it will return ``[b""]``, signaling the end of the stream, or ``[]``. - You aren't expected to call :meth:`events_received`; it won't return any new events. :meth:`receive_eof` is idempotent. N)rReoffeed_eofrWrVr]s r- receive_eofzProtocol.receive_eofs? ;?  F  T[r,finboolc |jstd|jtur.t d|jj| |_|tt||dS)a Send a `Continuation frame`_. .. _Continuation frame: https://datatracker.ietf.org/doc/html/rfc6455#section-5.6 Parameters: data: payload containing the same kind of data as the initial frame. fin: FIN bit; set it to :obj:`True` if this is the last frame of a fragmented message and to :obj:`False` otherwise. Raises: ProtocolError: If a fragmented message isn't in progress. unexpected continuation frameconnection is N) rIr r\r1r r5rCrD send_framerrrYrmrws r-send_continuationzProtocol.send_continuation#s"- A ?@@ @ ;d " "I 0E0E0G0GIIJJ J-0& gtS1122222r,Tc |jrtd|jtur.t d|jj| |_|tt||dS)a Send a `Text frame`_. .. _Text frame: https://datatracker.ietf.org/doc/html/rfc6455#section-5.6 Parameters: data: payload containing text encoded with UTF-8. 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. expected a continuation framer{N) rIr r\r1r r5rCrDr|rrr}s r- send_textzProtocol.send_text;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. rr{N) rIr r\r1r r5rCrDr|rrr}s r- send_binaryzProtocol.send_binaryRs  ) A ?@@ @ ;d " "I 0E0E0G0GIIJJ J-0& is3344444r,rerarfstrc|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. r{Nrez#cannot send a reason without a coder,)r\r1r r5rCrDr rrNO_STATUS_RCVD serializer|rrrMrNr2rYrarfcloserms r- send_closezProtocol.send_closeis$ ;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. r{N) r\r1r2r r5rCrDr|rrrqs 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. r{N) r\r1r2r r5rCrDr|rrrqs r- send_pongzProtocol.send_pongrr,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)r5r1rr`rrr|rrrNrMrOr2r8r)rQsend_eofdiscardrVrWrs 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. )rS)rYrSs 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. )rT)rYrTs 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)r5r1r2r3r0rQr]s r-close_expectedzProtocol.close_expectedsVD :  5 : 4 :  5zZ''''}r,Generator[None]c#K |jEd{Vr0|jr|jdt d|jd}n|j|j}n|j|jz }tj|jj |j tu||j Ed{V}|jr|jd|| |#t$r?}|t jt%|||_Yd}~nKd}~wt$r?}|t jt%|||_Yd}~nd}~wt*$r@}|t j|jd|j||_Yd}~nd}~wt2$rX}||j|t jt%|||_Yd}~n\d}~wt8$rL}|jdd |t j||_Yd}~nd}~wwxYwdVt?d ) 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)maskr6rKz< %sz at position z parser failed)exc_infoz"parse() shouldn't step after error) rRat_eofrGr7EOFErrorr6rHrrU read_exactr8r)rK recv_framer rrPROTOCOL_ERRORrrXr`UnicodeDecodeError INVALID_DATArfstartr set_current_sizeMESSAGE_TOO_BIG ExceptionerrorINTERNAL_ERRORAssertionError)rYr6framerks r-rUzProtocol.parse/s6 " '#{1133333333?z3 ))'222 ##=>>>=(#HH]*#}HH#}t}4E88 I96G I9 AH  I9-AI44I9c#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)r5r0rQrRrrrGr7r*rr3rr]s r-rzProtocol.discardws 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. NrrzFzincomplete fragmented messageTzunexpected opcode: 02x)opcoderrrHr rwlenrmrrrrr|rrrUrMr5r2rNrOr1r8r)rrrVrWrrSappend)rYr pong_frames r-rzProtocol.recv_frames <7 " "eli&?&?}(#$CDDD9 0 #EJ \W $ $}$#$CDDDy 1 $  UZ0 \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)rrK)rGr7rTrrr8r*rK)rYrs r-r|zProtocol.send_framesm : - K  fe , , ,  OOY&(?        r,c|jrJd|_|jr|jd|jt dS)NTz> EOF)rQrGr7rTrr!r]s r-rzProtocol.send_eofsS=    : ' K  g & & & 8$$$$$r,) r8rr5r r6r9r7r:r;r<)r;r )r5r r;r<)r;r9)r;rc)r;r)rmrnr;r<)r;r<)rmrnrwrxr;r<)T)Nre)rar9rfrr;r<)re)rarrfrr;r<)r;r)r;r)r;rx)r;r)rrr;r<)r$r%r&r'r1rZpropertyr5setterrbrgrlrrrvr~rrrrrrrrrrUrrr|rr+r,r-rrIsn  &$$( B1B1B1B1B1B1H   X  \\ (((X($***X*$!!!X!J *3333033333.55555."""""H.... .... -----b     ())))ZFCFCFCFCPCCCCcollections.abcrtypingr exceptionsrrr r r r rKr framesrrrrrrrrrrhttp11rrstreamsrrrr__all__EventIntEnumrr)r*r r0r1r2r3r!rr+r,r-rsp""""""  %%%%%%"!!!!!                        &%%%%%%%!!!!!!3333333333    gx&'?4<  11111DL111    z -  Em %m %m %m %m %m %m %m %m %m %r,