File: //usr/local/lib/python3.10/dist-packages/aiohttp/__pycache__/web_protocol.cpython-310.pyc
o
���g d �
@ s$ d dl Z d dlZ d dlZd dlZd dlZd dlmZ d dlmZ d dl m
Z d dl m
Z
d dlmZ d dlmZmZmZmZmZmZmZmZmZmZmZ d dl Z d dl Z dd l m Z m Z dd
l!m"Z" dd l#m$Z$ dd l m%Z%m&Z&m'Z'm(Z(m)Z) dd
l*m+Z+ ddl,m-Z-m.Z. ddl/m0Z0m1Z1 ddl2m3Z3 ddl4m5Z5m6Z6 ddl7m8Z8 ddl9m:Z: ddl;m<Z<m=Z= dZ>er�ddl?m@Z@ ee(e1de dge:f ZAee:gee= f ZBe(dde'i i ddd d e �Cd��
ZDG d d � d eE�ZFG d d � d eE�ZGeG� ZHe jIdddd!�G d"d#� d#��ZJeee(eJf e1f ZKG d$d� de"�ZLdS )%� N)�deque)�suppress)�escape)�
HTTPStatus)�Logger) �
TYPE_CHECKING�Any� Awaitable�Callable�Deque�Optional�Sequence�Tuple�Type�Union�cast� )�AbstractAccessLogger�AbstractStreamWriter)� BaseProtocol)� ceil_timeout)�HttpProcessingError�HttpRequestParser�
HttpVersion10�RawRequestMessage� StreamWriter)�
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__� r4 r4 �?/usr/local/lib/python3.10/dist-packages/aiohttp/web_protocol.pyr) P � r) c @ r. )r* z-Payload was accessed after response was sent.Nr/ r4 r4 r4 r5 r* T r6 r* )� auto_attribs�frozen�slotsc @ s& e Zd ZU eed< eed< eed< dS )�_ErrInfo�status�exc�messageN)r0 r1 r2 �int�__annotations__�
BaseException�strr4 r4 r4 r5 r: [ s
r: c ! sF e Zd ZdZdZddeeeejdddddd dd
d �d d
de j
de de de
dee de
dede dededede dede d e f � fd d �Zd efd d!�Zed e fd"d#��Zd\d%ee d d&fd'd(�Zd)e jd d&f� fd*d+� Zd,ee d d&f� fd-d.� Zd/e d d&fd0d1�Z d]d2d3�Z d4e d d&fd5d6�Z d7e d d&fd8d9�Z!d]d:d;�Z"d]d<d=�Z#d>e$d?e%d@e d d&fdAdB�Z&dCe dDe d d&fdEdF�Z'dCe dDe d d&fdGdH�Z(d]dIdJ�Z)d>e$dKe dLe*e$ge+e% f d e,e%e f fdMdN�Z-d]dOdP�Z.d>e$dQe%dKe d e,e%e f fdRdS�Z/ T & &d^d>e$dUed,ee dVee d e%f
dWdX�Z0dYe1d e*e$ge+e% f fdZd[�Z2� Z3S )_r( 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
) �_request_count�
_keepalive�_manager�_request_handler�_request_factory�_tcp_keepalive�_next_keepalive_close_time�_keepalive_handle�_keepalive_timeout�_lingering_time� _messages�
_message_tail�_handler_waiter�_waiter�
_task_handler�_upgrade�_payload_parser�_request_parser�_reading_paused�logger�debug�
access_logr �_close� _force_close�_current_request�_timeout_ceil_threshold�_request_in_progressi. TFi� i � g $@i � )�keepalive_timeoutr! rU �access_log_classrW �access_log_formatrV �
max_line_size� max_headers�max_field_size�lingering_time� read_bufsize�auto_decompress�timeout_ceil_threshold�managerr+ �loopr^ r! rU r_ rW r` rV ra rb rc rd re rf rg c
s t � �|� d| _d| _d | _|| _|j| _|j| _ || _
d| _ d | _ || _
t|
�| _t� | _d| _d | _d | _d | _d| _d | _t| |||
| | t|d�| _d| _zt|�| _W n t t fyc Y nw || _ | | _ || _ |rv|||�| _!nd | _!d| _"d| _#d| _$d S )Nr Fg � )ra rc rb �payload_exceptionrf r] )%�super�__init__rB rC rZ rD �request_handlerrE �request_factoryrF rG rH rI rJ �floatrK r rL rM rO rN rP rQ rR r r) rS r[ � TypeError�
ValueErrorrU rV rW r rX rY r\ )�selfrh ri r^ r! rU r_ rW r` rV ra rb rc rd re rf rg �� __class__r4 r5 rm � s\
� ��
zRequestHandler.__init__�returnc C s d� | jj| jd ur
d�S d�S )Nz<{} {}>� connected� disconnected)�formatru r0 � transport�rs r4 r4 r5 �__repr__� s ��zRequestHandler.__repr__c C s | j S �N)rJ r{ r4 r4 r5 r^ s z RequestHandler.keepalive_timeout� .@�timeoutNc � 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�� � )rY rI �cancelr\ �_loop�
create_futurerN r �asyncio�CancelledError� TimeoutError�sys� version_info� current_task�
cancellingrZ �_cancelrP �done�shield� force_close)rs r �taskr4 r4 r5 �shutdown sT �
(��
����
�(��
����
zRequestHandler.shutdownrz c s� t � �|� ttj|�}| jrt|� | jd usJ �| j�| |� | j}t j
dkr4tj | � � |dd�}n|�
| � � �}|| _d S )N�r� � T�ri � eager_start)rl �connection_mader r� � TransportrG r! rD r� r� r� �Task�start� create_taskrP )rs rz �real_transportri r� rt r4 r5 r� 6 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)rD �connection_lost�handler_cancellationr� rl rF rE rS rI r� rZ �ConnectionResetErrorr� rP rR �feed_eof)rs r< r� rt r4 r5 r� G s.
�z RequestHandler.connection_lost�parserc C s6 | j d u sJ �|| _ | jr| j �| j� d| _d S d S )Nrj )rR rM � feed_data)rs r� r4 r4 r5 �
set_parserg s
�zRequestHandler.set_parserc C s d S r} r4 r{ r4 r4 r5 � eof_receivedq s zRequestHandler.eof_received�datac
C sN | 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 |pGdD ]\}}| j
d7 _
| j � ||f� qH| j
}|rn|d urn|�� sn|�d � || _|rz|r||| _d S d S d S | jd u r�| jr�|r�| j|7 _d S |r�| j�|�\} }| r�| �� d S d S d S )Ni� )r; r< r= Frj r4 r )rY rX rR rQ rS r� r r: r= r rB rL �appendrO r� �
set_resultrM �close)
rs r� �messages�upgraded�tailr< �msg�payload�waiter�eofr4 r4 r5 �
data_receivedt s> ���
� �z RequestHandler.data_received�valc C s$ || _ | jr| j�� d| _dS dS )zMSet keep-alive connection mode.
:param bool val: new state.
N)rC rI r� )rs r� r4 r4 r5 �
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)rX rO r� r{ r4 r4 r5 r� � s �zRequestHandler.closec C s8 d| _ | jr | j�� | jdur| j�� d| _dS dS )z Forcefully close connection.TN)rY rO r� rz r� r{ r4 r4 r5 r� � s
�zRequestHandler.force_close�request�response�timec C s8 | j d ur| j jr| j �||| j�� | � d S d S d S r} )r �enabled�logr� r� )rs r� r� r� r4 r4 r5 �
log_access� s �zRequestHandler.log_access�args�kwc O s | j r| jj |i |�� d S d S r} )rV rU �rs r� r� r4 r4 r5 � log_debug� s �zRequestHandler.log_debugc O s | j j|i |�� d S r} )rU � exceptionr� r4 r4 r5 �
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} ) rI rY rC r� r� rH �call_at�_process_keepaliverO r� r� )rs ri �now�
close_timer4 r4 r5 r� � s �z!RequestHandler._process_keepalive�
start_timern 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� �� �__http_exception__Fzlreturning HTTPException object is deprecated (#2415) and will be removed, please raise the exception instead)r\ rZ r" �finish_responser� r� r� r� � handle_error� Exception�getattr�warnings�warn�DeprecationWarningrN r� )rs r� r� rn �respr< �resetr4 r4 r5 �_handle_request� sL �$� $�$�� ��
�
�z RequestHandler._handle_requestc
� s< �| j }t�|�}|dusJ �| j}|dusJ �| j}d}| jdus$J �| jdus+J �| j�s�| jsGz|� � | _
| j
I dH W d| _
nd| _
w | j� � \}}|� � }| j
d7 _
t| |�} t|t�rk| �|�}
t}n| j}
| �||| | |�} �zȐz[| �| ||
�} tjdkr�tj| |dd�}
n|�| �}
z |
I dH \}}W nJ ty� | �d� Y W W | jdu r�|dur�| �d� �n�| js�| jr�| j s�|dur�|� � }|| }|| _ | j du r�|� || j �| _ �n��n��n�w ~
|�r6| �d� W W | jdu �r|du�r| �d� �n�| j�s4| j�r2| j �s2|du�r0|� � }|| }|| _ | j du �r0|� || j �| _ �nT�nR�nPt!|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 �s�w Y |� � }|�#� �s�||k �sdW n! tj'tj(f�y� tjd
k�r�t�� }�r�|�)� �r�� Y nw |�#� �s�| j�s�| �d � | �*� |�+t,� W n, tj'�y� | �d� � t-�y } z| j.d |d
� | �/� W Y d}~nd}~ww W | jdu �r|du�r| �d� no| j�s@| j�r?| j �s?|du�r>|� � }|| }|| _ | j du �r>|� || j �| _ nDnFnB| jdu �rR|du�rR| �d� w | j�s�| j�r}| j �s}|du�r||� � }|| }|| _ | j du �r||� || j �| _ w Y nw | jr/| j�s�d| _0| jdu�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� )1r� r� r� rD rJ rF rE rY rL r� rO �popleftr� �requests_countr �
isinstancer: �_make_error_handler�ERRORr� r� r� r� r� �ConnectionErrorr� rz rC rX rH rI r� r� �boolr� �is_eofrK r �readanyr� r� r� r� �
set_exception�_PAYLOAD_ACCESS_ERRORr� r� r� rP )rs ri �handlerrh r^ r� r= r� r� �writerrn r� �coror� r� r� r� rd �end_t�tr< r4 r4 r5 r� s" �
2
����
,
��� �
�*���
����
���
����
�
��e �zRequestHandler.startr� c � s �|� � | jdur | j�d� d| _| jr | j�| j� d| _z|j}W n- tyR |du r6| �d� n| �d� |�� t
� }t |j |j
|j|jd�}|j}Y nw z||�I dH |�� I dH W n tyw | �|||� |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.
NFrj z+Missing return statement on request handlerz7Web-handler should return a response instance, got {!r})r; �reason�text�headersT)�_finishrS � set_upgradedrQ rM r� �prepare�AttributeErrorr� ry r# r& r; r� r� r� � write_eofr� r� )rs r� r� r� � prepare_methr< r4 r4 r5 r� ~ s@ �
��
�
�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 krd� t
j �}t
j j
}d}| jrVtt�� t�� }W d � n1 sQw Y d |j�d
d �v ru|rkt|�}d |� d
�}dj ||d�}d }n
|ry|}|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/plainz{0.value} {0.phrase}Nz text/html�Accept� z<h2>Traceback:</h2>
<pre>z</pre>zV<html><head><title>{title}</title></head><body>
<h1>{title}</h1>
{msg}
</body></html>
)�titler� z
)r; r� � content_type)rB r� r rU rV �remoter� r� � output_sizer� r �INTERNAL_SERVER_ERRORry � descriptionr r� � traceback�
format_excr� �get� html_escaper& r� )
rs r� r; r< r= �ctr� r� �tbr� r4 r4 r5 r� � sD �� �
� � zRequestHandler.handle_error�err_infoc s dt dtf� �fdd� }|S )Nr� rv c � s ��� | � j� j� j�S r} )r� r; r< r= )r� �r� rs r4 r5 r� � s ��z3RequestHandler._make_error_handler.<locals>.handler)r% r' )rs r� r� r4 r� r5 r� � s z"RequestHandler._make_error_handler)r~ )rv N)r� NN)4r0 r1 r2 r3 � __slots__r r$ r �
LOG_FORMATr� �AbstractEventLooprp r� r r r rA r> rm r| �propertyr^ r r� �
BaseTransportr� r@ r� r r� r� �bytesr� r� r� r� r% r' r� r� r� r� r
r r r� r� r� r� r: r� �
__classcell__r4 r4 rt r5 r( e s� '%������ �
� � �
�������N2
'
���
�
���
�
+}���
�-�����
�>��)Mr� �asyncio.streamsr� r� r� � collectionsr �
contextlibr �htmlr r� �httpr �loggingr �typingr r r r
r r r
r r r r �attr�yarl�abcr r �
base_protocolr �helpersr r r r r r �http_exceptionsr r� r r �streamsr r � tcp_helpersr! �web_exceptionsr" r# �web_logr$ � web_requestr% � web_responser& r' �__all__�
web_serverr+ �_RequestFactory�_RequestHandler�URLr� r� r) r* r� �sr: �_MsgTyper( r4 r4 r4 r5 �<module> st 4 ��� �