| 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 : /lib/python3/dist-packages/twisted/conch/test/__pycache__/ |
Upload File : |
o
�b�3 � @ sx d Z ddlZddlmZ ddlmZ zddlZW n ey# dZY nw eZzddl Z
W n ey7 dZ Y nw e
Z z
ddl m Z
mZ W n
eyQ d ZZ Y nw ee
ZZ ddlmZmZ ddlmZ G dd � d �ZG d
d � d ej�ZG d d
� d
e�ZG dd� de�Ze dur�G dd� de j�ZG dd� de�Z G dd� de�Z G dd� de�Z G dd� de�Z G dd� de�Z dS ) z'
Tests for L{twisted.conch.ssh.agent}.
� N)�iosim)�unittest)�agent�keys)�
ConchError�MissingKeyStoreError)�keydatac @ � e Zd ZdZdd� ZdS )� StubFactoryzb
Mock factory that provides the keys attribute required by the
SSHAgentServerProtocol
c C s
i | _ d S �N)r ��self� r �?/usr/lib/python3/dist-packages/twisted/conch/test/test_agent.py�__init__, s
zStubFactory.__init__N)�__name__�
__module__� __qualname__�__doc__r r r r r r
& s r
c @ s, e Zd ZdZedu sedu rdZdd� ZdS )�
AgentTestBasez*
Tests for SSHAgentServer/Client.
Nz)Cannot run without cryptography or PyASN1c C sj t �tjtj�\| _| _| _t� | j_ t
j � t
j�| _t
j � t
j�| _t
j � t
j�| _t
j � t
j�| _d S r )r �connectedServerAndClientr �SSHAgentServer�SSHAgentClient�client�server�pumpr
�factoryr �Key�
fromStringr �privateRSA_openssh�
rsaPrivate�privateDSA_openssh�
dsaPrivate�publicRSA_openssh� rsaPublic�publicDSA_openssh� dsaPublicr r r r �setUp8 s �
zAgentTestBase.setUp)r r r r r r �skipr' r r r r r 0 s
r c @ r )�&ServerProtocolContractWithFactoryTestsz�
The server protocol is stateful and so uses its factory to track state
across requests. This test asserts that the protocol raises if its factory
doesn't provide the necessary storage for that state.
c C s2 t �ddtj�}| jjjd= | �t| jj |� d S )Nz!LB� r )
�struct�packr �AGENTC_REQUEST_IDENTITIESr r �__dict__� assertRaisesr � dataReceived)r
�msgr r r �/test_factorySuppliesKeyStorageForServerProtocolQ s zVServerProtocolContractWithFactoryTests.test_factorySuppliesKeyStorageForServerProtocolN)r r r r r2 r r r r r) J s r) c @ �( e Zd ZdZdd� Zdd� Zdd� ZdS ) �"UnimplementedVersionOneServerTestsa!
Tests for methods with no-op implementations on the server. We need these
for clients, such as openssh, that try v1 methods before going to v2.
Because the client doesn't expose these operations with nice method names,
we invoke sendRequest directly with an op code.
c s0 � j �tjd�}� j�� � fdd�}|�|�S )zV
assert that we get the correct op code for an RSA identities request
� c s � � tjt| dd� �� d S )Nr r* )� assertEqualr �AGENT_RSA_IDENTITIES_ANSWER�ord)�packetr r r �_cbh s zRUnimplementedVersionOneServerTests.test_agentc_REQUEST_RSA_IDENTITIES.<locals>._cb)r � sendRequestr � AGENTC_REQUEST_RSA_IDENTITIESr �flush� addCallback)r
�dr: r r r �"test_agentc_REQUEST_RSA_IDENTITIESa s
zEUnimplementedVersionOneServerTests.test_agentc_REQUEST_RSA_IDENTITIESc C �( | j �tjd�}| j�� |�| jd�S )z[
assert that we get the correct op code for an RSA remove identity request
r5 )r r; r �AGENTC_REMOVE_RSA_IDENTITYr r= r> r6 �r
r? r r r � test_agentc_REMOVE_RSA_IDENTITYm s
zBUnimplementedVersionOneServerTests.test_agentc_REMOVE_RSA_IDENTITYc C rA )zj
assert that we get the correct op code for an RSA remove all identities
request.
r5 )r r; r � AGENTC_REMOVE_ALL_RSA_IDENTITIESr r= r> r6 rC r r r �%test_agentc_REMOVE_ALL_RSA_IDENTITIESu s
zHUnimplementedVersionOneServerTests.test_agentc_REMOVE_ALL_RSA_IDENTITIESN)r r r r r@ rD rF r r r r r4 X s
r4 c @ s e Zd ZdZdd� Zdd� ZdS )�
CorruptServerz�
A misbehaving server that returns bogus response op codes so that we can
verify that our callbacks that deal with these op codes handle such
miscreants.
c C � | � dd� d S �N� r5 �� sendResponse�r
�datar r r �agentc_REQUEST_IDENTITIES� � z'CorruptServer.agentc_REQUEST_IDENTITIESc C rH rI rK rM r r r �agentc_SIGN_REQUEST� rP z!CorruptServer.agentc_SIGN_REQUESTN)r r r r rO rQ r r r r rG � s rG c @ r3 ) �ClientWithBrokenServerTestszM
verify error handling code in the client using a misbehaving server
c C s2 t �| � t�ttj�\| _| _| _ t
� | j_ d S r ) r r' r r rG r r r r r r
r r r r r r' � s
�z!ClientWithBrokenServerTests.setUpc C s* | j �| j�� d�}| j�� | �|t�S )z�
Assert that L{SSHAgentClient.signData} raises a ConchError
if we get a response from the server whose opcode doesn't match
the protocol for data signing requests.
� John Hancock)r �signDatar$ �blobr r= �
assertFailurer rC r r r �"test_signDataCallbackErrorHandling� s
z>ClientWithBrokenServerTests.test_signDataCallbackErrorHandlingc C s | j �� }| j�� | �|t�S )z�
Assert that L{SSHAgentClient.requestIdentities} raises a ConchError
if we get a response from the server whose opcode doesn't match
the protocol for identity requests.
)r �requestIdentitiesr r= rV r rC r r r �+test_requestIdentitiesCallbackErrorHandling� s
zGClientWithBrokenServerTests.test_requestIdentitiesCallbackErrorHandlingN)r r r r r' rW rY r r r r rR � s
rR c @ �0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) �AgentKeyAdditionTestsz<
Test adding different flavors of keys to an agent.
c �2 � j �� j�� �}� j�� � fdd�}|�|�S )�@
L{SSHAgentClient.addIdentity} adds the private key it is called
with to the SSH agent server to which it is connected, associating
it with the comment it is called with.
This test asserts that omitting the comment produces an
empty string for the comment on the server.
c �: � j jj� j�� }� �� j|d � � �d|d � d S �Nr r5 r* �r r r r rU r6 ��ignored� serverKeyr r r �_check� � zBAgentKeyAdditionTests.test_addRSAIdentityNoComment.<locals>._check�r � addIdentityr � privateBlobr r= r> �r
r? rd r r r � test_addRSAIdentityNoComment� �
z2AgentKeyAdditionTests.test_addRSAIdentityNoCommentc r\ )r] c r^ r_ �r r r r" rU r6 ra r r r rd � re zBAgentKeyAdditionTests.test_addDSAIdentityNoComment.<locals>._check�r rg r" rh r r= r> ri r r r � test_addDSAIdentityNoComment� rk z2AgentKeyAdditionTests.test_addDSAIdentityNoCommentc �6 � j j� j�� dd�}� j�� � fdd�}|�|�S )�1
L{SSHAgentClient.addIdentity} adds the private key it is called
with to the SSH agent server to which it is connected, associating
it with the comment it is called with.
This test asserts that the server receives/stores the comment
as sent by the client.
� My special key��commentc r^ �Nr rq r* r` ra r r r rd � re zDAgentKeyAdditionTests.test_addRSAIdentityWithComment.<locals>._checkrf ri r r r � test_addRSAIdentityWithComment� �
�
z4AgentKeyAdditionTests.test_addRSAIdentityWithCommentc ro )rp rq rr c r^ rt rl ra r r r rd re zDAgentKeyAdditionTests.test_addDSAIdentityWithComment.<locals>._checkrm ri r r r � test_addDSAIdentityWithComment� rv z4AgentKeyAdditionTests.test_addDSAIdentityWithCommentN)r r r r rj rn ru rw r r r r r[ � s r[ c @ s e Zd Zdd� ZdS )�AgentClientFailureTestsc C s$ | j �dd�}| j�� | �|t�S )zK
verify that the client raises ConchError on AGENT_FAILURE
rJ r5 )r r; r r= rV r rC r r r �test_agentFailure s
z)AgentClientFailureTests.test_agentFailureN)r r r ry r r r r rx s rx c @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd S )
�AgentIdentityRequestsTestszJ
Test operations against a server with identities already loaded.
c C �B t �| � | jdf| jjj| j�� < | jdf| jjj| j�� < d S �N� a comment� another comment�r r' r" r r r rU r r r r r r' �
��z AgentIdentityRequestsTests.setUpc C sX | j �| j�� d�}| j�� | �|�}| j�d�}| � ||� | �
| j� |d�� dS )zc
Sign data with an RSA private key and then verify it with the public
key.
rS N) r rT r$ rU r r= �successResultOfr �signr6 �
assertTrue�verify)r
r? � signature�expectedr r r �test_signDataRSA" s
z+AgentIdentityRequestsTests.test_signDataRSAc s4 � j �� j�� d�}� j�� � fdd�}|�|�S )zb
Sign data with a DSA private key and then verify it with the public
key.
rS c s � � � j�| d�� d S )NrS )r� r& r� )�sigr r r rd 7 s z;AgentIdentityRequestsTests.test_signDataDSA.<locals>._check)r rT r& rU r r= r> ri r r r �test_signDataDSA/ s
z+AgentIdentityRequestsTests.test_signDataDSAc C s< | j jj| j�� = | j�| j�� d�}| j�� | � |t
�S )zm
Assert that we get an errback if we try to sign data using a key that
wasn't added.
rS ) r r r r$ rU r rT r r= rV r rC r r r �$test_signDataRSAErrbackOnUnknownBlob? s
z?AgentIdentityRequestsTests.test_signDataRSAErrbackOnUnknownBlobc �* � j �� }� j�� � fdd�}|�|�S )z}
Assert that we get all of the keys/comments that we add when we issue a
request for all identities.
c s^ i }d|� j �� < d|� j�� < i }| D ]}|d |tjj|d dd��� < q� �||� d S )Nr} r~ r* r rU )�type)r& rU r$ r r r r6 )�keytr� �received�kr r r rd Q s "zAAgentIdentityRequestsTests.test_requestIdentities.<locals>._check)r rX r r= r> ri r r r �test_requestIdentitiesI s
z1AgentIdentityRequestsTests.test_requestIdentitiesN) r r r r r' r� r� r� r� r r r r rz s
rz c @ rZ ) �AgentKeyRemovalTestsz<
Test support for removing keys in a remote server.
c C r{ r| r r r r r r' c r� zAgentKeyRemovalTests.setUpc r\ )z<
Assert that we can remove an RSA identity.
c sJ � � dt� jjj�� � �� j�� � jjj� � �� j �� � jjj� d S �Nr* )
r6 �lenr r r �assertInr" rU � assertNotInr �rb r r r rd v s z;AgentKeyRemovalTests.test_removeRSAIdentity.<locals>._check)r �removeIdentityr rU r r= r> ri r r r �test_removeRSAIdentityn s
z+AgentKeyRemovalTests.test_removeRSAIdentityc r\ )z;
Assert that we can remove a DSA identity.
c s2 � � dt� jjj�� � �� j�� � jjj� d S r� )r6 r� r r r r� r rU r� r r r rd � s z;AgentKeyRemovalTests.test_removeDSAIdentity.<locals>._check)r r� r" rU r r= r> ri r r r �test_removeDSAIdentity} s
z+AgentKeyRemovalTests.test_removeDSAIdentityc r� )z;
Assert that we can remove all identities.
c s � � dt� jjj�� d S )Nr )r6 r� r r r r� r r r rd � s z=AgentKeyRemovalTests.test_removeAllIdentities.<locals>._check)r �removeAllIdentitiesr r= r> ri r r r �test_removeAllIdentities� s
z-AgentKeyRemovalTests.test_removeAllIdentitiesN)r r r r r' r� r� r� r r r r r� ^ s r� )!r r+ � twisted.testr �
twisted.trialr � cryptography�
_cryptography� ImportError�pyasn1�_pyasn1�twisted.conch.sshr �_agentr �_keys�twisted.conch.errorr r �twisted.conch.testr r
�TestCaser r) r4 r rG rR r[ rx rz r� r r r r �<module> sF � � �
'#V
L