| Server IP : 80.87.202.40 / Your IP : 216.73.216.169 Web Server : Apache System : Linux rospirotorg.ru 5.14.0-539.el9.x86_64 #1 SMP PREEMPT_DYNAMIC Thu Dec 5 22:26:13 UTC 2024 x86_64 User : bitrix ( 600) PHP Version : 8.2.27 Disable Function : NONE MySQL : OFF | cURL : ON | WGET : ON | Perl : ON | Python : OFF | Sudo : ON | Pkexec : ON Directory : /lib64/python3.9/site-packages/mercurial/__pycache__/ |
Upload File : |
a
�+�b� � @ s d dl mZ d dlZd dlZddlmZ ddlmZ ddlm Z ddl
m Z m Z mZm
Z
mZ ddlmZmZ d Zd
ZdZd Zd Zeeed
�ZdZd ZdZdZdZ dZ d Z dZ eeeee e e e d�Z dZ!d Z"d Z#d Z$e!e"e#e$d�Z%dZ&d Z'e&e'd�Z(dZ)d Z*e)e*d�Z+dZ,d Z-e,e-d�Z.dZ/d Z0e/e0d�Z1ee%ee(ee+ei e i e i e e.e e1iZ2e�3d�Z4dd� Z5e j6dd�G dd � d e7��Z8e j6dd d �G d d � 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�ZDG d7d8� d8e7�ZEG d9d:� d:e7�ZFG d;d<� d<e7�ZGG d=d>� d>e7�ZHG d?d@� d@e7�ZIG dAdB� dBe7�ZJG dCdD� dDeJ�ZKG dEdF� dFe7�ZLG dGdH� dHeL�ZMi ZNg ZOdIdJ� ZPG dKdL� dLe7�ZQG dMdN� dNeQ�ZRG dOdP� dPeQ�ZSdQdR� ZTdSdTgiZUG dUdV� dVe7�ZVG dWdX� dXe7�ZWG dYdZ� dZe7�ZXdS )\� )�absolute_importN� )�_)�getattr)�attr)�encoding�error�pycompat�util�wireprototypes)�cborutil�
stringutil� i � � � )s stream-begins
stream-ends encoded� � � � � )s command-requests command-datas command-responses error-responses text-outputs progresss sender-protocol-settingss stream-settings)s new� continuations mores have-data)r � eosz<HHc C sV dd� t �| �D �}g }d}||krL||@ rB|�|�|d| �� |dK }q d�|�S )zFConvert a numeric flags value to a human value, using a mapping table.c S s i | ]\}}||�qS � r )�.0�k�vr r �@/usr/lib64/python3.9/site-packages/mercurial/wireprotoframing.py�
<dictcomp>~ � z humanflags.<locals>.<dictcomp>r � <unknown 0x%02x>� |)r � iteritems�append�get�join)�mapping�valueZnamemap�flags�valr r r �
humanflags| s
r) T)�slotsc @ s@ e Zd ZdZe�� Ze�� Ze�� Ze�� Z e�� Z
e�� Z dS )� frameheaderz&Represents the data in a frame header.N) �__name__�
__module__� __qualname__�__doc__r �ib�length� requestid�streamid� streamflags�typeidr' r r r r r+ � s r+ F)r* �reprc @ sN e Zd ZdZe�� Ze�� Ze�� Ze�� Z e�� Z
e�� Z e j
dd� �ZdS )�framezRepresents a parsed frame.c C sh d| j }t�t�D ]\}}|| j kr|} q0qdt| j�| j| jtt | j
�|tt � | j i �| j
�f S )Nr sH frame(size=%d; request=%d; stream=%d; streamflags=%s; type=%s; flags=%s))r5 r r! � FRAME_TYPES�len�payloadr2 r3 r) � STREAM_FLAGSr4 �FRAME_TYPE_FLAGSr# r' )�self�typename�namer& r r r �__repr__� s
���zframe.__repr__N)r, r- r. r/ r r0 r2 r3 r4 r5 r' r: r Z strmethodr@ r r r r r7 � s r7 c C sh t tt|� �}t�dt|��}|dd� |dd�< t�d|d| ||� |d> |B |d< ||dd�< |S ) z#Assemble a frame into a byte array.z<Ir r �<HBBr r r N)� bytearray�FRAME_HEADER_SIZEr9 �struct�pack� pack_into)r2 r3 r4 r5 r' r: r7 �lr r r � makeframe� s rH c C s | � dd�}|\}}}}}}t|�}t|�}d}|� d�D ]&} | tv rT|t| O }q:|t| �O }q:|tv rtt| }nt|�}d}
t| } |� d�D ]&} | | v r�|
| | O }
q�|
t| �O }
q�|�d�r�d�t�t �
|dd� ���}n
t � |�}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.
� r r r s cbor:r N)r2 r3 r4 r5 r' r: )
�split�intr; r8 r< �
startswithr$ r � streamencoder
ZevalpythonliteralZ unescapestrrH ) �s�fieldsr2 r3 r4 � frametype�
frameflagsr: Zfinalstreamflags�flagZ
finalflagsZ
validflagsr r r �makeframefromhumanstring� s>
�
�rS c C sb | d d| d d| 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 � r i @ r rA r r �� r � )rD � unpack_fromr+ )�dataZ framelengthr2 r3 r4 Z typeflagsrP rQ r r r � parseheader
s
�rY c C s� t t�}| �|�}|dkr dS |tkr<t�td�||f ��t|�}| �|j�}t |�|jkrzt�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.
r Ns+ received incomplete frame: got %d bytes: %ss' frame length error: expected %d; got %d)rB rC �readintor �Abortr rY �readr1 r9 r7 r2 r3 r4 r5 r' )Zfh�headerZ readcount�hr: r r r � readframe& s*
�� ���r_ c c s d|i}|r||d< |r ||d< d� t�|��}d}d} |sF| tO } n| tO } |rZ| tO } |||| � }
|t|
�7 }t|
�|kr�|t|�k r�| tO } | j|t | |
d�V | t@ s4q�q4|r�|�
t �}d} t|�t kr�t } nt
} d} | j|t| |d�V | r�q�q�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� redirectr r �r2 r5 r' r: FTN)r$ r rM �FLAG_COMMAND_REQUEST_NEW�!FLAG_COMMAND_REQUEST_CONTINUATION� FLAG_COMMAND_REQUEST_EXPECT_DATAr9 � FLAG_COMMAND_REQUEST_MORE_FRAMESrH �FRAME_TYPE_COMMAND_REQUESTr\ �DEFAULT_MAX_FRAME_SIZE� FLAG_COMMAND_DATA_CONTINUATION�FLAG_COMMAND_DATA_EOS�FRAME_TYPE_COMMAND_DATA) �streamr2 �cmd�args�datafh� maxframesize�redirectrX �offsetr' r: �doner r r �createcommandframesH sP
�
�ru c C sJ d� t�ddi��}| jr2| �|�}d}|s6d S nd}| j|tt||d�S )Nr � statuss okTF�r2 r5 r' r: �encoded)r$ r rM �streamsettingssent�encoderH �FRAME_TYPE_COMMAND_RESPONSE�"FLAG_COMMAND_RESPONSE_CONTINUATION)rm r2 �overallrx r r r � createcommandresponseokframe� s
�r~ c c sh | � � }d}|||| � }|t|�7 }|t|�k}|r>t}nt}| j|t|||dkd�V |r qdq dS )zACreate an empty payload frame representing command end-of-stream.r r rw N)�flushr9 �FLAG_COMMAND_RESPONSE_EOSr| rH r{ )rm r2 rq r: rs �chunkrt r' r r r � createcommandresponseeosframes� s" �r� c C s� d|j |jd�d�}dD ](}t||�}|d ur||d t�|�< qd�t�|��}| jrh| � |�}d}nd}| j
|t t ||d �S )
Nrb )s urls mediatype)rv � location)�sizeZ
fullhashesZ fullhashseedZserverdercertsZservercadercertsr� r TFrw )
�urlZ mediatyper r Zbytestrr$ r rM ry rz rH r{ r| )rm r2 �locationrX �ar& r: rx r r r �$createalternatelocationresponseframe� s* ��
�r� c c sF dd|id�}|r ||d d< d� t�|��}| j|tt|d�V d S )N� error� message)rv r� ra r rc )r$ r rM rH r{ r� )rm r2 �messagero �mr} r r r �createcommanderrorresponse� s �� �r� c c s4 d� t�|d|igd���}| j|td|d�V d S )Nr � msg)� typer� r rc )r$ r rM rH �FRAME_TYPE_ERROR_RESPONSE)rm r2 �msg�errtyper: r r r �createerrorframe s ��� �r� c c s� g }|D ]�\}}}t |t�s$td��|D ]}t |t�s(td��q(|D ]} t | t�sDtd��qD|�dd��d�}dd� |D �}dd� |D �}d |i}
|r�||
d
< |r�||
d < |�|
� qd �t�|��} t | �|kr�td
��| j
|t d| d�V dS )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 stringss must use bytes for argumentss must use bytes for labels�ascii�replacec S s g | ]}|� d d��d ��qS )zutf-8r� ��decoderz )r r� r r r �
<listcomp>7 r z)createtextoutputframe.<locals>.<listcomp>c S s g | ]}|� d d��d ��qS )r� �strictr� )r rG r r r r� : r r� ra s labelsr s$ cannot encode data in a single framer rc N) �
isinstance�bytes�
ValueErrorr� rz r" r$ r rM r9 rH �FRAME_TYPE_TEXT_OUTPUT) rm r2 Zatomsrq Z atomdictsZ
formattingro �labels�argZlabelZatomr: r r r �createtextoutputframe s8
�r� c @ s, e Zd ZdZefdd�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.
c C s" || _ || _|| _g | _d| _d S �Nr )�_stream�
_requestid�_maxsize�_chunks� _chunkssize)r= rm r2 rq r r r �__init__] s
z(bufferingcommandresponseemitter.__init__c c s |du r | � � D ]
}|V qdS | j�|�}|s4dS t|�| jkr�| � � D ]
}|V qJd}|||| j � }|t|�7 }| jj| jtt|dd�V |t|�krZdS qZt|�| j | jk r�| j
� |� | j t|�7 _ dS | � � D ]
}|V q�| 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.
Nr T�r5 r' r: rx ) �_flushr� rz r9 r� rH r� r{ r| r� r� r" )r= rX r7 rs r� r r r �sendd s< � z$bufferingcommandresponseemitter.sendc c sF d� | j�}g | jd d �< d| _|s(d S | jj| jtt|dd�V d S )Nr r Tr� )r$ r� r� r� rH r� r{ r| )r= r: r r r r� � s �z&bufferingcommandresponseemitter._flushN)r, r- r. r/ ri r� r� r� r r r r r� Q s Dr� c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) �identityencoderz3Encoder for the "identity" stream encoding profile.c C s d S �Nr �r= �uir r r r� � s zidentityencoder.__init__c C s |S r� r �r= rX r r r rz � s zidentityencoder.encodec C s dS �Nr r �r= r r r r � s zidentityencoder.flushc C s dS r� r r� r r r �finish� s zidentityencoder.finishN)r, r- r. r/ r� rz r r� r r r r r� � s
r� c @ s e Zd ZdZdd� Zdd� ZdS )�identitydecoderz3Decoder for the "identity" stream encoding profile.c C s |rt �td���d S )Ns6 identity decoder received unexpected additional values)r r[ r �r= r� � extraobjsr r r r� � s �zidentitydecoder.__init__c C s |S r� r r� r r r r� � s zidentitydecoder.decodeN)r, r- r. r/ r� r� r r r r r� � s r� c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
� zlibencoderc C s dd l }|| _|�� | _d S r� )�zlib�_zlib� compressobj� _compressor)r= r� r� r r r r� � s zzlibencoder.__init__c C s | j �|�S r� �r� �compressr� r r r rz � s zzlibencoder.encodec C s | j �| jj�S r� )r� r r� � Z_SYNC_FLUSHr� r r r r � s zzlibencoder.flushc C s | j �| jj�}d | _ |S r� )r� r r� �Z_FINISH�r= �resr r r r� � s zzlibencoder.finishN�r, r- r. r� rz r r� r r r r r� � s r� c @ s e Zd Zdd� Zdd� ZdS )� zlibdecoderc C s( dd l }|rt�td���|�� | _d S )Nr s2 zlib decoder received unexpected additional values)r� r r[ r �
decompressobj�
_decompressor)r= r� r� r� r r r r� � s �zzlibdecoder.__init__c C s$ t jst|t�rt|�}| j�|�S r� )r Zispy3r� rB r� r� �
decompressr� r r r r� � s zzlibdecoder.decodeN�r, r- r. r� r� r r r r r� � s
r� c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�zstdbaseencoderc C s, ddl m} || _|j|d�}|�� | _d S )Nr ��zstd)�level)� r� �_zstdZZstdCompressorr� r� )r= r� r� Zcctxr r r r� s zzstdbaseencoder.__init__c C s | j �|�S r� r� r� r r r rz
s zzstdbaseencoder.encodec C s | j �| jj�S r� )r� r r� ZCOMPRESSOBJ_FLUSH_BLOCKr� r r r r s zzstdbaseencoder.flushc C s | j �| jj�}d | _ |S r� )r� r r� ZCOMPRESSOBJ_FLUSH_FINISHr� r r r r� s zzstdbaseencoder.finishNr� r r r r r� s r� c s e Zd Z� fdd�Z� ZS )�zstd8mbencoderc s t t| ��d� d S )Nr )�superr� r� r� �� __class__r r r� s zzstd8mbencoder.__init__�r, r- r. r� �
__classcell__r r r� r r� s r� c @ s e Zd Zdd� Zdd� ZdS )�zstdbasedecoderc C s& ddl m} |j|d�}|�� | _d S )Nr r� )Zmax_window_size)r� r� ZZstdDecompressorr� r� )r= �
maxwindowsizer� Zdctxr r r r� " s zzstdbasedecoder.__init__c C s | j �|�S r� )r� r� r� r r r r� ( s zzstdbasedecoder.decodeNr� r r r r r� ! s r� c s e Zd Z� fdd�Z� ZS )�zstd8mbdecoderc s( |rt �td���tt| �jdd� d S )Ns5 zstd8mb decoder received unexpected additional valuesi � )r� )r r[ r r� r� r� r� r� r r r� - s
�zzstd8mbdecoder.__init__r� r r r� r r� , s r� c C s� t rd S zddlm} | j W n ty4 d } Y n0 | rPttft d< t�d� t t
ft d< t�d� t t ft d< t�d� d S )Nr r� s zstd-8mbs zlib� identity)
�STREAM_ENCODERSr� r� � __version__� ImportErrorr� r� �STREAM_ENCODERS_ORDERr" r� r� r� r� r� r r r �populatestreamencoders< s
r� c @ s" e Zd ZdZddd�Zdd� ZdS ) rm z5Represents a logical unidirectional series of frames.Fc C s || _ || _d S r� )r3 �_active�r= r3 �activer r r r� V s zstream.__init__c C s, d}| j s|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.
r T)r� �STREAM_FLAG_BEGIN_STREAMrH r3 )r= r2 r5 r' r: r4 r r r rH Z s �zstream.makeframeN)F)r, r- r. r/ r� rH r r r r rm S s
rm c s: e Zd ZdZd � fdd� Zdd� Zdd� Zd d
� Z� ZS ) � inputstreamz,Represents a stream used for receiving data.Fc s t t| �j||d� d | _d S )N�r� )r� r� r� �_decoderr� r� r r r� l s zinputstream.__init__c C s2 |t vrt�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.
s unknown stream decoder: %sr N)r� r r[ r r� )r= r� r? r� r r r �
setdecoderp s zinputstream.setdecoderc C s | j s
|S | j �|�S r� )r� r� r� r r r r� { s zinputstream.decodec C s | j s
dS | j �� S r� )r� r r� r r r r � s zinputstream.flush)F) r, r- r. r/ r� r� r� r r� r r r� r r� i s
r� c sT e Zd ZdZd� fdd� Zdd� Zdd� Zd d
� Zd d � Zdd
d�Z dd� Z
� Z S )� outputstreamz*Represents a stream used for sending data.Fc s* t t| �j||d� d| _d | _d | _d S )Nr� F)r� r� r� ry �_encoder� _encodernamer� r� r r r� � s zoutputstream.__init__c C s6 |t vrt�td�| ��t | d |�| _|| _dS )zTSet the encoder for this stream.
Receives the stream profile name.
s unknown stream encoder: %sr N)r� r r[ r r� r� )r= r� r? r r r �
setencoder� s zoutputstream.setencoderc C s | j s
|S | j �|�S r� )r� rz r� r r r rz � s zoutputstream.encodec C s | j s
dS | j �� S r� )r� r r� r r r r � s zoutputstream.flushc C s | j s
dS | j �� d S r� )r� r� r� r r r r� � s zoutputstream.finishc C s^ d}| j s|tO }d| _ |r4| js,t�d��|tO }|tkrJ|t@ rJd| _t|| j ||||�S )r� r Ts@ attempting to send encoded frame without sending stream settings)
r� r� ry r �ProgrammingError� STREAM_FLAG_ENCODING_APPLIED�FRAME_TYPE_STREAM_SETTINGS�!FLAG_STREAM_ENCODING_SETTINGS_EOSrH r3 )r= r2 r5 r' r: rx r4 r r r rH � s$ ����zoutputstream.makeframec C s2 | j r | jrdS d�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.
Nr ) r� ry r$ r rM r� rH r� r� )r= r2 r: r r r �makestreamsettingsframe� s �z$outputstream.makestreamsettingsframe)F)F) r, r- r. r/ r� r� rz r r� rH r� r� r r r� r r� � s
r� c C s | j d rt�d| j ��d S )Nr sA server should only write to even numbered streams; %d is not even)r3 r r� )rm r r r �ensureserverstream� s
��r� � contentencodingsr� c @ s� e Zd ZdZd(dd�Zdd� Zdd� Zd d
� Zd d � Zd
d� Z d)dd�Z
dd� Z dd� Z dd� Z
dd� Zdd� Zd d � Zd d � 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.
Fc C sR || _ || _d| _d| _g | _i | _i | _i | _t� | _ d| _
t t �| _
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.
� initialr N)�_ui� _deferoutput�_state�_nextoutgoingstreamid�_bufferedframegens�_incomingstreams�_outgoingstreams�_receivingcommands�set�_activecommands�_protocolsettingsdecoder�dict�DEFAULT_PROTOCOL_SETTINGS�_sendersettingsr� )r= r� Z deferoutputr r r r� - s
zserverreactor.__init__c C s� |j d s$d| _| �td�|j �S |j | jvr`|jt@ sNd| _| �td��S t|j �| j|j < |jt@ rzd| _t �
d��|jt @ r�| j|j = | j | j
| j| j| jd�}|�| j�}|s�t �
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: %dsN received frame on unknown inactive stream without beginning of stream flag sets8 support for decoding stream payloads not yet implemented)r� � protocol-settings-receiving� idle� command-receivingr s unhandled state: %s)r3 r� �_makeerrorresultr r� r4 r� r� r� r r� �STREAM_FLAG_END_STREAM�_onframeinitial�_onframeprotocolsettings� _onframeidle�_onframecommandreceiving�_onframeerroredr# )r= r7 �handlers�methr r r � onframerecvM sB
��
��
�
� zserverreactor.onframerecvc s8 t �� t� t�rt� �� � ���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
3 s� d} d}t ���}z t� �}W n� tyf |�d �D ]
}|V q6| r\t���D ]
}|V qPY �q�Y n� tjy� } z8t��|j|j �D ]
}|V q�W Y d }~�q�W Y d }~nbd }~0 t
�y } z@t ��dt �
|� dd�D ]
}|V q�W Y d }~�q�W Y d }~n
d }~0 0 �z(t|tj��rb| �r2t�d������}|�rH|V t��|�V d}d} W q|�rrt�d��| �s�����}|�r�|V t���}|�r�|V d} t|tj��r�|�|j�D ] }|V �q�nft|tj��rt�|j�D ] }|�|�D ] }|V �q��q�n*t�|�D ] }|�|�D ] }|V �q&�qW q t
�y� } z<t ��d| dd�D ] }|V �q`W Y d }~�q�W Y d }~qd }~0 0 q�j � �� d S )NFs %s� server�r� s: alternatelocationresponse seen after initial output objectTs( object follows alternatelocationresponse) r� �next�
StopIterationr� r� r ZWireprotoCommandErrorr� r� Z messageargs� Exceptionr� r
� forcebytestrr� r Zalternatelocationresponser� r� r� r~ ZencodedresponserX Z indefinitebytestringresponser Z streamencodebytestringfromiter�chunksrM r� �remove)ZemittedZalternatelocationsentZemitter�or7 �er� ��objsr2 r= rm r r �
sendframes� s�
�
�" �
$�
��
�� �
&z?serverreactor.oncommandresponsereadyobjects.<locals>.sendframes)r� r� �list�iter�_handlesendframes)r= rm r2 r r r r r � oncommandresponsereadyobjects~ s
qz+serverreactor.oncommandresponsereadyobjectsc s. � j r � jsdi fS � fdd�}dd|� ifS )z�Signals that end of input has been received.
No more frames will be received. All pending activity should be
completed.
� noopc 3 s � j D ]} | D ]
}|V qqd S r� )r� )�genr7 r� r r �makegen s
z)serverreactor.oninputeof.<locals>.makegen�
sendframes� framegen)r� r� )r= r" r r� r �
oninputeof s �zserverreactor.oninputeofc C s* | j r| j�|� di fS dd|ifS d S )Nr r# r$ )r� r� r" )r= Zframegenr r r r s �z serverreactor._handlesendframesc s& t �� � ���fdd�}��|� �S )Nc 3 s, t ��� dd�D ]
} | V q�j��� d S )Nr r )r� r� r �r7 �r� r2 r= rm r r r , s
�
z/serverreactor.onservererror.<locals>.sendframes�r� r )r= rm r2 r� r r r' r �
onservererror) s zserverreactor.onservererrorNc s( t �� � ����fdd�}��|� �S )zACalled when a command encountered an error before sending output.c 3 s* t ���� �D ]
} | V q�j��� d S r� )r� r� r r& �ro r� r2 r= rm r r r : s
�z0serverreactor.oncommanderror.<locals>.sendframesr( )r= rm r2 r� ro r r r* r �oncommanderror6 s z serverreactor.oncommanderrorc C sT | j }| j d7 _ t|�}|| j|< tD ]$}|| jd v r*|�| j|� qPq*|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.
r r� )r� r� r� r� r r� r� )r= r3 rN r? r r r �makeoutputstreamD s
z serverreactor.makeoutputstreamc C s dd|ifS )Nr� r� r )r= r� r r r r Z s �z serverreactor._makeerrorresultc C s� | j | }|d s"d| _t�d��| j |= | j r8d| _nd| _|d �d� t�|d �� �d }d|vr~d| _| �t d ��S d
|vr�i |d
< | j
� |� d ||d |d
|� d �|d
r�|d
�� nd d�fS )N� requestdoner s, should not be called without requestdone setr r � payloadr r` s$ command request missing "name" fieldra s
runcommandrb � data)s requestids commandra rb r/ )
r� r� r r� �seekr � decodeall�getvaluer r r� �addr# )r= r2 �entry�requestr r r �_makeruncommandresult_ s8
�� ��z#serverreactor._makeruncommandresultc C s dd| j ifS )Ns wantframes state)r� r� r r r �_makewantframeresult� s �z"serverreactor._makewantframeresultc C sP |j t@ }|j t@ }|r0|r0d| _| �td��S |sL|sLd| _| �td��S d S )Nr sG received command request frame with both new and continuation flags setsJ received command request frame with neither new nor continuation flags set)r' rd re r� r r )r= r7 �newZ continuationr r r � _validatecommandrequestframe� s
����z*serverreactor._validatecommandrequestframec C s\ |j tkr$d| _t�� | _| �|�S |j tkr>d| _| �|�S d| _| � t
d�|j �S d S )Nr r r sB expected sender protocol settings or command request frame; got %d) r5 �#FRAME_TYPE_SENDER_PROTOCOL_SETTINGSr� r �bufferingdecoderr� r rh r r r �r= r7 r r r r � s
���z serverreactor._onframeinitialc
C sL |j tkr$d| _| �td�|j �S |jt@ }|jt@ }|rT|rTd| _| �td��S |sp|spd| _| �td��S z| j� |j
� W nB t y� } z*d| _| �td�t �
|� �W Y d }~S d }~0 0 |r�| �� S | j�� }d | _|� s�d| _| �td��S t|�dk�r d| _| �td��S |d }d
|v �r>|d
| jd
< d | _| �� S ) Nr s/ expected sender protocol settings frame; got %dsX sender protocol settings frame cannot have both continuation and end of stream flags setsO sender protocol settings frame must have continuation or end of stream flag sets; error decoding CBOR from sender protocol settings frame: %ss8 sender protocol settings frame did not contain CBOR datar s= sender protocol settings frame contained multiple CBOR valuesr r� r )r5 r: r� r r r' �*FLAG_SENDER_PROTOCOL_SETTINGS_CONTINUATION�!FLAG_SENDER_PROTOCOL_SETTINGS_EOSr� r� r: r r
r r7 � getavailabler9 r )r= r7 �more�eosr �decoded�dr r r r � sn
��
���� ���
���
z&serverreactor._onframeprotocolsettingsc C s
|j tkr$d| _| �td�|j �S | �|�}|r6|S |j| jv r\d| _| �td�|j �S |j| jv r�d| _| �td�|j �S |j t
@ }|j t @ }|j t @ }|s�d| _| �td��S t
�� }|�|j� |d | t|�d�| j|j< |s�|s�| �|j�S d| _| �� S )Nr s& expected command request frame; got %ds# request with ID %d already receiveds$ request with ID %d is already actives3 received command request frame without new flag set)r. r/ r- �
expectingdatar )r5 rh r� r r r9 r2 r� r� r' rd rg rf r
�bytesio�writer: �boolr6 r7 )r= r7 r� r8 �
moreframes�
expectingdatar: r r r r sJ
�
� �
� � zserverreactor._onframeidlec C s� |j tkr0|jt@ r | �|�S | �|�}|r0|S |j| jv rVd| _| � t
d�|j �S |j| j vr|d| _| � t
d�|j �S | j |j }|j tk�r&|jt @ }t
|jt@ �}|d r�d| _| � t
d��S ||d kr�d| _| � t
d��S |d �|j� |�sd |d< |�s |�s | �|j�S | �� S |j tk�r||d �sVd| _| � t
d
�|j �S |d d u �rpt�� |d < | �||�S d| _| � t
d �|j �S d S )
Nr s3 received frame for request that is still active: %ds4 received frame for request that is not receiving: %dr- sG received command request frame when request frames were supposedly donerD s4 mismatch between expect data flag and previous framer. TsF received command data frame for request that is not expecting data: %dr/ s" received unexpected frame type: %d)r5 rh r' rd r r9 r2 r� r� r r r� rg rG rf rF r: r6 r7 rl r
rE �_handlecommanddataframe)r= r7 r� r4 rH rI r r r r
; sv
�� ��
�� �
��� �z&serverreactor._onframecommandreceivingc C s^ |d � |j� |jt@ r"| �� S |jt@ rF|d �d� | �|j�S d| _ | �
t d��S d S )Nr/ r r s command data frame without flags) rF r: r' rj r7 rk r0 r6 r2 r� r r )r= r7 r4 r r r rJ � s
z%serverreactor._handlecommanddataframec C s | � td��S )Ns server already errored)r r r< r r r r � s z serverreactor._onframeerrored)F)N)r, r- r. r/ r� r r r% r r) r+ r, r r6 r7 r9 r r r r
rJ r r r r r r� � s( @
1
+L5Nr� c @ s e Zd ZdZddd�ZdS )�commandrequestz&Represents a request to run a command.Nc C s( || _ || _|| _|| _|| _d| _d S )Ns pending)r2 r? ro rp rr �state)r= r2 r? ro rp rr r r r r� � s zcommandrequest.__init__)NN)r, r- r. r/ r� r r r r rK � s rK c @ sT e Zd ZdZddd�Zddd�Zd d
� Zd d � Zd
d� Zdd� Z dd� Z
dd� Z dS )�
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.
FTNc C s` || _ || _|| _|| _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.
TFr N)r� �_hasmultiplesend� _buffersends�_clientcontentencoders�_canissuecommands�_cansend�_protocolsettingssent�_nextrequestidr� �_outgoingstream� collections�deque�_pendingrequests�_activerequestsr� �_streamsettingsdecodersr� )r= r� ZhasmultiplesendZ buffersendsZclientcontentencodersr r r r� � s
zclientreactor.__init__c C s� | j st�d��| j}| jd7 _t|||||d�}| jrR| j�|� |di fS | jsbt�d��| j std| _d| _ |dd| �
|�ifS d S )
a Request 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).
s cannot issue new commandsr �rp rr r �* sends cannot be performed on this instanceFr# r$ N) rQ r r� rT rK rO rX r" rR rN �_makecommandframes)r= r? ro rp rr r2 r5 r r r � callcommand s,
�
�
��zclientreactor.callcommandc sJ � j sdi fS � js t�d��� js0d� _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.
r r\ Fc 3 s, � j r(� j �� } � �| �D ]
}|V qq d S r� )rX �popleftr] )r5 r7 r� r r �
makeframes? s
z/clientreactor.flushcommands.<locals>.makeframesr# r$ )rX rR r r� rN rQ )r= r` r r� r �
flushcommands' s
� �zclientreactor.flushcommandsc c s� || j |j< d|_| jsR| jrRd| _d�t�d| ji��}| jj |jt
t |d�V t | j|j|j
|j|j|jd�}|D ]
}|V qvd|_dS ) z�Emit frames to issue a command request.
As a side-effect, update request accounting to reflect its changed
state.
s sendingTr r� rc r[ s sentN)rY r2 rL rS rP r$ r rM rU rH r: r> ru r? ro rp rr )r= r5 r: r� r7 r r r r] I s6 ����� z clientreactor._makecommandframesc C s |j d r ddtd�|j ifS |j | jvrX|jt@ sFddtd�ifS t|j �| j|j < | j|j }|jt@ r||�|j�|_|jt @ r�| j|j = |j
t kr�| � |�S |j
| jvr�ddtd�|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.
r r� r� s. received frame with odd numbered stream ID: %dsE received frame on unknown stream without beginning of stream flag sets* received frame for inactive request ID: %ds receivings unhandled frame type: %d)r3 r r� r4 r� r� r� r� r: r r5 r� �_onstreamsettingsframer2 rY rL r{ �_oncommandresponseframer� �_onerrorresponseframer# r r� )r= r7 rm r5 r r
r r r r r sV
���
���
���
� �zclientreactor.onframerecvc
C sd |j t@ }|j t@ }|r,|r,ddtd�ifS |sD|sDddtd�ifS |j| jvrdt�� }|| j|j< | j|j }z|�|j � W n> t
y� } z&ddtd�t � |� ifW Y d }~S d }~0 0 |r�di fS |�
� }| j|j= |s�ddtd�ifS z(| j|j �| j|d |d d � � W n@ t
�yZ } z&ddtd
�t � |� ifW Y d }~S d }~0 0 di fS ) Nr� r� sX stream encoding settings frame cannot have both continuation and end of stream flags setsO stream encoding settings frame must have continuation or end of stream flag sets; error decoding CBOR from stream encoding settings frame: %sr s8 stream encoding settings frame did not contain CBOR datar r s error setting stream decoder: %s)r' �*FLAG_STREAM_ENCODING_SETTINGS_CONTINUATIONr� r r3 rZ r r; r� r: r r
r r? r� r� r� )r= r7 r@ rA �decoderr rB r r r rb � sr
��� ���
����
���
����
z$clientreactor._onstreamsettingsframec C s: |j t@ rd|_| j|j= d||j t@ |j t@ |jd�fS )Ns receiveds responsedata)� requests
expectmorer r/ )r' r� rL rY r2 r| r: )r= r5 r7 r r r rc s
��z%clientreactor._oncommandresponseframec C s8 d|_ | j|j= t�|j�d }d||d |d d�fS )Nr r r� r� r� )rg r� r� )rL rY r2 r r1 r: )r= r5 r7 r� r r r rd ! s
��z#clientreactor._onerrorresponseframe)FTN)NN) r, r- r. r/ r� r^ ra r] r rb rc rd r r r r rM � s 1 �
)
'")HXrM )N)YZ
__future__r rV rD Zi18nr r r Z
thirdpartyr r� r r r
r Zutilsr r
rC ri r� r r� r; rh rl r{ r� r� ZFRAME_TYPE_PROGRESSr: r� r8 rd re rg rf ZFLAGS_COMMAND_REQUESTrj rk ZFLAGS_COMMAND_DATAr| r� ZFLAGS_COMMAND_RESPONSEr= r> Z FLAGS_SENDER_PROTOCOL_SETTINGSre r� Z FLAGS_STREAM_ENCODING_SETTINGSr< �StructZARGUMENT_RECORD_HEADERr) rN �objectr+ r7 rH rS rY r_ ru r~ r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� rm r� r� r� r� r� rK rM r r r r �<module> s� �� ������
B'�
M�
%
�
8n
!U � 5