tf?dZddlZddlZddlZddlZddlZddlZddlZddlm Z m Z ddl m Z ddl mZddlmZddlmZddlmZmZdd lmZdd lmZmZmZmZmZdd lmZdd l m!Z!dd l"m#Z#m$Z$ddl%m&Z&ddl'm(Z(ddl)m*Z*ddl+m,Z,ddl-m.Z.ddl/m0Z0e.dZ1e.dZ2e1r9 ddl3m4Z4ddl5m6Z6ddl7m8Z8ddl9m:Z:ddl;mm?Z?ddlm@Z@n #eA$rYnwxYwdZBde2e1ejCedfvrdZBe eBdGd d!e0ZDGd"d#e0ZEGd$d%e!ZFGd&d'ZGGd(d)ZHeeGd*d+e ZIe eBdGd,d-e0ZJGd.d/ZKGd0d1ejLZMGd2d3eZNe eBdGd4d5eNZOe eBdGd6d7eNZPe eBde e,d8 d9e e,d: d;Gd<d=eNZQdS)>z* Tests for L{twisted.conch.scripts.cftp}. N)BytesIO TextIOWrapper)skipIf) implementer)ls) ISFTPFile)FileTransferTestAvatar SFTPTestBase)portal)defererror interfacesprotocolreactor)Clock)StringTransport)getProcessOutputAndValuegetProcessValue)log) UserDatabase)FilePath)which) requireModule)TestCase cryptographyztwisted.conch.unix)cftp) SSHSession) filetransfer)EXTENDED_DATA_STDERR) test_conchtest_ssh) FakeStdio)FileTransferForTestAvatarFTz*don't run w/o spawnProcess or cryptographyc*eZdZdZddZddZddZdS)SSHSessionTestsz= Tests for L{twisted.conch.scripts.cftp.SSHSession}. returnNct|_t|_|j|j_t |_t |j|_|j|j_dSN)r"stdiorchannelr stderrBufferrstderrselfs [/var/www/surfInsights/venv3-11/lib/python3.11/site-packages/twisted/conch/test/test_cftp.pysetUpzSSHSessionTests.setUp>sT[[ !|| !Z #II#D$566 "k cv|j||jjdS)z| L{twisted.conch.scripts.cftp.SSHSession.eofReceived} loses the write half of its stdio connection. N)r* eofReceived assertTruer) writeConnLostr-s r/test_eofReceivedz SSHSessionTests.test_eofReceivedFs5   """  011111r1cd}|d}|jt|dz||j|dzdS)z L{twisted.conch.scripts.cftp.SSHSession.extReceived} decodes stderr data using UTF-8 with the "backslashescape" error handling and writes the result to its own stderr. u☃utf-8s\xffN)encoder* extReceivedr assertEqualr+getvalue)r. errorText errorBytess r/test_extReceivedStderrz&SSHSessionTests.test_extReceivedStderrNs " %%g..            & & ( (  !     r1)r&N)__name__ __module__ __qualname____doc__r0r6r@r1r/r%r%8sZ****2222      r1r%cZeZdZdZeedddZdZdZdZ dZ d Z e j Z e je jd d Zn#e j$rd ZYnwxYwe je jen#e je jewxYweed dZdZdS) 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.tzsetc^d_fd}td|dtjvr]t jtjdtjdtj dSd}|dS)zo Patch the L{ls} module's time function so the results of L{lsLine} are deterministic. i[cjSr()nowr-sr/fakeTimez$ListingTests.setUp..fakeTimeps 8Or1timeTZcj tjd=n#t$rYnwxYwtjdS)NrN)osenvironKeyErrorrMrHrEr1r/cleanupz#ListingTests.setUp..cleanup|sC 4((D s  N) rKpatchrrPrQ addCleanupoperatorsetitemrMrH)r.rLrSs` r/r0zListingTests.setUpis       2vx((( 2:   OOH,bj$ 4@P Q Q Q OODJ ' ' ' ' '    OOG $ $ $ $ $r1cp|tjd<tjt jd|S)zl Call L{ls.lsLine} after setting the timezone to C{timezone} and return the result. rNfoo)rPrQrMrHrlsLine)r.timezonestats r/ _lsInTimezonezListingTests._lsInTimezones-$ 4 y%%%r1c |jdz }tjdddddddd|df }||d|d||d|ddS)z A file with an mtime six months (approximately) or more in the past has a listing including a low-resolution timestamp. rAmerica/New_Yorkz;!--------- 0 0 0 0 Apr 26 1973 fooPacific/Aucklandz;!--------- 0 0 0 0 Apr 27 1973 fooNrKrP stat_resultr<r]r.thenr\s r/ test_oldFilezListingTests.test_oldFile x01~q!Q1aAtQ?@@    14 8 8 I       14 8 8 I     r1c |jdz dz}tjdddddddd|df }||d|d||d|ddS) 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. r_irr`z;!--------- 0 0 0 0 May 01 1973 fooraz;!--------- 0 0 0 0 May 02 1973 fooNrbrds r/test_oldSingleDigitDayOfMonthz*ListingTests.test_oldSingleDigitDayOfMonthx015EF~q!Q1aAtQ?@@    14 8 8 I       14 8 8 I     r1c |jdz }tjdddddddd|df }||d|d||d|ddS)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. 逛zrr`;!--------- 0 0 0 0 Aug 28 17:33 foora;!--------- 0 0 0 0 Aug 29 09:33 fooNrbrds r/ test_newFilezListingTests.test_newFilergr1 es_AR.UTF8FTz'The es_AR.UTF8 locale is not installed.c |jdz }tjdddddddd|df }tj}tjtjd|tjtj||| d|d|| d|ddS) zC The month name in the date is locale independent. rmrrqr`rnraroN) rKrPrclocale getlocale setlocaleLC_ALLrUr<r])r.rer\ currentLocales r/test_localeIndependentz#ListingTests.test_localeIndependents x01~q!Q1aAtQ?@@(**  555 (&-GGG    14 8 8 I       14 8 8 I     r1c |jdz dz}tjdddddddd|df }||d|d||d|ddS) rirmiFrr`z;!--------- 0 0 0 0 Sep 01 17:33 fooraz;!--------- 0 0 0 0 Sep 02 09:33 fooNrbrds r/test_newSingleDigitDayOfMonthz*ListingTests.test_newSingleDigitDayOfMonthrkr1)rArBrCrDgetattrrMskipr0r]rfrjrprsrtrwrurv localeSkipErrorrrxrzrEr1r/rGrG`sY wtWd##+I%%%: & & &   $   (   (%F$&&M7  F V]L 9 9 9JJ|   JJJ   6666 6666 VJABB  CB ,     s)A A8 AA8AA88BrGc"eZdZdZfdZxZS)InMemorySSHChannelzp Minimal implementation of a L{SSHChannel} like class which only reads and writes data from memory. cd||_d|_tdS)zt @param conn: The SSH connection associated with this channel. @type conn: L{SSHConnection} rN)conn localClosedsuper__init__)r.r __class__s r/rzInMemorySSHChannel.__init__s0   r1)rArBrCrDr __classcell__)rs@r/rrsB r1rc$eZdZdZdZdZdZdS)FilesystemAccessExpectationszC A test helper used to support expected filesystem access. ci|_dSr(_cacher-s r/rz%FilesystemAccessExpectations.__init__s  r1c||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} Nr)r.pathflagsstreams r/putz FilesystemAccessExpectations.puts&, T5M"""r1c:|j||fS)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} )rpop)r.rrs r/rz FilesystemAccessExpectations.pop(s{e}---r1N)rArBrCrDrrrrEr1r/rrsK , , , . . . . .r1rceZdZdZdZdZdS)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} cPt||_ddd|_||_dS)N )requests buffersize)r transportoptions_availableFiles)r.availableFiless r/rzInMemorySFTPClient.__init__Bs5+D11    .r1c8|j||S)z} @see: L{filetransfer.FileTransferClient.openFile}. Retrieve and remove cached file based on flags. )rr)r.filenamerattrss r/openFilezInMemorySFTPClient.openFileJs #''%888r1N)rArBrCrDrrrEr1r/rr8s<...99999r1rc<eZdZdZdZdZdZdZdZdZ dZ d S) InMemoryRemoteFilez; An L{ISFTPFile} which handles all data in memory. c<||_tj|dS)zL @param name: Name of this file. @type name: L{str} N)namerr)r.rs r/rzInMemoryRemoteFile.__init__Ys"  r1c~||||tj|S)z/ @see: L{ISFTPFile.writeChunk} )seekwriter succeed)r.startdatas r/ writeChunkzInMemoryRemoteFile.writeChunkas8 % 4}T"""r1cd|_dS)zo @see: L{ISFTPFile.writeChunk} Keeps data after file was closed to help with testing. TN)_closedr-s r/closezInMemoryRemoteFile.closeis  r1cdSr(rEr-s r/getAttrszInMemoryRemoteFile.getAttrsq r1cdSr(rE)r.offsetlengths r/ readChunkzInMemoryRemoteFile.readChunkurr1cdSr(rE)r.rs r/setAttrszInMemoryRemoteFile.setAttrsyrr1c*tj|S)zb Get current data of file. Allow reading data event when file is closed. )rr=r-s r/r=zInMemoryRemoteFile.getvalue}s %%%r1N) rArBrCrDrrrrrrr=rEr1r/rrSs###         &&&&&r1rceZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZddZddZdZdZdZdZdS)StdioClientTestsz( Tests for L{cftp.StdioClient}. c>t|_t|j}tj||_d|j_tx|_|j_ | dd|jjj |j_ dS)zm Create a L{cftp.StdioClient} hooked up to dummy transport and a fake user database. /iN) rfakeFilesystemrr StdioClientclientcurrentDirectoryrdatabase_pwdsetKnownConsoleSizer)r. sftpClients r/r0zStdioClientTests.setUps ;<<'(;<< &z22 '* $+7>>9  (   b)))!% 2 < r1c |jtjdt jdddt j|j d}| |j d|S)zs The I{exec} command runs its arguments locally in a child process using the user's shell. secretrYbarzexec print(1 + 2)s3 ) raddUsergetpassgetuserrPgetuidsys executabler_dispatchCommand addCallbackr<r.ds r/ test_execzStdioClientTests.test_execsq  O  xdE5#.    K ( ()< = = d&///r1c |jtjdt jdddd|jd}||j d|S)zr If the local user has no shell, the I{exec} command runs its arguments using I{/bin/sh}. rrrYrzexec echo hellohello rrrrrPrrrrr<rs r/test_execWithoutShellz&StdioClientTests.test_execWithoutShellso  O  xdE5"    K ( (): ; ; d& 333r1c |jtjdt jdddd|jd}||j d|S)zO The I{exec} command is run for lines which start with C{"!"}. rrrYrz/bin/shz !echo hellorrrs r/ test_bangzStdioClientTests.test_bangsn  O  xdE5)    K ( ( 7 7 d& 333r1c~ddlGfdd}|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} rNceZdZfdZdS)7StdioClientTests.setKnownConsoleSize..FakeFcntlct|jkr|dtjdddS)Nz#Only window-size queries supported.4Hr) TIOCGWINSZfailstructpack)r.fdoptmutateheightttywidths r/ioctlz=StdioClientTests.setKnownConsoleSize..FakeFcntl.ioctls;#.((IICDDD{41===r1N)rArBrCr)rrrsr/ FakeFcntlrs8 > > > > > > > > >r1rfcntl)rrTr)r.rrrrs `` @r/rz$StdioClientTests.setKnownConsoleSizesp   > > > > > > > > > > > > 4))++.....r1c|ddtx}|j_t d}d|_t j|}d|_| }| d|xj dz c_ |j ||d}| |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. r"xsamplei(g@is# b'sample' 40% 4.0kB 2.0kBps 00:03 N)rrrrrrr FileWrappersizesecondsadvancetotal_printProgressBarr<rvaluer.clockwrappedwrapper startTimeresults r/test_printProgressBarReportingz/StdioClientTests.test_printProgressBarReportings   R(((&+gg- #$--  "7++  MMOO  c   %%gy9998 .4466?????r1c|ddtx}|j_t d}d|_t j|}|}|j ||d}| |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. rrrrs! b'sample' 0% 0.0B 0.0Bps 00:00 N) rrrrrrrrrrr<rrrs r/test_printProgressBarNoProgressz0StdioClientTests.test_printProgressBarNoProgresss   R(((&+gg- #$--  "7++MMOO  %%gy9996 .4466?????r1c"|ddt}d|_tj|}|j|dd}|||jj dS)z5 Print the progress for empty files. rrs empty-filers% b'empty-file'100% 0.0B 0.0Bps 00:00 N) rrrrrrrr<rr)r.rrrs r/test_printProgressBarEmptyFilez/StdioClientTests.test_printProgressBarEmptyFiles   R((())$ "7++ %%gq111: !6!!>?????r1cf|jd}|d|dS)zK Returns empty value for both filename and remaining data. z )rrNr _getFilenamer<r.rs r/test_getFilenameEmptyz&StdioClientTests.test_getFilenameEmptys5))$// 6*****r1cf|jd}|d|dS)zd Returns empty value for remaining data when line contains only a filename. only-local)r rNrr s r/test_getFilenameOnlyLocalz*StdioClientTests.test_getFilenameOnlyLocals6 )),77 +V44444r1cf|jd}|d|dS)ze Returns filename and remaining data striped of leading and trailing spaces. z local remote file )localz remote fileNrr s r/test_getFilenameNotQuotedz*StdioClientTests.test_getFilenameNotQuoted#s7 ))*ABB 16:::::r1cf|jd}|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 "Nrr s r/test_getFilenameQuotedz'StdioClientTests.test_getFilenameQuoted,s7 ))*NOO =vFFFFFr1Nr1c||}t|d5}||dddn #1swxYwY|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. Nwb)mktempopenr)r.rcontentfiles r/makeFilezStdioClientTests.makeFile5s <;;==D $    JJw                   sA  A A Fc|jj}|d}|d}g}g}|D]\}}}g} |D]} | |d| | d|d||| |ddd} | dd } | d d d } | | g}| dd D]} |  dd d} | dd } || dd| d || d |||r1| 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 r8z  z Transferred z to r N rz5There are still put responses which were not checked.) rrrdecodesplitappendrstripextendrsplitr<sortedlen)r. transfers randomOrderoutputexpectedOutput actualOutputrremoteexpectedexpectedTransferline progressPartsactuallastactualTransfers r/checkPutMessagez StdioClientTests.checkPutMessageIs&,,..w''f%% '0 0 0 #E68 "   ; ; ''5(9(94(9(9::::  # #$F5$F$Ff$F$F G G G  ! !"2 3 3 3"JJqMM//55;;DAAM"3B3'F $**40066t<r5r4rr<r=)r.r?rr@rArBs r/test_cmd_PUTSingleRemotePathz-StdioClientTests.test_cmd_PUTSingleRemotePaths MMOO &)??,BXX# ' 33   E5=3L3LMMM;&&)'J'Jj'J'J'JKK X&&& y*{mDEFFF  *+++ j113344444r1cN|}tj|}d}tj|}|tj||}t jt jzt j z}d|}d|}t|} t|} |j ||tj| |j ||tj| |jtj|d} || || j|d| || j|d| |||dgf||dgfgdd S) z When a gobbing expression is used local files are transferred with remote file names based on local names. second-namerr*r1rETr)N)rrPrr<dirnamer;rr8r9r:rrrr rrr=r>r4rr<r=r5) r.first firstName secondNameparentsecondrfirstRemotePathsecondRemotePathfirstRemoteFilesecondRemoteFilerBs r/ test_cmd_PUTMultipleNoRemotePathz1StdioClientTests.test_cmd_PUTMultipleNoRemotePaths  G$$U++ " ''BGLL$D$DEE&)??,BXX)i//+z++,_==-.>??  o8V8VWWW  eU]3C%D%D   ;&&rw||FC'@'@AA X&&& /000 o6688999 (0111 .7799::: +7)K=9       r1ct|}tj|}d}tj|}|tj||}t jt jzt j z}t|}t|}d|} d|} |j | |tj||j | |tj||jdtj|d} || ||j|d|||j|d||||dgf||dgfgd 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. rHrIz/remote/z {} remoterJr1rETrKN)rrPrr<rLr;rr8r9r:rrrr rrr=formatr>r4rr<r=r5) r.rMrNrOrPrQrrTrUrRrSrBs r/"test_cmd_PUTMultipleWithRemotePathz3StdioClientTests.test_cmd_PUTMultipleWithRemotePaths G$$U++ " ''BGLL$D$DEE&)??,BXX,Y77-j990Y002j22  o8V8VWWW  eU]3C%D%D   ;&&{'9'9"',,vs:S:S'T'TUU X&&& /000 o6688999 (0111 .7799:::  K=1k]3       r1)Nr1)F)rArBrCrDr0rrrrrrrr r rrrr5rCrFrVrYrEr1r/rrsD===&         ///*@@@0@@@$ @ @ @+++555;;;GGG(; ; ; ; z   0555&! ! ! F# # # # # r1rceZdZdZdZdS)FileTransferTestRealmc||_dSr()testDir)r.r]s r/rzFileTransferTestRealm.__init__s  r1cBt|j}|d|dfS)NrcdSr(rErEr1r/z5FileTransferTestRealm.requestAvatar..sr1)r r])r.avatarIDmindras r/ requestAvatarz#FileTransferTestRealm.requestAvatars$ "4< 0 0!}a--r1N)rArBrCrrdrEr1r/r[r[s2.....r1r[cNeZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d S) 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). cf|||_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.ris r/rzSFTPTestProcess.__init__ s< * !%"r1c0d|_g|_d|_dS)zR Clear any buffered data received from stdout. Should be private. r1N)buffer_linesReceived _lineBufferr-s r/rhzSFTPTestProcess.clearBuffers!  r1cttjd|z|j|zd}|d|_|j||j$|jdc}|_|||xj |z c_ | dS)zO Called by Twisted when the cftp client prints data to stdout. zgot %r rN) rmsgrpr!rror$ricallbackrn_checkForCommand)r.rlinesrs r/ outReceivedzSFTPTestProcess.outReceiveds 4   !D(//66 99R== ""5)))   )$($6 !At! JJt    t  r1c2d}|jr|j|krd|j}||r|t |d}||jdc}|_||dSdSdS)Nscftp> rr)rkrpr;ro startswithr'rhrt)r.promptbufrs r/ruz SFTPTestProcess._checkForCommand.s  ! d&6&&@&@**T011C~~f%% )#f++--(      (,(> %At% JJsOOOOO   &@&@r1c4tjd|zdS)zO Called by Twisted when the cftp client prints data to stderr. zerr: %sN)rrs)r.rs r/ errReceivedzSFTPTestProcess.errReceived8s  D !!!!!r1c|jS)zQ Return the contents of the buffer of data received from stdout. )rnr-s r/ getBufferzSFTPTestProcess.getBuffer>s {r1ctj|_|t |t r|d}|j|dz|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. r8rr) r Deferredrkrh isinstancestrr:rrr.commands r/ runCommandzSFTPTestProcess.runCommandDsl"'!1!1  gs # # .nnW--G Wu_---%%r1cttjdfd|D}tj|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. rcFg|]}j|SrE)runr).0rr.sems r/ z-SFTPTestProcess.runScript..bs) H H HGcggdow// H H Hr1)r DeferredSemaphore gatherResults)r.commandsdlrs` @r/ runScriptzSFTPTestProcess.runScriptVsC%a(( H H H H Hx H H H"2&&&r1c|jrtjdStj|_|jd|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}. NKILL)rlr rrrjr signalProcessr-s r/ killProcesszSFTPTestProcess.killProcessesN   '=&& &!N,, $$V,,,  r1cnd|_|jr&|jdc}|_|ddSdS)zF Called by Twisted when the cftp client process ends. TN)rlrjrt)r.reasonrs r/ processEndedzSFTPTestProcess.processEndedtsK"   #'#4d At JJt       r1N)rArBrCrDrrhrwrur}rrrrrrEr1r/rfrfs # # #   """"  &&&$ ' ' ' ! ! !r1rfc,eZdZdZdZdZdZdZdS)CFTPClientTestBasectdd5}|tjdddn #1swxYwYtdd5}|tjdddn #1swxYwYt jddtdd5}|dtjzdddn #1swxYwYtj |S)N dsa_test.pubrdsa_testikh_tests 127.0.0.1 ) rrr!publicDSA_opensshprivateDSA_opensshrPchmodpublicRSA_opensshr r0r.fs r/r0zCFTPClientTestBase.setUps .$ ' ' 01 GGH. / / / 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 *d # # 1q GGH/ 0 0 0 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 U### )T " " @a GGMH$>> ? ? ? @ @ @ @ @ @ @ @ @ @ @ @ @ @ @!$'''s3 =AA BB B4#C##C'*C'ct|j}tj|}|t jt j}||_tj d|d|_ dS)Nrz 127.0.0.1) interface) r[r]r PortalregisterCheckerr!conchTestPublicKeyCheckerConchTestServerFactoryr listenTCPserver)r.realmpfacs r/ startServerzCFTPClientTestBase.startServerso%dl33 M%  (<>>???-// '3+FFF r1ct|jjds|dSd|jjj_t j|jjjjj }| |j|S)Nprotor) hasattrrfactory _cbStopServerrexpectedLoseConnectionr maybeDeferredrloseConnectionrrs r/ stopServerzCFTPClientTestBase.stopServersut{*G44 ,%%d++ +;< !8   3 9 C R S S d()))r1c>tj|jjSr()r rr stopListeningr.ignoreds r/rz CFTPClientTestBase._cbStopServers"4;#<===r1c~dD]'} tj|#t$rY$wxYwtj|S)N)rrr)rPremove BaseExceptionr tearDownrs r/rzCFTPClientTestBase.tearDownsV8  A  !      $T***s  ((N)rArBrCr0rrrrrEr1r/rr~sb(((GGG>>>+++++r1rceZdZdZdZdZdZdZdZdZ dZ d Z d Z dd Z d ZdZdZdZdZdZdZdZdZdZdZd S)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. cNtd}jj}t j||zd}tj dtj d|tj}t|_|fdt$j}t$jtj|d<g}i}|D]A}t1|t2r|d}||B|D]c}||} t1|t2r|d}t1| t2r| d} | ||<dtj |tj |t9jjtj || |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.1rmodrunning rc6jSr()processProtocolrh_r.s r/r`z3OurServerCmdLineClientTests.setUp.. 4 @ @ B Br1 PYTHONPATHr8env)rr0rrgetHostportr _makeArgsr!rrsrrr rrfrrrPrQcopypathsepr;rrrr:r"r spawnProcess) r.cmdsrrr encodedCmds encodedEnvcmdvarvals ` r/r0z!OurServerCmdLineClientTests.setUps  &&&   {""$$)#TD[$7$7$9$9vFFF 23>22D22333 N  .q11 BBBBCCCjooJOOCH55L   $ $C#s## *jj))   s # # # # " "Cc(C#s## *jj))#s## *jj))!JsOO      #.+:    r1c`}|fd|S)Nc6jSr()rrrs r/r`z6OurServerCmdLineClientTests.tearDown..rr1)rrrs` r/rz$OurServerCmdLineClientTests.tearDowns3 OO   BBBBCCCr1cr |jjddS#tj$rYdSwxYw)Nr)rrrr ProcessExitedAlreadyrs r/ _killProcessz(OurServerCmdLineClientTests._killProcesssL   * 8 8 @ @ @ @ @)    DD s #66c6|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. )rrrs r/rz&OurServerCmdLineClientTests.runCommands #..w777r1c6|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. )rr)r.rs r/rz%OurServerCmdLineClientTests.runScripts #--h777r1c|j}|ddddd}d}||||j|jt jd|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. pwdlpwdzcd testDirectorycd ..cg}|D]A}t|tr|d}||B|dd|ddzS)zH Callback function for handling command output. r8N)rbytesr r")r*rrs r/ cmdOutputz8OurServerCmdLineClientTests.testCdPwd..cmdOutputskD ! !c5)).**W--C C    8d122h& &r1r)r]rrr<rrPgetcwd)r.homeDirrrs r/ testCdPwdz%OurServerCmdLineClientTests.testCdPwdsw, NN5&*._checks  " " $ $ $ OOGAJ11-@@ A A A   WQZ - - - OOGAJ11-@@'!* M M M   WQZ - - - - -r1zls -l testfile1zchmod 0 testfile1zchmod 644 testfile1rrr.rrs` r/ testChAttrsz'OurServerCmdLineClientTests.testChAttrssQ  . . . . . NN    !    }}V$$$r1cfd}ddjzddd}|d||S)z Check 'ls' works as expected. Checks for wildcards, hidden files, listing directories and listing empty directories. c2|dgd|dgd|dddg|dgd|d d gdS) Nrs testDirectorytestRemoveFiletestRenameFiles testfile1rrrrr)s.testHiddenFilerrrr1r<rs r/rz4OurServerCmdLineClientTests.testList.._check*s    VVV       VVV      WQZ*;=N)O P P P    VVV      WQZ# / / / / /r1rzls ../zls *Filez ls -a *Filezls -l testDirectorycd|DS)Nc8g|]}|dS)rr)r!)rxs r/rzJOurServerCmdLineClientTests.testList....@s"!=!=!=Q!''%..!=!=!=r1rE)xss r/r`z6OurServerCmdLineClientTests.testList..@s!=!="!=!=!=r1)rr]r<rrs` r/testListz$OurServerCmdLineClientTests.testList$s} 0 0 0 0 0 NN  t|,,.. .   !    ==>>>}}V$$$r1c,|d}tjdd}t |t r|d}||j ||S)zB Check that running the '?' command returns help. ?Nrr8) rrrcmd_HELPr#rrr:rr<)r.rhelpTexts r/testHelpz$OurServerCmdLineClientTests.testHelpCs~ OOC #D))22266<<>> h $ $ 0w//H d&111r1Nc|||||dS)zg Assert that the files at C{name1} and C{name2} contain exactly the same data. N)r< getContent)r.name1name2rss r/assertFilesEqualz,OurServerCmdLineClientTests.assertFilesEqualOs: ))++U-=-=-?-?EEEEEr1c^djjjjttrdfd}djjd}|||fd|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 {}/testfile1 to {}/test file2r8c |jdjdddS)N testfile1 test file2z get failedzrm "test file2")r4endswithr r]childrrr+r.s r/ _checkGetz6OurServerCmdLineClientTests.testGet.._checkGetdsw OOFOON;; < < <  ! ! "";// ""<00    ??#455 5r1zget testfile1 "z /test file2"cjdS)Nr assertFalser]rexistsrs r/r`z5OurServerCmdLineClientTests.testGet..p1d&&t|'9'9,'G'G'N'N'P'PQQr1)rXr]rrrr:rr)r.rrr+s` @r/testGetz#OurServerCmdLineClientTests.testGetVsEKK L  L    nc * * <+227;;N 6 6 6 6 6 6 OOMdl.?MMM N N i    Q Q Q Q   r1cbfd}d}||S)zQ Test that 'get' works correctly when given wildcard parameters. cjdtddjdtdddS)NtestRemoveFiletestRemoveFile get failedtestRenameFiletestRenameFile get failed)r r]rr)rr.s r/rz;OurServerCmdLineClientTests.testWildcardGet.._checkys  ! ! ""#344)**+     ! ! ""#344)**+     r1z get testR*)rrrs` r/testWildcardGetz+OurServerCmdLineClientTests.testWildcardGettsA      OOL ) )}}V$$$r1cBdjjzdzjjzdzfd}djjd}|||fd|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"file2c jdjd|dS)Nr test"file2zrm "test\"file2")r r]rr4rrrs r/ _checkPutz6OurServerCmdLineClientTests.testPut.._checkPutss  ! ! "";//1C1CL1Q1Q    OOFOON;; < < <??#677 7r1put z/testfile1 "test\"file2"cjdS)Nr$rrs r/r`z5OurServerCmdLineClientTests.testPut..rr1)r] asBytesModerrr)r.r%rr+s` @r/testPutz#OurServerCmdLineClientTests.testPuts l&&((- . l&&((- .    8 8 8 8 8 8 OOO4<#4OOO P P i    Q Q Q Q   r1cjdd5}|ddddn #1swxYwYjdd5}|ddddn #1swxYwYfd}d jjd }|||S) zb Check that 'put' uploads files correctly when overwriting a longer file. shorterFilewmodeaN longerFilebbcjdjddS)Nr+r0)r r]r)rr.s r/r%zEOurServerCmdLineClientTests.test_putOverLongerFile.._checkPutsJ  ! ! ""=114<3E3El3S3S     r1r&z/shorterFile longerFile)r]rrrrrr)r.rr%rs` r/test_putOverLongerFilez2OurServerCmdLineClientTests.test_putOverLongerFilesi \   . . 3 3 3 = =  GGDMMM                \   - - 2 2 2 < <  GGENNN                     OOM4<#4MMM N N i   s#AAA B--B14B1cDjddd5}|ddddn #1swxYwYjdd5}|ddddn #1swxYwYfd}d jjd }|||S) z Check that 'put' uploads files correctly when overwriting a longer file and you use a wildcard to specify the files to upload. dirrr,r-r/Nr1cdjddS)Nr)r rr])rr.someDirs r/r%zMOurServerCmdLineClientTests.test_putMultipleOverLongerFile.._checkPuts;  ! !'--"7"79K9KF9S9S T T T T Tr1r&z/dir/*)r]rcreateDirectoryrrrrr)r.rr%rr7s` @r/test_putMultipleOverLongerFilez:OurServerCmdLineClientTests.test_putMultipleOverLongerFiles ,$$U++!!! ]]6 " " ' 'S ' 1 1 Q GGDMMM                \   ' ' , ,# , 6 6 ! GGENNN                U U U U U U OO<4<#4<<< = = i   s$A<<BB5CCCcfd}ddjjddjz}|||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. c|dd|ddjdjddjdjdddS)Nrr1rrrrr )r<r r]rrPrs r/checkz:OurServerCmdLineClientTests.testWildcardPut..checks   WQZ - - -   WQZ - - -  ! ! ""#344 ##%%++,<==+     ! ! ""#344 ##%%++,<==+     r1rr&z/testR*zcd %s)rr]rr<r)r.r<rs` r/testWildcardPutz+OurServerCmdLineClientTests.testWildcardPutsv      NN  -4<$ - - - dl++-- -   er1cfd}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. c|dd|ddddS)Nrr1rlz link failedz rm testLink)rr<r4ryrrs r/rz4OurServerCmdLineClientTests.testLink.._checksg  " " $ $ $   WQZ - - - OOGAJ11$77 G G G??=11 1r1zln testLink testfile1zls -l testLinkr1rrr<rs` r/testLinkz$OurServerCmdLineClientTests.testLinksa  2 2 2 2 2 NN24D E E f d&,,,r1cfd}dd}|||jd|S)zV Test that we can create and remove directories with the cftp client. c|dd|dddS)Nrr1rdzrmdir testMakeDirectory)r<r4ryrrs r/rz?OurServerCmdLineClientTests.testRemoteDirectory.._checksT   WQZ - - - OOGAJ11$77 8 8 8??#<== =r1mkdir testMakeDirectoryzls -l testMakeDirector?r1rArs` r/testRemoteDirectoryz/OurServerCmdLineClientTests.testRemoteDirectorysa  > > > > > NN46O P P f d&,,,r1chfd}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. cx|dd|dddS)Nrr1rsremote error 11: mkdir failedrrs r/rzHOurServerCmdLineClientTests.test_existingRemoteDirectory.._checks>   WQZ - - -   WQZ)I J J J J Jr1rFrrs` r/test_existingRemoteDirectoryz8OurServerCmdLineClientTests.test_existingRemoteDirectory sO  K K K K K NN46O P P fr1cdjjd}|jd|fd|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 z/testLocalDirectoryr1c.dS)Nzrmdir testLocalDirectory)rrs r/r`z@OurServerCmdLineClientTests.testLocalDirectory.. s0J K Kr1)rr]rrr<rs` r/testLocalDirectoryz.OurServerCmdLineClientTests.testLocalDirectorysv OOLdl&7LLL M M d&,,, KKKKLLL d&,,,r1cfd}dd}|||jd|S)z1 Test that we can rename a file. c|dd|dddS)Nrr1rs testfile2zrename testfile2 testfile1)r<rrs r/rz6OurServerCmdLineClientTests.testRename.._check)sJ   WQZ - - -   WQZ 6 6 6??#?@@ @r1zrename testfile1 testfile2z ls testfile?r1rArs` r/ testRenamez&OurServerCmdLineClientTests.testRename$se  A A A A A NN7 H H f d&,,,r1r()rArBrCrDr0rrrrrrrrr rr!r)r3r9r=rBrGrJrMrPrEr1r/rrsc(((T  8888884%%%,%%%>   FFFF<%%%(8((<"              r1rc6eZdZdZdZdZdZdZdZdZ dS) OurServerBatchFileTestszm Functional tests which launch a SFTP server over localhost and checks csftp in batch interface. cbt||dSr(rr0rr-s r/r0zOurServerBatchFileTests.setUp;s.  &&& r1c^t||Sr()rrrr-s r/rz OurServerBatchFileTests.tearDown?s&##D)))   r1c|td5}||dddn #1swxYwY|jj}d|fz}t j|ddd}tj dtj d|tj}tjtj|d<d|jj_t+tj || }fd }|d |||S) Nr,z-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.1rrrrrrrc0tj|Sr()rPr)resfns r/_cleanupz9OurServerBatchFileTests._getBatchOutput.._cleanupZs IbMMMJr1c|dS)NrrE)rXs r/r`z9OurServerBatchFileTests._getBatchOutput..^s #a&r1)rrrrrrr rr!rrsrrrPrQrrr;rrrrraddBoth) r.rfprrrrrZrYs @r/_getBatchOutputz'OurServerBatchFileTests._getBatchOutputCs [[]] "c]] b HHQKKK               {""$$) !2J#DJJLLf===abbA 23>22D22333jooJOOCH55L56 2 $S^Ts C C C      (())) (sAA A cjd}fd}|}|||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 c*|d}tjdt|zjj|d|ddgddS)NrrzRES %srrr) r!rrsreprassertInr]r(rr<rXr.s r/_cbCheckResultz=OurServerBatchFileTests.testBatchFile.._cbCheckResultms))E""C GHtCyy( ) ) ) MM$,224493q6 B B B   AbD VVV     r1r^rr.rrers` r/ testBatchFilez%OurServerBatchFileTests.testBatchFilecsO          & & n%%%r1cjd}fd}|}|||S)zO Test that an error in the batch file stops running the batch. zchown 0 missingFile pwd exit cljj|dSr() assertNotInr]r(rrds r/rez9OurServerBatchFileTests.testError.._cbCheckResults0   T\5577._cbCheckResults. MM$,224493 ? ? ? ? ?r1rfrgs` r/testIgnoredErrorz(OurServerBatchFileTests.testIgnoredErrorsT   @ @ @ @ @   & & n%%%r1N) rArBrCrDr0rr^rhrlrorEr1r/rRrR4sy !!!@. r1rRsshz$no ssh command-line client availablesftpz%no sftp command-line client availablec$eZdZdZdZdZdZdS)OurServerSftpClientTestsz@ Test the sftp server against sftp command line client. c^t||Sr(rTr-s r/r0zOurServerSftpClientTests.setUps(  &&&!!!r1c*|Sr()rr-s r/rz!OurServerSftpClientTests.tearDowns   r1cT ttjt d5}|ddddn #1swxYwYjj tj fd} td|djj _ tdd} fd }fd }|||fd ||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. r,z ls . exitNc*||}d|d<|S)Nrext_foorE)r.sr oldGetAttrs r/ _getAttrszCOurServerSftpClientTests.test_extendedAttributes.._getAttrss#JtQ''E$E) Lr1r{Trp)-oPubkeyAcceptedKeyTypes=ssh-dssz-VcJ|dkrd}nd}|dddddddd dd fzd d f z }|S) Nr)r|r}rEz-Fz /dev/nullr|zIdentityFile=dsa_testzUserKnownHostsFile=kh_testzHostKeyAlgorithms=ssh-rsazPort=%iz-bztestuser@127.0.0.1rE)statusargsrYrs r/hasPAKTzAOurServerSftpClientTests.test_extendedAttributes..hasPAKTs^{{? ',+TG#$ DKr1c|dd|dddD]}||ddS)Nrrrasciir)r<r rc)rir.s r/r<z?OurServerSftpClientTests.test_extendedAttributes..checksi   VAY6!9+;+;G+D+D E E E , ,  a++++  , ,r1c&td|S)Nrq)r)rrs r/r`zBOurServerSftpClientTests.test_extendedAttributes..s#;FD##N#Nr1)dictrPrQrrrrrrr#r{rTrrrr) r.rr{rrr<rrYrzrs ` @@@@r/test_extendedAttributesz0OurServerSftpClientTests.test_extendedAttributess2: [[]] "c]] "a GGL ! ! ! " " " " " " " " " " " " " " "{""$$).8       ,k9EEE59 2 E#QSV W W      2 , , , , , g NNNNOOO}}U###sA%%A),A)N)rArBrCrDr0rrrEr1r/rsrssP"""!!!F$F$F$F$F$r1rs)RrDrrsrVrPrrrMiorrunittestrzope.interfacer twisted.conchrtwisted.conch.interfacesr$twisted.conch.test.test_filetransferr r twisted.credr twisted.internetr r rrrtwisted.internet.taskrtwisted.internet.testingrtwisted.internet.utilsrrtwisted.pythonrtwisted.python.fakepwdrtwisted.python.filepathrtwisted.python.procutilsrtwisted.python.reflectrtwisted.trial.unittestrrunixtwisted.conch.scriptsrtwisted.conch.scripts.cftprtwisted.conch.sshrtwisted.conch.ssh.connectionrtwisted.conch.testr r!twisted.conch.test.test_conchr"r# ImportError skipTestsIReactorProcessr%rGrrrrrr[ProcessProtocolrfrrrRrsrEr1r/rs   %%%%%%%%&&&&&&......UUUUUUUUHHHHHHHHHHHHHH''''''444444LLLLLLLL//////,,,,,,******000000++++++}^,, })**    ......999999222222EEEEEE;;;;;;;;;;;;;;RRRRRRR        D, : :7D I I JJJI ?@@$ $ $ $ $ h$ $ A@$ N_ _ _ _ _ 8_ _ _ D         #.#.#.#.#.#.#.#.L999999996 Y/&/&/&/&/&/&/&/&d ?@@p p p p p xp p A@p f ........xxxxxh.xxxv$+$+$+$+$+$+$+$+N ?@@KKKKK"4KKA@K\  ?@@ddddd0ddA@dN ?@@EE%LL@AAEE&MMBCCR$R$R$R$R$1R$R$DCBAA@R$R$R$s,,CC! C!