Current File : /opt/alt/python37/lib/python3.7/site-packages/urllib3/util/__pycache__//ssltransport.cpython-37.pyc
B
��`� � @ sF d dl Z d dlZd dlZd dlmZ d dlmZ dZG dd� d�ZdS )� N)�ProxySchemeUnsupported)�sixi @ c @ s� e Zd ZdZedd� �Zd:dd�Zdd � Zd
d � Zd d
� Z d;dd�Z
d<dd�Z d=dd�Z d>dd�Z
d?dd�Zd@dd �Zd d � Zd d � ZdAd"d#�Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zd.d/� Zd0d1� Zd2d3� Zd4d5� ZdBd6d7�Z d8d9� Z dS )C� 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.
c C s& t | d�s"tjrtd��ntd��dS )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_biozKTLS in TLS requires SSLContext.wrap_bio() which isn't supported on Python 2zXTLS in TLS requires SSLContext.wrap_bio() which isn't available on non-native SSLContextN)�hasattrr �PY2r )� ssl_context� r �J/opt/alt/python37/lib/python3.7/site-packages/urllib3/util/ssltransport.py�$_validate_ssl_context_for_tls_in_tls s
z1SSLTransport._validate_ssl_context_for_tls_in_tlsNTc C sH t �� | _t �� | _|| _|| _|j| j| j|d�| _| �| jj � dS )zV
Create an SSLTransport around socket using the provided ssl_context.
)�server_hostnameN)
�sslZ MemoryBIO�incoming�outgoing�suppress_ragged_eofs�socketr �sslobj� _ssl_io_loopZ do_handshake)�selfr r r r r r r
�__init__, s
zSSLTransport.__init__c C s | S )Nr )r r r r
� __enter__? s zSSLTransport.__enter__c G s | � � d S )N)�close)r �_r r r
�__exit__B s zSSLTransport.__exit__c C s
| j �� S )N)r �fileno)r r r r
r E s zSSLTransport.fileno� c C s | � ||�S )N)�_wrap_ssl_read)r �len�bufferr r r
�readH s zSSLTransport.readr c C s |dkrt d��| �|�S )Nr z+non-zero flags not allowed in calls to recv)�
ValueErrorr )r r �flagsr r r
�recvK s zSSLTransport.recvc C s> |dkrt d��|r&|d kr&t|�}n |d kr2d}| �||�S )Nr z0non-zero flags not allowed in calls to recv_intoi )r r r )r r �nbytesr! r r r
� recv_intoP s
zSSLTransport.recv_intoc
C sr |dkrt d��d}t|��L}|�d��6}t|�}x&||k rX| �||d � �}||7 }q4W W d Q R X W d Q R X d S )Nr z.non-zero flags not allowed in calls to sendall�B)r �
memoryview�castr �send)r �datar! �count�view� byte_view�amount�vr r r
�sendallY s
zSSLTransport.sendallc C s$ |dkrt d��| �| jj|�}|S )Nr z+non-zero flags not allowed in calls to send)r r r �write)r r) r! �responser r r
r( c s zSSLTransport.send�rc
C s t |�dddhks td|f ��d|k}d|kp4| }|sB|sBt�d|k}d} |rZ| d7 } |rf| d7 } t�| | �}
| j jd7 _|dkr�d}|d k r�tj}|d kr�|s�td
��|
S |r�|r�t�|
|
|�} n&|r�t� |
|�} n|s�t�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.
r2 �w�bz&invalid mode %r (only r, w, b allowed)� � N���r z!unbuffered streams must be binary)
�setr �AssertionErrorr �SocketIO�_io_refs�io�DEFAULT_BUFFER_SIZE�BufferedRWPair�BufferedReader�BufferedWriter�
TextIOWrapper�mode)
r rB � buffering�encoding�errors�newline�writing�reading�binary�rawmode�rawr �textr r r
�makefilei s@
zSSLTransport.makefilec C s | � | jj� d S )N)r r �unwrap)r r r r
rN � s zSSLTransport.unwrapc C s | j �� d S )N)r r )r r r r
r � s zSSLTransport.closeFc C s | j �|�S )N)r � getpeercert)r Z binary_formr r r
rO � s zSSLTransport.getpeercertc C s
| j �� S )N)r �version)r r r r
rP � s zSSLTransport.versionc C s
| j �� S )N)r �cipher)r r r r
rQ � s zSSLTransport.cipherc C s
| j �� S )N)r �selected_alpn_protocol)r r r r
rR � s z#SSLTransport.selected_alpn_protocolc C s
| j �� S )N)r �selected_npn_protocol)r r r r
rS � s z"SSLTransport.selected_npn_protocolc C s
| j �� S )N)r �shared_ciphers)r r r r
rT � s zSSLTransport.shared_ciphersc C s
| j �� S )N)r � compression)r r r r
rU � s zSSLTransport.compressionc C s | j �|� d S )N)r �
settimeout)r �valuer r r
rV � s zSSLTransport.settimeoutc C s
| j �� S )N)r �
gettimeout)r r r r
rX � s zSSLTransport.gettimeoutc C s | j �� d S )N)r �_decref_socketios)r r r r
rY � s z SSLTransport._decref_socketiosc
C sT y| � | jj||�S tjk
rN } z |jtjkr<| jr<dS � W d d }~X Y nX d S )Nr )r r r r
�SSLError�errnoZ
SSL_ERROR_EOFr )r r r �er r r
r � s zSSLTransport._wrap_ssl_readc
G s� d}d}x�|r�d}y ||� }W n@ t jk
r^ } z |jt jt jfkrH|�|j}W dd}~X Y nX | j�� }| j�|� |dkr�d}q
|t jkr
| j� t
�}|r�| j � |� q
| j �
� q
W |S )z>Performs an I/O loop between incoming/outgoing and the socket.TNF)r
rZ r[ ZSSL_ERROR_WANT_READZSSL_ERROR_WANT_WRITEr r r r/ r" �
SSL_BLOCKSIZEr r0 Z write_eof)r �func�argsZ should_loop�retr[ r\ �bufr r r
r � s(
zSSLTransport._ssl_io_loop)NT)r N)r r )Nr )r )r )r2 NNNN)F)N) �__name__�
__module__� __qualname__�__doc__� staticmethodr r r r r r r" r$ r/ r( rM rN r rO rP rQ rR rS rT rU rV rX rY r r r r r r
r s4
,
r ) r< r r
Zurllib3.exceptionsr Zurllib3.packagesr r] r r r r r
�<module> s 