https://t.me/RX1948
Server : Apache
System : Linux iad1-shared-b8-43 6.6.49-grsec-jammy+ #10 SMP Thu Sep 12 23:23:08 UTC 2024 x86_64
User : dh_edsupp ( 6597262)
PHP Version : 8.2.26
Disable Function : NONE
Directory :  /lib/python3/dist-packages/mercurial/__pycache__/

Upload File :
current_dir [ Writeable ] document_root [ Writeable ]

 

Current File : //lib/python3/dist-packages/mercurial/__pycache__/wireprotoframing.cpython-310.pyc
o

�]Lb��@sddlmZddlZddlZddlmZddlmZddlm	Z	ddl
mZmZmZm
Z
mZddlmZmZd	Zd
ZdZdZdZeeed
�ZdZdZdZdZdZdZd	ZdZeeeeeeeed�Z dZ!dZ"dZ#d	Z$e!e"e#e$d�Z%dZ&dZ'e&e'd�Z(dZ)dZ*e)e*d�Z+dZ,dZ-e,e-d�Z.dZ/dZ0e/e0d�Z1ee%ee(ee+eieieiee.ee1iZ2e�3d�Z4dd�Z5e	j6dd�Gdd�de7��Z8e	j6ddd�Gdd �d e7��Z9d!d"�Z:d#d$�Z;d%d&�Z<d'd(�Z=dedfd)d*�Z>d+d,�Z?efd-d.�Z@d/d0�ZAd[d1d2�ZBd3d4�ZCefd5d6�ZDGd7d8�d8e7�ZEGd9d:�d:e7�ZFGd;d<�d<e7�ZGGd=d>�d>e7�ZHGd?d@�d@e7�ZIGdAdB�dBe7�ZJGdCdD�dDeJ�ZKGdEdF�dFe7�ZLGdGdH�dHeL�ZMiZNgZOdIdJ�ZPGdKdL�dLe7�ZQGdMdN�dNeQ�ZRGdOdP�dPeQ�ZSdQdR�ZTdSdTgiZUGdUdV�dVe7�ZVGdWdX�dXe7�ZWGdYdZ�dZe7�ZXdS)\�)�absolute_importN�)�_)�getattr)�attr)�encoding�error�pycompat�util�wireprototypes)�cborutil�
stringutil�i���)sstream-begins
stream-endsencoded�����	)scommand-requestscommand-datascommand-responseserror-responsestext-outputsprogressssender-protocol-settingssstream-settings)snew�continuationsmores	have-data)r�eosz<HHcCs\dd�t�|�D�}g}d}||kr)||@r!|�|�|d|��|dK}||ksd�|�S)zFConvert a numeric flags value to a human value, using a mapping table.cSsi|]\}}||�qS�r)�.0�k�vrr�</usr/lib/python3/dist-packages/mercurial/wireprotoframing.py�
<dictcomp>~szhumanflags.<locals>.<dictcomp>r�<unknown 0x%02x>�|)r	�	iteritems�append�get�join)�mapping�value�namemap�flags�valrrr�
humanflags|s�
r)T)�slotsc@s@eZdZdZe��Ze��Ze��Ze��Z	e��Z
e��ZdS)�frameheaderz&Represents the data in a frame header.N)�__name__�
__module__�__qualname__�__doc__r�ib�length�	requestid�streamid�streamflags�typeidr'rrrrr+�sr+F)r*�reprc@sNeZdZdZe��Ze��Ze��Ze��Z	e��Z
e��Zej
dd��ZdS)�framezRepresents a parsed frame.cCshd|j}t�t�D]
\}}||jkr|}nq
dt|j�|j|jtt	|j
�|tt�|ji�|j
�fS)NrsHframe(size=%d; request=%d; stream=%d; streamflags=%s; type=%s; flags=%s))r5r	r �FRAME_TYPES�len�payloadr2r3r)�STREAM_FLAGSr4�FRAME_TYPE_FLAGSr"r')�self�typename�namer%rrr�__repr__�s 

�
���zframe.__repr__N)r,r-r.r/rr0r2r3r4r5r'r:r�	strmethodr@rrrrr7�sr7cCshttt|��}t�dt|��}|dd�|dd�<t�d|d|||�|d>|B|d<||dd�<|S)	z#Assemble a frame into a byte array.z<Irr�<HBBrrrN)�	bytearray�FRAME_HEADER_SIZEr9�struct�pack�	pack_into)r2r3r4r5r'r:r7�lrrr�	makeframe�s	rIc	Cs|�dd�}|\}}}}}}t|�}t|�}d}|�d�D]}	|	tvr*|t|	O}q|t|	�O}q|tvr:t|}nt|�}d}
t|}|�d�D]}	|	|vrV|
||	O}
qI|
t|	�O}
qI|�d�rrd�t�t	�
|dd����}nt	�|�}t|||||
|d�S)	a�Create a frame from a human readable string

    Strings have the form:

        <request-id> <stream-id> <stream-flags> <type> <flags> <payload>

    This can be used by user-facing applications and tests for creating
    frames easily without having to type out a bunch of constants.

    Request ID and stream IDs are integers.

    Stream flags, frame type, and flags can be specified by integer or
    named constant.

    Flags can be delimited by `|` to bitwise OR them together.

    If the payload begins with ``cbor:``, the following string will be
    evaluated as Python literal and the resulting object will be fed into
    a CBOR encoder. Otherwise, the payload is interpreted as a Python
    byte string literal.
    � rrrscbor:�N)r2r3r4r5r'r:)
�split�intr;r8r<�
startswithr#r�streamencoder
�evalpythonliteral�unescapestrrI)�s�fieldsr2r3r4�	frametype�
frameflagsr:�finalstreamflags�flag�
finalflags�
validflagsrrr�makeframefromhumanstring�s>

�
�rZcCsb|dd|dd|d}t�d|d�\}}}|d}|d	@d
?}|d@}t||||||�S)z�Parse a unified framing protocol frame header from a buffer.

    The header is expected to be in the buffer at offset 0 and the
    buffer is expected to be large enough to hold a full header.
    r�ri@rrBrr��r�)rE�unpack_fromr+)�data�framelengthr2r3r4�	typeflagsrTrUrrr�parseheader
s 
�rbcCs�tt�}|�|�}|dkrdS|tkrt�td�||f��t|�}|�|j�}t	|�|jkr=t�td�|jt	|�f��t
|j|j|j
|j|j|�S)z�Read a unified framing protocol frame from a file object.

    Returns a 3-tuple of (type, flags, payload) for the decoded frame or
    None if no frame is available. May raise if a malformed frame is
    seen.
    rNs+received incomplete frame: got %d bytes: %ss'frame length error: expected %d; got %d)rCrD�readintor�Abortrrb�readr1r9r7r2r3r4r5r')�fh�header�	readcount�hr:rrr�	readframe&s*
�����rjccs�d|i}|r||d<|r||d<d�t�|��}d}	d}	|s%|	tO}	n|	tO}	|r/|	tO}	||||�}
|t|
�7}t|
�|krM|t|�krM|	tO}	|j|t	|	|
d�V|	t@s\nq|r�	|�
t�}d}t|�tkrpt}	n
t
}	|�
d	�dks{J�d}|j|t|	|d�V|r�d
Sq`d
S)z�Create frames necessary to transmit a request to run a command.

    This is a generator of bytearrays. Each item represents a frame
    ready to be sent over the wire to a peer.
    �name�args�redirectrKrT�r2r5r'r:FrN)r#rrO�FLAG_COMMAND_REQUEST_NEW�!FLAG_COMMAND_REQUEST_CONTINUATION� FLAG_COMMAND_REQUEST_EXPECT_DATAr9� FLAG_COMMAND_REQUEST_MORE_FRAMESrI�FRAME_TYPE_COMMAND_REQUESTre�DEFAULT_MAX_FRAME_SIZE�FLAG_COMMAND_DATA_CONTINUATION�FLAG_COMMAND_DATA_EOS�FRAME_TYPE_COMMAND_DATA)�streamr2�cmd�args�datafh�maxframesize�redirectr_�offsetr'r:�donerrr�createcommandframesHs^�
��
���r�cCsJd�t�ddi��}|jr|�|�}d}|sdSnd}|j|tt||d�S)NrK�statussokTF�r2r5r'r:�encoded)r#rrO�streamsettingssent�encoderI�FRAME_TYPE_COMMAND_RESPONSE�"FLAG_COMMAND_RESPONSE_CONTINUATION)rxr2�overallr�rrr�createcommandresponseokframe�s
��r�ccsj�|��}d}	||||�}|t|�7}|t|�k}|r!t}nt}|j|t|||dkd�V|r4dSq)zACreate an empty payload frame representing command end-of-stream.rTrKr�N)�flushr9�FLAG_COMMAND_RESPONSE_EOSr�rIr�)rxr2r|r:r~�chunkrr'rrr�createcommandresponseeosframes�s(���r�cCs�d|j|jd�d�}dD]}t||�}|dur ||dt�|�<qd�t�|��}|jr4|�	|�}d}nd}|j
|tt||d	�S)
Nrm)surls	mediatype)r��location)�size�
fullhashes�fullhashseed�serverdercerts�servercadercertsr�rKTFr�)
�url�	mediatyperr	�bytestrr#rrOr�r�rIr�r�)rxr2�locationr_�ar%r:r�rrr�$createalternatelocationresponseframe�s,��
�
�r�ccsH�dd|id�}|r||dd<d�t�|��}|j|tt|d�VdS)N�error�message)r�r�rlrKrn)r#rrOrIr�r�)rxr2�messagerz�mr�rrr�createcommanderrorresponse�s����r�ccsF�t|�tks	J�d�t�|d|igd���}|j|td|d�VdS)NrK�msg)�typer�rrn)r9rtr#rrOrI�FRAME_TYPE_ERROR_RESPONSE)rxr2�msg�errtyper:rrr�createerrorframes����	�r�ccs��g}|D]V\}}}t|t�std��|D]}t|t�s td��q|D]}	t|	t�s.td��q#|�dd��d�}dd�|D�}dd�|D�}d	|i}
|rP||
d
<|rV||
d<|�|
�qd�t�|��}t	|�|krntd
��|j
|td|d�VdS)a�Create a text output frame to render text to people.

    ``atoms`` is a 3-tuple of (formatting string, args, labels).

    The formatting string contains ``%s`` tokens to be replaced by the
    corresponding indexed entry in ``args``. ``labels`` is an iterable of
    formatters to be applied at rendering time. In terms of the ``ui``
    class, each atom corresponds to a ``ui.write()``.
    s!must use bytes formatting stringssmust use bytes for argumentssmust use bytes for labels�ascii�replacecS�g|]}|�dd��d��qS)zutf-8r���decoder�)rr�rrr�
<listcomp>7�z)createtextoutputframe.<locals>.<listcomp>cSr�)r��strictr�)rrHrrrr�:r�r�rlslabelsrKs$cannot encode data in a single framerrnN)�
isinstance�bytes�
ValueErrorr�r�r!r#rrOr9rI�FRAME_TYPE_TEXT_OUTPUT)rxr2�atomsr|�	atomdicts�
formattingrz�labels�arg�label�atomr:rrr�createtextoutputframes>�

�
��r�c@s,eZdZdZefdd�Zdd�Zdd�ZdS)	�bufferingcommandresponseemittera�Helper object to emit command response frames intelligently.

    Raw command response data is likely emitted in chunks much smaller
    than what can fit in a single frame. This class exists to buffer
    chunks until enough data is available to fit in a single frame.

    TODO we'll need something like this when compression is supported.
    So it might make sense to implement this functionality at the stream
    level.
    cCs"||_||_||_g|_d|_dS�Nr)�_stream�
_requestid�_maxsize�_chunks�_chunkssize)r=rxr2r|rrr�__init__]s

z(bufferingcommandresponseemitter.__init__ccs�|dur|��D]}|Vq	dS|j�|�}|sdSt|�|jkrT|��D]}|Vq&d}	||||j�}|t|�7}|jj|jtt|dd�V|t|�krSdSq/t|�|j	|jkro|j
�|�|j	t|�7_	dS|��D]}|Vqs|j
�|�t|�|_	dS)z�Send new data for emission.

        Is a generator of new frames that were derived from the new input.

        If the special input ``None`` is received, flushes all buffered
        data to frames.
        NrT�r5r'r:r�)�_flushr�r�r9r�rIr�r�r�r�r�r!)r=r_r7r~r�rrr�senddsB�	��z$bufferingcommandresponseemitter.sendccsZ�d�|j�}t|�|jksJ�g|jdd�<d|_|sdS|jj|jtt	|dd�VdS)NrKrTr�)
r#r�r9r�r�r�rIr�r�r�)r=r:rrrr��s��z&bufferingcommandresponseemitter._flushN)r,r-r.r/rtr�r�r�rrrrr�Qs
Dr�c@s0eZdZdZdd�Zdd�Zdd�Zdd	�Zd
S)�identityencoderz3Encoder for the "identity" stream encoding profile.cCsdS�Nr�r=�uirrrr���zidentityencoder.__init__cC�|Sr�r�r=r_rrrr��r�zidentityencoder.encodecC�dS�NrKr�r=rrrr��r�zidentityencoder.flushcCr�r�rr�rrr�finish�r�zidentityencoder.finishN)r,r-r.r/r�r�r�r�rrrrr��sr�c@s eZdZdZdd�Zdd�ZdS)�identitydecoderz3Decoder for the "identity" stream encoding profile.cCs|r	t�td���dS)Ns6identity decoder received unexpected additional values)rrdr�r=r��	extraobjsrrrr��s
��zidentitydecoder.__init__cCr�r�rr�rrrr��r�zidentitydecoder.decodeN)r,r-r.r/r�r�rrrrr��sr�c@�,eZdZdd�Zdd�Zdd�Zdd�Zd	S)
�zlibencodercCsddl}||_|��|_dSr�)�zlib�_zlib�compressobj�_compressor)r=r�r�rrrr��szzlibencoder.__init__cC�|j�|�Sr��r��compressr�rrrr���zzlibencoder.encodecC�|j�|jj�Sr�)r�r�r��Z_SYNC_FLUSHr�rrrr��szzlibencoder.flushcC�|j�|jj�}d|_|Sr�)r�r�r��Z_FINISH�r=�resrrrr���zzlibencoder.finishN�r,r-r.r�r�r�r�rrrrr��s
r�c@�eZdZdd�Zdd�ZdS)�zlibdecodercCs(ddl}|r
t�td���|��|_dS)Nrs2zlib decoder received unexpected additional values)r�rrdr�
decompressobj�
_decompressor)r=r�r�r�rrrr��s�zzlibdecoder.__init__cCs$tjst|t�rt|�}|j�|�Sr�)r	�ispy3r�rCr�r��
decompressr�rrrr��szzlibdecoder.decodeN�r,r-r.r�r�rrrrr��s
r�c@r�)
�zstdbaseencodercCs,ddlm}||_|j|d�}|��|_dS)Nr��zstd)�level)�r��_zstd�ZstdCompressorr�r�)r=r�r��cctxrrrr�szzstdbaseencoder.__init__cCr�r�r�r�rrrr�
r�zzstdbaseencoder.encodecCr�r�)r�r�r��COMPRESSOBJ_FLUSH_BLOCKr�rrrr�szzstdbaseencoder.flushcCr�r�)r�r�r��COMPRESSOBJ_FLUSH_FINISHr�rrrr�r�zzstdbaseencoder.finishNr�rrrrr�s
r�c�eZdZ�fdd�Z�ZS)�zstd8mbencodercstt|��d�dS)Nr)�superr�r�r���	__class__rrr�szzstd8mbencoder.__init__�r,r-r.r��
__classcell__rrr�rr��r�c@r�)�zstdbasedecodercCs&ddlm}|j|d�}|��|_dS)Nrr�)�max_window_size)r�r��ZstdDecompressorr�r�)r=�
maxwindowsizer��dctxrrrr�"szzstdbasedecoder.__init__cCr�r�)r�r�r�rrrr�(r�zzstdbasedecoder.decodeNr�rrrrr�!sr�cr�)�zstd8mbdecodercs(|r	t�td���tt|�jdd�dS)Ns5zstd8mb decoder received unexpected additional valuesi�)r)rrdrr�rr�r�r�rrr�-s
�zzstd8mbdecoder.__init__r�rrr�rr,r�rcCs�trdSzddlm}|jWntyd}Ynw|r(ttftd<t�d�t	t
ftd<t�d�ttftd<t�d�dS)Nrr�szstd-8mbszlib�identity)
�STREAM_ENCODERSr�r��__version__�ImportErrorr�r�STREAM_ENCODERS_ORDERr!r�r�r�r�r�rrr�populatestreamencoders<s
�

rc@s"eZdZdZddd�Zdd�ZdS)	rxz5Represents a logical unidirectional series of frames.FcCs||_||_dSr�)r3�_active�r=r3�activerrrr�Vs
zstream.__init__cCs,d}|js|tO}d|_t||j||||�S)�}Create a frame to be sent out over this stream.

        Only returns the frame instance. Does not actually send it.
        rT)r	�STREAM_FLAG_BEGIN_STREAMrIr3)r=r2r5r'r:r4rrrrIZs�zstream.makeframeN�F)r,r-r.r/r�rIrrrrrxSs
rxcs:eZdZdZd�fdd�	Zdd�Zdd�Zd	d
�Z�ZS)�inputstreamz,Represents a stream used for receiving data.Fcstt|�j||d�d|_dS)N�r)r�rr��_decoderr
r�rrr�ls
zinputstream.__init__cCs2|tvr
t�td�|��t|d||�|_dS)z�Set the decoder for this stream.

        Receives the stream profile name and any additional CBOR objects
        decoded from the stream encoding settings frame payloads.
        sunknown stream decoder: %srN)rrrdrr)r=r�r?r�rrr�
setdecoderpszinputstream.setdecodercC�|js|S|j�|�Sr�)rr�r�rrrr�{szinputstream.decodecC�|jsdS|j��Sr�)rr�r�rrrr���
zinputstream.flushr)	r,r-r.r/r�rr�r�r�rrr�rrisrcsTeZdZdZd�fdd�	Zdd�Zdd�Zd	d
�Zdd�Zdd
d�Z	dd�Z
�ZS)�outputstreamz*Represents a stream used for sending data.Fcs*tt|�j||d�d|_d|_d|_dS)NrF)r�rr�r��_encoder�_encodernamer
r�rrr��s
zoutputstream.__init__cCs6|tvr
t�td�|��t|d|�|_||_dS)zTSet the encoder for this stream.

        Receives the stream profile name.
        sunknown stream encoder: %srN)rrrdrrr)r=r�r?rrr�
setencoder�s
zoutputstream.setencodercCrr�)rr�r�rrrr��szoutputstream.encodecCrr�)rr�r�rrrr��rzoutputstream.flushcCs|jsdS|j��dSr�)rr�r�rrrr��szoutputstream.finishcCs^d}|js|tO}d|_|r|jst�d��|tO}|tkr%|t@r%d|_t||j	||||�S)rrTs@attempting to send encoded frame without sending stream settings)
r	r
r�r�ProgrammingError�STREAM_FLAG_ENCODING_APPLIED�FRAME_TYPE_STREAM_SETTINGS�!FLAG_STREAM_ENCODING_SETTINGS_EOSrIr3)r=r2r5r'r:r�r4rrrrI�s"���zoutputstream.makeframecCs2|jr|jrdSd�t�|j��}|�|tt|�S)z�Create a stream settings frame for this stream.

        Returns frame data or None if no stream settings frame is needed or has
        already been sent.
        NrK)	rr�r#rrOrrIrr)r=r2r:rrr�makestreamsettingsframe�s�z$outputstream.makestreamsettingsframer)r,r-r.r/r�rr�r�r�rIrr�rrr�rr�s
rcCs|jdr
t�d|j��dS)NrsAserver should only write to even numbered streams; %d is not even)r3rr)rxrrr�ensureserverstream�s
���r�contentencodingsrc@s�eZdZdZd(dd�Zdd�Zdd�Zd	d
�Zdd�Zd
d�Z	d)dd�Z
dd�Zdd�Zdd�Z
dd�Zdd�Zdd�Zdd�Zd d!�Zd"d#�Zd$d%�Zd&d'�ZdS)*�
serverreactora>Holds state of a server handling frame-based protocol requests.

    This class is the "brain" of the unified frame-based protocol server
    component. While the protocol is stateless from the perspective of
    requests/commands, something needs to track which frames have been
    received, what frames to expect, etc. This class is that thing.

    Instances are modeled as a state machine of sorts. Instances are also
    reactionary to external events. The point of this class is to encapsulate
    the state of the connection and the exchange of frames, not to perform
    work. Instead, callers tell this class when something occurs, like a
    frame arriving. If that activity is worthy of a follow-up action (say
    *run a command*), the return value of that handler will say so.

    I/O and CPU intensive operations are purposefully delegated outside of
    this class.

    Consumers are expected to tell instances when events occur. They do so by
    calling the various ``on*`` methods. These methods return a 2-tuple
    describing any follow-up action(s) to take. The first element is the
    name of an action to perform. The second is a data structure (usually
    a dict) specific to that action that contains more information. e.g.
    if the server wants to send frames back to the client, the data structure
    will contain a reference to those frames.

    Valid actions that consumers can be instructed to take are:

    sendframes
       Indicates that frames should be sent to the client. The ``framegen``
       key contains a generator of frames that should be sent. The server
       assumes that all frames are sent to the client.

    error
       Indicates that an error occurred. Consumer should probably abort.

    runcommand
       Indicates that the consumer should run a wire protocol command. Details
       of the command to run are given in the data structure.

    wantframe
       Indicates that nothing of interest happened and the server is waiting on
       more frames from the client before anything interesting can be done.

    noop
       Indicates no additional action is required.

    Known Issues
    ------------

    There are no limits to the number of partially received commands or their
    size. A malicious client could stream command request data and exhaust the
    server's memory.

    Partially received commands are not acted upon when end of input is
    reached. Should the server error if it receives a partial request?
    Should the client send a message to abort a partially transmitted request
    to facilitate graceful shutdown?

    Active requests that haven't been responded to aren't tracked. This means
    that if we receive a command and instruct its dispatch, another command
    with its request ID can come in over the wire and there will be a race
    between who responds to what.
    FcCsR||_||_d|_d|_g|_i|_i|_i|_t�|_	d|_
tt�|_
t�dS)a�Construct a new server reactor.

        ``deferoutput`` can be used to indicate that no output frames should be
        instructed to be sent until input has been exhausted. In this mode,
        events that would normally generate output frames (such as a command
        response being ready) will instead defer instructing the consumer to
        send those frames. This is useful for half-duplex transports where the
        sender cannot receive until all data has been transmitted.
        �initialrN)�_ui�_deferoutput�_state�_nextoutgoingstreamid�_bufferedframegens�_incomingstreams�_outgoingstreams�_receivingcommands�set�_activecommands�_protocolsettingsdecoder�dict�DEFAULT_PROTOCOL_SETTINGS�_sendersettingsr)r=r��deferoutputrrrr�-s


zserverreactor.__init__cCs�|jdsd|_|�td�|j�S|j|jvr0|jt@s'd|_|�td��St|j�|j|j<|jt@r=d|_t	�
d��|jt@rG|j|j=|j|j
|j|j|jd�}|�|j�}|sdt	�
d|j��||�S)z�Process a frame that has been received off the wire.

        Returns a dict with an ``action`` key that details what action,
        if any, the consumer should take next.
        r�erroreds/received frame with even numbered stream ID: %dsNreceived frame on unknown inactive stream without beginning of stream flag sets8support for decoding stream payloads not yet implemented)r"�protocol-settings-receiving�idle�command-receivingr2sunhandled state: %s)r3r%�_makeerrorresultrr(r4r
rrrr�STREAM_FLAG_END_STREAM�_onframeinitial�_onframeprotocolsettings�_onframeidle�_onframecommandreceiving�_onframeerroredr")r=r7�handlers�methrrr�onframerecvMsB
��
��
�

�zserverreactor.onframerecvcs8t��t�t�r
t�������fdd�}��|��S)z�Signal that objects are ready to be sent to the client.

        ``objs`` is an iterable of objects (typically a generator) that will
        be encoded via CBOR and added to frames, which will be sent to the
        client.
        c
3sX�d}d}t���}	zt��}Wnfty2|�d�D]}|Vq|r0t���D]}|Vq*Yn�tjyS}zt��|j|j	�D]}|VqCWYd}~n�d}~wt
yw}zt��dt�
|�dd�D]}|VqgWYd}~n�d}~wwz�t|tj�r�|r�t�d������}|r�|Vt��|�Vd}d}Wq
|r�t�d��|s�����}|r�|Vt���}|r�|Vd}t|tj�r�|�|j�D]}|Vq�n.t|tj�r�t�|j�D]
}|�|�D]}|Vq�q�nt�|�D]
}|�|�D]}|Vq�q�Wn$t
�y"}zt��d|dd�D]}|V�qWYd}~nd}~wwq�j���dS)NFTs%s�server�r�s:alternatelocationresponse seen after initial output objects(object follows alternatelocationresponse)r��next�
StopIterationr�r�r�WireprotoCommandErrorr�r��messageargs�	Exceptionr�r
�forcebytestrr�r�alternatelocationresponserrr�r��encodedresponser_�indefinitebytestringresponser�streamencodebytestringfromiter�chunksrOr,�remove)�emitted�alternatelocationsent�emitter�or7�er���objsr2r=rxrr�
sendframes�s��
���
����
��

�����	��
�
���kz?serverreactor.oncommandresponsereadyobjects.<locals>.sendframes)rr��list�iter�_handlesendframes)r=rxr2rTrUrrSr�oncommandresponsereadyobjects~s

qz+serverreactor.oncommandresponsereadyobjectscs.�jr�js
difS�fdd�}dd|�ifS)z�Signals that end of input has been received.

        No more frames will be received. All pending activity should be
        completed.
        �noopc3s"��jD]
}|D]}|VqqdSr�)r')�genr7r�rr�makegens�
��z)serverreactor.oninputeof.<locals>.makegen�
sendframes�framegen)r$r')r=r\rr�r�
oninputeofs�zserverreactor.oninputeofcCs&|jr
|j�|�difSdd|ifS)NrZr]r^)r$r'r!)r=�framegenrrrrX s�zserverreactor._handlesendframescs&t������fdd�}��|��S)Nc3s.�t���dd�D]}|Vq	�j���dS)Nr@rA)r�r,rM�r7�r�r2r=rxrrrU,s�
�z/serverreactor.onservererror.<locals>.sendframes�rrX)r=rxr2r�rUrrbr�
onservererror)szserverreactor.onservererrorNcs(t�������fdd�}��|��S)zACalled when a command encountered an error before sending output.c3s,�t�����D]}|Vq�j���dSr�)r�r,rMra�rzr�r2r=rxrrrU:s��z0serverreactor.oncommanderror.<locals>.sendframesrc)r=rxr2r�rzrUrrer�oncommanderror6szserverreactor.oncommanderrorcCsV|j}|jd7_t|�}||j|<tD]}||jdvr(|�|j|�|Sq|S)z�Create a stream to be used for sending data to the client.

        If this is called before protocol settings frames are received, we
        don't know what stream encodings are supported by the client and
        we will default to identity.
        rr )r&rr)rr0rr#)r=r3rRr?rrr�makeoutputstreamDs
�zserverreactor.makeoutputstreamcCsdd|ifS)Nr�r�r)r=r�rrrr6Zs�zserverreactor._makeerrorresultcCs�|j|}|dsd|_t�d��|j|=|jrd|_nd|_|d�d�t�|d���d}d|vr?d|_|�t	d	��Sd
|vrGi|d
<||j
vsNJ�|j
�|�d||d|d
|�d�|d
rm|d
��d�fSdd�fS)N�requestdoner2s,should not be called without requestdone setr5r4�payloadrrks$command request missing "name" fieldrls
runcommandrm�data)s	requestidscommandrlrmrj)
r*r%rr�seekr�	decodeall�getvaluer6rr,�addr")r=r2�entry�requestrrr�_makeruncommandresult_s@
������z#serverreactor._makeruncommandresultcCsdd|jifS)Ns	wantframesstate)r%r�rrr�_makewantframeresult�s�z"serverreactor._makewantframeresultcCsT|jt@}|jt@}|r|rd|_|�td��S|s&|s(d|_|�td��SdSdS)Nr2sGreceived command request frame with both new and continuation flags setsJreceived command request frame with neither new nor continuation flags set)r'rorpr%r6r)r=r7�new�continuationrrr�_validatecommandrequestframe�s"

�����z*serverreactor._validatecommandrequestframecCsX|jtkrd|_t��|_|�|�S|jtkrd|_|�|�Sd|_|�	t
d�|j�S)Nr3r4r2sBexpected sender protocol settings or command request frame; got %d)r5�#FRAME_TYPE_SENDER_PROTOCOL_SETTINGSr%r�bufferingdecoderr-r9rsr:r6r�r=r7rrrr8�s




���zserverreactor._onframeinitialc
Csj|jdksJ�|jdusJ�|jtkr d|_|�td�|j�S|jt@}|jt@}|r8|r8d|_|�td��S|sF|sFd|_|�td��Sz	|j�	|j
�Wn!typ}zd|_|�td�t�
|��WYd}~Sd}~ww|rw|��S|s{J�|j��}d|_|s�d|_|�td��St|�dkr�d|_|�td	��S|d
}d|vr�|d|jd<d|_|��S)
Nr3r2s/expected sender protocol settings frame; got %dsXsender protocol settings frame cannot have both continuation and end of stream flags setsOsender protocol settings frame must have continuation or end of stream flag sets;error decoding CBOR from sender protocol settings frame: %ss8sender protocol settings frame did not contain CBOR datars=sender protocol settings frame contained multiple CBOR valuesrr r4)r%r-r5rvr6rr'�*FLAG_SENDER_PROTOCOL_SETTINGS_CONTINUATION�!FLAG_SENDER_PROTOCOL_SETTINGS_EOSr�r:rFr
rGrr�getavailabler9r0)r=r7�more�eosrR�decoded�drrrr9�sx
��

����	�����	
���z&serverreactor._onframeprotocolsettingscCs|jtkrd|_|�td�|j�S|�|�}|r|S|j|jvr.d|_|�td�|j�S|j|jvrAd|_|�td�|j�S|j	t
@}|j	t@}|j	t@}|s\d|_|�td��St
��}|�|j�|d|t|�d�|j|j<|s~|s~|�|j�S|s�|s�J�d|_|��S)Nr2s&expected command request frame; got %ds#request with ID %d already receiveds$request with ID %d is already actives3received command request frame without new flag set)rirjrh�
expectingdatar5)r5rsr%r6rrur2r*r,r'rorrrqr
�bytesio�writer:�boolrqrr)r=r7r�rs�
moreframes�
expectingdatar:rrrr:sL
�
��


��zserverreactor._onframeidlecCs�|jtkr|jt@r|�|�S|�|�}|r|S|j|jvr+d|_|�	t
d�|j�S|j|jvr>d|_|�	t
d�|j�S|j|j}|jtkr�|jt@}t
|jt@�}|drcd|_|�	t
d��S||dkrsd|_|�	t
d��S|d�|j�|s�d	|d<|s�|s�|�|j�S|��S|jtkr�|ds�d|_|�	t
d
�|j�S|ddur�t��|d<|�||�Sd|_|�	t
d�|j�S)
Nr2s3received frame for request that is still active: %ds4received frame for request that is not receiving: %drhsGreceived command request frame when request frames were supposedly doner�s4mismatch between expect data flag and previous frameriTsFreceived command data frame for request that is not expecting data: %drjs"received unexpected frame type: %d)r5rsr'ror:rur2r,r%r6rr*rrr�rqr�r:rqrrrwr
r��_handlecommanddataframe)r=r7r�ror�r�rrrr;;sv



����

���
����z&serverreactor._onframecommandreceivingcCsh|jtksJ�|d�|j�|jt@r|��S|jt@r*|d�d�|�	|j
�Sd|_|�t
d��S)Nrjrr2s command data frame without flags)r5rwr�r:r'rurrrvrkrqr2r%r6r)r=r7rorrrr��s

z%serverreactor._handlecommanddataframecCs|�td��S)Nsserver already errored)r6rrxrrrr<�szserverreactor._onframeerroredrr�)r,r-r.r/r�r?rYr_rXrdrfrgr6rqrrrur8r9r:r;r�r<rrrrr!�s*
@ 1	

+L5Nr!c@seZdZdZddd�ZdS)�commandrequestz&Represents a request to run a command.NcCs(||_||_||_||_||_d|_dS)Nspending)r2r?rzr{r}�state)r=r2r?rzr{r}rrrr��s
zcommandrequest.__init__�NN)r,r-r.r/r�rrrrr��sr�c@sZeZdZdZ			ddd�Zddd�Zd	d
�Zdd�Zd
d�Zdd�Z	dd�Z
dd�ZdS)�
clientreactora�Holds state of a client issuing frame-based protocol requests.

    This is like ``serverreactor`` but for client-side state.

    Each instance is bound to the lifetime of a connection. For persistent
    connection transports using e.g. TCP sockets and speaking the raw
    framing protocol, there will be a single instance for the lifetime of
    the TCP socket. For transports where there are multiple discrete
    interactions (say tunneled within in HTTP request), there will be a
    separate instance for each distinct interaction.

    Consumers are expected to tell instances when events occur by calling
    various methods. These methods return a 2-tuple describing any follow-up
    action(s) to take. The first element is the name of an action to
    perform. The second is a data structure (usually a dict) specific to
    that action that contains more information. e.g. if the reactor wants
    to send frames to the server, the data structure will contain a reference
    to those frames.

    Valid actions that consumers can be instructed to take are:

    noop
       Indicates no additional action is required.

    sendframes
       Indicates that frames should be sent to the server. The ``framegen``
       key contains a generator of frames that should be sent. The reactor
       assumes that all frames in this generator are sent to the server.

    error
       Indicates that an error occurred. The ``message`` key contains an
       error message describing the failure.

    responsedata
       Indicates a response to a previously-issued command was received.

       The ``request`` key contains the ``commandrequest`` instance that
       represents the request this data is for.

       The ``data`` key contains the decoded data from the server.

       ``expectmore`` and ``eos`` evaluate to True when more response data
       is expected to follow or we're at the end of the response stream,
       respectively.
    FTNcCs`||_||_||_||_d|_d|_d|_d|_td�|_	t
��|_i|_
i|_i|_t�dS)a�Create a new instance.

        ``hasmultiplesend`` indicates whether multiple sends are supported
        by the transport. When True, it is possible to send commands immediately
        instead of buffering until the caller signals an intent to finish a
        send operation.

        ``buffercommands`` indicates whether sends should be buffered until the
        last request has been issued.

        ``clientcontentencoders`` is an iterable of content encoders the client
        will advertise to the server and that the server can use for encoding
        data. If not defined, the client will not advertise content encoders
        to the server.
        TFrN)r#�_hasmultiplesend�_buffersends�_clientcontentencoders�_canissuecommands�_cansend�_protocolsettingssent�_nextrequestidr�_outgoingstream�collections�deque�_pendingrequests�_activerequestsr(�_streamsettingsdecodersr)r=r��hasmultiplesend�buffersends�clientcontentencodersrrrr��s


zclientreactor.__init__cCs�|jst�d��|j}|jd7_t|||||d�}|jr)|j�|�|difS|js1t�d��|j	s:d|_d|_|dd|�
|�ifS)	aRequest that a command be executed.

        Receives the command name, a dict of arguments to pass to the command,
        and an optional file object containing the raw data for the command.

        Returns a 3-tuple of (request, action, action data).
        scannot issue new commandsr�r{r}rZ�*sends cannot be performed on this instanceFr]r^)r�rrr�r�r�r�r!r�r��_makecommandframes)r=r?rzr{r}r2rprrr�callcommands,

�
�
��zclientreactor.callcommandcsJ�jsdifS�jst�d���jsd�_d�_�fdd�}dd|�ifS)a_Request that all queued commands be sent.

        If any commands are buffered, this will instruct the caller to send
        them over the wire. If no commands are buffered it instructs the client
        to no-op.

        If instances aren't configured for multiple sends, no new command
        requests are allowed after this is called.
        rZr�Fc3s6��jr�j��}��|�D]}|Vq�jsdSdSr�)r��popleftr�)rpr7r�rr�
makeframes?s�
�z/clientreactor.flushcommands.<locals>.makeframesr]r^)r�r�rrr�r�)r=r�rr�r�
flushcommands's
��zclientreactor.flushcommandsccs��||j|j<d|_|js*|jr*d|_d�t�d|ji��}|jj	|jt
t|d�Vt|j|j|j
|j|j|jd�}|D]}|Vq<d|_dS)	z�Emit frames to issue a command request.

        As a side-effect, update request accounting to reflect its changed
        state.
        ssendingTrKr rnr�ssentN)r�r2r�r�r�r#rrOr�rIrvrzr�r?rzr{r})r=rpr:r�r7rrrr�Is8������	
z clientreactor._makecommandframescCs|jdrddtd�|jifS|j|jvr,|jt@s#ddtd�ifSt|j�|j|j<|j|j}|jt@r>|�|j�|_|jt	@rH|j|j=|j
tkrR|�|�S|j
|jvrcddtd�|j
ifS|j|j
}d|_t|jt|ji}|�|j
�}|s�t�d|j
��|||�S)	z�Process a frame that has been received off the wire.

        Returns a 2-tuple of (action, meta) describing further action the
        caller needs to take as a result of receiving this frame.
        rr�r�s.received frame with odd numbered stream ID: %dsEreceived frame on unknown stream without beginning of stream flag sets*received frame for inactive request ID: %ds	receivingsunhandled frame type: %d)r3rr(r4r
rrr�r:r7r5r�_onstreamsettingsframer2r�r�r��_oncommandresponseframer��_onerrorresponseframer"rr)r=r7rxrpr=r>rrrr?rsV
���

���





���
��
zclientreactor.onframerecvc
Csv|jtksJ�|jt@}|jt@}|r|rddtd�ifS|s)|s)ddtd�ifS|j|jvr9t�	�}||j|j<|j|j}z|�
|j�Wntyf}zddtd�t
�|�ifWYd}~Sd}~ww|rmdifS|sqJ�|��}|j|j=|s�ddtd�ifSz|j|j�|j|d|d	d��WdifSty�}zddtd
�t
�|�ifWYd}~Sd}~ww)Nr�r�sXstream encoding settings frame cannot have both continuation and end of stream flags setsOstream encoding settings frame must have continuation or end of stream flag sets;error decoding CBOR from stream encoding settings frame: %srZs8stream encoding settings frame did not contain CBOR datarrs error setting stream decoder: %s)r5rr'�*FLAG_STREAM_ENCODING_SETTINGS_CONTINUATIONrrr3r�rrwr�r:rFr
rGr{r(rr#)r=r7r|r}�decoderrRr~rrrr��s~

������
������
���
�������z$clientreactor._onstreamsettingsframecCs:|jt@r
d|_|j|j=d||jt@|jt@|jd�fS)Nsreceivedsresponsedata)�requests
expectmorerrj)r'r�r�r�r2r�r:)r=rpr7rrrr�s

��z%clientreactor._oncommandresponseframecCs8d|_|j|j=t�|j�d}d||d|dd�fS)Nr2rr�r�r�)r�r�r�)r�r�r2rrlr:)r=rpr7r�rrrr�!s
��z#clientreactor._onerrorresponseframe)FTNr�)r,r-r.r/r�r�r�r�r?r�r�r�rrrrr��s1
�
)'")HXr�r�)Y�
__future__rr�rE�i18nrr	r�
thirdpartyrr�rrr
r�utilsrr
rDrtr
r7rr;rsrwr�r�r��FRAME_TYPE_PROGRESSrvrr8rorprrrq�FLAGS_COMMAND_REQUESTrurv�FLAGS_COMMAND_DATAr�r��FLAGS_COMMAND_RESPONSEryrz�FLAGS_SENDER_PROTOCOL_SETTINGSr�r�FLAGS_STREAM_ENCODING_SETTINGSr<�Struct�ARGUMENT_RECORD_HEADERr)rR�objectr+r7rIrZrbrjr�r�r�r�r�r�r�r�r�r�r�r�r�r�r�rrrrrxrrrr/r!r�r�rrrr�<module>s���������


 B'
�M
�
%
�8n
!U	�5

https://t.me/RX1948 - 2025