U  W[ã@sŠdZddlZddlmZddlmZddlmZedƒrNedƒrNddlm Z ndZ dd l m Z dd l m Z dd lmZGd d „d eƒZdS)z, Tests for L{twisted.conch.openssh_compat}. éN)ÚTestCase)ÚFilePath)Ú requireModuleZ cryptographyZpyasn1)ÚOpenSSHFactory)ÚgetDHGeneratorAndPrime)Úkeydata)ÚMockOSc@sZeZdZdZeeddƒdkr"dZn edkr.dZdd„Zdd „Z d d „Z d d „Z dd„Z dS)ÚOpenSSHFactoryTestsz& Tests for L{OpenSSHFactory}. ÚgeteuidNzgeteuid/seteuid not availablez)Cannot run without cryptography or PyASN1cCstƒ|_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) NZ ssh_host_foosfooZbar_keyZssh_host_one_keyÚssh_host_two_keyZssh_host_three_keysnot a key contentzssh_host_one_key.pubZmodulis # $OpenBSD: moduli,v 1.xx 2016/07/26 12:34:56 jhacker Exp $ # Time Type Tests Tries Size Generator Modulus 20030501000000 2 6 100 2047 2 FFFFFFFFFFFFFFFFC90FDAA22168C234C4C6628B80DC1CD129024E088A67CC74020BBEA63B139B22514A08798E3404DDEF9519B3CD3A431B302B0A6DF25F14374FE1356D6D51C245E485B576625E7EC6F44C42E9A637ED6B0BFF5CB6F406B7EDEE386BFB5A899FA5AE9F24117C4B1FE649286651ECE45B3DC2007CB8A163BF0598DA48361C55D39A69163FA8FD24CF5F83655D23DCA3AD961C62F356208552BB9ED529077096966D670C354E4ABC9804F1746C08CA18217C32905E462E36CE3BE39E772C180E86039B2783A2EC07A28FB5C55DF06F4C52C9DE2BCBF6955817183995497CEA956AE515D2261898FA051015728E5A8AACAA68FFFFFFFFFFFFFFFF ÚseteuidÚsetegid)rÚfactoryrZmktempÚkeysDirÚmakedirsÚpathZdataRootZ moduliDirZ moduliRootÚchildZ setContentrZprivateRSA_opensshZprivateDSA_opensshZpublicRSA_opensshrÚmockosÚpatchÚosr r )Úself©rúH/usr/lib/python3/dist-packages/twisted/conch/test/test_openssh_compat.pyÚsetUp"s2     ÿ ÿ ÿ ÿzOpenSSHFactoryTests.setUpcCs8|j ¡}| t|ƒd¡| ¡}| t|ƒdg¡dS)zw L{OpenSSHFactory.getPublicKeys} should return the available public keys in the data directory éóssh-rsaN)rZ getPublicKeysÚ assertEqualÚlenÚkeysÚlist©rrÚkeyTypesrrrÚtest_getPublicKeysCs z&OpenSSHFactoryTests.test_getPublicKeyscCs^|j ¡}| t|ƒd¡| ¡}| t|ƒtddgƒ¡| |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ÚgetPrivateKeysrrrÚsetrÚ seteuidCallsÚ setegidCallsr rrrÚtest_getPrivateKeysNs  z'OpenSSHFactoryTests.test_getPrivateKeyscs´|j d¡‰ˆ d¡| ˆjd¡tj‰‡‡fdd„}| td|¡|j ¡}|  t |ƒd¡|  ¡}|  t |ƒt dd gƒ¡|  |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. r réÿcsˆ d¡ˆ|ƒS)Nr*)Úchmod)Zeuid©ZkeyFileZ savedSeteuidrrr fs z>OpenSSHFactoryTests.test_getPrivateKeysAsRoot..seteuidr r#rr$N)rrr+Z addCleanuprr rrr%rrrr&rr'r r(Úgetegid)rr rr!rr,rÚtest_getPrivateKeysAsRoot[s   z-OpenSSHFactoryTests.test_getPrivateKeysAsRootcCs$|j ¡}| |dtdƒgi¡dS)zq L{OpenSSHFactory.getPrimes} should return the available primes in the moduli directory. isdiffie-hellman-group14-sha1N)rZ getPrimesrr)rZprimesrrrÚtest_getPrimesrs  ÿz"OpenSSHFactoryTests.test_getPrimes) Ú__name__Ú __module__Ú __qualname__Ú__doc__ÚgetattrrÚskiprrr"r)r.r/rrrrr s!  r )r3rZtwisted.trial.unittestrZtwisted.python.filepathrZtwisted.python.reflectrZ$twisted.conch.openssh_compat.factoryrZtwisted.conch.ssh._kexrZtwisted.conch.testrZtwisted.test.test_processrr rrrrÚs