o
    JAf:                     @   s   	 d gZ ddlZddlZddlZedZedZG dd deZdd Z	d	d
 Z
dd  ZG dd dZG dd dZG dd dZG dd dZG dd dZdd Zdd Zdd Zdd Zdd Zd d! Zd"d# Zd$d% ZdS )&	validator    Nz^[a-zA-Z][a-zA-Z0-9\-_]*$z[\000-\037]c                   @   s   e Zd ZdS )WSGIWarningN)__name__
__module____qualname__ r   r   '/usr/lib/python3.10/wsgiref/validate.pyr   y   s    r   c                 G   s   | st | d S N)AssertionError)condargsr   r   r   assert_~   s   r   c                 C   s$   t | tu r| S td|t| )Nz!{0} must be of type str (got {1}))typestrr
   formatrepr)valuetitler   r   r   check_string_type   s
   r   c                    s   	  fdd}|S )Nc                     s   t t| dkd t | d | \} t| g  fdd}t|d |d< t|d |d< ||}t |d uo=|dkd	 t| t|S )
N   zTwo arguments requiredNo keyword arguments allowedc                     s   t t| dkpt| dkd| f  t | d | d }| d }t| dkr+| d }nd }t| t| t|| t| d  t |  S )Nr      zInvalid number of arguments: %sr   r      )r   lencheck_statuscheck_headerscheck_content_typecheck_exc_infoappendWriteWrapper)r   kwstatusheadersexc_infoZstart_responseZstart_response_startedr   r   start_response_wrapper   s   


z;validator.<locals>.lint_app.<locals>.start_response_wrapper
wsgi.inputwsgi.errorsFz>The application must return an iterator, if only an empty list)r   r   check_environInputWrapperErrorWrappercheck_iteratorIteratorWrapper)r   r    environr%   iteratorapplicationr$   r   lint_app   s   

zvalidator.<locals>.lint_appr   )r0   r1   r   r/   r   r      s   
)c                   @   s<   e Zd Zdd Zdd Zdd Zdd Zd	d
 Zdd ZdS )r)   c                 C   
   || _ d S r	   )input)self
wsgi_inputr   r   r   __init__      
zInputWrapper.__init__c                 G   s0   t t|dk | jj| }t t|tu  |S Nr   )r   r   r3   readr   bytesr4   r   vr   r   r   r9         zInputWrapper.readc                 G   s0   t t|dk | jj| }t t|tu  |S r8   )r   r   r3   readliner   r:   r;   r   r   r   r>      r=   zInputWrapper.readlinec                 G   sJ   t t|dk | jj| }t t|tu  |D ]
}t t|tu  q|S r8   )r   r   r3   	readlinesr   listr:   )r4   r   linesliner   r   r   r?      s   zInputWrapper.readlinesc                 c   s    	 |   }|s
d S |V  qr	   )r>   )r4   rB   r   r   r   __iter__   s   zInputWrapper.__iter__c                 C      t dd d S )Nr   z input.close() must not be calledr   r4   r   r   r   close      zInputWrapper.closeN)	r   r   r   r6   r9   r>   r?   rC   rG   r   r   r   r   r)      s    r)   c                   @   4   e Zd Zdd Zdd Zdd Zdd Zd	d
 ZdS )r*   c                 C   r2   r	   )errors)r4   wsgi_errorsr   r   r   r6      r7   zErrorWrapper.__init__c                 C   s    t t|tu  | j| d S r	   )r   r   r   rJ   writer4   sr   r   r   rL      s   zErrorWrapper.writec                 C   s   | j   d S r	   )rJ   flushrF   r   r   r   rO      rH   zErrorWrapper.flushc                 C   s   |D ]}|  | qd S r	   )rL   )r4   seqrB   r   r   r   
writelines   s   zErrorWrapper.writelinesc                 C   rD   )Nr   z!errors.close() must not be calledrE   rF   r   r   r   rG      rH   zErrorWrapper.closeN)r   r   r   r6   rL   rO   rQ   rG   r   r   r   r   r*      s    r*   c                   @      e Zd Zdd Zdd ZdS )r   c                 C   r2   r	   )writer)r4   Zwsgi_writerr   r   r   r6      r7   zWriteWrapper.__init__c                 C   s   t t|tu  | | d S r	   )r   r   r:   rS   rM   r   r   r   __call__   s   zWriteWrapper.__call__N)r   r   r   r6   rT   r   r   r   r   r          r   c                   @   rR   )PartialIteratorWrapperc                 C   r2   r	   r.   )r4   wsgi_iteratorr   r   r   r6      r7   zPartialIteratorWrapper.__init__c                 C   s   t | jd S r	   )r,   r.   rF   r   r   r   rC     s   zPartialIteratorWrapper.__iter__N)r   r   r   r6   rC   r   r   r   r   rV      rU   rV   c                   @   rI   )r,   c                 C   s    || _ t|| _d| _|| _d S )NF)original_iteratoriterr.   closedcheck_start_response)r4   rX   r\   r   r   r   r6   	  s   

zIteratorWrapper.__init__c                 C   s   | S r	   r   rF   r   r   r   rC     s   zIteratorWrapper.__iter__c                 C   sT   t | j d t| j}t|turt dd|f  | jd ur(t | jd d | _|S )NzIterator read after closedFz$Iterator yielded non-bytestring (%r)zjThe application returns and we started iterating over its body, but start_response has not yet been called)r   r[   nextr.   r   r:   r\   )r4   r<   r   r   r   __next__  s   

zIteratorWrapper.__next__c                 C   s$   d| _ t| jdr| j  d S d S )NTrG   )r[   hasattrrY   rG   rF   r   r   r   rG     s   zIteratorWrapper.closec                 C   s"   | j s	tjd t| j d d S )Nz/Iterator garbage collected without being closed)r[   sysstderrrL   r   rF   r   r   r   __del__#  s   zIteratorWrapper.__del__N)r   r   r   r6   rC   r^   rG   rb   r   r   r   r   r,     s    r,   c                 C   s  t t| tu dt| | f  dD ]}t || v d|f  qdD ]}t || vd||dd  f  q d| vr<tdt |  D ]}d	|v rGq@t t| | tu d
|t| | | | f  q@t t| d tu d| d f  t | d dv d| d   t	| d  t
| d  | d dvrtd| d  t t | d p| d dd| d   t | d p| d dd| d   | drt t| d dkd| d   | dst d| v d t | ddkd d S )Nz:Environment is not of the right type: %r (environment: %r))	REQUEST_METHODZSERVER_NAMEZSERVER_PORTwsgi.versionr&   r'   zwsgi.multithreadzwsgi.multiprocesszwsgi.run_oncez$Environment missing required key: %r)ZHTTP_CONTENT_TYPEZHTTP_CONTENT_LENGTHz8Environment should not have the key: %s (use %s instead)   ZQUERY_STRINGzQUERY_STRING is not in the WSGI environment; the cgi module will use sys.argv when this variable is missing, so application errors are more likely.z9Environmental variable %s is not a string: %r (value: %r)rd   z#wsgi.version should be a tuple (%r)zwsgi.url_scheme)httphttpszwsgi.url_scheme unknown: %rr&   r'   rc   )GETZHEADZPOSTZOPTIONSZPATCHPUTZDELETEZTRACEzUnknown REQUEST_METHOD: %rZSCRIPT_NAME/z$SCRIPT_NAME doesn't start with /: %rZ	PATH_INFOz"PATH_INFO doesn't start with /: %rZCONTENT_LENGTHr   zInvalid CONTENT_LENGTH: %rzgOne of SCRIPT_NAME or PATH_INFO are required (PATH_INFO should at least be '/' if SCRIPT_NAME is empty)zOSCRIPT_NAME cannot be '/'; it should instead be '', and PATH_INFO should be '/')r   r   dictwarningswarnr   keysr   tuplecheck_inputcheck_errorsget
startswithint)r-   keyr   r   r   r(   *  sx   







r(   c                 C   &   dD ]}t t| |d| |f  qd S )N)r9   r>   r?   rC   z-wsgi.input (%r) doesn't have the attribute %sr   r_   )r5   attrr   r   r   rq   k     
rq   c                 C   rw   )N)rO   rL   rQ   z.wsgi.errors (%r) doesn't have the attribute %srx   )rK   ry   r   r   r   rr   q  rz   rr   c                 C   sz   t | d} | d dd }tt|dkd|  t|}t|dkd|  t| dk s1| d d	kr;td
|  t d S d S )NStatusr   r   r   z)Status codes must be three characters: %rd   zStatus code is invalid: %r    zjThe status string (%r) should be a three-digit integer followed by a single space and a status explanation)r   splitr   r   ru   rm   rn   r   )r!   Zstatus_codeZ
status_intr   r   r   r   w  s   
r   c                 C   s  t t| tu d| t| f  | D ]n}t t|tu d|t|f  t t|dk |\}}t|d}t|d}t | dkd|  t d|voKd	|vd
|  t t|d|  t |	d of|	d d|  t
|rt dd|t
|df  qd S )Nz%Headers (%r) must be of type list: %rz1Individual headers (%r) must be of type tuple: %rr   Header namezHeader valuer!   zyThe Status header cannot be used; it conflicts with CGI script, and HTTP status is not given through headers (value: %r).
:z,Header names may not contain ':' or '\n': %rzBad header name: %r-_z#Names may not end in '-' or '_': %rr   z#Bad header value: %r (bad char: %r))r   r   r@   rp   r   r   lower	header_researchendswithbad_header_value_regroup)r"   itemnamer   r   r   r   r     sB   




r   c                 C   s   t | d} t| d dd }d}|D ]\}}t |d}| dkr0||vr) d S tdd|  q||vr>tdd|  d S d S )	Nr{   r   r   )   i0  r   zcontent-typezJContent-Type header found in a %s response, which must not return content.z,No Content-Type header found in headers (%s))r   ru   r   r   r   )r!   r"   codeZNO_MESSAGE_BODYr   r   r   r   r   r     s   

r   c                 C   s*   t | d u p
t| tu d| t| f  d S )Nz exc_info (%r) is not a tuple: %r)r   r   rp   )r#   r   r   r   r     s   r   c                 C   s   t t| ttf d d S )NzwYou should not return a string as your application iterator, instead return a single-item list containing a bytestring.)r   
isinstancer   r:   rW   r   r   r   r+     s   r+   )__all__rer`   rm   compiler   r   Warningr   r   r   r   r)   r*   r   rV   r,   r(   rq   rr   r   r   r   r   r+   r   r   r   r   <module>   s0   j

7#		#A