a
    њhM:                     @  sv  d dl mZ 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mZ ddlmZ g d	Zd
jej Zejdde de Zejdde de ZeejddZeejddZeejddZdddddZedZedZ ej!G dd dZ"ej!G dd dZ#d d!d"d#d$Z$d d%d"d&d'Z%d(d)d d d d!d*d+d,Z&dS )-    )annotationsN)	Generator)Callable   )Headers)SecurityError)version)SERVER
USER_AGENTRequestResponsez{}.{}ZWEBSOCKETS_USER_AGENTzPython/z websockets/ZWEBSOCKETS_SERVERZWEBSOCKETS_MAX_NUM_HEADERSZ128ZWEBSOCKETS_MAX_LINE_LENGTHZ8192ZWEBSOCKETS_MAX_BODY_SIZEZ	1_048_576bytesstr)valuereturnc                 C  s   | j ddS )zG
    Decode a bytestring for interpolating into an error message.

    backslashreplaceerrors)decode)r    r   L/wd/webapps/venvs/v2025_4um/lib/python3.9/site-packages/websockets/http11.pyd2   s    r   s   [-!#$%&\'*+.^_`|~0-9a-zA-Z]+s   [\x09\x20-\x7e\x80-\xff]*c                   @  sb   e Zd ZU dZded< ded< dZded< edd	d
dZedddddZ	dd	ddZ
dS )r   z
    WebSocket handshake request.

    Attributes:
        path: Request path, including optional query.
        headers: Request headers.
    r   pathr   headersNException | None
_exceptionr   c                 C  s   t dt | jS )NzIRequest.exception is deprecated; use ServerProtocol.handshake_exc insteadwarningswarnDeprecationWarningr   selfr   r   r   	exception^   s
    zRequest.exception-Callable[[int], Generator[None, None, bytes]]zGenerator[None, None, Request]	read_liner   c           	   
   c  s   zt |E dH }W n. ty@ } ztd|W Y d}~n
d}~0 0 z|dd\}}}W n& ty~   tdt| dY n0 |dkrtdt| |dkrtd	t| |d
d}t|E dH }d|v rtdd|v rtd| ||S )a  
        Parse a WebSocket handshake request.

        This is a generator-based coroutine.

        The request path isn't URL-decoded or validated in any way.

        The request path and headers are expected to contain only ASCII
        characters. Other characters are represented with surrogate escapes.

        :meth:`parse` doesn't attempt to read the request body because
        WebSocket handshake requests don't have one. If the request contains a
        body, it may be read from the data stream after :meth:`parse` returns.

        Args:
            read_line: Generator-based coroutine that reads a LF-terminated
                line or raises an exception if there isn't enough data

        Raises:
            EOFError: If the connection is closed without a full HTTP request.
            SecurityError: If the request exceeds a security limit.
            ValueError: If the request isn't well formatted.

        Nz1connection closed while reading HTTP request line       zinvalid HTTP request line:    HTTP/1.1)unsupported protocol; expected HTTP/1.1: s   GETz+unsupported HTTP method; expected GET; got asciisurrogateescapeTransfer-Encodingz!transfer codings aren't supportedContent-Lengthzunsupported request body)
parse_lineEOFErrorsplit
ValueErrorr   r   parse_headersNotImplementedError)	clsr&   Zrequest_lineexcmethodraw_pathprotocolr   r   r   r   r   parsef   s*    # zRequest.parser   c                 C  s$   d| j  d }|| j 7 }|S )z;
        Serialize a WebSocket handshake request.

        zGET z HTTP/1.1
)r   encoder   	serialize)r"   requestr   r   r   r<      s    zRequest.serialize)__name__
__module____qualname____doc____annotations__r   propertyr#   classmethodr:   r<   r   r   r   r   r   N   s   
?r   c                   @  s~   e Zd ZU dZded< ded< ded< dZd	ed
< dZded< eddddZe	dddddddddZ
d	dddZdS )r   z
    WebSocket handshake response.

    Attributes:
        status_code: Response code.
        reason_phrase: Response reason.
        headers: Response headers.
        body: Response body.

    intstatus_coder   reason_phraser   r       r   bodyNr   r   r   c                 C  s   t dt | jS )NzJResponse.exception is deprecated; use ClientProtocol.handshake_exc insteadr   r!   r   r   r   r#      s
    zResponse.exceptionTr$   boolzGenerator[None, None, Response])r&   
read_exactread_to_eofinclude_bodyr   c              
   c  s^  zt |E dH }W n. ty@ } ztd|W Y d}~n
d}~0 0 z|dd\}}}	W n& ty~   tdt| dY n0 |dkrtdt| zt|}
W n& ty   tdt| dY n0 d	|
  krd
k sn tdt| t|	stdt|	 |	dd}t	|E dH }|rLt
|
||||E dH }nd}| |
|||S )a  
        Parse a WebSocket handshake response.

        This is a generator-based coroutine.

        The reason phrase and headers are expected to contain only ASCII
        characters. Other characters are represented with surrogate escapes.

        Args:
            read_line: Generator-based coroutine that reads a LF-terminated
                line or raises an exception if there isn't enough data.
            read_exact: Generator-based coroutine that reads the requested
                bytes or raises an exception if there isn't enough data.
            read_to_eof: Generator-based coroutine that reads until the end
                of the stream.

        Raises:
            EOFError: If the connection is closed without a full HTTP response.
            SecurityError: If the response exceeds a security limit.
            LookupError: If the response isn't well formatted.
            ValueError: If the response isn't well formatted.

        Nz0connection closed while reading HTTP status liner'   r(   zinvalid HTTP status line: r)   r*   z+invalid status code; expected integer; got d   iX  u-   invalid status code; expected 100–599; got zinvalid HTTP reason phrase: r+   r,   rH   )r/   r0   r1   r2   r   rE   	_value_re	fullmatchr   r3   	read_body)r5   r&   rK   rL   rM   Zstatus_liner6   r9   Zraw_status_codeZ
raw_reasonrF   reasonr   rI   r   r   r   r:      sD    ! 
zResponse.parsec                 C  s6   d| j  d| j d }|| j 7 }|| j7 }|S )z<
        Serialize a WebSocket handshake response.

        z	HTTP/1.1  z
)rF   rG   r;   r   r<   rI   )r"   responser   r   r   r<     s    
zResponse.serialize)T)r>   r?   r@   rA   rB   rI   r   rC   r#   rD   r:   r<   r   r   r   r   r      s   
 Fr   r$   zGenerator[None, None, bytes]r%   c                 c  sL   z| t E dH }W n ty,   tdY n0 |ds@td|dd S )ao  
    Parse a single line.

    CRLF is stripped from the return value.

    Args:
        read_line: Generator-based coroutine that reads a LF-terminated line
            or raises an exception if there isn't enough data.

    Raises:
        EOFError: If the connection is closed without a CRLF.
        SecurityError: If the response exceeds a security limit.

    Nzline too long   
zline without CRLF)MAX_LINE_LENGTHRuntimeErrorr   endswithr0   )r&   liner   r   r   r/   #  s    
r/   zGenerator[None, None, Headers]c           	      c  s  t  }ttd D ]}zt| E dH }W n. tyV } ztd|W Y d}~n
d}~0 0 |dkrf qz|dd\}}W n& ty   tdt| dY n0 t	|stdt| |
d}t	|std	t| |d
}|d
d}|||< qtd|S )a  
    Parse HTTP headers.

    Non-ASCII characters are represented with surrogate escapes.

    Args:
        read_line: Generator-based coroutine that reads a LF-terminated line
            or raises an exception if there isn't enough data.

    Raises:
        EOFError: If the connection is closed without complete headers.
        SecurityError: If the request exceeds a security limit.
        ValueError: If the request isn't well formatted.

    r   Nz,connection closed while reading HTTP headersrH      :zinvalid HTTP header line: zinvalid HTTP header name: s    	zinvalid HTTP header value: r+   r,   ztoo many HTTP headers)r   rangeMAX_NUM_HEADERSr/   r0   r1   r2   r   	_token_rerP   striprO   r   r   )	r&   r   _rZ   r6   raw_name	raw_valuenamer   r   r   r   r3   >  s,     




r3   rE   r   )rF   r   r&   rK   rL   r   c                 c  s  d|   krdk s&n | dks&| dkr*dS | d }d ur |dkrVtd| d	d}t|E d H }|d
dd }t|dkr|jdd}	td|	 dt|d}
|
dkrqt||
 tkrtd|
 dt| d|||
E d H 7 }|dE d H dkrZt	dqZt
|E d H  |S | d }d urt|dkrRtd| dt|}|tkrttd| d||E d H S z|tE d H W S  ty   tdt dY n0 d S )NrN         i0  rH   r-   chunkedztransfer coding z isn't supported   ;r   r      r   r   zchunk too large: 0xz bytes   zchunk too large: z bytes after r(   rU   zchunk without CRLFr.      zbody too large: zbody too large: over )getr4   r/   r1   lenr   r   rE   MAX_BODY_SIZEr2   r3   rX   )rF   r   r&   rK   rL   ZcodingrI   Zchunk_size_lineZraw_chunk_sizeZstr_chunk_size
chunk_sizeZraw_content_lengthcontent_lengthr   r   r   rQ   q  sD    &


rQ   )'
__future__r   dataclassesosresysr   collections.abcr   typingr   Zdatastructuresr   
exceptionsr   r   Zwebsockets_version__all__formatversion_infoPYTHON_VERSIONenvironrk   r
   r	   rE   r]   rW   rm   r   compiler^   rO   	dataclassr   r   r/   r3   rQ   r   r   r   r   <module>   sB   

cp3