Edit File: web_protocol.cpython-37.pyc
B �k�aX � @ s d dl Z d dlZ d dlZd dlZd dlmZ d dlmZ d dlm Z d dl m Z d dl mZ d dlmZmZmZmZmZmZmZmZmZmZmZ d dlZd 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*m+Z+ ddl,m-Z-m.Z. ddl/m0Z0 ddl1m2Z2 ddl3m4Z4 ddl5m6Z6 ddl7m8Z8m9Z9 dZ:e�rZddl;m<Z< ee'e.de dge6f Z=ee6gee9 f Z>e'dde&i i dddde �?d�� Z@G d d � d eA�ZBG d d � d eA�ZCejDdddd �G d!d"� d"��ZEeee'eEf e.f ZFG d#d� de!�ZGdS )$� 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)� access_logger� server_logger)� EMPTY_PAYLOAD� StreamReader)� tcp_keepalive)� HTTPException)� AccessLogger)� BaseRequest)�Response�StreamResponse)�RequestHandler�RequestPayloadError�PayloadAccessError)�Serverr& zasyncio.Task[None]�UNKNOWN�/TFc @ s e Zd ZdZdS )r' zPayload parsing error.N)�__name__� __module__� __qualname__�__doc__� r0 r0 �G/opt/alt/python37/lib64/python3.7/site-packages/aiohttp/web_protocol.pyr' N s r' c @ s e Zd ZdZdS )r( z-Payload was accessed after response was sent.N)r, r- r. r/ r0 r0 r0 r1 r( R s r( )Z auto_attribs�frozen�slotsc @ s& e Zd ZU eed< eed< eed< dS )�_ErrInfo�status�exc�messageN)r, r- r. �int�__annotations__� BaseException�strr0 r0 r0 r1 r4 V s r4 c s� e Zd ZdZdZdZddeeeej ddddd d dd � d e j e e eee eee eeee ee d �� 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%ge,e& f e-e&e f d9�d:d;�Z.dd�d<d=�Z/e%e&e e d>�d?d@�Z0dIe%eee ee e&dB�dCdD�Z1e2e+e%ge,e& f dE�dFdG�Z3� Z4S )Jr& 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. 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 r )�_request_count� _keepalive�_manager�_request_handler�_request_factory�_tcp_keepalive�_keepalive_time�_keepalive_handle�_keepalive_timeout�_lingering_time� _messages� _message_tail�_waiter� _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 rM �access_log_classrO �access_log_formatrN � max_line_size� max_headers�max_field_size�lingering_time� read_bufsize�auto_decompressr) )�manager�looprS r rM rT rO rU rN rV rW rX rY rZ r[ c s� t � �|� d| _d| _d | _|| _|j| _|j| _ || _ d| _ d | _ || _ t| �| _t� | _d| _d | _d | _d| _d | _t| ||| | | t|d�| _|| _| | _|| _ |r�|||�| _ nd | _ d| _ d| _ d S )Nr Fg � )rV rX rW Zpayload_exceptionr[ ) �super�__init__r< r= rR r> �request_handlerr? Zrequest_factoryr@ rA rB rC rD �floatrE r rF rG rH rI rJ rK r r' rL rM rN rO r rP rQ )�selfr\ r] rS r rM rT rO rU rN rV rW rX rY rZ r[ )� __class__r0 r1 r` � sH zRequestHandler.__init__)�returnc C s d� | jj| jd k rdnd�S )Nz<{} {}>Z connectedZ disconnected)�formatrd r, � transport)rc r0 r0 r1 �__repr__� s zRequestHandler.__repr__c C s | j S )N)rD )rc r0 r0 r1 rS � s z RequestHandler.keepalive_timeout� .@N)�timeoutre c � s� d| _ | jdk r| j�� | jr*| j�� ttjtj��b t|�4 I dH �@ | j dk rf| j � t�� � | j dk r�| j � � s�| j I dH W dQ I dH R X W dQ R X | j dk r�| j �� | j dk r�| j �� d| _ 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)rQ rC �cancelrH r �asyncio�CancelledError� TimeoutErrorr rR �_cancelrI �donerg �close)rc rj r0 r0 r1 �shutdown� s & zRequestHandler.shutdown)rg re c sX t � �|� ttj|�}| jr&t|� | j�| � � �| _ | j d k sFt �| j �| |� d S )N) r_ �connection_mader rl Z TransportrA r �_loop� create_task�startrI r> �AssertionError)rc rg Zreal_transport)rd r0 r1 rs s z RequestHandler.connection_made)r6 re 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 �� d | _ | jd k r�| j�� d | _d S )NTzConnection lost)r> �connection_lostr_ rQ r@ r? rL rC rk rR �ConnectionResetErrorro rI rH rK Zfeed_eof)rc r6 )rd r0 r1 rx s. z RequestHandler.connection_lost)�parserre c C s2 | j d kst�|| _ | jr.| j �| j� d| _d S )Nr^ )rK rw rG � feed_data)rc rz r0 r0 r1 � set_parser9 s zRequestHandler.set_parserc C s d S )Nr0 )rc r0 r0 r1 � eof_receivedC s zRequestHandler.eof_received)�datare c C sF | j s | jrd S | jd kr�| js�| jd k s.t�y| j�|�\}}}W n@ tk r� } z"td||j d�t fg}d}d}W d d }~X Y nX x2|p�dD ]&\}}| j d7 _ | j � ||f� q�W | j}|r�|d k r�|�� s�|�d � || _|r�|r�|| _nN| jd k�r | j�r |�r | j|7 _n$|�rB| j�|�\} }| �rB| �� d S )Ni� )r5 r6 r7 Fr^ r0 r )rQ rP rK rJ rL rw r{ r r4 r7 r r< rF �appendrH rp � set_resultrG rq ) rc r~ �messagesZupgraded�tailr6 �msg�payload�waiter�eofr0 r0 r1 � data_receivedF s2 z RequestHandler.data_received)�valre c C s || _ | jr | j�� d| _dS )zMSet keep-alive connection mode. :param bool val: new state. N)r= rC rk )rc r� r0 r0 r1 � keep_alivem s zRequestHandler.keep_alivec C s d| _ | jr| j�� dS )z�Close connection. Stop accepting new pipelining messages and close connection when handlers done processing messages. TN)rP rH rk )rc r0 r0 r1 rq w s zRequestHandler.closec C s4 d| _ | jr| j�� | jdk r0| j�� d| _dS )z Forcefully close connection.TN)rQ rH rk rg rq )rc r0 r0 r1 � force_close� s zRequestHandler.force_close)�request�response�timere c C s( | j d k r$| j �||| j�� | � d S )N)r �logrt r� )rc r� r� r� r0 r0 r1 � log_access� s zRequestHandler.log_access)�args�kwre c O s | j r| jj ||� d S )N)rN rM )rc r� r� r0 r0 r1 � log_debug� s zRequestHandler.log_debugc O s | j j||� d S )N)rM � exception)rc r� r� r0 r0 r1 � 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 )N) rQ r= rB rD rH rt r� r� Z call_later�KEEPALIVE_RESCHEDULE_DELAY�_process_keepaliverC )rc �nextr0 r0 r1 r� � s z!RequestHandler._process_keepalive)r� � start_timera re c � sP | j d k st�y&z|| _||�I d H }W d d | _X W n� tk rn } z |}| �|||�I d H }W d d }~X Y n� tjk r� � Y n� tjk r� } z2| jd|d� | � |d�}| �|||�I d H }W d d }~X Y nt t k �r } z&| � |d|�}| �|||�I d H }W d d }~X Y n0X t |dd��r4t � dt� | �|||�I d H }||fS )NzRequest handler timed out.)�exc_infoi� i� Z__http_exception__Fzlreturning HTTPException object is deprecated (#2415) and will be removed, please raise the exception instead)r? rw rR r! �finish_responserl rm rn r� � handle_error� Exception�getattr�warnings�warn�DeprecationWarning)rc r� r� ra �respr6 �resetr0 r0 r1 �_handle_request� s0 $ $&z RequestHandler._handle_requestc � s� | j }| j}|dk st�| j}|dk s*t�| j}d}| jdk sBt�| jdk sPt��x(| j�sz| js�z8y|� � | _ | j I dH W n t j k r� P Y nX W dd| _ X | j� � \}}|�� }| jd7 _t| |�} t|t�r�| �|�} t}n| j} | �||| | |�} �z��yR| j �| �| || ��} y| I dH \}} W n( t j tfk �r\ | �d� P Y nX ~ | �rr| �d� P t|j�| _|� � �sN| j }| j�s*|�r*| �d|� |�� }|| }t t j t j ��^ xV|� � �s ||k �r t || �4 I dH � |�!� I dH W dQ I dH R X |�� }�q�W W dQ R X |� � �sN| j�sN| �d� | �"� |�#t$� � W n� t j k �r� | �d� P Y nz t%k �r� } z"| j&�r�| j'd|d � | �(� W dd}~X Y n: t)k �r� } z| j'd |d � | �(� W dd}~X Y nX W d| j*dk�r |dk �r | �d � nV| j�sv| j�rt| j+�st|dk �rv| j �� }|| _,| j-dk�rv|�.|| | j/�| _-nP X qTW | j�s�d| _| j*dk �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 disconnectionz(Ignored premature client disconnection 2z'Start lingering close timer for %s sec.zUncompleted request.z'Ignored premature client disconnection zUnhandled runtime exception)r� zUnhandled exceptionz'Ignored premature client disconnection.)0rt rI rw r> rD r@ r? rQ rF Z create_futurerH rl rm �popleftr� Zrequests_countr � isinstancer4 �_make_error_handler�ERRORru r� �ConnectionErrorr� �boolr� r= Zis_eofrE r rn r Zreadanyrq � set_exceptionr( � RuntimeErrorrN r� r� r� rg rP rB rC Zcall_atr� )rc r] �handlerr\ rS r� r7 r� rv �writerra r� Ztaskr� rY �nowZend_tr6 r0 r0 r1 rv � s� zRequestHandler.start)r� r� r� re c � s� | j dk r6| j �d� d| _| jr6| j �| j� d| _y |j}W n4 tk rt |dkrbtd��ntd�|���Y nX y ||�I dH |� � I dH W n" t k r� | � |||� 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. NFr^ z+Missing return statement on request handlerz7Web-handler should return a response instance, got {!r}T) rL Z set_upgradedrJ rG r{ Zprepare�AttributeErrorr� rf Z write_eofr� r� )rc r� r� r� Z prepare_methr0 r0 r1 r� I s, z RequestHandler.finish_response�� )r� r5 r6 r7 re c C s� | j d|d� |jjdkr"td��d}|tjkr�d�tj�}tjj}d}| jrjt t �� t � � }W dQ R X d|j �d d �kr�|r�t|�}d |� d �}d j||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. zError handling request)r� r zMResponse is sent already, cannot send another response with the error messagez 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 )r5 �text� content_type)r� r� Z output_sizer� r ZINTERNAL_SERVER_ERRORrf � descriptionrN r r� � traceback� format_exc�headers�get� html_escaper$ r� ) rc r� r5 r6 r7 �ctr� r� �tbr� r0 r0 r1 r� n s2 zRequestHandler.handle_error)�err_infore c s t td�� �fdd� }|S )N)r� re c � s �� | � j� j� j�S )N)r� r5 r6 r7 )r� )r� rc r0 r1 r� � s z3RequestHandler._make_error_handler.<locals>.handler)r# r% )rc r� r� r0 )r� rc r1 r� � s z"RequestHandler._make_error_handler)ri )r� NN)5r, r- r. r/ r� � __slots__r r" r Z LOG_FORMATrl ZAbstractEventLooprb r� r r r r; r8 r` rh �propertyrS r rr Z BaseTransportrs r: rx r r| r} �bytesr� r� rq r� r# r% r� r� r� r� r r r r� rv r� r� r4 r� � __classcell__r0 r0 )rd r1 r& ` s\ " :2 ' "z& .)Hrl Zasyncio.streamsr� r� � collectionsr � contextlibr �htmlr r� �httpr �loggingr �typingr r r r r r r r r r r �attrZyarl�abcr r Z base_protocolr Zhelpersr 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_RequestHandler�URLr� r� r' r( �sr4 Z_MsgTyper& r0 r0 r0 r1 �<module> sb 4