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 )helpershttp) CeilTimeout)HttpProcessingErrorHttpRequestParser StreamWriter)
access_logger
server_logger)
EMPTY_PAYLOAD)tcp_cork
tcp_keepalive tcp_nodelay)
HTTPException) BaseRequest)ResponseRequestHandlerRequestPayloadErrorZUNKNOWN/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 loopNkeepalive_timeoutK r Tloggeraccess_log_class
access_logaccess_log_formatdebug
max_line_sizei max_headersi max_field_sizelingering_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_timeoutfloat_lingering_timer _messages
_message_tail_waiter_error_handler create_taskstart
_task_handler_upgrade_payload_parserr
r _request_parser transport_reading_pausedr! r% r# r _close _force_close)selfmanagerr 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)zWorker 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 ry | 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| rx9 | 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 dataZmessagesZupgradedtailrT msgpayloadwaitereofr 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 requestresponsetimer r r
log_access- s zRequestHandler.log_accessc O s | j r | j j | | d S)N)r% r! )rF argskwr 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_laterKEEPALIVE_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_readingAttributeErrorNotImplementedError 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 | | | | | } zyHy | 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 rt 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 rt# |
| | } t$ | d | | j% Id HWd QRX| j } qWWd 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 sd | _ | 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 exceptionexc_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\
isinstancewarningswarnDeprecationWarningprepare 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 Znowwriterrn 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)zHandle 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>statustextZ 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. !