U sÍ@gÐÆã@sìdZddlZddlZddlZddlZddlZddlZddlZddlm Z ddl m Z ddl m Z ddlmZe dƒZe dƒZe d ƒZdZeröerözPdd lmZdd lmZdd lmZdd lmZddlmZmZddlm Z Wne!k rôYnXddl"m#Z#ddl$m%Z%ddlm&Z&ddlm'Z'ddl(m)Z)ddl*m+Z+m,Z,m-Z-m.Z.m/Z/ddl0m1Z1m2Z2ddl3m4Z4ddl5m6Z6m7Z7ddl8m9Z9ddl:m;Z;ddlm?Z?Gdd„de?ƒZ@Gdd „d e?ƒZAGd!d"„d"e;eBƒZCGd#d$„d$eBƒZDGd%d&„d&eBƒZEee%ƒGd'd(„d(e ƒƒZFGd)d*„d*e?ƒZGGd+d,„d,ƒZHGd-d.„d.e,jIƒZJGd/d0„d0e&ƒZKGd1d2„d2eKƒZLGd3d4„d4eKƒZMGd5d6„d6eKƒZNdeeee- Oe+d¡fkrÌedkr¬d7ZeeL_PeeM_PeeN_PeeG_Pee@_Pndd8lQmRZReRd9ƒsèd:eN_PdS);z* Tests for L{twisted.conch.scripts.cftp}. éN)ÚBytesIO)ÚFilePath)Ú requireModule)Ú implementerÚpyasn1Ú cryptographyztwisted.conch.unix)Úcftp)Ú SSHSession)Ú filetransfer)ÚFileTransferForTestAvatar)Útest_sshÚ test_conch)Ú FakeStdio)Úls)Ú ISFTPFile)Ú SFTPTestBase)ÚFileTransferTestAvatar)Úportal)ÚreactorÚprotocolÚ interfacesÚdeferÚerror)ÚgetProcessOutputAndValueÚgetProcessValue)Úlog)Ú_PY3Úunicode)Ú UserDatabase)ÚStringTransport)ÚClock)ÚTestCasec@seZdZdZdd„ZdS)ÚSSHSessionTestsz= Tests for L{twisted.conch.scripts.cftp.SSHSession}. cCs*tƒ}tƒ}||_| ¡| |j¡dS)z| L{twisted.conch.scripts.cftp.SSHSession.eofReceived} loses the write half of its stdio connection. N)rr ÚstdioZ eofReceivedÚ assertTrueZ writeConnLost)Úselfr#Zchannel©r&ú>/usr/lib/python3/dist-packages/twisted/conch/test/test_cftp.pyÚtest_eofReceived6s z SSHSessionTests.test_eofReceivedN)Ú__name__Ú __module__Ú __qualname__Ú__doc__r(r&r&r&r'r"2sr"c@s´eZdZdZeeddƒdkr dZdd„Zdd„Zd d „Z d d „Z d d„Z dd„Z e  ¡ZzBze  e jd¡Wne jk rˆde _YnXe  e je¡Wn YnXdd„ZdS)Ú ListingTestszƒ Tests for L{lsLine}, the function which generates an entry for a file or directory in an SFTP I{ls} command's output. ÚtzsetNz8Cannot test timestamp formatting code without time.tzsetcshdˆ_‡fdd„}ˆ td|¡dtjkrRˆ tjtjdtjd¡ˆ tj ¡ndd„}ˆ |¡dS) zo Patch the L{ls} module's time function so the results of L{lsLine} are deterministic. iÍ[csˆjS©N)Únowr&©r%r&r'ÚfakeTimeRsz$ListingTests.setUp..fakeTimeÚtimeÚTZcSs.z tjd=Wntk r YnXt ¡dS)Nr4)ÚosÚenvironÚKeyErrorr3r.r&r&r&r'Úcleanup\s  z#ListingTests.setUp..cleanupN) r0Úpatchrr5r6Ú addCleanupÚoperatorÚsetitemr3r.)r%r2r8r&r1r'ÚsetUpLs  zListingTests.setUpcCs|tjd<t ¡t d|¡S)zl Call L{ls.lsLine} after setting the timezone to C{timezone} and return the result. r4Úfoo)r5r6r3r.rZlsLine)r%ÚtimezoneÚstatr&r&r'Ú _lsInTimezonefs zListingTests._lsInTimezonec CsT|jd}t dddddddd|df ¡}| | d|¡d¡| | d|¡d¡dS)z‘ A file with an mtime six months (approximately) or more in the past has a listing including a low-resolution timestamp. é€rúAmerica/New_Yorkz;!--------- 0 0 0 0 Apr 26 1973 fooúPacific/Aucklandz;!--------- 0 0 0 0 Apr 27 1973 fooN©r0r5Ú stat_resultÚ assertEqualrA©r%Úthenr@r&r&r'Ú test_oldFilers  þ þzListingTests.test_oldFilec CsX|jdd}t dddddddd|df ¡}| | d|¡d¡| | d|¡d¡dS) úæ A file with a high-resolution timestamp which falls on a day of the month which can be represented by one decimal digit is formatted with one padding 0 to preserve the columns which come after it. rBi€—rrCz;!--------- 0 0 0 0 May 01 1973 foorDz;!--------- 0 0 0 0 May 02 1973 fooNrErHr&r&r'Útest_oldSingleDigitDayOfMonthƒs þ þz*ListingTests.test_oldSingleDigitDayOfMonthc CsT|jd}t dddddddd|df ¡}| | d|¡d¡| | d|¡d¡dS)z¨ A file with an mtime fewer than six months (approximately) in the past has a listing including a high-resolution timestamp excluding the year. 逛zrrCú;!--------- 0 0 0 0 Aug 28 17:33 foorDú;!--------- 0 0 0 0 Aug 29 09:33 fooNrErHr&r&r'Ú test_newFile–s  þ þzListingTests.test_newFilec Cs||jd}t dddddddd|df ¡}t ¡}t tjd¡| tjtj|¡| |  d|¡d¡| |  d|¡d¡dS) zC The month name in the date is locale independent. rMrú es_AR.UTF8rCrNrDrON) r0r5rFÚlocaleÚ getlocaleÚ setlocaleÚLC_ALLr:rGrA)r%rIr@Ú currentLocaler&r&r'Útest_localeIndependent§s  þ þz#ListingTests.test_localeIndependentrQz'The es_AR.UTF8 locale is not installed.c CsX|jdd}t dddddddd|df ¡}| | d|¡d¡| | d|¡d¡dS) rKrMiFrrCz;!--------- 0 0 0 0 Sep 01 17:33 foorDz;!--------- 0 0 0 0 Sep 02 09:33 fooNrErHr&r&r'Útest_newSingleDigitDayOfMonthÉs þ þz*ListingTests.test_newSingleDigitDayOfMonth)r)r*r+r,Úgetattrr3Úskipr=rArJrLrPrWrRrSrVrTrUÚErrorrXr&r&r&r'r-Cs&  r-cs eZdZdZ‡fdd„Z‡ZS)ÚInMemorySSHChannelzp Minimal implementation of a L{SSHChannel} like class which only reads and writes data from memory. cs||_d|_tt|ƒ ¡dS)zt @param conn: The SSH connection associated with this channel. @type conn: L{SSHConnection} rN)ÚconnZ localClosedÚsuperr\Ú__init__)r%r]©Ú __class__r&r'r_ãszInMemorySSHChannel.__init__)r)r*r+r,r_Ú __classcell__r&r&r`r'r\Ýsr\c@s(eZdZdZdd„Zdd„Zdd„ZdS) ÚFilesystemAccessExpectationszC A test helper used to support expected filesystem access. cCs i|_dSr/©Ú_cacher1r&r&r'r_ósz%FilesystemAccessExpectations.__init__cCs||j||f<dS)zü @param path: Path at which the stream is requested. @type path: L{str} @param path: Flags with which the stream is requested. @type path: L{str} @param stream: A stream. @type stream: C{File} Nrd)r%ÚpathÚflagsÚstreamr&r&r'Úput÷s z FilesystemAccessExpectations.putcCs|j ||f¡S)a Remove a stream from the memory. @param path: Path at which the stream is requested. @type path: L{str} @param path: Flags with which the stream is requested. @type path: L{str} @return: A stream. @rtype: C{File} )reÚpop)r%rfrgr&r&r'rjs z FilesystemAccessExpectations.popN)r)r*r+r,r_rirjr&r&r&r'rcîsrcc@s eZdZdZdd„Zdd„ZdS)ÚInMemorySFTPClienta' A L{filetransfer.FileTransferClient} which does filesystem operations in memory, without touching the local disc or the network interface. @ivar _availableFiles: File like objects which are available to the SFTP client. @type _availableFiles: L{FilesystemRegister} cCs t|ƒ|_dddœ|_||_dS)Néé )ZrequestsÚ buffersize)r\Ú transportÚoptionsÚ_availableFiles)r%ZavailableFilesr&r&r'r_!s  þzInMemorySFTPClient.__init__cCs|j ||¡S)z} @see: L{filetransfer.FileTransferClient.openFile}. Retrieve and remove cached file based on flags. )rqrj)r%ÚfilenamergÚattrsr&r&r'ÚopenFile*szInMemorySFTPClient.openFileN)r)r*r+r,r_rtr&r&r&r'rks  rkc@s0eZdZdZdd„Zdd„Zdd„Zdd „Zd S) ÚInMemoryRemoteFilez; An L{ISFTPFile} which handles all data in memory. cCs||_t |¡dS)zL @param name: Name of this file. @type name: L{str} N)Únamerr_)r%rvr&r&r'r_:szInMemoryRemoteFile.__init__cCs| |¡| |¡t |¡S)z/ @see: L{ISFTPFile.writeChunk} )ÚseekÚwriterÚsucceed)r%ÚstartÚdatar&r&r'Ú writeChunkCs  zInMemoryRemoteFile.writeChunkcCs d|_dS)zo @see: L{ISFTPFile.writeChunk} Keeps data after file was closed to help with testing. TN)Ú_closedr1r&r&r'ÚcloseLszInMemoryRemoteFile.closecCs t |¡S)zb Get current data of file. Allow reading data event when file is closed. )rÚgetvaluer1r&r&r'rUszInMemoryRemoteFile.getvalueN)r)r*r+r,r_r|r~rr&r&r&r'ru4s    ruc@s¤eZdZdZdd„Zdd„Zdd„Zdd „Zd d „Zd d „Z dd„Z dd„Z dd„Z dd„Z dd„Zdd„Zd)dd„Zd*dd „Zd!d"„Zd#d$„Zd%d&„Zd'd(„ZdS)+ÚStdioClientTestsz( Tests for L{cftp.StdioClient}. cCsTtƒ|_t|jƒ}t |¡|_d|j_tƒ|_|j_ |  dd¡|jjj |j_ dS)zm Create a L{cftp.StdioClient} hooked up to dummy transport and a fake user database. ú/iôéN) rcÚfakeFilesystemrkrÚ StdioClientÚclientZcurrentDirectoryrÚdatabaseZ_pwdÚsetKnownConsoleSizero)r%Z sftpClientr&r&r'r=cs   zStdioClientTests.setUpc Cs@|j t ¡dt ¡dddtj¡|j  d¡}|  |j d¡|S)zs The I{exec} command runs its arguments locally in a child process using the user's shell. ÚsecretéÒr>Úbarzexec print(1 + 2)s3 ) r†ÚaddUserÚgetpassÚgetuserr5ÚgetuidÚsysÚ executabler…Ú_dispatchCommandÚ addCallbackrG©r%Údr&r&r'Ú test_execwsþ zStdioClientTests.test_execc Cs>|j t ¡dt ¡dddd¡|j d¡}| |j d¡|S)zr If the local user has no shell, the I{exec} command runs its arguments using I{/bin/sh}. rˆr‰r>rŠÚzexec echo helloóhello © r†r‹rŒrr5rŽr…r‘r’rGr“r&r&r'Útest_execWithoutShell…sÿ z&StdioClientTests.test_execWithoutShellc Cs>|j t ¡dt ¡dddd¡|j d¡}| |j d¡|S)zO The I{exec} command is run for lines which start with C{"!"}. rˆr‰r>rŠz/bin/shz !echo hellor—r˜r“r&r&r'Ú test_bang’sþ zStdioClientTests.test_bangcs4ddl‰G‡‡‡fdd„dtƒ}| td|ƒ¡dS)a For the duration of this test, patch C{cftp}'s C{fcntl} module to return a fixed width and height. @param width: the width in characters @type width: L{int} @param height: the height in characters @type height: L{int} rNcseZdZ‡‡‡fdd„ZdS)z7StdioClientTests.setKnownConsoleSize..FakeFcntlcs&|ˆjkr| d¡t dˆˆdd¡S)Nz#Only window-size queries supported.Z4Hr)Z TIOCGWINSZZfailÚstructZpack)r%ÚfdZoptZmutate©ÚheightÚttyÚwidthr&r'Úioctl¬s  z=StdioClientTests.setKnownConsoleSize..FakeFcntl.ioctlN)r)r*r+r¡r&rr&r'Ú FakeFcntl«sr¢Zfcntl)rŸÚobjectr9r)r%r ržr¢r&rr'r‡Ÿs z$StdioClientTests.setKnownConsoleSizecCsŒ| dd¡tƒ}|j_tdƒ}d|_t |¡}d|_|  ¡}|  d¡|j d7_ |j  ||¡t rpd}nd }| |jj ¡|¡d S) a( L{StdioClient._printProgressBar} prints a progress description, including percent done, amount transferred, transfer rate, and time remaining, all based the given start time, the given L{FileWrapper}'s progress information and the reactor's current time. rmé"óxósamplei(g@is# b'sample' 40% 4.0kB 2.0kBps 00:03 z sample 40% 4.0kB 2.0kBps 00:03 N)r‡r r…rrrvrÚ FileWrapperÚsizeÚsecondsZadvanceZtotalÚ_printProgressBarrrGroÚvalue©r%ZclockÚwrappedÚwrapperZ startTimeÚresultr&r&r'Útest_printProgressBarReporting³s   z/StdioClientTests.test_printProgressBarReportingcCsn| dd¡tƒ}|j_tdƒ}d|_t |¡}| ¡}|j  ||¡t rRd}nd}|  |jj   ¡|¡dS)z½ L{StdioClient._printProgressBar} prints a progress description that indicates 0 bytes transferred if no bytes have been transferred and no time has passed. rmr¤r¥r¦s! b'sample' 0% 0.0B 0.0Bps 00:00 z sample 0% 0.0B 0.0Bps 00:00 N)r‡r r…rrrvrr§r©rªrrGror«r¬r&r&r'Útest_printProgressBarNoProgressÏs  z0StdioClientTests.test_printProgressBarNoProgresscCsV| dd¡tƒ}d|_t |¡}|j |d¡tr:d}nd}| ||jj   ¡¡dS)z5 Print the progress for empty files. rmr¤s empty-filers% b'empty-file'100% 0.0B 0.0Bps 00:00 z" empty-file100% 0.0B 0.0Bps 00:00 N) r‡rrvrr§r…rªrrGror«)r%r­r®r¯r&r&r'Útest_printProgressBarEmptyFileås  z/StdioClientTests.test_printProgressBarEmptyFilecCs|j d¡}| d|¡dS)zK Returns empty value for both filename and remaining data. z )r–r–N©r…Z _getFilenamerG©r%r¯r&r&r'Útest_getFilenameEmpty÷s z&StdioClientTests.test_getFilenameEmptycCs|j d¡}| d|¡dS)zd Returns empty value for remaining data when line contains only a filename. ú only-local)r¶r–Nr³r´r&r&r'Útest_getFilenameOnlyLocals z*StdioClientTests.test_getFilenameOnlyLocalcCs|j d¡}| d|¡dS)ze Returns filename and remaining data striped of leading and trailing spaces. z local remote file )Úlocalz remote fileNr³r´r&r&r'Útest_getFilenameNotQuoted s z*StdioClientTests.test_getFilenameNotQuotedcCs|j d¡}| d|¡dS)z‰ Returns filename and remaining data not striped of leading and trailing spaces when quoted paths are requested. z" " local file " " remote file " )z local file z" remote file "Nr³r´r&r&r'Útest_getFilenameQuoteds z'StdioClientTests.test_getFilenameQuotedNóc Cs4|dkr| ¡}t|dƒ}| |¡W5QRX|S)ab Create a local file and return its path. When `path` is L{None}, it will create a new temporary file. @param path: Optional path for the new file. @type path: L{str} @param content: Content to be written in the new file. @type content: L{bytes} @return: Path to the newly create file. NÚwb)ÚmktempÚopenrx)r%rfÚcontentÚfiler&r&r'ÚmakeFiles  zStdioClientTests.makeFileFcCsf|jj ¡}tr| d¡}| d¡}g}g}|D]ö\}}}g} |D]} |  d|| f¡qB|  d||f¡| | ¡| d¡ d¡ d¡} | dd…} | d d ¡ d ¡} |   | ¡g}| dd…D]F} |  ¡  d d ¡d} |  ¡ d d ¡} | d| d| d  ¡f¡qÆ| | d¡| |¡q0|rD|  t |ƒt |ƒ¡n |  ||¡|  dt |ƒd ¡dS)a Check output of cftp client for a put request. @param transfers: List with tuple of (local, remote, progress). @param randomOrder: When set to C{True}, it will ignore the order in which put reposes are received úutf-8z z%s %szTransferred %s to %srú NéÿÿÿÿÚ ú érlz5There are still put responses which were not checked.)r…ror«rÚdecodeÚsplitÚappendrjÚstripÚextendÚrsplitrGÚsortedÚlen)r%Z transfersÚ randomOrderÚoutputÚexpectedOutputZ actualOutputr¸ZremoteZexpectedZexpectedTransferÚlineZ progressPartsZactualZlastZactualTransferr&r&r'ÚcheckPutMessage3s>         þz StdioClientTests.checkPutMessagecCs²d}|j|d}tjtjBtjB}tj dtj |¡¡}t |ƒ}|j   ||t   |¡¡d|jjjd<|j |¡}| |¡| || ¡¡| |j¡| ||dddgfg¡dS) z A name based on local path is used when remote path is not provided. The progress is updated while chunks are transferred. s Test Content)r¿rrmrnz 76% 10.0Bz 100% 13.0BN)rÁr Ú FXF_WRITEÚ FXF_CREATÚ FXF_TRUNCr5rfÚjoinÚbasenamerurƒrirryr…rpÚcmd_PUTÚsuccessResultOfrGrr$r}rÔ)r%r¿Ú localPathrgÚ remoteNameÚ remoteFileÚdeferredr&r&r'Útest_cmd_PUTSingleNoRemotePathqs* ÿþÿ   ÿÿz/StdioClientTests.test_cmd_PUTSingleNoRemotePathcCsŽ| ¡}tjtjBtjB}d}t|ƒ}|j ||t  |¡¡|j   d||f¡}|  |¡|  ||dgfg¡| |j¡| d| ¡¡dS)z€ Remote path is extracted from first filename after local file. Any other data in the line is ignored. z /remote-pathz %s %s ignoredú 100% 0.0Br»N)rÁr rÕrÖr×rurƒrirryr…rÚrÛrÔr$r}rGr)r%rÜrgrÝrÞrßr&r&r'Útest_cmd_PUTSingleRemotePathŽs"ÿþÿ ÿ  z-StdioClientTests.test_cmd_PUTSingleRemotePathc Cs| ¡}tj |¡}d}tj |¡}|jtj ||¡d}tjtjBtj B}d|f}d|f}t |ƒ} t |ƒ} |j   ||t  | ¡¡|j   ||t  | ¡¡|j tj |d¡¡} | | ¡| | j¡| d|  ¡¡| | j¡| d|  ¡¡|j||dgf||dgfgddd S) z„ When a gobbing expression is used local files are transferred with remote file names based on local names. ú second-name©rfz/%sÚ*r»ráT©rÐN©rÁr5rfrÙÚdirnamerØr rÕrÖr×rurƒrirryr…rÚrÛr$r}rGrrÔ) r%ÚfirstÚ firstNameÚ secondNameÚparentÚsecondrgÚfirstRemotePathÚsecondRemotePathÚfirstRemoteFileÚsecondRemoteFilerßr&r&r'Ú test_cmd_PUTMultipleNoRemotePath§sJ  ÿþÿ  ÿÿ     þüz1StdioClientTests.test_cmd_PUTMultipleNoRemotePathc Cs | ¡}tj |¡}d}tj |¡}|jtj ||¡d}tjtjBtj B}t |ƒ}t |ƒ}d|f} d|f} |j   | |t  |¡¡|j   | |t  |¡¡|j dtj |d¡f¡} | | ¡| |j¡| d| ¡¡| |j¡| d| ¡¡|j||dgf||dgfgdd d S) zõ When a gobbing expression is used local files are transferred with remote file names based on local names. when a remote folder is requested remote paths are composed from remote path and local filename. rãräz /remote/%sz %s remoterår»ráTræNrç) r%rérêrërìrírgrðrñrîrïrßr&r&r'Ú"test_cmd_PUTMultipleWithRemotePathÎsN  ÿþÿ  ÿÿÿ     þüz3StdioClientTests.test_cmd_PUTMultipleWithRemotePath)Nr»)F)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€_s&       >'r€c@seZdZdd„Zdd„ZdS)ÚFileTransferTestRealmcCs ||_dSr/)ÚtestDir)r%rõr&r&r'r_úszFileTransferTestRealm.__init__cGst|jƒ}|d|dd„fS)NrcSsdSr/r&r&r&r&r'Úr»z5FileTransferTestRealm.requestAvatar..)rrõ)r%ZavatarIDZmindrÚar&r&r'Ú requestAvatarþs z#FileTransferTestRealm.requestAvatarN)r)r*r+r_rør&r&r&r'rôùsrôc@s`eZdZdZdd„Zdd„Zdd„Zdd „Zd d „Zd d „Z dd„Z dd„Z dd„Z dd„Z dS)ÚSFTPTestProcessz³ Protocol for testing cftp. Provides an interface between Python (where all the tests are) and the cftp client process (which does the work that is being tested). cCs$| ¡||_d|_d|_d|_dS)zr @param onOutReceived: A L{Deferred} to be fired as soon as data is received from stdout. NF)Ú clearBufferÚ onOutReceivedÚ onProcessEndÚ_expectingCommandÚ _processEnded)r%rûr&r&r'r_ s zSFTPTestProcess.__init__cCsd|_g|_d|_dS)zR Clear any buffered data received from stdout. Should be private. r»N)ÚbufferÚ_linesReceivedÚ _lineBufferr1r&r&r'rúszSFTPTestProcess.clearBuffercCsrt d|¡|j| d¡}| d¡|_|j |¡|jdk rX|jd}|_| |¡|j |7_ |  ¡dS)zO Called by Twisted when the cftp client prints data to stdout. zgot %ró rÄN) rÚmsgrrÉrjrrÌrûÚcallbackrÿÚ_checkForCommand)r%r{Úlinesr”r&r&r'Ú outReceived s    zSFTPTestProcess.outReceivedcCs^d}|jrZ|j|krZd |j¡}| |¡r:|t|ƒd…}| ¡|jd}|_| |¡dS)Nscftp> r)rýrrØrÚ startswithrÏrúr)r%ÚpromptZbufr”r&r&r'r2s  z SFTPTestProcess._checkForCommandcCst d|¡dS)zO Called by Twisted when the cftp client prints data to stderr. zerr: %sN)rr)r%r{r&r&r'Ú errReceived=szSFTPTestProcess.errReceivedcCs|jS)zQ Return the contents of the buffer of data received from stdout. )rÿr1r&r&r'Ú getBufferDszSFTPTestProcess.getBuffercCs<t ¡|_| ¡t|tƒr&| d¡}|j |d¡|jS)a¬ Issue the given command via the cftp client. Return a C{Deferred} that fires when the server returns a result. Note that the C{Deferred} will callback even if the server returns some kind of error. @param command: A string containing an sftp command. @return: A C{Deferred} that fires when the sftp server returns a result. The payload is the server's response string. rÂr) rÚDeferredrýrúÚ isinstancerÚencoderorx©r%Úcommandr&r&r'Ú runCommandKs   zSFTPTestProcess.runCommandcs(t d¡‰‡‡fdd„|Dƒ}t |¡S)ax Run each command in sequence and return a Deferred that fires when all commands are completed. @param commands: A list of strings containing sftp commands. @return: A C{Deferred} that fires when all commands are completed. The payload is a list of response strings from the server, in the same order as the commands. rlcsg|]}ˆ ˆj|¡‘qSr&)Úrunr)Ú.0r©r%Zsemr&r'Ú jsz-SFTPTestProcess.runScript..)rZDeferredSemaphoreZ gatherResults)r%ÚcommandsZdlr&rr'Ú runScript^s zSFTPTestProcess.runScriptcCs,|jrt d¡St ¡|_|j d¡|jS)zø Kill the process if it is still running. If the process is still running, sends a KILL signal to the transport and returns a C{Deferred} which fires when L{processEnded} is called. @return: a C{Deferred}. NÚKILL)rþrryr rüroÚ signalProcessr1r&r&r'Ú killProcessns    zSFTPTestProcess.killProcesscCs(d|_|jr$|jd}|_| d¡dS)zF Called by Twisted when the cftp client process ends. TN)rþrür)r%Úreasonr”r&r&r'Ú processEnded~szSFTPTestProcess.processEndedN)r)r*r+r,r_rúrrr r rrrrr&r&r&r'rùs   rùc@s4eZdZdd„Zdd„Zdd„Zdd„Zd d „Zd S) ÚCFTPClientTestBasec Cs€tddƒ}| tj¡W5QRXtddƒ}| tj¡W5QRXt dd¡tddƒ}| dtj¡W5QRXt  |¡S)Nú dsa_test.pubr¼Údsa_testi€Úkh_tests 127.0.0.1 ) r¾rxr ZpublicDSA_opensshZprivateDSA_opensshr5ÚchmodZpublicRSA_opensshrr=©r%Úfr&r&r'r=Šs    zCFTPClientTestBase.setUpcCsFt|jƒ}t |¡}| t ¡¡t ¡}||_tj d|dd|_ dS)Nrz 127.0.0.1)Z interface) rôrõrZPortalZregisterCheckerr ZconchTestPublicKeyCheckerZConchTestServerFactoryrZ listenTCPÚserver)r%ZrealmÚpZfacr&r&r'Ú startServer•s   zCFTPClientTestBase.startServercCsHt|jjdƒs| d¡Sd|jjj_t |jjjjj ¡}|  |j¡|S)NÚprotorl) Úhasattrr$ÚfactoryÚ _cbStopServerr'ÚexpectedLoseConnectionrÚ maybeDeferredroZloseConnectionr’r“r&r&r'Ú stopServeržs   ÿ zCFTPClientTestBase.stopServercCst |jj¡Sr/)rr,r$Z stopListening©r%Úignoredr&r&r'r*¨sz CFTPClientTestBase._cbStopServercCs0dD] }zt |¡WqYqXqt |¡S)N)rrr )r5ÚremoverÚtearDownr"r&r&r'r1¬s zCFTPClientTestBase.tearDownN)r)r*r+r=r&r-r*r1r&r&r&r'r‰s    rc@sºeZdZdZdd„Zdd„Zdd„Zdd „Zd d „Zd d „Z dd„Z dd„Z dd„Z d-dd„Z dd„Zdd„Zdd„Zdd„Zdd „Zd!d"„Zd#d$„Zd%d&„Zd'd(„Zd)d*„Zd+d,„ZdS).ÚOurServerCmdLineClientTestszí Functional tests which launch a SFTP server over TCP on localhost and check cftp command line interface using a spawned process. Due to the spawned process you can not add a debugger breakpoint for the client code. c s,t ˆ¡ˆ ¡d}ˆj ¡j}tj|| ¡dd}t   dt j |f¡t  ¡}t|ƒˆ_| ‡fdd„¡tj ¡}tj t j¡|d<g}i}|D]"}t|tƒr°| d¡}| |¡q˜|D]<}||} t|tƒrà| d¡}t| tƒrô|  d¡} | ||<qÀt   |¡t   |¡tjˆjt j ||d |S) Nz‚-p %i -l testuser --known-hosts kh_test --user-authentications publickey --host-key-algorithms ssh-rsa -i dsa_test -a -v 127.0.0.1r©Úmodú running %s %scs ˆj ¡Sr/)ÚprocessProtocolrú©Ú_r1r&r'röÐr»z3OurServerCmdLineClientTests.setUp..Ú PYTHONPATHr©Úenv)rr=r&r$ÚgetHostÚportr Ú _makeArgsrÉrrrrrr rùr6r’r5r6ÚcopyÚpathseprØrfr rrrÊrZ spawnProcess) r%Úcmdsr=r”r;Z encodedCmdsZ encodedEnvÚcmdÚvarÚvalr&r1r'r=¿s<              ÿz!OurServerCmdLineClientTests.setUpcsˆ ¡}| ‡fdd„¡|S)Ncs ˆj ¡Sr/)r6rr7r1r&r'röér»z6OurServerCmdLineClientTests.tearDown..)r-r’r“r&r1r'r1çsz$OurServerCmdLineClientTests.tearDowncCs.z|jj d¡Wntjk r(YnXdS)Nr)r6rorrZProcessExitedAlreadyr.r&r&r'Ú _killProcessísz(OurServerCmdLineClientTests._killProcesscCs |j |¡S)z¼ Run the given command with the cftp client. Return a C{Deferred} that fires when the command is complete. Payload is the server's output for that command. )r6rrr&r&r'rôsz&OurServerCmdLineClientTests.runCommandcGs |j |¡S)zÓ Run the given commands with the cftp client. Returns a C{Deferred} that fires when the commands are all complete. The C{Deferred}'s payload is a list of output for each command. )r6r)r%rr&r&r'rýsz%OurServerCmdLineClientTests.runScriptcCsL|j}| ddddd¡}dd„}| |¡| |j|jt ¡d|jg¡|S)zý Test that 'pwd' reports the current remote directory, that 'lpwd' reports the current local directory, and that changing to a subdirectory then changing to its parent leaves you in the original remote directory. ÚpwdZlpwdzcd testDirectoryúcd ..cSsHg}|D]&}tr$t|tƒr$| d¡}| |¡q|dd…|dd…S)zH Callback function for handling command output. rÂNéé)rr ÚbytesrÈrÊ)rÑrArBr&r&r'Ú cmdOutputs   z8OurServerCmdLineClientTests.testCdPwd..cmdOutputr–)rõrr’rGrfr5Úgetcwd)r%ZhomeDirr”rKr&r&r'Ú testCdPwds ÿz%OurServerCmdLineClientTests.testCdPwdcs&‡fdd„}ˆ dddd¡}| |¡S)zŒ Check that 'ls -l' output includes the access permissions and that this output changes appropriately with 'chmod'. csZˆ ¡ˆ |d d¡¡ˆ |dd¡ˆ |d d¡|d¡ˆ |dd¡dS)Nrs -rw-r--r--rlr»rÇs ----------rH)ÚflushLoggedErrorsr$rrG©Zresultsr1r&r'Ú_check's z7OurServerCmdLineClientTests.testChAttrs.._checkzls -l testfile1zchmod 0 testfile1zchmod 644 testfile1©rr’©r%rPr”r&r1r'Ú testChAttrs"s  ÿz'OurServerCmdLineClientTests.testChAttrscs@‡fdd„}ˆ ddˆj ¡ddd¡}| dd „¡| |¡S) zŽ Check 'ls' works as expected. Checks for wildcards, hidden files, listing directories and listing empty directories. cspˆ |dddddg¡ˆ |dddddg¡ˆ |dddg¡ˆ |dd ddg¡ˆ |d d g¡dS) Nró testDirectoryótestRemoveFileótestRenameFileó testfile1rlrÇrHs.testHiddenFilerIr»©rGrOr1r&r'rP9sÿÿÿz4OurServerCmdLineClientTests.testList.._checkrzls ../zls *Filez ls -a *Filezls -l testDirectorycSsdd„|DƒS)NcSsg|]}| d¡‘qS)r)rÉ)rÚxr&r&r'rDszJOurServerCmdLineClientTests.testList....r&)Zxsr&r&r'röDr»z6OurServerCmdLineClientTests.testList..)rrõrÙr’rRr&r1r'ÚtestList4s ÿz$OurServerCmdLineClientTests.testListcCsD| d¡}t d¡ d¡ ¡}t|tƒr2| d¡}| |j |¡|S)zB Check that running the '?' command returns help. ú?Nr–rÂ) rrr„Zcmd_HELPrËr rrr’rG)r%r”ZhelpTextr&r&r'ÚtestHelpHs    z$OurServerCmdLineClientTests.testHelpNcCs| | ¡| ¡|¡dS)zg Assert that the files at C{name1} and C{name2} contain exactly the same data. N)rGZ getContent)r%Zname1Zname2rr&r&r'ÚassertFilesEqualUsz,OurServerCmdLineClientTests.assertFilesEqualcsjdˆjjˆjjf‰tˆtƒr(ˆ d¡‰‡‡fdd„}ˆ dˆjjf¡}| |¡| ‡fdd„¡|S)z³ Test that 'get' saves the remote file to the correct local location, that the output of 'get' is correct and that 'rm' actually removes the file. z)Transferred %s/testfile1 to %s/test file2rÂcs8ˆ | ˆ¡¡ˆ ˆj d¡ˆj d¡d¡ˆ d¡S)NÚ testfile1ú test file2z get failedzrm "test file2")r$Úendswithr]rõÚchildr©r¯©rÒr%r&r'Ú _checkGeths  þz6OurServerCmdLineClientTests.testGet.._checkGetzget testfile1 "%s/test file2"csˆ ˆj d¡ ¡¡S)Nr_©Z assertFalserõraÚexistsr7r1r&r'röqsÿz5OurServerCmdLineClientTests.testGet..)rõrfr rrrr’)r%rdr”r&rcr'ÚtestGet]sÿ   z#OurServerCmdLineClientTests.testGetcs ‡fdd„}ˆ d¡}| |¡S)zQ Test that 'get' works correctly when given wildcard parameters. cs8ˆ ˆj d¡tdƒd¡ˆ ˆj d¡tdƒd¡dS)NÚtestRemoveFileútestRemoveFile get failedÚtestRenameFileútestRenameFile get failed)r]rõrar)r/r1r&r'rPzsþþz;OurServerCmdLineClientTests.testWildcardGet.._checkz get testR*)rr’rRr&r1r'ÚtestWildcardGetvs  z+OurServerCmdLineClientTests.testWildcardGetcsfdˆj ¡jdˆj ¡jd‰‡‡fdd„}ˆ dˆjjf¡}| |¡| ‡fdd„¡|S) z— Check that 'put' uploads files correctly and that they can be successfully removed. Also check the output of the put command. s Transferred s/testfile1 to s /test"file2cs6ˆ ˆj d¡ˆj d¡¡ˆ | ˆ¡¡ˆ d¡S)Nr^ú test"file2zrm "test\"file2")r]rõrar$r`rrbrcr&r'Ú _checkPuts  ÿz6OurServerCmdLineClientTests.testPut.._checkPutzput %s/testfile1 "test\"file2"csˆ ˆj d¡ ¡¡S)Nrmrer7r1r&r'rö˜sÿz5OurServerCmdLineClientTests.testPut..)rõÚ asBytesModerfrr’)r%rnr”r&rcr'ÚtestPut†sÿ ÿþÿ z#OurServerCmdLineClientTests.testPutc s‚ˆj d¡jdd}| d¡W5QRXˆj d¡jdd}| d¡W5QRX‡fdd„}ˆ d ˆjjf¡}| |¡|S) zb Check that 'put' uploads files correctly when overwriting a longer file. Ú shorterFileÚw©ÚmodeóaÚ longerFileóbbcs ˆ ˆj d¡ˆj d¡¡dS)Nrqrv)r]rõrarbr1r&r'rn§s ÿzEOurServerCmdLineClientTests.test_putOverLongerFile.._checkPutzput %s/shorterFile longerFile)rõrar¾rxrrfr’©r%r#rnr”r&r1r'Útest_putOverLongerFiles ÿ z2OurServerCmdLineClientTests.test_putOverLongerFilec s–ˆj d¡‰ˆ ¡ˆ d¡jdd}| d¡W5QRXˆj d¡jdd}| d¡W5QRX‡‡fdd„}ˆ d ˆjjf¡}| |¡|S) z˜ Check that 'put' uploads files correctly when overwriting a longer file and you use a wildcard to specify the files to upload. ÚdirrÀrrrsrurwcsˆ ˆ d¡ˆj d¡¡dS)NrÀ)r]rarõrb©r%ZsomeDirr&r'rn½s  ÿzMOurServerCmdLineClientTests.test_putMultipleOverLongerFile.._checkPutz put %s/dir/*)rõraZcreateDirectoryr¾rxrrfr’rxr&r{r'Útest_putMultipleOverLongerFile±s ÿ z:OurServerCmdLineClientTests.test_putMultipleOverLongerFilecs<‡fdd„}ˆ ddˆjjfdˆj ¡¡}| |¡|S)zÍ What happens if you issue a 'put' command and include a wildcard (i.e. '*') in parameter? Check that all files matching the wildcard are uploaded to the correct directory. cshˆ |dd¡ˆ |dd¡ˆ ˆj d¡ˆj ¡ d¡d¡ˆ ˆj d¡ˆj ¡ d¡d¡dS)Nrr»rÇrhrirjrk)rGr]rõrarìrOr1r&r'ÚcheckÍsþþz:OurServerCmdLineClientTests.testWildcardPut..checkrGz put %s/testR*zcd %s)rrõrfrÙr’)r%r}r”r&r1r'ÚtestWildcardPutÇs   þ z+OurServerCmdLineClientTests.testWildcardPutcs4‡fdd„}ˆ dd¡}| |¡| ˆjd¡|S)z  Test that 'ln' creates a file which appears as a link in the output of 'ls'. Check that removing the new file succeeds without output. cs8ˆ ¡ˆ |dd¡ˆ |d d¡d¡ˆ d¡S)Nrr»rlólz link failedz rm testLink)rNrGr$rrrOr1r&r'rPäsz4OurServerCmdLineClientTests.testLink.._checkzln testLink testfile1zls -l testLinkr»©rr’rGrRr&r1r'ÚtestLinkßs    z$OurServerCmdLineClientTests.testLinkcs4‡fdd„}ˆ dd¡}| |¡| ˆjd¡|S)zV Test that we can create and remove directories with the cftp client. cs.ˆ |dd¡ˆ |d d¡¡ˆ d¡S)Nrr»rlódzrmdir testMakeDirectory)rGr$rrrOr1r&r'rPôsz?OurServerCmdLineClientTests.testRemoteDirectory.._checkúmkdir testMakeDirectoryzls -l testMakeDirector?r»r€rRr&r1r'ÚtestRemoteDirectoryðs ÿ z/OurServerCmdLineClientTests.testRemoteDirectorycs&‡fdd„}ˆ dd¡}| |¡|S)z— Test that a C{mkdir} on an existing directory fails with the appropriate error, and doesn't log an useless error server side. cs$ˆ |dd¡ˆ |dd¡dS)Nrr»rlsremote error 11: mkdir failedrXrOr1r&r'rPs ÿzHOurServerCmdLineClientTests.test_existingRemoteDirectory.._checkrƒrQrRr&r1r'Útest_existingRemoteDirectorys  ÿ z8OurServerCmdLineClientTests.test_existingRemoteDirectorycsFˆ dˆjjf¡}| ˆjd¡| ‡fdd„¡| ˆjd¡|S)z Test that we can create a directory locally and remove it with the cftp client. This test works because the 'remote' server is running out of a local directory. zlmkdir %s/testLocalDirectoryr»cs ˆ d¡S)Nzrmdir testLocalDirectory)rr7r1r&r'rör»z@OurServerCmdLineClientTests.testLocalDirectory..)rrõrfr’rGr“r&r1r'ÚtestLocalDirectorys z.OurServerCmdLineClientTests.testLocalDirectorycs4‡fdd„}ˆ dd¡}| |¡| ˆjd¡|S)z1 Test that we can rename a file. cs*ˆ |dd¡ˆ |dd¡ˆ d¡S)Nrr»rls testfile2zrename testfile2 testfile1)rGrrOr1r&r'rP!sz6OurServerCmdLineClientTests.testRename.._checkzrename testfile1 testfile2z ls testfile?r»r€rRr&r1r'Ú testRenames    z&OurServerCmdLineClientTests.testRename)N)r)r*r+r,r=r1rErrrMrSrZr\r]rgrlrpryr|r~rr„r…r†r‡r&r&r&r'r2¶s,(    r2c@s@eZdZdZdd„Zdd„Zdd„Zdd „Zd d „Zd d „Z dS)ÚOurServerBatchFileTestszm Functional tests which launch a SFTP server over localhost and checks csftp in batch interface. cCst |¡| ¡dSr/©rr=r&r1r&r&r'r=3s zOurServerBatchFileTests.setUpcCst |¡| ¡Sr/)rr1r-r1r&r&r'r18s z OurServerBatchFileTests.tearDownc sÌ| ¡‰tˆdƒ}| |¡W5QRX|j ¡j}d|ˆf}tj| ¡dddd…}t   dt j |f¡t j ¡}t j t j¡|d<d|jj_tt j ||d}‡fd d „}| d d „¡| |¡|S) Nrrzˆ-p %i -l testuser --known-hosts kh_test --user-authentications publickey --host-key-algorithms ssh-rsa -i dsa_test -a -v -b %s 127.0.0.1rr3rlr5r9r:cst ˆ¡|Sr/)r5r0©Úres©Úfnr&r'Ú_cleanupRs z9OurServerBatchFileTests._getBatchOutput.._cleanupcSs|dS)Nrr&rŠr&r&r'röVr»z9OurServerBatchFileTests._getBatchOutput..)r½r¾rxr$r<r=r r>rÉrrrrr5r6r?r@rØrfr)r+rr’ZaddBoth)r%r#Úfpr=rAr;r”rŽr&rŒr'Ú_getBatchOutput=s"  ú    z'OurServerBatchFileTests._getBatchOutputcs(d}‡fdd„}ˆ |¡}| |¡|S)zš Test whether batch file function of cftp ('cftp -b batchfile'). This works by treating the file as a list of commands to be run. z pwd ls exit csT| d¡}t dt|ƒ¡ˆ ˆj ¡j|d¡ˆ |dd…dddd g¡dS) NrzRES %srlrHéþÿÿÿrTrUrVrW) rÉrrÚreprÚassertInrõrorfrGrŠr1r&r'Ú_cbCheckResultfs ÿz=OurServerBatchFileTests.testBatchFile.._cbCheckResult©rr’©r%rAr”r”r&r1r'Ú testBatchFile\s    z%OurServerBatchFileTests.testBatchFilecs(d}‡fdd„}ˆ |¡}| |¡|S)zO Test that an error in the batch file stops running the batch. zchown 0 missingFile pwd exit csˆ ˆj ¡j|¡dSr/)Z assertNotInrõrorfrŠr1r&r'r”{sz9OurServerBatchFileTests.testError.._cbCheckResultr•r–r&r1r'Ú testErrorrs    z!OurServerBatchFileTests.testErrorcs(d}‡fdd„}ˆ |¡}| |¡|S)z_ Test that a minus sign '-' at the front of a line ignores any errors. z-chown 0 missingFile pwd exit csˆ ˆj ¡j|¡dSr/)r“rõrorfrŠr1r&r'r”Œsz@OurServerBatchFileTests.testIgnoredError.._cbCheckResultr•r–r&r1r'ÚtestIgnoredErrorƒs    z(OurServerBatchFileTests.testIgnoredErrorN) r)r*r+r,r=r1rr—r˜r™r&r&r&r'rˆ-srˆc@s(eZdZdZdd„Zdd„Zdd„ZdS) ÚOurServerSftpClientTestsz@ Test the sftp server against sftp command line client. cCst |¡| ¡Sr/r‰r1r&r&r'r=šs zOurServerSftpClientTests.setUpcCs| ¡Sr/)r-r1r&r&r'r1Ÿsz!OurServerSftpClientTests.tearDownc s¤ˆ ¡‰tˆdƒ}| d¡W5QRXˆj ¡j‰tj‰‡fdd„}ˆ td|¡dˆjj _ t ddƒ}‡‡fd d „}‡fd d „}|  |¡|  d d„¡|  |¡S)a Test the return of extended attributes by the server: the sftp client should ignore them, but still be able to parse the response correctly. This test is mainly here to check that L{filetransfer.FILEXFER_ATTR_EXTENDED} has the correct value. rrz ls . exitcsˆ||ƒ}d|d<|S)NrŠZext_foor&)r%Úsrs)Ú oldGetAttrr&r'Ú _getAttrs±s zCOurServerSftpClientTests.test_extendedAttributes.._getAttrsrTZssh)ú-oúPubkeyAcceptedKeyTypes=ssh-dssz-Vcs>|dkrd}nd}|dddddddd dd ˆfd ˆd f 7}|S) Nr)ržrŸr&z-Fz /dev/nullržzIdentityFile=dsa_testzUserKnownHostsFile=kh_testzHostKeyAlgorithms=ssh-rsazPort=%iz-bztestuser@127.0.0.1r&)ZstatusÚargs)rr=r&r'ÚhasPAKT¿s"üzAOurServerSftpClientTests.test_extendedAttributes..hasPAKTcs.ˆ |dd¡dD]}ˆ ||d¡qdS)NrÇr)rTrUrVrW)rGr“)r¯Úir1r&r'r}Îsz?OurServerSftpClientTests.test_extendedAttributes..checkcSs td|ƒS)NÚsftp)r)r r&r&r'röÔr»zBOurServerSftpClientTests.test_extendedAttributes..) r½r¾rxr$r<r=r rr9r)r+rr’)r%r#rr”r¡r}r&)rrœr=r%r'Útest_extendedAttributes£s"    ÿ  z0OurServerSftpClientTests.test_extendedAttributesN)r)r*r+r,r=r1r¤r&r&r&r'rš•sršz4don't run w/o spawnProcess or cryptography or pyasn1)Úwhichr£z%no sftp command-line client available)Sr,rRr3rr5r;rŒr›ÚiorZtwisted.python.filepathrZtwisted.python.reflectrZzope.interfacerrrZunixZ_reasonZtwisted.conch.scriptsrZtwisted.conch.scripts.cftpr Ztwisted.conch.sshr Z$twisted.conch.test.test_filetransferr Ztwisted.conch.testr r Ztwisted.conch.test.test_conchrÚ ImportErrorZ twisted.conchrZtwisted.conch.interfacesrrrZ twisted.credrZtwisted.internetrrrrrZtwisted.internet.utilsrrZtwisted.pythonrZtwisted.python.compatrrZtwisted.python.fakepwdrZtwisted.test.proto_helpersrZtwisted.internet.taskr Ztwisted.trial.unittestr!r"r-r£r\rcrkrur€rôZProcessProtocolrùrr2rˆršZIReactorProcessrZZtwisted.python.procutilsr¥r&r&r&r'Ús†0                  )* -yhD ÿ