o
    'h"                     @  s   d dl mZ d dlZd dlZd dlZd dlZddlmZ ejr-d dl	m
Z
 ddlmZmZ ejeef ZedZd	ZG d
d dZdS )    )annotationsN   )ProxySchemeUnsupported)Self   )_TYPE_PEER_CERT_RET_TYPE_PEER_CERT_RET_DICT_ReturnValuei @  c                   @  s  e Zd ZdZedsddZ			dtduddZdvddZdwddZdxddZ	dydzd!d"Z
d{d|d&d'Z		#d}d~d,d-Zddd0d1Zddd2d3Z	ddddd4dd<d=Zdd>d?Zdd@dAZej	BdddFdGZejddJdGZdddMdGZddNdOZddQdRZddSdTZddVdWZddXdYZdd\d]Zdd^d_Zdd`daZdddcddZejddgdhZejddkdhZejddndhZ		dddrdhZdS )SSLTransportaL  
    The SSLTransport wraps an existing socket and establishes an SSL connection.

    Contrary to Python's implementation of SSLSocket, it allows you to chain
    multiple TLS connections together. It's particularly useful if you need to
    implement TLS within TLS.

    The class supports most of the socket API operations.
    ssl_contextssl.SSLContextreturnNonec                 C  s   t | ds	tddS )z
        Raises a ProxySchemeUnsupported if the provided ssl_context can't be used
        for TLS in TLS.

        The only requirement is that the ssl_context provides the 'wrap_bio'
        methods.
        wrap_biozXTLS in TLS requires SSLContext.wrap_bio() which isn't available on non-native SSLContextN)hasattrr   )r    r   U/var/www/html/olx_land/venv/lib/python3.10/site-packages/urllib3/util/ssltransport.py$_validate_ssl_context_for_tls_in_tls!   s
   

z1SSLTransport._validate_ssl_context_for_tls_in_tlsNTsocketsocket.socketserver_hostname
str | Nonesuppress_ragged_eofsboolc                 C  sH   t  | _t  | _|| _|| _|j| j| j|d| _| | jj	 dS )zV
        Create an SSLTransport around socket using the provided ssl_context.
        )r   N)
ssl	MemoryBIOincomingoutgoingr   r   r   sslobj_ssl_io_loopdo_handshake)selfr   r   r   r   r   r   r   __init__1   s   



zSSLTransport.__init__r   c                 C  s   | S Nr   r!   r   r   r   	__enter__H      zSSLTransport.__enter___
typing.Anyc                 G  s   |    d S r#   )close)r!   r'   r   r   r   __exit__K      zSSLTransport.__exit__intc                 C  
   | j  S r#   )r   filenor$   r   r   r   r.   N      
zSSLTransport.fileno   lenbuffertyping.Any | Noneint | bytesc                 C  s   |  ||S r#   )_wrap_ssl_read)r!   r1   r2   r   r   r   readQ   r+   zSSLTransport.readr   buflenflagsc                 C  s   |dkrt d| |S )Nr   z+non-zero flags not allowed in calls to recv)
ValueErrorr5   )r!   r7   r8   r   r   r   recvT   s   
zSSLTransport.recv_WriteBuffernbytes
int | NoneNone | int | bytesc                 C  s,   |dkrt d|d u rt|}| ||S )Nr   z0non-zero flags not allowed in calls to recv_into)r9   r1   r6   )r!   r2   r<   r8   r   r   r   	recv_intoY   s
   zSSLTransport.recv_intodatabytesc              	   C  s   |dkrt dd}t|?}|d!}t|}||k r.| ||d  }||7 }||k sW d    n1 s8w   Y  W d    d S W d    d S 1 sPw   Y  d S )Nr   z.non-zero flags not allowed in calls to sendallB)r9   
memoryviewcastr1   send)r!   r@   r8   countview	byte_viewamountvr   r   r   sendalle   s   PzSSLTransport.sendallc                 C  s    |dkrt d| | jj|S )Nr   z+non-zero flags not allowed in calls to send)r9   r   r   write)r!   r@   r8   r   r   r   rE   o   s   zSSLTransport.send)encodingerrorsnewlinemodestr	bufferingrM   rN   rO   1typing.BinaryIO | typing.TextIO | socket.SocketIOc                C  s  t |h dkstd|dd|v }d|v p| }|s!|s!J d|v }d}	|r-|	d7 }	|r3|	d7 }	t| |	}
| j jd7  _|d	u rGd
}|dk rNtj}|dkrZ|sXtd|
S |rf|rft|
|
|}n|rot|
|}n
|ssJ t	|
|}|r}|S t
||||}||_|S )a  
        Python's httpclient uses makefile and buffered io when reading HTTP
        messages and we need to support it.

        This is unfortunately a copy and paste of socket.py makefile with small
        changes to point to the socket directly.
        >   brwzinvalid mode z (only r, w, b allowed)rV   rU   rT    r   Nr   z!unbuffered streams must be binary)setr9   r   SocketIO_io_refsioDEFAULT_BUFFER_SIZEBufferedRWPairBufferedReaderBufferedWriterTextIOWrapperrP   )r!   rP   rR   rM   rN   rO   writingreadingbinaryrawmoderawr2   textr   r   r   makefilet   s@   zSSLTransport.makefilec                 C  s   |  | jj d S r#   )r   r   unwrapr$   r   r   r   ri      s   zSSLTransport.unwrapc                 C     | j   d S r#   )r   r)   r$   r   r   r   r)         zSSLTransport.close.binary_formtyping.Literal[False]_TYPE_PEER_CERT_RET_DICT | Nonec                 C     d S r#   r   r!   rl   r   r   r   getpeercert   s   zSSLTransport.getpeercerttyping.Literal[True]bytes | Nonec                 C  ro   r#   r   rp   r   r   r   rq      r&   Fr   c                 C  s   | j |S r#   )r   rq   rp   r   r   r   rq      r+   c                 C  r-   r#   )r   versionr$   r   r   r   rt      r/   zSSLTransport.versiontuple[str, str, int] | Nonec                 C  r-   r#   )r   cipherr$   r   r   r   rv      r/   zSSLTransport.cipherc                 C  r-   r#   )r   selected_alpn_protocolr$   r   r   r   rw      r/   z#SSLTransport.selected_alpn_protocol!list[tuple[str, str, int]] | Nonec                 C  r-   r#   )r   shared_ciphersr$   r   r   r   ry      r/   zSSLTransport.shared_ciphersc                 C  r-   r#   )r   compressionr$   r   r   r   rz      r/   zSSLTransport.compressionvaluefloat | Nonec                 C  s   | j | d S r#   )r   
settimeout)r!   r{   r   r   r   r}      s   zSSLTransport.settimeoutc                 C  r-   r#   )r   
gettimeoutr$   r   r   r   r~      r/   zSSLTransport.gettimeoutc                 C  rj   r#   )r   _decref_socketiosr$   r   r   r   r      rk   zSSLTransport._decref_socketiosbytearray | Nonec              
   C  sR   z
|  | jj||W S  tjy( } z|jtjkr#| jr#W Y d }~dS  d }~ww Nr   )r   r   r6   r   SSLErrorerrnoSSL_ERROR_EOFr   )r!   r1   r2   er   r   r   r5      s   zSSLTransport._wrap_ssl_readfunctyping.Callable[[], None]c                 C  ro   r#   r   )r!   r   r   r   r   r      r&   zSSLTransport._ssl_io_looptyping.Callable[[bytes], int]arg1c                 C  ro   r#   r   )r!   r   r   r   r   r   r      r&   /typing.Callable[[int, bytearray | None], bytes]arg2c                 C  ro   r#   r   )r!   r   r   r   r   r   r   r      s   "typing.Callable[..., _ReturnValue]None | bytes | intr	   c           	   
   C  s   d}d}|rsd}z|du r|du r| }n|du r||}n|||}W n! t jyE } z|jt jt jfvr8||j}W Y d}~nd}~ww | j }| j| |du rXd}n|t jkrq| j	t
}|rl| j| n| j  |stt|S )z>Performs an I/O loop between incoming/outgoing and the socket.TNF)r   r   r   SSL_ERROR_WANT_READSSL_ERROR_WANT_WRITEr   r6   r   rK   r:   SSL_BLOCKSIZEr   rL   	write_eoftypingrD   r	   )	r!   r   r   r   should_loopretr   r   bufr   r   r   r      s8   




)r   r   r   r   )NT)
r   r   r   r   r   r   r   r   r   r   )r   r   )r'   r(   r   r   )r   r,   )r0   N)r1   r,   r2   r3   r   r4   )r0   r   )r7   r,   r8   r,   r   r4   r   )r2   r;   r<   r=   r8   r,   r   r>   )r   )r@   rA   r8   r,   r   r   )r@   rA   r8   r,   r   r,   r#   )rP   rQ   rR   r=   rM   r   rN   r   rO   r   r   rS   )r   r   ).)rl   rm   r   rn   )rl   rr   r   rs   )F)rl   r   r   r   )r   r   )r   ru   )r   rx   )r{   r|   r   r   )r   r|   )r1   r,   r2   r   r   r4   )r   r   r   r   )r   r   r   rA   r   r,   )r   r   r   r,   r   r   r   rA   )NN)r   r   r   r   r   r   r   r	   )__name__
__module____qualname____doc__staticmethodr   r"   r%   r*   r.   r6   r:   r?   rK   rE   rh   ri   r)   r   overloadrq   rt   rv   rw   ry   rz   r}   r~   r   r5   r   r   r   r   r   r
      sb    





4










r
   )
__future__r   r\   r   r   r   
exceptionsr   TYPE_CHECKINGtyping_extensionsr   ssl_r   r   Union	bytearrayrC   r;   TypeVarr	   r   r
   r   r   r   r   <module>   s    
