Your IP : 216.73.216.213
���]T� �
@ 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 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( � Z/ e0 d) k r�e/ � d S)*� 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 �0/opt/alt/python35/lib64/python3.5/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
�socket�
setsockopt�
SOL_SOCKET� 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 r4 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�intr: �tupler( �create_connection�TIMEOUT� dtp_handlerr- r )r rA �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 � �.rD rF z$227 entering passive mode (%s,%d,%d)r ) r( �bind� getsockname�listen�
settimeoutrN �replacer �acceptrO r- ) r rA �sockrQ rR Zp1Zp2r rP 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' rH r z&200 active data connection establishedr8 )r: rK r( rM rN rO r- r )r rA �afrQ rR rS 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_INET6rW rX rY rZ rN r r\ rO r- )r rA r] rR r rP r r r �cmd_epsv� s
zDummyFTPHandler.cmd_epsvc C s | j | � d S)N)r )r rA r r r �cmd_echo� s zDummyFTPHandler.cmd_echoc C s | j d � d S)Nz 200 noop ok)r )r rA r r r �cmd_noop� s zDummyFTPHandler.cmd_noopc C s | j d � d S)Nz331 username ok)r )r rA r r r �cmd_user� s zDummyFTPHandler.cmd_userc C s | j d � d S)Nz230 password ok)r )r rA r r r �cmd_pass� s zDummyFTPHandler.cmd_passc C s | j d � d S)Nz 230 acct ok)r )r rA r r r �cmd_acct� s zDummyFTPHandler.cmd_acctc C s | j d � d S)Nz 350 rnfr ok)r )r rA r r r �cmd_rnfr� s zDummyFTPHandler.cmd_rnfrc C s | j d � d S)Nz 250 rnto ok)r )r rA r r r �cmd_rnto� s zDummyFTPHandler.cmd_rntoc C s | j d � d S)Nz 250 dele ok)r )r rA r r r �cmd_dele� s zDummyFTPHandler.cmd_delec C s | j d � d S)Nz
250 cwd ok)r )r rA r r r �cmd_cwd� s zDummyFTPHandler.cmd_cwdc C s | j d � d S)Nz250 1000)r )r rA r r r �cmd_size� s zDummyFTPHandler.cmd_sizec C s | j d | � d S)Nz257 "%s")r )r rA r r r �cmd_mkd� s zDummyFTPHandler.cmd_mkdc C s | j d � d S)Nz
250 rmd ok)r )r rA r r r �cmd_rmd� s zDummyFTPHandler.cmd_rmdc C s | j d � d S)Nz 257 "pwd ok")r )r rA r r r �cmd_pwd� s zDummyFTPHandler.cmd_pwdc C s | j d � d S)Nz 200 type ok)r )r rA r r r �cmd_type� s zDummyFTPHandler.cmd_typec C s | j d � | j � d S)Nz 221 quit ok)r r )r rA r r r �cmd_quit� s
zDummyFTPHandler.cmd_quitc C s | j d � d S)Nz 226 abor ok)r )r rA r r r �cmd_abor� s zDummyFTPHandler.cmd_aborc C s | j d � d S)Nz 125 stor ok)r )r rA r r r �cmd_stor� s zDummyFTPHandler.cmd_storc C s | | _ | j d � d S)Nz 350 rest ok)r0 r )r rA 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 r0 rK r- r2 r )r rA �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 rA 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 rA r r r �cmd_nlst� s
zDummyFTPHandler.cmd_nlstc C s | j d � d S)Nz 200 opts ok)r )r rA 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 rA r r r �cmd_mlsd� s
zDummyFTPHandler.cmd_mlsdc C s$ d t | � | _ | j d � d S)N�xz125 setlongretr ok)rK r2 r )r rA r r r �cmd_setlongretr� s z DummyFTPHandler.cmd_setlongretrN)&r# r$ r% r rO r
r5 rC r" r rT r^ r` rb rc rd re rf rg rh ri rj rk rl rm rn ro rp rq rr rs rt rv rx rz r{ r} 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)NrG FrU )r �Threadr
�asyncore�
dispatcherZ
create_socketr( � SOCK_STREAMrW rY �active�Lock� active_lockrX �hostrR �handler_instance)r �addressr_ 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)NTrH 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� r9 )r r r r �stop s zDummyFTPServer.stopc C s | j | � | _ d S)N)�handlerr� )r r rP 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 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)�
SSLConnectionFc 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 r4 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 r6 ) 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% 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 d � Z d S)�DummyTLS_DTPHandlerc 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� � s r� c @ sR e Z d Z e Z d d � Z d d � Z d d � Z d d � Z d d
� Z d S) �DummyTLS_FTPHandlerc 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)Nz234 AUTH TLS successful)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)Nz200 PBSZ=0 successful.)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)N�Cz200 Protection set to ClearF�Pz 200 Protection set to PrivateTz(502 Unrecognized PROT type (use C or P).)�upperr r� )r r? rA r r r �cmd_prot� s
z DummyTLS_FTPHandler.cmd_protN)
r# r$ r% r� rO 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 rH ) r� r �serverr� �ftplib�FTPrN �client�connectr� rR )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� �deleter� 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� �quitr] )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 )r3 r )r4 )r� r r �callback1 s z.TestFTPClass.test_retrbinary.<locals>.callback�retrr
)r� �
retrbinaryr� r9 r1 )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 )r3 r )r4 )r� r r r� 8 s z3TestFTPClass.test_retrbinary_rest.<locals>.callbackr �
� r� r0 r
)r r� r� )r� r� r� r9 r1 )r r� r0 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� � retrlinesr3 r� r9 r1 r[ )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)r3 )r~ )�flagr r �<lambda>K s z.TestFTPClass.test_storbinary.<locals>.<lambda>) �io�BytesIOr1 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 r0 )r
r )
r r r1 r[ r r r� r r� r� r� r0 �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)r3 )r~ )r r r r \ s z-TestFTPClass.test_storlines.<locals>.<lambda>r
�quietT)r r r1 r[ 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' r8 )r� Znlstr� ry r: )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)r3 )r~ )�lr r r j s z'TestFTPClass.test_dir.<locals>.<lambda>r
z
)r� r� r� r9 rw r[ )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 )r4 )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)
rI 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)NrR )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)NrH Zpasv)
r� �makepasvr( rM rN r r� r� r� r. )r r� rR 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� r] r� r� r� r )r r r �is_client_connected� s z=TestFTPClass.test_with_statement.<locals>.is_client_connectedrH r. r� z550 error on quitzException not raisedg�������?)r� r� r� r� rN r� r� r� rR r� r� r� r� r. r� r/ r� r r( �time�sleep)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� rR r r� r] rX r� r� �
EADDRINUSE�skipTest)r rR �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)NrI r' zcouldn't bind to port %d) r r4 r r� r3 � transfercmdr� rX r� r� r5 r6 )r rR r] r7 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~ rU )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 %drU r� )r� r� r< r� r� r� r� r3 )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� xrU r )r r r� r< 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- r2 r8 r: r; r= r> 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_ rH )
r� r r( ra r� r� r� r� rN r� r� r� rR )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( ra )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)NrH Zepsv)
r� r, r( rM rN r r� r� r� r. )r r� rR 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 )r3 r )r4 )r� r r r� 1 s zATestIPv6Environment.test_transfer.<locals>.retr.<locals>.callbackr� r
)r� r� r� r� r9 r1 )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� rB r+ r- rC r r r r rA s rA zSSL not availablec @ s e Z d Z d d � Z d S)�TestTLS_FTPClassMixinc 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 rH )
r� r r� r� r� �FTP_TLSrN r� r� r� rR �auth�prot_p)r r r r r� C s
zTestTLS_FTPClassMixin.setUpN)r# r$ r% r� r r r r rD = s rD 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 S)�TestTLS_FTPClassc C sZ t t d f � | _ | j j � t j d t � | _ | j j | j j | j j
� d S)Nr rH ) r� r r� r� r� rE rN r� r� r� rR )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� r] r� r� rF 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)NrI z226 transfer complete)
r� r9 rI r� r� r� ZvoidresprG r$ Zprot_c)r r] 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)rI r� r] 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� rF 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� rF 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� rH rI )r� r� r� �
SSLContextrM r� r� r� rE r� rN r� r� r� rR rI r] r� rF ZassertIsrO r$ rG r9 )r �ctxr] 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� �cccr� r$ r] 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)NTrO rH rI � localhost)r� r� r� rP rM Z
CERT_REQUIREDZ verify_modeZcheck_hostnameZload_verify_locations�CAFILEr� rE rN r� r� r� rR r� ZCertificateErrorrF rG r9 )r rQ r] r r r �test_check_hostname� s,
z$TestTLS_FTPClass.test_check_hostnameN)
r# r$ r% r� r� rJ rK r� rL rN rR rT rW r r r r rH M s rH 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� r] rZ r Z bind_portrR 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)r] r� r� rR r[ r9 )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
) r] rY rZ r� r\ r( rH �sendall�shutdown�SHUT_WRr )r r rP 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� r] �
gettimeoutrZ 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
rH ) ra r( rb rc r� r� r r] rd rZ r� r )r re r r r �testTimeoutNone� s
z TestTimeouts.testTimeoutNonec C sI t j t d d �} | j | j j � d � | j j � | j � d S)NrH r
) r� r� r r� r] rd rZ r� r )r re 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)NrH r
)
r� r� r� r r� r] rd rZ r� r )r re 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)NrH r
)
r� r� r� r r� r] rd rZ r� r )r re 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� rH r� r r� r] rd rZ r� r )r re r r r �testTimeoutDirectAccess s
z$TestTimeouts.testTimeoutDirectAccessN) r# r$ r% r� r� r� rf rg rh ri rj rk r r r r rX � s
rX c C sG t t t t t g } t j � } z t j | � Wd t j | � Xd S)N) r� rX rA rD rH r Zthreading_setupZ run_unittestZthreading_cleanup)Ztests� thread_infor r r � test_main s rm �__main__)1r� r� r r( r r� �osr0 r� � ImportErrorZunittestr r Ztestr Z test.supportr r �
import_moduler rN r1 rw ry r| r r r& r� r� r� r r9 �dirname�__file__r� rV r� r� r� r� r� Z IPV6_ENABLEDrA rD rH rX rm r# r r r r �<module> sT
� 1 !!d (� I ( t^