CasperSecurity
o
0�PjX � @ sv d dl mZ d dlZG dd� de�ZG dd� de�ZG dd� de�ZG d d
� d
e�Zd efd d
�ZG dd� de�Z dS )� )�ProtocolNc @ � e Zd ZdZdS )�DecodeRatioErrorzBOutput-to-input amplification ratio exceeded the configured limit.N��__name__�
__module__� __qualname__�__doc__� r
r
�1/usr/lib/python3/dist-packages/httplib2/decode.pyr � r c @ r )�DecodeLimitErrorz,Total output length exceeded the hard limit.Nr r
r
r
r r
r r
c @ sV e Zd Zedefdd��Zdedefdd�Zdefdd�Zdd
ed e defd d
�Z
dS )�DecoderProtocol�returnc C � d S �Nr
��selfr
r
r � needs_input s zDecoderProtocol.needs_input�bc C r r r
�r r r
r
r �decode � zDecoderProtocol.decodec C r r r
r r
r
r �flush r zDecoderProtocol.flush� �data�
chunk_sizec C sb t � }|dkr t|�}tdt|�|�D ]}|||| � }|�| �|�� q|�| �� � t|�S )Nr )� bytearray�len�range�extendr r �bytes)r r r �out�i�chunkr
r
r �
consume_bytes s z DecoderProtocol.consume_bytesN)r ) r r r �property�boolr r! r r �intr% r
r
r
r r
s r c @ sh e Zd ZdZdZdZdZdZdZefde fdd �Z
e d
e fd d ��Z
d
ed
efdd�Zd
efdd�ZdS )� ZlibDecodera#
Thin wrapper around zlib.Decompressor conforming to the Decoder interface.
Note: zlib pushes all available decompressed data immediately upon receiving
input. It never holds back output requiring `decode(b"")` to extract it.
Thus, `needs_input` naturally remains True.
)�_decoderi�� � �/ �wbitsc C s t �|�| _d S r )�zlib�
decompressobjr* )r r. r
r
r �__init__3 s zZlibDecoder.__init__r c C s | j d u r td��| j j S �Nzused after flush())r* � RuntimeError�eofr r
r
r r 6 s
zZlibDecoder.needs_inputr c C s | j d u r td��| j �|�S r2 )r* r3 �
decompressr r
r
r r < s
zZlibDecoder.decodec C s& | j d u r td��| j �� }d | _ |S r2 )r* r3 r )r �resultr
r
r r A s
zZlibDecoder.flushN)r r r r � __slots__�
WBITS_DEFLATE�
WBITS_ZLIB�
WBITS_GZIP�WBITS_AUTO_GZIP_ZLIBr( r1 r& r' r r! r r r
r
r
r r) # s r) r c C s
t t j�S r )r) r8 r
r
r
r �DeflateDecoderI s
r< c
@ s� e Zd ZdZ ddededed ed
ed d f d
d�Zddd�Ze d e
fdd��Z de d e fdd�Z
d e fdd�Zde d efdd�Zd S ) � LimitDecoder) r* �_ratio� _chunk_size� _safe_limit� _hard_limit�_consumed_length�_output_length�
_input_buffer�_flushed�d r � � �
�decoder�ratior �
safe_limit�
hard_limitr Nc C s� |dk r t d|� d���|dk rt d|� d���|dk r$t d|� d���|dk r0t d|� d���|| _|| _|| _|| _|| _d| _d| _t� | _ d| _
d S )Nr zLimitDecoder() ratio=z expected >= 0zLimitDecoder() chunk_size=zLimitDecoder() safe_limit=F) �
ValueErrorr* r> r? r@ rA rB rC r rD rE )r rI rJ r rK rL r
r
r r1 Z s"
zLimitDecoder.__init__c C s� | j dkr| j| j krtd| j� d| j � ���| jdkr$| j| jk r$d S | jdkrU| j| j| j krW| jdkr=| j| j ntd�}td|d�d| j� d| j� d | j� ���d S d S )
Nr zOutput length z exceeds hard limit �infzAmplification ratio z.1fz (�/z) exceeds limit )rA rC r
r@ r> rB �floatr )r � actual_ratior
r
r �
_check_limitsu s ���zLimitDecoder._check_limitsc C s | j jS r )r* r r r
r
r r � s zLimitDecoder.needs_inputr c C s | j rtd��| �|�}t|�S )Nz decode() called after flush())rE r3 �_pumpr! )r r �outputr
r
r r � s
zLimitDecoder.decodec C sT | j rtd��d| _ | �d�}| j�� }|�|� | jt|�7 _| �� t |�S )Nz flush() called more than onceT� )
rE r3 rS r* r r rC r rR r! )r rT r r
r
r r � s
zLimitDecoder.flushc C s� | j �|� t� } | jjs)| j�d�}|r)|�|� | jt|�7 _| �� q | j rat | j d | j
� �}| j d | j
�= | j�|�}| j t|�7 _ |r`|�|� | jt|�7 _| �� q |S )NTrU ) rD r r r* r r rC r rR r! r? rB )r r rT r r$ r
r
r rS � s,
zLimitDecoder._pump)rF r rG rH )r N)r r r r7 r rP r( r1 rR r& r' r r! r r r rS r
r
r
r r= M s2 ������
�
r= )
�typingr r/ � Exceptionr r
r r) r< r= r
r
r
r �<module> s &