File: //usr/lib/python3/dist-packages/twisted/conch/test/__pycache__/test_session.cpython-310.pyc
o
�b�� � @ s� d Z ddlZddlZddlZddlZddlmZ ddlmZ ddl m
Z
m Z m Z ddl
mZ ddlmZmZ ddlmZmZ dd lmZ dd
lmZ dd lmZ dd lm Z ed
�Z e rgddl m Z m Z m!Z! nG dd� d�Z!G dd� d�Z"G dd� d�Z#ee!j$�G dd� d��Z%ee!j&�G dd� de%��Z'G dd� d�Z(G dd � d e j)�Z*G d d � d �Z+G d d � d �Z,G d!d"� d"e,�Z-G d#d$� d$�Z.G d%d&� d&ee �Z/G d'd(� d(ee �Z0G d)d*� d*e �Z1G d+d,� d,e �Z2G d-d.� d.e �Z3G d/d0� d0e �Z4dS )1zb
Tests for the 'session' channel implementation in twisted.conch.ssh.session.
See also RFC 4254.
� N)�skipIf)� implementer)�defer�error�protocol�� IPv4Address)� ProcessDone�ProcessTerminated)�
components�failure)�Failure)�
requireModule)�RegistryUsingMixin)�TestCase� cryptography)�common�
connection�sessionc @ s e Zd ZddlmZmZmZ dS )r r )� EnvironmentVariableNotPermitted�ISession�ISessionSetEnvN)�__name__�
__module__� __qualname__�twisted.conch.interfacesr r r � r r �A/usr/lib/python3/dist-packages/twisted/conch/test/test_session.pyr " s r c @ � e Zd ZdZdd� ZdS )�SubsystemOnlyAvatarz^
A stub class representing an avatar that is only useful for
getting a subsystem.
c C s |dksJ �t � S )z�
If the other side requests the 'subsystem' subsystem, allow it by
returning a MockProtocol to implement it. Otherwise raise an assertion.
� subsystem)� MockProtocol��self�name�datar r r �lookupSubsystem0 s z#SubsystemOnlyAvatar.lookupSubsystemN�r r r �__doc__r& r r r r r * � r c @ r )�
StubAvatarz�
A stub class representing the avatar representing the authenticated user.
It implements the I{ISession} interface.
c C s" |dkrt � | _|| j_| jS dS )z�
If the user requests the TestSubsystem subsystem, connect them to a
MockProtocol. If they request neither, then None is returned which is
interpreted by SSHSession as a failure.
�
TestSubsystemN)r! � subsystem�
packetDatar" r r r r& ? s
�zStubAvatar.lookupSubsystemNr' r r r r r* 9 r) r* c @ sH e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dS )�StubSessionForStubAvatara2
A stub ISession implementation for our StubAvatar. The instance
variables generally keep track of method invocations so that we can test
that the methods were called.
@ivar avatar: the L{StubAvatar} we are adapting.
@ivar ptyRequest: if present, the terminal, window size, and modes passed
to the getPty method.
@ivar windowChange: if present, the window size passed to the
windowChangned method.
@ivar shellProtocol: if present, the L{SSHSessionProcessProtocol} passed
to the openShell method.
@ivar shellTransport: if present, the L{EchoTransport} connected to
shellProtocol.
@ivar execProtocol: if present, the L{SSHSessionProcessProtocol} passed
to the execCommand method.
@ivar execTransport: if present, the L{EchoTransport} connected to
execProtocol.
@ivar execCommandLine: if present, the command line passed to the
execCommand method.
@ivar gotEOF: if present, an EOF message was received.
@ivar gotClosed: if present, a closed message was received.
c C s || _ d| _dS )z2
Store the avatar we're adapting.
N)�avatar�
shellProtocol�r# r/ r r r �__init__e �
z!StubSessionForStubAvatar.__init__c C s |dkr |||f| _ dS td��)zw
If the terminal is 'bad', fail. Otherwise, store the information in
the ptyRequest variable.
� badznot getting a ptyN)�
ptyRequest� RuntimeError)r# �terminal�window�modesr r r �getPtyl s z StubSessionForStubAvatar.getPtyc C s |dkrt d��|| _dS )zw
If all the window sizes are 0, fail. Otherwise, store the size in the
windowChange variable.
�r r r r z not changing the window sizeN)r6 � windowChange)r# r8 r r r �
windowChangedv s
z&StubSessionForStubAvatar.windowChangedc C s& | j dur td��|| _ t|�| _dS )z�
If we have gotten a shell request before, fail. Otherwise, store the
process protocol in the shellProtocol variable, connect it to the
EchoTransport and store that as shellTransport.
Nz not getting a shell this time)r0 r6 �
EchoTransport�shellTransport)r# �ppr r r � openShell� s
z"StubSessionForStubAvatar.openShellc C sV || _ |dkr || _dS |dd� dkr'|| _t|�| _|�|dd� � dS td��)z�
If the command is 'true', store the command, the process protocol, and
the transport we connect to the process protocol. Otherwise, just
store the command and raise an error.
� successN� s repeat� znot getting a command)�execCommandLine� execProtocolr> �
execTransport� outReceivedr6 )r# r@ �commandr r r � execCommand� s
z$StubSessionForStubAvatar.execCommandc C �
d| _ dS )z2
Note that EOF has been received.
TN)�gotEOF�r# r r r � eofReceived� �
z$StubSessionForStubAvatar.eofReceivedc C rK )z4
Note that close has been received.
TN)� gotClosedrM r r r �closed� rO z StubSessionForStubAvatar.closedN) r r r r( r2 r: r= rA rJ rN rQ r r r r r. K s
r. c s0 e Zd ZdZ� fdd�Zdd� Zdd� Z� ZS )� StubSessionForStubAvatarWithEnva�
Same as StubSessionForStubAvatar, but supporting environment variables
setting.
End users would want to have the same class annotated with
C{@implementer(session.ISession, session.ISessionSetEnv)}. The interfaces
are split for backwards compatibility, so we split it here to test
this compatibility too.
@ivar environ: a L{dict} of environment variables passed to the setEnv
method.
c s t � �|� i | _i | _d S �N)�superr2 �environ� environAtPtyr1 �� __class__r r r2 � s
z(StubSessionForStubAvatarWithEnv.__init__c C s0 |dkrt d��|dkrt�d��|| j|< dS )ak
If the requested environment variable is 'FAIL', fail. If it is
'IGNORED', raise EnvironmentVariableNotPermitted, which should cause
it to be silently ignored. Otherwise, store the requested
environment variable.
(Real applications should normally implement an allowed list rather
than a blocked list.)
� FAILz$disallowed environment variable name� IGNOREDz!ignored environment variable nameN)r6 r r rU )r# r$ �valuer r r �setEnv� s
�z&StubSessionForStubAvatarWithEnv.setEnvc C s | j �� | _dS )zs
Just a simple implementation which records the current environment
when PTY is requested.
N)rU �copyrV )r# �term�
windowSizer9 r r r r: � s z&StubSessionForStubAvatarWithEnv.getPty)r r r r( r2 r\ r: �
__classcell__r r rW r rR � s
rR c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) r> ay
A transport for a ProcessProtocol which echos data that is sent to it with
a Window newline (CR LF) appended to it. If a null byte is in the data,
disconnect. When we are asked to disconnect, disconnect the
C{ProcessProtocol} with a 0 exit code.
@ivar proto: the C{ProcessProtocol} connected to us.
@ivar data: a L{bytes} of data written to us.
c C s || _ d| _d| _|�| � dS )z�
Initialize our instance variables.
@param processProtocol: a C{ProcessProtocol} to connect to ourself.
F� N)�protorQ r% �makeConnection)r# �processProtocolr r r r2 � s zEchoTransport.__init__c C s> | j |7 _ | j�|� | j�d� d|v r | �� dS dS )z�
We got some data. Give it back to our C{ProcessProtocol} with
a newline attached. Disconnect if there's a null byte.
�
� N)r% rb rH �loseConnection�r# r% r r r �write� s �zEchoTransport.writec C sN | j rdS d| _ | j�� | j�� | j�� | j�t�t� ddd��� dS )z�
If we're asked to disconnect (and we haven't already) shut down
the C{ProcessProtocol} with a 0 exit code.
N� r )
rQ rb �inConnectionLost�outConnectionLost�errConnectionLost� processEndedr r
r r
rM r r r rg � s
z EchoTransport.loseConnectionN)r r r r( r2 ri rg r r r r r> � s
r> c @ s, e Zd ZdZdZdd� Zdd� Zdd� Zd S )
r! a>
A sample Protocol which stores the data passed to it.
@ivar packetData: a L{bytes} of data to be sent when the connection is
made.
@ivar data: a L{bytes} of the data passed to us.
@ivar open: True if the channel is open.
@ivar reason: if not None, the reason the protocol was closed.
ra c C s, d| _ d| _d| _| jr| �| j� dS dS )zc
Set up the instance variables. If we have any packetData, send it
along.
ra TN)r% �open�reasonr- � dataReceivedrM r r r �connectionMade s �zMockProtocol.connectionMadec C s" | j |7 _ | j�|d � dS )z�
Store the received data and write it back with a tilde appended.
The tilde is appended so that the tests can verify that we processed
the data.
� ~N)r% � transportri rh r r r rq " s zMockProtocol.dataReceivedc C s d| _ || _dS )z:
Close the protocol and store the reason.
FN)ro rp )r# rp r r r �connectionLost+ r3 zMockProtocol.connectionLostN)r r r r( r- rr rq ru r r r r r! s
r! c @ sL e Zd ZdZddd�Zdd� Zdd� Zd d
� Zdd d
�Zdd� Z dd� Z
dS )�StubConnectiona�
A stub for twisted.conch.ssh.connection.SSHConnection. Record the data
that channels send, and when they try to close the connection.
@ivar data: a L{dict} mapping C{SSHChannel}s to a C{list} of L{bytes} of
data they sent.
@ivar extData: a L{dict} mapping L{SSHChannel}s to a C{list} of L{tuple} of
(L{int}, L{bytes}) of extended data they sent.
@ivar requests: a L{dict} mapping L{SSHChannel}s to a C{list} of L{tuple}
of (L{str}, L{bytes}) of channel requests they made.
@ivar eofs: a L{dict} mapping L{SSHChannel}s to C{true} if they have sent
an EOF.
@ivar closes: a L{dict} mapping L{SSHChannel}s to C{true} if they have sent
a close.
Nc C s( i | _ i | _i | _i | _i | _|| _dS )z4
Initialize our instance variables.
N)r% �extData�requests�eofs�closesrt )r# rt r r r r2 D s
zStubConnection.__init__c C � dS )z,
Return our logging prefix.
�MockConnectionr rM r r r � logPrefixO s zStubConnection.logPrefixc C s | j �|g ��|� dS )z'
Record the sent data.
N)r% �
setdefault�append)r# �channelr% r r r �sendDataU s zStubConnection.sendDatac C s | j �|g ��||f� dS )z0
Record the sent extended data.
N)rw r~ r )r# r� �typer% r r r �sendExtendedData[ � z StubConnection.sendExtendedDataFc C s, | j �|g ��|||f� |rt�d�S dS )z2
Record the sent channel request.
N)rx r~ r r �succeed)r# r� �requestr% � wantReplyr r r � sendRequesta s
�zStubConnection.sendRequestc C � d| j |< dS )z&
Record the sent EOF.
TN)ry �r# r� r r r �sendEOFi � zStubConnection.sendEOFc C r� )z(
Record the sent close.
TN)rz r� r r r � sendCloseo r� zStubConnection.sendCloserS )F) r r r r( r2 r} r� r� r� r� r� r r r r rv 3 s
rv c @ s@ e Zd ZdZdZdZdd� Zdd� Zdd � Zd
d � Z d d
� Z
dS )�
StubTransportz�
A stub transport which records the data written.
@ivar buf: the data sent to the transport.
@type buf: L{bytes}
@ivar close: flags indicating if the transport has been closed.
@type close: L{bool}
ra Fc C � t ddd�S )�2
Return an arbitrary L{IAddress}.
�TCP�
remotehosti�" r rM r r r �getPeer� � zStubTransport.getPeerc C r� )r� r� � localhosti' r rM r r r �getHost� r� zStubTransport.getHostc C � | j |7 _ dS )z,
Record data in the buffer.
N)�bufrh r r r ri � s zStubTransport.writec C rK )z6
Note that the connection was closed.
TN)�closerM r r r rg � rO z StubTransport.loseConnectionc C r{ )z0
Pretend to set C{TCP_NODELAY}.
Nr )r# �enabledr r r �
setTcpNoDelay� s zStubTransport.setTcpNoDelayN) r r r r( r� r� r� r� ri rg r� r r r r r� v s
r� c @ s e Zd ZdZdZdd� ZdS )�StubTransportWithWriteErrz�
A version of StubTransport which records the error data sent to it.
@ivar err: the extended data sent to the transport.
@type err: L{bytes}
ra c C r� )z�
Record the extended data in the buffer. This was an old interface
that allowed the Transports from ISession.openShell() or
ISession.execCommand() to receive extended data from the client.
N)�errrh r r r �writeErr� s z"StubTransportWithWriteErr.writeErrN)r r r r( r� r� r r r r r� � s r� c @ r )�
StubClientz�
A stub class representing the client to a SSHSession.
@ivar transport: A L{StubTransport} object which keeps track of the data
passed to it.
c C s t � | _d S rS )r� rt rM r r r r2 � s zStubClient.__init__N)r r r r( r2 r r r r r� � s r� c @ s� e Zd ZdZes
dZd;dd�Zd;dd�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-� Zd.d/� Zd0d1� Z d2d3� Z d4d5� Z d6d7� Z d8d9� Z d:S )<�SessionInterfaceTestsz�
Tests for the SSHSession class interface. This interface is not ideal, but
it is tested in order to maintain backwards compatibility.
� cannot run without cryptographyTc C s6 t �| � | �� | _|rt�tttj� | �� | _dS )z�
Make an SSHSession object to test. Give the channel some window
so that it's allowed to send packets. 500 and 100 are arbitrary
values.
N) r �setUp�
getSSHSessionr r �registerAdapterrR r* r �r# �register_adaptersr r r r� � s
�zSessionInterfaceTests.setUpc C s t jddt� t� d�S )z+
Return a new SSH session.
�� �d )� remoteWindow�remoteMaxPacket�connr/ )r �
SSHSessionrv r* r� r r r r� � s �z#SessionInterfaceTests.getSSHSessionc C s | � | jjt� dS )zj
Asserts that self.session.session is an instance of
StubSessionForStubOldAvatar.
N)�assertIsInstancer r. rM r r r �assertSessionIsStubSession� s z0SessionInterfaceTests.assertSessionIsStubSessionc C s6 t jtd�}| �|jd� | �|j� | �|j � dS )z�
SSHSession initializes its buffer (buf), client, and ISession adapter.
The avatar should not need to be adaptable to an ISession immediately.
)r/ ra N)r r� �object� assertEqualr� � assertIsNone�client)r# �sr r r � test_init� s z SessionInterfaceTests.test_initc C s: | j �d� t� | j _| j �d� | �| j jjjd� dS )z�
SSHSession.dataReceived() passes data along to a client. If the data
comes before there is a client, the data should be discarded.
� 1� 2N)r rq r� r� r� rt r� rM r r r �test_client_dataReceived� s
z.SessionInterfaceTests.test_client_dataReceivedc C sP | j �tjd� | j �dd� t� | j _| j �tjd� | �| j jjjd� dS )z�
SSHSession.extReceived() passed data of type EXTENDED_DATA_STDERR along
to the client. If the data comes before there is a client, or if the
data is not of type EXTENDED_DATA_STDERR, it is discared.
r� � r� � 3N) r � extReceivedr �EXTENDED_DATA_STDERRr� r� r� rt r� rM r r r �test_client_extReceived s
z-SessionInterfaceTests.test_client_extReceivedc C s* t � }| j_t� |_| j�tjd� dS )z�
SSHSession.extReceived() should handle the case where the transport
on the client doesn't have a writeErr method.
� ignoredN)r� r r� r� rt r� r r� �r# r� r r r �&test_client_extReceivedWithoutWriteErr s z<SessionInterfaceTests.test_client_extReceivedWithoutWriteErrc C s6 t � | j_| j�� | �| jjjj� d| jjj_dS )z}
SSHSession.closed() should tell the transport connected to the client
that the connection was lost.
FN)r� r r� rQ �
assertTruert r� rM r r r �test_client_closed s
z(SessionInterfaceTests.test_client_closedc C s0 | j �dt�d��}| �|� | �| j j� dS )zV
When a subsystem request fails, SSHSession.client should not be set.
r s BadSubsystemN)r �requestReceivedr �NS� assertFalser� r� �r# �retr r r �$test_badSubsystemDoesNotCreateClient! s
z:SessionInterfaceTests.test_badSubsystemDoesNotCreateClientc C sR | j �dt�d�d �}| �|� | �| j jtj� | � | j jj
j | j j j
� dS )z�
When a client requests a subsystem, the SSHSession object should get
the subsystem by calling avatar.lookupSubsystem, and attach it as
the client.
r r+ � dataN)r r� r r� r� r� r� r �ProcessProtocol�assertIsrt rb r/ r, r� r r r �test_lookupSubsystem) s �
�z*SessionInterfaceTests.test_lookupSubsystemc C s� t jt� t� d�}|�t�d�d �}| �|� | �|j � | �
|j j �
|�� |�� | �|j j �
|�� |�� |�� dS )z�
Previously, if one only wanted to implement a subsystem, an ISession
adapter wasn't needed because subsystems were looked up using the
lookupSubsystem method on the avatar.
)r/ r� r r� N)r r� r rv �request_subsystemr r� r� �assertIsNotNoner� r� r� rz �getrN rg rQ )r# r� r� r r r �'test_lookupSubsystemDoesNotNeedISession8 s
z=SessionInterfaceTests.test_lookupSubsystemDoesNotNeedISessionc C s^ | j �dt�d�d � | �| j jj| j dg� | j �d� | �| j jj| j d d� dS ) aD
After having looked up a subsystem, data should be passed along to the
client. Additionally, subsystems were passed the entire request packet
as data, instead of just the additional data.
We check for the additional tidle to verify that the data passed
through the client.
r r+ r� s
TestSubsystemdata~s more data���s
more data~N)r r� r r� r� r� r% rq rM r r r �test_lookupSubsystem_dataI s
�� z/SessionInterfaceTests.test_lookupSubsystem_datac C s< | j �dt�d�d � | j �� | �| j jj| j � dS )zd
SSHSession.closeReceived() should sent a close message to the remote
side.
r r+ r� N)r r� r r� �
closeReceivedr� r� rz rM r r r �"test_lookupSubsystem_closeReceiveda s
�
z8SessionInterfaceTests.test_lookupSubsystem_closeReceivedc C sB | � t�}| �t|�ddd�dd� |D �� � |d �t� dS )zo
Assert that the request we just made raised a RuntimeError (and only a
RuntimeError).
rj z!Multiple RuntimeErrors raised: %s�
c S s g | ]}t |��qS r )�repr)�.0r r r r �
<listcomp>w s zISessionInterfaceTests.assertRequestRaisedRuntimeError.<locals>.<listcomp>r N)�flushLoggedErrorsr6 r� �len�join�trap)r# �errorsr r r � assertRequestRaisedRuntimeErrorm s
��z5SessionInterfaceTests.assertRequestRaisedRuntimeErrorc C sh | j �dd�}| �|� | �� | �| j jt j� | �| j j j| j j� | � | j �dd�� | �
� dS )z�
When a client requests a shell, the SSHSession object should get
the shell by getting an ISession adapter for the avatar, then
calling openShell() with a ProcessProtocol to attach.
� shellra N) r r� r� r� r� r� �SSHSessionProcessProtocolr� r0 r� r� r� r r r �test_requestShell{ s
z'SessionInterfaceTests.test_requestShellc C s� | j �dd�}| �|� | �� | j �d� | �| j j jjd� | �| j jj| j ddg� | �| j j jj � | �| j jj
| j dg� dS )z�
When a client executes a shell, it should be able to give pass data
back and forth between the local and the remote side.
r� ra s
some data re �� exit-statuss FN) r r� r� r� rq r� r? r% r� rQ rx r� r r r �test_requestShellWithData� s
��z/SessionInterfaceTests.test_requestShellWithDatac C s� | j �dt�d��}| �|� | �� | �| j j� | �| j �dt�d��� | � � | �
| j jt j � | � | j j j
| j j� | �| j j jd� dS )z�
When a client requests a command, the SSHSession object should get
the command by getting an ISession adapter for the avatar, then
calling execCommand with a ProcessProtocol to attach and the
command line.
� execs failurerB N)r r� r r� r� r� r� r� r� r� r� r� r� rF r� rE r� r r r �test_requestExec� s
z&SessionInterfaceTests.test_requestExecc C s� | j �dt�d��}| �|� | �� | j �d� | �| j j jj d� | �| j j
j | j g d�� | j � � | j � � | j �
� | �| j j jj
� | �| j j
j| j dg� dS )zo
When a client executes a command, it should be able to give pass data
back and forth.
r� s repeat hello� some data)s hellor� re r� N)r r� r r� r� r� rq r� rG r% r� rN r� rQ rx r� r r r �test_requestExecWithData� s
�
�z.SessionInterfaceTests.test_requestExecWithDatac
C s� | � � | jdd� t�tttj� | �� }|� dt�
ddd��}| � |� | � |jt� | �
� | �|� dt�
ddd��� | �|jjddg f� dS ) a
When a client requests a PTY, the SSHSession object should make
the request by getting an ISession adapter for the avatar, then
calling getPty with the terminal type, the window size, and any modes
the client gave us.
F�r� � pty_reqr4 �rj � � � ra s goodN)�
doCleanupsr� r r� r. r* r r r� r� �packRequest_pty_reqr� r� r� r� r� r5 )r# � test_sessionr� r r r �test_requestPty� s$ ��
��z%SessionInterfaceTests.test_requestPtyc C s� | � | j�dt�d�t�d� �� | �| jjt� | �� | � | j�dt�d�t�d� �� | �| � � g � | �
| j�dt�d�t�d� �� | �| jjj ddi� dS ) a
When a client requests passing an environment variable, the
SSHSession object should make the request by getting an
ISessionSetEnv adapter for the avatar, then calling setEnv with the
environment variable name and value.
� envrY r4 rZ r� � NAME� valueN) r� r r� r r� r� rR r� r� r� r� rU rM r r r � test_setEnv� s" �����z!SessionInterfaceTests.test_setEnvc C st | � � }| �|�dt�d�t�d� �� | �|�dt�d�t�d� �� | �|jt� | �ddd�|jj � dS )zG
Multiple setenv requests will share the same session.
r� � Key1� Value 1� Key2� Value2�r� r� N)
r� r� r� r r� r� r rR r� rU �r# r� r r r �test_setEnvSessionShare s �����z-SessionInterfaceTests.test_setEnvSessionSharec C s~ | � � }|�dt�d�t�d� � |�dt�d�t�d� � |�dt�ddd �� | �|jt� | �ddd
�|jj � d S ) z�
Calling another session service after setenv will provide the
previous session with the environment variables.
r� r� r� r� r� r� � termr; ra r� N)
r� r� r r� r r� r� rR r� rV r� r r r �test_setEnvMultiplexShare s ��z/SessionInterfaceTests.test_setEnvMultiplexSharec C sL | � � | jdd� t�tttj� | �| j� dt
� d�t
� d� �� dS )z�
If the avatar does not have an ISessionSetEnv adapter, then a
request to pass an environment variable fails gracefully.
Fr� r� r� r� N) r� r� r r� r. r* r r r� r� r r� rM r r r �%test_setEnvNotProvidingISessionSetEnv) s ���z;SessionInterfaceTests.test_setEnvNotProvidingISessionSetEnvc C s^ | j �dt �d��}| �|� | �� | �� | �| j �dt �d��� | �| j j jd� dS )z�
When the client requests to change the window size, the SSHSession
object should make the request by getting an ISession adapter for the
avatar, then calling windowChanged with the new window size.
�
window_changer; r� N) r r� �packRequest_window_changer� r� r� r� r� r< r� r r r �test_requestWindowChange; s
�
��z.SessionInterfaceTests.test_requestWindowChangec C �0 t �| j j�| j _ | j �� | �| j j j� dS )z
When an EOF is received and an ISession adapter is present, it should
be notified of the EOF message.
N)r r r/ rN r� rL rM r r r �test_eofReceivedN �
z&SessionInterfaceTests.test_eofReceivedc C s. | j �� }| �|� | �| j jj| j � dS )zT
When a close is received, the session should send a close message.
N)r r� r� r� r� rz r� r r r �test_closeReceivedW s
z(SessionInterfaceTests.test_closeReceivedc C r )z�
When a close is received and an ISession adapter is present, it should
be notified of the close message.
N)r r r/ rQ r� rP rM r r r � test_closed_ r z!SessionInterfaceTests.test_closedN)T)!r r r r( r �skipr� 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 r� � s>
r� c @ sH e Zd ZdZes
dZdd� Zdd� Zdd� Zd d
� Z d d � Z
d
d� Z dS )�SessionWithNoAvatarTestsa
Test for the SSHSession interface. Several of the methods (request_shell,
request_exec, request_pty_req, request_env, request_window_change) would
create a 'session' instance variable from the avatar if one didn't exist
when they were called.
r� c C s@ t �| � t�tttj� t�� | _t� | j_ | �
| jj� d S rS ) r r� r r� r. r* r r r� r/ r� rM r r r r� t s
�
z SessionWithNoAvatarTests.setUpc C s$ | � tj�| jj�d| jj � dS )zB
self.session.session should provide I{ISession}.
zISession not provided by %rN)r� r r �
providedByrM r r r � assertSessionProvidesISession} s
�z6SessionWithNoAvatarTests.assertSessionProvidesISessionc C s | j �dd� | �� dS )ze
If an ISession adapter isn't already present, request_shell should get
one.
r� ra N)r r� r rM r r r � test_requestShellGetsSession� s z5SessionWithNoAvatarTests.test_requestShellGetsSessionc C s | j �dt�d�� | �� dS )zd
If an ISession adapter isn't already present, request_exec should get
one.
r� rB N)r r� r r� r rM r r r �test_requestExecGetsSession� s z4SessionWithNoAvatarTests.test_requestExecGetsSessionc C s$ | j �dt �ddd�� | �� dS )zg
If an ISession adapter isn't already present, request_pty_req should
get one.
r� r� r; ra N)r r� r� r rM r r r � test_requestPtyReqGetsSession� s � z6SessionWithNoAvatarTests.test_requestPtyReqGetsSessionc C s | j �dt �d�� | �� dS )zm
If an ISession adapter isn't already present, request_window_change
should get one.
r )rj rj rj rj N)r r� r r rM r r r �#test_requestWindowChangeGetsSession� s
� z<SessionWithNoAvatarTests.test_requestWindowChangeGetsSessionN) r r r r( r r r� r r r
r r r r r r r i s
r c @ s( e Zd ZdZes
dZdd� Zdd� ZdS )�
WrappersTestszH
A test for the wrapProtocol and wrapProcessProtocol functions.
r� c C sz t � }t� |_|�� t�|�}|�d� | �|jjd� |� d� |�
ddg� |� � | �|j d� |j
�tj� dS )z�
L{wrapProtocol}, when passed a L{Protocol} should return something that
has write(), writeSequence(), loseConnection() methods which call the
Protocol's dataReceived() and connectionLost() methods, respectively.
s dataReceivedr� r� r� s data12N)r! r� rt rr r � wrapProtocolrq r� r� ri �
writeSequencerg r% rp r� r �ConnectionDone)r# r �wrappedr r r �test_wrapProtocol� s
z WrappersTests.test_wrapProtocolc C sf t � }t� |_t�|�}|�� |�d� | �|jjd� |� t
� t �
ddd��� |j�t j
� dS )a.
L{wrapPRocessProtocol}, when passed a L{Protocol} should return
something that follows the L{IProcessProtocol} interface, with
connectionMade() mapping to connectionMade(), outReceived() mapping to
dataReceived() and processEnded() mapping to connectionLost().
r� s data~r N)r! r� rt r �wrapProcessProtocolrr rH r� r� rn r r
r r
rp r� )r# r �process_protocolr r r �!test_wrapProcessProtocol_Protocol� s
�z/WrappersTests.test_wrapProcessProtocol_ProtocolN)r r r r( r r r r r r r r r � s r c @ s@ e Zd ZdZes
dZdd� Zdd� Zdd� Zd d
� Z d d � Z
d
S )� HelpersTestszM
Tests for the 4 helper functions: parseRequest_* and packRequest_*.
r� c C sJ | � t�t�d�t�ddddd� t�t�ddd �� �dd
d gf� d S )
a1
The payload of a pty-req message is::
string terminal
uint32 columns
uint32 rows
uint32 x pixels
uint32 y pixels
string modes
Modes are::
byte mode number
uint32 mode value
� xterm�>4Lrj r� r� r� �>BL� rC �r� rj r� r� )r rC N)r� r �parseRequest_pty_reqr r� �struct�packrM r r r �test_parseRequest_pty_req� s ���
�z&HelpersTests.test_parseRequest_pty_reqc C �J t �ddd�}| �|t�d�t�ddddd� t�t�d d
d �� � d S �
zG
See test_parseRequest_pty_req for the payload format.
r r s r rj r� r� r� r r rC N�r r� r� r r� r r! �r# �packedr r r � test_packRequest_pty_req_old� s ����z)HelpersTests.test_packRequest_pty_req_oldc C r# r$ r% r& r r r �test_packRequest_pty_req s ����z%HelpersTests.test_packRequest_pty_reqc C s$ | � t�t�ddddd��d� dS )a
The payload of a window_change request is::
uint32 columns
uint32 rows
uint32 x pixels
uint32 y pixels
parseRequest_window_change() returns (rows, columns, x pixels,
y pixels).
r rj r� r� r� r N)r� r �parseRequest_window_changer r! rM r r r � test_parseRequest_window_change s �z,HelpersTests.test_parseRequest_window_changec
C s$ | � t�d�t�ddddd�� dS )zM
See test_parseRequest_window_change for the payload format.
r r rj r� r� r� N)r� r r r r! rM r r r � test_packRequest_window_change$ s �z+HelpersTests.test_packRequest_window_changeN) r r r r( r r r" r( r) r+ r, r r r r r � s r c @ s e Zd ZdZes
dZdd� Zdd� Zdd� Zd d
� Z d d � Z
d
d� Z dd� Z e
eed� d�dd� �Ze
eed� d�dd� �Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Ze
eed)� d*�d+d,� �Z e
eed)� d*�d-d.� �Z d/S )0� SSHSessionProcessProtocolTestsz1
Tests for L{SSHSessionProcessProtocol}.
r� c C s@ t � | _tjt| j�ddd�| _t�| j�| _| j�| j� d S )Nr� r� )r� r� r� )r� rt r r� rv r� r@ rc rM r r r r� 6 s �z$SSHSessionProcessProtocolTests.setUpc C s | � | jjj| j � dS )z8
Assert that C{self.session} is closed.
N)r� r r� rz rM r r r �assertSessionClosed> s z2SSHSessionProcessProtocolTests.assertSessionClosedc C s | � | jjj| j |� dS )zO
Assert that C{self.session} has sent the C{expectedRequests}.
N)r� r r� rx )r# �expectedRequestsr r r �assertRequestsEqualD r� z2SSHSessionProcessProtocolTests.assertRequestsEqualc C s | � | jj| j� dS )zy
SSHSessionProcessProtocol should set self.session to the session passed
to the __init__ method.
N)r� r@ r rM r r r r� J s z(SSHSessionProcessProtocolTests.test_initc C � | � | jjj�� | j�� � dS )zu
SSHSessionProcessProtocol.getHost() just delegates to its
session.conn.transport.getHost().
N)r� r r� rt r� r@ rM r r r � test_getHostQ � z+SSHSessionProcessProtocolTests.test_getHostc C r1 )zu
SSHSessionProcessProtocol.getPeer() just delegates to its
session.conn.transport.getPeer().
N)r� r r� rt r� r@ rM r r r � test_getPeerX r3 z+SSHSessionProcessProtocolTests.test_getPeerc C s& d| j _| j�� | �| jjd� dS )z�
SSHSessionProcessProtocol.connectionMade() should check if there's a
'buf' attribute on its session and write it to the transport if so.
s bufferN)r r� r@ rr r� rt rM r r r �test_connectionMade_ s
z2SSHSessionProcessProtocolTests.test_connectionMade�SIGALRMzNot all signals availablec C sF t jD ] }d| }tt|�}| j�|�}| �||d|||f � qdS )zi
_getSignalName should return the name of a signal when given the
signal number.
�SIGz %i: %s != %sN)r �SUPPORTED_SIGNALS�getattr�signalr@ �_getSignalNamer� )r# �
signalName� signalValue�sshNamer r r �test_getSignalNameh s
��z1SSHSessionProcessProtocolTests.test_getSignalNamec C s4 t jd t _d| j_| �| j�t j�dtj � dS )z�
If there are signals in the signal module which aren't in the SSH RFC,
we map their name to [signal name]@[platform].
rj NzSIGTwistedTest@) r: �NSIG�SIGTwistedTestr@ �_signalValuesToNamesr� r; �sys�platformrM r r r �!test_getSignalNameWithLocalSignalv s �z@SSHSessionProcessProtocolTests.test_getSignalNameWithLocalSignalc C �* | j �d� | �| jjj| j dg� dS )zy
When data is passed to the outReceived method, it should be sent to
the session's write method.
� test dataN)r@ rH r� r r� r% rM r r r �test_outReceived� � z/SSHSessionProcessProtocolTests.test_outReceivedc C rF )z{
When data is passed to the write method, it should be sent to the
session channel's write method.
rG N)r@ ri r� r r� r% rM r r r �
test_write� rI z)SSHSessionProcessProtocolTests.test_writec C s. | j �ddg� | �| jjj| j dg� dS )z�
When a sequence is passed to the writeSequence method, it should be
joined together and sent to the session channel's write method.
s test r� rG N)r@ r r� r r� r% rM r r r �test_writeSequence� s z1SSHSessionProcessProtocolTests.test_writeSequencec C s* | j �d� | �| jjj| j dg� dS )z�
When data is passed to the errReceived method, it should be sent to
the session's writeExtended method.
rG )rj rG N)r@ � errReceivedr� r r� rw rM r r r �test_errReceived� rI z/SSHSessionProcessProtocolTests.test_errReceivedc C �D | j �� | �| j| jjjv � | j �� | �| jjj| j � dS )zv
When outConnectionLost and errConnectionLost are both called, we should
send an EOF message.
N)r@ rl r� r r� ry rm r� rM r r r �test_outConnectionLost� �
z5SSHSessionProcessProtocolTests.test_outConnectionLostc C rN )zh
Make sure reverse ordering of events in test_outConnectionLost also
sends EOF.
N)r@ rm r� r r� ry rl r� rM r r r �test_errConnectionLost� rP z5SSHSessionProcessProtocolTests.test_errConnectionLostc C s$ | j �� | �| jjj| j � dS )zp
When loseConnection() is called, it should call loseConnection
on the session channel.
N)r@ rg r� r r� rz rM r r r �test_loseConnection� s
z2SSHSessionProcessProtocolTests.test_loseConnectionc C s | j �t�td��� dS )zr
When connectionLost() is called, it should call loseConnection()
on the session channel.
r N)r@ ru r r
r rM r r r �test_connectionLost� s z2SSHSessionProcessProtocolTests.test_connectionLostc C s: | j �ttd��� | �dt�dd�dfg� | �� dS )z�
When processEnded is called, if there is an exit code in the reason
it should be sent in an exit-status method. The connection should be
closed.
Nr� z>Ir F)r@ rn r
r r0 r r! r. rM r r r � test_processEndedWithExitCode� s z<SSHSessionProcessProtocolTests.test_processEndedWithExitCode� WCOREDUMPz can't run this w/o os.WCOREDUMPc C �V | j �ttdtjd��� | �dt�d�d t�d� t�d� dfg� | � � dS ) z�
When processEnded is called, if there is an exit signal in the reason
it should be sent in an exit-signal message. The connection should be
closed.
rj � � exit-signal� TERM� ra FN�
r@ rn r
r
r: �SIGTERMr0 r r� r. rM r r r �'test_processEndedWithExitSignalCoreDump� s" ������� zFSSHSessionProcessProtocolTests.test_processEndedWithExitSignalCoreDumpc C rV ) z�
When processEnded is called, if there is an exit signal in the
reason it should be sent in an exit-signal message. If no
core was dumped, don't set the core-dump bit.
rj r rX rY rf ra FNr[ rM r r r �)test_processEndedWithExitSignalNoCoreDump� s ��� zHSSHSessionProcessProtocolTests.test_processEndedWithExitSignalNoCoreDumpN) r r r r( r r r� r. r0 r� r2 r4 r5 r �hasattrr: r? rE rH rJ rK rM rO rQ rR rS rT �osr] r^ r r r r r- . s8
r- c @ s e Zd ZdZes
dZdd� ZdS )�SSHSessionClientTestsza
SSHSessionClient is an obsolete class used to connect standard IO to
an SSHSession.
r� c C s. t �� }t� |_|�d� | �|jjd� dS )zL
When data is received, it should be sent to the transport.
rG N)r �SSHSessionClientr� rt rq r� r� r� r r r �test_dataReceived s
z'SSHSessionClientTests.test_dataReceivedN)r r r r( r r rc r r r r ra � s
ra )5r( r` r: r rC �unittestr �zope.interfacer �twisted.internetr r r �twisted.internet.addressr �twisted.internet.errorr r
�twisted.pythonr r �twisted.python.failurer
�twisted.python.reflectr �#twisted.python.test.test_componentsr �twisted.trial.unittestr r �twisted.conch.sshr r r r r* r r. r rR r> �Protocolr! rv r� r� r� r� r r r r- ra r r r r �<module> sR ]0/*C- *B-V R