Your IP : 216.73.216.213
�
f f}� �
@ 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 re 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) n 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/ Gd( d) � d) e � Z0 d* d+ � Z1 e2 d, k r�e1 � n 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/python34/lib64/python3.4/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 s3 | j s/ | j j d � | j � d | _ n 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 _ n | 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 � d S)Nr )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 rI | j | j � d | _ n | 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 � d S)Nr )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 r0 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�intr6 �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 d � | j t � | j � d d � \ } } | j d d � } | d } | d } | j d | | | f � | j � \ } } | j | d | �| _
Wd QXd S) Nr rC � �.r@ rB z$227 entering passive mode (%s,%d,%d)r ) r' �bind� getsockname�listen�
settimeoutrJ �replacer �acceptrK r) ) r r= �sockrM rN Zp1Zp2r rL 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& rD r z&200 active data connection establishedr4 )r6 rG r' rI rJ rK r) r )r r= �afrM rN rO r r r �cmd_eprt� s
& zDummyFTPHandler.cmd_eprtc C s� t j t j � �� } | j | j j � d d f � | j d � | j t � | j � d } | j d | � | j � \ } } | j | d | �| _
Wd QXd S)Nr rC r& z+229 entering extended passive mode (|||%d|)r ) r' �AF_INET6rS rT rU rV rJ r rX rK r) )r r= rY rN r rL 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, rG 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)rG r. r )r r= r r r �cmd_setlongretr� s z DummyFTPHandler.cmd_setlongretrN)&r" r# r$ r rK r
r1 r? r! r rP rZ r\ r^ r_ r` ra rb rc rd re rf rg rh ri rj rk rl rm rn ro rp rr rt rv rw ry 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)NrC FrQ )r �Threadr
�asyncore�
dispatcherZ
create_socketr' � SOCK_STREAMrS rU �activeZLock� active_lockrT �hostrN �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)NTrD 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� r5 )r r r r �stop s zDummyFTPServer.stopc C s | j | � | _ d S)N)�handlerr� )r r rL 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 � d S)Nr )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 r0 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 r2 ) r r� r r� r� r� r� r� r� r� r )r Z buffer_sizer� )r r r r � s
zSSLConnection.recvc C s � d S)Nr )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 s0 t j | | | � | j j r, | j � n 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� rK 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 rD ) r| r �serverr� �ftplib�FTPrJ �client�connectr� rN )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 �
exceptionsrz 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� �quitrY )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 )r/ r )r0 )r� r r �callback1 s z.TestFTPClass.test_retrbinary.<locals>.callback�retrr
)r� �
retrbinaryr� r5 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 )r/ r )r0 )r� r r r� 8 s z3TestFTPClass.test_retrbinary_rest.<locals>.callbackr �
� r� r, r
)r r� r� )r� r� r� r5 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� � retrlinesr/ r� r5 r- rW )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)r/ )rz )�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- rW 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 QXd S)
Nz
r r r� r zstor foor� c s
� j d � S)N)r/ )rz )r� r r r� \ s z-TestFTPClass.test_storlines.<locals>.<lambda>r
�quietT)r� r� r- rW 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& r4 )r� Znlstr� ru r6 )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)r/ )rz )�lr r r� j s z'TestFTPClass.test_dir.<locals>.<lambda>r
z
)r� r� r� r5 rs rW )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( t
� q`Wd S))N�path�/�factsr� �type�perm�uniquec s | � j j _ d S)N)r� r� r )r0 )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)rE r� r �assertIsInstancer �dictr� ZassertInr �islowerr� �
StopIteration�failr0 )r Zlsr r r r rz r )r r r � test_mlsdm sF %
!
zTestFTPClass.test_mlsdc
C s3 | j j � � | j | j j j d � Wd QXd S)NrN )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)NrD Zpasv)
r� �makepasvr' rI rJ r r� r� r� r* )r r� rN 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 QX� j � j j j
d � � j | � � t j d t � �F � _ � j j � j j � j j � � j j d � � j j � Wd QX� j � j j j
d � � j | � � y` t j d t � �H � _ � j j � j j � j j � � j j d � d � j j _ Wd QXWn> 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� rY r� r� r� r )r r r �is_client_connected� s z=TestFTPClass.test_with_statement.<locals>.is_client_connectedrD r( r� z550 error on quitzException not raisedg�������?)r� r� r� r� rJ r� r� r� rN 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 � WnN t k
r� } z. | j
t
j k r� | j d | � n � WYd d } ~ Xn Xd S)N�source_addressr& zcouldn't bind to port %d)r� r� r �find_unused_portr� r� r� rN r r� rY rT r� r� �
EADDRINUSE�skipTest)r rN �er r r �test_source_address� s
z TestFTPClass.test_source_addressc C s� t j � } t | f | j _ y9 | j j d � �! } | j | j � d | � Wd QXWnN t k
r� } z. | j t j
k r� | j d | � n � WYd d } ~ Xn Xd S)NrE r& zcouldn't bind to port %d) r r- r r� r, � transfercmdr� rT r� r� r. r/ )r rN rY r0 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)Nrz rQ )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 %drQ r� )r� r� r5 r� r� r� r� r/ )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� xrQ r� )r� r� r� r5 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+ r1 r3 r4 r6 r7 r9 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[ rD )
r| r r' r] r� r� r� r� rJ r� r� r� rN )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 s3 | j j � � | j | j j j d � Wd QXd 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)NrD Zepsv)
r� r& r' rI rJ r r� r� r� r* )r r� rN 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 )r/ r )r0 )r� r r r� 1 s zATestIPv6Environment.test_transfer.<locals>.retr.<locals>.callbackr� r
)r� r� r� r� r5 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 rD )
r� r r� r� r� �FTP_TLSrJ r� r� r� rN �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 rD ) r� r r� r� r� r> rJ r� r� r� rN )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� rY 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 QX| j | j j � d � | j j � | j j d � � } | j | t j � Wd QX| j | j j � d � | j j � | j j d � � } | j | t j � Wd QX| j | j j � d � d S)NrE z226 transfer complete)
r� r2 rB r� r� r� Zvoidrespr@ r Zprot_c)r rY 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)rB r� rY 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 sf | 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 QXd S)NZkeyfile�contextr� rD rE )r� r� r� �
SSLContextrF r� r� r� r> r� rJ r� r� r� rN rB rY r� r? ZassertIsrH r r@ r2 )r �ctxrY 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 rY 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 QX| j j
| j j | j j � | j j � | j t j � � | j j d � � } Wd QXWd QX| j j � | j j
d | j j � | j j � | j j � | j j
d | j j � | j j � | j j d � � } Wd QXd S)NTrH rD rE Z localhost)r� r� r� rI rF Z
CERT_REQUIREDZ verify_modeZcheck_hostnameZload_verify_locations�CAFILEr� r> rJ r� r� r� rN r� ZCertificateErrorr? r@ r2 )r rJ rY r r r �test_check_hostname� s,
z$TestTLS_FTPClass.test_check_hostnameN)r" r# r$ r� r� r� rC rD r� rE rG rK rL rN r r r r rA M s rA 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� rY rV r Z bind_portrN 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)rT r� r� rN rR r5 )r r r r r� � s zTestTimeouts.tearDownc C s� | j j d � | j j � zi y | j j � \ } } Wn t j k
rP Yn5 X| j d � | j t j � | j j � | j
� Wd | j j
� Xd S)NrC s
1 Hola mundo
) rY rU rQ r� rX r' rD ZsendallZshutdownZSHUT_WRr )r r rL 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� rY �
gettimeoutrQ 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 rD ) rU r' rV rW r� r� r rY rX rQ r� r )r rY r r r �testTimeoutNone� s
z TestTimeouts.testTimeoutNonec C sI t j t d d �} | j | j j � d � | j j � | j � d S)NrD r ) r� r� r r� rY rX rQ r� r )r rY 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)NrD r )
r� r� r� r r� rY rX rQ r� r )r rY 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)NrD r )
r� r� r� r r� rY rX rQ r� r )r rY 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� rD r� r r� rY rX rQ r� r )r rY r r r �testTimeoutDirectAccess s
z$TestTimeouts.testTimeoutDirectAccessN) r" r# r$ r� r� r� rZ r[ r\ r] r^ r_ r r r r rO � s
rO c @ s e Z d Z d d � Z d S)�TestNetrcDeprecationc C so t j � �] t j � �J } t j � | d <t d d � j � | j t � � t j
� Wd QXWd QXWd QXd S)N�HOMEz.netrc�w) r Ztemp_cwdZEnvironmentVarGuard�os�getcwd�openr Z assertWarns�DeprecationWarningr� ZNetrc)r �envr r r �test_deprecation" s
z%TestNetrcDeprecation.test_deprecationN)r" r# r$ rh r r r r r` s r` c C sJ t t t t t t g } t j � } z t j | � Wd t j | � Xd S)N)
r� rO r` r: r= rA r Zthreading_setupZ run_unittestZthreading_cleanup)Ztests� thread_infor r r � test_main+ s rj �__main__)3r� r� r~ r r' r� r� rc r* r� � ImportErrorZunittestr r Ztestr Z test.supportr r �
import_moduler rJ r- rs ru rx r r r% r r} r| r r5 �dirname�__file__r� rM r� r� r� r� r� Z IPV6_ENABLEDr: r= rA rO r` rj r" r r r r �<module> sX
� 1 !!d (� I ( t^