=Xcq\ddlmZddlZddlZddlZddlmZmZmZm Z m Z ddl m Z m Z mZmZmZmZddlmZddlmZmZmZmZmZmZmZmZmZddlmZm Z dd l!m"Z"dd lm#Z#m$Z$m%Z%gd Z&e ee efZ' Gd d ejPZ)e)jTZ*e)jVZ+GddejPZ,e,jZZ-e,j\Z.e,j^Z/e,j`Z0dZ1 GddZ2y)) annotationsN) GeneratorListOptionalTypeUnion)ConnectionClosedConnectionClosedErrorConnectionClosedOK InvalidState PayloadTooBig ProtocolError) Extension) OK_CLOSE_CODES OP_BINARYOP_CLOSEOP_CONTOP_PINGOP_PONGOP_TEXTCloseFrame)RequestResponse) StreamReader) LoggerLikeOrigin Subprotocol) ConnectionSideStateSEND_EOFc&eZdZdZed\ZZy)r!z6A WebSocket connection is either a server or a client.N)__name__ __module__ __qualname____doc__rangeSERVERCLIENT7/usr/lib/python3/dist-packages/websockets/connection.pyr!r!,s@1XNFFr.r!c*eZdZdZed\ZZZZy)r"z6A WebSocket connection is in one of these four states.N) r&r'r(r)r* CONNECTINGOPENCLOSINGCLOSEDr-r.r/r"r"6s@(-a%Jgvr.r"r.c0eZdZdZeddf ddZeddZejddZeddZ eddZ ed d Z d!d Z d"d Z d#d Zd$d#d Zd$d#dZd%d&dZd!dZd!dZd'd(dZd)dZd*dZd+dZd,dZd,dZd-dZd-dZd"dZy).r a5 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` to disable 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. iNcHtj|_ |0tjd|j j }||_ |jtj|_ ||_ ||_ ||_ d|_d|_d|_g|_d|_d|_d|_d|_d|_ d|_t1|_g|_g|_|j9|_t=|j:d|_y)Nz websockets.F) uuiduuid4idlogging getLoggernamelowerlogger isEnabledForDEBUGdebugsidestatemax_sizecur_sizeexpect_continuation_frameorigin extensions subprotocol close_rcvd close_sentclose_rcvd_then_sent handshake_exceof_sentrreadereventswritesparseparsernext parser_exc)selfrCrDrEr?s r/__init__zConnection.__init__Vs"ZZ\B >&&TYY__5F4G'HIF"( )((7   ! (, */&)- +-26,0+/48!37  #n #% #% jjl  T[[/3r.c|jS)zm WebSocket connection state. Defined in 4.1, 4.2, 7.1.3, and 7.1.4 of :rfc:`6455`. )_staterWs r/rDzConnection.states{{r.cv|jr&|jjd|j||_y)Nz= connection is %s)rBr?r=rZ)rWrDs r/rDzConnection.states) :: KK  2EJJ ? r.cn|jtury|jy|jjS)z `WebSocket close code`_. .. _WebSocket close code: https://www.rfc-editor.org/rfc/rfc6455.html#section-7.1.5 :obj:`None` if the connection isn't closed yet. N)rDr5rKcoder[s r/ close_codezConnection.close_codes0 ::V # __ $??'' 'r.cn|jtury|jy|jjS)z `WebSocket close reason`_. .. _WebSocket close reason: https://www.rfc-editor.org/rfc/rfc6455.html#section-7.1.6 :obj:`None` if the connection isn't closed yet. N)rDr5rKreasonr[s r/ close_reasonzConnection.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.connection.State.CLOSING`; wait until it's :attr:`~websockets.connection.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) rDr5rKrLr_rr r rMrV __cause__)rWexc_typeexcs r/ close_exczConnection.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)rP feed_datarUrTrWdatas r/ receive_datazConnection.receive_datas" d# T[[r.cb|jjt|jy)aw 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. - You aren't expected to call :meth:`events_received`; it won't return any new events. Raises: EOFError: if :meth:`receive_eof` was called earlier. N)rPfeed_eofrUrTr[s r/ receive_eofzConnection.receive_eofs   T[[r.c|js td| |_|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 frameN)rGr send_framerrrWrmfins r/send_continuationzConnection.send_continuations8"-- ?@ @-0& gtS12r.c|jr td| |_|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 frameN)rGrrtrrrus r/ send_textzConnection.send_text,s8  ) ) ?@ @-0& gtS12r.c|jr td| |_|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. ryN)rGrrtrrrus r/ send_binaryzConnection.send_binaryAs8  ) ) ?@ @-0& is34r.c|jr td||dk7r tdtdd}d}nt||}|j}|j t t |||_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 a fragmented message is being sent, if the code isn't valid, or if a reason is provided without a code ryNrbz#cannot send a reason without a codeir.) rGrr serializertrrrLr4rDrWr_rccloserms r/ send_closezConnection.send_closeVs|  ) ) ?@ @ <|#$IJJ$OED$'E??$D h-. r.cB|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. N)rtrrrls r/ send_pingzConnection.send_pingv gt,-r.cB|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. N)rtrrrls r/ send_pongzConnection.send_pongrr.c||jturR|dk7rMt||}|j}|j t t |||_t|_|jtur|js|j|j|_t|jy)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. r^N)rDr3rr~rtrrrLr4rCr+rOsend_eofdiscardrTrUrs r/failzConnection.fails& :: t|dF+(h 56"'$ 99 t}} MMO lln  T[[r.c.|jgc}|_|S)a7 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: List[Event]: Events read from the connection. )rQ)rWrQs r/events_receivedzConnection.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.connection.SEND_EOF` signals the end of the data stream. When you receive it, half-close the TCP connection. Returns: List[bytes]: Data to write to the connection. )rR)rWrRs r/ data_to_sendzConnection.data_to_sends"#kk2  r.cF|jtuxs|jduS)a Tell if the TCP connection is expected to close soon. Call this method immediately after any of the ``receive_*()`` 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: bool: Whether the TCP connection is expected to close soon. N)rDr4rNr[s r/close_expectedzConnection.close_expecteds#,zzW$F(:(:$(FFr.c#K |jjEd{r2|jr|jjdt d|j d}n2|j |j }n|j |j z }tj|jj|jtu||jEd{}|jr|jjd||j|77B#t$r,}|jdt!|||_Yd}~nd}~wt$r,}|jdt!|||_Yd}~nd}~wt$$r<}|jd |j&d |j(||_Yd}~nd}~wt*$r,}|jd t!|||_Yd}~nNd}~wt,$r?}|jj/d d |jd||_Yd}~nd}~wwxYwdt1dw)z Parse incoming data into frames. :meth:`receive_data` and :meth:`receive_eof` run this generator coroutine until it needs more data or reaches EOF. TN< EOFzunexpected end of stream)maskrErIz< %sir^iz at position iz parser failed)exc_infoiz"parse() shouldn't step after error)rPat_eofrBr?EOFErrorrErFrrS read_exactrCr+rI recv_framerrstrrVUnicodeDecodeErrorrcstartr ExceptionerrorAssertionError)rWrEframerhs r/rSzConnection.parses5 "#{{11333zz ))'2 ##=>>==(#H]]*#}}H#}}t}}C*A$C*(C*c|jtus|jturH|j t d|j r d|_nt |j|_n|jturV|j t d|j r d|_n|xjt |jz c_n|jtur-tt|j}|j|nM|jturn9|jtur tj|j|_|j"t$ur|j&Jd|_|j t d|j"t*urL|jtt|j|j |_d|_t$|_|j,t.ur|j1|j3|_t7|j4nt9d|jd|j:j=|y) z- Process an incoming frame. NryrsFzincomplete fragmented messageTzunexpected opcode: 02x)opcoderrrFrrvlenrmrrrrrtrrrSrKrDr4rLrMr3rCr+rrrTrUrrQappend)rWr pong_frames r/rzConnection.recv_frame[s <<7 "elli&?}}(#$CDDyy $ #EJJ \\W $}}$#$CDDyy $  UZZ0 \\W $w 3J OOJ ' \\W $  \\X %$kk%**5DOzzW$222,1)}}(#$CDDzzT! h ;<"&//,0)$ yyF"  ,,.DK  !#6u||C6H!IJ J 5!r.cJ|jtur#td|jjd|jr|j j d||j j|j|jtu|jy)Nz#cannot write to a WebSocket in the z statez> %s)rrI) rDr3r r=rBr?rRrr~rCr,rI)rWrs r/rtzConnection.send_frames| ::T !5djjoo5FfM  :: KK  fe ,  OOf!4O Q r.c|jrJd|_|jr|jjd|jj t y)NTz> EOF)rOrBr?rRrr#r[s r/rzConnection.send_eofsA==   :: KK  g & 8$r.) rCr!rDr"rE Optional[int]r?zOptional[LoggerLike]returnNone)rr")rDr"rr)rr)rz Optional[str])rr )rmbytesrr)rr)rmrrvboolrr)T)Nrb)r_rrcrrr)rb)r_intrcrrr)rz List[Event])rz List[bytes])rr)rzGenerator[None, None, None])rrrr)r&r'r(r)r3rXpropertyrDsetterr`rdrirnrqrwrzr|rrrrrrrrSrrrtrr-r.r/r r Fs $"''+ A4A4A4 A4 % A4  A4F \\ (("**"!!J &3,3*5*@ . .'V  (G4BCHCrs"  99"   &!33  gx&'A4<<  1DLL1     zz --  Ex %x %r.