U hU@sddlmZddlZddlZddlZddlZddlmZddlm Z m Z m Z dddZ e d ZGd d d ZGd d d ZGdddeZGdddeZdS))dequeN)ReadOnlyFileBasedBuffer)build_http_datelogger queue_loggerCONTENT_LENGTH CONTENT_TYPE)rr ) connection keep-alivezproxy-authenticatezproxy-authorizationteZtrailersztransfer-encodingupgradec@sReZdZdZdZdZeZeZddZddZ ddZ d d Z d d Z dddZ dS)ThreadedTaskDispatcherz6A Task Dispatcher that creates a thread for each task.rcCs:t|_t|_t|_t|j|_t|j|_ dSN) setthreadsrqueue threadingLocklock Conditionqueue_cvthread_exit_cvselfr?/var/www/html/venv/lib/python3.8/site-packages/waitress/task.py__init__3s  zThreadedTaskDispatcher.__init__cCs*tj|d||fd}d|_|dS)Nz waitress-)targetnameargsT)rThreaddaemonstart)rr thread_notrrrstart_new_thread:sz'ThreadedTaskDispatcher.start_new_threadc Cs|j|js@|jdkr@|jd8_|j|jd7_q|jdkr|jd8_|jd8_|j||j W5QRq|j }W5QRXz | Wqt k r|j d|YqXqdS)NrrzException when servicing %r)rr stop_count active_countrwaitrdiscardrnotifypopleftservice BaseExceptionr exception)rr$taskrrrhandler_threadAs       z%ThreadedTaskDispatcher.handler_threadc Cs|j|j}d}t||j}||krr||kr:|d}q(|||d7}||j||jd7_|d}q ||kr|j||7_|j W5QRXdS)Nrr) rrlenr'addr&r1r(r notify_all)rcountrr$runningrrrset_thread_countXs   z'ThreadedTaskDispatcher.set_thread_countc Csf|jV|j||jt|j}t|j|j|j}||krX|j d||W5QRXdS)NzTask queue depth is %d) rrappendrr+r2rr'r(rwarning)rr0Z queue_sizeZ idle_threadsrrradd_taskks   zThreadedTaskDispatcher.add_taskTc Cs|d|j}t|}|j|rVt|krH|jdt|qV|jdq$|r|j }t|dkr~|jdt||r| }| q~|j W5QRdSW5QRXdS)Nrz%d thread(s) still runningg?zCanceling %d pending task(s)TF)r7rtimerrr9r2rr)rr,cancelrr4)rZcancel_pendingtimeoutrZ expirationrr0rrrshutdownvs&      zThreadedTaskDispatcher.shutdownN)Tr;)__name__ __module__ __qualname____doc__r'r(rrrr&r1r7r:r?rrrrr+s rc@seZdZdZdZdZdZdZdZdZ dZ dZ dZ e Z ddZddZed d Zdd d d ZddZddZddZddZddZdS)TaskFz200 OKrNcCs.||_||_g|_|j}|dkr$d}||_dS)N)1.01.1rE)channelrequestresponse_headersversion)rrGrHrJrrrrsz Task.__init__cCsHz|||Wn&tk rBd|_|jjjr>YnXdS)NT)r#executefinishOSErrorclose_on_finishrGadjZlog_socket_errorsrrrrr-s  z Task.servicecCs&|jdp"|jdp"|jd S)N1Z204Z304)status startswithrrrrhas_bodys    z Task.has_body)returncCsL|jsBd}|jD]\}}|dkr|}q|dkrB|jdd|_dS)N Connection)rUcloseT) wrote_headerrI capitalizelowerr8rN)rZconnection_close_header headername headervalrrrset_close_on_finishs   zTask.set_close_on_finishcCs|j}|jjdd}g}d}d}d}|jD]^\}}ddd|dD}|dkrh|jr0|}nq0|dkrt|}|dkr|}| ||fq0||_|dkr|j dk r|jrt |j }|j d|f|d kr|d kr|s| q|j d n| nV|d krL|d kr| |sT|jr:|j dd|_ |jsT| ntd|jjjj} |s~| r|j d| fn|j d| pdf|s|j dt|jfd|jd|j} ddt|jdddD} | g| } dd| } | dS)NZ CONNECTION-cSsg|] }|qSr)rX).0xrrr sz.Task.build_response_header..zContent-LengthDateServerrEr )rUz Keep-AliverFrV)zTransfer-EncodingchunkedTzneither HTTP/1.0 or HTTP/1.1ZViaZwaitresszHTTP/ cSsg|] }d|qS)z%s: %sr)r_ZhvrrrrascSs|dS)Nrr)r`rrrz,Task.build_response_header..)keyz%s z latin-1)rJrHheadersgetrYrIjoinsplitrSr8content_lengthstrr\chunked_responserNAssertionErrorrGserverrOidentr start_timerQsortedencode)rrJr rIZcontent_length_headerZ date_headerZ server_headerrZr[rs first_lineZ next_lineslinesresrrrbuild_response_headersp         zTask.build_response_headercCs:g}|jD]$\}}|dkr q |||fq ||_dS)Ncontent-length)rIrYr8)rrI header_name header_valuerrrremove_content_length_headers  z!Task.remove_content_length_headercCst|_dSr)r<rtrrrrr#(sz Task.startcCs&|js|d|jr"|jddS)Nrgs0 )rWwriterprG write_soonrrrrrL+s z Task.finishcCs|jstd|j}|js2|}||d|_|r|jr|}|j}|jr|t t |dd dd}||d7}nP|dk r|d||j }|j t |7_ ||kr|js|jd|d|_|r||n8|r|j t |7_ |js|jd|jd|_dS)Nz1start_response was not called before body writtenTris z`application-written content exceeded the number of bytes specified by Content-Length header (%s)zfapplication-written content was ignored due to HTTP response that may not contain a message-body: (%s))complete RuntimeErrorrGrWrzrrSrnrphexr2upperrvcontent_bytes_writtenlogged_write_excessrr9logged_write_no_bodyrQ)rdatarGZrhZtowriteclrrrr2sD  " z Task.write)r@rArBrNrQrWrtrnrrrrrprrr-propertyrSr\rzr~r#rLrrrrrrDs*    [ rDc@seZdZdZdZddZdS) ErrorTaskz(An error task produces an error responseTcCsV|jjjj}|jj}||\}}}||_|j || t ||_ | |dSr)rGrrrOrsrHerrorZ to_responserQrIextendr\r2rnr)rrserQrjbodyrrrrK`s   zErrorTask.executeN)r@rArBrCrrKrrrrr[src@s$eZdZdZdZddZddZdS)WSGITaskz8A WSGI task produces a response from a WSGI application.Nc s\}d fdd }jj||}d}z|jtkrj}| |}|r||krj|dk rd |_ dj |d}WdSd}|D]P}|dkrt |}jdkrd} t|drt |} | dkr|_|r |qj}|dk r8j|kr8jjd kr8jjd kr8jd j|W5|rVt|drV|XdS) Ncs8jr|std|r8zjr(|dng_W5d}Xd_|jtk rTtd|d|ksdd|krltd|_|D]\}}|jtk rtd|d ||f|jtk rtd |d ||fd|ksd|krtd d|ksd|krtd |}|d krt |_ qv|t krvtd|qvj |j S)Nz?start_response called a second time without providing exc_info.rTzstatus %s is not a string  z5carriage return/line feed character present in statusz Header name z is not a string in z Header value z;carriage return/line feed character present in header valuez:carriage return/line feed character present in header namer{zS%s is a "hop-by-hop" header; it cannot be used by a WSGI application (see PEP 3333))rrqrWrI __class__ro ValueErrorrQrYintrn hop_by_hoprr)rQrjexc_infokvklrrrstart_responsess\           z(WSGITask.execute..start_responseTrVrgF__len__rHEADzVapplication returned too few bytes (%s) for specified Content-Length (%s) via app_iter)N)get_environmentrGrrZ applicationhasattrrVrrrnpreparer~rrr2rrHcommandr\rr9) renvironrZapp_iterZcan_close_app_iterrsizeZfirst_chunk_lenchunkZ app_iter_lenrrrrKpsRA        zWSGITask.executec CsL|j}|dk r|S|j}|j}|j}|j}|jj}|drJd|d}|r~||kr\d}n"|d}||r~|t |d}|j d|j dt |j d|j t |j|j|jjd|j|||j|j|jdtjdd d |tdd }t|jD]>\}} | } t|d} | dkr$d |} | |kr| || <q|jj|d <||_|S) zReturns a WSGI environment.N/r]rrzHTTP/%s)rrTF)Z REMOTE_ADDRZ REMOTE_HOSTZ REMOTE_PORTREQUEST_METHODZ SERVER_PORTZ SERVER_NAMESERVER_SOFTWAREZSERVER_PROTOCOLZ SCRIPT_NAMEZ PATH_INFOZ REQUEST_URIZ QUERY_STRINGzwsgi.url_schemez wsgi.versionz wsgi.errorszwsgi.multithreadzwsgi.multiprocessz wsgi.run_oncez wsgi.inputzwsgi.file_wrapperzwsgi.input_terminatedZHTTP_zwaitress.client_disconnected) rrHpathrGrrrO url_prefixrRlstripr2addrrorrZeffective_portZ server_namersrJ request_uriquery url_schemesysstderrZget_body_streamrdictrjitemsstriprename_headersrkZcheck_client_disconnected) rrrHrrGrrrZurl_prefix_with_trailing_slashrhvalueZmykeyrrrrs`       zWSGITask.get_environment)r@rArBrCrrKrrrrrrks~r) collectionsrsocketrrr<buffersrZ utilitiesrrrr frozensetrrrDrrrrrrs"  cN