a
    hh                     @   sh  d dl mZmZmZmZmZmZmZmZm	Z	m
Z
 ddlmZmZmZmZmZmZmZ ddlmZmZmZ ddlmZmZ ddlmZ ddlmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z& ddl'm(Z(m)Z)m*Z* dd	l+m,Z,m-Z- g d
Z.G dd de*e*dZ/G dd de*e*dZ0dZ1e
eef e2dddZ3e4e
eef ee5e
ed ee6 f f dddZ7G dd dZ8dS )    )
AnyCallablecastDictListOptionaloverloadTupleTypeUnion   )ConnectionClosedDataEndOfMessageEventInformationalResponseRequestResponse)get_comma_headerhas_expect_100_continueset_comma_header)READERSReadersType)ReceiveBuffer)
_SWITCH_CONNECT_SWITCH_UPGRADECLIENTConnectionStateDONEERRORMIGHT_SWITCH_PROTOCOL	SEND_BODYSERVERSWITCHED_PROTOCOL)LocalProtocolErrorRemoteProtocolErrorSentinel)WRITERSWritersType)
Connection	NEED_DATAPAUSEDc                   @   s   e Zd ZdS )r*   N__name__
__module____qualname__ r0   r0   K/var/www/html/assistant/venv/lib/python3.9/site-packages/h11/_connection.pyr*   3   s   r*   )	metaclassc                   @   s   e Zd ZdS )r+   Nr,   r0   r0   r0   r1   r+   7   s   r+   i @  eventreturnc                 C   s0   t | jd}d|v rdS t| dddk r,dS dS )N
   connection   closeFhttp_version   1.1T)r   headersgetattr)r4   
connectionr0   r0   r1   _keep_aliveT   s    r=   r0   )request_methodr4   r5   c                 C   s   t |ttfv sJ t |tu rf|jdv sT| dksT| dkrXd|j  krPdk rXn ndS |jdksfJ t|jd}|r|dgksJ d	S t|jd
}|rdt|d ffS t |tu rdS dS d S )N)   i0     HEAD   CONNECT   ,  )content-length)r      transfer-encoding   chunked)chunkedr0      content-lengthrD   r   )http/1.0r0   )typer   r   status_coder   r:   int)r>   r4   Ztransfer_encodingsZcontent_lengthsr0   r0   r1   _body_framing]   s0    
rM   c                   @   s4  e Zd ZdZefee eddddZe	e
ee ee f dddZe	ee dd	d
Ze	ee dddZe	edddZddddZee ddddZeeee  dddZee eddddZee ee eeef eedef  dddZd6e
ee ee f ee dddd Ze	eeef dd!d"Zedd#d$d%Z eeee! ee" f dd&d'Z#eeee! ee" f dd(d)Z$e%e&ddd*d+Z'e%ee(e)e*e+e,f edd,d+Z'e%eee dd-d+Z'eee dd.d+Z'eee-e  dd/d0Z.ddd1d2Z/e*e*d3d4d5Z0dS )7r)   aB  An object encapsulating the state of an HTTP connection.

    Args:
        our_role: If you're implementing a client, pass :data:`h11.CLIENT`. If
            you're implementing a server, pass :data:`h11.SERVER`.

        max_incomplete_event_size (int):
            The maximum number of bytes we're willing to buffer of an
            incomplete event. In practice this mostly sets a limit on the
            maximum size of the request/response line + headers. If this is
            exceeded, then :meth:`next_event` will raise
            :exc:`RemoteProtocolError`.

    N)our_rolemax_incomplete_event_sizer5   c                 C   s   || _ |ttfvr td||| _|  |tu r:t| _nt| _t | _| | jd t	| _
| | jd t| _t | _d| _d | _d | _d| _d S )Nzexpected CLIENT or SERVER, not F)_max_incomplete_event_sizer   r"   
ValueErrorrN   
their_roler   _cstate_get_io_objectr'   _writerr   _readerr   _receive_buffer_receive_buffer_closedtheir_http_version_request_method"client_is_waiting_for_100_continue)selfrN   rO   r0   r0   r1   __init__   s     zConnection.__init__)r5   c                 C   s   t | jjS )zA dictionary like::

           {CLIENT: <client state>, SERVER: <server state>}

        See :ref:`state-machine` for details.

        )dictrS   statesr\   r0   r0   r1   r_      s    	zConnection.statesc                 C   s   | j j| j S )zjThe current state of whichever role we are playing. See
        :ref:`state-machine` for details.
        )rS   r_   rN   r`   r0   r0   r1   	our_state   s    zConnection.our_statec                 C   s   | j j| j S )znThe current state of whichever role we are NOT playing. See
        :ref:`state-machine` for details.
        )rS   r_   rR   r`   r0   r0   r1   their_state   s    zConnection.their_statec                 C   s   | j tu o| jS N)rR   r   r[   r`   r0   r0   r1   !they_are_waiting_for_100_continue   s    z,Connection.they_are_waiting_for_100_continuec                 C   s4   t | jj}| j  d| _| jr&J | | dS )a  Attempt to reset our connection state for a new request/response
        cycle.

        If both client and server are in :data:`DONE` state, then resets them
        both to :data:`IDLE` state in preparation for a new request/response
        cycle on this same connection. Otherwise, raises a
        :exc:`LocalProtocolError`.

        See :ref:`keepalive-and-pipelining`.

        N)r^   rS   r_   start_next_cyclerZ   r[   _respond_to_state_changes)r\   
old_statesr0   r0   r1   re      s
    

zConnection.start_next_cycle)roler5   c                 C   s&   t | jj}| j| | | d S rc   )r^   rS   r_   Zprocess_errorrf   )r\   rh   rg   r0   r0   r1   _process_error   s    zConnection._process_errorr3   c                 C   sT   t |tu r|jdkrtS t |tu rPt| jjv rPd|j  krHdk rPn ntS d S )Ne   rB   rC   )rJ   r   rK   r   r   r   rS   Zpending_switch_proposalsr\   r4   r0   r0   r1   _server_switch_event  s    

zConnection._server_switch_event)rh   r4   r5   c                 C   s\  t | jj}|tu rNt|tu rN|jdkr6| jt t	|j
drN| jt d }|tu rd| |}| j|t|| t|tu r|j| _|| ju rt|tttfv rtttttf |}|j| _t|ttfv rtttttf |s| j  t|tu rt|rd| _t|ttfv r*d| _|tu rLt|ttfv rLd| _| || d S )NrA   s   upgradeTF)r^   rS   r_   r   rJ   r   methodZprocess_client_switch_proposalr   r   r:   r   r"   rl   Zprocess_eventrZ   rR   r   r   r   r   r8   rY   r=   Zprocess_keep_alive_disabledr   r[   r   r   rf   )r\   rh   r4   rg   Zserver_switch_eventr0   r0   r1   _process_event  s<    


zConnection._process_event.)rh   r4   io_dictr5   c                 C   sZ   | j j| }|tu rHttt| jtttt	f |\}}|t | | S |
||fS d S rc   )rS   r_   r!   rM   r   bytesrZ   r   r   r   get)r\   rh   r4   ro   stateframing_typeargsr0   r0   r1   rT   @  s    zConnection._get_io_object)rg   r4   r5   c                 C   sH   | j || j kr"| | j|t| _| j|| j krD| | j|t| _d S rc   )	ra   rN   rT   r'   rU   rb   rR   r   rV   )r\   rg   r4   r0   r0   r1   rf   V  s    z$Connection._respond_to_state_changesc                 C   s   t | j| jfS )ab  Data that has been received, but not yet processed, represented as
        a tuple with two elements, where the first is a byte-string containing
        the unprocessed data itself, and the second is a bool that is True if
        the receive connection was closed.

        See :ref:`switching-protocols` for discussion of why you'd want this.
        )rp   rW   rX   r`   r0   r0   r1   trailing_dataa  s    	zConnection.trailing_data)datar5   c                 C   s,   |r"| j rtd|  j|7  _nd| _ dS )a  Add data to our internal receive buffer.

        This does not actually do any processing on the data, just stores
        it. To trigger processing, you have to call :meth:`next_event`.

        Args:
            data (:term:`bytes-like object`):
                The new data that was just received.

                Special case: If *data* is an empty byte-string like ``b""``,
                then this indicates that the remote side has closed the
                connection (end of file). Normally this is convenient, because
                standard Python APIs like :meth:`file.read` or
                :meth:`socket.recv` use ``b""`` to indicate end-of-file, while
                other failures to read are indicated using other mechanisms
                like raising :exc:`TimeoutError`. When using such an API you
                can just blindly pass through whatever you get from ``read``
                to :meth:`receive_data`, and everything will work.

                But, if you have an API where reading an empty string is a
                valid non-EOF condition, then you need to be aware of this and
                make sure to check for such strings and avoid passing them to
                :meth:`receive_data`.

        Returns:
            Nothing, but after calling this you should call :meth:`next_event`
            to parse the newly received data.

        Raises:
            RuntimeError:
                Raised if you pass an empty *data*, indicating EOF, and then
                pass a non-empty *data*, indicating more data that somehow
                arrived after the EOF.

                (Calling ``receive_data(b"")`` multiple times is fine,
                and equivalent to calling it once.)

        z(received close, then received more data?TN)rX   RuntimeErrorrW   )r\   rv   r0   r0   r1   receive_datal  s
    'zConnection.receive_datac                 C   s   | j }|tu r| jrtS |tu s(|tu r,tS | jd us:J | | j}|d u rx| jsx| jrxt| jdrr| j	 }nt
 }|d u rt}|S )Nread_eof)rb   r   rW   r+   r    r#   rV   rX   hasattrry   r   r*   )r\   rr   r4   r0   r0   r1   _extract_next_receive_event  s    z&Connection._extract_next_receive_eventc              
   C   s   | j tu rtdz`|  }|ttfvr<| | jtt	| |tu rnt
| j| jkr`tddd| jrntd|W S  ty } z.| | j t|tr|  n W Y d}~n
d}~0 0 dS )aF  Parse the next event out of our receive buffer, update our internal
        state, and return it.

        This is a mutating operation -- think of it like calling :func:`next`
        on an iterator.

        Returns:
            : One of three things:

            1) An event object -- see :ref:`events`.

            2) The special constant :data:`NEED_DATA`, which indicates that
               you need to read more data from your socket and pass it to
               :meth:`receive_data` before this method will be able to return
               any more events.

            3) The special constant :data:`PAUSED`, which indicates that we
               are not in a state where we can process incoming data (usually
               because the peer has finished their part of the current
               request/response cycle, and you have not yet called
               :meth:`start_next_cycle`). See :ref:`flow-control` for details.

        Raises:
            RemoteProtocolError:
                The peer has misbehaved. You should close the connection
                (possibly after sending some kind of 4xx response).

        Once this method returns :class:`ConnectionClosed` once, then all
        subsequent calls will also return :class:`ConnectionClosed`.

        If this method raises any exception besides :exc:`RemoteProtocolError`
        then that's a bug -- if it happens please file a bug report!

        If this method raises any exception then it also sets
        :attr:`Connection.their_state` to :data:`ERROR` -- see
        :ref:`error-handling` for discussion.

        z+Can't receive data when peer state is ERRORzReceive buffer too longi  )Zerror_status_hintz#peer unexpectedly closed connectionN)rb   r   r%   r{   r*   r+   rn   rR   r   r   lenrW   rP   rX   BaseExceptionri   
isinstancer$   Z!_reraise_as_remote_protocol_error)r\   r4   excr0   r0   r1   
next_event  s&    (


zConnection.next_eventc                 C   s   d S rc   r0   rk   r0   r0   r1   send  s    zConnection.sendc                 C   s   d S rc   r0   rk   r0   r0   r1   r     s    c                 C   s   d S rc   r0   rk   r0   r0   r1   r     s    c                 C   s$   |  |}|du rdS d|S dS )a  Convert a high-level event into bytes that can be sent to the peer,
        while updating our internal state machine.

        Args:
            event: The :ref:`event <events>` to send.

        Returns:
            If ``type(event) is ConnectionClosed``, then returns
            ``None``. Otherwise, returns a :term:`bytes-like object`.

        Raises:
            LocalProtocolError:
                Sending this event at this time would violate our
                understanding of the HTTP/1.1 protocol.

        If this method raises any exception then it also sets
        :attr:`Connection.our_state` to :data:`ERROR` -- see
        :ref:`error-handling` for discussion.

        N    )send_with_data_passthroughjoin)r\   r4   	data_listr0   r0   r1   r     s    
c                 C   s   | j tu rtdzbt|tu r*| |}| j}| | j| t|t	u rPW dS |dus\J g }|||j
 |W S W n   | | j  Y n0 dS )ah  Identical to :meth:`send`, except that in situations where
        :meth:`send` returns a single :term:`bytes-like object`, this instead
        returns a list of them -- and when sending a :class:`Data` event, this
        list is guaranteed to contain the exact object you passed in as
        :attr:`Data.data`. See :ref:`sendfile` for discussion.

        z'Can't send data when our state is ERRORN)ra   r   r$   rJ   r   &_clean_up_response_headers_for_sendingrU   rn   rN   r   appendri   )r\   r4   writerr   r0   r0   r1   r      s     


z%Connection.send_with_data_passthroughc                 C   s   |  | j dS )zNotify the state machine that we failed to send the data it gave
        us.

        This causes :attr:`Connection.our_state` to immediately become
        :data:`ERROR` -- see :ref:`error-handling` for discussion.

        N)ri   rN   r`   r0   r0   r1   send_failedA  s    zConnection.send_failed)responser5   c                 C   s   t |tu sJ |j}d}tt| j}|dkr2d}t||\}}|dv rt|dg }| jd u sh| jdk rt|dg }| jdkrd}nt|dd	g}| j	j
r|rtt|d
}|d |d t|d
t|}t||j|j|jdS )NFr@   s   GET)rG   rI   rH   r9   rE   TrF   r6   s
   keep-aliver7   )r:   rK   r8   reason)rJ   r   r:   r   rp   rZ   rM   r   rY   rS   
keep_alivesetr   discardaddsortedrK   r8   r   )r\   r   r:   Z
need_closeZmethod_for_choosing_headersrs   _r<   r0   r0   r1   r   Z  s2    	


z1Connection._clean_up_response_headers_for_sending)N)1r-   r.   r/   __doc__!DEFAULT_MAX_INCOMPLETE_EVENT_SIZEr
   r&   rL   r]   propertyr   r_   ra   rb   boolrd   re   ri   r   r   rl   rn   r   r   r(   r   r   rT   rf   r	   rp   ru   rx   r*   r+   r{   r   r   r   r   r   r   r   r   r   r   r   r   r   r0   r0   r0   r1   r)      s`   ' 
2
 
0 A!r)   N)9typingr   r   r   r   r   r   r   r	   r
   r   Z_eventsr   r   r   r   r   r   r   _headersr   r   r   _readersr   r   Z_receivebufferr   _stater   r   r   r   r   r   r    r!   r"   r#   Z_utilr$   r%   r&   _writersr'   r(   __all__r*   r+   r   r   r=   rp   strrL   rM   r)   r0   r0   r0   r1   <module>   s    0$	0
: