�
Ϫ�f� � � � d Z ddlZddlmZ ddlmZmZmZmZ ddl m
Z
ddl m Z m
Z
mZmZ ddlmZ ddlmZ G d � d
ej( � Zy) z�
A Factory for SSH servers.
See also L{twisted.conch.openssh_compat.factory} for OpenSSH compatibility.
Maintainer: Paul Swartz
� N)�chain)�Dict�List�Optional�Tuple)�error)�_kex�
connection� transport�userauth)�protocol)�Loggerc � � e Zd ZU dZeeeeeeef f e d< e
� Z e j Zej ej$ d�Zdd�Zd� Zd� Zd � Zdeeeeeeef f fd
�Zd edeeef fd �Zd
� Zy)�
SSHFactoryz$
A Factory for SSH servers.
�primes)� ssh-userauths ssh-connection�returnNc �& � t | d� s| j � | _ t | d� s| j � | _ | j r | j st j d� �t | d� s| j � | _ yy)z4
Check for public and private keys.
�
publicKeys� privateKeyszno host keys, failingr N) �hasattr�
getPublicKeysr �getPrivateKeysr r �
ConchError� getPrimesr ��selfs �;/usr/lib/python3/dist-packages/twisted/conch/ssh/factory.py� startFactoryzSSHFactory.startFactory& sw � � �t�\�*� "� 0� 0� 2�D�O��t�]�+� #� 2� 2� 4�D� ����d�&6�&6��"�"�#:�;� ;��t�X�&� �.�.�*�D�K� '� c � � t j j | |� }t t j
d� | j j � D � � � |_ | j sd| j j d� |j D �cg c]. }t j |� st j |� r|��0 c}|_ |S c c}w )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.
c 3 �< K � | ] }|j � �� � y �w�N)� supportedSignatureAlgorithms)�.0�keys r � <genexpr>z+SSHFactory.buildProtocol.<locals>.<genexpr>? s � �� � �7:��0�0�2� �s � zTdisabling non-fixed-group key exchange algorithms because we cannot find moduli file)r
�Factory�
buildProtocol�listr �
from_iterabler �values�supportedPublicKeysr �_log�info�supportedKeyExchangesr � isFixedGroup�isEllipticCurve)r �addr�t� kexAlgorithms r r) zSSHFactory.buildProtocol3 s� � �
� � � *� *�4�� 6�� $� � � � �>B�>N�>N�>U�>U�>W� �
�!
�� �
�{�{� �I�I�N�N�5�
� %&�$;�$;�'� ��$�$�\�2�d�6J�6J�<�6X� �'�A� #�
��� 's �3Cc � � t d� �)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 unimplemented��NotImplementedErrorr s r r zSSHFactory.getPublicKeysO s � � "�"?�@�@r c � � t d� �)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}
z getPrivateKeys unimplementedr7 r s r r zSSHFactory.getPrivateKeysY s � � "�"@�A�Ar c � � y)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).
N� r s r r zSSHFactory.getPrimesc s � r �bitsc �� �� dt dt f�fd� }| j �J d� �t | j j � |�� }|d }t j
| 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.
�ir c � �� t | �z
� S r# )�abs)r> r< s �r �keyfuncz&SSHFactory.getDHPrime.<locals>.keyfuncp s �� ��q�4�x�=� r z(Factory should have been started by now.)r& r )�intr �sorted�keys�random�choice)r r< rA �
primesKeys�realBitss ` r �
getDHPrimezSSHFactory.getDHPrimej sh �� � !�s� !�s� !� �{�{�&�R�(R�R�&��D�K�K� ,� ,� .�G�<�
� �a�=���}�}�T�[�[�� 2�3�3r c �D � |dk( s t |d� r| j | S y)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 �avatarN)r �services)r r �services r �
getServicezSSHFactory.getServicex s* � � �o� %���H�)E��=�=��)� )� *Fr )r N)�__name__�
__module__� __qualname__�__doc__r r rB r r �__annotations__r r. r �SSHServerTransportr
r �SSHUserAuthServerr
�
SSHConnectionrL r r) r r r rI rN r; r r r r s� � ��
�T�#�t�E�#�s�(�O� 4�4�5� 6�6� �8�D��+�+�H� "�3�3�%�3�3��H�
+��8A�B� �8�D��d�5��c��?�.C�)C�$D�E� � 4�s� 4�u�S�#�X�� 4� *r r )rR rE � itertoolsr �typingr r r r �
twisted.conchr �twisted.conch.sshr r
r r �twisted.internetr
�twisted.loggerr r( r r; r r �<module>r] s; ��� � � .� .� � C� C� %� !�j*��!�!� j*r |