Your IP : 216.73.217.129
���]�@ � @ s d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l m Z m Z d d l
m Z e j
d � Z e j Z d Z d Z e e d � rd d l Z d Z e j j e j j e � p� e j d � Z e j j e j j e � p
e j d � Z e e d
� Z d Z d Z Gd
d � d e j � Z Gd d � d e j e j! � Z" Gd d � d e � Z# e r�d d l$ m% Z% Gd d � d e% e � Z& e Gd d � d e# � � Z' e Gd d � d e# � � Z( Gd d � d e � Z) d d � Z* e+ d k re* � d S) � N)�TestCase�
skipUnless)�support� threadingF�POP3_SSLTz keycert3.pemz pycacert.pemzSSL not supporteds 1 1
2 2
3 3
4 4
5 5
.
st From: postmaster@python.org
Content-Type: text/plain
MIME-Version: 1.0
Subject: Dummy
line1
line2
line3
.
c @ sE e Z d Z d g d d g i Z d Z d d � Z d d � Z d d
� Z d d � Z d
d � Z d d � Z
d d � Z d d � Z d d � Z
d d � Z e Z d d � Z e Z d d � Z d d � Z d d � Z d! d" � Z d# d$ � Z d% d& � Z d' d( � Z d) d* � Z e rAd+ d, � Z d- d. � Z d/ d0 � Z d1 S)2�DummyPOP3HandlerZUIDL�IMPLEMENTATIONzpython-testlib-pop-serverFc C sL t j j | | � | j d � g | _ | j d � d | _ d | _ d S)Ns
z(+OK dummy pop3 server ready. <timestamp>F)�asynchat�
async_chat�__init__�set_terminator� in_buffer�push�
tls_active� tls_starting)�self�conn� r �0/opt/alt/python35/lib64/python3.5/test_poplib.pyr 1 s
zDummyPOP3Handler.__init__c C s | j j | � d S)N)r
�append)r �datar r r �collect_incoming_data9 s z&DummyPOP3Handler.collect_incoming_datac C s� d j | j � } t | d � } g | _ | j d � d j � } | j d � } | d k ru | | d d � } n d } t | d | � r� t | d | � } | | � n | j d | � d S)
N� z
ISO-8859-1� r � � Zcmd_z$-ERR unrecognized POP3 command "%s".���) �joinr
�str�split�lower�find�hasattr�getattrr )r �line�cmdZspace�arg�methodr r r �found_terminator< s
z!DummyPOP3Handler.found_terminatorc C s � d S)Nr )r r r r � handle_errorL s z DummyPOP3Handler.handle_errorc C s$ t j j | | j d � d � d S)Nz
ISO-8859-1s
)r r
r �encode)r r r r r r O s zDummyPOP3Handler.pushc C s | j | � d S)N)r )r r& r r r �cmd_echoR s zDummyPOP3Handler.cmd_echoc C s* | d k r | j d � | j d � d S)N�guidoz-ERR no such userz+OK password required)r )r r& r r r �cmd_userV s
zDummyPOP3Handler.cmd_userc C s* | d k r | j d � | j d � d S)N�pythonz-ERR wrong passwordz+OK 10 messages)r )r r& r r r �cmd_pass[ s
zDummyPOP3Handler.cmd_passc C s | j d � d S)Nz
+OK 10 100)r )r r& r r r �cmd_stat` s zDummyPOP3Handler.cmd_statc C sD | r | j d | | f � n | j d � t j j | t � d S)Nz +OK %s %sz+OK)r r r
� LIST_RESP)r r& r r r �cmd_listc s
zDummyPOP3Handler.cmd_listc C s. | j d t t � � t j j | t � d S)Nz +OK %s bytes)r �len� RETR_RESPr r
)r r& r r r �cmd_retrl s zDummyPOP3Handler.cmd_retrc C s | j d � d S)Nz +OK message marked for deletion.)r )r r& r r r �cmd_deler s zDummyPOP3Handler.cmd_delec C s | j d � d S)Nz+OK done nothing.)r )r r& r r r �cmd_noopu s zDummyPOP3Handler.cmd_noopc C s | j d � d S)Nz+OK done nothing.)r )r r& r r r �cmd_rpopx s zDummyPOP3Handler.cmd_rpopc C s | j d � d S)Nz+OK done nothing.)r )r r& r r r �cmd_apop{ s zDummyPOP3Handler.cmd_apopc C s | j d � | j � d S)Nz +OK closing.)r Zclose_when_done)r r& r r r �cmd_quit~ s
zDummyPOP3Handler.cmd_quitc C s- t | j � } | j r) t r) g | d <| S)N�STLS)�dict�CAPASr � SUPPORTS_SSL)r Z_capasr r r �
_get_capas� s
zDummyPOP3Handler._get_capasc C s | j d � | j � rn xR | j � j � D]> \ } } | g } | rT | j | � | j d j | � � q, W| j d � d S)Nz+OK Capability list followsr �.)r r? �items�extendr )r r& Zcap�paramsZ_lnr r r �cmd_capa� s
zDummyPOP3Handler.cmd_capac C s | j | j r d n d � d S)Nz+OK I know RFC6856z-ERR What is UTF8?!)r � enable_UTF8)r r& r r r �cmd_utf8� s zDummyPOP3Handler.cmd_utf8c
C s� | j d k r� | j d � t j | j d t d d d d d d �} | j � | j | � d | _ d | _ g | _ | j
� n
| j d � d S) NFz+OK Begin TLS negotiation�certfileZ server_sideTZdo_handshake_on_connectZsuppress_ragged_eofsz*-ERR Command not permitted when TLS active) r r �sslZ wrap_socket�socket�CERTFILEZ del_channelZ
set_socketr r
�_do_tls_handshake)r r& Ztls_sockr r r �cmd_stls� s
zDummyPOP3Handler.cmd_stlsc C s� y | j j � Wn� t j k
r� } zK | j d t j t j f k rL d S| j d t j k rl | j � S� WYd d } ~ XnW t k
r� } z% | j d t
j k r� | j � SWYd d } ~ Xn Xd | _ d | _
d S)Nr TF)rI Z do_handshakerH ZSSLError�argsZSSL_ERROR_WANT_READZSSL_ERROR_WANT_WRITEZ
SSL_ERROR_EOF� handle_close�OSError�errnoZ ECONNABORTEDr r )r �errr r r rK � s
z"DummyPOP3Handler._do_tls_handshakec C sP | j r | j � n6 y t j j | � Wn t j k
rK | j � Yn Xd S)N)r rK r r
� handle_readrH Z SSLEOFErrorrN )r r r r rR � s
z DummyPOP3Handler.handle_readN) �__name__�
__module__� __qualname__r= rE r r r( r) r r+ r- r/ r0 r2 Zcmd_uidlr5 Zcmd_topr6 r7 r8 r9 r: r? rD rF r> rL rK rR r r r r r , s6
r c @ s� e Z d Z e Z e j d d � Z d d � Z d d � Z d d � Z
d d
� Z d d � Z e Z
d
d � Z d d � Z d S)�DummyPOP3Serverc C s� t j j | � t j j | � | j | t j � | j | � | j d � d | _
t j � | _ | j j
� d d � \ | _ | _ d | _ d S)N� F� )r �Threadr �asyncore�
dispatcherZ
create_socketrI � SOCK_STREAM�bind�listen�active�Lock� active_lock� getsockname�host�port�handler_instance)r �address�afr r r r � s
%zDummyPOP3Server.__init__c C s0 t j � | _ t j j | � | j j � d S)N)r �Event�_DummyPOP3Server__flagrY �start�wait)r r r r rj � s zDummyPOP3Server.startc C ss d | _ | j j � xF | j r^ t j r^ | j j � t j d d d d � | j j � q Wt j d d � d S)NT�timeoutg�������?�countr Z
ignore_all)
r_ ri �setrZ Z
socket_mapra �acquireZloop�releaseZ close_all)r r r r �run� s
zDummyPOP3Server.runc C s d | _ | j � d S)NF)r_ r )r r r r �stop� s zDummyPOP3Server.stopc C s | j | � | _ d S)N)�handlerre )r r �addrr r r �handle_accepted� s z DummyPOP3Server.handle_acceptedc C s | j � d S)N)�close)r r r r �handle_connect� s z DummyPOP3Server.handle_connectc C s d S)Nr r )r r r r �writable� s zDummyPOP3Server.writablec C s � d S)Nr )r r r r r) � s z DummyPOP3Server.handle_errorN)rS rT rU r rs rI �AF_INETr rj rq rr ru rw rR rx r) r r r r rV � s rV c @ sN e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d d � Z d
d � Z d d � Z
d d � Z d d � Z d d � Z
d d � Z d d � Z d d � Z d d � Z d d � Z d! d" � Z d# d$ � Z d% d& � Z d' d( � Z d) d* � Z d+ d, � Z e d- d. � � Z e d/ d0 � � Z e d1 d2 � � Z d3 S)4�
TestPOP3Classc C s | j | j d � � d S)Ns +OK)�
assertTrue�
startswith)r �respr r r �assertOK� s zTestPOP3Class.assertOKc C sM t t t f � | _ | j j � t j | j j | j j d d �| _ d S)Nrl � )
rV �HOST�PORT�serverrj �poplib�POP3rc rd �client)r r r r �setUp� s
zTestPOP3Class.setUpc C s | j j � | j j � d S)N)r� rv r� rr )r r r r �tearDown� s
zTestPOP3Class.tearDownc C s | j | j j � d � d S)Ns( +OK dummy pop3 server ready. <timestamp>)� assertEqualr� Z
getwelcome)r r r r �test_getwelcome s z TestPOP3Class.test_getwelcomec C s | j t j | j j d � d S)Nz echo -err)� assertRaisesr� � error_protor� � _shortcmd)r r r r �test_exceptions s z TestPOP3Class.test_exceptionsc C s9 | j | j j d � � | j t j | j j d � d S)Nr, �invalid)r~ r� �userr� r� r� )r r r r � test_user s zTestPOP3Class.test_userc C s9 | j | j j d � � | j t j | j j d � d S)Nr. r� )r~ r� Zpass_r� r� r� r� )r r r r �
test_pass_ s zTestPOP3Class.test_pass_c C s | j | j j � d � d S)N�
�d )r� r� )r� r� �stat)r r r r � test_stat s zTestPOP3Class.test_statc C s^ | j | j j � d d � d d d d d g d f � | j | j j d � j d � � d S)
Nr s 1 1s 2 2s 3 3s 4 4s 5 5� �1s OK 1 1)r� r� �listr{ �endswith)r r r r � test_list s
zTestPOP3Class.test_listc C sM d d d d d d d d d g d
f } | j j d � } | j | | � d S) Ns
+OK 116 bytess From: postmaster@python.orgs Content-Type: text/plains MIME-Version: 1.0s Subject: Dummyr s line1s line2s line3�q �foo)r� Zretrr� )r �expectedr� r r r � test_retr s zTestPOP3Class.test_retrc C s/ | j t j | j j d t j d d � d S)Nzecho +%sr� �a)r� r� r� r� r� Z_MAXLINE)r r r r �test_too_long_lines" s z!TestPOP3Class.test_too_long_linesc C s | j | j j d � � d S)Nr� )r~ r� Zdele)r r r r � test_dele& s zTestPOP3Class.test_delec C s | j | j j � � d S)N)r~ r� �noop)r r r r � test_noop) s zTestPOP3Class.test_noopc C s | j | j j d � � d S)Nr� )r~ r� Zrpop)r r r r � test_rpop, s zTestPOP3Class.test_rpopc C s | j | j j d d � � d S)Nr� Z
dummypassword)r~ r� �apop)r r r r �test_apop_normal/ s z TestPOP3Class.test_apop_normalc C sQ d d d } t j | j d | � �$ | j t j | j j d d � Wd QRXd S)Ns +OK� <i@B Zwelcomer� Zkb)� test_supportZ swap_attrr� r� r� r� r� )r Z evil_welcomer r r �test_apop_REDOS2 s z TestPOP3Class.test_apop_REDOSc C sJ d d d d d d d d d g d
f } | j | j j d d � | � d S) Ns
+OK 116 bytess From: postmaster@python.orgs Content-Type: text/plains MIME-Version: 1.0s Subject: Dummyr s line1s line2s line3r� r )r� r� �top)r r� r r r �test_top< s zTestPOP3Class.test_topc C s! | j j � | j j d � d S)Nr� )r� Zuidl)r r r r � test_uidlD s
zTestPOP3Class.test_uidlc C s, d | j j _ | j t j | j j � d S)NF)r� rs rE r� r� r� r� �utf8)r r r r � test_utf8_raises_if_unsupportedH s z-TestPOP3Class.test_utf8_raises_if_unsupportedc C s8 d | j j _ d } | j j � } | j | | � d S)NTs +OK I know RFC6856)r� rs rE r� r� r� )r r� �resultr r r � test_utf8L s zTestPOP3Class.test_utf8c C s, | j j � } | j d | j � k � d S)Nr )r� �capar{ �keys)r r� r r r � test_capaR s zTestPOP3Class.test_capac C sF | j j � } | j | � | j | j j � | j | j j � d S)N)r� �quitr{ � assertIsNone�sock�file)r r} r r r � test_quitV s
zTestPOP3Class.test_quitc C s, | j j � } | j d | j � k � d S)Nr; )r� r� r{ r� )r r� r r r �test_stls_capa\ s z TestPOP3Class.test_stls_capac C s) d } | j j � } | j | | � d S)Ns +OK Begin TLS negotiation)r� �stlsr� )r r� r} r r r � test_stlsa s zTestPOP3Class.test_stlsc C s� d } t j t j � } | j t � t j | _ d | _ | j t j � � | j
j d | � } Wd QRXt j
d | j j d d �| _
| j
j d | � } | j | | � d S)Ns +OK Begin TLS negotiationT�context� localhostrl r )rH �
SSLContext�PROTOCOL_TLSv1Zload_verify_locations�CAFILEZ
CERT_REQUIREDZ verify_modeZcheck_hostnamer� ZCertificateErrorr� r� r� r� r� rd r� )r r� �ctxr} r r r �test_stls_contextg s
!z TestPOP3Class.test_stls_contextN) rS rT rU r~ r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� � requires_sslr� r� r� r r r r rz � s2
rz )�
SSLConnectionc @ s e Z d Z d d � Z d S)�DummyPOP3_SSLHandlerc C sV t j j | | � | j � | j d � g | _ | j d � d | _ d | _ d S)Ns
z(+OK dummy pop3 server ready. <timestamp>TF) r r
r Zsecure_connectionr r
r r r )r r r r r r z s
z DummyPOP3_SSLHandler.__init__N)rS rT rU r r r r r r� x s r� c @ sR e Z d Z d d � Z d d � Z d d � Z d d � Z e Z d d
� Z d S) �TestPOP3_SSLClassc C sS t t t f � | _ t | j _ | j j � t j | j j | j j
� | _ d S)N) rV r� r� r� r� rs rj r� r rc rd r� )r r r r r� � s
zTestPOP3_SSLClass.setUpc C s | j d t j � d S)Nr )ZassertInr� �__all__)r r r r � test__all__� s z TestPOP3_SSLClass.test__all__c C s4 t j t j � } | j t t j | j j | j j d t
d | �| j t t j | j j | j j d t
d | �| j t t j | j j | j j d t
d t
d | �| j j � t j | j j | j j d | �| _ | j
| j j t j � | j | j j j | � | j | j j � j d � � d S)NZkeyfiler� rG s +OK)rH r� r� r� �
ValueErrorr� r r� rc rd rJ r� r� ZassertIsInstancer� Z SSLSocketZassertIsr� r{ r� r| )r r� r r r � test_context� s
z TestPOP3_SSLClass.test_contextc C s | j t j | j j � d S)N)r� r� r� r� r� )r r r r r� � s zTestPOP3_SSLClass.test_stlsc C s, | j j � } | j d | j � k � d S)Nr; )r� r� � assertFalser� )r r� r r r r� � s z TestPOP3_SSLClass.test_stls_capaN) rS rT rU r� r� r� r� r� r� r r r r r� � s r� c @ sF e Z d Z d d � Z d d � Z d d � Z e Z d d � Z d S)
�TestPOP3_TLSClassc C sZ t t t f � | _ | j j � t j | j j | j j d d �| _ | j j
� d S)Nrl r ) rV r� r� r� rj r� r� rc rd r� r� )r r r r r� � s
'zTestPOP3_TLSClass.setUpc C sk | j j d k rZ | j j d k rZ y | j j � Wn" t j k
rY | j j � Yn X| j j � d S)N) r� r� r� r� r� r� rv r� rr )r r r r r� � s $zTestPOP3_TLSClass.tearDownc C s | j t j | j j � d S)N)r� r� r� r� r� )r r r r r� � s zTestPOP3_TLSClass.test_stlsc C s, | j j � } | j d | j � k � d S)Ns STLS)r� r� r� r� )r r� r r r r� � s z TestPOP3_TLSClass.test_stls_capaN)rS rT rU r� r� r� r� r� r r r r r� � s
r� c @ sX e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d d � Z d
S)� TestTimeoutsc C s� t j � | _ t j t j t j � | _ | j j d � t j | j � | _
t j d | j d | j | j f � | _
| j
j d � | j
j � | j j � d S)N�< �targetrM T)r rh �evtrI ry r\ r� �
settimeoutr� Z bind_portrd rY r� �thread� setDaemonrj rk )r r r r r� � s *
zTestTimeouts.setUpc C s | j j � | ` d S)N)r� r )r r r r r� � s
zTestTimeouts.tearDownc C so | j � | j � zI y- | j � \ } } | j d � | j � Wn t j k
r[ Yn XWd | j � Xd S)Ns
+ Hola mundo
)r^ rn �accept�sendrv rI rl )r r� Zservr rt r r r r� � s
zTestTimeouts.serverc C st | j t j � � t j d � z t j t | j � } Wd t j d � X| j | j j
� d � | j j � d S)N� ) r� rI �getdefaulttimeout�setdefaulttimeoutr� r� r� rd r� r� �
gettimeoutrv )r �popr r r �testTimeoutDefault� s
z TestTimeouts.testTimeoutDefaultc C sw | j t j � � t j d � z t j t | j d d �} Wd t j d � X| j | j j � � | j j
� d S)Nr� rl ) r� rI r� r� r� r� r� rd r� r� rv )r r� r r r �testTimeoutNone� s
z TestTimeouts.testTimeoutNonec C sE t j t | j d d �} | j | j j � d � | j j � d S)Nrl r� )r� r� r� rd r� r� r� rv )r r� r r r �testTimeoutValue� s z TestTimeouts.testTimeoutValueN) rS rT rU r� r� r� r� r� r� r r r r r� � s
r� c C sD t t t t g } t j � } z t j | � Wd t j | � Xd S)N)rz r� r� r� r� Zthreading_setupZ run_unittestZthreading_cleanup)Ztests� thread_infor r r � test_main s r� �__main__),r� rZ r rI �os�timerP Zunittestr r Ztestr r� �
import_moduler r� r� r> r" rH �pathr �dirname�__file__�curdirrJ r� r� r1 r4 r
r r[ rY rV rz Ztest.test_ftplibr� r� r� r� r� r� rS r r r r �<module> sD ** � 1� ' 6