U
'�7`�Z �
@ s� d dl Z d dlZ d dlZd dlZd dlmZ d dlmZ d dlm Z
d dl m Z d dl
mZ d dlmZmZmZmZmZmZmZmZ d dlZdd lmZmZ dd
l m Z dd l m Z m Z dd l m!Z!m"Z"m#Z#m$Z$m%Z% dd
l&m'Z'm(Z( ddl)m*Z*m+Z+ ddl,m-Z- ddl.m/Z/ ddl0m1Z1 ddl2m3Z3 ddl4m5Z5m6Z6 dZ7e�rJddl8m9Z9 ee$e+dedge3f Z:ee3gee6 f Z;e$dde#i i dddde�<d��
Z=G d d � d e>�Z?G d d � d e>�Z@G d d� de �ZAdS )!� N)�deque)�suppress)�escape)�
HTTPStatus)�Logger)�
TYPE_CHECKING�Any� Awaitable�Callable�Optional�Tuple�Type�cast� )�AbstractAccessLogger�AbstractStreamWriter)� BaseProtocol)� CeilTimeout� current_task)�HttpProcessingError�HttpRequestParser�
HttpVersion10�RawRequestMessage� StreamWriter)�
access_logger�
server_logger)�
EMPTY_PAYLOAD� StreamReader)�
tcp_keepalive)�
HTTPException)� AccessLogger)� BaseRequest)�Response�StreamResponse)�RequestHandler�RequestPayloadError�PayloadAccessError)�Serverr$ zasyncio.Task[None]ZUNKNOWN�/TFc @ s e Zd ZdZdS )r% zPayload parsing error.N��__name__�
__module__� __qualname__�__doc__� r. r. �G/opt/alt/python38/lib64/python3.8/site-packages/aiohttp/web_protocol.pyr% 9 s r% c @ s e Zd ZdZdS )r& z-Payload was accessed after response was sent.Nr) r. r. r. r/ r& = s r& c s� e Zd ZdZdZdZddeeeej ddddd d
d � d e
j e e
eee eee
eeee ed
�� fdd�Zed�dd�Zee d�dd��ZdHee dd�dd�Ze
jdd�� fdd � Zee dd �� fd d � Z e dd �d!d"�Z dd�d#d$�Z e dd%�d&d'�Z!e
dd(�d)d*�Z"dd�d+d,�Z#dd�d-d.�Z$e%e&e dd/�d0d1�Z'e e dd2�d3d4�Z(e e dd2�d5d6�Z)dd�d7d8�Z*e%e e+e&e
f d9�d:d;�Z,dd�d<d=�Z-e%e&e e
d>�d?d@�Z.dIe%eee ee e&dB�dCdD�Z/dJe0eee ee ddE�dFdG�Z1� Z2S )Kr$ a� 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.
:param keepalive_timeout: number of seconds before closing
keep-alive connection
:type keepalive_timeout: int or None
:param bool tcp_keepalive: TCP keep-alive is on, default is on
:param bool debug: enable debug mode
:param logger: custom logger object
:type logger: aiohttp.log.server_logger
:param access_log_class: custom class for access_logger
:type access_log_class: aiohttp.abc.AbstractAccessLogger
:param access_log: custom logging object
:type access_log: aiohttp.log.server_logger
:param str access_log_format: access log format string
:param loop: Optional event loop
:param int max_line_size: Optional maximum header line size
:param int max_field_size: Optional maximum header field size
:param int max_headers: Optional maximum header size
r )�_request_count�
_keepalive�_manager�_request_handler�_request_factory�_tcp_keepalive�_keepalive_time�_keepalive_handle�_keepalive_timeout�_lingering_time� _messages�
_message_tail�_waiter�_error_handler�
_task_handler�_upgrade�_payload_parser�_request_parserZ_reading_paused�logger�debug�
access_logr �_close� _force_close�_current_requestg �R@TFi� i � g $@i ) �keepalive_timeoutr rB �access_log_classrD �access_log_formatrC �
max_line_size� max_headers�max_field_size�lingering_time� read_bufsizer' )�manager�looprH r rB rI rD rJ rC rK rL rM rN rO c
s� t � �|� d| _d| _d | _|| _|j| _|j| _ || _
d| _ d | _ || _
t|
�| _t� | _d| _d | _d | _d | _d| _d | _t| |||
| | td�| _|| _| | _ || _ |r�|||�| _ nd | _ d| _ d| _ d S )Nr Fg � )rK rM rL Zpayload_exception)!�super�__init__r0 r1 rG r2 Zrequest_handlerr3 Zrequest_factoryr4 r5 r6 r7 r8 �floatr9 r r: r; r< r= r> r? r@ r r% rA rB rC rD r rE rF )�selfrP rQ rH r rB rI rD rJ rC rK rL rM rN rO �� __class__r. r/ rT � sR ��
�
�zRequestHandler.__init__)�returnc C s d� | jj| jd k rdnd�S )Nz<{} {}>Z connectedZ disconnected)�formatrX r* � transport�rV r. r. r/ �__repr__� s �zRequestHandler.__repr__c C s | j S �N)r8 r\ r. r. r/ rH � s z RequestHandler.keepalive_timeout� .@N)�timeoutrY c
� s� d| _ | jdk r| j�� | jr*| j�� ttjtj��z t|| j d��` | j
dk rj| j
� � sj| j
I dH | j dk r�| j �
t�� � | jdk r�| j� � s�| jI dH W 5 Q R X W 5 Q R X | jdk r�| j�� | jdk r�| j�� d| _dS )z�Worker process is about to exit, we need cleanup everything and
stop accepting requests. It is especially important for keep-alive
connections.TN�rQ )rF r7 �cancelr< r �asyncio�CancelledError� TimeoutErrorr �_loopr= �donerG �_cancelr> r[ �close)rV r` r. r. r/ �shutdown� s$
zRequestHandler.shutdown)r[ rY c sX t � �|� ttj|�}| jr&t|� | j�| � � �| _
| j d k sFt �| j �| |� d S r^ )
rS �connection_mader rc Z Transportr5 r rf � create_task�startr> r2 �AssertionError)rV r[ Zreal_transportrW r. r/ rk � s z RequestHandler.connection_made)�excrY c s� | j d krd S | j �| |� t� �|� d | _ d| _d | _d | _d | _| jd k rZ| j�� | j d k r�|d krtt
d�}| j � |� | j d k r�| j �� | j
d k r�| j
�� | jd k r�| j�� d | _
| jd k r�| j�� d | _d S )NTzConnection lost)r2 �connection_lostrS rF r4 r3 rA r7 rb rG �ConnectionResetErrorrh r= r> r< r@ Zfeed_eof)rV ro rW r. r/ rp s2
z RequestHandler.connection_lost)�parserrY c C s2 | j d kst�|| _ | jr.| j �| j� d| _d S )NrR )r@ rn r; � feed_data)rV rr r. r. r/ �
set_parser& s
zRequestHandler.set_parserc C s d S r^ r. r\ r. r. r/ � eof_received0 s zRequestHandler.eof_received)�datarY c
C s� | j s | jrd S | jd k�rP| j�sP| jd k s2t�z| j�|�\}}}W n� tk
r� } z0| j� | �
t | | j�d||j ��| _
| �� W 5 d }~X Y n� tk
r� } z,| j� | �
t | | j�d|��| _
| �� W 5 d }~X Y npX |�r6|D ]&\}}| jd7 _| j�||f� q�| j}|d k �r6|�� �s6|�d � || _|�r�|�r�|| _nN| jd k�rz| j�rz|�rz| j|7 _n$|�r�| j�|�\} }| �r�| �� d S )Ni� �� r )rF rE r@ r? rA rn rs r rf rl �handle_parse_errorr �messager= ri � Exceptionr0 r: �appendr< rg Z
set_resultr; )
rV rv ZmessagesZupgraded�tailro �msg�payloadZwaiter�eofr. r. r/ �
data_received3 sL
���
z RequestHandler.data_received)�valrY c C s || _ | jr | j�� d| _dS )zMSet keep-alive connection mode.
:param bool val: new state.
N)r1 r7 rb )rV r� r. r. r/ �
keep_alived s
zRequestHandler.keep_alivec C s d| _ | jr| j�� dS )ziStop accepting new pipelinig messages and close
connection when handlers done processing messagesTN)rE r< rb r\ r. r. r/ ri n s zRequestHandler.closec C s4 d| _ | jr| j�� | jdk r0| j�� d| _dS )zForce close connectionTN)rF r< rb r[ ri r\ r. r. r/ � force_closeu s
zRequestHandler.force_close)�request�response�timerY c C s( | j d k r$| j �||| j�� | � d S r^ )r �logrf r� )rV r� r� r� r. r. r/ �
log_access~ s
zRequestHandler.log_access)�args�kwrY c O s | j r| jj ||� d S r^ )rC rB �rV r� r� r. r. r/ � log_debug� s zRequestHandler.log_debugc O s | j j||� d S r^ )rB Z exceptionr� r. r. r/ �
log_exception� s z RequestHandler.log_exceptionc C sT | j s | jsd S | j| j }| jr<| j�� |kr<| �� d S | j�| j | j
�| _ d S r^ ) rF r1 r6 r8 r< rf r� r� Z
call_later�KEEPALIVE_RESCHEDULE_DELAY�_process_keepaliver7 )rV �nextr. r. r/ r� � s �z!RequestHandler._process_keepalive)r� �
start_timerY c
� sL | j d k st�z(z|| _| � |�I d H }W 5 d | _X W n� tk
r� } z0t|j|j|j|jd�}| � |||�I d H }W 5 d }~X Y n� t
j k
r� � Y n� t
j k
r� } z2| j
d|d� | �|d�}| � |||�I d H }W 5 d }~X Y nZ tk
�r. } z&| �|d|�}| � |||�I d H }W 5 d }~X Y nX | � |||�I d H }||fS )N)�status�reason�text�headerszRequest handler timed out.��exc_infoi� rw )r3 rn rG r r" r� r� r� r� �finish_responserc rd re r� � handle_errorrz )rV r� r� �respro �resetr. r. r/ �_handle_request� s2 �$ $&z RequestHandler._handle_requestc � s� | j }| j}|dk st�| j}|dk s*t�| j}d}| jdk sBt�| jdk sPt�| j�s�| js�z@z|�
� | _ | j I dH W n t j k
r� Y W ��q�Y nX W 5 d| _ X | j�
� \}}|�� }| jd7 _t| |�} | �||| | |�}
d�z$�zv| j �| �|
|��} z| I dH \}}
W n6 t j tfk
�rN | �d� Y W W ��� �q�Y nX t |dd��rjt � dt � ~ |
�r�| �d� W W ��� �q�t |j!�| _|�"� �sX| j#}| j�s4|�r4| �d |� |�� } | | }t$t j%t j ��P |�"� �s*| |k �r*t&|| |d
�� |�'� I dH W 5 Q R X |�� } �q�W 5 Q R X |�"� �sX| j�sX| �d � | �(� |�)t*� � W n� t j k
�r� | �d � Y W �� �q�Y nz t+k
�r� } z"| j,�r�| j-d
|d� | �.� W 5 d}~X Y n: t/k
�r } z| j-d|d� | �.� W 5 d}~X Y nX W 5 | jdk�r4|dk �r4| �d� n\| j�s�| j�r�| j�s�|dk �r�| j �� } | | _| jdk�r�|�| | | j�| _n� �q�X qP| j�s�d| _| jdk �r�| j0dk�r�| j�(� 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 z'Ignored premature client disconnection.z&Ignored premature client disconnectionZ__http_exception__Fzlreturning HTTPException object is deprecated (#2415) and will be removed, please raise the exception insteadz(Ignored premature client disconnection 2z'Start lingering close timer for %s sec.ra zUncompleted request.z'Ignored premature client disconnection zUnhandled runtime exceptionr� zUnhandled exception)1rf r> rn r2 r8 r4 r3 rF r: r< Z
create_futurerc rd �popleftr� Zrequests_countr r[ r� r1 rE r6 r7 Zcall_atr� rl r� �ConnectionError�getattr�warnings�warn�DeprecationWarning�boolr� Zis_eofr9 r re r Zreadanyri Z
set_exceptionr& � RuntimeErrorrC r� r� rz r= )rV rQ ZhandlerrP rH r� ry r~ rm �writerr� Znow�taskr� rN Zend_tro r. r. r/ rm � s�
�
�
�zRequestHandler.start)r� r� r� rY c � s� | j dk r6| j �d� d| _| jr6| j �| j� d| _z
|j}W n4 tk
rt |dkrbtd��ntd�|���Y nX z ||�I dH |� � I dH W n$ t
k
r� | � |||� Y dS X | � |||� dS dS )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.
NFrR z+Missing return statement on request handlerz7Web-handler should return a response instance, got {!r}T) rA Z set_upgradedr? r; rs �prepare�AttributeErrorr� rZ � write_eofr� r� )rV r� r� r� Z prepare_methr. r. r/ r� 4 s0
��
z RequestHandler.finish_responserw )r� r� ro ry rY c
C s� | j d|d� d}|tjkr�d�tj�}tjj}d}| jrVtt�� t� � }W 5 Q R X d|j
� dd�kr�|r�t |�}d |� d
�}d j||d �}d}n|r�|}|d
| }t
|||d�} | �� |jjdks�| jdkr�| �� | S )z�Handle errors.
Returns HTTP response with specific status code. Logs additional
information. It always closes current connection.zError handling requestr� z
text/plainz{0.value} {0.phrase}Nz text/htmlZAccept� 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� Z content_typer )r� r ZINTERNAL_SERVER_ERRORrZ � descriptionrC r rz � traceback�
format_excr� �get� html_escaper" r� r� Z output_sizer[ )
rV r� r� ro ry Zctr� r} �tbr� r. r. r/ r� X s6
� � zRequestHandler.handle_error)r� r� ro ry rY c � sr t � }|d k st�ttt| ||| j�}| �||||�}|�|�I d H |�� I d H | j d k rh| j �
� d | _ d S r^ ) r rn r! �ERRORr rf r� r� r� r[ ri r= )rV r� r� ro ry r� r� r� r. r. r/ rx � s �
z!RequestHandler.handle_parse_error)r_ )rw NN)NN)3r* r+ r, r- r� � __slots__r r r Z
LOG_FORMATrc ZAbstractEventLooprU r� r r
r �str�intrT r] �propertyrH r rj Z
BaseTransportrk �
BaseExceptionrp r rt ru �bytesr� r� ri r� r! r# r� r� r� r� r r� rm r� r� r rx �
__classcell__r. r. rW r/ r$ A s� '"��G "
1
�
� y � ' �� 3 ��)Brc Zasyncio.streamsr� r� � collectionsr �
contextlibr Zhtmlr r� Zhttpr Zloggingr �typingr r r r
r r r
r Zyarl�abcr r Z
base_protocolr Zhelpersr r r r r r r r� r r Zstreamsr r Z tcp_helpersr Zweb_exceptionsr Zweb_logr Z web_requestr! Z web_responser" r# �__all__Z
web_serverr' Z_RequestFactoryZ_RequestHandlerZURLr� rz r% r& r$ r. r. r. r/ �<module> sd ( ��� �