File: //usr/lib/python3/dist-packages/twisted/conch/test/__pycache__/test_agent.cpython-38.pyc
U
W[ 3 � @ sp d Z ddlmZmZ ddlZddlmZ ddlmZ z ddl Z W n e
k
rX dZ Y nX z ddl Z W n e
k
r~ dZ Y nX e r�e r�ddl m
Z
mZ nd Z
ZddlmZ ddlmZmZ G d d
� d
e�ZG d d � d ej�ZG d
d� de�ZG dd� de�Zedk �r G dd� dej�ZG dd� de�Z G dd� de�Z G dd� de�Z G dd� de�Z G dd � d e�Z dS ) z'
Tests for L{twisted.conch.ssh.agent}.
� )�absolute_import�divisionN)�unittest)�iosim)�keys�agent)�keydata)�
ConchError�MissingKeyStoreErrorc @ s 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 ZdZedkrdZnedks*edkr.dZdd� ZdS )�
AgentTestBasez*
Tests for SSHAgentServer/Client.
Nz,iosim requires SSL, but SSL is not availablez)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 ZprivateRSA_openssh�
rsaPrivateZprivateDSA_openssh�
dsaPrivateZpublicRSA_openssh� rsaPublicZpublicDSA_openssh� dsaPublicr
r r r �setUp4 s �
zAgentTestBase.setUp) r r r r r �skipr r r$ r r r r r + s r c @ s e Zd ZdZdd� ZdS )�&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 )
�structZpackr ZAGENTC_REQUEST_IDENTITIESr r �__dict__Z assertRaisesr
Z dataReceived)r �msgr r r �/test_factorySuppliesKeyStorageForServerProtocolL s �zVServerProtocolContractWithFactoryTests.test_factorySuppliesKeyStorageForServerProtocolN)r r r r r+ r r r r r& F s r& c @ s( 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 ZAGENT_RSA_IDENTITIES_ANSWER�ord)Zpacketr
r r �_cbd s �zRUnimplementedVersionOneServerTests.test_agentc_REQUEST_RSA_IDENTITIES.<locals>._cb)r � sendRequestr Z AGENTC_REQUEST_RSA_IDENTITIESr �flush� addCallback)r �dr0 r r
r �"test_agentc_REQUEST_RSA_IDENTITIES^ s
zEUnimplementedVersionOneServerTests.test_agentc_REQUEST_RSA_IDENTITIESc C s( | j �tjd�}| j�� |�| jd�S )z[
assert that we get the correct op code for an RSA remove identity request
r- )r r1 r ZAGENTC_REMOVE_RSA_IDENTITYr r2 r3 r. �r r4 r r r � test_agentc_REMOVE_RSA_IDENTITYj s
zBUnimplementedVersionOneServerTests.test_agentc_REMOVE_RSA_IDENTITYc C s( | j �tjd�}| j�� |�| jd�S )zj
assert that we get the correct op code for an RSA remove all identities
request.
r- )r r1 r Z AGENTC_REMOVE_ALL_RSA_IDENTITIESr r2 r3 r. r6 r r r �%test_agentc_REMOVE_ALL_RSA_IDENTITIESs s
zHUnimplementedVersionOneServerTests.test_agentc_REMOVE_ALL_RSA_IDENTITIESN)r r r r r5 r7 r8 r r r r r, U s r, 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 s | � dd� d S �N� r- �Z sendResponse�r �datar r r �agentc_REQUEST_IDENTITIES� s z'CorruptServer.agentc_REQUEST_IDENTITIESc C s | � dd� d S r: r< r= r r r �agentc_SIGN_REQUEST� s z!CorruptServer.agentc_SIGN_REQUESTN)r r r r r? r@ r r r r r9 s r9 c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �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 r9 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 r2 �
assertFailurer r6 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 r2 rE r r6 r r r �+test_requestIdentitiesCallbackErrorHandling� s
zGClientWithBrokenServerTests.test_requestIdentitiesCallbackErrorHandlingN)r r r r r$ rF rH r r r r rA � s rA c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) �AgentKeyAdditionTestsz<
Test adding different flavors of keys to an agent.
c s2 � 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 s: � j jj� j�� }� �� j|d � � �d|d � d S �Nr r- r' �r r r r rD r. ��ignoredZ serverKeyr
r r �_check� s zBAgentKeyAdditionTests.test_addRSAIdentityNoComment.<locals>._check�r � addIdentityr � privateBlobr r2 r3 �r r4 rO r r
r � test_addRSAIdentityNoComment� s
z2AgentKeyAdditionTests.test_addRSAIdentityNoCommentc s2 � j �� j�� �}� j�� � fdd�}|�|�S )rJ c s: � j jj� j�� }� �� j|d � � �d|d � d S rK �r r r r! rD r. rM r
r r rO � s zBAgentKeyAdditionTests.test_addDSAIdentityNoComment.<locals>._check�r rQ r! rR r r2 r3 rS r r
r � test_addDSAIdentityNoComment� s
z2AgentKeyAdditionTests.test_addDSAIdentityNoCommentc s6 � 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�Zcommentc s: � j jj� j�� }� �� j|d � � �d|d � d S �Nr rY r' rL rM r
r r rO � s zDAgentKeyAdditionTests.test_addRSAIdentityWithComment.<locals>._checkrP rS r r
r � test_addRSAIdentityWithComment� s �
z4AgentKeyAdditionTests.test_addRSAIdentityWithCommentc s6 � j j� j�� dd�}� j�� � fdd�}|�|�S )rX rY rZ c s: � j jj� j�� }� �� j|d � � �d|d � d S r[ rU rM r
r r rO � s zDAgentKeyAdditionTests.test_addDSAIdentityWithComment.<locals>._checkrV rS r r
r � test_addDSAIdentityWithComment� s �
z4AgentKeyAdditionTests.test_addDSAIdentityWithCommentN)r r r r rT rW r\ r] r r r r rI � s
rI c @ s e Zd Zdd� ZdS )�AgentClientFailureTestsc C s$ | j �dd�}| j�� | �|t�S )zK
verify that the client raises ConchError on AGENT_FAILURE
r; r- )r r1 r r2 rE r r6 r r r �test_agentFailure s
z)AgentClientFailureTests.test_agentFailureN)r r r r_ r r r r r^ s r^ c @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd S )
�AgentIdentityRequestsTestszJ
Test operations against a server with identities already loaded.
c C sB t �| � | jdf| jjj| j�� < | jdf| jjj| j�� < d S �N� a comment� another comment�r r$ r! r r r rD r r
r r r r$ s
� �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.
rB N) r rC r" rD r r2 ZsuccessResultOfr Zsignr. �
assertTrue�verify)r r4 Z 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.
rB c s � � � j�| d�� d S )NrB )re r# rf )Zsigr
r r rO 0 s z;AgentIdentityRequestsTests.test_signDataDSA.<locals>._check)r rC r# rD r r2 r3 rS 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.
rB ) r r r r" rD r rC r r2 rE r r6 r r r �$test_signDataRSAErrbackOnUnknownBlob8 s
z?AgentIdentityRequestsTests.test_signDataRSAErrbackOnUnknownBlobc s* � 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 )Nrb rc r' r rD )�type)r# rD r" r r r r. )Zkeytrg Zreceived�kr
r r rO J s "zAAgentIdentityRequestsTests.test_requestIdentities.<locals>._check)r rG r r2 r3 rS r r
r �test_requestIdentitiesC s
z1AgentIdentityRequestsTests.test_requestIdentitiesN) r r r r r$ rh ri rj rm r r r r r` s r` c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) �AgentKeyRemovalTestsz<
Test support for removing keys in a remote server.
c C sB t �| � | jdf| jjj| j�� < | jdf| jjj| j�� < d S ra rd r
r r r r$ \ s
� �zAgentKeyRemovalTests.setUpc s2 � j �� j�� �}� j�� � fdd�}|�|�S )z<
Assert that we can remove an RSA identity.
c sJ � � dt� jjj�� � �� j�� � jjj� � �� j �� � jjj� d S �Nr' )
r. �lenr r r �assertInr! rD Z assertNotInr �rN r
r r rO l s z;AgentKeyRemovalTests.test_removeRSAIdentity.<locals>._check)r �removeIdentityr rD r r2 r3 rS r r
r �test_removeRSAIdentityd s
z+AgentKeyRemovalTests.test_removeRSAIdentityc s2 � j �� j�� �}� j�� � fdd�}|�|�S )z;
Assert that we can remove a DSA identity.
c s2 � � dt� jjj�� � �� j�� � jjj� d S ro )r. rp r r r rq r rD rr r
r r rO { s z;AgentKeyRemovalTests.test_removeDSAIdentity.<locals>._check)r rs r! rD r r2 r3 rS r r
r �test_removeDSAIdentitys s
z+AgentKeyRemovalTests.test_removeDSAIdentityc s* � j �� }� j�� � fdd�}|�|�S )z;
Assert that we can remove all identities.
c s � � dt� jjj�� d S )Nr )r. rp r r r rr r
r r rO � s z=AgentKeyRemovalTests.test_removeAllIdentities.<locals>._check)r ZremoveAllIdentitiesr r2 r3 rS r r
r �test_removeAllIdentities� s
z-AgentKeyRemovalTests.test_removeAllIdentitiesN)r r r r r$ rt ru rv r r r r rn W s
rn )!r Z
__future__r r r( Z
twisted.trialr Z twisted.testr Z cryptography� ImportErrorZpyasn1Ztwisted.conch.sshr r Ztwisted.conch.testr Ztwisted.conch.errorr r
�objectr ZTestCaser r& r, r r9 rA rI r^ r` rn r r r r �<module> s8
)
%P I