File: //usr/lib64/python3.6/site-packages/M2Crypto/__pycache__/BIO.cpython-36.pyc
3
���[�* � @ s� d dl mZ d dlZd dlmZmZmZ erHd dlmZm Z m
Z
m Z m Z ej
d�ZG dd� de�Zeje� G dd� de�ZG d d
� d
e�ZG d d � d e�Zddd�ZG dd� de�ZG dd� de�ZG dd� de�ZdS )� )�absolute_importN)�m2�py27plus�six)�AnyStr�Callable�Iterable�Optional�Union�BIOc @ s e Zd ZdS )�BIOErrorN)�__name__�
__module__� __qualname__� r r �/usr/lib64/python3.6/BIO.pyr s r c @ s� e Zd ZdZejZd0dd�Zdd� Zdd � Z e Z
d
d � Z d d
� Z d1dd�Z
d2dd�Zd3dd�Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zd.d/� Z dS )4r z)Abstract object interface to the BIO API.Nr c C s" || _ || _|| _d| _d| _d S )Nr )�bio�_pyfree� _close_cb�closed� write_closed)�selfr r r r r r �__init__ s
z BIO.__init__c C s | j r| j| j� d S )N)r � m2_bio_freer )r r r r �__del__$ s z BIO.__del__c C s | j S )N)r )r r r r �_ptr( s zBIO._ptrc C s t j| j�S )N)r Z
bio_get_fdr )r r r r �fileno. s z
BIO.filenoc C s | j S )N)r )r r r r �readable2 s z BIO.readablec C sx | j � std��|d krDt� }x tj| jd�}|s4P ||7 }q W |S |dkrPdS |dk rbtd��nttj| j|��S d S )Nz cannot readi r � zread count is negative)r �IOError� bytearrayr �bio_readr �
ValueError�bytes)r �size�buf�datar r r �read6 s
zBIO.read� c C s8 | j � std��tj| j|�}|d kr*dn|}tj|�S )Nz cannot read� )r r r �bio_getsr r �
ensure_binary)r r$ r% r r r �readlineI s
z BIO.readline�ignoredc C sF | j � std��g }x,tj| jd�}|d kr.P |jtj|�� qW |S )Nz cannot readi )r r r r* r �appendr r+ )r �sizehint�linesr% r r r � readlinesQ s z
BIO.readlinesc C s | j o| j S )N)r r )r r r r � writeable] s z
BIO.writeablec C s4 | j � std��t|tj�r&|jd�}tj| j|�S )z�Write data to BIO.
:return: either data written, or [0, -1] for nothing written,
-2 not implemented
z cannot write�utf8) r2 r �
isinstancer Z text_type�encoder � bio_writer )r r& r r r �writea s
z BIO.writec C s
d| _ d S )N� )r )r r r r � write_closen s zBIO.write_closec C s t j| j� dS )zTFlush the buffers.
:return: 1 for success, and 0 or -1 for failure
N)r Z bio_flushr )r r r r �flushr s z BIO.flushc C s t j| j�S )zcSet the bio to its initial state.
:return: 1 for success, and 0 or -1 for failure
)r Z bio_resetr )r r r r �resetz s z BIO.resetc C s d| _ | jr| j� d S )Nr8 )r r )r r r r �close� s z BIO.closec C s t j| j�S )za
Can the call be attempted again, or was there an error
ie do_handshake
)r Zbio_should_retryr )r r r r � should_retry� s zBIO.should_retryc C s t j| j�S )zShould we read more data?)r Zbio_should_readr )r r r r � should_read� s zBIO.should_readc C s t j| j�S )zShould we write more data?)r Zbio_should_writer )r r r r � should_write� s zBIO.should_writec C s t j| j�S )zReturn the current offset.)r Zbio_tellr )r r r r �tell� s zBIO.tellc C s t j| j|�S )z&Seek to the specified absolute offset.)r Zbio_seekr )r Zoffr r r �seek� s zBIO.seekc C s | S )Nr )r r r r � __enter__� s z
BIO.__enter__c G s | j � d S )N)r< )r �argsr r r �__exit__� s z BIO.__exit__)Nr N)N)r( )r- ) r
r r �__doc__r �bio_freer r r r Zbio_ptrr r r' r, r1 r2 r7 r9 r: r; r< r= r> r? r@ rA rB rD r r r r r s0
c sL e Zd ZdZd � fdd� Zdd� Zd
dd �Ze ZZ� fd
d �Z e Z
� Z S )� MemoryBufferan Object interface to BIO_s_mem.
Empirical testing suggests that this class performs less well than
cStringIO, because cStringIO is implemented in C, whereas this class
is implemented in Python. Thus, the recommended practice is to use
cStringIO for regular work and convert said cStringIO object to
a MemoryBuffer object only when necessary.
Nc sh t t| �j| � |d k r8t|t� r8tdt|�jf ��tj tj
� �| _ d| _ |d k rdtj
| j |� d S )Nz"data must be bytes or None, not %sr8 )�superrG r r4 r# � TypeError�typer
r �bio_newZ bio_s_memr r r6 )r r& )� __class__r r r � s zMemoryBuffer.__init__c C s t j| j�S )N)r �bio_ctrl_pendingr )r r r r �__len__� s zMemoryBuffer.__len__r c C s< | j � std��|r"tj| j|�S tj| jtj| j��S d S )Nz cannot read)r r r r! r rM )r r$ r r r r' � s
zMemoryBuffer.readc s t t| �j� tj| jd� d S )Nr )rH rG r9 r Zbio_set_mem_eof_returnr )r )rL r r r9 � s zMemoryBuffer.write_close)N)r ) r
r r rE r rN r' �getvalueZread_allr9 r< �
__classcell__r r )rL r rG � s
rG c sN e Zd ZdZd� fdd� Z� fdd�Z� fdd �Z� fd
d �Zd d
� Z� Z S )�Filez�Object interface to BIO_s_pyfd.
This class interfaces Python to OpenSSL functions that expect BIO. For
general file manipulation in Python, use Python's builtin file object.
r8 �rbc s� t t| �j| dd� t|tj�r*t||�}|j� |j| _ || _
t t d�r`t j
|j� t j�| _nt j|t j�| _|| _d| _d S )Nr8 )r � bio_new_pyfdF)rH rQ r r4 r Z string_types�openr: �nameZfname�pyfile�hasattrr rS r � bio_nocloser Zbio_new_pyfile� close_pyfiler )r rV rY �mode)rL r r r � s
z
File.__init__c s t t| �j� | jj� d S )N)rH rQ r: rV )r )rL r r r: � s z
File.flushc s* | j � tt| �j� | jr&| jj� d S )N)r: rH rQ r< rY rV )r )rL r r r< s z
File.closec s t t| �j� S )z^Set the bio to its initial state.
:return: 0 for success, and -1 for failure
)rH rQ r; )r )rL r r r; s z
File.resetc C s | j stj| j� d S )N)r r rF r )r r r r r s z File.__del__)r8 rR )
r
r r rE r r: r< r; r rP r r )rL r rQ � s rQ rR c C sD yt | |�}W n, tk
r: } zt|j��W Y d d }~X nX t|�S )N)rT r r rC rQ )�filenamerZ �fZexr r r �openfile s
r] c s> e Zd ZdZejZejZd
� fdd� Z dd� Z
dd � Z � Z S ) �IOBufferz�Object interface to BIO_f_buffer.
Its principal function is to be BIO_push()'ed on top of a BIO_f_ssl, so
that makefile() of said underlying SSL socket works.
�rwbr8 c sX t t| �j| |d� tjtj� �| _tj| j|j� �| _ || _
d|krNd| _ nd| _ d S )N)r �wr r8 ) rH r^ r r rK Z bio_f_buffer�io�bio_pushr r Z
_under_bior )r Z under_biorZ r )rL r r r ( s zIOBuffer.__init__c C s( t | dd�r| j| j� | j| j� d S )Nr r )�getattr�
m2_bio_popr r ra )r r r r r 4 s zIOBuffer.__del__c C s t j| � d S )N)r r< )r r r r r< : s zIOBuffer.close)r_ r8 )
r
r r rE r �bio_poprd rF r r r r< rP r r )rL r r^ s r^ c sR e Zd ZdZejZejZej Z
� fdd�Z dd� Z dd� Z
dd � Zd
d � Z� ZS ) � CipherStreamz!Object interface to BIO_f_cipher.c s4 t t| �j| dd� || _tjtj� �| _d| _d S )Nr8 )r r ) rH rf r �obior rK Z bio_f_cipherr r )r rg )rL r r r G s zCipherStream.__init__c C s t | dd�s| j� d S )Nr r8 )rc r< )r r r r r N s zCipherStream.__del__c C s" | j | j� | j| j� d| _d S )Nr8 )rd r r r )r r r r r< S s zCipherStream.closec C s | j j� d S )N)rg r9 )r r r r r9 Y s zCipherStream.write_closec C sz t t|d �}|d kr td|��n(t|t�s4|jd�}t|t�sH|jd�}tj| j|� ||t|�� tj | j| j
j � � d S )Nzunknown cipherr3 ) rc r r" r4 r# r5 Zbio_set_cipherr �intrb rg r )r Zalgo�keyZiv�opZcipherr r r �
set_cipher] s
zCipherStream.set_cipher)r
r r rE r ZPKCS5_SALT_LENZSALT_LENre rd rF r r r r< r9 rk rP r r )rL r rf ? s rf c s8 e Zd ZdZd � fdd� Zejfdd�Zdd� Z� Z S )
�SSLBioz Object interface to BIO_f_ssl.r8 c s. t t| �j| |d� tjtj� �| _d| _d S )N)r r )rH rl r r rK Z bio_f_sslr r )r r )rL r r r n s zSSLBio.__init__c C s2 d| _ tj| j|j|� |tjkr.|jtj� dS )zf
Sets the bio to the SSL pointer which is
contained in the connection object.
r N)r r Z bio_set_sslr ZsslrX Zset_ssl_close_flagZ bio_close)r ZconnZ
close_flagr r r �set_sslt s
zSSLBio.set_sslc C s t j| j�S )z�Do the handshake.
Return 1 if the handshake completes
Return 0 or a negative number if there is a problem
)r Zbio_do_handshaker )r r r r � do_handshake s zSSLBio.do_handshake)r8 )
r
r r rE r r rX rm rn rP r r )rL r rl k s rl )rR )Z
__future__r ZloggingZM2Cryptor r r Ztypingr r r r r
Z getLogger�logr" r Zbio_init�objectr rG rQ r] r^ rf rl r r r r �<module> s
-;
!,