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d€d0d1„Zdd�d2d3„Z	d‚dddd4œdƒd<d=„Zd„d>d?„Zd„d@dA„Zej	Bd…d†dFdG„ƒZejd‡dJdG„ƒZdˆd‰dMdG„ZdŠdNdO„Zd‹dQdR„ZdŠdSdT„ZdŒdVdW„ZdŠdXdY„Zd�d\d]„ZdŽd^d_„Zd„d`da„Zd‚d�dcdd„Zejd�dgdh„ƒZejd‘dkdh„ƒZejd’dndh„ƒZ		d“d”drdh„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   úY/var/www/html/premium_crap/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   þ€Pþz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   Néÿÿÿÿr   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    
