CasperSecurity
o
VƧ\ # � @ s� d Z ddlmZ ddlmZ G dd� d�ZG dd� d�ZG dd � d �ZG d
d � d �ZG d d
� d
�Z G dd� d�Z
G dd� d�Z G dd� d�Z G dd� d�Z
G dd� d�ZzddlZW n eyf dZY nw G dd� d�ZG dd � d �ZdS ) z;$Id: producers.py,v 1.9 2004/04/21 13:56:28 akuchling Exp $� )�find_prefix_at_end)�as_bytesc @ �" e Zd ZdZddd�Zdd� ZdS ) �simple_producerzproducer for a string� c C � || _ || _d S �N)�data� buffer_size��selfr r
� r
�=/usr/lib/python3/dist-packages/supervisor/medusa/producers.py�__init__ �
zsimple_producer.__init__c C sF t | j�| jkr| jd | j� }| j| jd � | _|S | j}d| _|S �N� )�lenr r
�r �resultr
r
r �more s zsimple_producer.moreN�r ��__name__�
__module__� __qualname__�__doc__r r r
r
r
r r s
r c @ r ) �scanning_producerz:like simple_producer, but more efficient for large stringsr c C � || _ || _d| _d S �Nr )r r
�posr r
r
r r # �
zscanning_producer.__init__c C sV | j t| j�k r)| j }tt| j�| j | j �}| j||� }| j t|�7 _ |S dS r )r r r �minr
)r �lp�rpr r
r
r r ( s
�zscanning_producer.moreNr r r
r
r
r r ! s
r c @ � e Zd ZdZdd� Zdd� ZdS )�lines_producerz producer for a list of linesc C �
|| _ d S r )�lines�r r( r
r
r r 8 �
zlines_producer.__init__c C s6 | j r| j d d� }| j dd � | _ d�|�d S dS )N�2 �
� )r( �join)r �chunkr
r
r r ; s
zlines_producer.moreNr r
r
r
r r& 5 s r& c @ r% )�buffer_list_producerz producer for a list of stringsc C � d| _ || _d S r )�index�buffers)r r3 r
r
r r H r z buffer_list_producer.__init__c C s2 | j t| j�kr
dS | j| j }| j d7 _ |S �Nr � )r2 r r3 �r r r
r
r r L s
zbuffer_list_producer.moreNr r
r
r
r r0 C s r0 c @ s$ e Zd ZdZdZdd� Zdd� ZdS )�
file_producerz(producer wrapper for file[-like] objects� c C r1 r )�done�file)r r: r
r
r r Z r zfile_producer.__init__c C s8 | j rdS | j�| j�}|s| j�� | `d| _ dS |S r4 )r9 r: �read�out_buffer_size�closer6 r
r
r r ^ s
zfile_producer.moreN)r r r r r< r r r
r
r
r r7 T s
r7 c @ sH e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dS )�output_producerz;Acts like an output file; suitable for capturing sys.stdoutc C s
d| _ d S r �r �r r
r
r r t r* zoutput_producer.__init__c C s& |� d�}d�|�}| j|7 _d S )N�
r, )�splitr. r )r r r( r
r
r �writew s
zoutput_producer.writec C s | j | d | _ d S �Nr, r? )r �liner
r
r � writeline| s zoutput_producer.writelinec C s | j d�|� d | _ d S rD )r r. r) r
r
r �
writelines s zoutput_producer.writelinesc C � d S r r
r@ r
r
r �flush� � zoutput_producer.flushc G rH r r
)r �argsr
r
r � softspace� rJ zoutput_producer.softspacec C s, | j r| j d d� }| j dd � | _ |S dS )Ni r- r? r r
r
r r � s
zoutput_producer.moreN) r r r r r rC rF rG rI rL r r
r
r
r r> r s r> c @ r% )�composite_producerz$combine a fifo of producers into onec C r' r )� producers)r rN r
r
r r � r* zcomposite_producer.__init__c C s> t | j�r | jd }|�� }|r|S | j�d� t | j�sdS )Nr r )r rN r �pop)r �p�dr
r
r r � s
�zcomposite_producer.moreNr r
r
r
r rM � s rM c @ r ) �globbing_producerz�
'glob' the output from a producer into a particular buffer size.
helps reduce the number of calls to send(). [this appears to
gain about 30% performance on requests to a single channel]
r8 c C s || _ d| _|| _d S r )�producer�bufferr
)r rS r
r
r
r r � r! zglobbing_producer.__init__c C sN t | j�| jk r | j�� }|r| j| | _nnt | j�| jk s| j}d| _|S r )r rT r
rS r )r r �rr
r
r r � s
�zglobbing_producer.moreN)r8 r r
r
r
r rR � s
rR c @ r% )�hooked_producerz�
A producer that will call <function> when it empties,.
with an argument of the number of bytes produced. Useful
for logging/instrumentation purposes.
c C r r )rS �function�bytes)r rS rW r
r
r r � r! zhooked_producer.__init__c C sD | j r | j �� }|sd | _ | �| j� |S | jt|�7 _|S dS )Nr- )rS r rW rX r r r
r
r r � s
�zhooked_producer.moreNr r
r
r
r rV � s rV c @ s" e Zd ZdZddd�Zdd� ZdS )�chunked_producera' A producer that implements the 'chunked' transfer coding for HTTP/1.1.
Here is a sample usage:
request['Transfer-Encoding'] = 'chunked'
request.push (
producers.chunked_producer (your_producer)
)
request.done()
Nc C r r )rS �footers)r rS rZ r
r
r r � r zchunked_producer.__init__c C s^ | j r-| j �� }|rdt|� }t|�d | d S d | _ | jr+d�dg| j �d S dS dS )Nz%xs
� 0s
s 0
r )rS r r r rZ r. )r r �sr
r
r r � s
zchunked_producer.morer r r
r
r
r rY � s
rY Nc @ r ) �compressed_producerz:
Compress another producer on-the-fly, using ZLIB
� c C s || _ t�|�| _d S r )rS �zlib� compressobj�
compressor)r rS �levelr
r
r r s z compressed_producer.__init__c C sD | j r d}|s | j �� }|sd | _ | j�� S | j�|�}|r|S dS r )rS r ra rI �compress)r �cdatar r
r
r r s
�zcompressed_producer.moreN)r^ r r
r
r
r r] � s
r] c @ s" e Zd ZdZd dd�Zdd� ZdS )
�escaping_producerz0A producer that escapes a sequence of characters�
.�
..c C s" || _ || _|| _d| _t| _d S r )rS �esc_from�esc_torT r )r rS rh ri r
r
r r s
zescaping_producer.__init__c C sh | j }| j}| j| j�� }|r2|�||�}| �||�}|r-|| d � | _|d | � S d| _|S |S r )rh ri rT rS r �replacer )r rh ri rT �ir
r
r r ' s zescaping_producer.moreN)rf rg r r
r
r
r re s
re )�RCS_ID�asynchatr �supervisor.compatr r r r& r0 r7 r> rM rR rV rY r_ � ImportErrorr] re r
r
r
r �<module> s* �