U  W[@sndZddlmZmZddlmZddlmZddlm Z ddl m Z m Z m Z mZddlZGdd d ejZdS) z A Factory for SSH servers. See also L{twisted.conch.openssh_compat.factory} for OpenSSH compatibility. Maintainer: Paul Swartz )divisionabsolute_import)protocol)log)error)_kex transportuserauth connectionNc@s\eZdZdZejZeje j dZ ddZ ddZ ddZd d Zd d Zd dZddZdS) SSHFactoryz$ A Factory for SSH servers. ) ssh-userauthsssh-connectioncCsVt|ds||_t|ds(||_|jr4|js>tdt|dsR||_dS)z4 Check for public and private keys. publicKeys privateKeyszno host keys, failingprimesN) hasattr getPublicKeysr getPrivateKeysrrZ ConchError getPrimesrselfr;/usr/lib/python3/dist-packages/twisted/conch/ssh/factory.py startFactory!s       zSSHFactory.startFactorycCs@tj||}|j|_|js>s  z,SSHFactory.buildProtocol..) rFactory buildProtocolrkeysZsupportedPublicKeysrrmsgZsupportedKeyExchanges)rZaddrtrrrr/s   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 unimplementedNNotImplementedErrorrrrrrEszSSHFactory.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 rrrrrPszSSHFactory.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). @rtype: L{dict} Nrrrrrr[szSSHFactory.getPrimescs2t|jfddd}|d}t|j|S)z Return a tuple of (g, p) for a Diffe-Hellman process, with p being as close to bits bits as possible. @type bits: L{int} @rtype: L{tuple} cs t|S)N)abs)ibitsrrmz'SSHFactory.getDHPrime..)keyr)sortedrrrandomZchoice)rr%Z primesKeysZrealBitsrr$r getDHPrimeeszSSHFactory.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} r ZavatarN)rservices)rrservicerrr getServicerszSSHFactory.getServiceN)__name__ __module__ __qualname____doc__rZSSHServerTransportrr ZSSHUserAuthServerr Z SSHConnectionr,rrrrrr+r.rrrrr s    r )r2Z __future__rrZtwisted.internetrZtwisted.pythonrZ twisted.conchrZtwisted.conch.sshrrr r r*rr rrrrs