U ucT@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_cvselfr1/tmp/pip-unpacked-wheel-yalm6hg1/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$Zrunningrrrset_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)r6rtimerrr8r2rr)rr,cancelrr4)rZcancel_pendingtimeoutrZ expirationrr0rrrshutdownvs&      zThreadedTaskDispatcher.shutdownN)Tr:)__name__ __module__ __qualname____doc__r'r(rrrr&r1r6r9r>rrrrr+s rc@s|eZdZdZdZdZdZdZdZdZ dZ dZ dZ e Z ddZddZed d Zd d Zd dZddZddZddZdS)TaskFz200 OKrNcCs.||_||_g|_|j}|dkr$d}||_dS)N)1.01.1rD)channelrequestresponse_headersversion)rrFrGrIrrrrsz Task.__init__cCsHz|||Wn&tk rBd|_|jjjr>YnXdS)NT)r#executefinishOSErrorclose_on_finishrFadjZlog_socket_errorsrrrrr-s  z Task.servicecCs&|jdp"|jdp"|jd S)N1Z204Z304)status startswithrrrrhas_bodys    z Task.has_bodycsj}jjdd}gd}d}d}djD]n\}}ddd|dD}|dkrljr4|}nq4|dkrx|}|dkr|}|d kr| ||fq4|dkrԈj dk rԈjrt j } d|ffd d }|d kr|d kr|s|n  dn|nP|dkrb|dkr4||sjjrR dd_ j sj|ntdjjjj} |s| r d| fn d| pdf|s dtjf_djdj} ddtjdddD} | g| } dd| } | dS)NZ CONNECTION-cSsg|] }|qSr) capitalize).0xrrr sz.Task.build_response_header..zContent-LengthDateServer Connectioncsdkrdd_dSN)r[closeT)r7rMrZconnection_close_headerrHrrrrMs z3Task.build_response_header..close_on_finishrDr )r[z Keep-AliverEr])zTransfer-EncodingchunkedTzneither HTTP/1.0 or HTTP/1.1ZViaZwaitresszHTTP/ cSsg|] }d|qS)z%s: %sr)rVZhvrrrrXscSs|dS)Nrr)rWrrrz,Task.build_response_header..)keyz%s z latin-1)rIrGheadersgetlowerrHjoinsplitrRr7content_lengthstrchunked_responserMAssertionErrorrFserverrNidentr start_timerPsortedencode)rrIr Zcontent_length_headerZ date_headerZ server_headerZ headernameZ headervalrMro first_lineZ next_lineslinesresrr^rbuild_response_headersx         zTask.build_response_headercCs:g}|jD]$\}}|dkr q |||fq ||_dS)Ncontent-length)rHrgr7)rrH header_name header_valuerrrremove_content_length_headers  z!Task.remove_content_length_headercCst|_dSr)r;rprrrrr##sz Task.startcCs&|js|d|jr"|jddS)Nrbs0 ) wrote_headerwriterlrF write_soonrrrrrK&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 writtenTrds 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 RuntimeErrorrFr{rvr}rRrjrlhexr2upperrrcontent_bytes_writtenlogged_write_excessrr8logged_write_no_bodyrP)rdatarFZrhZtowriteclrrrr|-sD  " z Task.write)r?r@rArMrPr{rprjrrrrrlrrr-propertyrRrvrzr#rKr|rrrrrCs(   b rCc@seZdZdZdZddZdS) ErrorTaskz(An error task produces an error responseTcCsR|jj}|\}}}||_|j||jdd|_t||_ | |dSr\) rGerrorZ to_responserPrHextendr7rMr2rjr|)rerPrebodyrrrrJ[s   zErrorTask.executeN)r?r@rArBrrJrrrrrVsrc@s$eZdZdZdZddZddZdS)WSGITaskz8A WSGI task produces a response from a WSGI application.Nc sP}d fdd }jj||}d}z|jtkrj}| |}|r||krj|dk rd |_ dj |d}WdSd}|D]P}|dkrt |}jdkrd} t|drt |} | dkr|_|r |qj}|dk r,j|kr,d_jjd kr,jd j|fW5|rJt|drJ|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 namerwzS%s is a "hop-by-hop" header; it cannot be used by a WSGI application (see PEP 3333))rrmr{rH __class__rk ValueErrorrPrgintrj hop_by_hoprr|)rPreexc_infokvklrrrstart_responseqs\           z(WSGITask.execute..start_responseTr]rbF__len__rHEADzVapplication returned too few bytes (%s) for specified Content-Length (%s) via app_iter)N)get_environmentrFrnZ applicationhasattrr]rrrjpreparerzr|r}r2rrMrGcommandrr8) renvironrZapp_iterZcan_close_app_iterrsizeZfirst_chunk_lenchunkZ app_iter_lenrrrrJnsRA         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/rSrrzHTTP/%s)rrTF)Z REMOTE_ADDRZ REMOTE_HOSTZ REMOTE_PORTREQUEST_METHODZ SERVER_PORTZ SERVER_NAMESERVER_SOFTWAREZSERVER_PROTOCOLZ SCRIPT_NAMEZ PATH_INFOZ REQUEST_URI QUERY_STRINGzwsgi.url_schemez wsgi.versionz wsgi.errorszwsgi.multithreadzwsgi.multiprocessz wsgi.run_oncez wsgi.inputzwsgi.file_wrapperzwsgi.input_terminatedZHTTP_zwaitress.client_disconnected) rrGpathrFrnrN url_prefixrQlstripr2addrrkrrZeffective_portZ server_namerorI request_uriqueryZ url_schemesysstderrZget_body_streamrdictreitemsstriprename_headersrfZcheck_client_disconnected) rrrGrrFrnrZurl_prefix_with_trailing_slashrcvalueZmykeyrrrrs`       zWSGITask.get_environment)r?r@rArBrrJrrrrrris}r) collectionsrsocketrrr;buffersrZ utilitiesrrrr frozensetrrrCrrrrrrs"  cI