← Back
Editing: 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
Save File
Cancel