File: //lib/python3/dist-packages/twisted/conch/test/__pycache__/test_ssh.cpython-38.pyc
U
��W[y~ � @ s� d Z ddlmZmZ ddlZddlmZ ed�Zed�Zerfddl m
Z
m Z m Z m
Z
ddlmZmZ nG d d
� d
�Zdd lmZmZ dd lmZmZ dd
lmZ ddlmZ ddlmZm Z m Z ddl m Z ddl m!Z!m"Z" ddl#m$Z$ ddl%m&Z& G dd� de'�Z(G dd� dej)�Z*G dd� de'�Z+ddl m,Z, e�rHe,�-e+e*e j.� G dd � d e j/�Z0G d d � d �Z1G d d � d �Z2G d!d"� d"�Z3G d#d$� d$�Z4edk �r�edk �r�dd%lm5Z5 dd&l m6Z6m7Z7m8Z8m9Z9 dd'l m:Z:m;Z; G d(d)� d)�Z<G d*d+� d+e5j=�Z>G d,d-� d-e8j?�Z@G d.d/� d/�ZAG d0d1� d1eAe:jB�ZCG d2d3� d3eAe:jD�ZEG d4d5� d5e;jF�ZGG d6d7� d7e7jH�ZIG d8d9� d9e6jJ�ZKd:d;� ZLG d<d=� d=e$jM�ZNG d>d?� d?e$jM�ZOG d@dA� dAe$jM�ZPG dBdC� dCe$jM�ZQdS )Dz!
Tests for L{twisted.conch.ssh}.
� )�division�absolute_importN)�
requireModule� cryptography�pyasn1)�common�
forwarding�session�_kex)�avatar�errorc @ s e Zd ZG dd� d�ZdS )r c @ s e Zd ZdS )zavatar.ConchUserN)�__name__�
__module__� __qualname__� r r �=/usr/lib/python3/dist-packages/twisted/conch/test/test_ssh.py� ConchUser s r N)r
r r r r r r r r s r )�publicRSA_openssh�privateRSA_openssh)�publicDSA_openssh�privateDSA_openssh)�portal)�UnauthorizedLogin)�defer�protocol�reactor)�ProcessTerminated)�failure�log)�unittest)�
LoopbackRelayc @ s$ e Zd ZdZdZdd� Zdd� ZdS )�ConchTestRealma"
A realm which expects a particular avatarId to log in once and creates a
L{ConchTestAvatar} for that request.
@ivar expectedAvatarID: The only avatarID that this realm will produce an
avatar for.
@ivar avatar: A reference to the avatar after it is requested.
Nc C s
|| _ d S �N)�expectedAvatarID)�selfr# r r r �__init__1 s zConchTestRealm.__init__c G sN || j kr8| jdk r td��t� | _|d | j| jjfS td| j |f ��dS )z�
Return a new L{ConchTestAvatar} if the avatarID matches the expected one
and this is the first avatar request.
NzOnly one login allowedr zOnly %r may log in, not %r)r# r r �ConchTestAvatar�logout)r$ ZavatarIDZmindZ
interfacesr r r �
requestAvatar5 s
�z ConchTestRealm.requestAvatar)r
r r �__doc__r r% r( r r r r r! % s r! c @ sL e Zd ZdZesdZdZdd� Zdd� Zdd � Z d
d � Z
d d
� Z dd� Z dS )r& z�
An avatar against which various SSH features can be tested.
@ivar loggedOut: A flag indicating whether the avatar logout method has been
called.
� cannot run without cryptographyFc C sB t j�| � i | _i | _| j�tjt j
d�� | j �dt i� d S )N)� sessions direct-tcpip� crazy)
r r r% � listeners�globalRequestsZ
channelLookup�updater Z
SSHSessionr ZopenConnectForwardingClientZsubsystemLookup�CrazySubsystem�r$ r r r r% P s ��zConchTestAvatar.__init__c C s || j d< dS )N�foo� �r. �r$ �datar r r �
global_fooZ s
zConchTestAvatar.global_fooc C s || j d< dS )N�foo_2)r3 � datar4 r5 r r r � global_foo_2_ s
z ConchTestAvatar.global_foo_2c C sf t �|�\}}z&tj|t �| j||ft j�|d�}W n t�d d� Y dS X || j ||f< dS d S )N)Z interfacez2something went wrong with remote->local forwardingr r3 )
r �unpackGlobal_tcpip_forwardr Z listenTCPZSSHListenForwardingFactory�connZ SSHListenServerForwardingChannelr �errr- �r$ r6 ZhostZport�listenerr r r �global_tcpip_forwardd s ��
z$ConchTestAvatar.global_tcpip_forwardc C s@ t �|�\}}| j�||fd �}|s(dS | j||f= |�� dS )Nr r3 )r r; r- �get�
stopListeningr> r r r �global_cancel_tcpip_forwardt s z+ConchTestAvatar.global_cancel_tcpip_forwardc C s0 d| _ | j�� D ]}t�d| � |�� qd S )NTzstopListening %s)� loggedOutr- �valuesr �msgrB )r$ r? r r r r' ~ s zConchTestAvatar.logoutN)
r
r r r) r �skiprD r% r7 r: r@ rC r' r r r r r& D s
r& c @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dS )�ConchSessionForTestAvatarz2
An ISession adapter for ConchTestAvatar.
c C s4 || _ | | j _d| _d| _d| _d| _t�� | _dS )zq
Initialize the session and create a reference to it on the avatar for
later inspection.
NFr ) r � _testSession�cmd�proto�ptyReq�eofr �Deferred�onClose)r$ r r r r r% � s z"ConchSessionForTestAvatar.__init__c C s t �d� || _|| _d| _d S )Nzpty reqT)r rF �
_terminalType� _windowSizerL )r$ ZtermZ
windowSizeZattrsr r r �getPty� s
z ConchSessionForTestAvatar.getPtyc C s" t �d� || _t|� d| _d S )Nz
opening shell� shell)r rF rK �
EchoTransportrJ )r$ rK r r r � openShell� s
z#ConchSessionForTestAvatar.openShellc C s� || _ || _|�� d }|dkr8t|�}t�d|j� n�|dkrdt|�}|�|dd � � |�� nb|dkr�t |�}|�|dd � � |�� n6|dkr�t
|�}|�|dd � � |�� n
t � d��d | j
jj_d S )
Nr � falses echo� s secho� s eecho�bad execr3 )rJ rK �split�FalseTransportr Z callLater�loseConnectionrT �write�SuperEchoTransport�ErrEchoTransportr �
ConchErrorr r< � transport�expectedLoseConnection)r$ rK rJ �f�tr r r � execCommand� s(
z%ConchSessionForTestAvatar.execCommandc C s
d| _ d S �Nr3 )rM r1 r r r � eofReceived� s z%ConchSessionForTestAvatar.eofReceivedc C s, t �d| j � | jjj| _| j�d � d S )Nzclosed cmd "%s") r rF rJ rK r ZremoteWindowLeft�remoteWindowLeftAtCloserO �callbackr1 r r r �closed� s z ConchSessionForTestAvatar.closedN)
r
r r r) r% rR rU re rg rj r r r r rH � s rH )�
componentsc @ s e Zd Zdd� Zdd� ZdS )r0 c O s d S r" r )r$ �args�kwr r r r% � s zCrazySubsystem.__init__c C s dS )z
good ... good
Nr r1 r r r �connectionMade� s z CrazySubsystem.connectionMadeN)r
r r r% rn r r r r r0 � s r0 c @ s e Zd ZdZdd� Zdd� ZdS )r[ a
False transport should act like a /bin/false execution, i.e. just exit with
nonzero status, writing nothing to the terminal.
@ivar proto: The protocol associated with this transport.
@ivar closed: A flag tracking whether C{loseConnection} has been called yet.
c C s || _ |�| � d| _dS )zY
@type p L{twisted.conch.ssh.session.SSHSessionProcessProtocol} instance
r N�rK �makeConnectionrj �r$ �pr r r r% � s
zFalseTransport.__init__c C sL | j r
dS d| _ | j�� | j�� | j�� | j�t�tddd��� dS )zI
Disconnect the protocol associated with this transport.
Nr3 � � rj rK ZinConnectionLostZoutConnectionLostZerrConnectionLostZ processEndedr ZFailurer r1 r r r r\ � s
z FalseTransport.loseConnectionN)r
r r r) r% r\ r r r r r[ � s r[ c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )rT c C s || _ |�| � d| _d S �Nr ro rq r r r r% � s
zEchoTransport.__init__c C s: t �t|�� | j�|� | j�d� d|kr6| �� d S )N�
� )r rF �reprrK � outReceivedr\ r5 r r r r] s
zEchoTransport.writec C sL | j r
d S d| _ | j�� | j�� | j�� | j�t�tdd d ��� d S �Nr3 r rt r1 r r r r\ s
z EchoTransport.loseConnectionN�r
r r r% r] r\ r r r r rT � s rT c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )r_ c C s || _ |�| � d| _d S ru ro rq r r r r% s
zErrEchoTransport.__init__c C s | j �|� | j �d� d S �Nrv )rK � errReceivedr5 r r r r] s zErrEchoTransport.writec C sL | j r
d S d| _ | j�� | j�� | j�� | j�t�tdd d ��� d S rz rt r1 r r r r\ s
z ErrEchoTransport.loseConnectionNr{ r r r r r_ s r_ c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )r^ c C s || _ |�| � d| _d S ru ro rq r r r r% & s
zSuperEchoTransport.__init__c C s4 | j �|� | j �d� | j �|� | j �d� d S r| )rK ry r} r5 r r r r] + s zSuperEchoTransport.writec C sL | j r
d S d| _ | j�� | j�� | j�� | j�t�tdd d ��� d S rz rt r1 r r r r\ 1 s
z!SuperEchoTransport.loseConnectionNr{ r r r r r^ $ s r^ )�checkers)�channel�
connection�factory�keys)ra �userauthc @ s e Zd ZejfZdd� ZdS )�ConchTestPasswordCheckerc C s. |j dkr |jdkr t�|j �S t�td��S )N� testuser� testpasszBad credentials)ZusernameZpasswordr �succeed�fail� Exception)r$ Z credentialsr r r �requestAvatarIdB s z(ConchTestPasswordChecker.requestAvatarIdN)r
r r r~ ZIUsernamePasswordZcredentialInterfacesr� r r r r r� ? s r� c @ s e Zd Zdd� ZdS )�ConchTestSSHCheckerc C s" |dkst | j| �dk r dS dS )Nr� � FT)�lenZsuccessfulCredentials)r$ ZavatarIdr r r �areDoneJ s zConchTestSSHChecker.areDoneN)r
r r r� r r r r r� H s r� c @ sF e Zd ZdZejejd�Zdd� Z dd� Z
dd� Z d d
� Z d d � Z
d
S )�ConchTestServerFactoryr )s ssh-userauth� ssh-connectionc C s4 t � }| j�� |_| |_t| d�r*| j|_|| _|S )Nrb )�ConchTestServerZ privateKeysr� ZsupportedPublicKeysr� �hasattrrb rK )r$ ZaddrrK r r r �
buildProtocolW s
z$ConchTestServerFactory.buildProtocolc C s t j�t�t j�t�d�S �N)s ssh-rsas ssh-dss)r� �Key�
fromStringr r r1 r r r �
getPublicKeysb s
�z$ConchTestServerFactory.getPublicKeysc C s t j�t�t j�t�d�S r� )r� r� r� r r r1 r r r �getPrivateKeysh s
�z%ConchTestServerFactory.getPrivateKeysc C s dt �d�giS )a%
Diffie-Hellman primes that can be used for the
diffie-hellman-group-exchange-sha1 key exchange.
@return: The primes and generators.
@rtype: L{dict} mapping the key size to a C{list} of
C{(generator, prime)} tupple.
i s diffie-hellman-group14-sha1)r
ZgetDHGeneratorAndPrimer1 r r r � getPrimesn s ���z ConchTestServerFactory.getPrimesc C s t j�| ||�S r" )r� �
SSHFactory�
getService)r$ Ztrans�namer r r r� � s z!ConchTestServerFactory.getServiceN)r
r r Znoisyr� ZSSHUserAuthServerr� �
SSHConnection�servicesr� r� r� r� r� r r r r r� O s � r� c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �
ConchTestBaser c C s0 | j r
d S t| d�s&t�d| |f ��d| _ d S )Nrb z"unexpectedly lost connection %s
%sr3 )�doner� r �FailTest�r$ �reasonr r r �connectionLost� s
�z ConchTestBase.connectionLostc C s4 d| _ |tjkr(t�td| ||f �� | �� d S )Nr3 z*got disconnect for %s: reason %s, desc: %s)rb ra ZDISCONNECT_BY_APPLICATIONr r= r� r\ )r$ Z
reasonCodeZdescr r r � receiveError� s
���zConchTestBase.receiveErrorc C s" t �d|f ��d| _| �� d S )Nzgot unimplemented: seqid %sr3 )r r� rb r\ )r$ ZseqIDr r r �receiveUnimplemented� s z"ConchTestBase.receiveUnimplementedN)r
r r r� r� r� r� r r r r r� � s r� c @ s e Zd Zdd� ZdS )r� c C s t �| |� tj�| |� d S r" )r� r� ra �SSHServerTransportr� r r r r� � s z ConchTestServer.connectionLostN)r
r r r� r r r r r� � s r� c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) �ConchTestClientz�
@ivar _channelFactory: A callable which accepts an SSH connection and
returns a channel which will be attached to a new channel on that
connection.
c C s
|| _ d S r" )�_channelFactory�r$ ZchannelFactoryr r r r% � s zConchTestClient.__init__c C s t �| |� tj�| |� d S r" )r� r� ra �SSHClientTransportr� r r r r� � s z ConchTestClient.connectionLostc C s< |t j�t��� k}|dk}|r.|r.t�d�S t�td��S )Ns/ 85:25:04:32:58:55:96:9f:57:ee:fb:a8:1a:ea:69:dar3 zKey or fingerprint mismatch) r� r� r� r Zblobr r� r� r� )r$ �key�fpZkeyMatchZfingerprintMatchr r r �
verifyHostKey� s �
z ConchTestClient.verifyHostKeyc C s | � tdt| j��� d S )Nr� )ZrequestService�ConchTestClientAuth�ConchTestClientConnectionr� r1 r r r �connectionSecure� s �z ConchTestClient.connectionSecureN)r
r r r) r% r� r� r� r r r r r� � s
r� c @ s8 e Zd ZdZdZdZdd� Zdd� Zdd� Zdd � Z d
S ) r� r c C s( | j s| jrt�d��tj�| |� d S )Nz2got USERAUTH_SUCCESS before password and publickey)�canSucceedPassword�canSucceedPublicKeyr r� r� �SSHUserAuthClient�ssh_USERAUTH_SUCCESS)r$ Zpacketr r r r� � s
�z(ConchTestClientAuth.ssh_USERAUTH_SUCCESSc C s d| _ t�d�S )Nr3 r� )r� r r� r1 r r r � getPassword� s z ConchTestClientAuth.getPasswordc C s d| _ t�tj�t��S rf )r� r r� r� r� r� r r1 r r r �
getPrivateKey� s z!ConchTestClientAuth.getPrivateKeyc C s t j�t�S r" )r� r� r� r r1 r r r � getPublicKey� s z ConchTestClientAuth.getPublicKeyN)
r
r r Z hasTriedNoner� r� r� r� r� r� r r r r r� � s r� c @ s, e Zd ZdZdZdZdZdd� Zdd� Zd S )
r� z�
@ivar _completed: A L{Deferred} which will be fired when the number of
results collected reaches C{totalResults}.
r� r � c C s t j�| � || _d S r" )r� r� r% r� r� r r r r% � s z"ConchTestClientConnection.__init__c C s | � | j| d�� d S )N�r< )Z openChannelr� r1 r r r �serviceStarted� s z(ConchTestClientConnection.serviceStartedN) r
r r r) r� ZresultsZ totalResultsr% r� r r r r r� � s r� c @ sL e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dS )�SSHTestChannelc O s: || _ || _g | _g | _t�� | _tjj | f|�|� d S r" )
r� �_opened�received� receivedExtr rN rO r �
SSHChannelr% )r$ r� Zopenedrl �kwargsr r r r% � s
zSSHTestChannel.__init__c C s | j �|� d S r" )r� Zerrbackr� r r r �
openFailed� s zSSHTestChannel.openFailedc C s | j �| � d S r" )r� ri )r$ �ignorer r r � channelOpen� s zSSHTestChannel.channelOpenc C s | j �|� d S r" )r� �appendr5 r r r � dataReceived s zSSHTestChannel.dataReceivedc C s, |t jkr| j�|� nt�d|f � d S )Nz Unrecognized extended data: %r)r� ZEXTENDED_DATA_STDERRr� r� r rF )r$ ZdataTyper6 r r r � extReceived s
zSSHTestChannel.extReceivedc C s t �d|�\| _d S )Nz>L)�structZunpack�status)r$ r� r r r �request_exit_status s z"SSHTestChannel.request_exit_statusc C s
d| _ d S )NT)� eofCalledr1 r r r rg s zSSHTestChannel.eofReceivedc C s | j �d � d S r" )rO ri r1 r r r rj s zSSHTestChannel.closedN) r
r r r% r� r� r� r� r� rg rj r r r r r� � s r� c C s" t �dtj�t�gi�} t �| �S )z�
Produces a SSHPublicKeyChecker with an in-memory key mapping with
a single use: 'testuser'
@return: L{twisted.conch.checkers.SSHPublicKeyChecker}
r� )r~ ZInMemorySSHKeyDBr� r� r� r ZSSHPublicKeyChecker)ZconchTestPublicKeyDBr r r �conchTestPublicKeyChecker s �r� c @ sj e Zd ZdZesdZes dZddd�Zdd� Zd d
� Z d d � Z
d
d� Z dd� Z dd� Z
dd� Zdd� ZdS )�SSHProtocolTestsz^
Tests for communication between L{SSHServerTransport} and
L{SSHClientTransport}.
� can't run without cryptography�Cannot run without PyASN1r+ c s� t �� �td�| _t�| j�}t� }|�t� � |�t � � |�|� t
� }||_|� � |� d�| _
t| j
�| _t� ��fdd��| _t| j�| _| j
�| j� | j�| j� �S )z�
Create a connected SSH client and server protocol pair and return a
L{Deferred} which fires with an L{SSHTestChannel} instance connected to
a channel on that SSH connection.
r� Nc s t ��fd| i� ��S )Nr< )r� r� �r� r� �resultr r �<lambda>C � z:SSHProtocolTests._ourServerOurClientTest.<locals>.<lambda>)r rN r! �realmr ZPortalr� ZregisterCheckerr� r� r� � startFactoryr� Zserverr ZclientTransportr� ZclientZserverTransportrp )r$ r� r� rr ZsshpcZfacr r� r �_ourServerOurClientTest0 s&
� z(SSHProtocolTests._ourServerOurClientTestc sd � � � }� fdd�}|�|� � fdd�}|�|� � fdd�}|�|� � fdd�}|�|� |S ) z�
Run the Conch server against the Conch client. Set up several different
channels which exercise different behaviors and wait for them to
complete. Verify that the channels with errors log them.
c s& | � _ � �| j�| dt�d�d�t�S )N� subsystems not-crazyr3 �r �
assertFailurer< � sendRequestr �NSr� �r r1 r r � cbSubsystemT s ��zFSSHProtocolTests.test_subsystemsAndGlobalRequests.<locals>.cbSubsystemc s � j }|j�|dt�d�d�S )Nr� r, r3 �r r< r� r r� ��ignoredr r1 r r �cbNotCrazyFailed\ s �zKSSHProtocolTests.test_subsystemsAndGlobalRequests.<locals>.cbNotCrazyFailedc s\ � j }|j�ddd�}|j�ddd�}|�� jd� � �|j�ddd�t�}t�|||g�S )N� foo� barr3 s foo-2� bar2r9 ) r r< ZsendGlobalRequest� addCallback� assertEqualr� r� r �
gatherResults)r� r Zd1Zd2Zd3r1 r r �cbGlobalRequestsb s �zKSSHProtocolTests.test_subsystemsAndGlobalRequests.<locals>.cbGlobalRequestsc s> � � � jjjddd�� � j}d|jj_|j�� |� � d S )Nr� r� )r2 r8 T)
r� r� r r. r r< ra rb ZserviceStoppedr\ r� r1 r r �
disconnectp s �
zESSHProtocolTests.test_subsystemsAndGlobalRequests.<locals>.disconnect�r� r� )r$ r r� r� r� r� r r1 r � test_subsystemsAndGlobalRequestsL s
z1SSHProtocolTests.test_subsystemsAndGlobalRequestsc st �� � }t�ddd�� � �fdd�}|�|� �fdd�}|�|� �fdd �}|�|� �fd
d �}|�|� |S ) z�
L{SSHChannel.sendRequest} can open a shell with a I{pty-req} request,
specifying a terminal type and window size.
� conch-test-term�� �P r r r� c s | �_ | j�| d� d�S )Ns pty-reqr3 )r r< r� r� �r6 r$ r r � cbChannel� s z.SSHProtocolTests.test_shell.<locals>.cbChannelc sd � j jjjd j}� �|j� j j� � �|jd� � �|jd� � � |j
� � j }|j� |ddd�S )Nr r� r� rS r� r3 )
r� r r< Zchannelsr ZassertIsr� rP rQ �
assertTruerL r r� )r� r r r1 r r �cbPty� s z*SSHProtocolTests.test_shell.<locals>.cbPtyc s6 � j �d� � j j�� j � t�� j j� jjj jg�S )Ns testing the shell! )
r r] r< ZsendEOFr r� rO r� r rI �r� r1 r r �cbShell� s
�z,SSHProtocolTests.test_shell.<locals>.cbShellc sZ � j jdkr t�d� j jf � � �d�� j j�d� � �� j j� � �� j j
j j � d S )Nr z shell exit status was not 0: %ir� s testing the shell!
)
r r� r rF r� �joinr� r� r� r� r rI rM r� r1 r r �cbExited� s � �
�z-SSHProtocolTests.test_shell.<locals>.cbExited)r� r ZpackRequest_pty_reqr� )r$ r r� r� r� r� r r� r �
test_shell} s �
zSSHProtocolTests.test_shellc s8 � � � }� fdd�}|�|� � fdd�}|�|� |S )z�
If L{SSHChannel.sendRequest} issues an exec which the server responds to
with an error, the L{Deferred} it returns fires its errback.
c s& | � _ � �| j�| dt�d�d�t�S )N� execs jumboliahr3 r� r� r1 r r r� � s ��z3SSHProtocolTests.test_failedExec.<locals>.cbChannelc s$ � � tj�}� �|d jjd� d S )Nr )rY N)�flushLoggedErrorsr r` r� �valuerl �r� �errorsr1 r r �cbFailed� s z2SSHProtocolTests.test_failedExec.<locals>.cbFailedr� )r$ r r� r� r r1 r �test_failedExec� s
z SSHProtocolTests.test_failedExecc sN � � � }� fdd�}|�|� � fdd�}|�|� � fdd�}|�|� |S )z�
When the process started by a L{SSHChannel.sendRequest} exec request
exits, the exit status is reported to the channel.
c s | � _ | j�| dt�d�d�S )Nr� rV r3 r� r� r1 r r r� � s �z5SSHProtocolTests.test_falseChannel.<locals>.cbChannelc s � j jS r" �r rO r� r1 r r �cbExec� s z2SSHProtocolTests.test_falseChannel.<locals>.cbExecc s$ � � � jjg � � �� jjd� d S ru )r� r r� ZassertNotEqualr� r� r1 r r �cbClosed� s z4SSHProtocolTests.test_falseChannel.<locals>.cbClosedr� �r$ r r� r r r r1 r �test_falseChannel� s
z"SSHProtocolTests.test_falseChannelc sT � j ddd�}� fdd�}|�|� � fdd�}|�|� � fdd �}|�|� |S )
z�
Bytes sent over the extended channel for stderr data are delivered to
the channel's C{extReceived} method.
� rW �Z localWindowZlocalMaxPacketc s | � _ | j�| dt�d�d�S )Nr� s eecho hellor3 r� r� r1 r r r� � s �z5SSHProtocolTests.test_errorChannel.<locals>.cbChannelc s t �� jj� jjjjg�S r" �r r� r rO r� r rI r� r1 r r r � s
�z2SSHProtocolTests.test_errorChannel.<locals>.cbExecc sp � � � jjg � � � d�� jj�d� � � � jjd� � �� jj� � � � jjd� � � � jj� j j
j j � d S )Nr� � hello
r r )
r� r r� r� r� r� r� r� �localWindowLeftr� r rI rh r� r1 r r r � s
�z4SSHProtocolTests.test_errorChannel.<locals>.cbClosedr� r r r1 r �test_errorChannel� s
z"SSHProtocolTests.test_errorChannelc s, � � � �d�t�}� fdd�}|�|� |S )z�
When an attempt is made to open an unknown channel type, the L{Deferred}
returned by L{SSHChannel.sendRequest} fires its errback.
s crazy-unknown-channelc s4 � � tj�}� �|d jjd� � �t|�d� d S )Nr )� zunknown channelr3 )r� r r` r� r� rl r� r� r1 r r r� s z6SSHProtocolTests.test_unknownChannel.<locals>.cbFailed)r� r� r� r� )r$ �dr� r r1 r �test_unknownChannel� s �
z$SSHProtocolTests.test_unknownChannelc sT � j ddd�}� fdd�}|�|� � fdd�}|�|� � fdd �}|�|� |S )
zc
An L{SSHChannel} can be configured with a maximum packet size to
receive.
� r3 r c s | � _ | j�| dt�d�d�S )Nr� s secho hellor3 r� r� r1 r r r� s �z2SSHProtocolTests.test_maxPacket.<locals>.cbChannelc s � j jS r" r r� r1 r r r s z/SSHProtocolTests.test_maxPacket.<locals>.cbExecc s^ � � � jjd� � � d�� jj�d� � � d�� jj�d� � � � jjd� � �� jj� d S )Nr r� r r ) r� r r� r� r� r� r r� r� r� r1 r r r s
z1SSHProtocolTests.test_maxPacket.<locals>.cbClosedr� r r r1 r �test_maxPacket s �
z SSHProtocolTests.test_maxPacketc sT � j ddd�}� fdd�}|�|� � fdd�}|�|� � fdd �}|�|� |S )
ze
Normal standard out bytes are sent to the channel's C{dataReceived}
method.
r rW r c s | � _ | j�| dt�d�d�S )Nr� s
echo hellor3 r� r� r1 r r r� 1 s �z-SSHProtocolTests.test_echo.<locals>.cbChannelc s t �� jj� jjjjg�S r" r r� r1 r r �cbEcho7 s
�z*SSHProtocolTests.test_echo.<locals>.cbEchoc s` � � � jjd� � � d�� jj�d� � � � jjd� � �� jj� � � � jj� jj j
j � d S )Nr r� r r ) r� r r� r� r� r r� r� r� r rI rh r� r1 r r r = s
�z,SSHProtocolTests.test_echo.<locals>.cbClosedr� )r$ r r� r r r r1 r � test_echo* s
zSSHProtocolTests.test_echoN)r+ )r
r r r) r rG r r� r� r� r� r r
r
r r r r r r r� $ s
1/ r� c @ sF e Zd ZesdZesdZddd�Zdd� Zdd � Zd
d � Z d d
� Z
dS )�SSHFactoryTestsr� r� Nc s6 t �� }dd� }� fdd�|_| |_|_|�� |S )Nc S s dt �d �iS )Nzssh-rsa)r� r� r r r r r� T r� z0SSHFactoryTests.makeSSHFactory.<locals>.<lambda>c s � S r" r r ��primesr r r� U r� )r� r� r� r� r� r� )r$ r Z
sshFactoryZgpkr r r �makeSSHFactoryR s z SSHFactoryTests.makeSSHFactoryc C s$ | � � }|�d�}| �|tj� dS )zc
By default, buildProtocol() constructs an instance of
SSHServerTransport.
N)r r� ZassertIsInstancera r� )r$ r� r r r r �test_buildProtocol[ s
z"SSHFactoryTests.test_buildProtocolc s: g � � fdd�}| � � }||_|�d� | �dg� � dS )zc
buildProtocol() calls 'self.protocol()' to construct a protocol
instance.
c s � � | � t�� S r" )r� ra r� )rl �Zcallsr r � makeProtocolk s
zHSSHFactoryTests.test_buildProtocolRespectsProtocol.<locals>.makeProtocolNr )r r r� r� )r$ r r� r r r �"test_buildProtocolRespectsProtocole s
z2SSHFactoryTests.test_buildProtocolRespectsProtocolc C s6 | j dd�}|�d�}| �d|j� | �d|j� dS )zg
Group key exchanges are not supported when we don't have the primes
database.
Nr �" diffie-hellman-group-exchange-sha1�$ diffie-hellman-group-exchange-sha256)r r� Z assertNotIn�supportedKeyExchanges)r$ �f1Zp1r r r �test_buildProtocolNoPrimest s
� �z*SSHFactoryTests.test_buildProtocolNoPrimesc C s: | j ddid�}|�d�}| �d|j� | �d|j� dS )zU
Group key exchanges are supported when we have the primes database.
r3 )r� r r Nr r )r r� ZassertInr )r$ �f2Zp2r r r � test_buildProtocolWithPrimes� s
� �z,SSHFactoryTests.test_buildProtocolWithPrimes)N) r
r r r rG r r r r r r r r r r r J s
r c @ sV e Zd ZdZesdZes dZer*eej �Z dd� Z
dd� Z dd � Z d
d � Z
d d
� ZdS )�MPTestszs
Tests for L{common.getMP}.
@cvar getMP: a method providing a MP parser.
@type getMP: C{callable}
r� r� c C s | � | �d�d� dS )z�
L{common.getMP} should parse the a multiple precision integer from a
string: a 4-byte length followed by length bytes of the integer.
s )r3 r� N�r� �getMPr1 r r r �
test_getMP� s �zMPTests.test_getMPc C s | � | �d�d� dS )zv
L{common.getMP} should be able to parse a big enough integer
(that doesn't fit on one byte).
s )ir� Nr" r1 r r r �test_getMPBigInteger� s �z MPTests.test_getMPBigIntegerc C s | � | �dd�d� dS )zg
L{common.getMP} has the ability to parse multiple integer in the same
string.
s r� )r3 r� r� Nr" r1 r r r �test_multipleGetMP� s ��zMPTests.test_multipleGetMPc C s | � | �d�d� dS )zu
When more data than needed is sent to L{common.getMP}, it should return
the remaining data.
s foo)r3 r� Nr" r1 r r r �test_getMPRemainingData� s �z MPTests.test_getMPRemainingDatac C s | � tj| jd� dS )z�
When the string passed to L{common.getMP} doesn't even make 5 bytes,
it should raise a L{struct.error}.
s N)Z assertRaisesr� r r# r1 r r r �test_notEnoughData� s zMPTests.test_notEnoughDataN)r
r r r) r rG r � staticmethodr r# r$ r% r& r' r( r r r r r! � s
r! c @ s e Zd ZdZesdZdd� ZdS )�GMPYInstallDeprecationTestszI
Tests for the deprecation of former GMPY accidental public API.
r* c C s> t �� | �| jg�}| �t|�d� | �|d d d� dS )zD
L{twisted.conch.ssh.common.install} is deprecated.
r3 r �messagezAtwisted.conch.ssh.common.install was deprecated in Twisted 16.5.0N)r �installZ
flushWarnings�test_deprecatedr� r� )r$ �warningsr r r r- � s
�z+GMPYInstallDeprecationTests.test_deprecatedN)r
r r r) r rG r- r r r r r* � s r* )Rr) Z
__future__r r r� Ztwisted.python.reflectr r r Ztwisted.conch.sshr r r r
Z
twisted.conchr r Ztwisted.conch.test.keydatar r r r Z twisted.credr Ztwisted.cred.errorr Ztwisted.internetr r r Ztwisted.internet.errorr Ztwisted.pythonr r Z
twisted.trialr Ztwisted.conch.test.loopbackr �objectr! r r& rH rk ZregisterAdapterZISessionZProtocolr0 r[ rT r_ r^ r~ r r� r� r� ra r� r� ZSSHProtocolCheckerr� r� r� r� r� r� r� r� r� r� r� r� r� r� r� ZTestCaser� r r! r* r r r r �<module> sf BC � 5 *
(HA