Edit File: web_protocol.cpython-37.opt-1.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]ZUNKNOWN�/TFc @ s e Zd ZdZdS )r' zPayload parsing error.N)�__name__� __module__� __qualname__�__doc__� r/ r/ �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. r/ r/ r/ r0 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�strr/ r/ r/ r0 r3 V s r3 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 rL �access_log_classrN �access_log_formatrM � max_line_size� max_headers�max_field_size�lingering_time� read_bufsize�auto_decompressr) )�manager�looprR r rL rS rN rT rM rU rV rW rX rY rZ 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 � )rU rW rV Zpayload_exceptionrZ ) �super�__init__r; r< rQ r= �request_handlerr> Zrequest_factoryr? r@ rA rB rC �floatrD r rE rF rG rH rI rJ r r' rK rL rM rN r rO rP )�selfr[ r\ rR r rL rS rN rT rM rU rV rW rX rY rZ )� __class__r/ r0 r_ � sH zRequestHandler.__init__)�returnc C s d� | jj| jd k rdnd�S )Nz<{} {}>Z connectedZ disconnected)�formatrc r+ � transport)rb r/ r/ r0 �__repr__� s zRequestHandler.__repr__c C s | j S )N)rC )rb r/ r/ r0 rR � s z RequestHandler.keepalive_timeout� .@N)�timeoutrd 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)rP rB �cancelrG r �asyncio�CancelledError� TimeoutErrorr rQ �_cancelrH �donerf �close)rb ri r/ r/ r0 �shutdown� s & zRequestHandler.shutdown)rf rd c sX t � �|� ttj|�}| jr&t|� | j�| � � �| _ | j d k sFt �| j �| |� d S )N) r^ �connection_mader rk Z Transportr@ r �_loop� create_task�startrH r= �AssertionError)rb rf Zreal_transport)rc r/ r0 rr s z RequestHandler.connection_made)r5 rd 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^ rP r? r> rK rB rj rQ �ConnectionResetErrorrn rH rG rJ Zfeed_eof)rb r5 )rc r/ r0 rw s. z RequestHandler.connection_lost)�parserrd c C s2 | j d kst�|| _ | jr.| j �| j� d| _d S )Nr] )rJ rv rF � feed_data)rb ry r/ r/ r0 � set_parser9 s zRequestHandler.set_parserc C s d S )Nr/ )rb r/ r/ r0 � eof_receivedC s zRequestHandler.eof_received)�datard 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� )r4 r5 r6 Fr] r/ r )rP rO rJ rI rK rv rz r r3 r6 r r; rE �appendrG ro Z set_resultrF rp ) rb r} ZmessagesZupgraded�tailr5 �msg�payload�waiter�eofr/ r/ r0 � data_receivedF s2 z RequestHandler.data_received)�valrd c C s || _ | jr | j�� d| _dS )zMSet keep-alive connection mode. :param bool val: new state. N)r< rB rj )rb r� r/ r/ r0 � 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)rO rG rj )rb r/ r/ r0 rp w s zRequestHandler.closec C s4 d| _ | jr| j�� | jdk r0| j�� d| _dS )z Forcefully close connection.TN)rP rG rj rf rp )rb r/ r/ r0 � force_close� s zRequestHandler.force_close)�request�response�timerd c C s( | j d k r$| j �||| j�� | � d S )N)r �logrs r� )rb r� r� r� r/ r/ r0 � log_access� s zRequestHandler.log_access)�args�kwrd c O s | j r| jj ||� d S )N)rM rL )rb r� r� r/ r/ r0 � log_debug� s zRequestHandler.log_debugc O s | j j||� d S )N)rL Z exception)rb r� r� r/ r/ r0 � 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) rP r< rA rC rG rs r� r� Z call_later�KEEPALIVE_RESCHEDULE_DELAY�_process_keepaliverB )rb �nextr/ r/ r0 r� � s z!RequestHandler._process_keepalive)r� � start_timer` rd 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> rv rQ r! �finish_responserk rl rm r� � handle_error� Exception�getattr�warnings�warn�DeprecationWarning)rb r� r� r` �respr5 �resetr/ r/ r0 �_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.)0rs rH rv r= rC r? r> rP rE Z create_futurerG rk rl �popleftr� Zrequests_countr � isinstancer3 �_make_error_handler�ERRORrt r� �ConnectionErrorr� �boolr� r< Zis_eofrD r rm r Zreadanyrp Z set_exceptionr( � RuntimeErrorrM r� r� r� rf rO rA rB Zcall_atr� )rb r\ �handlerr[ rR r� r6 r� ru �writerr` r� Ztaskr� rX ZnowZend_tr5 r/ r/ r0 ru � s� zRequestHandler.start)r� r� r� rd 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) rK Z set_upgradedrI rF rz Zprepare�AttributeErrorr� re Z write_eofr� r� )rb r� r� r� Z prepare_methr/ r/ r0 r� I s, z RequestHandler.finish_response�� )r� r4 r5 r6 rd 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 )r4 �textZ content_type)r� r� Z output_sizer� r ZINTERNAL_SERVER_ERRORre Z descriptionrM r r� � traceback� format_excZheaders�get� html_escaper$ r� ) rb r� r4 r5 r6 Zctr� r� �tbr� r/ r/ r0 r� n s2 zRequestHandler.handle_error)�err_inford c s t td�� �fdd� }|S )N)r� rd c � s �� | � j� j� j�S )N)r� r4 r5 r6 )r� )r� rb r/ r0 r� � s z3RequestHandler._make_error_handler.<locals>.handler)r# r% )rb r� r� r/ )r� rb r0 r� � s z"RequestHandler._make_error_handler)rh )r� NN)5r+ r, r- r. r� � __slots__r r" r Z LOG_FORMATrk ZAbstractEventLoopra r� r r r r: r7 r_ rg �propertyrR r rq Z BaseTransportrr r9 rw r r{ r| �bytesr� r� rp r� r# r% r� r� r� r� r r r r� ru r� r� r3 r� � __classcell__r/ r/ )rc r0 r& ` s\ " :2 ' "z& .)Hrk Zasyncio.streamsr� r� � collectionsr � contextlibr Zhtmlr r� Zhttpr Zloggingr �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_RequestHandlerZURLr� r� r' r( �sr3 Z_MsgTyper& r/ r/ r/ r0 �<module> sb 4