File: //lib/python3.9/site-packages/rhn/__pycache__/SSL.cpython-39.pyc
a
ꑐc-+ � @ s� d Z ddlmZ ddlmZ ddlZddlZddlZddlmZ ddl Z dZ
e ed�r^ej Z
n ddlm
Z
G d d
� d
�Zd d � ZG d
d� dejej�ZdS )zK
rhn.SSL builds an abstraction on top of the objects provided by pyOpenSSL
� )�SSL)�cryptoN)�bstr�x �sslerror)� socket_errorc @ s� e Zd ZdZd(dd�Zdd� Zdd� Zd)d d
�Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zd*dd�Zd+d d �Zd d � Zd d!� Zd"d#� Zd$d%� ZeZeZd,d&d'�ZdS )-� SSLSocketzM
Class that wraps a pyOpenSSL Connection object, adding more methods
Nc C sj d | _ d | _|| _g | _|p g }|D ]}| �|� q$tj| _tj| _ d| _
d| _ t d�| _
d| _d | _d S )Ni r � )�_ctx� _connection�_sock�_trusted_certs�add_trusted_certr Z
SSLv23_METHOD� _ssl_methodZ VERIFY_PEER�_ssl_verify_flags� _buffer_size�_posr �_buffer�_makefile_called�_closed)�self�socketZ
trusted_certs�f� r �+/usr/lib/python3.9/site-packages/rhn/SSL.py�__init__/ s
zSSLSocket.__init__c C s0 t �|t j�std| ��| j�|�d�� dS )zh
Adds a trusted certificate to the certificate store of the SSL context
object.
z"Unable to read certificate file %szutf-8N)�os�access�R_OK�
ValueErrorr
�append�encode)r �filer r r r L s zSSLSocket.add_trusted_certc C s� | � � t�| j�| _| j�tj� | j�tj� | jrR| jD ]}| j� |� q>nd| _
| j� | j
t � t
td�r�| j�tj� t�| j| j�| _| j�� dS )z1
Initializes the SSL connection.
r � OP_DONT_INSERT_EMPTY_FRAGMENTSN)�
_check_closedr ZContextr r
Z set_optionsZ OP_NO_SSLv2Z OP_NO_SSLv3r
Zload_verify_locationsr Z
set_verify�ssl_verify_callback�hasattrr# Z
Connectionr r Zset_connect_state)r r r r r �init_sslU s
zSSLSocket.init_sslc C s |r
|| _ | jd | _| S )zG
Returns self, since we are a file-like object already
� )r r )r �mode�bufsizer r r �makefiles s zSSLSocket.makefilec C s, | j r
dS | js | �� dS | jd | _dS )z+
Closes the SSL connection
Nr( )r r �
_really_close�r r r r �close� s zSSLSocket.closec C s d S �Nr r- r r r �flush� s zSSLSocket.flushc C s� | j d u rd S d }zt| j d�}W n ty@ t| j d�}Y n0 |d ur�tjd dkrn|� dkr�| j �� n|� dkr�| j �� | j �� d| _d S )NZ state_stringZget_state_stringr � s% SSL negotiation finished successfullyz%SSL negotiation finished successfullyr( )r �getattr�AttributeError�sys� version_infoZshutdownr. r )r Z get_stater r r r, � s
zSSLSocket._really_closec C s | j rtd��d S )Nz I/O operation on closed file)r r r- r r r r$ � s zSSLSocket._check_closedc C s$ t | j|�rt| j|�S t|��d S r/ )r& r r2 r3 )r �namer r r � __getattr__� s zSSLSocket.__getattr__c C s dS )z'
Returns false always.
r r r- r r r �isatty� s zSSLSocket.isattyc C s | j S r/ )r r- r r r �tell� s zSSLSocket.tellr c C s t d��d S )N�seek)�NotImplementedError)r �posr) r r r r: � s zSSLSocket.seekc C sX | � � | j}t| j�}||k s*|du �r|dur@t|| |�}z>| j�|�}| j| | _t| j�}| j�� }|dkr|W �qW q tj y� Y �qY q tj
y� t � � d }t
d|j� Y �qY q tjy� | �tjd� Y q tj�y
| �tjd� Y q0 q|�r4| jd|� }| j|d� | _n| j}td�| _| jt|� | _|S )z@
Reads up to amt bytes from the SSL connection.
Nr r( z
SSL exception�readr )r$ r �lenr �minr �recv�pendingr �ZeroReturnErrorZ SysCallErrorr4 �exc_info�print�args�WantWriteError�_poll�select�POLLOUT�
WantReadError�POLLINr r )r �amtZ buffer_sizeZ
buffer_length�datarA �e�retr r r r= � s<
zSSLSocket.readc C s | � t|��|d d �< t|�S r/ )r= r> )r �bufr r r �readinto� s zSSLSocket.readintoc C sB t �� }|�| j|� |�| j�� d �}|g kr>td| ��d S )Ni� zConnection timed out on %s)rH Zpoll�registerr Z
gettimeout�TimeoutException)r Z filter_typeZ caller_nameZpoller�resr r r rG � s
zSSLSocket._pollc C s� | � � t|�}z,| j�|�}|t|�kr.W q�||d� }W q tjy^ | �tjd� Y q tj y~ | �tj
d� Y q0 q|S )z/
Writes to the SSL connection.
N�write) r$ r> r �sendr rF rG rH rI rJ rK )r rM Zoriglen�sentr r r rU � s zSSLSocket.writec C s
| � |�S r/ )r= )r rL r r r r@ s zSSLSocket.recvc C s: | � � d}| j�td��}|dkr.|d }n|rDt| j�|krD|}|dur~| jd|� }| j|d� | _| jt|� | _|S | j}|r�t| j|t| j� �}z | j� |�}| j| | _W q t
j y� Y �qY q t
j y� | �
tjd� Y q t
j�y | �
tjd� Y q0 q| j}d| _| jt|� | _|S )zg
Reads a single line (up to `length' characters long) from the SSL
connection.
N�
r r( �readliner )r$ r �findr r> r r r? r r@ r rB rF rG rH rI rJ rK )r �lengthZ charcount�irO r* rM r r r rY s8
zSSLSocket.readline)N)N)r )N)N)�__name__�
__module__� __qualname__�__doc__r r r' r+ r. r0 r, r$ r7 r8 r9 r: r= rQ rG rU r@ rV ZsendallrY r r r r r + s*
3r c C s |S )zb
Verify callback, which will be called for each certificate in the
certificate chain.
r )ZconnZcertZerrnum�depth�okr r r r% K s r% c @ s e Zd Zdd� Zdd� ZdS )rS c G s
|| _ d S r/ )rE )r rE r r r r U s zTimeoutException.__init__c C s dS )NzTimeout Exceptionr r- r r r �__str__X s zTimeoutException.__str__N)r] r^ r_ r rc r r r r rS S s rS )r` ZOpenSSLr r r r rH Zrhn.i18nr r4 ZDEFAULT_TIMEOUTr& r r Zsslr r% �Error�timeoutrS r r r r �<module> s
"