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