o KK&hg@s@ddlmZddlZddlZddlZddlmZddlmZddl m Z m Z m Z m Z mZmZddlmZddlmZmZmZmZmZmZmZmZmZmZdd lmZmZdd l m!Z!dd lm"Z"m#Z#m$Z$gd Z%eeeefZ& Gd ddej'Z(e(j)Z)e(j*Z*Gdddej'Z+e+j,Z,e+j-Z-e+j.Z.e+j/Z/dZ0 GdddZ1dS)) 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@seZdZdZed\ZZdS)rz6A WebSocket connection is either a server or a client.N)__name__ __module__ __qualname____doc__rangeSERVERCLIENTr)r)F/var/www/html/venv/lib/python3.10/site-packages/websockets/protocol.pyr/src@s eZdZdZed\ZZZZdS)rz6A WebSocket connection is in one of these four states.N) r"r#r$r%r& CONNECTINGOPENCLOSINGCLOSEDr)r)r)r*r9src@seZdZdZeddddMddZedNddZejdOddZedPddZ edQddZ edRddZ dSddZ dTd d!Z dUd$d%ZdVdUd'd(ZdVdUd)d*ZdWdXd/d0ZdSd1d2ZdSd3d4ZdYdZd6d7Zd[d9d:Zd\dd?Zd^dAdBZd^dCdDZd_dGdHZd_dIdJZdTdKdLZdS)`ra6 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_sizeloggersiderr1rr2 int | Noner3LoggerLike | NonereturnNonecCst|_ |durtd|j}||_ |tj |_ ||_ ||_ ||_ d|_d|_d|_g|_d|_d|_d|_d|_d|_ d|_t|_g|_g|_||_t|jd|_dS)Nz websockets.F) uuiduuid4idlogging getLoggernamelowerr3 isEnabledForDEBUGdebugr4r1r2cur_sizeexpect_continuation_frameorigin extensions subprotocol close_rcvd close_sentclose_rcvd_then_sent handshake_exceof_sentrreadereventswritesparseparsernext parser_exc)selfr4r1r2r3r)r)r*__init__Ys6    zProtocol.__init__cCs|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 )_staterTr)r)r*r1s zProtocol.statecCs |jr |jd|j||_dS)Nz= connection is %s)rBr3r>rV)rTr1r)r)r*r1s cCs&|jturdS|jdurtjS|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)r1r/rHrABNORMAL_CLOSUREcoderWr)r)r* close_codes  zProtocol.close_code str | NonecCs$|jturdS|jdurdS|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)r1r/rHreasonrWr)r)r* close_reasons  zProtocol.close_reasonrcCsf|jtus Jd|jdur"|jdur"|jjtvr"|jjtvr"t}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 yetN) r1r/rHrIrYr rrrJrS __cause__)rTexc_typeexcr)r)r* close_excs    zProtocol.close_excdatabytescCs|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)rM feed_datarRrQrTrcr)r)r* receive_datas zProtocol.receive_datacCs$|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)rMeoffeed_eofrRrQrWr)r)r* receive_eofs zProtocol.receive_eoffinboolcCsL|jstd|jturtd|jj| |_|t t ||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) rDr rVr-r r1r>r? send_framerrrTrcrkr)r)r*send_continuation#s  zProtocol.send_continuationTcCL|jrtd|jturtd|jj| |_|t t ||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 framernN) rDr rVr-r r1r>r?rorrrpr)r)r* send_text;  zProtocol.send_textcCrr)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. rsrnN) rDr rVr-r r1r>r?rorrrpr)r)r* send_binaryRruzProtocol.send_binaryr\rYr]strcCs|jturtd|jj|dur%|dkrtdttj d}d}n t||}| }| t t ||jdus=J||_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. rnNr\z#cannot send a reason without a coder0)rVr-r r1r>r?r rrNO_STATUS_RCVD serializerorrrHrIr.rTrYr]closercr)r)r* send_closeis    zProtocol.send_closecC>|jtur|jturtd|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. rnN) rVr-r.r r1r>r?rorrrfr)r)r* send_ping zProtocol.send_pingcCr})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. rnN) rVr-r.r r1r>r?rorrrfr)r)r* send_pongrzProtocol.send_pongintcCs|jtur)|tjkr)t||}|}|tt|||_ |j dur&d|_ t |_|j tur5|js5|||_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)r1r-rrXrryrorrrIrHrJr.r4r'rLsend_eofdiscardrQrRrzr)r)r*fails     z Protocol.fail list[Event]cC|jg}|_|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. )rN)rTrNr)r)r*events_receiveds zProtocol.events_received list[bytes]cCr)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. )rO)rTrOr)r)r* data_to_sendszProtocol.data_to_sendcCs>|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)r1r-r.r/r,rLrWr)r)r*close_expecteds "  zProtocol.close_expectedGenerator[None]c cszO |jEdHr|jr|jdtd|jdur d}n|jdur)|j}n|j|j}tj|jj |j t u||j dEdH}|jrK|jd|| |qtyn}z|tjt|||_WYd}~nd}~wty}z|tjt|||_WYd}~nld}~wty}z|tj|jd|j||_WYd}~nJd}~wty}z||j|tjt|||_WYd}~n'd}~wty}z|jjddd |tj||_WYd}~nd}~wwdVtd ) 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)maskr2rFz< %sz at position z parser failed)exc_infoz"parse() shouldn't step after error) rMat_eofrBr3EOFErrorr2rCrrP read_exactr4r'rF recv_framer rrPROTOCOL_ERRORrwrSrXUnicodeDecodeError INVALID_DATAr]startr set_current_sizeMESSAGE_TOO_BIG ExceptionerrorINTERNAL_ERRORAssertionError)rTr2framerar)r)r*rP/s^          zProtocol.parseccs|jtup |jtu|jksJ|jEdHs%|j|jEdHr|jr.|j d|jt ur<|jtur<| t |_dVt d)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)r4r'r1r,rLrMrrrBr3r(rr/rrWr)r)r*rws  zProtocol.discardrrcCsj|jtus |jtur|jdurtd|jst|j|_n|jtur=|jdur+td|jr2d|_n{|jt|j7_np|jt urNt t |j}| |n_|jt urTnY|jt urt|j|_|jturo|jduslJd|_|jdurxtd|jtur| t t |j|j|_d|_t|_|jtur|||_t|jn td|jd|j|dS) z- Process an incoming frame. NrsrmFzincomplete fragmented messageTzunexpected opcode: 02x)opcoderrrCr rklenrcrrrrrorrrPrHr1r.rIrJr-r4r'rrrQrRrrNappend)rTr pong_framer)r)r*rsF               zProtocol.recv_framecCs6|jr |jd||j|j|jtu|jddS)Nz> %s)rrF)rBr3rOrryr4r(rF)rTrr)r)r*roszProtocol.send_framecCs2|jrJd|_|jr|jd|jtdS)NTz> EOF)rLrBr3rOrr rWr)r)r*rs  zProtocol.send_eof) r4rr1rr2r5r3r6r7r8)r7r)r1rr7r8)r7r5)r7r[)r7r)rcrdr7r8)r7r8)rcrdrkrlr7r8)T)Nr\)rYr5r]rwr7r8)r\)rYrr]rwr7r8)r7r)r7r)r7rl)r7r)rrr7r8)r"r#r$r%r-rUpropertyr1setterrZr^rbrgrjrqrtrvr|r~rrrrrrPrrrorr)r)r)r*rIsBD      %      $   1   - H  R r)2 __future__renumr<r9collections.abcrtypingr exceptionsrrrr r r rFr framesr rrrrrrrrrhttp11rrstreamsrrrr__all__EventIntEnumrr'r(rr,r-r.r/r rr)r)r)r*s4     0