���ZCF � @ s� d d l Z d d l Z d d l Z d d l Z d d l Z d d l m Z d d l m Z d d l
m Z d d l
Z
d d l m Z m Z d d l m Z d d l m Z m Z m Z d d l m Z m Z d d
l m Z d d l m Z m Z m Z d d l m Z d d
l m Z d d l! m" Z" d Z# e j$ d d e j% i i d d d d e
j& d � �
Z' Gd d � d e( � Z) Gd d � d e j j* e j+ � Z, d S)� N)�deque)�suppress)�escape� )�helpers�http)� CeilTimeout)�HttpProcessingError�HttpRequestParser� StreamWriter)�
access_logger�
server_logger)�
EMPTY_PAYLOAD)�tcp_cork�
tcp_keepalive� tcp_nodelay)�
HTTPException)� BaseRequest)�Response�RequestHandler�RequestPayloadErrorZUNKNOWN�/TFc @ s e Z d Z d Z d S)r zPayload parsing error.N)�__name__�
__module__� __qualname__�__doc__� r r �/web_protocol.pyr s c s� e Z d Z d Z d Z d Z d Z d d d d d d
d e d e j d
e
d e j j d d d d d d d d d d � f d d � Z d d � Z
e d d � � Z d d d � Z � f d d! � Z � f d" d# � Z d$ d% � Z d& d' � Z d( d) � Z d* d+ � Z d, d- � Z d d. d/ � Z d0 d1 � Z d2 d3 � Z d4 d5 � Z d6 d7 � Z d8 d9 � Z d: d; � Z d< d= � Z d> d d d? d@ � Z d d dA dB � Z! � S)Cr 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 Fr �loopN�keepalive_timeout�K r T�logger�access_log_class�
access_log�access_log_format�debug�
max_line_sizei� � max_headersi � �max_field_size�lingering_timeg $@c sm t � j d | � | d k r% | n t j � | _ | | _ | j | _ | j | _ | | _
d | _ d | _ | | _
t |
� | _ t � | _ d | _ d | _ d | _ | j j | j � � | _ d | _ d | _ t | | d |
d | d | d t �| _ d | _ d | _ | | _ | | _ | | _! | rN| | | � | _" n d | _" d | _# d | _$ d S)Nr � Fr& r( r' Zpayload_exception)%�super�__init__�asyncioZget_event_loop�_loop�_managerZrequest_handler�_request_handlerZrequest_factory�_request_factory�_tcp_keepalive�_keepalive_time�_keepalive_handle�_keepalive_timeout�float�_lingering_timer � _messages�
_message_tail�_waiter�_error_handler� create_task�start�
_task_handler�_upgrade�_payload_parserr
r �_request_parser� transport�_reading_pausedr! r% r# r �_close� _force_close)�self�managerr r r r! r" r# r$ r% r& r' r( r) )� __class__r r r, M sD
! zRequestHandler.__init__c C s+ d j | j j | j d k r$ d n d � S)Nz<{} {}>Z connectedZ disconnected)�formatrH r rB )rF r r r �__repr__� s zRequestHandler.__repr__c C s | j S)N)r5 )rF r r r r � s z RequestHandler.keepalive_timeoutg .@c � s� d | _ | j d k r% | j j � | j r; | j j � t t j t j � �l t | d | j ��O | j
r� | j
j � r� | j
Id H| j r� | j j � r� | j Id HWd QRXWd QRX| j r� | j j � | j
d k r� | j
j � d | _
d S)z�Worker process is about to exit, we need cleanup everything and
stop accepting requests. It is especially important for keep-alive
connections.TNr )rE r4 �cancelr: r r- �CancelledError� TimeoutErrorr r. r; �doner> rB �close)rF �timeoutr r r �shutdown� s
zRequestHandler.shutdownc s] t � j | � | | _ | j r, t | � t | d � t | d � | j j | | � d S)NFT)r+ �connection_maderB r2 r r r r/ )rF rB )rH r r rR � s
z RequestHandler.connection_madec s� | j j | | � t � j | � d | _ d | _ d | _ d | _ d | _ d | _ | j d k ru | j j � | j
r� | j
j � | j d k r� | j j � d | _
| j d k r� | j j
� d | _ d S)NT)r/ �connection_lostr+ rE r1 r0 rA rB r4 rK r> r; r@ Zfeed_eof)rF �exc)rH r r rS � s$
z RequestHandler.connection_lostc C sG | j d k s t � | | _ | j rC | j j | j � d | _ d S)Nr* )r@ �AssertionErrorr9 � feed_data)rF �parserr r r �
set_parser� s
zRequestHandler.set_parserc C s d S)Nr )rF r r r � eof_received� s zRequestHandler.eof_receivedc
C s | j s | j r d S| j d k r�| j r�y | j j | � \ } } } Wn� t k
r� } zK | j j | j t
| | j | j � d | | j � � | _
| j � WYd d } ~ Xqt k
r } zE | j j | j t
| | j | j � d | � � | _
| j � WYd d } ~ XqX| r�x9 | D]1 \ } } | j d 7_ | j j | | f � q-W| j } | d k r�| j � s�| j d � | | _ | r| r| | _ n^ | j d k r�| j r�| r�| j | 7_ n. | r| j j | � \ } } | r| j � d S)Ni� i� r )rE rD r@ r? rA rV r r. r< �handle_parse_errorr rB �messager; rO � Exception�_request_countr8 �appendr: rN �
set_resultZ _upgradedr9 )
rF �dataZmessagesZupgraded�tailrT �msg�payload�waiter�eofr r r �
data_received� sD
z RequestHandler.data_receivedc C s, | | _ | j r( | j j � d | _ d S)zMSet keep-alive connection mode.
:param bool val: new state.
N)�
_keepaliver4 rK )rF �valr r r �
keep_alive s
zRequestHandler.keep_alivec C s# d | _ | j r | j j � d S)ziStop accepting new pipelinig messages and close
connection when handlers done processing messagesTN)rD r: rK )rF r r r rO s zRequestHandler.closec C s^ d | _ | j r | j j � | j d k rZ | rD | j j d � | j j � d | _ d S)zForce close connectionTNs
)rE r: rK rB �writerO )rF �send_last_heartbeatr r r � force_close" s
zRequestHandler.force_closec C s) | j d k r% | j j | | | � d S)N)r �log)rF �request�response�timer r r �
log_access- s zRequestHandler.log_accessc O s | j r | j j | | � d S)N)r% r! )rF �args�kwr r r � log_debug1 s zRequestHandler.log_debugc O s | j j | | � d S)N)r! � exception)rF rr rs r r r �
log_exception5 s z RequestHandler.log_exceptionc C s{ | j s | j r d S| j | j } | j rY | j j � | k rY | j d d � d S| j j | j | j
� | _ d S)Nrk T) rE rg r3 r5 r: r. rp rl Z
call_later�KEEPALIVE_RESCHEDULE_DELAY�_process_keepaliver4 )rF �nextr r r rx 8 s z!RequestHandler._process_keepalivec
C sE | j sA y | j j � Wn t t t f k
r7 Yn Xd | _ d S)NT)rC rB �
pause_reading�AttributeError�NotImplementedError� RuntimeError)rF r r r rz I s z RequestHandler.pause_readingc
C sE | j rA y | j j � Wn t t t f k
r7 Yn Xd | _ d S)NF)rC rB �resume_readingr{ r| r} )rF r r r r~ Q s z RequestHandler.resume_readingc 8 � s� | j } | j } | j } | j } xW| j s}| j s� z< y | j � | _ | j Id HWn t j k
rs PYn XWd d | _ X| j j
� \ } } | j r� | j � } | j
d 7_
t | | j | � } | j | | | | | � } z�yHy | j | � Id H}
Wn� t k
r6} z | }
WYd d } ~ Xn� t j k
rX| j d � PYn� t j k
r�| j d � | j | d � }
YnY t k
r�} z | j | d | � }
WYd d } ~ Xn Xt |
t � r�t j d t � |
j | � Id H|
j � Id H|
j | _ | j r7| j | |
| j � | � | j � s5| j! } | j r| r| j d | � | j � } | | }
t" t j t j � �m xe | j � r�| |
k r�t# |
| | � } t$ | d | �� | j% � Id HWd QRX| j � } q�WWd QRX| j � r5| j r5| j d
� | j& � Wn� t j k
r[| j d � PYn� t' k
r�} z+ | j( r�| j) d d
| �| j* � WYd d } ~ XnB t k
r�} z" | j) d d
| �| j* � WYd d } ~ Xn XWd | j d k r
| j d � no | j sy| j rx| j+ rx| d k ry| j j � } | | _, | j- d k ry| j. | | | j/ � | _- n PXq' W| j s�d | _ | j d k r�| j0 d k r�| j j& � d S)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 disconnectionzRequest handler timed out.i� i� zlreturning HTTPException object is deprecated (#2415) and will be removed, please raise the exception insteadz'Start lingering close timer for %s sec.r zUncompleted request.z'Ignored premature client disconnection zUnhandled runtime exception�exc_infozUnhandled exceptionz'Ignored premature client disconnection.)1r. r> r/ r5 rE r8 Z
create_futurer: r- rL �popleftr# rp Zrequests_countr rB r1 r0 r rt rM � handle_errorr\ �
isinstance�warnings�warn�DeprecationWarning�prepare� write_eofri rg rq Zis_eofr7 r �minr ZreadanyrO r} r% rv rl rD r3 r4 Zcall_atrx r; )rF r �handlerrG r r[ rc Znow�writerrn �resprT r) Zend_trP r r r r= Y s�
(
! zRequestHandler.starti� c C s� | j d d | �| d k r� d } | j r{ t t � �; t j � } t | � } | d 7} | | 7} | d 7} Wd QRXq� | d 7} d | d
} n | } t d | d | d
d � } | j � | j j
d k s� | j d k r� | j � | S)z�Handle errors.
Returns HTTP response with specific status code. Logs additional
information. It always closes current connection.zError handling requestr i� z"<h1>500 Internal Server Error</h1>z <br><h2>Traceback:</h2>
<pre>z</pre>Nz Server got itself in troublezA<html><head><title>500 Internal Server Error</title></head><body>z</body></html>�status�textZ content_typez text/htmlr ) rv r% r r\ � traceback�
format_exc� html_escaper rl r� Z output_sizerB )rF rn r� rT r[ rb �tbr� r r r r� � s$
!
zRequestHandler.handle_errorc � s� t t t | | d | j � } | j | | | | � } | j | � Id H| j � Id H| j d k rs | j j � d | _ d S)N)
r �ERRORr r. r� r� r� rB rO r; )rF r� r� rT r[ rn r� r r r rZ � s
z!RequestHandler.handle_parse_error)"r r r r r] rg rw r
r Z AccessLoggerr Z
LOG_FORMATr, rJ �propertyr rQ rR rS rX rY rf ri rO rl rq rt rv rx rz r~ r= r� rZ r r )rH r r " sH & / 1
| )r r )-r- Zasyncio.streamsZ http.serverr r� r� � collectionsr �
contextlibr Zhtmlr r� Zyarl� r r r r
r rm r r
Zstreamsr Z tcp_helpersr r r Zweb_exceptionsr Z web_requestr Z web_responser �__all__ZRawRequestMessageZ
HttpVersion10ZURLr� r\ r ZFlowControlMixinZProtocolr r r r r �<module> s. !