o
    ii;                     @  s  U d dl mZ dZd dlZd dlZd dlZd dlZd dlmZm	Z	 d dl
mZ d dlmZ d dlmZ d dlmZmZmZ d	d
lmZmZmZmZmZ d	dlmZmZ d	dlmZmZmZm Z m!Z!m"Z" ej#dkrqd dlm$Z$m%Z% nd dl&m$Z$m%Z% ej#dkrd dlm'Z' nd dl&m'Z' edZ(e$dZ)e*e*e+e+f df Z,de-d< e*e,df Z.de-d< G dd deZ/eddG dd deZ0eddG dd de!e0 Z1G dd  d e Z2dS )!    )annotations)TLSAttributeTLSConnectableTLSListener	TLSStreamN)CallableMapping)	dataclass)wraps)
SSLContext)Any	TypeAliasTypeVar   )BrokenResourceErrorEndOfStreamaclose_forcefullyget_cancelled_exc_class	to_thread)TypedAttributeSettyped_attribute)AnyByteStreamAnyByteStreamConnectable
ByteStreamByteStreamConnectableListener	TaskGroup)      )TypeVarTupleUnpack)r      )overrideT_RetvalPosArgsT.r   _PCTRTT_PCTRTTTc                   @  s   e Zd ZU dZe Zded< e Zded< e Zded< e Z	ded	< e Z
d
ed< e Zded< e Zded< e Zded< e Zded< e Zded< dS )r   z5Contains Transport Layer Security related attributes.
str | Nonealpn_protocolbyteschannel_binding_tls_uniqueztuple[str, str, int]cipherz*None | dict[str, str | _PCTRTTT | _PCTRTT]peer_certificatezbytes | Nonepeer_certificate_binaryboolserver_sidez!list[tuple[str, str, int]] | Noneshared_ciphersssl.SSLObject
ssl_objectstandard_compatiblestrtls_versionN)__name__
__module____qualname____doc__r   r(   __annotations__r*   r+   r,   r-   r/   r0   r2   r3   r5    r;   r;   Y/var/www/scraper.webshop2u.hu/html/venv/lib/python3.10/site-packages/anyio/streams/tls.pyr   5   s   
 r   F)eqc                   @  s   e Zd ZU dZded< ded< ded< ded	< ded
< edddddd2ddZd3ddZd4dd Zd5d"d#Z	d6d7d(d)Z
d8d+d,Zd5d-d.Zed9d0d1ZdS ):r   a  
    A stream wrapper that encrypts all sent data and decrypts received data.

    This class has no public initializer; use :meth:`wrap` instead.
    All extra attributes from :class:`~TLSAttribute` are supported.

    :var AnyByteStream transport_stream: the wrapped stream

    r   transport_streamr.   r3   r1   _ssl_objectzssl.MemoryBIO	_read_bio
_write_bioNT)r/   hostnamessl_contextr3   r/   bool | NonerB   r'   rC   ssl.SSLContext | Nonereturnc                  s   |du r| }|s'|rt jjnt jj}t |}tt dr'| jt j M  _t  }t  }t	|t j
u r@|j||||d}	nt|j||||dI dH }	| |||	||d}
|
|	jI dH  |
S )a  
        Wrap an existing stream with Transport Layer Security.

        This performs a TLS handshake with the peer.

        :param transport_stream: a bytes-transporting stream to wrap
        :param server_side: ``True`` if this is the server side of the connection,
            ``False`` if this is the client side (if omitted, will be set to ``False``
            if ``hostname`` has been provided, ``False`` otherwise). Used only to create
            a default context when an explicit context has not been provided.
        :param hostname: host name of the peer (if host name checking is desired)
        :param ssl_context: the SSLContext object to use (if not provided, a secure
            default will be created)
        :param standard_compatible: if ``False``, skip the closing handshake when
            closing the connection, and don't raise an exception if the peer does the
            same
        :raises ~ssl.SSLError: if the TLS handshake fails

        NOP_IGNORE_UNEXPECTED_EOF)r/   server_hostname)r>   r3   r?   r@   rA   )sslPurposeCLIENT_AUTHSERVER_AUTHcreate_default_contexthasattroptionsrG   	MemoryBIOtyper   wrap_bior   run_sync_call_sslobject_methoddo_handshake)clsr>   r/   rB   rC   r3   purposebio_inbio_outr2   wrapperr;   r;   r<   wrapc   s@   


	zTLSStream.wrapfunc&Callable[[Unpack[PosArgsT]], T_Retval]argsUnpack[PosArgsT]r#   c                   s  	 z|| }W n t jyY   z| jjr!| j| j I d H  | j I d H }W n& ty8   | j	
  Y n tyP } z| j	
  | j
  t|d }~ww | j	| Y no t jyn   | j| j I d H  Y nZ t jy } z| j	
  | j
  t|d }~w t jy } z"| j	
  | j
  t|t js|jrd|jv r| jrt|td  d }~ww | jjr| j| j I d H  |S q)NTUNEXPECTED_EOF_WHILE_READING)rI   SSLWantReadErrorrA   pendingr>   sendreadreceiver   r@   	write_eofOSErrorr   writeSSLWantWriteErrorSSLSyscallErrorSSLError
isinstanceSSLEOFErrorstrerrorr3   )selfr\   r^   resultdataexcr;   r;   r<   rT      sV   






z TLSStream._call_sslobject_methodtuple[AnyByteStream, bytes]c                   s:   |  | jjI dH  | j  | j  | j| j fS )z
        Does the TLS closing handshake.

        :return: a tuple of (wrapped byte stream, bytes left in the read buffer)

        N)rT   r?   unwrapr@   rf   rA   r>   rd   ro   r;   r;   r<   rt      s
   

zTLSStream.unwrapNonec                   sP   | j rz	|  I d H  W n ty   t| jI d H   w | j I d H  d S N)r3   rt   BaseExceptionr   r>   acloseru   r;   r;   r<   ry      s   zTLSStream.aclose   	max_bytesintr)   c                   s$   |  | jj|I d H }|st|S rw   )rT   r?   rd   r   )ro   r{   rq   r;   r;   r<   re      s
   zTLSStream.receiveitemc                   s   |  | jj|I d H  d S rw   )rT   r?   rh   )ro   r}   r;   r;   r<   rc      s   zTLSStream.sendc                   sb   |  tj}td|}|r-t|dt|dpd}}||fdk r-td| td)NzTLSv(\d+)(?:\.(\d+))?   r   r   )r~   r   z;send_eof() requires at least TLSv1.3; current session uses z7send_eof() has not yet been implemented for TLS streams)extrar   r5   rematchr|   groupNotImplementedError)ro   r5   r   majorminorr;   r;   r<   send_eof   s   "zTLSStream.send_eofMapping[Any, Callable[[], Any]]c                   s   i  j jtj jjtj jjtj jjtj	 fddtj
 fddtj fddtj fddtj fddtj fddtj jji
S )Nc                         j dS )NFr?   getpeercertr;   ru   r;   r<   <lambda>  s    z,TLSStream.extra_attributes.<locals>.<lambda>c                     r   )NTr   r;   ru   r;   r<   r     s    c                     s    j jS rw   )r?   r/   r;   ru   r;   r<   r     s    c                     s    j jr	 j  S d S rw   )r?   r/   r0   r;   ru   r;   r<   r     s   c                         j S rw   r3   r;   ru   r;   r<   r         c                     r   rw   )r?   r;   ru   r;   r<   r     r   )r>   extra_attributesr   r(   r?   selected_alpn_protocolr*   get_channel_bindingr+   r,   r-   r/   r0   r3   r2   r5   versionru   r;   ru   r<   r     s   


zTLSStream.extra_attributes)r>   r   r/   rD   rB   r'   rC   rE   r3   r.   rF   r   )r\   r]   r^   r_   rF   r#   )rF   rs   rF   rv   )rz   )r{   r|   rF   r)   )r}   r)   rF   rv   rF   r   )r6   r7   r8   r9   r:   classmethodr[   rT   rt   ry   re   rc   r   propertyr   r;   r;   r;   r<   r   Q   s*   
 

F
.



r   c                   @  sn   e Zd ZU dZded< ded< dZded< d	Zd
ed< ed ddZ	d!d"ddZ	d#ddZ
ed$ddZdS )%r   a  
    A convenience listener that wraps another listener and auto-negotiates a TLS session
    on every accepted connection.

    If the TLS handshake times out or raises an exception,
    :meth:`handle_handshake_error` is called to do whatever post-mortem processing is
    deemed necessary.

    Supports only the :attr:`~TLSAttribute.standard_compatible` extra attribute.

    :param Listener listener: the listener to wrap
    :param ssl_context: the SSL context object
    :param standard_compatible: a flag passed through to :meth:`TLSStream.wrap`
    :param handshake_timeout: time limit for the TLS handshake
        (passed to :func:`~anyio.fail_after`)
    zListener[Any]listenerzssl.SSLContextrC   Tr.   r3      floathandshake_timeoutrr   rx   streamr   rF   rv   c                   sL   t |I dH  t| t sttjd| d t| tr#t| t r$ dS )a  
        Handle an exception raised during the TLS handshake.

        This method does 3 things:

        #. Forcefully closes the original stream
        #. Logs the exception (unless it was a cancellation exception) using the
           ``anyio.streams.tls`` logger
        #. Reraises the exception if it was a base exception or a cancellation exception

        :param exc: the exception
        :param stream: the original stream

        NzError during TLS handshake)exc_info)r   rl   r   logging	getLoggerr6   	exception	Exception)rr   r   r;   r;   r<   handle_handshake_error8  s   
z"TLSListener.handle_handshake_errorNhandlerCallable[[TLSStream], Any]
task_groupTaskGroup | Nonec                   s2   t  d fdd}j||I d H  d S )Nr   r   rF   rv   c              
     s   ddl m} z$|j tj| jjdI d H }W d    n1 s%w   Y  W n tyG } z|| I d H  W Y d }~d S d }~ww  |I d H  d S )Nr   )
fail_after)rC   r3   )	 r   r   r   r[   rC   r3   rx   r   )r   r   wrapped_streamrr   r   ro   r;   r<   handler_wrapper]  s     z*TLSListener.serve.<locals>.handler_wrapper)r   r   rF   rv   )r
   r   serve)ro   r   r   r   r;   r   r<   r   X  s   zTLSListener.servec                   s   | j  I d H  d S rw   )r   ry   ru   r;   r;   r<   ry   o  s   zTLSListener.acloser   c                   s   t j fddiS )Nc                     r   rw   r   r;   ru   r;   r<   r   u  r   z.TLSListener.extra_attributes.<locals>.<lambda>)r   r3   ru   r;   ru   r<   r   r  s   zTLSListener.extra_attributes)rr   rx   r   r   rF   rv   rw   )r   r   r   r   rF   rv   r   r   )r6   r7   r8   r9   r:   r3   r   staticmethodr   r   ry   r   r   r;   r;   r;   r<   r      s   
 "
r   c                   @  s2   e Zd ZdZdddddddZedddZdS )r   a  
    Wraps another connectable and does TLS negotiation after a successful connection.

    :param connectable: the connectable to wrap
    :param hostname: host name of the server (if host name checking is desired)
    :param ssl_context: the SSLContext object to use (if not provided, a secure default
        will be created)
    :param standard_compatible: if ``False``, skip the closing handshake when closing
        the connection, and don't raise an exception if the server does the same
    NTrB   rC   r3   connectabler   rB   r'   rC   rE   r3   r.   rF   rv   c                C  sN   || _ |pttjj| _t| jtjstdt	| jj
 || _|| _d S )Nz7ssl_context must be an instance of ssl.SSLContext, not )r   rI   rM   rJ   rL   rC   rl   r   	TypeErrorrQ   r6   rB   r3   )ro   r   rB   rC   r3   r;   r;   r<   __init__  s   

zTLSConnectable.__init__r   c                   sR   | j  I d H }ztj|| j| j| jdI d H W S  ty(   t|I d H   w )Nr   )	r   connectr   r[   rB   rC   r3   rx   r   )ro   r   r;   r;   r<   r     s   zTLSConnectable.connect)
r   r   rB   r'   rC   rE   r3   r.   rF   rv   )rF   r   )r6   r7   r8   r9   r   r"   r   r;   r;   r;   r<   r   y  s    r   )3
__future__r   __all__r   r   rI   syscollections.abcr   r   dataclassesr	   	functoolsr
   r   typingr   r   r   r   r   r   r   r   r   _core._typedattrr   r   abcr   r   r   r   r   r   version_infor   r    typing_extensionsr"   r#   r$   tupler4   r%   r:   r&   r   r   r   r   r;   r;   r;   r<   <module>   s>     
	
 OX