o Q_@sdZddlZddlZddlZddlZddlmZmZmZddl m Z m Z ddl m Z ddl mZddl mZddlmZmZdd lmZdd lmZmZmZmZmZmZmZGd d d eZGd ddeZ GdddeZ!Gdddej"Z#Gdddej$Z%GdddeZ&dS)zFClient and server implementations of HTTP/1.x. .. versionadded:: 4.0 N)Futurefuture_add_done_callback"future_set_result_unless_cancelled) native_strutf8)gen)httputil)iostream)gen_logapp_log)GzipDecompressor)castOptionalType AwaitableCallableUnionTuplec@seZdZdddZdS)_QuietExceptionreturnNcCdSNselfrr9/usr/lib/python3/dist-packages/tornado/http1connection.py__init__+z_QuietException.__init__rN)__name__ __module__ __qualname__rrrrrr*src@sNeZdZdZdejddfddZdddZd d d ee d e j ddfd dZ dS)_ExceptionLoggingContextzUsed with the ``with`` statement when calling delegate methods to log any exceptions with the given logger. Any exceptions caught are converted to _QuietException loggerrNcCs ||_dSr)r#)rr#rrrr5 z!_ExceptionLoggingContext.__init__cCrrrrrrr __enter__8rz"_ExceptionLoggingContext.__enter__typzOptional[Type[BaseException]]valuetbcCs2|dur|dus J|jjd|||fdtdS)NUncaught exceptionexc_info)r#errorr)rr&r'r(rrr__exit__;s  z!_ExceptionLoggingContext.__exit__r) rr r!__doc__loggingLoggerrr%r BaseExceptiontypes TracebackTyper-rrrrr"/s r"c@s^eZdZdZ       ddedeedeedeedeed eed ed dfd d ZdS)HTTP1ConnectionParameterszDParameters for `.HTTP1Connection` and `.HTTP1ServerConnection`. FN no_keep_alive chunk_sizemax_header_sizeheader_timeout max_body_size body_timeout decompressrcCs6||_|pd|_|p d|_||_||_||_||_dS)a; :arg bool no_keep_alive: If true, always close the connection after one request. :arg int chunk_size: how much data to read into memory at once :arg int max_header_size: maximum amount of data for HTTP headers :arg float header_timeout: how long to wait for all headers (seconds) :arg int max_body_size: maximum amount of data for body :arg float body_timeout: how long to wait while reading body (seconds) :arg bool decompress: if true, decode incoming ``Content-Encoding: gzip`` iN)r5r6r7r8r9r:r;)rr5r6r7r8r9r:r;rrrrKs   z"HTTP1ConnectionParameters.__init__)FNNNNNF) rr r!r.boolrintfloatrrrrrr4Gs4 r4c @seZdZdZ  dAdejdedeedee ddf dd Z d e j de efd d Zd e j defd dZdBddZdeegdfddfddZdBddZdBddZdejfddZdeddfddZdeddfddZ dCd ee je jfd!e jd"eedd#fd$d%Zd"edefd&d'Z d"edd#fd(d)Z!dBd*d+Z"dDd-d.Z#d e jd!e jdefd/d0Z$dEd2d3Z%d4ede&e'e jffd5d6Z(d7ed!e jd e j dee dfd8d9Z)d:ed e j ddfd;d<Z*d e j ddfd=d>Z+d e j ddfd?d@Z,dS)FHTTP1ConnectionzImplements the HTTP/1.x protocol. This class can be on its own for clients, or via `HTTP1ServerConnection` for servers. Nstream is_clientparamscontextrcCs||_||_|dur t}||_||_|j|_|jjp|jj|_|jj |_ d|_ d|_ t |_d|_|d|_d|_d|_d|_d|_d|_dS)a :arg stream: an `.IOStream` :arg bool is_client: client or server :arg params: a `.HTTP1ConnectionParameters` instance or ``None`` :arg context: an opaque application-defined object that can be accessed as ``connection.context``. NF)rAr@r4rBrCr5r9max_buffer_size_max_body_sizer: _body_timeout_write_finished_read_finishedr_finish_future_disconnect_on_finish_clear_callbacks_request_start_line_response_start_line_request_headers_chunking_output_expected_content_remaining_pending_write)rr@rArBrCrrrrps(  zHTTP1Connection.__init__delegatecCs |jjr t||jj}||S)azRead a single HTTP response. Typical client-mode usage is to write a request using `write_headers`, `write`, and `finish`, and then call ``read_response``. :arg delegate: a `.HTTPMessageDelegate` Returns a `.Future` that resolves to a bool after the full response has been read. The result is true if the stream is still open. )rBr;_GzipMessageDelegater6 _read_message)rrRrrr read_responses  zHTTP1Connection.read_responsecsd}zz-|jjd|jjd}|jjdur|IdH}nHztj|jj|jj|t j dIdH}Wn1tj yd| YWW|r\t t |Wdn1sWwYd}|dSw||\}}|jr}t|}||_|}d|_nt|} | |_||_| }|| | |_d}t t|||} | dur| IdHWdn1swY|jdurd}WW|rt t |Wdn1swYd}|dSd} |jr/t|tjsJ|jdur|jjdkrd} |j} | dkrd} d| krd kr.nn,d |vsd |vr&t d | |!|IdHn|"d dkrA|j#sA|j$d| s|%|jrM|jnd||} | dur|j&durc| IdHnRztj|jj|j&| t j dIdHWn<tj yt'(d|j)|j YWW|rt t |Wdn 1swYd}|dSwd|_*|j#r|jrd}t t |+Wdn 1swY|j,-s|jdur|j.s|j/|j0|j,IdH|jr|jr| |jdur2WW|r*t t |Wdn 1s%wYd}|dSWnQtj y}zCt'(d|j)||jsR|j$dIdH| WYd}~W|rxt t |Wdn 1sswYd}|dSd}~wwW|rt t |Wdn 1swYd}|dS|rt t |Wdn 1swYd}|w)NFs ? ?  max_bytes)quiet_exceptionsTHEAD0dContent-LengthTransfer-Encodingz!Response code %d cannot have bodyExpectz 100-continuesHTTP/1.1 100 (Continue) rzTimeout reading body from %sz"Malformed HTTP message from %s: %ssHTTP/1.1 400 Bad Request )1r@read_until_regexrBr7r8r with_timeoutio_looptimer StreamClosedError TimeoutErrorcloser"r on_connection_closerK_parse_headersrArparse_response_start_linerMrJparse_request_start_linerLrN_can_keep_aliveheaders_received isinstanceResponseStartLinemethodcodeHTTPInputErrorrTgetrGwrite _read_bodyrFr inforCrHfinishrIdoneclosedset_close_callback_on_connection_close)rrRneed_delegate_close header_future header_datastart_line_strheadersresp_start_line start_linereq_start_lineheader_recv_future skip_bodyrp body_futureerrrrTs(  b         H                               zHTTP1Connection._read_messagecCs0d|_d|_d|_|jdur|jddSdS)zClears the callback attributes. This allows the request handler to be garbage collected more quickly in CPython by breaking up reference cycles. N)_write_callback _write_future_close_callbackr@ryrrrrrK.s  z HTTP1Connection._clear_callbackscallbackcC ||_dS)a$Sets a callback that will be run when the connection is closed. Note that this callback is slightly different from `.HTTPMessageDelegate.on_connection_close`: The `.HTTPMessageDelegate` method is called when the connection is closed while receiving a message. This callback is used when there is not an active delegate (for example, on the server side this callback is used if the client closes the connection after sending its request but before receiving all the response. N)rrrrrrry:s z"HTTP1Connection.set_close_callbackcCs>|jdur|j}d|_||jst|jd|dSr)rrIrwrrKrrrrrzHs    z$HTTP1Connection._on_connection_closecCs:|jdur |j||jst|jddSdSr)r@rfrKrIrwrrrrrrfTs   zHTTP1Connection.closecCs.||j}d|_|jst|jd|S)a>Take control of the underlying stream. Returns the underlying `.IOStream` object and stops all further HTTP processing. May only be called during `.HTTPMessageDelegate.headers_received`. Intended for implementing protocols like websockets that tunnel over an HTTP handshake. N)rKr@rIrwr)rr@rrrdetach[s   zHTTP1Connection.detachtimeoutcCr)ztSets the body timeout for a single request. Overrides the value from `.HTTP1ConnectionParameters`. N)rF)rrrrrset_body_timeoutj z HTTP1Connection.set_body_timeoutr9cCr)zwSets the body size limit for a single request. Overrides the value from `.HTTP1ConnectionParameters`. N)rE)rr9rrrset_max_body_sizeqrz!HTTP1Connection.set_max_body_sizerrchunk Future[None]c Cshg}|jr4t|tjs J||_|td|d|df|jdvo1d|vo1d|vp1|ddk|_nnt|tj ss z0HTTP1Connection.write_headers..css|]}|dVqdS)latin1N)encode)rlinerrrrs zNewline in header:  s )&rArmrRequestStartLinerLappendrrorOrnrNrMversionrprJrrlowerr rPr=get_allextend ValueErrorreprr@rxrr set_exceptionr rd exceptionjoin _format_chunkrsrQr_on_write_complete) rrrrlines header_linesrfuturedatarrr write_headersxs             zHTTP1Connection.write_headerscCsd|jdur|jt|8_|jdkr|jtd|jr0|r0tdt|d|dS|S)Nrz,Tried to write more data than Content-Lengthz%xr)rPlenr@rfrHTTPOutputErrorrOr)rrrrrrs    zHTTP1Connection._format_chunkcCsjd}|jrt}|_|jt|j|St}|_|j| ||_ t |j |j |S)zImplements `.HTTPConnection.write`. For backwards compatibility it is allowed but deprecated to skip `write_headers` and instead call `write()` with a pre-encoded header block. N) r@rxrrrr rdrrsrrQrr)rrrrrrrss    zHTTP1Connection.writecCs|jdur|jdkr|js|jtd|j|jr2|js2|jd|_|j |j d|_ |j s;d|_ |jd|jdurM|ddSt|j|jdS)z$Implements `.HTTPConnection.finish`.Nrz0Tried to write %d bytes less than Content-Lengths0 T)rPr@rxrfrrrOrsrQadd_done_callbackrrGrHrJ set_nodelay_finish_requestrrrrrrvs*      zHTTP1Connection.finishrcCsp|}|durt|tjs||jdur$|j}d|_|jj||j dur6|j }d|_ t |ddSdSr) rrmr rdresultrr@rb add_callbackrr)rrexcrrrrrs  z"HTTP1Connection._on_write_completecCsp|jjrdS|d}|dur|}|jdkr|dkSd|vs2|dddks2t|d dd vr6|d kSdS) NFrrrfr]r^rrro)rYGETr)rBr5rrrrgetattr)rrrconnection_headerrrrrk$s  zHTTP1Connection._can_keep_aliveOptional[Future[None]]cCsJ||js|jr|dS|jd|js#t|jddSdS)NF) rKrArJrfr@rrIrwr)rrrrrr8s   zHTTP1Connection._finish_requestrcCsLt|dd}|d}|d|d}tj||d}||fS)Nrz   )rdecodelstripfindrstripr HTTPHeadersparse)rrdata_streolrrrrrrhCs  zHTTP1Connection._parse_headersrpcs,d|vr\d|vr tdd|dvr5td|dtfddDr/td|dd |d<zt|d}WntyMtd |dwtt||jkr[td nd}|d krsd|vsj|d vrqtd|d }|dur}| ||S| dd dkr| |S|j r||SdS)Nr]r^z7Response with both Transfer-Encoding and Content-Length,z,\s*c3s|] }|dkVqdS)rNr)ripiecesrrrbsz-HTTP1Connection._read_body..z$Multiple unequal Content-Lengths: %rrz*Only integer Content-Length is allowed: %szContent-Length too longr)Nrz*Response with code %d should not have bodyrr)rrqresplitanyr=rr rE_read_fixed_bodyrrr_read_chunked_bodyrA_read_body_until_close)rrprrRcontent_lengthrrrrtOsP       zHTTP1Connection._read_bodyrcs|dkrI|jjt|jj|ddIdH}|t|8}|jr!|jrCtt | |}|dur4|IdHWdn1s>wY|dksdSdS)NrTpartial) r@ read_bytesminrBr6rrGrAr"r data_received)rrrRbodyretrrrrs     z HTTP1Connection._read_fixed_bodyc s d} |jjdddIdH}t|d}|dkr.|jdIdH}|dkr,tddS||7}||jkrrr=rrrrnrbytesrrrsrvrrkrrstrrhrtrrrrrrrr?is  5 z    ]        : r?c@szeZdZdZdejdeddfddZdeej ej fd ej de e dfd d Zd eddfd dZdddZdddZdS)rSzIWraps an `HTTPMessageDelegate` to decode ``Content-Encoding: gzip``. rRr6rNcCs||_||_d|_dSr) _delegate _chunk_size _decompressor)rrRr6rrrrs z_GzipMessageDelegate.__init__rrcCs:|ddkrt|_|d|d|d=|j||S)NzContent-EncodinggzipzX-Consumed-Content-Encoding)rrr raddrrl)rrrrrrrls z%_GzipMessageDelegate.headers_receivedrcs|jr4|}|r2|j||j}|r!|j|}|dur!|IdH|jj}|r.|s.td|sdSdS|j|}|durE|IdHdSdS)Nz8encountered unconsumed gzip data without making progress)rr;rrrunconsumed_tailrrq)rrcompressed_data decompressedrrrrrs*   z"_GzipMessageDelegate.data_receivedcCs*|jdur|j}|rtd|jS)Nz:decompressor.flush returned data; possible truncated input)rflushrrrv)rtailrrrrvs   z_GzipMessageDelegate.finishcCs |jSr)rrgrrrrrgr$z(_GzipMessageDelegate.on_connection_closer)rr r!r.rrr=rrrrnrrrrlrrrvrgrrrrrSs   rSc @sleZdZdZ  ddejdeedeeddfddZ dd d Z d e j ddfd d Z d e j ddfddZdS)HTTP1ServerConnectionzAn HTTP/1.x server.Nr@rBrCrcCs*||_|dur t}||_||_d|_dS)z :arg stream: an `.IOStream` :arg params: a `.HTTP1ConnectionParameters` or None :arg context: an opaque application-defined object that is accessible as ``connection.context`` N)r@r4rBrC_serving_future)rr@rBrCrrrrs   zHTTP1ServerConnection.__init__csB|j|jdus Jz |jIdHWdSty YdSw)zmCloses the connection. Returns a `.Future` that resolves after the serving loop has exited. N)r@rfr Exceptionrrrrrfs  zHTTP1ServerConnection.closerRcCs>t|tjsJt||}||_|jj |dddS)znStarts serving requests on this connection. :arg delegate: a `.HTTPServerConnectionDelegate` cSs|Sr)r)frrr-sz5HTTP1ServerConnection.start_serving..N) rmrHTTPServerConnectionDelegaterconvert_yielded_server_request_looprr@rb add_future)rrRfutrrr start_serving$sz#HTTP1ServerConnection.start_servingc szr t|jd|j|j}|||}z ||IdH}WnCtjtjt j fy3YW| |dSt yF| YW| |dSty`tjddd| YW| |dSw|skW| |dSt dIdHq| |w)NTFr)r*r)r?r@rBrC start_requestrUr rdUnsatisfiableReadErrorasyncioCancelledErroron_closerrfrr r,sleep)rrRconnrequest_delegaterrrrr/s<     z*HTTP1ServerConnection._server_request_looprr)rr r!r.r rrr4rrrfrrrrrrrrrs(  r)'r.r r/rr2tornado.concurrentrrrtornado.escaperrtornadorrr tornado.logr r tornado.utilr typingr rrrrrrrrrr"r4HTTPConnectionr?rrSrrrrrs.    $"\?