File: //opt/saltstack/salt/lib/python3.10/site-packages/aiohttp/__pycache__/http_writer.cpython-310.pyc
o
;j:1 � @ s� d Z ddlZddlZddlZddlmZmZmZmZm Z m
Z
m Z m Z ddl
mZ ddlmZ ddlmZ ddlmZ dd lmZ dd
lmZ d Zejd krTdd
lm Z nddlm Z e e e ddf Z dZ dej kondk n Z!ejdk Z"e!pye"Z#G dd� de �Z$e$dd�Z%e$dd�Z&e ee ged f Z'e edged f Z(G dd� de�Z)e�*d�Z+de,d e,fd d �Z-d e,d dd e fd!d"�Z.e.Z/zddl0m1Z1 e1j/Z2es�e2Z/W dS W dS e3y� Y dS w )#z"Http related parsers and protocol.� N)�
TYPE_CHECKING�Any� Awaitable�Callable�Iterable�List�
NamedTuple�Optional)� CIMultiDict� )�AbstractStreamWriter)� BaseProtocol)�ClientConnectionResetError)�ZLibCompressor)�
NO_EXTENSIONS)� StreamWriter� HttpVersion�
HttpVersion10�
HttpVersion11)� � )�Buffer)�Unionzmemoryview[int]zmemoryview[bytes]i )r �
r )r r � )r r � c @ s e Zd ZU eed< eed< dS )r �major�minorN)�__name__�
__module__� __qualname__�int�__annotations__� r# r# �G/opt/saltstack/salt/lib/python3.10/site-packages/aiohttp/http_writer.pyr . s
r �CIMultiDict[str]c @ sx e Zd ZU dZedB ed< dZeed< dZeed< dZ e
dB ed< d9de de j
d ed
ed df
d d
�Zed e jdB fdd��Zed e fdd��Zd:dd�Z d;dededB d dfdd�Zded dfdd�Zd ee d dfd d �Zded dfd d �Z ded!ed dfd"d#�Z d$d%d&�ded'ed(ed dfd)d*�Z d+ed,d-d dfd.d/�Z d:d0d1�Z d:d2d3�Z!d<de"d dfd5d6�Z#d:d7d8�Z$dS )=r N�lengthF�chunked�_eof� _compress�protocol�loop�
on_chunk_sent�on_headers_sent�returnc C s( || _ || _|| _|| _d | _d| _d S )NF)� _protocolr+ �_on_chunk_sent�_on_headers_sent� _headers_buf�_headers_written)�selfr* r+ r, r- r# r# r$ �__init__B s
zStreamWriter.__init__c C s | j jS �N)r/ � transport�r4 r# r# r$ r7 P s zStreamWriter.transportc C s | j S r6 )r/ r8 r# r# r$ r* T s zStreamWriter.protocolc C s
d| _ d S )NT)r' r8 r# r# r$ �enable_chunkingX s
z StreamWriter.enable_chunking�deflate�encoding�strategyc C s t ||d�| _d S )N)r; r< )r r) )r4 r; r< r# r# r$ �enable_compression[ s z StreamWriter.enable_compression�chunkc C sR t |�}| j|7 _| j|7 _| jj}|d u s |�� r"td��|�|� d S )N�!Cannot write to closing transport)�len� buffer_size� output_sizer/ r7 �
is_closingr �write)r4 r> �sizer7 r# r# r$ �_write` s zStreamWriter._write�chunksc C s� d}|D ]}|t |�7 }q| j|7 _| j|7 _| jj}|d u s'|�� r+td��ts1|tk r;|� d�
|�� d S |� |� d S )Nr r? � ) r@ rA rB r/ r7 rC r �SKIP_WRITELINES�MIN_PAYLOAD_FOR_WRITELINESrD �join�
writelines)r4 rG rE r> r7 r# r# r$ � _writelinesi s zStreamWriter._writelinesc C s* t |�d�d��d�}| �||df� dS )z+Write a chunk with proper chunked encoding.�x�
�ascii�
N)r@ �encoderM )r4 r> �
chunk_len_prer# r# r$ �_write_chunked_payloadw s z#StreamWriter._write_chunked_payload�is_eofc C s� d| _ | j}d| _tr|dusJ �| js&|r | �||f� dS | �|� dS |rKt|�d�d��d�}|r@| �|||df� dS | �|||df� dS |rV| �|df� dS | �|� dS ) zASend buffered headers with payload, coalescing into single write.TNrN rO rP �
0
rQ � 0
)r3 r2 r r' rM rF r@ rR )r4 r> rU � headers_bufrS r# r# r$ �_send_headers_with_payload| s&
�z'StreamWriter._send_headers_with_payloadTi )�drain�LIMITrZ r[ c � s8 �| j dur| � |�I dH t|t�r |jt|�kr |�d�}| jdur1| j�|�I dH }|s1dS | jdurTt|�}| j|krF| j| | _n|d| j� }d| _|sTdS | j rs| j
ss| � |d� |rq| j |krqd| _ | �
� I dH dS |r�| jr~| �|� n| �|� |r�| j |kr�d| _ | �
� I dH dS dS dS dS )z�
Writes chunk of data to a stream.
write_eof() indicates end of stream.
writer can't be used after write_eof() method being called.
write() return drain future.
N�cr F)r0 �
isinstance�
memoryview�nbytesr@ �castr) �compressr& r2 r3 rY rA rZ r' rT rF )r4 r> rZ r[ Z chunk_lenr# r# r$ rD � sB �
�zStreamWriter.write� status_line�headersr% c � s6 �| j dur| � |�I dH t||�}d| _|| _dS )z Write headers to the stream.NF)r1 �_serialize_headersr3 r2 )r4 rb rc �bufr# r# r$ �
write_headers� s �
zStreamWriter.write_headersc C s@ | j r| jrdS d| _| j }d| _ tr|dusJ �| �|� dS )z3Force sending buffered headers if not already sent.NT)r2 r3 r rF �r4 rX r# r# r$ � send_headers� s zStreamWriter.send_headersc C sz | j rdS | jr-| js-d| _| j}d| _tr |dus J �| jr'| �|df� n| �|� n | jr8| jr8| �d� d| _ dS )z&Indicate that the message is complete.NTrW )r( r2 r3 r r' rM rF rg r# r# r$ �set_eof� s
zStreamWriter.set_eofrH c � s �| j rd S |r| jd ur| �|�I d H | jr�g }d}|r2| j�|�I d H }r2t|�}|�|� | j�� }|t|�7 }|�|� |sFJ �| jr�| js�d| _| j}d | _| j rn|d�d��
d�}| � ||g|�d�R � n | � |g|�R � | � � I d H d| _ d S | j r�|d�d��
d�}| � |g|�d�R � nt|�dkr�| � |� n| �
|d � | � � I d H d| _ d S | jr�| js�| �|d� | � � I d H d| _ d S | j r�|r�| � t|�d�d��
d�|df� n| �
d� | � � I d H d| _ d S |�r | �
|� | � � I d H d| _ d S ) Nr TrN rO rP rV r rW )r( r0 r) ra r@ �append�flushr2 r3 r' rR rM rZ rF rY )r4 r> rG Z
chunks_lenZcompressed_chunkZ flush_chunkrX rS r# r# r$ � write_eof sn �
� �
zStreamWriter.write_eofc � s2 �| j }|jdur|jr|�� I dH dS dS dS )zFlush the write buffer.
The intended use is to write
await w.write(data)
await w.drain()
N)r/ r7 Z_pausedZ
_drain_helper)r4 r* r# r# r$ rZ R s
��zStreamWriter.drain)NN)r. N)r: N)rH )%r r r r& r! r" r' �boolr( r) r r
�asyncioZAbstractEventLoop�_T_OnChunkSent�_T_OnHeadersSentr5 �propertyZ Transportr7 r* r9 �strr= r rF r rM rT rY rD rf rh ri �bytesrl rZ r# r# r# r$ r ; sp
�����
�
���
� !����
�3��
�
Nr z[\x00-\x08\x0a-\x1f\x7f]�stringr. c C s t �| �d ur td��| S )NzSForbidden control character detected in headers. Potential header injection attack.)�_FORBIDDEN_HEADER_CHARS_RE�search�
ValueError)rt r# r# r$ � _safe_headerd s
�rx rb rc c C s: t | � dd� |�� D �}| d d�|� d }|�d�S )Nc s s( � | ]\}}t |�d t |� V qdS )z: N)rx )�.0�k�vr# r# r$ � <genexpr>o s �& z(_py_serialize_headers.<locals>.<genexpr>rO z
zutf-8)rx �itemsrK rR )rb rc Z headers_gen�liner# r# r$ �_py_serialize_headersm s
r )4�__doc__rn �re�sys�typingr r r r r r r r Z multidictr
�abcr Z
base_protocolr
Zclient_exceptionsr Zcompression_utilsr Zhelpersr �__all__� version_info�collections.abcr r rs � bytearrayrJ ZIS_PY313_BEFORE_313_2ZIS_PY_BEFORE_312_9rI r r r ro rp r �compileru rr rx r rd Zaiohttp._http_writerZ _http_writerZ_c_serialize_headers� ImportErrorr# r# r# r$ �<module> sR (
(
� �