Your IP : 216.73.216.213
��YfT� �
@ s� d 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 Z d d l Z y d d l Z Wn e
k
r� d Z Yn Xd d l m Z m
Z
d d l m Z d d l m Z m Z e j d � Z d Z d d Z d
Z d
Z d Z Gd d
� d
e j � Z Gd d � d e j � Z Gd d � d e j e j � Z e d k r�e j j! e j j" e# � d � Z$ e j j! e j j" e# � d � Z% Gd d � d e j � Z& Gd d � d e& e � Z' Gd d � d e& e � Z( Gd d � d e � Z) Gd d � d e � Z* e
e j+ d � Gd d � d e � � Z, e
e d! � Gd" d# � d# e* � � Z- e
e d! � Gd$ d% � d% e � � Z. Gd& d' � d' e � Z/ d( d) � Z0 e1 d* k r�e0 � d S)+z Test script for ftplib module.� N)�TestCase�
skipUnless)�support)�HOST�HOSTv6� threading� z abcde12345
i� z
foo
bar
a� type=cdir;perm=el;unique==keVO1+ZF4; test
type=pdir;perm=e;unique==keVO1+d?3; ..
type=OS.unix=slink:/foobar;perm=;unique==keVO1+4G4; foobar
type=OS.unix=chr-13/29;perm=;unique==keVO1+5G4; device
type=OS.unix=blk-11/108;perm=;unique==keVO1+6G4; block
type=file;perm=awr;unique==keVO1+8G4; writable
type=dir;perm=cpmel;unique==keVO1+7G4; promiscuous
type=dir;perm=;unique==keVO1+1t2; no-exec
type=file;perm=r;unique==keVO1+EG4; two words
type=file;perm=r;unique==keVO1+IH4; leading space
type=file;perm=r;unique==keVO1+1G4; file1
type=dir;perm=cpmel;unique==keVO1+7G4; incoming
type=file;perm=r;unique==keVO1+1G4; file2
type=file;perm=r;unique==keVO1+1G4; file3
type=file;perm=r;unique==keVO1+1G4; file4
c sX e Z d Z d Z d d � Z d d � Z d d � Z � f d d � Z d
d � Z � S) �DummyDTPHandlerFc C s, t j j | | � | | _ d | j _ d S)N� )�asynchat�
async_chat�__init__� baseclass�last_received_data)�self�connr � r �5/opt/alt/python35/lib64/python3.5/test/test_ftplib.pyr
2 s zDummyDTPHandler.__init__c C s( | j j | j d � j d � 7_ d S)Ni �ascii)r r �recv�decode)r r r r � handle_read7 s zDummyDTPHandler.handle_readc C s0 | j s, | j j d � | j � d | _ d S)Nz226 transfer completeT)�dtp_conn_closedr �push�close)r r r r � handle_close: s
z DummyDTPHandler.handle_closec s] | j j d k r* | j j } d | j _ | s: | j � St t | � j | j d � � d S)Nr )r � next_data�close_when_done�superr r �encode)r Zwhat)� __class__r r r C s
zDummyDTPHandler.pushc C s
t � d S)N)� Exception)r r r r � handle_errorK s z DummyDTPHandler.handle_error) �__name__�
__module__� __qualname__r r
r r r r" r r )r r r / s r c @ s� e Z 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 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/ d0 � Z d1 d2 � Z d3 d4 � Z d5 d6 � Z d7 d8 � Z d9 d: � Z! d; d< � Z" d= d> � Z# d? d@ � Z$ dA dB � Z% dC S)D�DummyFTPHandlerc C s� t j j | | � | j j t j t j d � | j d � g | _ d | _ d | _
d | _ d | _ d | _
d | _ t | _ | j d � d S)N� s
r
z 220 welcome)r r r
�socketZ
setsockoptZ
SOL_SOCKETZ SO_OOBINLINEZset_terminator� in_buffer�dtp�last_received_cmdr �
next_responser �rest� RETR_DATA�next_retr_datar )r r r r r r
S s
zDummyFTPHandler.__init__c C s | j j | � d S)N)r) �append)r �datar r r �collect_incoming_datab s z%DummyFTPHandler.collect_incoming_datac C s� d j | j � j d � } g | _ | j rF | j | j � d | _ | j d � d j � } | | _ | j d � } | d k r� | | d d � } n d } t | d | � r� t
| d | � } | | � n | j d | � d S)
N� r r
� r r' Zcmd_z 550 command "%s" not understood.���) �joinr) r r, r �split�lowerr+ �find�hasattr�getattr)r �line�cmdZspace�arg�methodr r r �found_terminatore s
z DummyFTPHandler.found_terminatorc C s
t � d S)N)r! )r r r r r" x s z DummyFTPHandler.handle_errorc C s$ t j j | | j d � d � d S)Nr s
)r r r r )r r1 r r r r { s zDummyFTPHandler.pushc C s� t t t | j d � � � } d t | d d � � } | d d | d } t j | | f d t �} | j | d | �| _ | j
d � d S) N�,z %d.%d.%d.%d� � � �timeoutr z&200 active data connection established) �list�map�intr7 �tupler( �create_connection�TIMEOUT� dtp_handlerr* r )r r> �addr�ip�port�sr r r �cmd_port~ s zDummyFTPHandler.cmd_portc C s� t j � �� } | j | j j � d d f � | j � | j t � | j � d d � \ } } | j d d � } | d } | d } | j d | | | f � | j � \ } } | j | d | �| _
Wd QRXd S)Nr � �.rA rC z$227 entering passive mode (%s,%d,%d)r ) r( �bind� getsockname�listen�
settimeoutrK �replacer �acceptrL r* ) r r> �sockrN rO Zp1Zp2r rM r r r �cmd_pasv� s
zDummyFTPHandler.cmd_pasvc C sv | j | d � d d � \ } } } t | � } t j | | f d t �} | j | d | �| _ | j d � d S)Nr r' rE r z&200 active data connection establishedr5 )r7 rH r( rJ rK rL r* r )r r> �afrN rO rP r r r �cmd_eprt� s
& zDummyFTPHandler.cmd_eprtc C s� t j t j � �� } | j | j j � d d f � | j � | j t � | j � d } | j d | � | j � \ } } | j | d | �| _
Wd QRXd S)Nr r' z+229 entering extended passive mode (|||%d|)r ) r( �AF_INET6rT rU rV rW rK r rY rL r* )r r> rZ rO r rM r r r �cmd_epsv� s
zDummyFTPHandler.cmd_epsvc C s | j | � d S)N)r )r r> r r r �cmd_echo� s zDummyFTPHandler.cmd_echoc C s | j d � d S)Nz 200 noop ok)r )r r> r r r �cmd_noop� s zDummyFTPHandler.cmd_noopc C s | j d � d S)Nz331 username ok)r )r r> r r r �cmd_user� s zDummyFTPHandler.cmd_userc C s | j d � d S)Nz230 password ok)r )r r> r r r �cmd_pass� s zDummyFTPHandler.cmd_passc C s | j d � d S)Nz 230 acct ok)r )r r> r r r �cmd_acct� s zDummyFTPHandler.cmd_acctc C s | j d � d S)Nz 350 rnfr ok)r )r r> r r r �cmd_rnfr� s zDummyFTPHandler.cmd_rnfrc C s | j d � d S)Nz 250 rnto ok)r )r r> r r r �cmd_rnto� s zDummyFTPHandler.cmd_rntoc C s | j d � d S)Nz 250 dele ok)r )r r> r r r �cmd_dele� s zDummyFTPHandler.cmd_delec C s | j d � d S)Nz
250 cwd ok)r )r r> r r r �cmd_cwd� s zDummyFTPHandler.cmd_cwdc C s | j d � d S)Nz250 1000)r )r r> r r r �cmd_size� s zDummyFTPHandler.cmd_sizec C s | j d | � d S)Nz257 "%s")r )r r> r r r �cmd_mkd� s zDummyFTPHandler.cmd_mkdc C s | j d � d S)Nz
250 rmd ok)r )r r> r r r �cmd_rmd� s zDummyFTPHandler.cmd_rmdc C s | j d � d S)Nz 257 "pwd ok")r )r r> r r r �cmd_pwd� s zDummyFTPHandler.cmd_pwdc C s | j d � d S)Nz 200 type ok)r )r r> r r r �cmd_type� s zDummyFTPHandler.cmd_typec C s | j d � | j � d S)Nz 221 quit ok)r r )r r> r r r �cmd_quit� s
zDummyFTPHandler.cmd_quitc C s | j d � d S)Nz 226 abor ok)r )r r> r r r �cmd_abor� s zDummyFTPHandler.cmd_aborc C s | j d � d S)Nz 125 stor ok)r )r r> r r r �cmd_stor� s zDummyFTPHandler.cmd_storc C s | | _ | j d � d S)Nz 350 rest ok)r- r )r r> r r r �cmd_rest� s zDummyFTPHandler.cmd_restc C sk | j d � | j d k r. t | j � } n d } | j j | j | d � � | j j � d | _ d S)Nz 125 retr okr )r r- rH r* r/ r )r r> �offsetr r r �cmd_retr� s
zDummyFTPHandler.cmd_retrc C s. | j d � | j j t � | j j � d S)Nz 125 list ok)r r* � LIST_DATAr )r r> r r r �cmd_list� s
zDummyFTPHandler.cmd_listc C s. | j d � | j j t � | j j � d S)Nz 125 nlst ok)r r* � NLST_DATAr )r r> r r r �cmd_nlst� s
zDummyFTPHandler.cmd_nlstc C s | j d � d S)Nz 200 opts ok)r )r r> r r r �cmd_opts� s zDummyFTPHandler.cmd_optsc C s. | j d � | j j t � | j j � d S)Nz 125 mlsd ok)r r* � MLSD_DATAr )r r> r r r �cmd_mlsd� s
zDummyFTPHandler.cmd_mlsdc C s$ d t | � | _ | j d � d S)N�xz125 setlongretr ok)rH r/ r )r r> r r r �cmd_setlongretr� s z DummyFTPHandler.cmd_setlongretrN)&r# r$ r% r rL r
r2 r@ r" r rQ r[ r] r_ r` ra rb rc rd re rf rg rh ri rj rk rl rm rn ro rp rq rs ru rw rx rz r| r r r r r& O sD
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)�DummyFTPServerc C s� t j j | � t j j | � | j | t j � | j | � | j d � d | _
t j � | _ | j j
� d d � \ | _ | _ d | _ d S)NrD FrR )r �Threadr
�asyncore�
dispatcherZ
create_socketr( � SOCK_STREAMrT rV �activeZLock� active_lockrU �hostrO �handler_instance)r Zaddressr\ r r r r
s
%zDummyFTPServer.__init__c C s0 t j � | _ t j j | � | j j � d S)N)r �Event�_DummyFTPServer__flagr~ �start�wait)r r r r r� s zDummyFTPServer.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)NTrE g�������?�countr' Z
ignore_all)
r� r� �setr Z
socket_mapr� �acquireZloop�releaseZ close_all)r r r r �run s
zDummyFTPServer.runc C s d | _ | j � d S)NF)r� r6 )r r r r �stop s zDummyFTPServer.stopc C s | j | � | _ d S)N)�handlerr� )r r rM r r r �handle_accepted s z DummyFTPServer.handle_acceptedc C s | j � d S)N)r )r r r r �handle_connect# s z DummyFTPServer.handle_connectc C s d S)Nr r )r r r r �writable' s zDummyFTPServer.writablec C s
t � d S)N)r! )r r r r r" * s zDummyFTPServer.handle_errorN)r# r$ r% r& r� r( �AF_INETr
r� r� r� r� r� r r� r" r r r r r} � s r} z keycert3.pemz pycacert.pemc s� e Z d Z d Z d Z d Z d d � Z d d � Z � f d d � Z � f d d
� Z � f d d � Z
� f d
d � Z � f d d � Z d d � Z
� f d d � Z � S)�
SSLConnectionz3An asyncore.dispatcher subclass supporting TLS/SSL.Fc C sW t j | j d d d t d d d d d t j �} | j � | j | � d | _ d S)NZsuppress_ragged_eofsF�certfileZ server_sideTZdo_handshake_on_connect� ssl_version)�sslZ wrap_socketr( �CERTFILE�PROTOCOL_SSLv23Z del_channelZ
set_socket�_ssl_accepting)r r( r r r �secure_connection9 s
z SSLConnection.secure_connectionc 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 } ~ XnN t k
r� } z% | j d t
j k r� | j � SWYd d } ~ Xn
Xd | _ d S)Nr F)
r( Z do_handshaker� �SSLError�args�SSL_ERROR_WANT_READ�SSL_ERROR_WANT_WRITE�
SSL_ERROR_EOFr �OSError�errnoZ ECONNABORTEDr� )r �errr r r �_do_ssl_handshakeB s
z SSLConnection._do_ssl_handshakec s� d | _ y | j j � | _ Wno t j k
rl } z( | j d t j t j f k rZ d SWYd d } ~ Xn% t k
r� } z WYd d } ~ Xn Xd | _ t | d d � d k r� t
t | � j � n d S)NTr F�_ccc)
� _ssl_closingr( �unwrapr� r� r� r� r� r� r; r r� r )r r� )r r r �_do_ssl_shutdownR s z SSLConnection._do_ssl_shutdownc sC | j r | j � n) | j r, | j � n t t | � j � d S)N)r� r� r� r� r r� �handle_read_event)r )r r r r� f s
z SSLConnection.handle_read_eventc sC | j r | j � n) | j r, | j � n t t | � j � d S)N)r� r� r� r� r r� �handle_write_event)r )r r r r� n s
z SSLConnection.handle_write_eventc s{ y t t | � j | � SWnZ t j k
rv } z7 | j d t j t j t j t j f k ra d S� WYd d } ~ Xn Xd S)Nr )
r r� �sendr� r� r� r� �SSL_ERROR_ZERO_RETURNr� r� )r r1 r� )r r r r� v s zSSLConnection.sendc s� y t t | � j | � SWn{ t j k
r� } zX | j d t j t j f k rU d S| j d t j t j f k r� | j
� d S� WYd d } ~ Xn Xd S)Nr r3 ) r r� r r� r� r� r� r� r� r� r )r Z buffer_sizer� )r r r r � s
zSSLConnection.recvc C s
t � d S)N)r! )r r r r r" � s zSSLConnection.handle_errorc sK t | j t j � r4 | j j d k r4 | j � n t t | � j � d S)N) �
isinstancer( r� � SSLSocketZ_sslobjr� r r� r )r )r r r r � s
zSSLConnection.close)r# r$ r% �__doc__r� r� r� r� r� r� r� r� r r" r r r )r r r� 3 s
r� c @ s" e Z d Z d Z d d � Z d S)�DummyTLS_DTPHandlerz.A DummyDTPHandler subclass supporting TLS/SSL.c C s- t j | | | � | j j r) | j � d S)N)r r
r �secure_data_channelr� )r r r r r r r
� s z DummyTLS_DTPHandler.__init__N)r# r$ r% r� r
r r r r r� � s r� c @ sX e Z d Z d Z e Z d d � Z d d � Z d d � Z d d � Z d
d � Z
d S)
�DummyTLS_FTPHandlerz.A DummyFTPHandler subclass supporting TLS/SSL.c C s& t j | | � d | _ d | _ d S)NF)r& r
r� r� )r r r r r r
� s z DummyTLS_FTPHandler.__init__c C s | j d � | j � d S)z Set up secure control channel.z234 AUTH TLS successfulN)r r� )r r< r r r �cmd_auth� s
z DummyTLS_FTPHandler.cmd_authc C s$ | j d � d | _ | j � d S)Nz 220 Reverting back to clear-textT)r r� r� )r r< r r r �cmd_ccc� s
zDummyTLS_FTPHandler.cmd_cccc C s | j d � d S)z�Negotiate size of buffer for secure data transfer.
For TLS/SSL the only valid value for the parameter is '0'.
Any other value is accepted but ignored.
z200 PBSZ=0 successful.N)r )r r< r r r �cmd_pbsz� s z DummyTLS_FTPHandler.cmd_pbszc C sg | j � } | d k r1 | j d � d | _ n2 | d k rV | j d � d | _ n
| j d � d S) z Setup un/secure data channel.�Cz200 Protection set to ClearF�Pz 200 Protection set to PrivateTz(502 Unrecognized PROT type (use C or P).N)�upperr r� )r r< r> r r r �cmd_prot� s
z DummyTLS_FTPHandler.cmd_protN) r# r$ r% r� r� rL r
r� r� r� r� r r r r r� � s r� c @ s e Z d Z e Z d S)�DummyTLS_FTPServerN)r# r$ r% r� r� r r r r r� � s r� c @ s� 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 d- d. � Z d/ d0 � Z d1 d2 � Z d3 d4 � Z d5 d6 � Z d7 d8 � Z d9 d: � Z d; d< � Z d= d> � Z! d? d@ � Z" dA dB � Z# dC dD � Z$ dE dF � Z% dG dH � Z& dI dJ � Z' dK dL � Z( dM S)N� TestFTPClassc C sZ t t d f � | _ | j j � t j d t � | _ | j j | j j | j j
� d S)Nr rE ) r} r �serverr� �ftplib�FTPrK �client�connectr� rO )r r r r �setUp� s
zTestFTPClass.setUpc C s | j j � | j j � d S)N)r� r r� r� )r r r r �tearDown� s
zTestFTPClass.tearDownc C s0 | j t | � t | � � | j | | � d S)N)� assertEqual�len)r �receivedZexpectedr r r �
check_data� s zTestFTPClass.check_datac C s | j | j j � d � d S)Nz 220 welcome)r� r� Z
getwelcome)r r r r �test_getwelcome� s z TestFTPClass.test_getwelcomec C sj | j | j j d � t d � � | j | j j d � t d � � | j | j j d � t d � � d S)N�fooz
pass 12345z
pass *****z
PASS 12345z
PASS *****)r� r� Zsanitize�repr)r r r r �
test_sanitize� s ""zTestFTPClass.test_sanitizec C s� | j t | j j d � | j t | j j d � | j t | j j d � | j t j | j j d � | j t j | j j d � | j t j | j j d � | j t j | j j d � | j t j | j j d � d S) Nz
echo 40
0z echo 40
0z echo 40
0zecho 400zecho 499zecho 500zecho 599zecho 999)� assertRaises�
ValueErrorr� �sendcmdr� �
error_temp�
error_perm� error_proto)r r r r �test_exceptions� s z TestFTPClass.test_exceptionsc C sg t j t j t j t j t j t t f } x6 | D]. } y | d � � Wq1 t j k
r^ Yq1 Xq1 Wd S)Nz(exception not included in all_errors set) r� � error_replyr� r� r� �Errorr� �EOFErrorZ
all_errors)r �
exceptionsr{ r r r �test_all_errors� s
z TestFTPClass.test_all_errorsc C s] | j | j j � | j j d � | j | j j � | j j d � | j | j j � d S)NTF)�
assertTruer� Z
passiveserver�set_pasv� assertFalse)r r r r �
test_set_pasv� s
zTestFTPClass.test_set_pasvc C s\ | j j d � | j j d � | j t j | j j d � | j t j | j j d � d S)Nzecho 200zecho 299zecho 199zecho 300)r� Zvoidcmdr� r� r� )r r r r � test_voidcmd s zTestFTPClass.test_voidcmdc C s | j j � d S)N)r� �login)r r r r �
test_login s zTestFTPClass.test_loginc C s | j j d � d S)NZpasswd)r� Zacct)r r r r � test_acct s zTestFTPClass.test_acctc C sE | j j d d � d | j j _ | j t j | j j d d � d S)N�a�bZ200)r� �renamer� r� r, r� r� r� )r r r r � test_rename s zTestFTPClass.test_renamec C s? | j j d � d | j j _ | j t j | j j d � d S)Nr� Z199)r� Zdeleter� r� r, r� r� r� )r r r r � test_delete s zTestFTPClass.test_deletec C s | j j d � d S)Nr� )r� �size)r r r r � test_size s zTestFTPClass.test_sizec C s& | j j d � } | j | d � d S)Nz/foo)r� Zmkdr� )r �dirr r r �test_mkd s zTestFTPClass.test_mkdc C s | j j d � d S)Nr� )r� Zrmd)r r r r �test_rmd s zTestFTPClass.test_rmdc C s& | j j d � } | j | d � d S)Nz/fooz
250 cwd ok)r� �cwdr� )r r� r r r �test_cwd s zTestFTPClass.test_cwdc C s# | j j � } | j | d � d S)Nzpwd ok)r� �pwdr� )r r� r r r �test_pwd$ s zTestFTPClass.test_pwdc C s3 | j | j j � d � | j | j j d � d S)Nz 221 quit ok)r� r� �quitrZ )r r r r � test_quit( s zTestFTPClass.test_quitc C s | j j � d S)N)r� �abort)r r r r �
test_abort- s zTestFTPClass.test_abortc sH � f d d � } g � | j j d | � | j d j � � t � d S)Nc s � j | j d � � d S)Nr )r0 r )r1 )r� r r �callback1 s z.TestFTPClass.test_retrbinary.<locals>.callback�retrr
)r� �
retrbinaryr� r6 r. )r r� r )r� r �test_retrbinary0 s z TestFTPClass.test_retrbinaryc si � f d d � } xP d D]H } g � | j j d | d | �| j d j � � t | d � � q Wd S)
Nc s � j | j d � � d S)Nr )r0 r )r1 )r� r r r� 8 s z3TestFTPClass.test_retrbinary_rest.<locals>.callbackr �
� r� r- r
)r r� r� )r� r� r� r6 r. )r r� r- r )r� r �test_retrbinary_rest7 s
z!TestFTPClass.test_retrbinary_restc C sE g } | j j d | j � | j d j | � t j d d � � d S)Nr� r
z
)r� � retrlinesr0 r� r6 r. rX )r r� r r r �test_retrlines? s zTestFTPClass.test_retrlinesc s� t j t j d � � } | j j d | � | j | j j j t � g � | j
d � | j j d | d � f d d � �| j � � d S)Nr �storr r� c s
� j d � S)N)r0 )r{ )�flagr r �<lambda>K s z.TestFTPClass.test_storbinary.<locals>.<lambda>) �io�BytesIOr. r r� �
storbinaryr� r� r� r �seekr� )r �fr )r� r �test_storbinaryD s
%z TestFTPClass.test_storbinaryc C s~ t j t j d d � j d � � } xS d D]K } | j d � | j j d | d | �| j | j j
j t | � � q+ Wd S)
Nz
�
r � �30r r� r- )r r )
r� r� r. rX r r r� r r� r� r� r- �str)r r �rr r r �test_storbinary_restN s
$
z!TestFTPClass.test_storbinary_restc s� t j t j d d � j d � � } | j j d | � | j | j j j
t � g � | j d � | j j d | d � f d d � �| j � � t j
t j d d � � } t j d
t f d d ��! | j t | j j d | � Wd QRXd S)
Nz
r r r� r zstor foor� c s
� j d � S)N)r0 )r{ )r� r r r� \ s z-TestFTPClass.test_storlines.<locals>.<lambda>r
�quietT)r� r� r. rX r r� � storlinesr� r� r� r r r� �StringIOr Zcheck_warnings� BytesWarningr� � TypeError)r r r )r� r �test_storlinesU s $
%
zTestFTPClass.test_storlinesc C s= | j j � | j | j j � t j d � d d � � d S)Nz
r' r5 )r� Znlstr� rv r7 )r r r r � test_nlstd s
zTestFTPClass.test_nlstc sK g � | j j � f d d � � | j d j � � t j d d � � d S)Nc s
� j | � S)N)r0 )r{ )�lr r r� j s z'TestFTPClass.test_dir.<locals>.<lambda>r
z
)r� r� r� r6 rt rX )r r )r r �test_dirh s zTestFTPClass.test_dirc s t � j j � � t � j j d d � � t � j j d d d d d g � � t � j j � � } xq | D]i \ } } � j | t � � j | t � � j | � � j d | � � j d | � � j d | � qm W� f d d � � d d d d � � f d
d � } | d � | d
d d �| d d d �| d d d �| d d d �| d d d �| d d d �| d d d �| d d d �| d d d �| d d d! �| d" d d# �| d$ d d% �� d& � t � j j � � \ } } x! | D] } � j | j � � qW� d' � � j
t t � j j � � � d' � x( � j j � D] } � j d( | � q`Wd S))N�path�/�factsr� �type�perm�uniquec s | � j j _ d S)N)r� r� r )r1 )r r r �set_data{ s z(TestFTPClass.test_mlsd.<locals>.set_datac s� | d k r d n | } | d k r* d n | } | d k rB d n | } | d k rZ d n | } � | � t � j j � � \ } } � j | | � � j | d | � � j | d | � � j | d | � d S)Nr r r �name)�nextr� �mlsdr� )r< r r r r �_namer )r r r r �
test_entry~ s
z*TestFTPClass.test_mlsd.<locals>.test_entryz)type=type;perm=perm;unique=unique; name
z*type=ty=pe;perm=perm;unique=unique; name
zty=pez*type==type;perm=perm;unique=unique; name
z=typez+type=t=y=pe;perm=perm;unique=unique; name
zt=y=pez)type=====;perm=perm;unique=unique; name
z====z*type=type;perm=perm;unique=unique; na me
r zna mez*type=type;perm=perm;unique=unique; name
zname z*type=type;perm=perm;unique=unique; name
z namez,type=type;perm=perm;unique=unique; n am e
zn am ez*type=type;perm=perm;unique=unique; na;me
zna;mez*type=type;perm=perm;unique=unique; ;name
z;namez+type=type;perm=perm;unique=unique; ;name;
z;name;z)type=type;perm=perm;unique=unique; ;;;;
z;;;;z)Type=type;TyPe=perm;UNIQUE=unique; name
r
zunexpected data %s)
rF r� r �assertIsInstancer �dictr� ZassertInr �islowerr� �
StopIteration�fail)r Zlsr r r r r{ r )r r r � test_mlsdm sF %
!
zTestFTPClass.test_mlsdc
C s4 | j j � � | j | j j j d � Wd QRXd S)NrO )r� �makeportr� r� r� r+ )r r r r �
test_makeport� s zTestFTPClass.test_makeportc C sW | j j � \ } } t j | | f d t �} | j � | j | j j j d � d S)NrE Zpasv)
r� �makepasvr( rJ rK r r� r� r� r+ )r r� rO r r r r �
test_makepasv� s
zTestFTPClass.test_makepasvc s � j j � � f d d � } t j d t � �I � _ � j j � j j � j j � � j j d � � j
| � � Wd QRX� j � j j j
d � � j | � � t j d t � �F � _ � j j � j j � j j � � j j d � � j j � Wd QRX� j � j j j
d � � j | � � ya t j d t � �H � _ � j j � j j � j j � � j j d � d � j j _ Wd QRXWn> t j k
r�} z � j t | � d � WYd d } ~ Xn X� j d � t j d � � j � j j j
d � � j | � � d S) Nc sM � j j d k r d Sy � j j d � Wn t t f k
rH d SYn Xd S)NF�noopT)r� rZ r� r� r� r )r r r �is_client_connected� s z=TestFTPClass.test_with_statement.<locals>.is_client_connectedrE r) r� z550 error on quitzException not raisedg�������?)r� r� r� r� rK r� r� r� rO r� r� r� r� r+ r� r, r� r r# �timeZsleep)r r* r� r )r r �test_with_statement� s2
)
z TestFTPClass.test_with_statementc C s� | j j � t j � } y\ | j j | j j | j j d t | f �| j | j j
j � d | � | j j � WnK t k
r� } z+ | j
t
j k r� | j d | � � WYd d } ~ Xn Xd S)N�source_addressr' zcouldn't bind to port %d)r� r� r �find_unused_portr� r� r� rO r r� rZ rU r� r� �
EADDRINUSE�skipTest)r rO �er r r �test_source_address� s
z TestFTPClass.test_source_addressc C s� t j � } t | f | j _ y: | j j d � �! } | j | j � d | � Wd QRXWnK t k
r� } z+ | j t j
k r� | j d | � � WYd d } ~ Xn Xd S)NrF r' zcouldn't bind to port %d) r r. r r� r- � transfercmdr� rU r� r� r/ r0 )r rO rZ r1 r r r �&test_source_address_passive_connection� s %z3TestFTPClass.test_source_address_passive_connectionc C s� | j t j d � d � | j t j d � d � | j t j d � d � | j t j d � d � | j t j t j d � | j t j d � d � | j t j d
� d � d S) Nz257 "/foo/bar"z/foo/barz257 "/foo/bar" createdz257 ""r
z257 "" createdz250 "/foo/bar"z257 "/foo/b""ar"z /foo/b"arz257 "/foo/b""ar" created)r� r� Zparse257r� r� )r r r r �
test_parse257� s zTestFTPClass.test_parse257c C s. | j t j | j j d | j j d � d S)Nr{ rR )r� r� r� r� r� �maxline)r r r r �test_line_too_long s z TestFTPClass.test_line_too_longc C sJ | j j d | j j d � g } | j t j | j j d | j � d S)NzSETLONGRETR %drR r� )r� r� r6 r� r� r� r� r0 )r r� r r r �test_retrlines_too_long s z$TestFTPClass.test_retrlines_too_longc C s@ t j d | j j d � } | j t j | j j d | � d S)N� xrR r� )r� r� r� r6 r� r� r� r )r r r r r �test_storlines_too_long s z$TestFTPClass.test_storlines_too_longN))r# r$ r% r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r r$ r& r( r, r2 r4 r5 r7 r8 r: r r r r r� � sL
: -
r� zIPv6 not enabledc @ 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)�TestIPv6Environmentc C sc t t d f d t j �| _ | j j � t j d t � | _ | j j
| j j | j j � d S)Nr r\ rE )
r} r r( r^ r� r� r� r� rK r� r� r� rO )r r r r r� s
zTestIPv6Environment.setUpc C s | j j � | j j � d S)N)r� r r� r� )r r r r r� s
z TestIPv6Environment.tearDownc C s | j | j j t j � d S)N)r� r� r\ r( r^ )r r r r �test_af! s zTestIPv6Environment.test_afc
C s4 | j j � � | j | j j j d � Wd QRXd S)NZeprt)r� r% r� r� r� r+ )r r r r r& $ s z!TestIPv6Environment.test_makeportc C sW | j j � \ } } t j | | f d t �} | j � | j | j j j d � d S)NrE Zepsv)
r� r' r( rJ rK r r� r� r� r+ )r r� rO r r r r r( ) s
z!TestIPv6Environment.test_makepasvc sD � f d d � } � j j d � | � � j j d � | � d S)Nc sm � f d d � } g � � j j d | � � j t d j � � � t t � � � j d j � � t � d S)Nc s � j | j d � � d S)Nr )r0 r )r1 )r� r r r� 1 s zATestIPv6Environment.test_transfer.<locals>.retr.<locals>.callbackr� r
)r� r� r� r� r6 r. )r� )r )r� r r� 0 s
%z/TestIPv6Environment.test_transfer.<locals>.retrTF)r� r� )r r� r )r r �
test_transfer/ s
z!TestIPv6Environment.test_transferN) r# r$ r% r� r� r< r& r( r= r r r r r; s r; zSSL not availablec @ s" e Z d Z d Z d d � Z d S)�TestTLS_FTPClassMixinzfRepeat TestFTPClass tests starting the TLS layer for both control
and data connections first.
c C st t t d f � | _ | j j � t j d t � | _ | j j | j j | j j
� | j j � | j j � d S)Nr rE )
r� r r� r� r� �FTP_TLSrK r� r� r� rO �auth�prot_p)r r r r r� C s
zTestTLS_FTPClassMixin.setUpN)r# r$ r% r� r� r r r r r> = s r> c @ s� e Z d 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 S)�TestTLS_FTPClassz Specific TLS_FTP class tests.c C sZ t t d f � | _ | j j � t j d t � | _ | j j | j j | j j
� d S)Nr rE ) r� r r� r� r� r? rK r� r� r� rO )r r r r r� Q s
zTestTLS_FTPClass.setUpc C s | j j � | j j � d S)N)r� r r� r� )r r r r r� W s
zTestTLS_FTPClass.tearDownc C sC | j | j j t j � | j j � | j | j j t j � d S)N)�assertNotIsInstancer� rZ r� r� r@ r )r r r r �test_control_connection[ s
z(TestTLS_FTPClass.test_control_connectionc C s� | j j d � � } | j | t j � Wd QRX| j | j j � d � | j j � | j j d � � } | j | t j � Wd QRX| j | j j � d � | j j � | j j d � � } | j | t j � Wd QRX| j | j j � d � d S)NrF z226 transfer complete)
r� r3 rC r� r� r� ZvoidresprA r Zprot_c)r rZ r r r �test_data_connection` s
z%TestTLS_FTPClass.test_data_connectionc C sP | j | j j t j � | j j � | j | j j t j � | j j � d S)N)rC r� rZ r� r� r� r )r r r r r� r s
zTestTLS_FTPClass.test_loginc C s' | j j � | j t | j j � d S)N)r� r@ r� r� )r r r r �test_auth_issued_twicez s
z'TestTLS_FTPClass.test_auth_issued_twicec C sM z6 t j | j _ | j j � | j t | j j � Wd t j | j _ Xd S)N)r� r� r� r� r@ r� r� �PROTOCOL_TLSv1)r r r r �
test_auth_ssl~ s
z TestTLS_FTPClass.test_auth_sslc
C sg | j j � t j t j � } | j t t j d t d | �| j t t j d t d | �| j t t j d t d t d | �t j d | d t
� | _ | j j | j j
| j j � | j | j j t j � | j j � | j | j j j | � | j | j j t j � | j j � | j j d � �- } | j | j | � | j | t j � Wd QRXd S)NZkeyfile�contextr� rE rF )r� r� r� �
SSLContextrG r� r� r� r? r� rK r� r� r� rO rC rZ r� r@ ZassertIsrI r rA r3 )r �ctxrZ r r r � test_context� s$
z TestTLS_FTPClass.test_contextc C sl | j t | j j � | j j d d � | j | j j t j � | j j � | j t | j j j � d S)NZsecureT)
r� r� r� Zcccr� r rZ r� r� r� )r r r r �test_ccc� s
zTestTLS_FTPClass.test_cccc C s� | j j � t j t j � } t j | _ d | _ | j t � t
j d | d t � | _ | j j
| j j | j j � | j t j � � | j j � Wd QRX| j j
| j j | j j � | j j � | j t j � �! | j j d � � } Wd QRXWd QRX| j j � | j j
d | j j � | j j � | j j � | j j
d | j j � | j j � | j j d � � } Wd QRXd S)NTrI rE rF Z localhost)r� r� r� rJ rG Z
CERT_REQUIREDZ verify_modeZcheck_hostnameZload_verify_locations�CAFILEr� r? rK r� r� r� rO r� ZCertificateErrorr@ rA r3 )r rK rZ r r r �test_check_hostname� s,
z$TestTLS_FTPClass.test_check_hostnameN)r# r$ r% r� r� r� rD rE r� rF rH rL rM rO r r r r rB M s rB c @ s| 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 S)� TestTimeoutsc C s� t j � | _ t j t j t j � | _ | j j d � t j | j � | _
t j d | j � | _
| j
j � | j j � | j j � t j j
| _ | j
t j _
d S)Nr� �target)r r� �evtr( r� r� rZ rW r Z bind_portrO r~ r� �
server_threadr� r� �clearr� r� �old_port)r r r r r� � s
zTestTimeouts.setUpc C s | j t j _ | j j � d S)N)rU r� r� rO rS r6 )r r r r r� � s zTestTimeouts.tearDownc C s� | j j � | j j � zi y | j j � \ } } Wn t j k
rM Yn5 X| j d � | j t j � | j j � | j
� Wd | j j
� Xd S)Ns
1 Hola mundo
) rZ rV rR r� rY r( rE ZsendallZshutdownZSHUT_WRr )r r rM r r r r� � s
zTestTimeouts.serverc C sx | j t j � � t j d � z t j t � } Wd t j d � X| j | j j � d � | j
j � | j � d S)Nr )
� assertIsNoner( �getdefaulttimeout�setdefaulttimeoutr� r� r r� rZ �
gettimeoutrR r� r )r �ftpr r r �testTimeoutDefault� s
z TestTimeouts.testTimeoutDefaultc C s{ | j t j � � t j d � z t j t d d �} Wd t j d � X| j | j j � � | j j
� | j � d S)Nr rE ) rV r( rW rX r� r� r rZ rY rR r� r )r rZ r r r �testTimeoutNone� s
z TestTimeouts.testTimeoutNonec C sI t j t d d �} | j | j j � d � | j j � | j � d S)NrE r ) r� r� r r� rZ rY rR r� r )r rZ r r r �testTimeoutValue s
z TestTimeouts.testTimeoutValuec C sS t j � } | j t d d �| j | j j � d � | j j � | j � d S)NrE r )
r� r� r� r r� rZ rY rR r� r )r rZ r r r �testTimeoutConnect s
z TestTimeouts.testTimeoutConnectc C sS t j d d � } | j t � | j | j j � d � | j j � | j � d S)NrE r )
r� r� r� r r� rZ rY rR r� r )r rZ r r r �testTimeoutDifferentOrder s
z&TestTimeouts.testTimeoutDifferentOrderc C sV t j � } d | _ | j t � | j | j j � d � | j j � | j
� d S)Nr ) r� r� rE r� r r� rZ rY rR r� r )r rZ r r r �testTimeoutDirectAccess s
z$TestTimeouts.testTimeoutDirectAccessN) r# r$ r% r� r� r� r[ r\ r] r^ r_ r` r r r r rP � s
rP c C sG t t t t t g } t j � } z t j | � Wd t j | � Xd S)N) r� rP r; r> rB r Zthreading_setupZ run_unittestZthreading_cleanup)Ztests� thread_infor r r � test_main s rb �__main__)2r� r� r r r( r� r� �osr+ r� � ImportErrorZunittestr r Ztestr Z test.supportr r �
import_moduler rK r. rt rv ry r r r& r� r~ r} r r6 �dirname�__file__r� rN r� r� r� r� r� Z IPV6_ENABLEDr; r> rB rP rb r# r r r r �<module> sV
� 1 !!d (� I ( t^