File: //opt/saltstack/salt/lib/python3.10/site-packages/aiohttp/__pycache__/web_protocol.cpython-310.pyc
o
;jty �
@ s< d dl Z d dlZ d dlZd dlZd dlZd dlmZ d dlmZm Z m
Z
d dl m Z d dl
mZ d dlmZ d dlmZ d dlmZmZmZmZ d dlZd dlZd d lm Z d
d l m Z m Z d
d l m!Z!m"Z" d
d
l#m$Z$m%Z% d
dlm&Z&m'Z'm(Z(m)Z)m*Z*m+Z+ d
dl,m-Z- d
dl.m/Z/m0Z0 d
dl1m2Z2m3Z3 d
dl4m5Z5 d
dl6m7Z7m8Z8 d
dl9m:Z: d
dl;m<Z< d
dl=m>Z>m?Z? dZ@dZAer�d dlBZBd
dlCmDZD e e)e3de dge<f ZEe e<gee? f ZFe)d d e(i i d dd d e�Gd ��
ZHG d d!� d!eI�ZJG d"d#� d#eI�ZKeK� ZLejMd d d d$�G d%d&� d&��ZNeOe)eNB e3f ZPG d'd� de"�ZQdS )(� N)�deque)� Awaitable�Callable�Sequence)�suppress)�escape)�
HTTPStatus)�Logger)�
TYPE_CHECKING�Any�Optional�cast)�under_cached_property� )�AbstractAccessLogger�AbstractStreamWriter)�PAUSE_RESUME_READING_ERRORS� BaseProtocol)�DEFAULT_CHUNK_SIZE� ceil_timeout)�HttpProcessingError�HttpRequestParser�
HttpVersion10�RawRequestMessage� StreamWriter�WebSocketReader)�
BadHttpMethod)�
access_logger�
server_logger)�
EMPTY_PAYLOAD� StreamReader)�
tcp_keepalive)�
HTTPException�HTTPInternalServerError)� AccessLogger)� BaseRequest)�Response�StreamResponse)�RequestHandler�RequestPayloadError�PayloadAccessError� )�Serverr( zasyncio.Task[None]�UNKNOWN�/TFc @ � e Zd ZdZdS )r) zPayload parsing error.N��__name__�
__module__� __qualname__�__doc__� r5 r5 �H/opt/saltstack/salt/lib/python3.10/site-packages/aiohttp/web_protocol.pyr) N � r) c @ r/ )r* z-Payload was accessed after response was sent.Nr0 r5 r5 r5 r6 r* R r7 r* )Z auto_attribs�frozen�slotsc @ s& e Zd ZU eed< eed< eed< dS )�_ErrInfo�status�exc�messageN)r1 r2 r3 �int�__annotations__�
BaseException�strr5 r5 r5 r6 r: Y s
r: c ! s e Zd ZdZdZddeeeejddddde dd d
�d d d
e
j de de
dedee dedede
dededede dede
de f � fd d �Zd efd d �Zed ed! fd"d#��Zed eeeeeef B eeef B d$B fd%d&��Zed eeeeeef B eeef B d$B fd'd(��Zed e fd)d*��Z did,e d$B d d$fd-d.�Z d/e
j d d$f� fd0d1� Z d2e d$B d d$f� fd3d4� Z! $djd5e"d6e#g d$f d$B d d$fd7d8�Z$dkd9d:�Z%d;e&d d$fd<d=�Z'd e
fd>d?�Z(dkd@dA�Z)dkdBdC�Z*dDe
d d$fdEdF�Z+dkdGdH�Z,dkdIdJ�Z-dKe.dLe/dMe d$B d d$fdNdO�Z0dPe1dQe1d d$fdRdS�Z2dPe1dQe1d d$fdTdU�Z3dkdVdW�Z4dKe.dXe d$B dYe#e.ge5e/ f d ee/e
f fdZd[�Z6dkd\d]�Z7dKe.d^e/dXe d$B d ee/e
f fd_d`�Z8 a $ $dldKe.dbed2e d$B dced$B d e/f
ddde�Z9dfe:d e#e.ge5e/ f fdgdh�Z;� Z<S )mr( aj HTTP protocol implementation.
RequestHandler handles incoming HTTP request. It reads request line,
request headers and request payload and calls handle_request() method.
By default it always returns with 404 response.
RequestHandler handles errors in incoming request, like bad
status line, bad headers or incomplete payload. If any error occurs,
connection gets closed.
keepalive_timeout -- number of seconds before closing
keep-alive connection
tcp_keepalive -- TCP keep-alive is on, default is on
debug -- enable debug mode
logger -- custom logger object
access_log_class -- custom class for access_logger
access_log -- custom logging object
access_log_format -- access log format string
loop -- Optional event loop
max_line_size -- Optional maximum header line size
max_field_size -- Optional maximum header field size
max_headers -- Optional maximum header size
timeout_ceil_threshold -- Optional value to specify
threshold to ceil() timeout
values
)"�max_field_size� max_headers�
max_line_size�_request_count�
_keepalive�_manager�_request_handler�_request_factory�_tcp_keepalive�_next_keepalive_close_time�_keepalive_handle�_keepalive_timeout�_lingering_time� _messages�_max_msg_queue_size�_msg_queue_resume_size�_msg_queue_paused�
_message_tail�_handler_waiter�_waiter�
_task_handler�_payload_parser�_data_received_cb�logger�debug�
access_logr �_close� _force_close�_current_request�_timeout_ceil_threshold�_request_in_progress�_logging_enabled�_cachei. TFi� � g $@� )�keepalive_timeoutr! rY �access_log_classr[ �access_log_formatrZ rD rC rB �lingering_time� read_bufsize�auto_decompress�timeout_ceil_threshold�managerr, �loopre r! rY rf r[ rg rZ rD rC rB rh ri rj rk c sH t | _t d | _d| _t| |||
| | t|t d� }t� �||� d| _d| _ d | _
|| _ |j | _
|j| _|
| _| | _| | _|| _d| _d | _|| _t|
�| _t� | _d| _d | _ d | _ d | _ d | _ d | _ d| _!zt|�| _!W n t"t#fyx Y nw || _$| | _%|| _&|r�|||�| _'| j'j(| _)nd | _'d| _)d| _*d| _+d| _,i | _-d S )N� F)rD rB rC Zpayload_exceptionrj Zmax_msg_queue_sizer g � rd ).�MAX_MSG_QUEUE_SIZErP rQ rR r r) �super�__init__rE rF r^ rG �request_handlerrH Zrequest_factoryrI rD rC rB rJ rK rL rM �floatrN r rO rS rX rU rT rV rW r_ � TypeError�
ValueErrorrY rZ r[ r �enabledra r\ r] r` rb )�selfrl rm re r! rY rf r[ rg rZ rD rC rB rh ri rj rk �parser�� __class__r5 r6 rr � sp
�
��
zRequestHandler.__init__�returnc C s d� | jj| jd ur
d�S d�S )Nz<{} {}>� connectedZ disconnected)�formatr{ r1 � transport�rx r5 r5 r6 �__repr__ s ��zRequestHandler.__repr__zssl.SSLContextc C � | j du rdS | j �d�S )z Return SSLContext if available.NZ
sslcontext�r Zget_extra_infor� r5 r5 r6 � ssl_context s
��
�zRequestHandler.ssl_contextNc C r� )z Return peername if available.N�peernamer� r� r5 r5 r6 r� �
��
�zRequestHandler.peernamec C r� )z Return sockname if available.N�socknamer� r� r5 r5 r6 r� & r� zRequestHandler.socknamec C � | j S �N)rM r� r5 r5 r6 re 1 s z RequestHandler.keepalive_timeout� .@�timeoutc � s� �d| _ | jdur| j�� | jr^| j�� | _z&t|�4 I dH � | jI dH W d �I dH n 1 I dH s7w Y W n tj tj
fy] d| _t j dkr[t�
� }r[|�� r[� Y nw z@t|�4 I dH �+ | jduru| j�t� � � | jdur�| j�� s�t�| j�I dH W d �I dH n 1 I dH s�w Y W n tj tj
fy� t j dkr�t�
� }r�|�� r�� Y nw | jdur�| j�� | �� dS )z�Do worker process exit preparations.
We need to clean up everything and stop accepting requests.
It is especially important for keep-alive connections.
TN�� � )r] rL �cancelr` �_loop�
create_futurerT r �asyncio�CancelledError� TimeoutError�sys� version_info� current_task�
cancellingr^ �_cancelrV �doneZshield� force_close)rx r� �taskr5 r5 r6 �shutdown5 sT �
(��
����
�(��
����
zRequestHandler.shutdownr c s� t � �|� ttj|�}| jrt|� | jd usJ �| j�| |� | j}t j
dkr4tj | � � |dd�}n|�
| � � �}|| _d S )N�r� � T�rm Z eager_start)rq �connection_mader
r� Z TransportrJ r! rG r� r� r� �Task�start� create_taskrV )rx r Zreal_transportrm r� rz r5 r6 r� g s
z RequestHandler.connection_mader< c s� | j d u rd S | j �| |� | j j}| �� t� �|� d | _ d | _d | _d | _| jd ur2| j� � | j
d urE|d u r?t d�}| j
� |� |rQ| j
d urQ| j
� � d | _
| jd urc| j�� d | _d S d S )NzConnection lost)rG �connection_lost�handler_cancellationr� rq rI rH �_parserrL r� r^ �ConnectionResetErrorr� rV rW Zfeed_eof)rx r< r� rz r5 r6 r� x s.
�z RequestHandler.connection_lostry �data_received_cbc C s< | j d u sJ �|| _ || _| jr | j �| j� d| _d S d S )Nro )rW rX rS � feed_data)rx ry r� r5 r5 r6 �
set_parser� s
�zRequestHandler.set_parserc C s d S r� r5 r� r5 r5 r6 � eof_received� s zRequestHandler.eof_received�datac
C sz | j s| jrd S | jd u r�| js�| jd usJ �z | j�|�\}}}W n! tyC } ztd||jd�t fg}d}d}W Y d }~nd }~ww |D ]\}}| j
d7 _
| j � ||f� qF| j
}|rl|d url|�� sl|�d � | js{t| j �| jkr{| �� || _|r�|r�|| _d S d S d S | jd u r�| jr�|r�| j|7 _d S |r�| jd ur�| �� | j�|�\} }| r�| �� d S d S d S )Ni� )r; r< r= Fro r )r] r\ rW � _upgradedr� r� r r: r= r rE rO �appendrU r� �
set_resultrR �lenrP �_pause_msg_queue_readingrS rX �close)
rx r� �messagesZupgraded�tailr< �msg�payload�waiter�eofr5 r5 r6 �
data_received� sJ ���
�
�
�z RequestHandler.data_receivedc C r� r� )rR r� r5 r5 r6 � _reading_paused_for_msg_queue� s z,RequestHandler._reading_paused_for_msg_queuec C s: d| _ | jd urz| j�� W d S ty Y d S w d S )NT)rR r Z
pause_readingr r� r5 r5 r6 r� � s
��z'RequestHandler._pause_msg_queue_readingc C sh | j s| �d� t| j�| jkrd S d| _| js0| jd ur2z| j�� W d S t y/ Y d S w d S d S )Nro F)
r� r� r� rO rP rR Z_reading_pausedr Zresume_readingr r� r5 r5 r6 �_resume_msg_queue_reading� s
��z(RequestHandler._resume_msg_queue_reading�valc C s$ || _ | jr| j�� d| _dS dS )zMSet keep-alive connection mode.
:param bool val: new state.
N)rF rL r� )rx r� r5 r5 r6 �
keep_alive� s
�zRequestHandler.keep_alivec C s d| _ | jr
| j�� dS dS )z�Close connection.
Stop accepting new pipelining messages and close
connection when handlers done processing messages.
TN)r\ rU r� r� r5 r5 r6 r� s �zRequestHandler.closec C s8 d| _ | jr | j�� | jdur| j�� d| _dS dS )z Forcefully close connection.TN)r] rU r� r r� r� r5 r5 r6 r� s
�zRequestHandler.force_close�request�response�timec C sF | j r | jd ur!tr|d usJ �| j�||| j�� | � d S d S d S r� )ra r r
�logr� r� )rx r� r� r� r5 r5 r6 �
log_access s
�zRequestHandler.log_access�args�kwc O s | j r| jj |i |�� d S d S r� )rZ rY �rx r� r� r5 r5 r6 � log_debug s �zRequestHandler.log_debugc O s | j j|i |�� d S r� )rY � exceptionr� r5 r5 r6 �
log_exception! s z RequestHandler.log_exceptionc C sj d | _ | js | js d S | j}|�� }| j}||k r#|�|| j�| _ d S | jr1| j� � s3| �
� d S d S d S r� ) rL r] rF r� r� rK �call_at�_process_keepaliverU r� r� )rx rm �now�
close_timer5 r5 r6 r� $ s �z!RequestHandler._process_keepalive�
start_timers c
� s� �d| _ z�zz|| _||�I d H }W d | _nd | _w W nw ty; } z|}| �|||�I d H \}}W Y d }~ntd }~w tjyC � tjyn } z | jd|d� | �|d�}| �|||�I d H \}}W Y d }~nAd }~w t y� } z| �|d|�}| �|||�I d H \}}W Y d }~n d }~ww t
|dd�r�t � dt
� | �|||�I d H \}}W d| _ | jd ur�| j�d � ||fS d| _ | jd ur�| j�d � w w ) NTzRequest handler timed out.��exc_infoi� �� Z__http_exception__Fzlreturning HTTPException object is deprecated (#2415) and will be removed, please raise the exception instead)r` r^ r" �finish_responser� r� r� r� � handle_error� Exception�getattr�warnings�warn�DeprecationWarningrT r� )rx r� r� rs �respr< �resetr5 r5 r6 �_handle_request5 sL �$� $�$�� ��
�
�z RequestHandler._handle_requestc
� s| �| j }| j}|dus
J �| j}d}| jdusJ �| jdus J �| j�s&| js<z|�� | _| jI dH W d| _nd| _w | j� � \}}| j
durM| j
� � | j r\t
| j�| jkr\| �� | jrc|�� nd}| jd7 _t| |�}t|t�r~| �|�} t}n| j} | �||| || jp�t�|��}
�zV�z| �|
|| �} t j dkr�tj | |dd�} n|� | �} z | I dH \}}
W n" t y� | �!d� Y W W d|
_"| j#du r�|dur�| �!d� �nMw ~ |
r�| �!d� W W d|
_"| j#du r�|dur�| �!d� �n/t$|j%�| _&|�'� �s�| j(}| j�s~|�r~| �!d |� |�� }|| }zB|�'� �s[||k �r[t)|| �4 I dH � |�*� I dH W d �I dH n 1 I dH �sHw Y |�� }|�'� �s[||k �s%W n! tj+tj,f�y} t j d
k�r{t�� }�r{|�-� �r{� Y nw |�'� �s�| j�s�| �!d � | �.� |�/t0� W n< tj+�y� | �!d� | �1� � t2�y� } z| j3d |d
� | �1� W Y d}~nd}~w t4�y� | �1� � w W d|
_"| j#du �r�|du�r�| �!d� nd|
_"| j#du �r�|du�r�| �!d� w w w | j&�r"| j5�s"| j�s"|�� | }|| _6| j7du �r!|�8|| j9�| _7nn| jr$| j�s:d| _| j#du�r<| j#�.� dS dS dS )a_ Process incoming request.
It reads request line, request headers and request payload, then
calls handle_request() method. Subclass has to override
handle_request(). start() handles various exceptions in request
or response handling. Connection is being closed always unless
keep_alive(True) specified.
Nr r� Tr� z&Ignored premature client disconnectionz'Ignored premature client disconnection.z(Ignored premature client disconnection 2z'Start lingering close timer for %s sec.r� zUncompleted request.zUnhandled exceptionr� ):r� rG rM rI rH r] rO r� rU �popleftr� Zmessage_consumedrR r� rQ r� ra r� Zrequests_countr �
isinstancer: �_make_error_handler�ERRORrV r� r� r� r� r� r� r� �ConnectionErrorr� Z_taskr �boolr� rF Zis_eofrN r Zreadanyr� r� r� r� Z
set_exception�_PAYLOAD_ACCESS_ERRORr� r� r� r@ r\ rK rL r� r� )rx rm rl re r� r= r� r� �writerrs r� �coror� r� rh r� Zend_t�tr< r� r5 r5 r6 r� ` s� �
�
�
6
��
0
� �
�*���
����
���
�� � ��� �zRequestHandler.startr� c � sF �|� � | jdur@| j�d� d| _| jr@| j�| j�\}}}|| _|D ]\}}| jd7 _| j�||f� q%| j du s@J �z|j
} W n- t yr |du rV| � d� n| � d|��� t
� }
t|
j|
j|
j|
jd�}|j
} Y nw z| |�I dH |�� I dH W n ty� | �|||� |df Y S w | �|||� |dfS )a Prepare the response and write_eof, then log access.
This has to
be called within the context of any exception so the access logger
can get exception information. Returns True if the client disconnects
prematurely.
NFr z+Missing return statement on request handlerz3Web-handler should return a response instance, got )r; �reason�text�headersT)Z_finishr� Z set_upgradedr� rS r� rE rO r� rU Zprepare�AttributeErrorr� r# r& r; r� r� r� � write_eofr� r� ) rx r� r� r� r� r� r� r� r� Z prepare_methr< r5 r5 r6 r� � sF �
��
� �z RequestHandler.finish_responser� r; r= c
C s, | j dkrt|t�r| jjd|j|d� n | jd|j|d� |jjdkr(t d��d}|t
j kr�t
j j � dt
j j
� �}t
j j}d}| jr[tt�� t�� }W d � n1 sVw Y d |j�d
d �v r|rpt|�}d |� d
�}d|� d|� d|� d�}d }n
|r�|}|d | }t|||d�} | �� | S )z�Handle errors.
Returns HTTP response with specific status code. Logs additional
information. It always closes current connection.
r z Error handling request from %sr� r zMResponse is sent already, cannot send another response with the error messagez
text/plain� Nz text/htmlZAccept� z<h2>Traceback:</h2>
<pre>z</pre>z<html><head><title>z</title></head><body>
<h1>z</h1>
z
</body></html>
z
)r; r� � content_type)rE r� r rY rZ Zremoter� r� Z output_sizer� r �INTERNAL_SERVER_ERROR�value�phrase� descriptionr r� � traceback�
format_excr� �get� html_escaper& r� )
rx r� r; r< r= �ct�titler� �tbr� r5 r5 r6 r� $ sN �� �
� ���� zRequestHandler.handle_error�err_infoc s dt dtf� �fdd� }|S )Nr� r| c � s ��� | � j� j� j�S r� )r� r; r< r= )r� �r� rx r5 r6 �handlere s ��z3RequestHandler._make_error_handler.<locals>.handler)r% r' )rx r� r� r5 r� r6 r� b s z"RequestHandler._make_error_handler)r� r� )r| N)r� NN)=r1 r2 r3 r4 � __slots__r r$ r Z
LOG_FORMATr r� ZAbstractEventLooprt r� r �typer rA r> rr r� r r r� �tupler� r� �propertyre r� Z
BaseTransportr� r@ r� r r r� r� �bytesr� r� r� r� r� r� r� r% r' r� r r� r� r� r r� r� r� r� r: r� �
__classcell__r5 r5 rz r6 r( c s
'+������ �
� � �
�������\" �
" �
2#���
�
1
���
�
���
�
+ ���
�2�����
�>��)Rr� Zasyncio.streamsr� r� r� � collectionsr �collections.abcr r r �
contextlibr �htmlr r� �httpr �loggingr �typingr
r r r
�attrZyarlZ propcacher �abcr r Z
base_protocolr r Zhelpersr r r r r r r r Zhttp_exceptionsr r� r r Zstreamsr r Z tcp_helpersr! Zweb_exceptionsr" r# Zweb_logr$ Z web_requestr% Z web_responser& r' �__all__rp �sslZ
web_serverr, Z_RequestFactoryZ_RequestHandlerZURLr� r� r) r* r� �sr: r� Z_MsgTyper( r5 r5 r5 r6 �<module> s| ��� �