File: //usr/lib/python3/dist-packages/twisted/conch/test/__pycache__/test_connection.cpython-38.pyc
U
W[�m � @ s� d Z ddlmZmZ ddlZddlmZ ed�Zddlm Z erRddl
m Z m Z nG dd � d �Z dd
l
m
Z
dd lmZ dd lmZ dd
lmZ G dd� de
j�ZG dd� d�ZG dd� de j�ZG dd� dej�ZG dd� dej�ZdS )z1
This module tests twisted.conch.ssh.connection.
� )�division�absolute_importN)�
requireModule� cryptography)�error)�common�
connectionc @ s e Zd ZG dd� d�ZdS )r c @ s e Zd ZdS )zconnection.SSHConnectionN)�__name__�
__module__� __qualname__� r r �D/usr/lib/python3/dist-packages/twisted/conch/test/test_connection.py�
SSHConnection s r N)r r
r r r r r r
r s r )�channel)�long)�unittest)�
test_userauthc @ s` e Zd ZdZdZdZdd� Zdd� Zdd � Zd
d � Z d d
� Z
dd� Z dd� Z dd� Z
dd� ZdS )� TestChannela�
A mocked-up version of twisted.conch.ssh.channel.SSHChannel.
@ivar gotOpen: True if channelOpen has been called.
@type gotOpen: L{bool}
@ivar specificData: the specific channel open data passed to channelOpen.
@type specificData: L{bytes}
@ivar openFailureReason: the reason passed to openFailed.
@type openFailed: C{error.ConchError}
@ivar inBuffer: a C{list} of strings received by the channel.
@type inBuffer: C{list}
@ivar extBuffer: a C{list} of 2-tuples (type, extended data) of received by
the channel.
@type extBuffer: C{list}
@ivar numberRequests: the number of requests that have been made to this
channel.
@type numberRequests: L{int}
@ivar gotEOF: True if the other side sent EOF.
@type gotEOF: L{bool}
@ivar gotOneClose: True if the other side closed the connection.
@type gotOneClose: L{bool}
@ivar gotClosed: True if the channel is closed.
@type gotClosed: L{bool}
� TestChannelFc C s
d| j S )NzTestChannel %i)�id��selfr r r
� logPrefix8 s zTestChannel.logPrefixc C s4 d| _ || _g | _g | _d| _d| _d| _d| _dS )zF
The channel is open. Set up the instance variables.
Tr FN)�gotOpen� specificData�inBuffer� extBuffer�numberRequests�gotEOF� gotOneClose� gotClosed)r r r r r
� channelOpen; s zTestChannel.channelOpenc C s
|| _ dS )zD
Opening the channel failed. Store the reason why.
N)�openFailureReason)r �reasonr r r
�
openFailedH s zTestChannel.openFailedc C s | j d7 _ |dkS )z_
A test request. Return True if data is 'data'.
@type data: L{bytes}
� � data)r �r �datar r r
� request_testN s zTestChannel.request_testc C s | j �|� dS )z=
Data was received. Store it in the buffer.
N)r �appendr' r r r
� dataReceivedW s zTestChannel.dataReceivedc C s | j �||f� dS )zF
Extended data was received. Store it in the buffer.
N)r r* )r �coder( r r r
� extReceived] s zTestChannel.extReceivedc C s
d| _ dS )z1
EOF was received. Remember it.
TN)r r r r r
� eofReceivedc s zTestChannel.eofReceivedc C s
d| _ dS )z3
Close was received. Remember it.
TN)r r r r r
�
closeReceivedi s zTestChannel.closeReceivedc C s
d| _ dS )z7
The channel is closed. Rembember it.
TN)r r r r r
�closedo s zTestChannel.closedN)r r
r �__doc__�namer r r! r$ r) r+ r- r. r/ r0 r r r r
r s
r c @ s$ e Zd ZdZdZdd� Zdd� ZdS )�
TestAvatarz?
A mocked-up version of twisted.conch.avatar.ConchUser
�{ c C s4 |t jkrt |||| d�S |dkr0t�| jd��dS )z�
The server wants us to return a channel. If the requested channel is
our TestChannel, return it, otherwise return None.
)� remoteWindow�remoteMaxPacketr( �avatar� conch-error-args�error args in wrong orderN)r r2 r �
ConchError�_ARGS_ERROR_CODE)r Z channelType�
windowSize� maxPacketr( r r r
�
lookupChannel{ s
� �zTestAvatar.lookupChannelc C s$ |dkr dS |dkr d|fS dS dS )z�
The client has made a global request. If the global request is
'TestGlobal', return True. If the global request is 'TestData',
return True and the request-specific data we received. Otherwise,
return False.
�
TestGlobalT� TestDataFNr )r Z requestTyper( r r r
�gotGlobalRequest� s
zTestAvatar.gotGlobalRequestN)r r
r r1 r; r> rA r r r r
r3 u s r3 c @ s@ e Zd ZdZesdZdd� Zdd� Zdd� Zd d
� Z d d � Z
d
S )�TestConnectionz}
A subclass of SSHConnection for testing.
@ivar channel: the current channel.
@type channel. C{TestChannel}
� Cannot run without cryptographyc C s dS )NrB r r r r r
r � s zTestConnection.logPrefixc C s dS )zT
The other side made the 'TestGlobal' global request. Return True.
Tr r' r r r
�global_TestGlobal� s z TestConnection.global_TestGlobalc C s d|fS )zt
The other side made the 'Test-Data' global request. Return True and
the data we received.
Tr r' r r r
�global_Test_Data� s z TestConnection.global_Test_Datac C s t |||d�| _| jS )z�
The other side is requesting the TestChannel. Create a C{TestChannel}
instance, store it, and return it.
)r5 r6 r( )r r �r r< r= r( r r r
�channel_TestChannel� s
�z"TestConnection.channel_TestChannelc C s t d��dS )zU
The other side is requesting the ErrorChannel. Raise an exception.
z
no such thingN)�AssertionErrorrF r r r
�channel_ErrorChannel� s z#TestConnection.channel_ErrorChannelN) r r
r r1 r �skipr rD rE rG rI r r r r
rB � s rB c @ s2 e Zd ZesdZejdkr dZdd� Zdd� Zdd � 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.d/� Z d0d1� Z d2d3� Z d4d5� Z d6d7� Z d8d9� Z!d:d;� Z"d<d=� Z#d>d?� Z$d@dA� Z%dBdC� Z&dDdE� Z'dFdG� Z(dS )H�ConnectionTestsrC N�/Cannot run without both cryptography and pyasn1c C s6 t �d �| _t� | j_t� | _| j| j_| j�� d S �N)r �
FakeTransport� transportr3 r7 rB �conn�serviceStartedr r r r
�setUp� s
zConnectionTests.setUpc C s@ | j �|� | jjdd� | j_| j �t�d|jd�d � dS )z?
Open the channel with the default connection.
N���z>2L� s � )rP � openChannelrO �packets� ssh_CHANNEL_OPEN_CONFIRMATION�structZpackr �r r r r r
� _openChannel� s ��z ConnectionTests._openChannelc C s | j �� d S rM )rP �serviceStoppedr r r r
�tearDown� s zConnectionTests.tearDownc C s | � | jjj| j� dS )z_
Test that the connection links itself to the avatar in the
transport.
N)ZassertIsrO r7 rP r r r r
�test_linkAvatar� s z ConnectionTests.test_linkAvatarc C sb t � }t � }| j�|� | j�|� | j�d� | �|j� | �|j� | j�� | �|j� dS )zF
Test that serviceStopped() closes any open channels.
s N) r rP rU rW �
assertTruer � assertFalser[ r )r Zchannel1�channel2r r r
�test_serviceStopped� s
z#ConnectionTests.test_serviceStoppedc C s� | j �t�d�d � | �| jjtjdfg� g | j_| j �t�d�d d � | �| jjtjdfg� g | j_| j �t�d�d � | �| jjtj dfg� g | j_| j �t�d�d � | �| jjg � dS ) z�
Test that global request packets are dispatched to the global_*
methods and the return values are translated into success or failure
messages.
r? � �� r@ s test datas TestBad� N)
rP Zssh_GLOBAL_REQUESTr �NS� assertEqualrO rV r ZMSG_REQUEST_SUCCESSZMSG_REQUEST_FAILUREr r r r
�test_GLOBAL_REQUEST� s&
��
�
�z#ConnectionTests.test_GLOBAL_REQUESTc sB � j �ddd�}� j �d� � fdd�}|�|� |�� j� |S )zh
Test that global request success packets cause the Deferred to be
called back.
� requestr& Tc s � � | d� d S �Nr& )rf )r( r r r
�check s z3ConnectionTests.test_REQUEST_SUCCESS.<locals>.check)rP �sendGlobalRequestZssh_REQUEST_SUCCESS� addCallback�
addErrback�fail�r �drj r r r
�test_REQUEST_SUCCESS s
z$ConnectionTests.test_REQUEST_SUCCESSc sB � j �ddd�}� j �d� � fdd�}|�� j� |�|� |S )zg
Test that global request failure packets cause the Deferred to be
erred back.
rh r& Tc s � � | jjd� d S ri )rf �valuer( )�fr r r
rj " s z3ConnectionTests.test_REQUEST_FAILURE.<locals>.check)rP rk Zssh_REQUEST_FAILURErl rn rm ro r r r
�test_REQUEST_FAILURE s
z$ConnectionTests.test_REQUEST_FAILUREc C sH | j `| j�t�d�d � | �| jjj� | � | jjj| j� | � | jjj
d� | � | jjj d� | � | jjj d� | � | jjj
d� | � | j jtjdfg� g | j _| j�t�d�d � | �� | � | j jtjdt�d � t�d
� fg� g | j _| j�t�d �d � | �� | � | j jtjd t�d
� t�d
� fg� dS )z�
Test that open channel packets cause a channel to be created and
opened or a failure message to be returned.
r � � r% s � �
BadChannels s s unknown channelrc s ErrorChannels s unknown failureN)rO r7 rP �ssh_CHANNEL_OPENr re r^ r r rf r( r �remoteWindowLeftr6 rV r Z MSG_CHANNEL_OPEN_CONFIRMATION�flushLoggedErrors�MSG_CHANNEL_OPEN_FAILUREr r r r
�test_CHANNEL_OPEN( sV �
���
�����
����z!ConnectionTests.test_CHANNEL_OPENc C s� || j j_| j�t�d�d � | �tj �}| �
t |�dd|f � | �
|d j j
td�df� | �
| j jtjdt�d � t�d
� fg� d S ) z�
Deliver a request for a channel open which will result in an exception
being raised during channel lookup. Assert that an error response is
delivered as a result.
r8 ru r% zExpected one error, got: %rr r4 r9 s {s error args in wrong orderrc N)rO r7 r; rP rx r re rz r r: rf �lenrr �argsr rV r r{ )r r, �errorsr r r
�_lookupChannelErrorTestL s*
� � ����z'ConnectionTests._lookupChannelErrorTestc C s | � d� dS )a�
If a C{lookupChannel} implementation raises L{error.ConchError} with the
arguments in the wrong order, a C{MSG_CHANNEL_OPEN} failure is still
sent in response to the message.
This is a temporary work-around until L{error.ConchError} is given
better attributes and all of the Conch code starts constructing
instances of it properly. Eventually this functionality should be
deprecated and then removed.
r4 N)r� r r r r
�test_lookupChannelErrorc s z'ConnectionTests.test_lookupChannelErrorc C s | � td�� dS )z�
Like L{test_lookupChannelError}, but for the case where the failure code
is represented as a L{long} instead of a L{int}.
r4 N)r� r r r r r
� test_lookupChannelErrorLongCodeq s z/ConnectionTests.test_lookupChannelErrorLongCodec C st t � }| j�|� | j�d� | �|jd� | �|jd� | �|jd� | �| jj| d� | �| jj d d� dS )zv
Test that channel open confirmation packets cause the channel to be
notified that it's open.
s r � N)
r rP rU rW rf ry r6 r ZchannelsToRemoteChannelZlocalToRemoteChannelrY r r r
� test_CHANNEL_OPEN_CONFIRMATIONy s �z.ConnectionTests.test_CHANNEL_OPEN_CONFIRMATIONc C sP t � }| j�|� | j�dt�d� � | �|jjd� | � | jj
� |�� dS )zz
Test that channel open failure packets cause the channel to be
notified that its opening failed.
� � failure!)r� r% N) r rP rU Zssh_CHANNEL_OPEN_FAILUREr re rf r" r~ Z assertIsNoneZchannels�getrY r r r
�test_CHANNEL_OPEN_FAILURE� s �z)ConnectionTests.test_CHANNEL_OPEN_FAILUREc C s8 t � }| �|� |j}| j�d� | �|j|d � dS )zc
Test that channel window adjust messages add bytes to the channel
window.
r� r% N)r rZ ry rP Zssh_CHANNEL_WINDOW_ADJUSTrf )r r Z
oldWindowSizer r r
�test_CHANNEL_WINDOW_ADJUST� s
z*ConnectionTests.test_CHANNEL_WINDOW_ADJUSTc C s t ddd�}| �|� | j�dt�d� � | �|jdg� | �| jj t
j dfg� g | j_ d|j d }| j�dt�|� � | �|jdg� | �| jj t
j
d fg� t � }| �|� d|jd }g | j_ | j�d
t�|� � | �|jg � | �| jj t
j
d fg� d S ) z�
Test that channel data messages are passed up to the channel, or
cause the channel to be closed if the data is too large.
� � �� localWindow�localMaxPacketr� r& � � � ar% � �rv N)r rZ rP Zssh_CHANNEL_DATAr re rf r rO rV r �MSG_CHANNEL_WINDOW_ADJUST�localWindowLeft�MSG_CHANNEL_CLOSEr� �r r ZlongDataZbigDatar r r
�test_CHANNEL_DATA� s.
�
�
�z!ConnectionTests.test_CHANNEL_DATAc C s t ddd�}| �|� | j�dt�d� � | �|jdg� | �| jj t
j dfg� g | j_ d|j d }| j�dt�|� � | �|jdg� | �| jj t
j
d
fg� t � }| �|� d|jd }g | j_ | j�d t�|� � | �|jg � | �| jj t
j
d
fg� d S )
z�
Test that channel extended data messages are passed up to the channel,
or cause the channel to be closed if they're too big.
r� r� r� s r& )r r& r� r� r% r� s N)r rZ rP Zssh_CHANNEL_EXTENDED_DATAr re rf r rO rV r r� r� r� r� r� r r r
�test_CHANNEL_EXTENDED_DATA� s:
�
��
�
�
�z*ConnectionTests.test_CHANNEL_EXTENDED_DATAc C s, t � }| �|� | j�d� | �|j� dS )zN
Test that channel eof messages are passed up to the channel.
r� N)r rZ rP Zssh_CHANNEL_EOFr^ r rY r r r
�test_CHANNEL_EOF� s
z ConnectionTests.test_CHANNEL_EOFc C sD t � }| �|� | j�|� | j�d� | �|j� | �|j� dS )z�
Test that channel close messages are passed up to the channel. Also,
test that channel.close() is called if both sides are closed when this
message is received.
r� N)r rZ rP � sendCloseZssh_CHANNEL_CLOSEr^ r r rY r r r
�test_CHANNEL_CLOSE� s
z"ConnectionTests.test_CHANNEL_CLOSEc sp t � }� �|� � j�dt�d� d � � �|jd� � j�dt�d� d d �}� fdd�}|�|� |S ) zS
Test that channel requests that succeed send MSG_CHANNEL_SUCCESS.
r� � testrd r% rb r& c s � � � jjtjdfg� d S �Nr� )rf rO rV r ZMSG_CHANNEL_SUCCESS��resultr r r
rj � s
�z;ConnectionTests.test_CHANNEL_REQUEST_success.<locals>.check) r rZ rP �ssh_CHANNEL_REQUESTr re rf r rl �r r rp rj r r r
� test_CHANNEL_REQUEST_success� s
� ���
z,ConnectionTests.test_CHANNEL_REQUEST_successc sP t � }� �|� � j�dt�d� d �}� fdd�}|�� j� |�|� |S )zP
Test that channel requests that fail send MSG_CHANNEL_FAILURE.
r� r� rb c s � � � jjtjdfg� d S r� )rf rO rV r ZMSG_CHANNEL_FAILUREr� r r r
rj s
�z;ConnectionTests.test_CHANNEL_REQUEST_failure.<locals>.check) r rZ rP r� r re rl rn rm r� r r r
� test_CHANNEL_REQUEST_failure s
��
z,ConnectionTests.test_CHANNEL_REQUEST_failurec s> t � }� �|� � j�|ddd�}� j�d� � fdd�}|S )zj
Test that channel request success messages cause the Deferred to be
called back.
r� r& Tr� c s � � | � d S rM )r^ r� r r r
rj s z;ConnectionTests.test_CHANNEL_REQUEST_SUCCESS.<locals>.check)r rZ rP � sendRequestZssh_CHANNEL_SUCCESSr� r r r
� test_CHANNEL_REQUEST_SUCCESS s
z,ConnectionTests.test_CHANNEL_REQUEST_SUCCESSc sT t � }� �|� � j�|ddd�}� j�d� � fdd�}|�� j� |�|� |S )zi
Test that channel request failure messages cause the Deferred to be
erred back.
r� rc Tr� c s � � | jjd� d S )Nzchannel request failed)rf rr r� r r r
rj ' s z;ConnectionTests.test_CHANNEL_REQUEST_FAILURE.<locals>.check)r rZ rP r� Zssh_CHANNEL_FAILURErl rn rm r� r r r
� test_CHANNEL_REQUEST_FAILURE s
z,ConnectionTests.test_CHANNEL_REQUEST_FAILUREc C s| | j �ddd�}|�dd� � | j �ddd� | �| jjtjt� d�d ftjt� d�d
fg� | �| j j
d |gi� d S )
zQ
Test that global request messages are sent in the right format.
s wantReplyr& Tc S s d S rM r �Zfailurer r r
�<lambda>3 rc z8ConnectionTests.test_sendGlobalRequest.<locals>.<lambda>s noReplyrc Fs �datard �globalN) rP rk rm rf rO rV r ZMSG_GLOBAL_REQUESTr re � deferreds�r rp r r r
�test_sendGlobalRequest- s
� ���z&ConnectionTests.test_sendGlobalRequestc C sX t � }| j�|d� | �| jjtjt� d�d fg� | �|j
d� | �| jj d� dS )zO
Test that open channel messages are sent in the right format.
s aaaar s � aaaar r% N) r rP rU rf rO rV r ZMSG_CHANNEL_OPENr re r ZlocalChannelIDrY r r r
�test_openChannel< s
��z ConnectionTests.test_openChannelc C s� t � }| �|� | j�|ddd�}|�dd� � | j�|ddd� d|_| j�|ddd� | �| jjt j
d t � d� d
ft j
d t � d� d fg� | �| jj
d |g� d
S )zR
Test that channel request messages are sent in the right format.
r� Tc S s d S rM r r� r r r
r� P rc z2ConnectionTests.test_sendRequest.<locals>.<lambda>� test2rc Fs test3r� s testrd r N)r rZ rP r� rm � localClosedrf rO rV r ZMSG_CHANNEL_REQUESTr re r� �r r rp r r r
�test_sendRequestH s*
������z ConnectionTests.test_sendRequestc C st t dd�}| �|� d|_| j�|d� | �|jd� d|_| j�|d� | �|jd� | �| jjt j
dfg� dS ) zi
Test that channel window adjust messages cause bytes to be added
to the window.
r� )r� r r% T� � � N) r rZ r� rP Z adjustWindowrf r� rO rV r r� rY r r r
�test_adjustWindow[ s
�z!ConnectionTests.test_adjustWindowc C sX t � }| �|� | j�|d� d|_| j�|d� | �| jjtj dt
� d� fg� dS )zO
Test that channel data messages are sent in the right format.
r� T� br� N) r rZ rP ZsendDatar� rf rO rV r ZMSG_CHANNEL_DATAr re rY r r r
�
test_sendDatal s
��z ConnectionTests.test_sendDatac C s\ t � }| �|� | j�|dd� d|_| j�|dd� | �| jjtj dt
� d� fg� dS )zX
Test that channel extended data messages are sent in the right format.
r% r� Tr� r� r� N) r rZ rP ZsendExtendedDatar� rf rO rV r ZMSG_CHANNEL_EXTENDED_DATAr re rY r r r
�test_sendExtendedDatay s
��z%ConnectionTests.test_sendExtendedDatac C sb t � }| �|� | j�|� | �| jjtjdfg� d|_ | j�|� | �| jjtjdfg� dS )zN
Test that channel EOF messages are sent in the right format.
r� TN)
r rZ rP ZsendEOFrf rO rV r ZMSG_CHANNEL_EOFr� rY r r r
� test_sendEOF� s
�
�z ConnectionTests.test_sendEOFc C s� t � }| �|� | j�|� | �|j� | �| jjt j
dfg� | j�|� | �| jjt j
dfg� t � }| �|� d|_ | j�|� | �|j � dS )zP
Test that channel close messages are sent in the right format.
r� TN)
r rZ rP r� r^ r� rf rO rV r r� Z remoteClosedr )r r r` r r r
�test_sendClose� s
�
�
z ConnectionTests.test_sendClosec C sZ | j �dddd�}| �|jd� | �|jd� | �|jd� | �tj| j jdddd� dS )z�
Test that getChannel dispatches to the avatar when an avatar is
present. Correct functioning without the avatar is verified in
test_CHANNEL_OPEN.
r �2 � r& rw N) rP Z
getChannelrf r( ry r6 Z assertRaisesr r: rY r r r
�test_getChannelWithAvatar� s �z)ConnectionTests.test_getChannelWithAvatarc C sH | j `| �| j�dd�� | �| j�dd�d� | �| j�dd�� dS )zW
Test that gotGlobalRequests dispatches to global_* without an avatar.
r? r& s Test-Data)Tr& s BadGlobalN)rO r7 r^ rP rA rf r_ r r r r
�"test_gotGlobalRequestWithoutAvatar� s �z2ConnectionTests.test_gotGlobalRequestWithoutAvatarc C sB t � }| �|� | jj|dddd�}| �|tj�}| j�|� |S )z�
Whenever an SSH channel gets closed any Deferred that was returned by a
sendRequest() on its parent connection must be errbacked.
� dummyrequest� dummydatar% �Z wantReply)r rZ rP r� �
assertFailurer r: Z
channelClosedr� r r r
�9test_channelClosedCausesLeftoverChannelDeferredsToErrback� s
� zIConnectionTests.test_channelClosedCausesLeftoverChannelDeferredsToErrback))r r
r r rJ r rO rR rZ r\ r] ra rg rq rt 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
rK � sL
$
rK c @ s6 e Zd ZdZesdZejdkr"dZdd� Zdd� Z dS ) � CleanConnectionShutdownTestszL
Check whether correct cleanup is performed on connection shutdown.
rC NrL c C s, t �d �| _t� | j_t� | _| j| j_d S rM )r rN rO r3 r7 rB rP r r r r
rR � s
z"CleanConnectionShutdownTests.setUpc C s8 | j �� | j jdddd�}| �|tj�}| j �� |S )z�
Once the service is stopped any leftover global deferred returned by
a sendGlobalRequest() call must be errbacked.
r� r� r% r� )rP rQ rk r� r r: r[ r� r r r
�9test_serviceStoppedCausesLeftoverGlobalDeferredsToErrback� s
�
zVCleanConnectionShutdownTests.test_serviceStoppedCausesLeftoverGlobalDeferredsToErrback)
r r
r r1 r rJ r rO rR r� r r r r
r� � s
r� ) r1 Z
__future__r r rX Ztwisted.python.reflectr r Z
twisted.conchr Ztwisted.conch.sshr r r Ztwisted.python.compatr Z
twisted.trialr Ztwisted.conch.testr Z
SSHChannelr r3 r rB ZTestCaserK r� r r r r
�<module> s* Y(,