관리-도구
편집 파일: SSL.cpython-36.pyc
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 "