�
�+Xc@s�dZd
dl
mZ
d
dl
mZ
d
dlZd
dlZd
dlZd
dlmZ
d
dl	Z	dZ
eed�r�ejZ
nd
dlm
Z

d	fd
��YZd�Zdejejfd
��YZdS(sK
rhn.SSL builds an abstraction on top of the objects provided by pyOpenSSL
i����(
tSSL(
tcryptoN(
tbstrixtsslerror(
tsocket_errort	SSLSocketcBs�eZ
dZdd
�
Zd�Zd�Zdd�
Zd�Zd�Z	d�Z
d�Zd	�Zd
�Z
dd�
Zdd
�
Zd�Zd�Zd�Zd�ZeZeZdd�
ZRS(sM
    Class that wraps a pyOpenSSL Connection object, adding more methods
    cCs�d|_
d|_|
|_g|_|p-g}x|D]}|j|�

q7Wtj|_tj	|_
d
|_d|_t
d�
|_d|_d|_dS(Ni it(tNonet_ctxt_connectiont_sockt_trusted_certstadd_trusted_certRt
SSLv23_METHODt_ssl_methodtVERIFY_PEERt_ssl_verify_flagst_buffer_sizet_posRt_buffert_makefile_calledt_closed(tselftsockett
trusted_certst
f((s+/usr/lib/python2.7/site-packages/rhn/SSL.pyt__init__/s		
	
	


			cCsEtj
|
tj�s(td
|
�
�
n|jj|
jd�
�

dS(sh
        Adds a trusted certificate to the certificate store of the SSL context
        object.
        s"Unable to read certificate file %ssutf-8N(tostaccesstR_OKt
ValueErrorRtappendtencode(Rtfile((s+/usr/lib/python2.7/site-packages/rhn/SSL.pyRLs

c
Cs�|j�
t
j|j�
|_|jrOx-|jD]}
|jj|
�

q2Wn	d
|_|jj|jt	�
t
t
d�r�|jjt
j�

nt
j
|j|j�|_|jj�
dS(s1
        Initializes the SSL connection.
        itOP_DONT_INSERT_EMPTY_FRAGMENTSN(t
_check_closedRtContextRRRtload_verify_locationsRt
set_verifytssl_verify_callbackthasattrtset_optionsR"t
ConnectionR
R	tset_connect_state(RR((s+/usr/lib/python2.7/site-packages/rhn/SSL.pytinit_sslUs

	
	
cCs&|r||_n|j
d
|_
|S(sG
        Returns self, since we are a file-like object already
        i
(RR(Rtmodetbufsize((s+/usr/lib/python2.7/site-packages/rhn/SSL.pytmakefileqs

c

Cs8|jr
d
S|j
s$|j�
d
S|j
d|_
d
S(s+
        Closes the SSL connection
        Ni
(RRt
_really_close(
R((s+/usr/lib/python2.7/site-packages/rhn/SSL.pytclose~s	
	



c

Cs'|jj
�
|jj�
d
|_dS(Ni
(R	tshutdownR1R(
R((s+/usr/lib/python2.7/site-packages/rhn/SSL.pyR0�s




c

Cs|jrt
d
�
�
ndS(NsI/O operation on closed file(RR(
R((s+/usr/lib/python2.7/site-packages/rhn/SSL.pyR#�s
	
cCs2t|j
|
�r"t|j
|
�St|
�
�
dS(
N(R(R	tgetattrtAttributeError(Rtname((s+/usr/lib/python2.7/site-packages/rhn/SSL.pyt__getattr__�s


c


Csd
S(s'
        Returns false always.
        i((
R((s+/usr/lib/python2.7/site-packages/rhn/SSL.pytisatty�sc


Cs|jS(
N(
R(
R((s+/usr/lib/python2.7/site-packages/rhn/SSL.pyttell�s
icCstd
�
�
dS(Ntseek(
tNotImplementedError(RtposR-((s+/usr/lib/python2.7/site-packages/rhn/SSL.pyR9�s
cCs�
|j�
|j
}t|j�
}x(
||
ks=|
dkrL
|
dk	r_t|
||�}nyT|jj|�
}|j||_t|j�
}|jj�}|d
kr�PnWq%t	j
k
r�


Pq%t	jk
r�


tj
�d}d|jfGHPq%t	jk
r"



|jtjd�
q%t	jk
rH



|jtjd�
q%Xq%W|
rs
|j|
 }|j|
|_n|j}td�
|_|jt|�
|_|S(s@
        Reads up to amt bytes from the SSL connection.
        ii
s
SSL exceptiontreadRN(R#RtlenRRtminR	trecvtpendingRtZeroReturnErrortSysCallErrortsystexc_infotargstWantWriteErrort_polltselecttPOLLOUTt
WantReadErrortPOLLINRR(Rtamttbuffer_sizet
buffer_lengthtdataR@t
etret((s+/usr/lib/python2.7/site-packages/rhn/SSL.pyR<�s<
	















	

cCs |jt
|
�
�
|
(t
|
�
S(
N(R<R=(Rtbuf((s+/usr/lib/python2.7/site-packages/rhn/SSL.pytreadinto�s

cCs^tj
�}|j|j|
�
|j
|jj�d
�
}|gkrZtd|�
�
ndS(Ni�sConnection timed out on %s(RHtpolltregisterR
t
gettimeouttTimeoutException(Rtfilter_typetcaller_nametpollertres((s+/usr/lib/python2.7/site-packages/rhn/SSL.pyRG�s





cCs�|j�
t
|
�
}x�tr�y6|jj|
�
}|t
|
�
krJPn|
|}
Wqtjk
r~


|jtj	d
�
qtj
k
r�


|jtjd
�
qXqW|S(s/
        Writes to the SSL connection.
        twrite(R#R=tTrueR	tsendRRFRGRHRIRJRK(RROtoriglentsent((s+/usr/lib/python2.7/site-packages/rhn/SSL.pyR\�s

	








cCs
|j|
�
S(
N(
R<(RRL((s+/usr/lib/python2.7/site-packages/rhn/SSL.pyR?�s
cCs�
|j�
xc
t
ro
d}|jjtd
�
�
}|dkrJ|d}n$|
rnt|j�
|
krn|
}n|dk	r�|j| }|j||_|jt|�
|_|S|j}|
r�t	|j|
t|j�
�}ny&|j
j|�
}|j||_Wq
tj
k
r



Pq
tjk
rE



|jtjd�
q
tjk
rk



|jtjd�
q
Xq
W|j}d|_|jt|�
|_|S(sg
        Reads a single line (up to `length' characters long) from the SSL
        connection.
        s

ii
treadlineRN(R#R]RRtfindRR=RRR>R	R?RRARFRGRHRIRJRK(Rtlengtht	charcountt
iRQR.RO((s+/usr/lib/python2.7/site-packages/rhn/SSL.pyRa
s:

	




	




	
"






	
	

N(t__name__t
__module__t__doc__RRRR,R/R1R0R#R6R7R8R9R<RSRGR\R?R^tsendallRa(((s+/usr/lib/python2.7/site-packages/rhn/SSL.pyR+s(
			
						3				c
Cs|S(
sb
    Verify callback, which will be called for each certificate in the
    certificate chain.
    ((tconntcertterrnumtdepthtok((s+/usr/lib/python2.7/site-packages/rhn/SSL.pyR'2
sRWc
BseZ
d�Zd
�ZRS(c
Gs
|
|_dS(
N(
RE(RRE((s+/usr/lib/python2.7/site-packages/rhn/SSL.pyR<
s
c


Csd
S(NsTimeout Exception((
R((s+/usr/lib/python2.7/site-packages/rhn/SSL.pyt__str__?
s
(RfRgRRo(((s+/usr/lib/python2.7/site-packages/rhn/SSL.pyRW:
s	(RhtOpenSSLRR
RRRHtrhn.i18nRRCtDEFAULT_TIMEOUTR(RRtsslRR'tErrorttimeoutRW(((s+/usr/lib/python2.7/site-packages/rhn/SSL.pyt<module>s




�