o

�
b�@s�dZd
dl
Z
d
dlmZ
d
dlmZ
d
dlmZ
d
dlm	Z	
d
dl
mZ
d
dlm
Z

d
d	lmZ
d
ZdZed�
rCed
�
rCd
dlmZ
ndZdZee
d�sPdZdZeee�Gdd�de��
ZdS)z,
Tests for L{twisted.conch.openssh_compat}.
�N)
�skipIf)
�getDHGeneratorAndPrime)
�keydata)
�FilePath)
�
requireModule)
�MockOS)
�TestCaseF��cryptography�pyasn1)
�OpenSSHFactoryTz)Cannot run without cryptography or PyASN1�geteuidzgeteuid/seteuid not availablec@s8eZ
dZd
Zdd�Zdd�Zdd�Zdd	�Zd
d�ZdS)
�OpenSSHFactoryTestsz&
    Tests for L{OpenSSHFactory}.
    c

Cs
t�|_
t|���
|_|j��
|jj|j
_t|���
|_|j��
|jj|j
_	|j�
d
�
�d�

|j�
d�
�d�

|j�
d�
�tj
�

|j�
d�
�tj�

|j�
d�
�d�

|j�
d�
�tj�

|j�
d	�
�d
�

t�|_|�td|jj�
|�td|jj�
dS)
N�ssh_host_foosfoo�bar_key�ssh_host_one_key�ssh_host_two_key�ssh_host_three_keysnot a key contentzssh_host_one_key.pub�modulis�
#    $OpenBSD: moduli,v 1.xx 2016/07/26 12:34:56 jhacker Exp $i
# Time Type Tests Tries Size Generator Modulus
20030501000000 2 6 100 2047 2 FFFFFFFFFFFFFFFFC90FDAA22168C234C4C6628B80DC1CD129024E088A67CC74020BBEA63B139B22514A08798E3404DDEF9519B3CD3A431B302B0A6DF25F14374FE1356D6D51C245E485B576625E7EC6F44C42E9A637ED6B0BFF5CB6F406B7EDEE386BFB5A899FA5AE9F24117C4B1FE649286651ECE45B3DC2007CB8A163BF0598DA48361C55D39A69163FA8FD24CF5F83655D23DCA3AD961C62F356208552BB9ED529077096966D670C354E4ABC9804F1746C08CA18217C32905E462E36CE3BE39E772C180E86039B2783A2EC07A28FB5C55DF06F4C52C9DE2BCBF6955817183995497CEA956AE515D2261898FA051015728E5A8AACAA68FFFFFFFFFFFFFFFF
�seteuid�setegid)r�factoryr�mktemp�keysDir�makedirs�path�dataRoot�	moduliDir�
moduliRoot�child�
setContentr�privateRSA_openssh�privateDSA_openssh�publicRSA_opensshr�mockos�patch�osrr)
�self�r(�H/usr/lib/python3/dist-packages/twisted/conch/test/test_openssh_compat.py�setUp%s&













�

zOpenSSHFactoryTests.setUpc
Cs8|j�
�}
|�t|
�
d
�
|
��}|�t|�
dg
�
dS)zw
        L{OpenSSHFactory.getPublicKeys} should return the available public keys
        in the data directory
        �
�ssh-rsaN)r�
getPublicKeys�assertEqual�len�keys�list�r'r0�keyTypesr(r(r)�test_getPublicKeysJs



z&OpenSSHFactoryTests.test_getPublicKeysc
CsZ|j�
�}
|�t|
�
d
�
|
��}|�t|�
ddh�
|�|jjg�
|�|jjg�
dS)z�
        Will return the available private keys in the data directory, ignoring
        key files which failed to be loaded.
        �r,�ssh-dssN)	r�getPrivateKeysr.r/r0�setr$�seteuidCalls�setegidCallsr2r(r(r)�test_getPrivateKeysTs





z'OpenSSHFactoryTests.test_getPrivateKeysc
s�|j�
d
�
���d�

|��jd�
tj�
��
fdd�}
|�td|
�
|j��}|�	t
|�
d�
|��}|�	t|�
dd	h�
|�	|j
jdt��g�
|�	|j
jdt��g�
d
S)z�
        L{OpenSSHFactory.getPrivateKeys} should switch to root if the keys
        aren't readable by the current user.
        rr
�
c

s��d
�

�
|�
S)Nr<)
�chmod)
�euid��keyFile�savedSeteuidr(r)rls


z>OpenSSHFactoryTests.test_getPrivateKeysAsRoot.<locals>.seteuidrr5r,r6N)rrr=�
addCleanupr&rr%rr7r.r/r0r8r$r9r
r:�getegid)r'rr0r3r(r?r)�test_getPrivateKeysAsRoot`s








z-OpenSSHFactoryTests.test_getPrivateKeysAsRootc
Cs$|j�
�}
|�|
d
td�
g
i
�
dS)zq
        L{OpenSSHFactory.getPrimes} should return the available primes
        in the moduli directory.
        isdiffie-hellman-group14-sha1N)r�	getPrimesr.r)r'�primesr(r(r)�test_getPrimesxs



��z"OpenSSHFactoryTests.test_getPrimesN)	�__name__�
__module__�__qualname__�__doc__r*r4r;rDrGr(r(r(r)rs%
r)rKr&�unittestr�twisted.conch.ssh._kexr�twisted.conch.testr�twisted.python.filepathr�twisted.python.reflectr�twisted.test.test_processr�twisted.trial.unittestr�doSkip�
skipReason�$twisted.conch.openssh_compat.factoryr�hasattrrr(r(r(r)�<module>
s(