| Server IP : 209.38.156.173 / Your IP : 216.73.216.122 [ Web Server : Apache/2.4.52 (Ubuntu) System : Linux lakekumayuhotel 5.15.0-136-generic #147-Ubuntu SMP Sat Mar 15 15:53:30 UTC 2025 x86_64 User : root ( 0) PHP Version : 8.1.2-1ubuntu2.22 Disable Function : NONE Domains : 2 Domains MySQL : OFF | cURL : ON | WGET : ON | Perl : ON | Python : OFF | Sudo : ON | Pkexec : ON Directory : /usr/lib/python3/dist-packages/twisted/conch/test/__pycache__/ |
Upload File : |
o
�b�� � @ s� d Z ddlZddlmZmZmZmZ ddlmZ ddl m
Z
ddl m Z ddl
mZmZmZ ddlmZ dd lmZmZ dd
lmZ dd lmZ ed �Zed
�Zergddl m Z m Z ddl m Z m!Z!m"Z"m#Z# nG dd� d�Z G dd� d�Z$G dd� de j%�Z&G dd� d�Z'ddlm(Z( er�e(�)e'e&e#j*� G dd� dej+�Z,G dd � d �Z-G d d � d �Z.G d d � d �Z/G d!d"� d"�Z0edu�r-edu�r-dd#l m1Z1 dd$l m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7 G d%d&� d&�Z8G d'd(� d(e1j9�Z:G d)d*� d*e4j;�Z<G d+d,� d,�Z=G d-d.� d.e=e6j>�Z?G d/d0� d0e=e6j@�ZAG d1d2� d2e7jB�ZCG d3d4� d4e3jD�ZEG d5d6� d6e2jF�ZGd7d8� ZHG d9d:� d:ejI�ZJG d;d<� d<ejI�ZKG d=d>� d>ejI�ZLG d?d@� d@ejI�ZMdS )Az!
Tests for L{twisted.conch.ssh}.
� N)�privateDSA_openssh�privateRSA_openssh�publicDSA_openssh�publicRSA_openssh)�
LoopbackRelay)�portal)�UnauthorizedLogin)�defer�protocol�reactor)�ProcessTerminated)�failure�log)�
requireModule)�unittest� cryptography�pyasn1)�avatar�error)�_kex�common�
forwarding�sessionc @ 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 c @ 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 �
|| _ d S �N)�expectedAvatarID)�selfr" r r r �__init__4 �
zConchTestRealm.__init__c G sN || j kr | jdurtd��t� | _|d | j| jjfS td| j �d|����)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 z may log in, not )r" r r �ConchTestAvatar�logout)r# �avatarID�mind�
interfacesr r r �
requestAvatar7 s
�z ConchTestRealm.requestAvatar)r r r �__doc__r r$ r+ r r r r r ' s
r c @ sL e Zd ZdZes
dZdZdd� 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�globalRequests�
channelLookup�updater �
SSHSessionr �openConnectForwardingClient�subsystemLookup�CrazySubsystem�r# r r r r$ S s ��zConchTestAvatar.__init__c C � || j d< dS )N�foo� �r1 �r# �datar r r �
global_foo_ �
zConchTestAvatar.global_fooc C r9 )N�foo_2)r; � datar< r= r r r � global_foo_2c r@ z ConchTestAvatar.global_foo_2c C sh t �|�\}}ztj|t �| j||ft j�|d�}W n ty* t� d d� Y dS w || j
||f< dS )N)� interfacez2something went wrong with remote->local forwardingr r; ) r �unpackGlobal_tcpip_forwardr � listenTCP�SSHListenForwardingFactory�conn� SSHListenServerForwardingChannel�
BaseExceptionr �errr0 �r# r> �host�port�listenerr r r �global_tcpip_forwardg s �
� �z$ConchTestAvatar.global_tcpip_forwardc C s@ t �|�\}}| j�||fd �}|sdS | j||f= |�� dS )Nr r; )r rE r0 �get�
stopListeningrL r r r �global_cancel_tcpip_forwardx s z+ConchTestAvatar.global_cancel_tcpip_forwardc C s0 d| _ | j�� D ]
}t�d| � |�� qd S )NTzstopListening %s)� loggedOutr0 �valuesr �msgrR )r# rO r r r r' � s
�zConchTestAvatar.logoutN)
r r r r, r �skiprT r$ r? rC rP rS r' r r r r r& F 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 rV �
_terminalType� _windowSizer\ )r# �term�
windowSize�attrsr r r �getPty� s
z ConchSessionForTestAvatar.getPtyc C s" t �d� || _t|� d| _d S )Nz
opening shell� shell)r rV r[ �
EchoTransportrZ )r# r[ r r r � openShell� s
z#ConchSessionForTestAvatar.openShellc C s� || _ || _|�� d }|dkr t|�}t�d|j� nG|dkr2t|�}|�|dd � � |�� n1|dkrHt |�}|�|dd � � |�� n|dkr^t
|�}|�|dd � � |�� nt � d��d | j
jj_d S )
Nr � falses echo� s secho� s eecho�bad execr; )rZ r[ �split�FalseTransportr � callLater�loseConnectionrg �write�SuperEchoTransport�ErrEchoTransportr �
ConchErrorr rH � transport�expectedLoseConnection)r# r[ rZ �f�tr r r � execCommand� s(
z%ConchSessionForTestAvatar.execCommandc C �
d| _ d S �Nr; )r] r8 r r r � eofReceived� r% z%ConchSessionForTestAvatar.eofReceivedc C s, t �d| j � | jjj| _| j�d � d S )Nzclosed cmd "%s") r rV rZ r[ r �remoteWindowLeft�remoteWindowLeftAtCloser_ �callbackr8 r r r �closed� s z ConchSessionForTestAvatar.closedN)
r r r r, r$ re rh ry r| r� r r r r rX � s
rX )�
componentsc @ s e Zd Zdd� Zdd� ZdS )r7 c O s d S r! r )r# �args�kwr r r r$ � s zCrazySubsystem.__init__c C s dS )z
good ... good
Nr r8 r r r �connectionMade� � z CrazySubsystem.connectionMadeN)r r r r$ r� r r r r r7 � s r7 c @ s e Zd ZdZdd� Zdd� ZdS )rn 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�r[ �makeConnectionr� �r# �pr r r r$ � s
zFalseTransport.__init__c C sL | j rdS d| _ | j�� | j�� | j�� | j�t�tddd��� dS )zI
Disconnect the protocol associated with this transport.
Nr; � � r� r[ �inConnectionLost�outConnectionLost�errConnectionLost� processEndedr
�Failurer r8 r r r rp � s
z FalseTransport.loseConnectionN)r r r r, r$ rp r r r r rn � s rn c @ �$ e Zd Zdd� Zdd� Zdd� ZdS )rg c C � || _ |�| � d| _d S �Nr r� r� r r r r$ � �
zEchoTransport.__init__c C s> t �t|�� | j�|� | j�d� d|v r | �� d S d S )N�
� )r rV �reprr[ � outReceivedrp r= r r r rq � s �zEchoTransport.writec C �L | j rd S d| _ | j�� | j�� | j�� | j�t�tdd d ��� d S �Nr; r r� r8 r r r rp �
z EchoTransport.loseConnectionN�r r r r$ rq rp r r r r rg � s rg c @ r� )rs c C r� r� r� r� r r r r$ r� zErrEchoTransport.__init__c C s | j �|� | j �d� d S �Nr� )r[ � errReceivedr= r r r rq s zErrEchoTransport.writec C r� r� r� r8 r r r rp r� z ErrEchoTransport.loseConnectionNr� r r r r rs s rs c @ r� )rr c C r� r� r� r� r r r r$ $ r� zSuperEchoTransport.__init__c C s4 | j �|� | j �d� | j �|� | j �d� d S r� )r[ r� r� r= r r r rq ) s zSuperEchoTransport.writec C r� r� r� r8 r r r rp / r� z!SuperEchoTransport.loseConnectionNr� r r r r rr # s rr )�checkers)�channel�
connection�factory�keysru �userauthc @ s e Zd ZejfZdd� ZdS )�ConchTestPasswordCheckerc C s. |j dkr|jdkrt�|j �S t�td��S )N� testuser� testpasszBad credentials)�username�passwordr �succeed�fail� Exception)r# � credentialsr r r �requestAvatarIdG s
z(ConchTestPasswordChecker.requestAvatarIdN)r r r r� �IUsernamePassword�credentialInterfacesr� r r r r r� D s r� c @ � e Zd Zdd� ZdS )�ConchTestSSHCheckerc C s" |dks
t | j| �dk rdS dS )Nr� � FT)�len�successfulCredentials)r# �avatarIdr r r �areDoneP s zConchTestSSHChecker.areDoneN)r r r r� r r r r r� O � r� c @ sF e Zd ZdZejejd�Zdd� Z dd� Z
dd� Z d d
� Z d d � Z
d
S )�ConchTestServerFactoryF)s ssh-userauth� ssh-connectionc C s4 t � }| j�� |_| |_t| d�r| j|_|| _|S )Nrv )�ConchTestServer� privateKeysr� �supportedPublicKeysr� �hasattrrv r[ )r# �addrr[ r r r �
buildProtocol] s
z$ConchTestServerFactory.buildProtocolc C � t j�t�t j�t�d�S �N)s ssh-rsas ssh-dss)r� �Key�
fromStringr r r8 r r r �
getPublicKeysh �
�z$ConchTestServerFactory.getPublicKeysc C r� r� )r� r� r� r r r8 r r r �getPrivateKeysn r� 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 �getDHGeneratorAndPrimer8 r r r � getPrimest s
z ConchTestServerFactory.getPrimesc C s t j�| ||�S r! )r� �
SSHFactory�
getService)r# �trans�namer r r r� � � z!ConchTestServerFactory.getServiceN)r r r �noisyr� �SSHUserAuthServerr� �
SSHConnection�servicesr� r� r� r� r� r r r r r� U s � r� c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �
ConchTestBaser c C s4 | j rd S t| d�st�d| � d|� ���d| _ d S )Nrv z unexpectedly lost connection �
r; )�doner� r �FailTest�r# �reasonr r r �connectionLost� s
�
z ConchTestBase.connectionLostc C s4 d| _ |tjkrt�td| ||f �� | �� d S )Nr; z*got disconnect for %s: reason %s, desc: %s)rv ru �DISCONNECT_BY_APPLICATIONr rK r� rp )r# �
reasonCode�descr r r � receiveError� s
��� zConchTestBase.receiveErrorc C s t �d|� ���)Nzgot unimplemented: seqid )r r� )r# �seqIDr r r �receiveUnimplemented� r� z"ConchTestBase.receiveUnimplementedN)r r r r� r� r� r� r r r r r� � s
r� c @ r� )r� c C � t �| |� tj�| |� d S r! )r� r� ru �SSHServerTransportr� r r r r� � � z ConchTestServer.connectionLostN)r r r r� r r r r r� � r� 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 r r! )�_channelFactory�r# �channelFactoryr r r r$ � r% zConchTestClient.__init__c C r� r! )r� r� ru �SSHClientTransportr� r r r r� � r� z ConchTestClient.connectionLostc C s< |t j�t��� k}|dk}|r|rt�d�S t�td��S )Ns/ 85:25:04:32:58:55:96:9f:57:ee:fb:a8:1a:ea:69:dar; zKey or fingerprint mismatch) r� r� r� r �blobr r� r� r� )r# �key�fp�keyMatch�fingerprintMatchr r r �
verifyHostKey� s
z ConchTestClient.verifyHostKeyc C s | � tdt| j��� d S )Nr� )�requestService�ConchTestClientAuth�ConchTestClientConnectionr� r8 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 | jr t�d��tj�| |� d S )Nz2got USERAUTH_SUCCESS before password and publickey)�canSucceedPassword�canSucceedPublicKeyr r� r� �SSHUserAuthClient�ssh_USERAUTH_SUCCESS)r# �packetr r r r� � s
�z(ConchTestClientAuth.ssh_USERAUTH_SUCCESSc C s d| _ t�d�S )Nr; r� )r� r r� r8 r r r � getPassword� s
z ConchTestClientAuth.getPasswordc C s d| _ t�tj�t��S r{ )r� r r� r� r� r� r r8 r r r �
getPrivateKey� � z!ConchTestClientAuth.getPrivateKeyc C s t j�t�S r! )r� r� r� r r8 r r r � getPublicKey� s z ConchTestClientAuth.getPublicKeyN)
r r r � 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�rH )� openChannelr� r8 r r r �serviceStarted� s z(ConchTestClientConnection.serviceStartedN) r r r r, r� �results� 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 | g|�R i |�� d S r! )
r� �_opened�received� receivedExtr r^ r_ r� �
SSHChannelr$ )r# r� �openedr� �kwargsr r r r$ � s
zSSHTestChannel.__init__c C � | j �|� d S r! )r �errbackr� r r r �
openFailed� r� zSSHTestChannel.openFailedc C s | j �| � d S r! )r r )r# �ignorer r r � channelOpen� r� zSSHTestChannel.channelOpenc C r r! )r �appendr= r r r � dataReceived� r� zSSHTestChannel.dataReceivedc C s. |t jkr
| j�|� d S t�d|��� d S )Nz Unrecognized extended data: )r� �EXTENDED_DATA_STDERRr r r rV )r# �dataTyper> r r r � extReceived s
zSSHTestChannel.extReceivedc C s t �d|�\| _d S )Nz>L)�struct�unpack�status)r# r r r r �request_exit_status s z"SSHTestChannel.request_exit_statusc C rz )NT)� eofCalledr8 r r r r| r% zSSHTestChannel.eofReceivedc C s | j �d � d S r! )r_ r r8 r r r r� r� zSSHTestChannel.closedN) r r r r$ r r
r r r r| r� 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� �InMemorySSHKeyDBr� r� r� r �SSHPublicKeyChecker)�conchTestPublicKeyDBr r r �conchTestPublicKeyChecker s �
r c @ sj e Zd ZdZes
dZesdZddd�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 )NrH )r r� �r r� �resultr r �<lambda>= s z:SSHProtocolTests._ourServerOurClientTest.<locals>.<lambda>)r r^ r �realmr �Portalr� �registerCheckerr� r r� � startFactoryr� �serverr �clientTransportr� �client�serverTransportr� )r# r� r r� �sshpc�facr r r �_ourServerOurClientTest* 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 �& | � _ � �| j�| dt�d�d�t�S )N� subsystems not-crazyr; �r� �
assertFailurerH � sendRequestr �NSr� �r� r8 r r � cbSubsystemN s ��zFSSHProtocolTests.test_subsystemsAndGlobalRequests.<locals>.cbSubsystemc s � j }|j�|dt�d�d�S )Nr. r/ r; �r� rH r1 r r2 ��ignoredr� r8 r r �cbNotCrazyFailedY � �zKSSHProtocolTests.test_subsystemsAndGlobalRequests.<locals>.cbNotCrazyFailedc s\ � j }|j�ddd�}|j�ddd�}|�� jd� � �|j�ddd�t�}t�|||g�S )N� foo� barr; s foo-2� bar2rB ) r� rH �sendGlobalRequest� addCallback� assertEqualr0 r� r �
gatherResults)r7 r� �d1�d2�d3r8 r r �cbGlobalRequestsa s �zKSSHProtocolTests.test_subsystemsAndGlobalRequests.<locals>.cbGlobalRequestsc s> � � � jjjddd�� � j}d|jj_|j�� |� � d S )Nr; r< )r: rA T)
r? r" r r1 r� rH ru rv �serviceStoppedrp r6 r8 r r �
disconnectp s �
zESSHProtocolTests.test_subsystemsAndGlobalRequests.<locals>.disconnect�r, r> )r# r� r4 r8 rD rF r r8 r � test_subsystemsAndGlobalRequestsF 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 � c s | �_ | j�| d� d�S )Ns pty-reqr; )r� rH r1 r3 �r> r# r r � cbChannel� r� z.SSHProtocolTests.test_shell.<locals>.cbChannelc sd � j jjjd j}� �|j� j j� � �|jd� � �|jd� � � |j
� � j }|j� |ddd�S )Nr rI rJ rf rM r; )
r" r rH �channelsr �assertIsr? r` ra �
assertTruer\ r� r1 )r7 r r� r8 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� rq rH �sendEOFr r@ r_ r" r rY �r7 r8 r r �cbShell� s
�z,SSHProtocolTests.test_shell.<locals>.cbShellc sZ � j jdkrt�d� j jf � � �d�� j j�d� � �� j j� � �� j j
j j � d S )Nr z shell exit status was not 0: %irM s testing the shell!
)
r� r r rV r? �joinr rR r r" r rY r] rU r8 r r �cbExited� s �z-SSHProtocolTests.test_shell.<locals>.cbExited)r, r �packRequest_pty_reqr> )r# r� rO rS rV rX r rN 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 r- )N� execs jumboliahr; r/ r3 r8 r r rO � s
�z3SSHProtocolTests.test_failedExec.<locals>.cbChannelc s$ � � tj�}� �|d jjd� d S )Nr )rl N)�flushLoggedErrorsr rt r? �valuer� �r7 �errorsr8 r r �cbFailed� s z2SSHProtocolTests.test_failedExec.<locals>.cbFailedrG )r# r� rO r` r r8 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 � | � _ | j�| dt�d�d�S )Nr[ ri r; r5 r3 r8 r r rO � s z5SSHProtocolTests.test_falseChannel.<locals>.cbChannelc � � j jS r! �r� r_ rU r8 r r �cbExec� � z2SSHProtocolTests.test_falseChannel.<locals>.cbExecc s$ � � � jjg � � �� jjd� d S r� )r? r� r �assertNotEqualr rU r8 r r �cbClosed� s z4SSHProtocolTests.test_falseChannel.<locals>.cbClosedrG �r# r� rO re rh r r8 r �test_falseChannel� s
z"SSHProtocolTests.test_falseChannelc �T � 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.
� rj �� localWindow�localMaxPacketc rb )Nr[ s eecho hellor; r5 r3 r8 r r rO � r9 z5SSHProtocolTests.test_errorChannel.<locals>.cbChannelc � t �� jj� jjjjg�S r! �r r@ r� r_ r" r rY rU r8 r r re � � �z2SSHProtocolTests.test_errorChannel.<locals>.cbExecc sp � � � jjg � � � d�� jj�d� � � � jjd� � �� jj� � � � jjd� � � � jj� j j
j j � d S )NrM � hello
r rl )
r? r� r rW r r rR r �localWindowLeftr" r rY r~ rU r8 r r rh � s
�z4SSHProtocolTests.test_errorChannel.<locals>.cbClosedrG ri r r8 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 channelr; )r\ r rt r? r] r� r� r^ r8 r r r`
s z6SSHProtocolTests.test_unknownChannel.<locals>.cbFailed)r0 r, r� r> )r# �dr` r r8 r �test_unknownChannel s
�
z$SSHProtocolTests.test_unknownChannelc rk )
zc
An L{SSHChannel} can be configured with a maximum packet size to
receive.
� r; rm c rb )Nr[ s secho hellor; r5 r3 r8 r r rO r9 z2SSHProtocolTests.test_maxPacket.<locals>.cbChannelc rc r! rd rU r8 r r re & rf z/SSHProtocolTests.test_maxPacket.<locals>.cbExecc s^ � � � jjd� � � d�� jj�d� � � d�� jj�d� � � � jjd� � �� jj� d S )Nr rM rs ry ) r? r� r rW r r rt rR r rU r8 r r rh + s
z1SSHProtocolTests.test_maxPacket.<locals>.cbClosedrG ri r r8 r �test_maxPacket s
z SSHProtocolTests.test_maxPacketc rk )
ze
Normal standard out bytes are sent to the channel's C{dataReceived}
method.
rl rj rm c rb )Nr[ s
echo hellor; r5 r3 r8 r r rO < r9 z-SSHProtocolTests.test_echo.<locals>.cbChannelc rp r! rq rU r8 r r �cbEchoD rr z*SSHProtocolTests.test_echo.<locals>.cbEchoc s` � � � jjd� � � d�� jj�d� � � � jjd� � �� jj� � � � jj� jj j
j � d S )Nr rM rs rl ) r? r� r rW r rt rR r r" r rY r~ rU r8 r r rh K s
�z,SSHProtocolTests.test_echo.<locals>.cbClosedrG )r# r� rO r{ rh r r8 r � test_echo5 s
zSSHProtocolTests.test_echoN)r. )r r r r, r rW r r, rH rZ ra rj ru rx rz r| r r r r r s
70$ r c @ sN e Zd ZesdZes dZddd�Zdd� Zdd � Zd
d � Z d d
� Z
dd� 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! c s z0SSHFactoryTests.makeSSHFactory.<locals>.<lambda>c s � S r! r r ��primesr r r! d r� )r� r� r� r� r� r% )r# r �
sshFactory�gpkr r~ r �makeSSHFactorya s z SSHFactoryTests.makeSSHFactoryc C s$ | � � }|�d�}| �|tj� dS )zc
By default, buildProtocol() constructs an instance of
SSHServerTransport.
N)r� r� �assertIsInstanceru r� )r# r� r
r r r �test_buildProtocoli 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 ru r� )r� ��callsr r � makeProtocoly s
zHSSHFactoryTests.test_buildProtocolRespectsProtocol.<locals>.makeProtocolNr )r� r
r� r? )r# r� r� r r� r �"test_buildProtocolRespectsProtocolr 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� � assertNotIn�supportedKeyExchanges)r# �f1�p1r r r �test_buildProtocolNoPrimes� 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.
r; )r� rv r~ Nr� r� )r� r� �assertInr� �r# �f2�p2r r r � test_buildProtocolWithPrimes� s
z,SSHFactoryTests.test_buildProtocolWithPrimesc C s* | � � }|�d�}| �dd�|j�� dS )z_
ECDSA key exchanges are listed with 256 having a higher priority among ECDSA.
Ns8 ecdh-sha2-nistp256,ecdh-sha2-nistp384,ecdh-sha2-nistp521� ,)r� r� r� rW r� r� r r r �test_buildProtocolKexECDSA� s
�z*SSHFactoryTests.test_buildProtocolKexECDSAr! ) r r r r rW r r� r� r� r� r� r� r r r r r} Y s
r} c @ sV e Zd ZdZes
dZesdZereej �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 � | � | �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 )r; rM N�r? �getMPr8 r r r �
test_getMP� � zMPTests.test_getMPc C r� )zv
L{common.getMP} should be able to parse a big enough integer
(that doesn't fit on one byte).
s )irM Nr� r8 r r r �test_getMPBigInteger� �
�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� )r; r� rM Nr� r8 r r r �test_multipleGetMP� s ��zMPTests.test_multipleGetMPc C r� )zu
When more data than needed is sent to L{common.getMP}, it should return
the remaining data.
s foo)r; r: Nr� r8 r r r �test_getMPRemainingData� r� 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)� assertRaisesr r r� r8 r r r �test_notEnoughData� r� zMPTests.test_notEnoughDataN)r r r r, r rW r � staticmethodr r� r� r� r� r� r� r r r r r� � s
r� c @ s e Zd ZdZes
dZdd� 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.
r; r �messagezAtwisted.conch.ssh.common.install was deprecated in Twisted 16.5.0N)r �install�
flushWarnings�test_deprecatedr? r� )r# �warningsr r r r� � s
�z+GMPYInstallDeprecationTests.test_deprecatedN)r r r r, r rW r� r r r r r� � s
r� )Nr, r �twisted.conch.test.keydatar r r r �twisted.conch.test.loopbackr � twisted.credr �twisted.cred.errorr �twisted.internetr r
r �twisted.internet.errorr �twisted.pythonr
r �twisted.python.reflectr �
twisted.trialr r r �
twisted.conchr r �twisted.conch.sshr r r r r r r& rX r� �registerAdapter�ISession�Protocolr7 rn rg rs rr r� r� r� r� r� ru r� r� �SSHProtocolCheckerr� r� r� r� r� r� r� r� r� r� r� r� r r r �TestCaser r} r� r� r r r r �<module> sb B @�
1 !
=T?