U n5{i@sdZddlZddlmZddlmZmZmZmZddl m Z ddl m Z m Z mZmZddlmZddlmZGd d d ejZdS) z A Factory for SSH servers. See also L{twisted.conch.openssh_compat.factory} for OpenSSH compatibility. Maintainer: Paul Swartz N)chain)DictListOptionalTuple)error)_kex connection transportuserauth)protocol)Loggerc@seZdZUdZeeeeeeeffe d<e Z e j ZejejdZddddZdd Zd d Zd d ZeeeeeeeffdddZeeeefdddZddZdS) SSHFactoryz$ A Factory for SSH servers. primes) ssh-userauthsssh-connectionN)returncCsVt|ds||_t|ds(||_|jr4|js>tdt|dsR||_dS)z4 Check for public and private keys. publicKeys privateKeyszno host keys, failingrN) hasattr getPublicKeysrgetPrivateKeysrrZ ConchError getPrimesrselfr=/tmp/pip-unpacked-wheel-mnhg7r0q/twisted/conch/ssh/factory.py startFactory&s       zSSHFactory.startFactorycCsVtj||}ttdd|jD|_|j sR|j ddd|j D|_ |S)a= Create an instance of the server side of the SSH protocol. @type addr: L{twisted.internet.interfaces.IAddress} provider @param addr: The address at which the server will listen. @rtype: L{twisted.conch.ssh.transport.SSHServerTransport} @return: The built transport. css|]}|VqdSN)ZsupportedSignatureAlgorithms).0keyrrr ?sz+SSHFactory.buildProtocol..zTdisabling non-fixed-group key exchange algorithms because we cannot find moduli filecSs$g|]}t|st|r|qSr)rZ isFixedGroupZisEllipticCurve)rZ kexAlgorithmrrr Hs z,SSHFactory.buildProtocol..) r Factory buildProtocollistr from_iterablervaluesZsupportedPublicKeysr_loginfoZsupportedKeyExchanges)raddrtrrrr#3s  zSSHFactory.buildProtocolcCs tddS)z Called when the factory is started to get the public portions of the servers host keys. Returns a dictionary mapping SSH key types to public key strings. @rtype: L{dict} zgetPublicKeys unimplementedNNotImplementedErrorrrrrrOszSSHFactory.getPublicKeyscCs tddS)z Called when the factory is started to get the private portions of the servers host keys. Returns a dictionary mapping SSH key types to L{twisted.conch.ssh.keys.Key} objects. @rtype: L{dict} zgetPrivateKeys unimplementedNr+rrrrrYszSSHFactory.getPrivateKeyscCsdS)z Called when the factory is started to get Diffie-Hellman generators and primes to use. Returns a dictionary mapping number of bits to lists of tuple of (generator, prime). NrrrrrrcszSSHFactory.getPrimes)bitsrcsPttdfdd }|jdk s&tdt|j|d}|d}t|j|S)z Return a tuple of (g, p) for a Diffe-Hellman process, with p being as close to C{bits} bits as possible. )ircs t|Sr)abs)r.r-rrkeyfuncpsz&SSHFactory.getDHPrime..keyfuncNz(Factory should have been started by now.)rr)intrAssertionErrorsortedkeysrandomchoice)rr-r1Z primesKeysZrealBitsrr0r getDHPrimejs zSSHFactory.getDHPrimecCs |dkst|dr|j|SdS)z Return a class to use as a service for the given transport. @type transport: L{transport.SSHServerTransport} @type service: L{bytes} @rtype: subclass of L{service.SSHService} rZavatarN)rservices)rr Zservicerrr getServicexszSSHFactory.getService)__name__ __module__ __qualname____doc__rrr2rr__annotations__r r'r ZSSHServerTransportr r ZSSHUserAuthServerr Z SSHConnectionr9rr#rrrr8r:rrrrrs     &r)r>r6 itertoolsrtypingrrrrZ twisted.conchrZtwisted.conch.sshrr r r Ztwisted.internetr Ztwisted.loggerr r"rrrrrs