jDjddlmZddlZddlZddlZddlmZddlmZm 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 ejNZ(e(jRZ)e(jTZ*GddejNZ+e+jXZ,e+jZZ-e+j\Z.e+j^Z/dZ0 GddZ1y)) annotationsN) Generator)ConnectionClosedConnectionClosedErrorConnectionClosedOK InvalidState PayloadTooBig ProtocolError) Extension) BINARYCLOSECONTOK_CLOSE_CODESPINGPONGTEXTClose CloseCodeFrame)RequestResponse) StreamReader) BytesLike LoggerLikeOrigin Subprotocol)ProtocolSideStateSEND_EOFc&eZdZdZed\ZZy)rz6A WebSocket connection is either a server or a client.N)__name__ __module__ __qualname____doc__rangeSERVERCLIENTZ/var/www/html/streamforge-license/venv/lib/python3.12/site-packages/websockets/protocol.pyrr-s@1XNFFr,rc*eZdZdZed\ZZZZy)r z6A WebSocket connection is in one of these four states.N) r$r%r&r'r( CONNECTINGOPENCLOSINGCLOSEDr+r,r-r r 7s@(-a%Jgvr,r r,c2eZdZdZeddd ddZeddZejddZeddZ ed d Z ed!d Z d"d Z d#d Z d$d Zd%d$dZd%d$dZd&d'dZd(dZd(dZd)d*dZd+dZd,dZd-dZd.dZd.dZd/dZd/dZd#dZy)0ra 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_sizeloggerctj|_ |0tjd|j j }||_ |jtj|_ ||_ ||_ t|ts||dc|_|_n|\|_|_d|_d|_d|_g|_d|_d|_d|_d|_d|_ d|_t7|_g|_g|_|j?|_ tC|j@d|_"y)Nz websockets.F)#uuiduuid4idlogging getLoggernamelowerr7 isEnabledForDEBUGdebugsider5 isinstanceintmax_message_sizemax_fragment_size current_sizeexpect_continuation_frameorigin extensions subprotocol close_rcvd close_sentclose_rcvd_then_sent handshake_exceof_sentrreadereventswritesparseparsernext parser_exc)selfrCr5r6r7s r-__init__zProtocol.__init__Ys:"ZZ\B >&&TYY__5F4G'HIF"( )((7    h $(8
r\)rYr5s r-r5zProtocol.states) :: KK  2EJJ ? r,c|jtury|jtjS|jj S)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_codes9 ::V # __ $-- -??'' 'r,cn|jtury|jy|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_reasons0 ::V # __ $??)) )r,cd|jtusJd|jK|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) r5r3rMrNrarrrrOrX __cause__)rYexc_typeexcs r- close_exczProtocol.close_excs zzV#B%BB# OO '+$$6$$6)H,H ( OO OO  % %!    r,cd|jj|t|jy)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_datarWrVrYdatas r- receive_datazProtocol.receive_datas" d# T[[r,c|jjry|jjt|jy)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. ;;??   T[[r,c|js td|jtur0t d|j j j| |_|jtt||y)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 r5r>r? send_framerrrYrofins r-send_continuationzProtocol.send_continuation&sh"-- ?@ @ ;;d " 0E0E0G/HIJ J-0& dD#./r,c|jr td|jtur0t d|j j j| |_|jtt||y)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 framerwN) rIr r\r1r r5r>r?rxrrrys r- send_textzProtocol.send_text>sh  ) ) ?@ @ ;;d " 0E0E0G/HIJ J-0& dD#./r,c|jr td|jtur0t d|j j j| |_|jtt||y)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. r}rwN) rIr r\r1r r5r>r?rxrr rys r- send_binaryzProtocol.send_binaryUsh  ) ) ?@ @ ;;d " 0E0E0G/HIJ J-0& fdC01r,c|jtur0td|jjj |-|dk7r t dttjd}d}nt||}|j}|jtt||jJ||_t |_y)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. rwNrdz#cannot send a reason without a coder,)r\r1r r5r>r?r rrNO_STATUS_RCVD serializerxrrrMrNr2rYrarecloseros r- send_closezProtocol.send_closels$ ;;d " 0E0E0G/HIJ J <|#$IJJ)22B7ED$'E??$D eT*+&&& r,c|jturB|jtur0td|jj j |jtt|y)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. rwN) r\r1r2r r5r>r?rxrrrns r- send_pingzProtocol.send_pingR ;;d "t{{''A 0E0E0G/HIJ J dD)*r,c|jturB|jtur0td|jj j |jtt|y)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. rwN) r\r1r2r r5r>r?rxrrrns r- send_pongzProtocol.send_pongrr,c|jturs|tjk7r`t ||}|j }|j tt|||_ |jd|_ t|_|jtur|js|j!|j#|_t'|j$y)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`rrrxrrrNrMrOr2rCr)rQsend_eofdiscardrVrWrs r-failz Protocol.fails& :: y111dF+(eT 23"' ??.04D-$ 99 t}} MMO lln  T[[r,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#kk2  r,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"#kk2  r,c|jtury|jtury|jtury|jtusJ|j S)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_expectedsND ::  ::  :: zzZ'''}}r,c#K |jjEd{r2|jr|jjdt dd}|j 2|j |j }n|j |j z }|j%| |j}nt||j}tj|jj|jtu||jEd{}|jr|jjd||j|O727B#t $r;}|j#t$j&t)|||_Yd}~nKd}~wt$r;}|j#t$j,t)|||_Yd}~n d}~wt.$rJ}|j#t$j0|j2d|j4||_Yd}~nd}~wt6$rU}|j9|j |j#t$j:t)|||_Yd}~n\d}~wt<$rM}|jj?dd |j#t$j@||_Yd}~nd}~wwxYwdtCd w) 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. TNz< EOFzunexpected end of stream)maskr6rKz< %sz at position z parser failed)exc_infoz"parse() shouldn't step after error)"rRat_eofrBr7EOFErrorrFrHrGminrrU read_exactrCr)rK recv_framer rrPROTOCOL_ERRORstrrXr`UnicodeDecodeError INVALID_DATArestartr set_current_sizeMESSAGE_TOO_BIG ExceptionerrorINTERNAL_ERRORAssertionError)rYr6framerjs r-rUzProtocol.parse2sH= "#{{11333zz ))'2 ##=>>((4((0#'#8#8#'#8#84;L;L#L))5'#'#9#9#&x1G1G#H $);;KK**f,%# $::KK%%fe4&K34 " IIi..C 9!DOO " IIi00#c( ;!DOO! " IIi,, M#)).U V!DOO "  !2!2 3 IIi//S :!DOO " KK  o  = IIi.. /!DOO  " ABBsK2EEC1EEAEE K "0FK2 K $0GK2 K &AH+&K2+ K 7A JK2 K AKK2K  K2c#0K|jtuxs|jtu|jk(sJ|j j Ed{s=|j j|j j Ed{s=|jtur"|jtur|jt|_|jtur|jjddtd77w)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). Nzconnection closedz"discard() shouldn't step after EOF)rCr)r5r0rQrRrrr*rr3r7inforr]s r-rzProtocol.discards V#?tzzZ'?T]]SSS"kk00222 KK   !#kk00222 99 4::Z#? MMO 99  KK  0 1 ABB32s+ADD;DDDA %s)rrK)rBr7rTrrrCr*rK)rYrs r-rxzProtocol.send_framesS :: KK  fe ,  OOYY&(??   r,c|jrJd|_|jr|jjd|jj t y)NTz> EOF)rQrBr7rTrr!r]s r-rzProtocol.send_eofsA==   :: KK  g & 8$r,) rCrr5r r6z*tuple[int | None, int | None] | int | Noner7zLoggerLike | NonereturnNone)rr )r5r rr)rz int | None)rz str | None)rr)rozbytes | bytearrayrr)rr)rorrzboolrr)T)Nrd)razCloseCode | int | Nonererrr)rorrr)rd)razCloseCode | intrerrr)rz list[Event])rz list[bytes])rr)rzGenerator[None])rrrr)r$r%r&r'r1rZpropertyr5setterrbrfrkrprtr{r~rrrrrrrrrUrrrxrr+r,r-rrGs*?D$( E1E1 E1 = E1 " E1 E1N   \\ (($**$!!J *000.2."H+ + -b  ()ZMC^Crs"  %"   &!>>  (U"?4<<  1DLL1     zz --  Ey %y %r,