REDROOM

Edit File Size: 4.13 KB

//lib/python3/dist-packages/twisted/conch/ssh/factory.py

Text Base64

# Copyright (c) Twisted Matrix Laboratories.
# See LICENSE for details.

"""
A Factory for SSH servers.

See also L{twisted.conch.openssh_compat.factory} for OpenSSH compatibility.

Maintainer: Paul Swartz
"""

import random
from itertools import chain
from typing import Dict, List, Optional, Tuple

from twisted.conch import error
from twisted.conch.ssh import _kex, connection, transport, userauth
from twisted.internet import protocol
from twisted.logger import Logger

class SSHFactory(protocol.Factory):
    """
    A Factory for SSH servers.
    """

primes: Optional[Dict[int, List[Tuple[int, int]]]]

_log = Logger()
    protocol = transport.SSHServerTransport

services = {
        b"ssh-userauth": userauth.SSHUserAuthServer,
        b"ssh-connection": connection.SSHConnection,
    }

def startFactory(self) -> None:
        """
        Check for public and private keys.
        """
        if not hasattr(self, "publicKeys"):
            self.publicKeys = self.getPublicKeys()
        if not hasattr(self, "privateKeys"):
            self.privateKeys = self.getPrivateKeys()
        if not self.publicKeys or not self.privateKeys:
            raise error.ConchError("no host keys, failing")
        if not hasattr(self, "primes"):
            self.primes = self.getPrimes()

def buildProtocol(self, addr):
        """
        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.
        """
        t = protocol.Factory.buildProtocol(self, addr)
        t.supportedPublicKeys = list(
            chain.from_iterable(
                key.supportedSignatureAlgorithms() for key in self.privateKeys.values()
            )
        )
        if not self.primes:
            self._log.info(
                "disabling non-fixed-group key exchange algorithms "
                "because we cannot find moduli file"
            )
            t.supportedKeyExchanges = [
                kexAlgorithm
                for kexAlgorithm in t.supportedKeyExchanges
                if _kex.isFixedGroup(kexAlgorithm) or _kex.isEllipticCurve(kexAlgorithm)
            ]
        return t

def getPublicKeys(self):
        """
        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}
        """
        raise NotImplementedError("getPublicKeys unimplemented")

def getPrivateKeys(self):
        """
        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}
        """
        raise NotImplementedError("getPrivateKeys unimplemented")

def getPrimes(self) -> Optional[Dict[int, List[Tuple[int, int]]]]:
        """
        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).
        """

def getDHPrime(self, bits: int) -> Tuple[int, int]:
        """
        Return a tuple of (g, p) for a Diffe-Hellman process, with p being as
        close to C{bits} bits as possible.
        """

def keyfunc(i: int) -> int:
            return abs(i - bits)

assert self.primes is not None, "Factory should have been started by now."
        primesKeys = sorted(self.primes.keys(), key=keyfunc)
        realBits = primesKeys[0]
        return random.choice(self.primes[realBits])

def getService(self, transport, service):
        """
        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}
        """
        if service == b"ssh-userauth" or hasattr(transport, "avatar"):
            return self.services[service]

Exit & Reset Text mode: syntax highlighting auto-detects file type.

Name	Size	Perms	Modified	Actions
__pycache__ DIR	-	drwxr-xr-x	2026-01-08 12:56:22	Edit Download recursive recursive
address.py	1.08 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
agent.py	9.29 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
channel.py	9.73 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
common.py	1.91 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
connection.py	24.96 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
factory.py	4.13 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
filetransfer.py	37.19 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
forwarding.py	8.03 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
keys.py	62.38 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
service.py	1.52 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
session.py	13.33 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
sexpy.py	944 B	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
transport.py	80.03 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
userauth.py	27.06 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
_kex.py	8.37 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
__init__.py	182 B	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive

Directory Contents