File: //usr/local/lib/python3.7/test/__pycache__/test_ftplib.cpython-37.opt-1.pyc
B
��g�� � @ s d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z y ddl
Z
W n e k
rp dZ
Y nX ddl m
Z
mZ ddlmZ ddlmZmZ dZdd Zd Zd Zd
ZG d d � d ej�ZG d
d� dej�ZG dd� dej ej �Z e
dk �rlej � ej �!e"�d�Z#ej � ej �!e"�d�Z$G dd� dej �Z%G dd� de%e�Z&G dd� de%e�Z'G dd� de �Z(G dd � d e
�Z)eej*d �G d d � d e
��Z+ee
d �G d!d"� d"e)��Z,ee
d �G d#d$� d$e
��Z-G d%d&� d&e
�Z.G d'd(� d(e
�Z/d)d*� Z0e1d+k�re0� dS ),z Test script for ftplib module.� N)�TestCase�
skipUnless)�support)�HOST�HOSTv6� 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 j�| |� || _d| j_d S )N� )�asynchat�
async_chat�__init__� baseclass�last_received_data)�self�connr
� r �,/usr/local/lib/python3.7/test/test_ftplib.pyr 2 s zDummyDTPHandler.__init__c C s | j j| �d��d�7 _d S )Ni �ascii)r
r �recv�decode)r r r r � handle_read7 s zDummyDTPHandler.handle_readc C s$ | j s | j�d� | �� 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(| �� S tt| ��|�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! �
__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 j�| |� | j�tjtjd� | �d� g | _d | _ d | _
d| _ d| _ d | _
d | _t| _| �d� 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 �|� d S )N)r) �append)r �datar r r �collect_incoming_dataf s z%DummyFTPHandler.collect_incoming_datac C s� d� | j��d�}g | _| jr0| �| j� d| _|�d�d �� }|| _|�d�}|dkrl||d d � }nd}t | d| �r�t
| d| �}||� n| �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_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 j�| |�d�d � d S )Nr s
)r
r r r )r r2 r r r r s zDummyFTPHandler.pushc C sl t tt|�d���}dt|d d� � }|d d |d }tj||ftd�}| j|| d�| _ | �
d� d S ) N�,z %d.%d.%d.%d� � � )�timeout)r
z&200 active data connection established) �list�map�intr8 �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 �� d df� |�� |�t� |�� d }| j}|�dd�}|d }|d }| �d|||f � |� � \}}| j
|| d�| _ W d Q R X d S )Nr r' �.rB rD z$227 entering passive mode (%s,%d,%d))r
) r( �bind� getsockname�listen�
settimeoutrL r0 �replacer �acceptrM r* ) r r? �sockrP rO Zp1Zp2r rN r r r �cmd_pasv� s
zDummyFTPHandler.cmd_pasvc C sT |� |d �dd� \}}}t|�}tj||ftd�}| j|| d�| _| �d� d S )Nr r' r6 )rF )r
z&200 active data connection established)r8 rI r( rK rL rM r* r )r r? �afrO rP rQ r r r �cmd_eprt� s
zDummyFTPHandler.cmd_eprtc C s| t � t j��f}|�| j �� d df� |�� |�t� |�� d }| �d| � |�� \}}| j || d�| _
W d Q R X d S )Nr r' z+229 entering extended passive mode (|||%d|))r
) r( �AF_INET6rT rU rV rW rL r rY rM r* )r r? rZ rP r rN r r r �cmd_epsv� s
zDummyFTPHandler.cmd_epsvc C s | � |� d S )N)r )r r? r r r �cmd_echo� s zDummyFTPHandler.cmd_echoc C s | � d� d S )Nz 200 noop ok)r )r r? r r r �cmd_noop� s zDummyFTPHandler.cmd_noopc C s | � d� d S )Nz331 username ok)r )r r? r r r �cmd_user� s zDummyFTPHandler.cmd_userc C s | � d� d S )Nz230 password ok)r )r r? r r r �cmd_pass� s zDummyFTPHandler.cmd_passc C s | � d� d S )Nz 230 acct ok)r )r r? r r r �cmd_acct� s zDummyFTPHandler.cmd_acctc C s | � d� d S )Nz 350 rnfr ok)r )r r? r r r �cmd_rnfr� s zDummyFTPHandler.cmd_rnfrc C s | � d� d S )Nz 250 rnto ok)r )r r? r r r �cmd_rnto� s zDummyFTPHandler.cmd_rntoc C s | � d� d S )Nz 250 dele ok)r )r r? r r r �cmd_dele� s zDummyFTPHandler.cmd_delec C s | � d� d S )Nz
250 cwd ok)r )r r? r r r �cmd_cwd� s zDummyFTPHandler.cmd_cwdc C s | � d� d S )Nz250 1000)r )r r? r r r �cmd_size� s zDummyFTPHandler.cmd_sizec C s | � d| � d S )Nz257 "%s")r )r r? r r r �cmd_mkd� s zDummyFTPHandler.cmd_mkdc C s | � d� d S )Nz
250 rmd ok)r )r r? r r r �cmd_rmd� s zDummyFTPHandler.cmd_rmdc C s | � d� d S )Nz 257 "pwd ok")r )r r? r r r �cmd_pwd� s zDummyFTPHandler.cmd_pwdc C s | � d� d S )Nz 200 type ok)r )r r? r r r �cmd_type� s zDummyFTPHandler.cmd_typec C s | � d� | �� d S )Nz 221 quit ok)r r )r r? r r r �cmd_quit� s
zDummyFTPHandler.cmd_quitc C s | � d� d S )Nz 226 abor ok)r )r r? r r r �cmd_abor� s zDummyFTPHandler.cmd_aborc C s | � d� d S )Nz 125 stor ok)r )r r? r r r �cmd_stor� s zDummyFTPHandler.cmd_storc C s || _ | �d� d S )Nz 350 rest ok)r- r )r r? r r r �cmd_rest� s zDummyFTPHandler.cmd_restc C sN | � d� | jd k r t| j�}nd}| j� | j|d � � | j�� d | _d S )Nz 125 retr okr )r r- rI r* r/ r )r r? �offsetr r r �cmd_retr� s
zDummyFTPHandler.cmd_retrc C s$ | � d� | j� t� | j�� d S )Nz 125 list ok)r r* � LIST_DATAr )r r? r r r �cmd_list� s
zDummyFTPHandler.cmd_listc C s$ | � d� | j� t� | j�� d S )Nz 125 nlst ok)r r* � NLST_DATAr )r r? r r r �cmd_nlst� s
zDummyFTPHandler.cmd_nlstc C s | � d� d S )Nz 200 opts ok)r )r r? r r r �cmd_opts� s zDummyFTPHandler.cmd_optsc C s$ | � d� | j� t� | j�� d S )Nz 125 mlsd ok)r r* � MLSD_DATAr )r r? r r r �cmd_mlsd� s
zDummyFTPHandler.cmd_mlsdc C s dt |� | _| �d� d S )N�xz125 setlongretr ok)rI r/ r )r r? r r r �cmd_setlongretr� s z DummyFTPHandler.cmd_setlongretrN)&r" r# r$ r rM r r3 rA r! r rR 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 @ sZ e Zd ZeZejfdd�Zdd� Zdd� Z dd� Z
d d
� Z d d � Z e Z
d
d� Zdd� ZdS )�DummyFTPServerc C st t j�| � tj�| � d| _| �|tj� | � |� | �
d� d| _ t � � | _
| j�� d d� \| _| _d | _d S )NTrE F� )� threading�Threadr �asyncore�
dispatcher�daemonZ
create_socketr( � SOCK_STREAMrT rV �activeZLock� active_lockrU �hostrP �handler_instance)r Zaddressr\ r r r r s
zDummyFTPServer.__init__c C s$ t �� | _t j�| � | j�� d S )N)r �Event�_DummyFTPServer__flagr� �start�wait)r r r r r� s
zDummyFTPServer.startc C sT d| _ | j�� x2| j rBtjrB| j�� tjddd� | j�� qW tj dd� d S )NTg�������?r' )rF �count)�
ignore_all)
r� r� �setr� Z
socket_mapr� �acquireZloop�release� close_all)r r r r �run s
zDummyFTPServer.runc C s d| _ | �� d S )NF)r� r7 )r r r r �stop! s zDummyFTPServer.stopc C s | � |�| _d S )N)�handlerr� )r r rN r r r �handle_accepted& s z DummyFTPServer.handle_acceptedc C s | � � 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! 0 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 �� }|�t� |j| jdddd�}| �� | �|� 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 �� W n� tjk
r| } zN|jd tjtjfkr<d S |jd tjkrT| �� S d|jd krj| �� S � W d d }~X Y n@ t k
r� } z |jd t
j kr�| �� S W d d }~X Y nX d| _ d S )Nr ZSSLV3_ALERT_BAD_CERTIFICATEr' 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_handshakeJ s
z SSLConnection._do_ssl_handshakec
s� d| _ y| j�� | _W n^ tjk
rT } z |jd tjtjfkrDd S W d d }~X Y n" tk
rt } zW d d }~X Y nX d| _ t | dd�dkr�t
t | �� � 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_shutdown] s
z SSLConnection._do_ssl_shutdownc s2 | j r| �� n | jr | �� ntt| ��� d S )N)r� r� r� r� r r� �handle_read_event)r )r r r r� q s
z SSLConnection.handle_read_eventc s2 | j r| �� n | jr | �� ntt| ��� d S )N)r� r� r� r� r r� �handle_write_event)r )r r r r� y s
z SSLConnection.handle_write_eventc
s^ yt t| ��|�S tjk
rX } z(|jd tjtjtjtj fkrFdS � W d d }~X Y nX d S )Nr )
r r� �sendr� r� r� r� �SSL_ERROR_ZERO_RETURNr� r� )r r2 r� )r r r r� � s
zSSLConnection.sendc
sx yt t| ��|�S tjk
rr } zB|jd tjtjfkr>dS |jd tjtj fkr`| �
� dS � W d d }~X Y nX d S )Nr r4 ) 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$| �� ntt| ��� 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� 9 s
r� c @ s e Zd ZdZdd� ZdS )�DummyTLS_DTPHandlerz.A DummyDTPHandler subclass supporting TLS/SSL.c C s" t �| ||� | jjr | �� 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 �| |� d| _d| _d S )NF)r& r r� r� )r r r r r r � s z DummyTLS_FTPHandler.__init__c C s | � d� | �� 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 | � d� d| _| �� 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 | � 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 |� � }|dkr"| �d� d| _n$|dkr<| �d� d| _n
| �d� 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� rM 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�| _| j�� tjtd�| _| j�| jj | jj
� d S )Nr )rF ) r} r �serverr� �ftplib�FTPrL �client�connectr� rP )r r r r �setUp� s
zTestFTPClass.setUpc C s* | j �� | j�� d | _tjdd� d S )NT)r� )r� r r� r� r� r� )r r r r �tearDown� s
zTestFTPClass.tearDownc C s$ | � t|�t|�� | � ||� d S )N)� assertEqual�len)r �receivedZexpectedr r r �
check_data� s zTestFTPClass.check_datac C s | � | j�� d� d S )Nz 220 welcome)r� r� Z
getwelcome)r r r r �test_getwelcome� s z TestFTPClass.test_getwelcomec C sL | � | j�d�td�� | � | j�d�td�� | � | j�d�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� | � t| jjd� | � t| jjd� | � t| jjd� | � tj| jjd� | � tj| jjd� | � tj| jjd� | � tj| jjd� | � 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 | � | jj� | j�d� | � | jj� | j�d� | �| 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 �d� | j �d� | �tj| j jd� | �tj| 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 �� d S )N)r� �login)r r r r �
test_login s zTestFTPClass.test_loginc C s | j �d� d S )NZpasswd)r� Zacct)r r r r � test_acct s zTestFTPClass.test_acctc C s2 | j �dd� d| jj_| �tj| 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 �d� d| jj_| �tj| 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 �d� d S )Nr� )r� �size)r r r r � test_size$ s zTestFTPClass.test_sizec C s | j �d�}| �|d� d S )Nz/foo)r� Zmkdr� )r �dirr r r �test_mkd' s zTestFTPClass.test_mkdc C s | j �d� d S )Nr� )r� Zrmd)r r r r �test_rmd+ s zTestFTPClass.test_rmdc C s | j �d�}| �|d� d S )Nz/fooz
250 cwd ok)r� �cwdr� )r r� r r r �test_cwd. s zTestFTPClass.test_cwdc C s | j �� }| �|d� d S )Nzpwd ok)r� �pwdr� )r r� r r r �test_pwd2 s
zTestFTPClass.test_pwdc C s& | � | j�� d� | � | jjd � d S )Nz 221 quit ok)r� r� �quitrZ )r r r r � test_quit6 s zTestFTPClass.test_quitc C s | j �� d S )N)r� �abort)r r r r �
test_abort; s zTestFTPClass.test_abortc s4 � fdd�}g � | j �d|� | �d�� �t� d S )Nc s � � | �d�� d S )Nr )r1 r )r2 )r� r r �callback? s z.TestFTPClass.test_retrbinary.<locals>.callback�retrr )r� �
retrbinaryr� r7 r. )r r� r )r� r �test_retrbinary> s z TestFTPClass.test_retrbinaryc sN � fdd�}x<dD ]4}g � | j jd||d� | �d�� �t|d � � qW d S )Nc s � � | �d�� d S )Nr )r1 r )r2 )r� r r r� F s z3TestFTPClass.test_retrbinary_rest.<locals>.callback)r �
� r� )r- r )r� r� r� r7 r. )r r� r- r )r� r �test_retrbinary_restE s
z!TestFTPClass.test_retrbinary_restc C s2 g }| j �d|j� | �d�|�t�dd�� d S )Nr� r z
)r� � retrlinesr1 r� r7 r. rX )r r� r r r �test_retrlinesM s zTestFTPClass.test_retrlinesc sf t �t�d��}| j�d|� | �| jjj t� g � |�
d� | jjd|� fdd�d� | � � � d S )Nr �storr c s
� � d �S )N)r1 )r{ )�flagr r �<lambda>Y r4 z.TestFTPClass.test_storbinary.<locals>.<lambda>)r� ) �io�BytesIOr. r r� �
storbinaryr� r� r� r �seekr� )r �fr )r r �test_storbinaryR s
z TestFTPClass.test_storbinaryc C s\ t �t�dd��d��}x>dD ]6}|�d� | jjd||d� | �| j j
j t |�� q W d S )Nz
�
r )� Z30r r )r- )
r r r. rX r r r� r r� r� r� r- �str)r r �rr r r �test_storbinary_rest\ s
z!TestFTPClass.test_storbinary_restc s� t �t�dd��d��}| j�d|� | �| jj j
t� g � |� d� | jjd|� fdd�d � | � � � t �
t�dd��}tjd
tfd d �� | �t| jjd|� W d Q R X d S )
Nz
r
r r r zstor fooc s
� � d �S )N)r1 )r{ )r r r r j r4 z-TestFTPClass.test_storlines.<locals>.<lambda>)r� r T)�quiet)r r r. rX r r� � storlinesr� r� r� r r r� �StringIOr Zcheck_warnings� BytesWarningr� � TypeError)r r r )r r �test_storlinesc s
zTestFTPClass.test_storlinesc C s. | j �� | �| j �� t�d�d d� � d S )Nz
r6 )r� Znlstr� rv r8 )r r r r � test_nlstr s
zTestFTPClass.test_nlstc s6 g � | j �� fdd�� | �d�� �t�dd�� d S )Nc s
� � | �S )N)r1 )r{ )�lr r r x r4 z'TestFTPClass.test_dir.<locals>.<lambda>r z
)r� r� r� r7 rt rX )r r )r r �test_dirv s zTestFTPClass.test_dirc s� t � j�� � t � jjdd�� t � jjdddgd�� t � j�� �}xV|D ]N\}}� �|t� � �|t� � �|� � �d|� � �d|� � �d|� 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� j�� �\}}x |D ]}� �|� � � �qrW �d(� � �
t t� j�� � �d(� x"� j�� D ]}� � d)| � �q�W d S )+N�/)�pathr� �type)r �facts�perm�uniquec s | � j j_d S )N)r� r� r )r2 )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 � j�� �\}}� �||� � �|d |� � �|d |� � �|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)
rG r� r! �assertIsInstancer �dictr� ZassertInr �islowerr� �
StopIteration�fail)r Zlsr r r# r" r{ r )r r r � test_mlsd{ sF
zTestFTPClass.test_mlsdc C s, | j �� � | �| jjjd� W d Q R X d S )NrP )r� �makeportr� r� r� r+ )r r r r �
test_makeport� s zTestFTPClass.test_makeportc C s> | j �� \}}tj||ftd�}|�� | �| jjj d� d S )N)rF Zpasv)
r� �makepasvr( rK rL r r� r� r� r+ )r r� rP r r r r �
test_makepasv� s zTestFTPClass.test_makepasvc C sN d| j _| j �� \}}| �|| jjj� tj| j j �
� d |ft d�� � dS )z/Test the opt-in to the old vulnerable behavior.Tr )rF N)
r� � trust_server_pasv_ipv4_addressr, r� r� r� r0 r( rK rZ Z getpeernamerL r )r Zbad_hostrP r r r �*test_makepasv_issue43285_security_disabled� s z7TestFTPClass.test_makepasv_issue43285_security_disabledc C sH | � | jj� | j�� \}}| �|| jjj� tj ||ft
d�� � d S )N)rF ) r� r� r. r, ZassertNotEqualr� r� r0 r( rK rL r )r Z trusted_hostrP r r r �1test_makepasv_issue43285_security_enabled_default� s
z>TestFTPClass.test_makepasv_issue43285_security_enabled_defaultc
s� � j �� � fdd�}tjtd��6� _ � j �� jj� jj� � j � d� � �
|� � W d Q R X � � � jj j
d� � �|� � tjtd��4� _ � j �� jj� jj� � j � d� � j �� W d Q R X � � � jj j
d� � �|� � yJtjtd��4� _ � j �� jj� jj� � j � d� d� jj _W d Q R X W n6 tjk
�r` } z� � t|�d� W d d }~X Y n X � �d� t�d� � � � jj j
d� � �|� � d S ) Nc s> � j jd krdS y� j �d� W n ttfk
r8 dS X dS )NF�noopT)r� rZ r� r� r� r )r r r �is_client_connected� s z=TestFTPClass.test_with_statement.<locals>.is_client_connected)rF r1 r� z550 error on quitzException not raisedg�������?)r� r� r� r� rL r� r� r� rP 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 �� t�� }yD| j j| jj| jjt|fd� | � | j j
� � d |� | j �� W n> t k
r� } z |j
t
jkr�| �d| � � W d d }~X Y nX d S )N)�source_addressr' zcouldn't bind to port %d)r� r� r �find_unused_portr� r� r� rP r r� rZ rU r� r� �
EADDRINUSE�skipTest)r rP �er r r �test_source_address s
z TestFTPClass.test_source_addressc
C s� t �� }t|f| j_y0| j�d��}| �|�� d |� W d Q R X W n> tk
r� } z |j t j
krp| � d| � � W d d }~X Y nX d S )NrG r' zcouldn't bind to port %d) r r6 r r� r5 � transfercmdr� rU r� r� r7 r8 )r rP rZ r9 r r r �&test_source_address_passive_connection s " z3TestFTPClass.test_source_address_passive_connectionc C s� | � t�d�d� | � t�d�d� | � t�d�d� | � t�d�d� | �tjtjd� | � t�d�d � | � t�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� �parse257r� r� )r r r r �
test_parse257 s zTestFTPClass.test_parse257c C s$ | � 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 �d| j jd � g }| �tj| j jd|j� d S )NzSETLONGRETR %dr~ r� )r� r� r? r� r� r� r� r1 )r r� r r r �test_retrlines_too_long+ s z$TestFTPClass.test_retrlines_too_longc C s0 t �d| jj d �}| �tj| jjd|� d S )N� xr~ r )r r r� r? r� r� r� r )r r r r r �test_storlines_too_long1 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�| _| j�� tjtd�| _ | j �
| jj | jj � d S )Nr )r\ )rF )
r} r r( r^ r� r� r� r� rL r� r� r� rP )r r r r r� 9 s
zTestIPv6Environment.setUpc C s* | j �� | j�� d | _tjdd� d S )NT)r� )r� r r� r� r� r� )r r r r r� ? s
z TestIPv6Environment.tearDownc C s | � | jjtj� d S )N)r� r� r\ r( r^ )r r r r �test_afF s zTestIPv6Environment.test_afc C s, | j �� � | �| jjjd� W d Q R X d S )NZeprt)r� r* r� r� r� r+ )r r r r r+ I s z!TestIPv6Environment.test_makeportc C s> | j �� \}}tj||ftd�}|�� | �| jjj d� d S )N)rF Zepsv)
r� r, r( rK rL r r� r� r� r+ )r r� rP r r r r r- N s z!TestIPv6Environment.test_makepasvc s4 � fdd�}� j �d� |� � j �d� |� d S )Nc sN � fdd�} g � �j �d| � ��td�� ��tt�� ��d�� �t� d S )Nc s � � | �d�� d S )Nr )r1 r )r2 )r� r r r� V s zATestIPv6Environment.test_transfer.<locals>.retr.<locals>.callbackr� r )r� r� r� r� r7 r. )r� )r )r� r r� U s
z/TestIPv6Environment.test_transfer.<locals>.retrTF)r� r� )r r� r )r r �
test_transferT s
z!TestIPv6Environment.test_transferN) r" r# r$ r� r� rE r+ r- rF r r r r rD 6 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�| _| j�� tjtd�| _| j�| jj | jj
� | j� � | j� � d S )Nr )rF )
r� r r� r� r� �FTP_TLSrL r� r� r� rP �auth�prot_p)r r r r r� h s
zTestTLS_FTPClassMixin.setUpN)r" r# r$ r� r� r r r r rG b s rG c @ sb e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z e dd�dd� �Z
dS )�TestTLS_FTPClassz Specific TLS_FTP class tests.c C s@ t tdf�| _| j�� tjtd�| _| j�| jj | jj
� d S )Nr )rF ) r� r r� r� r� rH rL r� r� r� rP )r r r r r� v s
zTestTLS_FTPClass.setUpc C s* | j �� | j�� d | _tjdd� d S )NT)r� )r� r r� r� r� r� )r r r r r� | s
zTestTLS_FTPClass.tearDownc C s2 | � | jjtj� | j�� | �| jjtj� d S )N)�assertNotIsInstancer� rZ r� r� rI r$ )r r r r �test_control_connection� s
z(TestTLS_FTPClass.test_control_connectionc C s | j �d��,}| �|tj� | �|�d�t�d�� W d Q R X | �| j � � d� | j �
� | j �d��,}| � |tj� | �|�d�t�d�� W d Q R X | �| j � � d� | j � � | j �d��,}| �|tj� | �|�d�t�d�� W d Q R X | �| j � � d� d S )NrG i r z226 transfer complete)
r� r; rL r� r� r� r rt r ZvoidresprJ r$ Zprot_c)r rZ r r r �test_data_connection� s "
"
"z%TestTLS_FTPClass.test_data_connectionc C s< | � | jjtj� | j�� | �| jjtj� | j�� d S )N)rL r� rZ r� r� r� r$ )r r r r r� � s
zTestTLS_FTPClass.test_loginc C s | j �� | �t| 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
| j �� t�tj�}d|_tj|_| jt t
j t |d� | jt t
j t |d� | jt t
j t t |d� t
j |t
d�| _ | j �| jj| jj� | �| j jtj� | j �� | �| j jj|� | �| j jtj� | j �� | j �d��"}| �|j|� | �|tj� W d Q R X d S )NF)�keyfiler� )�certfiler� )rQ rP r� )r� rF rG )r� r� r� r� �PROTOCOL_TLS_CLIENT�check_hostnameZ CERT_NONE� verify_moder� r� r� rH r� rL r� r� r� rP rL rZ r� rI ZassertIsr� r$ rJ r; )r �ctxrZ r r r � test_context� s(
z TestTLS_FTPClass.test_contextc C sP | � t| jj� | jjdd� | �| jjtj� | j�� | � t| jjj � d S )NT)Zsecure)
r� r� r� Zcccr� r$ rZ r� r� r� )r r r r �test_ccc� s
zTestTLS_FTPClass.test_cccFzFIXME: bpo-32706c
C s> | j �� t�tj�}| �|jtj� | �|jd� |� t
� t j |t
d�| _ | j �| jj| jj� | �tj�� | j �� W d Q R X | j �| jj| jj� | j �� | �tj�� | j �d��}W d Q R X W d Q R X | j �� | j �d| jj� | j �� | j �� | j �d| jj� | j �� | j �d��}W d Q R X d S )NT)r� rF rG Z localhost)r� r� r� r� rR r� rT Z
CERT_REQUIREDrS Zload_verify_locations�CAFILEr� rH rL r� r� r� rP r� ZCertificateErrorrI rJ r; )r rU rZ r r r �test_check_hostname� s,
z$TestTLS_FTPClass.test_check_hostnameN)r" r# r$ r� r� r� rM rN r� rO rV rW r rY r r r r rK r 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 �� | _t�tjtj�| _| j�d� t� | j�| _
t j | j d�| _
d| j
_| j
�� | j�� | j�� tjj
| _| j
tj_
d S )Nr� )�targetT)r r� �evtr( r� r� rZ rW r Z bind_portrP r� r� �
server_threadr� r� r� �clearr� r� �old_port)r r r r r� � s
zTestTimeouts.setUpc C s | j tj_| j�� d | _d S )N)r_ r� r� rP r] r7 )r r r r r� � s
zTestTimeouts.tearDownc C s| | j �� | j�� zVy| j �� \}}W n tjk
r> Y n*X |�d� |�tj � | j�� |�
� W d | j �
� X d S )Ns
1 Hola mundo
) rZ rV r\ r� rY r( rF ZsendallZshutdownZSHUT_WRr )r r rN r r r r� s
zTestTimeouts.serverc C s\ | � t�� � t�d� zt�t�}W d t�d � X | �|j� � d� | j
� � |� � d S )Nr )
� assertIsNoner( �getdefaulttimeout�setdefaulttimeoutr� r� r r� rZ �
gettimeoutr\ r� r )r �ftpr r r �testTimeoutDefault s
z TestTimeouts.testTimeoutDefaultc C s^ | � t�� � t�d� ztjtd d�}W d t�d � X | � |j�� � | j �
� |� � d S )Nr )rF ) r` r( ra rb r� r� r rZ rc r\ r� r )r rd r r r �testTimeoutNone! s
z TestTimeouts.testTimeoutNonec C s6 t jtdd�}| �|j�� d� | j�� |�� d S )Nr )rF ) r� r� r r� rZ rc r\ r� r )r rd r r r �testTimeoutValue- s
z TestTimeouts.testTimeoutValuec C s> t �� }|jtdd� | �|j�� d� | j�� |� � d S )Nr )rF )
r� r� r� r r� rZ rc r\ r� r )r rd r r r �testTimeoutConnect4 s
z TestTimeouts.testTimeoutConnectc C s> t jdd�}|�t� | �|j�� d� | j�� |� � d S )Nr )rF )
r� r� r� r r� rZ rc r\ r� r )r rd r r r �testTimeoutDifferentOrder; s
z&TestTimeouts.testTimeoutDifferentOrderc C s@ t �� }d|_|�t� | �|j�� d� | j� � |�
� d S )Nr ) r� r� rF r� r r� rZ rc r\ r� r )r rd r r r �testTimeoutDirectAccessB s
z$TestTimeouts.testTimeoutDirectAccessN) r" r# r$ r� r� r� re rf rg rh ri rj r r r r rZ � s rZ 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 rm r r r � test__all__L s
zMiscTestCase.test__all__N)r" r# r$ rn r r r r rk K s rk c C s8 t tttttg} t�� }ztj| � W d tj |� X d S )N)
r� rZ rD rG rK rk r Zthreading_setupZ run_unittestZthreading_cleanup)Ztests� thread_infor r r � test_mainS s rp �__main__)2r� r� r� r
r( r r� �osr r3 r� � ImportErrorZunittestr r rl r Z test.supportr r rL r. rt rv ry r r r& r� r� r} r r7 �dirname�__file__r� rX r� r� r� r� r� Z IPV6_ENABLEDrD rG rK rZ rk rp r" r r r r �<module> s\
42
i ( a+wa