File: //usr/lib/python3/dist-packages/twisted/conch/test/__pycache__/test_filetransfer.cpython-38.pyc
U
\�[7r � @ s� d Z ddlmZmZ ddlZddlZddlZddlmZ ddl m
Z
ed�Z ed�Z e r~ddl
mZ dd lmZmZmZmZ nG d
d � d �Zdd lmZ dd
lmZ ddlmZ ddlmZm Z ddl m Z G dd� dej �Z G dd� de �Z!G dd� d�Z"e �rNe#e d��s,ddl$Z$e$�%de f � dZ n"G dd� de j&�Z'e�(e'e ej)� G dd � d e
j*�Z+G d d � d e+�Z,G d d � d �Z-G d!d"� d"e
j*�Z.G d#d$� d$e
j*�Z/G d%d&� d&e
j*�Z0dS )'z.
Tests for L{twisted.conch.ssh.filetransfer}.
� )�division�absolute_importN)�
requireModule)�unittest� cryptographyztwisted.conch.unix��avatar)�common�
connection� filetransfer�sessionc @ s e Zd ZG dd� d�ZdS )r c @ s e Zd ZdS )zavatar.ConchUserN)�__name__�
__module__� __qualname__� r r �F/usr/lib/python3/dist-packages/twisted/conch/test/test_filetransfer.py� ConchUser s r N)r
r r r r r r r r s r )�defer)�loopback)�
components)�long� _PY37PLUS)�FilePathc @ s e Zd Zdd� Zdd� ZdS )�
TestAvatarc C s( t j�| � tj| jd< tj| jd< d S )N� session� sftp) r r �__init__r �
SSHSessionZ
channelLookupr �FileTransferServerZsubsystemLookup��selfr r r r $ s zTestAvatar.__init__c O s~ z t |�}W n tk
r, |||fg}Y nX |D ]F}|d }t|�dkrR|d pTd}t|�dkrj|d pli }|||�}q2|S )Nr � r � )�iter� TypeError�len)r �f�args�kw�i�func�rr r r �
_runAsUser) s zTestAvatar._runAsUserN)r
r r r r, r r r r r # s r c @ s e Zd Zdd� Zdd� ZdS )�FileTransferTestAvatarc C s t �| � || _d S �N)r r �homeDir)r r/ r r r r 8 s
z FileTransferTestAvatar.__init__c C s t t�� ��| jj�S r. )r �os�getcwd� preauthChildr/ �pathr r r r �
getHomeDir< s z!FileTransferTestAvatar.getHomeDirN)r
r r r r4 r r r r r- 6 s r- c @ s e Zd Zdd� ZdS )�ConchSessionForTestAvatarc C s
|| _ d S r. r )r r r r r r B s z"ConchSessionForTestAvatar.__init__N)r
r r r r r r r r5 @ s r5 �SFTPServerForUnixConchUserzNtwisted.conch.unix imported %r, but doesn't define SFTPServerForUnixConchUser'c @ s e Zd Zdd� Zdd� ZdS )�FileTransferForTestAvatarc C s ddiS )N� conchTest� ext datar )r �versionZotherExtr r r �
gotVersionQ s z$FileTransferForTestAvatar.gotVersionc C s |dkr dS t �d S )N� testExtendedRequest� bar)�NotImplementedError)r ZextName�extDatar r r �extendedRequestT s z)FileTransferForTestAvatar.extendedRequestN)r
r r r; r@ r r r r r7 O s r7 c @ s e Zd Zdd� ZdS )� SFTPTestBasec
C s t | �� �| _| j�d�| _| j�d��d� | j�d�jdd��6}|�d� tdd ��}|�|�d
�� W 5 Q R X W 5 Q R X | j�d��d � | j�d �jdd��}|�d
� W 5 Q R X | j�d�jdd��}|�d
� W 5 Q R X | j�d�jdd��}|�d
� W 5 Q R X d S )NZextraZ
testDirectoryT� testfile1�wb)�mode� aaaaaaaaaabbbbbbbbbbz /dev/urandom�rbi i� ZtestRemoveFile� aZtestRenameFilez.testHiddenFile) r Zmktemp�testDir�child�makedirs�open�write�read�chmod)r r& �f2r r r �setUp_ s
$zSFTPTestBase.setUpN)r
r r rP r r r r rA ] s rA c @ s� e Zd ZesdZdd� Zdd� Zdd� Zdd � Zd
d � Z d d
� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zej d.d/� �Z e r�d0e _ej d1d2� �Z d3S )4�OurServerOurClientTests� can't run on non-posix computersc s� t �� � t� j�� _tj� jd�� _t� � j�}t�
� � _ d � _ d � _
� fdd�}|� j _t� � j �}� j �|� � j�|� |� _|� _� �� d S )Nr c s | � _ |� _d S r. )�_serverVersion�_extData)Z
serverVersionr? r r r �_� s z(OurServerOurClientTests.setUp.<locals>._)rA rP r- rH r r r �serverr Z
LoopbackRelay�FileTransferClient�clientrS rT ZgotServerVersionZmakeConnection�clientTransport�serverTransport�
_emptyBuffers)r rY rU rZ r r r rP x s
z OurServerOurClientTests.setUpc C s* | j js| jjr&| j �� | j�� q d S r. )rZ �bufferrY � clearBufferr r r r r[ � s
z%OurServerOurClientTests._emptyBuffersc C s, | j �� | j�� | j �� | j�� d S r. )rZ �loseConnectionrY r] r r r r �tearDown� s
z OurServerOurClientTests.tearDownc C s$ | � | jd� | � | jddi� d S )N� r8 r9 )� assertEqualrS rT r r r r �test_serverVersion� s z*OurServerOurClientTests.test_serverVersionc C s& | � tj�| jj�d| jjf � dS )z:
It implements the ISFTPServer interface.
z ISFTPServer not provided by %rN)�
assertTruer � ISFTPServerZ
providedByrV rX r r r r � test_interface_implementation� s �z5OurServerOurClientTests.test_interface_implementationc sb �j �dtjtjB i �}��� tj�g � � �fdd�}��td|� � �fdd�}|� |� |S )zV
A file opened with C{openFile} is close when the connection is lost.
� testfile1c s � � | � �| � d S r. ��append)�fd)�closed�oldCloser r �close� s
zJOurServerOurClientTests.test_openedFileClosedWithConnection.<locals>.closerl c s` �j j| jdd � j}�j�� �j�� �j�� �j�� ���j ji � �� |� � d S )N� )
rV Z openFilesZhandleri rZ r^ rY r] ra ZassertIn)�openFileri )rj r r r � _fileOpened� s
zPOurServerOurClientTests.test_openedFileClosedWithConnection.<locals>._fileOpened)
rX rn r �FXF_READ� FXF_WRITEr[ r0 rl Zpatch� addCallback)r �drl ro r )rj rk r r �#test_openedFileClosedWithConnection� s ��
z;OurServerOurClientTests.test_openedFileClosedWithConnectionc s. � j �d�}� �� � fdd�}|�|� |S )zh
A directory opened with C{openDirectory} is close when the connection
is lost.
� c s< � j �� � j�� � j �� � j�� � �� jji � d S r. )rZ r^ rY r] ra rV ZopenDirs��openDirr r r � _getFiles� s
zSOurServerOurClientTests.test_openedDirectoryClosedWithConnection.<locals>._getFiles�rX �
openDirectoryr[ rr )r rs rx r r r �(test_openedDirectoryClosedWithConnection� s
z@OurServerOurClientTests.test_openedDirectoryClosedWithConnectionc sd �j �dtjtjB i �}��� � ��fdd�}�fdd�� ��fdd���fdd ��|�|� |S )
Nrf c s* �� | t�| �� � | �}|��| � |S r. )ra r Z ISFTPFilerr �rn rs )�
_readChunk� _writeChunkr r r ro � s z<OurServerOurClientTests.test_openFileIO.<locals>._fileOpenedc s& | � dd�}� �� |�� jd� |S )Nr � rE �Z readChunkr[ rr ra r| r r r r} � s z;OurServerOurClientTests.test_openFileIO.<locals>._readChunkc s$ |� dd�}��� |�� |� |S )Nr s
cccccccccc)Z
writeChunkr[ rr �rU rn rs )� _readChunk2r r r r~ � s z<OurServerOurClientTests.test_openFileIO.<locals>._writeChunkc s& |� dd�}� �� |�� jd� |S )Nr � s aaaaaaaaaabbbbbbbbbbccccccccccr� r� r r r r� � s z<OurServerOurClientTests.test_openFileIO.<locals>._readChunk2�rX rn r rp rq r[ rr )r rs ro r )r} r� r~ r r �test_openFileIO� s ��
z'OurServerOurClientTests.test_openFileIOc sV �j �dtjtjB i �}��� �fdd���fdd�� � ��fdd�}|�|� |S )Nrf c s |� � }� �� |S r. )�getAttrsr[ r� r r r � _getAttrs s zBOurServerOurClientTests.test_closedFileGetAttrs.<locals>._getAttrsc s � � � | S r. )ZflushLoggedErrors�r& r r r �_err s z=OurServerOurClientTests.test_closedFileGetAttrs.<locals>._errc s4 | � � }��� |��| � |�� � ��|tj�S r. )rl r[ rr �
addErrback�
assertFailurer � SFTPErrorr| �r� r� r r r �_close
s
z?OurServerOurClientTests.test_closedFileGetAttrs.<locals>._closer� )r rs r� r r� r �test_closedFileGetAttrs� s ��
z/OurServerOurClientTests.test_closedFileGetAttrsc sD �j �dtjtjB i �}��� � �fdd�}�fdd�� |�|�S )Nrf c s | � � }��� |�� � |S r. �r� r[ rr r| ��
_getAttrs2r r r r� s
zBOurServerOurClientTests.test_openFileAttributes.<locals>._getAttrsc s& � j �d�}� �� |�� j| � |S �Nrf �rX r� r[ rr ra )Zattrs1rs r r r r� s zCOurServerOurClientTests.test_openFileAttributes.<locals>._getAttrs2r� �r rs r� r r� r �test_openFileAttributes s �� z/OurServerOurClientTests.test_openFileAttributesc sV �j �dtjtjB i �}��� ��fdd�}� �fdd���fdd�� |�|� |S )Nrf c s | � � }��� |�� � |S r. r� r| )� _setAttrsr r r r� 0 s
z@OurServerOurClientTests.test_openFileSetAttrs.<locals>._getAttrsc s: d| d< �j �d| �}��� |�� � |��j| � |S )Nr Zatimerf )rX ZsetAttrsr[ rr ra ��attrsrs r� r r r� 6 s
z@OurServerOurClientTests.test_openFileSetAttrs.<locals>._setAttrsc s � j �d�}� �� |S r� �rX r� r[ �rU rs r r r r� > s zAOurServerOurClientTests.test_openFileSetAttrs.<locals>._getAttrs2r� r� r )r� r� r r �test_openFileSetAttrs) s ��
z-OurServerOurClientTests.test_openFileSetAttrsc sb i ��j jj� � �fdd�}|�j j_�j�dtjtjB ddi�}��� ��fdd�}|�|�S )z�
Check that L{filetransfer.FileTransferClient.openFile} can send
extended attributes, that should be extracted server side. By default,
they are ignored, so we just verify they are correctly parsed.
c s �� |� � | ||�S r. )�update)�filename�flagsr� )� oldOpenFile�savedAttributesr r rn O s
zIOurServerOurClientTests.test_openFileExtendedAttributes.<locals>.openFilerf �ext_foor= c s �� � ddi� d S )Nr� r= �ra )Zign)r� r r r �checkX s zFOurServerOurClientTests.test_openFileExtendedAttributes.<locals>.check)rV rX rn r rp rq r[ rr )r rn rs r� r )r� r� r r � test_openFileExtendedAttributesG s
��z7OurServerOurClientTests.test_openFileExtendedAttributesc sB � j �d�}� �� � fdd�}|�|� |�|� � �|tj�S )N� testRemoveFilec s � j �d�}� �� |S )Nr� )rX Z
removeFiler[ )�ignoredrs r r r � _removeFileb s z<OurServerOurClientTests.test_removeFile.<locals>._removeFile)rX r� r[ rr r� r r� )r rs r� r r r �test_removeFile^ s
z'OurServerOurClientTests.test_removeFilec s8 �j �d�}��� � �fdd�}�fdd�� |�|�S )N� testRenameFilec s& �j �dd�}��� |�� | � |S )Nr� � testRenamedFile)rX Z
renameFiler[ rr r� �� _testRenamedr r r �_renamep s z8OurServerOurClientTests.test_renameFile.<locals>._renamec s& � j �d�}� �� |�� j|� d S )Nr� r� )rU r� rs r r r r� v s z=OurServerOurClientTests.test_renameFile.<locals>._testRenamed)rX r� r[ rr )r rs r� r r� r �test_renameFilel s
z'OurServerOurClientTests.test_renameFilec C s" | j �d�}| �� | �|tj�S �N� testMakeDirectory�rX r� r[ r� r r� �r rs r r r �test_directoryBad~ s z)OurServerOurClientTests.test_directoryBadc sZ � j �di �}� �� � fdd�}� fdd�}|�|� |�|� |�|� � �|tj�S )Nr� c s � j �d�}� �� |S r� r� r� r r r r� � s zAOurServerOurClientTests.test_directoryCreation.<locals>._getAttrsc s � j �d�}� �� |S r� )rX ZremoveDirectoryr[ r� r r r �_removeDirectory� s zHOurServerOurClientTests.test_directoryCreation.<locals>._removeDirectory)rX Z
makeDirectoryr[ rr r� r r� )r rs r� r� r r r �test_directoryCreation� s
z.OurServerOurClientTests.test_directoryCreationc s\ �j �d�}��� g �� ���fdd����fdd�}�fdd�� |��� |�|� |S )N� c sF �� fdd�}t �� j�}��� |�|� |��� |��� � |S )Nc s � � | � �S r. rg r� )�filesrw r r rh � s
zMOurServerOurClientTests.test_openDirectory.<locals>._getFiles.<locals>.append)r Z
maybeDeferred�nextr[ rr r� )rw rh rs �r� rx r� r rv r rx � s
z=OurServerOurClientTests.test_openDirectory.<locals>._getFilesc s6 t t t� � �d �}|�� ��|dddddg� d S )Nr � .testHiddenFile�
testDirectoryr� r� rf )�list�zip�sortra )r� Zfs)r� r r r � _checkFiles� s ��z?OurServerOurClientTests.test_openDirectory.<locals>._checkFilesc s |� � }� �� |S r. )rl r[ )rU rw rs r r r r� � s z:OurServerOurClientTests.test_openDirectory.<locals>._closery )r rs r� r r� r �test_openDirectory� s
z*OurServerOurClientTests.test_openDirectoryc C s" | j �d�}| �� | �|tj�S )N� testLinkr� r� r r r �test_linkDoesntExist� s z,OurServerOurClientTests.test_linkDoesntExistc sB � j �dd�}� �� � fdd�}� fdd�}|�|� |�|�S )Nr� rf c s � j �dd�}� �� |S )Nr� r! r� r� r r r �_getFirstAttrs� s zDOurServerOurClientTests.test_linkSharesAttrs.<locals>._getFirstAttrsc s& � j �d�}� �� |�� j| � |S r� r� )Z
firstAttrsrs r r r �_getSecondAttrs� s zEOurServerOurClientTests.test_linkSharesAttrs.<locals>._getSecondAttrs�rX ZmakeLinkr[ rr )r rs r� r� r r r �test_linkSharesAttrs� s
z,OurServerOurClientTests.test_linkSharesAttrsc sF � j �dd�}� �� � fdd�}� fdd�}|�|� |�|� |S )Nr� rf c sH � j �d�}� �� tt�� ��� jj�}|� d�}|�
� j |j� |S �Nr� rB ) rX ZreadLinkr[ r r0 r1 r2 rH r3 rI rr ra )rU rs ZtestFiler r r � _readLink� s
�z8OurServerOurClientTests.test_linkPath.<locals>._readLinkc sH � j �d�}� �� tt�� ��� jj�}|� d�}|�
� j |j� |S r� ) rX ZrealPathr[ r r0 r1 r2 rH r3 rI rr ra )rU rs ZtestLinkr r r � _realPath� s
�z8OurServerOurClientTests.test_linkPath.<locals>._realPathr� )r rs r� r� r r r �
test_linkPath� s
z%OurServerOurClientTests.test_linkPathc C s4 | j �dd�}| �� |�| jd� |�| j� |S )Nr< s foor= )rX r@ r[ rr ra �_cbTestExtendedRequestr� r r r �test_extendedRequest� s
z,OurServerOurClientTests.test_extendedRequestc C s" | j �dd�}| �� | �|t�S )Ns testBadRequestr� )rX r@ r[ r� r> )r r� rs r r r r� � s z.OurServerOurClientTests._cbTestExtendedRequestc c s� | j �d�}| �� |V }t� }z,|D ]"}| �� |V \}}}|�|� q&W 5 |�� }| �� |V X | �� | �|tdddddg�� dS )z�
Check that the object returned by
L{filetransfer.FileTransferClient.openDirectory} can be used
as an iterator.
r� r� r� r� r� rf N)rX rz r[ �setrl �addra )r rs rw � filenamesr& r� rU Z fileattrsr r r �test_openDirectoryIterator s(
��z2OurServerOurClientTests.test_openDirectoryIteratorz!Broken by PEP 479 and deprecated.c c sj | j �d�}| �� |V }|�� | �� }d}| �dt|�� | �t|d d � | �||d d � dS )zJ
Using client.openDirectory as an iterator is deprecated.
r� zeUsing twisted.conch.ssh.filetransfer.ClientDirectory as an iterator was deprecated in Twisted 18.9.0.r! r �category�messageN)rX rz r[ r� Z
flushWarningsra r% �DeprecationWarning)r rs rw �warningsr� r r r �$test_openDirectoryIteratorDeprecated+ s �z<OurServerOurClientTests.test_openDirectoryIteratorDeprecatedN) r
r r �unix�skiprP r[ r_ rb re rt r{ r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r ZinlineCallbacksr� r r� r r r r rQ s s> ! "
#�rQ c @ s e Zd Zdd� ZdS )�FakeConnc C s d S r. r )r Zchannelr r r � sendCloseA s zFakeConn.sendCloseN)r
r r r� r r r r r� @ s r� c @ sL e Zd ZesdZdd� Zdd� Zdd� Zdd � Zd
d � Z d d
� Z
dd� Z dS )�FileTransferCloseTestsrR c C s t � | _d S r. )r r r r r r rP J s z FileTransferCloseTests.setUpc C s, t �� }G dd� d�}|� |_| j|j_|S )Nc @ s$ e Zd Zdd� Zdd� Zdd� ZdS )zDFileTransferCloseTests.buildServerConnection.<locals>.DummyTransportc S s
| | _ d S r. )� transportr r r r r R s zMFileTransferCloseTests.buildServerConnection.<locals>.DummyTransport.__init__c S s d S r. r )r Zkind�datar r r �
sendPacketT s zOFileTransferCloseTests.buildServerConnection.<locals>.DummyTransport.sendPacketc S s dS )Nzdummy transportr r r r r � logPrefixV s zNFileTransferCloseTests.buildServerConnection.<locals>.DummyTransport.logPrefixN)r
r r r r� r� r r r r �DummyTransportQ s r� )r
Z
SSHConnectionr� r )r �connr� r r r �buildServerConnectionM s
z,FileTransferCloseTests.buildServerConnectionc s$ d�_ |j� � �fdd�}||_d S )NFc s d�_ � | � d S )NT)�connectionLostFired)�reason�ZorigConnectionLostr r r �connectionLost` s zFFileTransferCloseTests.interceptConnectionLost.<locals>.connectionLost)r� r� )r �
sftpServerr� r r� r �interceptConnectionLost] s z.FileTransferCloseTests.interceptConnectionLostc C s | � | jd� d S )Nz*sftpServer's connectionLost was not called)rc r� r r r r �assertSFTPConnectionLostf s �z/FileTransferCloseTests.assertSFTPConnectionLostc C sJ t jt� | jd�}|�t�d�� |jjj }| �
|� |� � | � � dS )ze
Closing a session should notify an SFTP subsystem launched by that
session.
)r� r r N)
r r r� r �request_subsystemr �NSrX r� �protor� Z
closeReceivedr� )r Z testSessionr� r r r �test_sessionClosek s
z(FileTransferCloseTests.test_sessionClosec C s� | � � }t�d�t�dd�d }|�|� |jd }|�t�d�� |jj j
}| � |� | � |� |� t�dd�� | �
� dS )zy
A client sending CHANNEL_CLOSE should trigger closeReceived on the
associated channel instance.
r �>Lr r` r N)r� r r� �struct�pack�ssh_CHANNEL_OPEN�channelsr� rX r� r� r� Zssh_CHANNEL_CLOSEr� �r r� ZpacketZsessionChannelr� r r r �%test_clientClosesChannelOnConnnection� s
z<FileTransferCloseTests.test_clientClosesChannelOnConnnectionc C sn | � � }t�d�t�dd�d }|�|� |jd }|�t�d�� |jj j
}| � |� |� � | �
� dS )zP
Closing an SSH connection should close all sessions within it.
r r� r r` r N)r� r r� r� r� r� r� r� rX r� r� r� ZserviceStoppedr� r� r r r �'test_stopConnectionServiceClosesChannel� s
z>FileTransferCloseTests.test_stopConnectionServiceClosesChannelN) r
r r r� r� rP r� r� r� r� r� r� r r r r r� E s r� c @ s, e Zd ZdZesdZddddgZdd� Zd S )
�ConstantsTestsag
Tests for the constants used by the SFTP protocol implementation.
@ivar filexferSpecExcerpts: Excerpts from the
draft-ietf-secsh-filexfer-02.txt (draft) specification of the SFTP
protocol. There are more recent drafts of the specification, but this
one describes version 3, which is what conch (and OpenSSH) implements.
� Cannot run without cryptographya�
The following values are defined for packet types.
#define SSH_FXP_INIT 1
#define SSH_FXP_VERSION 2
#define SSH_FXP_OPEN 3
#define SSH_FXP_CLOSE 4
#define SSH_FXP_READ 5
#define SSH_FXP_WRITE 6
#define SSH_FXP_LSTAT 7
#define SSH_FXP_FSTAT 8
#define SSH_FXP_SETSTAT 9
#define SSH_FXP_FSETSTAT 10
#define SSH_FXP_OPENDIR 11
#define SSH_FXP_READDIR 12
#define SSH_FXP_REMOVE 13
#define SSH_FXP_MKDIR 14
#define SSH_FXP_RMDIR 15
#define SSH_FXP_REALPATH 16
#define SSH_FXP_STAT 17
#define SSH_FXP_RENAME 18
#define SSH_FXP_READLINK 19
#define SSH_FXP_SYMLINK 20
#define SSH_FXP_STATUS 101
#define SSH_FXP_HANDLE 102
#define SSH_FXP_DATA 103
#define SSH_FXP_NAME 104
#define SSH_FXP_ATTRS 105
#define SSH_FXP_EXTENDED 200
#define SSH_FXP_EXTENDED_REPLY 201
Additional packet types should only be defined if the protocol
version number (see Section ``Protocol Initialization'') is
incremented, and their use MUST be negotiated using the version
number. However, the SSH_FXP_EXTENDED and SSH_FXP_EXTENDED_REPLY
packets can be used to implement vendor-specific extensions. See
Section ``Vendor-Specific-Extensions'' for more details.
a�
The flags bits are defined to have the following values:
#define SSH_FILEXFER_ATTR_SIZE 0x00000001
#define SSH_FILEXFER_ATTR_UIDGID 0x00000002
#define SSH_FILEXFER_ATTR_PERMISSIONS 0x00000004
#define SSH_FILEXFER_ATTR_ACMODTIME 0x00000008
#define SSH_FILEXFER_ATTR_EXTENDED 0x80000000
a�
The `pflags' field is a bitmask. The following bits have been
defined.
#define SSH_FXF_READ 0x00000001
#define SSH_FXF_WRITE 0x00000002
#define SSH_FXF_APPEND 0x00000004
#define SSH_FXF_CREAT 0x00000008
#define SSH_FXF_TRUNC 0x00000010
#define SSH_FXF_EXCL 0x00000020
a�
Currently, the following values are defined (other values may be
defined by future versions of this protocol):
#define SSH_FX_OK 0
#define SSH_FX_EOF 1
#define SSH_FX_NO_SUCH_FILE 2
#define SSH_FX_PERMISSION_DENIED 3
#define SSH_FX_FAILURE 4
#define SSH_FX_BAD_MESSAGE 5
#define SSH_FX_NO_CONNECTION 6
#define SSH_FX_CONNECTION_LOST 7
#define SSH_FX_OP_UNSUPPORTED 8
c C s� i }| j D ]<}|�� D ].}t�d|�}|rt|�d�d�||�d�< qq
| �t|�dkd� |�� D ]\}}| � |t
t |�� qddS )z�
The constants used by the SFTP protocol implementation match those
found by searching through the spec.
z)^\s*#define SSH_([A-Z_]+)\s+([0-9x]*)\s*$r" r r! z,No constants found (the test must be buggy).N) �filexferSpecExcerpts�
splitlines�re�matchr �grouprc r% �itemsra �getattrr )r Z constantsZexcerpt�line�m�k�vr r r �test_constantsAgainstSpec s
�z(ConstantsTests.test_constantsAgainstSpecN)r
r r �__doc__r r� r� r r r r r r� � s &
�Lr� c @ sP e Zd ZdZesdZdd� Zdd� Zdd� Zd d
� Z d d � Z
d
d� Z dd� Z dS )�RawPacketDataTestsz�
Tests for L{filetransfer.FileTransferClient} which explicitly craft certain
less common protocol messages to exercise their handling.
r� c C s t �� | _d S r. )r rW �ftcr r r r rP " s zRawPacketDataTests.setUpc C sT t �� }|�| j� || jjd< t�ddtj �t
� d� t
� d� }| j� |� |S )a/
A STATUS packet containing a result code, a message, and a language is
parsed to produce the result of an outstanding request L{Deferred}.
@see: U{section 9.1<http://tools.ietf.org/html/draft-ietf-secsh-filexfer-13#section-9.1>}
of the SFTP Internet-Draft.
r! �!LL� msg� lang)
r �Deferredrr �_cbTestPacketSTATUSr � openRequestsr� r� r �FX_OKr r� �
packet_STATUS�r rs r� r r r �test_packetSTATUS& s $ z$RawPacketDataTests.test_packetSTATUSc C s$ | � |d d� | � |d d� dS )z{
Assert that the result is a two-tuple containing the message and
language from the STATUS packet.
r r r! r Nr� �r �resultr r r r 6 s z&RawPacketDataTests._cbTestPacketSTATUSc C s@ t �� }|�| j� || jjd< t�ddtj �}| j�
|� |S )az
A STATUS packet containing only a result code can also be parsed to
produce the result of an outstanding request L{Deferred}. Such packets
are sent by some SFTP implementations, though not strictly legal.
@see: U{section 9.1<http://tools.ietf.org/html/draft-ietf-secsh-filexfer-13#section-9.1>}
of the SFTP Internet-Draft.
r! r ) r r
rr �_cbTestPacketSTATUSShortr r r� r� r r
r r r r r �test_packetSTATUSShort? s z)RawPacketDataTests.test_packetSTATUSShortc C s$ | � |d d� | � |d d� dS )z�
Assert that the result is a two-tuple containing empty strings, since
the STATUS packet had neither a message nor a language.
r r� r! Nr� r r r r r P s z+RawPacketDataTests._cbTestPacketSTATUSShortc C sJ t �� }|�| j� || jjd< t�ddtj �t
� d� }| j� |� |S )a�
A STATUS packet containing a result code and a message but no language
can also be parsed to produce the result of an outstanding request
L{Deferred}. Such packets are sent by some SFTP implementations, though
not strictly legal.
@see: U{section 9.1<http://tools.ietf.org/html/draft-ietf-secsh-filexfer-13#section-9.1>}
of the SFTP Internet-Draft.
r! r r )
r r
rr � _cbTestPacketSTATUSWithoutLangr r r� r� r r
r r� r r r r r � test_packetSTATUSWithoutLangY s
z/RawPacketDataTests.test_packetSTATUSWithoutLangc C s$ | � |d d� | � |d d� dS )z�
Assert that the result is a two-tuple containing the message from the
STATUS packet and an empty string, since the language was missing.
r r r! r� Nr� r r r r r k s z1RawPacketDataTests._cbTestPacketSTATUSWithoutLangN)
r
r r r r r� rP r r r r r r r r r r r s r )1r Z
__future__r r r0 r� r� Ztwisted.python.reflectr Z
twisted.trialr r r� Z
twisted.conchr Ztwisted.conch.sshr r
r r Ztwisted.internetr Ztwisted.protocolsr Ztwisted.pythonr Ztwisted.python.compatr r Ztwisted.python.filepathr r r r- r5 �hasattrr� �warnr6 r7 ZregisterAdapterrd ZTestCaserA rQ r� r� r� r r r r r �<module> sR
�
� Pjj