Your IP : 18.191.223.30
3
�U�f-+ � @ 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 sn d | _ d | _|| _g | _|p g }x|D ]}| j|� q&W 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.6/SSL.py�__init__/ s
zSSLSocket.__init__c C s0 t j|t j�std| ��| jj|jd�� 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� | j � tj| j�| _| jjtj� | jjtj� | jrVx"| jD ]}| jj |� q@W nd| _
| jj | j
t � t
td�r�| jjtj� tj| j| j�| _| jj� 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 | j� 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 krd S d }yt| j d�}W n tk
rB t| j d�}Y nX |d k r�tjd dkrp|� dkr�| j j� n|� dkr�| j j� | 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 )N)r& r r0 r1 )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 )N)r )r r r r �tell� s zSSLSocket.tellr c C s t d��d S )N�seek)�NotImplementedError)r �posr) r r r r8 � s zSSLSocket.seekc C sX | j � | j}t| j�}x�||k s,|dk�r |dk rBt|| |�}y:| jj|�}| j| | _t| j�}| jj� }|dkrzP W q tj k
r� P Y q tj
k
r� t j � d }t
d|j� P Y q tjk
r� | jtjd� Y q tjk
�r | jtjd� Y qX qW |�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 SysCallErrorr2 �exc_info�print�args�WantWriteError�_poll�select�POLLOUT�
WantReadError�POLLINr r )r �amt� buffer_sizeZ
buffer_length�datar? �e�retr r r r; � s<
zSSLSocket.readc C s | j t|��|d d �< t|�S )N)r; r< )r Zbufr r r �readinto� s zSSLSocket.readintoc C sB t j� }|j| j|� |j| jj� d �}|g kr>td| ��d S )Ni� zConnection timed out on %s)rF Zpoll�registerr Z
gettimeout�TimeoutException)r Z filter_typeZ caller_nameZpoller�resr r r rE � s
zSSLSocket._pollc C s� | j � t|�}xvy*| jj|�}|t|�kr.P ||d� }W q tjk
r` | jtjd� Y q tj k
r� | jtj
d� Y qX qW |S )z/
Writes to the SSL connection.
N�write) r$ r< r �sendr rD rE rF rG rH rI )r rL ZoriglenZsentr r r rS � s zSSLSocket.writec C s
| j |�S )N)r; )r rJ r r r r> s zSSLSocket.recvc C sB | j � �xd}| jjtd��}|dkr2|d }n|rHt| j�|krH|}|dk r�| jd|� }| j|d� | _| jt|� | _|S | j}|r�t| j|t| j� �}y | jj |�}| j| | _W q t
j k
r� P Y q t
j k
r� | j
tjd� Y q t
jk
�r | j
tjd� Y q X q W | 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 r@ rD rE rF rG rH rI )r ZlengthZ charcount�irN r* rL r r r rV s:
zSSLSocket.readline)N)N)r )N)N)�__name__�
__module__� __qualname__�__doc__r r r' r+ r- r. r, r$ r5 r6 r7 r8 r; rO rE rS r> rT ZsendallrV 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 )rQ c G s
|| _ d S )N)rC )r rC r r r r U s zTimeoutException.__init__c C s dS )NzTimeout Exceptionr )r r r r �__str__X s zTimeoutException.__str__N)rY rZ r[ r r_ r r r r rQ S s rQ )r\ ZOpenSSLr r r r rF Zrhn.i18nr r2 ZDEFAULT_TIMEOUTr& r r Zsslr r% �ErrorZtimeoutrQ r r r r �<module> s
"