U ~Hj" @sUdZddlmZddlmZmZmZmZddlm Z m Z ddl m Z m Z mZddlmZe rjddlmZGdd d e ZGd d d e ZGd d d eZGdddeZGdddeZeeGdddZeeGdddZeeGdddZeeGdddZeeGdddZeeGdddZeeGdddZeeGd d!d!Zeeeeeeeed"Zd#e d$<d%d d&d'd(Z!d%d)d&d*d+Z"d%d)d&d,d-Z#d%d d&d.d/Z$d%d0d&d1d2Z%d3d4d5d6Z&d7S)8z SSH key exchange handling. ) annotations)sha1sha256sha384sha512) TYPE_CHECKINGProtocol) Attribute Interface implementer)error)_Hashc@seZdZddddddZdS) _HashFactory.bytesr )datareturncCsdSN)selfrrr:/tmp/pip-unpacked-wheel-ewod1tgb/twisted/conch/ssh/_kex.py__call__sz_HashFactory.__call__N).)__name__ __module__ __qualname__rrrrrrsrc@s2eZdZUdZedZded<edZded<dS) _IKexAlgorithmzB An L{_IKexAlgorithm} describes a key exchange algorithm. zAn L{int} giving the preference of the algorithm when negotiating key exchange. Algorithms with lower precedence values are more preferred.int preferencezqA callable hash algorithm constructor (e.g. C{hashlib.sha256}) suitable for use with this key exchange algorithm.r hashProcessorN)rrr__doc__r r__annotations__rrrrrrs  rc@s eZdZdZedZedZdS)_IFixedGroupKexAlgorithmzu An L{_IFixedGroupKexAlgorithm} describes a key exchange algorithm with a fixed prime / generator group. zdAn L{int} giving the prime number used in Diffie-Hellman key exchange, or L{None} if not applicable.zAn L{int} giving the generator number used in Diffie-Hellman key exchange, or L{None} if not applicable. (This is not related to Python generator functions.)N)rrrrr prime generatorrrrrr -sr c@seZdZdZdS)#_IEllipticCurveExchangeKexAlgorithmz An L{_IEllipticCurveExchangeKexAlgorithm} describes a key exchange algorithm that uses an elliptic curve exchange between the client and server. Nrrrrrrrrr#?sr#c@seZdZdZdS)_IGroupExchangeKexAlgorithmz An L{_IGroupExchangeKexAlgorithm} describes a key exchange algorithm that uses group exchange between the client and server. A prime / generator group should be chosen at run time based on the requested size. See RFC 4419. Nr$rrrrr%Fsr%c@seZdZdZdZeZdS)_Curve25519SHA256z Elliptic Curve Key Exchange using Curve25519 and SHA256. Defined in U{https://datatracker.ietf.org/doc/draft-ietf-curdle-ssh-curves/}. Nrrrrrrrrrrrr&Psr&c@seZdZdZdZeZdS)_Curve25519SHA256LibSSHzN As L{_Curve25519SHA256}, but with a pre-standardized algorithm name. Nr(rrrrr)[sr)c@seZdZdZdZeZdS)_ECDH256aX Elliptic Curve Key Exchange with SHA-256 as HASH. Defined in RFC 5656. Note that C{ecdh-sha2-nistp256} takes priority over nistp384 or nistp512. This is the same priority from OpenSSH. C{ecdh-sha2-nistp256} is considered preety good cryptography. If you need something better consider using C{curve25519-sha256}. Nr(rrrrr+es r+c@seZdZdZdZeZdS)_ECDH384zT Elliptic Curve Key Exchange with SHA-384 as HASH. Defined in RFC 5656. N)rrrrrrrrrrrr-vsr-c@seZdZdZdZeZdS)_ECDH512zT Elliptic Curve Key Exchange with SHA-512 as HASH. Defined in RFC 5656. N)rrrrrrrrrrrr/sr/c@seZdZdZdZeZdS)_DHGroupExchangeSHA256zc Diffie-Hellman Group and Key Exchange with SHA-256 as HASH. Defined in RFC 4419, 4.2. Nr(rrrrr1sr1c@seZdZdZdZeZdS)_DHGroupExchangeSHA1za Diffie-Hellman Group and Key Exchange with SHA-1 as HASH. Defined in RFC 4419, 4.1. N)rrrrrrrrrrrr3sr3c@s$eZdZdZdZeZedZdZ dS)_DHGroup14SHA1z Diffie-Hellman key exchange with SHA-1 as HASH and Oakley Group 14 (2048-bit MODP Group). Defined in RFC 4253, 8.2. Ai32317006071311007300338913926423828248817941241140239112842009751400741706634354222619689417363569347117901737909704191754605873209195028853758986185622153212175412514901774520270235796078236248884246189477587641105928646099411723245426622522193230540919037680524235519125679715870117001058055877651038861847280257976054903569732561526167081339361799541336476559160368317896729073178384589680639671900977202194168647225871031411336429319536193471636533209717077448227988588565369208645296636077250268955505928362751121174096972998068410554359584866583291642136218231078990999448652468262416972035911852507045361090559r*N) rrrrrrrrr!r"rrrrr5s r5)curve25519-sha256scurve25519-sha256@libssh.orgs$diffie-hellman-group-exchange-sha256s"diffie-hellman-group-exchange-sha1sdiffie-hellman-group14-sha1secdh-sha2-nistp256secdh-sha2-nistp384secdh-sha2-nistp521zdict[bytes, _IKexAlgorithm]_kexAlgorithmsr) kexAlgorithmrcCs |tkrtd|t|S)aY Get a description of a named key exchange algorithm. @param kexAlgorithm: The key exchange algorithm name. @type kexAlgorithm: L{bytes} @return: A description of the key exchange algorithm named by C{kexAlgorithm}. @rtype: L{_IKexAlgorithm} @raises ConchError: if the key exchange algorithm is not found. z$Unsupported key exchange algorithm: )r8r Z ConchErrorr9rrrgetKexs r;boolcCstt|S)z Returns C{True} if C{kexAlgorithm} is an elliptic curve. @param kexAlgorithm: The key exchange algorithm name. @return: C{True} if C{kexAlgorithm} is an elliptic curve, otherwise C{False}. )r# providedByr;r:rrrisEllipticCurves r>cCstt|S)z Returns C{True} if C{kexAlgorithm} has a fixed prime / generator group. @param kexAlgorithm: The key exchange algorithm name. @return: C{True} if C{kexAlgorithm} has a fixed prime / generator group, otherwise C{False}. )r r=r;r:rrr isFixedGroups r?cCst|}|jS)z Get the hash algorithm callable to use in key exchange. @param kexAlgorithm: The key exchange algorithm name. @return: A callable hash algorithm constructor (e.g. C{hashlib.sha256}). )r;rr9ZkexrrrgetHashProcessorsrAztuple[int, int]cCstt|}|j|jfS)z Get the generator and the prime to use in key exchange. @param kexAlgorithm: The key exchange algorithm name. @type kexAlgorithm: L{bytes} @return: A L{tuple} containing L{int} generator and L{int} prime. @rtype: L{tuple} )r r;r"r!r@rrrgetDHGeneratorAndPrimes rBz list[bytes])rcsddlm}ddlm}ddlm}|}ttD]V}| drj| dd}| | ||}n| dr~| }nd}|s:|q:tfd d d S) z Get a list of supported key exchange algorithm names in order of preference. @return: A C{list} of supported key exchange algorithm names. r)default_backend)ec) _curveTablesecdhsecdsar7Tcs |jSr)rr:Z kexAlgorithmsrr(z*getSupportedKeyExchanges..)key)Zcryptography.hazmat.backendsrCZ)cryptography.hazmat.primitives.asymmetricrDZtwisted.conch.ssh.keysrEr8copylist startswithreplaceZ+elliptic_curve_exchange_algorithm_supportedZECDHZx25519_supportedpopsorted)rCrDrEbackendZ keyAlgorithmZkeyAlgorithmDsa supportedrrFrgetSupportedKeyExchanges s*          rRN)'r __future__rhashlibrrrrtypingrrZzope.interfacer r r Z twisted.conchr r rrr r#r%r&r)r+r-r/r1r3r5r8rr;r>r?rArBrRrrrrsV