Your IP : 216.73.216.52
3
� f�� �
@ s
d Z ddlZddlZddlZddlZddlZddlZddlZddlZy ddl Z W n e
k
rh dZ Y nX ddl m Z m
Z
ddlmZ ddlmZmZ ejd�ZdZdd Zd
Zd
Zd ZG d d
� d
ej�ZG dd� dej�Z G dd� dej ej �Z e dk �rnej j!ej j"e#�d�Z$ej j!ej j"e#�d�Z%G dd� dej �Z&G dd� de&e�Z'G dd� de&e �Z(G dd� de �Z)G d d � d e �Z*e
ej+d �G d d � d e ��Z,e
e d!�G d"d#� d#e*��Z-e
e d!�G d$d%� d%e ��Z.G d&d'� d'e �Z/G d(d)� d)e �Z0d*d+� Z1e2d,k�re1� dS )-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 s@ e Zd ZdZdd� Zdd� Zdd� Z� fdd �Zd
d � Z� Z S ) �DummyDTPHandlerFc C s t jj| |� || _d| j_d S )N� )�asynchat�
async_chat�__init__� baseclass�last_received_data)�self�connr � r �5/opt/alt/python36/lib64/python3.6/test/test_ftplib.pyr
2 s zDummyDTPHandler.__init__c C s | j j| jd�jd�7 _d S )Ni �ascii)r r �recv�decode)r r r r � handle_read7 s zDummyDTPHandler.handle_readc C s$ | j s | jjd� | j� d| _ d S )Nz226 transfer completeT)�dtp_conn_closedr �push�close)r r r r � handle_close: s z DummyDTPHandler.handle_closec sB | j jd k r | j j}d | j _|s(| j� S tt| �j|jd�� 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" �
__classcell__r r )r r r / s r c @ s e Zd ZeZdd� Zdd� Zdd� Zdd� Zd d
� Z d d � Z
d
d� Z dd� Z dd� Z
dd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Z d1d2� Z d3d4� Z d5d6� Z d7d8� Z d9d:� Z!d;d<� Z"d=d>� Z#d?d@� Z$dAdB� Z%dCS )D�DummyFTPHandlerc C sp t jj| |� | jjtjtjd� | jd� g | _d | _ d | _
d| _ d| _ d | _
d | _t| _| jd� d| _d S )N� s
r
z 220 welcomez252.253.254.255)r r r
�socketZ
setsockoptZ
SOL_SOCKETZ SO_OOBINLINEZset_terminator� in_buffer�dtp�last_received_cmdr �
next_responser �rest� RETR_DATA�next_retr_datar �fake_pasv_server_ip)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_dataf s z%DummyFTPHandler.collect_incoming_datac C s� dj | j�jd�}g | _| jr0| j| j� d| _|jd�d j� }|| _|jd�}|d krl||d d � }nd}t | d| �r�t
| d| �}||� n| jd| � 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_terminatori s
z DummyFTPHandler.found_terminatorc C s t �d S )N)r! )r r r r r" | s z DummyFTPHandler.handle_errorc C s t jj| |jd�d � d S )Nr s
)r r r r )r r3 r r r r s zDummyFTPHandler.pushc C sl t tt|jd���}dt|d d� � }|d d |d }tj||ftd�}| j|| d�| _ | j
d� d S ) N�,z %d.%d.%d.%d� � � )�timeout)r z&200 active data connection established) �list�map�intr9 �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 df� |j� |jt� |j� d }| j}|jdd�}|d }|d }| jd|||f � |j � \}}| j
|| d�| _ W d Q R X d S )Nr r( �.rC rE z$227 entering passive mode (%s,%d,%d))r ) r) �bind� getsockname�listen�
settimeoutrM r1 �replacer �acceptrN r+ ) r r@ �sockrQ rP Zp1Zp2r rO r r r �cmd_pasv� s
zDummyFTPHandler.cmd_pasvc C sT |j |d �dd� \}}}t|�}tj||ftd�}| j|| d�| _| jd� d S )Nr r( )rG )r z&200 active data connection establishedr7 )r9 rJ r) rL rM rN r+ r )r r@ �afrP rQ rR r r r �cmd_eprt� s
zDummyFTPHandler.cmd_eprtc C s| t j t j��f}|j| j j� d df� |j� |jt� |j� d }| jd| � |j� \}}| j || d�| _
W d Q R X d S )Nr r( z+229 entering extended passive mode (|||%d|))r ) r) �AF_INET6rU rV rW rX rM r rZ rN r+ )r r@ r[ rQ r rO 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 || _ | jd� d S )Nz 350 rest ok)r. r )r r@ r r r �cmd_rest� s zDummyFTPHandler.cmd_restc C sN | j d� | jd k r t| j�}nd}| jj | j|d � � | jj� d | _d S )Nz 125 retr okr )r r. rJ r+ r0 r )r r@ �offsetr r r �cmd_retr� s
zDummyFTPHandler.cmd_retrc C s$ | j d� | jj t� | jj� 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� | jj t� | jj� 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� | jj t� | jj� d S )Nz 125 mlsd ok)r r+ � MLSD_DATAr )r r@ r r r �cmd_mlsd� s
zDummyFTPHandler.cmd_mlsdc C s dt |� | _| jd� d S )N�xz125 setlongretr ok)rJ r0 r )r r@ r r r �cmd_setlongretr� s z DummyFTPHandler.cmd_setlongretrN)&r# r$ r% r rN r
r4 rB r" r rS r\ r^ r` ra rb rc rd re rf rg rh ri rj rk rl rm rn ro rp rq rr rt rv rx ry r{ r} r r r r r' O sD
r' c @ sZ e Zd ZeZejfdd�Zdd� Zdd� Z dd� Z
d d
� Z d d � Z e Z
d
d� Zdd� ZdS )�DummyFTPServerc C sn t jj| � tjj| � | j|tj� | j|� | j d� d| _
t j � | _ | jj
� d d� \| _| _d | _d S )NrF F� )r �Threadr
�asyncore�
dispatcherZ
create_socketr) � SOCK_STREAMrU rW �activeZLock� active_lockrV �hostrQ �handler_instance)r Zaddressr] r r r r
s
zDummyFTPServer.__init__c C s$ t j� | _t jj| � | jj� d S )N)r �Event�_DummyFTPServer__flagr� �start�wait)r r r r r� s
zDummyFTPServer.startc C sT d| _ | jj� x2| j rBtjrB| jj� tjddd� | jj� qW tj dd� d S )NTg�������?r( )rG �count)�
ignore_all)
r� r� �setr� Z
socket_mapr� �acquireZloop�release� close_all)r r r r �run s
zDummyFTPServer.runc C s d| _ | j� d S )NF)r� r8 )r r r r �stop s zDummyFTPServer.stopc C s | j |�| _d S )N)�handlerr� )r r rO 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 dS )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 Zd ZdZdZdZdd� Zdd� Z� fdd�Z� fd d
�Z � fd d �Z
� fd
d�Z � fdd�Z dd� Z
� fdd�Z� ZS )�
SSLConnectionz3An asyncore.dispatcher subclass supporting TLS/SSL.Fc C sB t j� }|jt� |j| jdddd�}| j� | j|� d| _d S )NFT)Zsuppress_ragged_eofsZ server_sideZdo_handshake_on_connect) �ssl�
SSLContextZload_cert_chain�CERTFILEZ wrap_socketr) Z del_channelZ
set_socket�_ssl_accepting)r �contextr) r r r �secure_connection> s
z SSLConnection.secure_connectionc C s� y| j j� W n� tjk
rf } z:|jd tjtjfkr<d S |jd tjkrT| j� S � W Y d d }~X n@ t k
r� } z |jd t
j kr�| j� S W Y d d }~X nX d| _ 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_handshakeI s
z SSLConnection._do_ssl_handshakec s� d| _ y| jj� | _W n^ tjk
rT } z |jd tjtjfkrDd S W Y d d }~X n" tk
rt } zW Y d d }~X nX d| _ t | dd�dkr�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_shutdownY s
z SSLConnection._do_ssl_shutdownc s2 | j r| j� n | jr | j� ntt| �j� d S )N)r� r� r� r� r r� �handle_read_event)r )r r r r� m s
z SSLConnection.handle_read_eventc s2 | j r| j� n | jr | j� ntt| �j� d S )N)r� r� r� r� r r� �handle_write_event)r )r r r r� u s
z SSLConnection.handle_write_eventc s^ yt t| �j|�S tjk
rX } z*|jd tjtjtjtj fkrFdS � W Y d d }~X nX d S )Nr )
r r� �sendr� r� r� r� �SSL_ERROR_ZERO_RETURNr� r� )r r3 r� )r r r r� } s
zSSLConnection.sendc sx yt t| �j|�S tjk
rr } zD|jd tjtjfkr>dS |jd tjtj fkr`| j
� dS � W Y d d }~X nX d S )Nr r5 ) r r� r r� r� r� r� r� r� r� r )r � 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 s6 t | jtj�r$| jjd k r$| j� ntt| �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 r� 8 s
r� c @ s e Zd ZdZdd� ZdS )�DummyTLS_DTPHandlerz.A DummyDTPHandler subclass supporting TLS/SSL.c C s" t j| ||� | jjr | 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 @ s< e Zd ZdZeZdd� Zdd� Zdd� Zdd � 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� dS )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� dS )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 sJ |j � }|dkr"| jd� d| _n$|dkr<| jd� d| _n
| jd� dS ) 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� rN r
r� r� r� r� r r r r r� � s r� c @ s e Zd ZeZdS )�DummyTLS_FTPServerN)r# r$ r% r� r� r r r r r� � s r� c @ sL e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zd1d2� Zd3d4� Z d5d6� Z d7d8� Z d9d:� Z d;d<� Z d=d>� Z!d?d@� Z"dAdB� Z#dCdD� Z$dEdF� Z%dGdH� Z&dIdJ� Z'dKdL� Z(dMdN� Z)dOdP� Z*dQS )R� TestFTPClassc C s@ t tdf�| _| jj� tjtd�| _| jj| jj | jj
� d S )Nr )rG ) r~ r �serverr� �ftplib�FTPrM �client�connectr� rQ )r r r r �setUp� s
zTestFTPClass.setUpc C s* | j j� | jj� d | _tjdd� d S )NT)r� )r� r r� r� r� r� )r r r r �tearDown� s
zTestFTPClass.tearDownc C s$ | j t|�t|�� | j ||� d S )N)� assertEqual�len)r �receivedZexpectedr r r �
check_data� s zTestFTPClass.check_datac C s | j | jj� d� d S )Nz 220 welcome)r� r� Z
getwelcome)r r r r �test_getwelcome� s z TestFTPClass.test_getwelcomec C sL | j | jjd�td�� | j | jjd�td�� | j | jjd�td�� 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| jjd� | j t| jjd� | j t| jjd� | j tj| jjd� | j tj| jjd� | j tj| jjd� | j tj| jjd� | j tj| jjd� 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 sR t jt jt jt jt jttf}x0|D ](}y |d��W q" t jk
rH Y q"X q"W d 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 sF | j | jj� | jjd� | j | jj� | jjd� | j| jj� d S )NTF)�
assertTruer� Z
passiveserver�set_pasv� assertFalse)r r r r �
test_set_pasv s
zTestFTPClass.test_set_pasvc C sD | j jd� | j jd� | jtj| j jd� | jtj| j jd� 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 jd� d S )NZpasswd)r� Zacct)r r r r � test_acct s zTestFTPClass.test_acctc C s2 | j jdd� d| jj_| jtj| j jdd� 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 jd� d| jj_| jtj| j jd� d S )Nr� Z199)r� �deleter� r� r- r� r� r� )r r r r � test_delete s
zTestFTPClass.test_deletec C s | j jd� d S )Nr� )r� �size)r r r r � test_size s zTestFTPClass.test_sizec C s | j jd�}| j|d� d S )Nz/foo)r� Zmkdr� )r �dirr r r �test_mkd# s zTestFTPClass.test_mkdc C s | j jd� d S )Nr� )r� Zrmd)r r r r �test_rmd' s zTestFTPClass.test_rmdc C s | j jd�}| 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 s& | j | jj� d� | j | jjd � d S )Nz 221 quit ok)r� r� �quitr[ )r r r r � test_quit2 s zTestFTPClass.test_quitc C s | j j� d S )N)r� �abort)r r r r �
test_abort7 s zTestFTPClass.test_abortc s4 � fdd�}g � | j jd|� | jdj� �t� d S )Nc s � j | jd�� d S )Nr )r2 r )r3 )r� r r �callback; s z.TestFTPClass.test_retrbinary.<locals>.callback�retrr
)r� �
retrbinaryr� r8 r/ )r r� r )r� r �test_retrbinary: s z TestFTPClass.test_retrbinaryc sN � fdd�}x<d D ]4}g � | j jd||d� | jdj� �t|d � � qW d S )
Nc s � j | jd�� d S )Nr )r2 r )r3 )r� r r r� B s z3TestFTPClass.test_retrbinary_rest.<locals>.callbackr �
� r� )r. r
)r r� r� )r� r� r� r8 r/ )r r� r. r )r� r �test_retrbinary_restA s
z!TestFTPClass.test_retrbinary_restc C s2 g }| j jd|j� | jdj|�tjdd�� d S )Nr� r
z
)r� � retrlinesr2 r� r8 r/ rY )r r� r r r �test_retrlinesI s zTestFTPClass.test_retrlinesc sf t jtjd��}| jjd|� | j| jjj t� g � |j
d� | jjd|� fdd�d� | j � � d S )Nr �storr c s
� j d �S )N)r2 )r| )�flagr r �<lambda>U s z.TestFTPClass.test_storbinary.<locals>.<lambda>)r� ) �io�BytesIOr/ r r� �
storbinaryr� r� r� r �seekr� )r �fr )r r �test_storbinaryN s
z TestFTPClass.test_storbinaryc C s\ t jtjdd�jd��}x>d D ]6}|jd� | jjd||d� | j| j j
j t |�� q W d S )
Nz
�
r � �30r r )r. )r
r )
r r r/ rY r r r� r r� r� r� r. �str)r r �rr r r �test_storbinary_restX s
z!TestFTPClass.test_storbinary_restc s� t jtjdd�jd��}| jjd|� | j| jj j
t� g � |j d� | jjd|� fdd�d � | j � � t j
tjdd��}tjd
tfd d �� | jt| jjd|� W d Q R X d S )
Nz
r r r r zstor fooc s
� j d �S )N)r2 )r| )r r r r f s z-TestFTPClass.test_storlines.<locals>.<lambda>)r� r
T)�quiet)r r r/ rY r r� � storlinesr� r� r� r r r� �StringIOr Zcheck_warnings� BytesWarningr� � TypeError)r r r )r r �test_storlines_ s
zTestFTPClass.test_storlinesc C s. | j j� | j| j j� tjd�d d� � d S )Nz
r( r7 )r� Znlstr� rw r9 )r r r r � test_nlstn s
zTestFTPClass.test_nlstc s6 g � | j j� fdd�� | jdj� �tjdd�� d S )Nc s
� j | �S )N)r2 )r| )�lr r r t s z'TestFTPClass.test_dir.<locals>.<lambda>r
z
)r� r� r� r8 ru rY )r r )r r �test_dirr s zTestFTPClass.test_dirc s� t � jj� � t � jjdd�� t � jjdddgd�� t � jj� �}xV|D ]N\}}� j|t� � j|t� � j|� � jd|� � jd|� � jd|� qLW � fdd ��d*� �fd
d � }|d � |d
dd� |ddd� |ddd� |ddd� |ddd� |ddd� |dd d� |d d d� |d d d� |d!d"d� |d#d$d� |d%d&d� �d'� t� jj� �\}}x |D ]}� j|j � � �qrW �d(� � j
t t� jj� � �d(� x"� jj� D ]}� j d)| � �q�W d S )+N�/)�pathr� �type)r �facts�perm�uniquec s | � j j_d S )N)r� r� r )r3 )r r r �set_data� s z(TestFTPClass.test_mlsd.<locals>.set_datac s� |d kr dn|}|d kr dn|}|d kr,dn|}|d kr<dn|}�| � t � jj� �\}}� 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=pe)r z*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
zna me)r z*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)NNNN)
rH r� r! �assertIsInstancer �dictr� ZassertInr �islowerr� �
StopIteration�fail)r Zlsr r r# r" r| r )r r r � test_mlsdw sF
zTestFTPClass.test_mlsdc
C s, | j j� � | j| jjjd� W d Q R X d S )NrQ )r� �makeportr� r� r� r, )r r r r �
test_makeport� s zTestFTPClass.test_makeportc C s> | j j� \}}tj||ftd�}|j� | j| jjj d� d S )N)rG Zpasv)
r� �makepasvr) rL rM r r� r� r� r, )r r� rQ r r r r �
test_makepasv� s zTestFTPClass.test_makepasvc C sN d| j _| j j� \}}| j|| jjj� tj| j j j
� d |ft d�j � dS )z/Test the opt-in to the old vulnerable behavior.Tr )rG N)
r� � trust_server_pasv_ipv4_addressr, r� r� r� r1 r) rL r[ Z getpeernamerM r )r Zbad_hostrQ r r r �*test_makepasv_issue43285_security_disabled� s z7TestFTPClass.test_makepasv_issue43285_security_disabledc C sH | j | jj� | jj� \}}| j|| jjj� tj ||ft
d�j � d S )N)rG ) r� r� r. r, ZassertNotEqualr� r� r1 r) rL rM r )r Z trusted_hostrQ r r r �1test_makepasv_issue43285_security_enabled_default� s
z>TestFTPClass.test_makepasv_issue43285_security_enabled_defaultc s� � j j� � fdd�}tjtd��6� _ � j j� jj� jj� � j j d� � j
|� � W d Q R X � j � jj j
d� � j|� � tjtd��4� _ � j j� jj� jj� � j j d� � j j� W d Q R X � j � jj j
d� � j|� � yJtjtd��4� _ � j j� jj� jj� � j j d� d� jj _W d Q R X W n6 tjk
�r` } z� j t|�d� W Y d d }~X n X � jd� tjd� � j � jj j
d� � j|� � d S ) Nc s> � j jd krdS y� j jd� W n ttfk
r8 dS X dS )NF�noopT)r� r[ r� r� r� r )r r r �is_client_connected� s z=TestFTPClass.test_with_statement.<locals>.is_client_connected)rG r1 r� z550 error on quitzException not raisedg�������?)r� r� r� r� rM r� r� r� rQ r� r� r� r� r, r� r- r� r r( �timeZsleep)r r2 r� r )r r �test_with_statement� s2
"
z TestFTPClass.test_with_statementc C s� | j j� tj� }yD| j j| jj| jjt|fd� | j | j j
j � d |� | j j� W n> t k
r� } z"|j
t
jkr�| jd| � � W Y d d }~X nX d S )N)�source_addressr( zcouldn't bind to port %d)r� r� r �find_unused_portr� r� r� rQ r r� r[ rV r� r� �
EADDRINUSE�skipTest)r rQ �er r r �test_source_address� s
z TestFTPClass.test_source_addressc C s� t j� }t|f| j_y0| jjd��}| j|j� d |� W d Q R X W n> tk
r� } z"|j t j
krp| j d| � � W Y d d }~X nX d S )NrH r( zcouldn't bind to port %d) r r6 r r� r5 � transfercmdr� rV r� r� r7 r8 )r rQ r[ r9 r r r �&test_source_address_passive_connection s " z3TestFTPClass.test_source_address_passive_connectionc C s� | j tjd�d� | j tjd�d� | j tjd�d� | j tjd�d� | jtjtjd� | j tjd�d � | j tjd
�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� �parse257r� r� )r r r r �
test_parse257 s zTestFTPClass.test_parse257c C s$ | j tj| jjd| jj d � d S )Nr| r )r� r� r� r� r� �maxline)r r r r �test_line_too_long# s z TestFTPClass.test_line_too_longc C s8 | j jd| j jd � g }| jtj| j jd|j� d S )NzSETLONGRETR %dr r� )r� r� r? r� r� r� r� r2 )r r� r r r �test_retrlines_too_long' s z$TestFTPClass.test_retrlines_too_longc C s0 t jd| jj d �}| jtj| jjd|� d S )N� xr 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- r/ r0 r4 r: r< r> r@ rA rC r r r r r� � sP
: -
r� zIPv6 not enabledc @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )�TestIPv6Environmentc C sF t tdftjd�| _| jj� tjtd�| _ | j j
| jj | jj � d S )Nr )r] )rG )
r~ r r) r_ r� r� r� r� rM r� r� r� rQ )r r r r r� 5 s
zTestIPv6Environment.setUpc C s* | j j� | jj� d | _tjdd� d S )NT)r� )r� r r� r� r� r� )r r r r r� ; s
z TestIPv6Environment.tearDownc C s | j | jjtj� d S )N)r� r� r] r) r_ )r r r r �test_afB s zTestIPv6Environment.test_afc
C s, | j j� � | j| jjjd� W d Q R X d S )NZeprt)r� r* r� r� r� r, )r r r r r+ E s z!TestIPv6Environment.test_makeportc C s> | j j� \}}tj||ftd�}|j� | j| jjj d� d S )N)rG Zepsv)
r� r, r) rL rM r r� r� r� r, )r r� rQ r r r r r- J s z!TestIPv6Environment.test_makepasvc s4 � fdd�}� j jd� |� � j jd� |� d S )Nc sN � fdd�} g � �j jd| � �jtdj� ��tt�� �jdj� �t� d S )Nc s � j | jd�� d S )Nr )r2 r )r3 )r� r r r� R s zATestIPv6Environment.test_transfer.<locals>.retr.<locals>.callbackr� r
)r� r� r� r� r8 r/ )r� )r )r� r r� Q s
z/TestIPv6Environment.test_transfer.<locals>.retrTF)r� r� )r r� r )r r �
test_transferP s
z!TestIPv6Environment.test_transferN) r# r$ r% r� r� rE r+ r- rF r r r r rD 2 s rD zSSL not availablec @ s e Zd ZdZdd� ZdS )�TestTLS_FTPClassMixinzfRepeat TestFTPClass tests starting the TLS layer for both control
and data connections first.
c C sT t tdf�| _| jj� tjtd�| _| jj| jj | jj
� | jj � | jj � d S )Nr )rG )
r� r r� r� r� �FTP_TLSrM r� r� r� rQ �auth�prot_p)r r r r r� d s
zTestTLS_FTPClassMixin.setUpN)r# r$ r% r� r� r r r r rG ^ s rG c @ s` e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dS )�TestTLS_FTPClassz Specific TLS_FTP class tests.c C s@ t tdf�| _| jj� tjtd�| _| jj| jj | jj
� d S )Nr )rG ) r� r r� r� r� rH rM r� r� r� rQ )r r r r r� r s
zTestTLS_FTPClass.setUpc C s* | j j� | jj� d | _tjdd� d S )NT)r� )r� r r� r� r� r� )r r r r r� x s
zTestTLS_FTPClass.tearDownc C s2 | j | jjtj� | jj� | j| jjtj� d S )N)�assertNotIsInstancer� r[ r� r� rI r$ )r r r r �test_control_connection s
z(TestTLS_FTPClass.test_control_connectionc C s | j jd��,}| j|tj� | j|jd�tjd�� W d Q R X | j| j j � d� | j j
� | j jd��,}| j |tj� | j|jd�tjd�� W d Q R X | j| j j � d� | j j � | j jd��,}| j|tj� | j|jd�tjd�� W d Q R X | j| j j � d� d S )NrH i r z226 transfer complete)
r� r; rL r� r� r� r ru r ZvoidresprJ r$ Zprot_c)r r[ r r r �test_data_connection� s "
"
"z%TestTLS_FTPClass.test_data_connectionc C s< | j | jjtj� | jj� | j| jjtj� | jj� d S )N)rL r� r[ r� r� r� r$ )r r r r r� � s
zTestTLS_FTPClass.test_loginc C s | j j� | jt| j j� d S )N)r� rI r� r� )r r r r �test_auth_issued_twice� s
z'TestTLS_FTPClass.test_auth_issued_twicec C s: z(t j| j_| jj� | jt| jj� W d t j| j_X d S )N)r� ZPROTOCOL_SSLv23r� Z ssl_versionrI r� r� � PROTOCOL_TLS)r r r r �
test_auth_ssl� s
z TestTLS_FTPClass.test_auth_sslc
C s� | j j� tjtj�}| jttjt |d� | jttjt |d� | jttjt t |d� tj|t
d�| _ | j j | j j
| j j� | j| j jtj� | j j� | j| j jj|� | j| j jtj� | j j� | j jd��"}| j|j|� | j|tj� W d Q R X d S )N)�keyfiler� )�certfiler� )rS rR r� )r� rG rH )r� r� r� r� rP r� r� r� rH r� rM r� r� r� rQ rL r[ r� rI ZassertIsr� r$ rJ r; )r �ctxr[ r r r � test_context� s$
z TestTLS_FTPClass.test_contextc C sP | j t| jj� | jjdd� | j| jjtj� | jj� | j t| jjj � d S )NT)Zsecure)
r� r� r� Zcccr� r$ r[ r� r� r� )r r r r �test_ccc� s
zTestTLS_FTPClass.test_cccc C s. | j j� tjtj�}tj|_d|_|jt � t
j |t d�| _ | j j
| jj| jj� | jtj�� | j j� W d Q R X | j j
| jj| jj� | j j� | jtj�� | j jd��}W d Q R X W d Q R X | j j� | j j
d| jj� | j j� | j j� | j j
d| jj� | j j� | j jd��}W d Q R X d S )NT)r� rG rH Z localhost)r� r� r� r� rP Z
CERT_REQUIREDZ verify_modeZcheck_hostnameZload_verify_locations�CAFILEr� rH rM r� r� r� rQ r� ZCertificateErrorrI rJ r; )r rT r[ r r r �test_check_hostname� s,
z$TestTLS_FTPClass.test_check_hostnameN)r# r$ r% r� r� r� rM rN r� rO rQ rU rV rX r r r r rK n s rK c @ sT e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dS )� TestTimeoutsc C s| t j� | _tjtjtj�| _| jjd� tj | j�| _
t j | j d�| _
| j
j� | jj� | jj� tjj
| _| j
tj_
d S )Nr� )�target)r r� �evtr) r� r� r[ rX r Z bind_portrQ r� r� �
server_threadr� r� �clearr� r� �old_port)r r r r r� � s
zTestTimeouts.setUpc C s | j tj_| jj� d | _d S )N)r^ r� r� rQ r\ r8 )r r r r r� � s
zTestTimeouts.tearDownc C s| | j j� | jj� zVy| j j� \}}W n tjk
r> Y n*X |jd� |jtj � | jj� |j
� W d | j j
� X d S )Ns
1 Hola mundo
) r[ rW r[ r� rZ r) rG ZsendallZshutdownZSHUT_WRr )r r rO r r r r� s
zTestTimeouts.serverc C s\ | j tj� � tjd� ztjt�}W d tjd � X | j|jj � d� | j
j � |j � d S )Nr
)
� assertIsNoner) �getdefaulttimeout�setdefaulttimeoutr� r� r r� r[ �
gettimeoutr[ r� r )r �ftpr r r �testTimeoutDefault s
z TestTimeouts.testTimeoutDefaultc C s^ | j tj� � tjd� ztjtd d�}W d tjd � X | j |jj� � | j j
� |j � d S )Nr
)rG ) r_ r) r` ra r� r� r r[ rb r[ r� r )r rc r r r �testTimeoutNone! s
z TestTimeouts.testTimeoutNonec C s6 t jtdd�}| j|jj� d� | jj� |j� d S )Nr
)rG ) r� r� r r� r[ rb r[ r� r )r rc r r r �testTimeoutValue- s
z TestTimeouts.testTimeoutValuec C s> t j� }|jtdd� | j|jj� d� | jj� |j � d S )Nr
)rG )
r� r� r� r r� r[ rb r[ r� r )r rc r r r �testTimeoutConnect4 s
z TestTimeouts.testTimeoutConnectc C s> t jdd�}|jt� | j|jj� d� | jj� |j � d S )Nr
)rG )
r� r� r� r r� r[ rb r[ r� r )r rc r r r �testTimeoutDifferentOrder; s
z&TestTimeouts.testTimeoutDifferentOrderc C s@ t j� }d|_|jt� | j|jj� d� | jj � |j
� d S )Nr
) r� r� rG r� r r� r[ rb r[ r� r )r rc r r r �testTimeoutDirectAccessB s
z$TestTimeouts.testTimeoutDirectAccessN) r# r$ r% r� r� r� rd re rf rg rh ri r r r r rY � s
rY c @ s e Zd Zdd� ZdS )� MiscTestCasec
C s2 ddddddddd d
d d d
h
}t j| t|d� d S )NZMSG_OOBZFTP_PORTZMAXLINEZCRLFZB_CRLFr� Zparse150Zparse227Zparse229r= Z
print_lineZftpcp�test)� blacklist)r Z check__all__r� )r rl r r r � test__all__L s
zMiscTestCase.test__all__N)r# r$ r% rm r r r r rj K s rj c C s8 t tttttg} tj� }ztj| � W d tj |� X d S )N)
r� rY rD rG rK rj r Zthreading_setupZ run_unittestZthreading_cleanup)Ztests� thread_infor r r � test_mainS s ro �__main__)3r� r� r� r r) r r� �osr3 r� � ImportErrorZunittestr r rk r Z test.supportr r �
import_moduler rM r/ ru rw rz r r r' r� r� r~ r r8 �dirname�__file__r� rW r� r� r� r� r� Z IPV6_ENABLEDrD rG rK rY rj ro r# r r r r �<module> s\
41
f ( a+|`