File: //lib/python3/dist-packages/twisted/conch/scripts/__pycache__/ckeygen.cpython-310.pyc
o
�b1- � @ sv d Z ddlZddlZddlZddlZddlmZ ddlmZ ddl m
Z
ddl m Z m
Z
mZmZ ejejkrWz ddlZejejf W n eefyV dejd< ee� Y nw e� Zdd � ZG d
d � d ej�Zd d
� Z dd� Z dd� Z ed�dd� �Z ed�dd� �Z ed�dd� �Z!ed�d d � �Z"d d � Z#d d!� Z$d"d#� Z%d$d%� Z&d&e'd'e'fd(d)�Z(d*d+� Z)e*d,kr�e � dS dS )-z2
Implementation module for the `ckeygen` command.
� N)�wraps)�reload)�keys)�failure�filepath�log�usage�termiosc s � fdd�}|S )Nc s t � �� fdd��}|t�< |S )Nc s � | i |��S )N� )�args�kwargs�� keygeneratorr
�?/usr/lib/python3/dist-packages/twisted/conch/scripts/ckeygen.py�wrapper" s z:_keyGenerator.<locals>.assignkeygenerator.<locals>.wrapper)r �supportedKeyTypes)r r ��keyTyper
r �assignkeygenerator! s z)_keyGenerator.<locals>.assignkeygeneratorr
)r r r
r r �
_keyGenerator s r c @ s� e Zd ZdZdZg d�g d�g d�g d�g d�g d�g d �g d
�gZg d �g d �g d
�g d�g d�gZeje� e
e � � ��e� ddg�d�d�Z
dS )�GeneralOptionsz Usage: ckeygen [options]
z8ckeygen manipulates public/private keys in various ways.)�bits�bNz$Number of bits in the key to create.)�filename�fNzFilename of the key file.)�type�tNz Specify type of key to create.)�comment�CNzProvide new comment.)�newpass�NNzProvide new passphrase.)�pass�PNzProvide old passphrase.)�format�o�
sha256-base64z Fingerprint format of key file.)�private-key-subtypeNNz5OpenSSH private key subtype to write ("PEM" or "v1").)� fingerprint�lz Show fingerprint of key file.)�
changepass�pz&Change passphrase of private key file.)�quiet�qzQuiet.)�
no-passphraseNz"Create the key with no passphrase.)�showpub�yz+Read private key file and print public key.�PEM�v1)r r& )�
optActionsN)�__name__�
__module__� __qualname__�synopsis�longdesc�
optParameters�optFlagsr � Completions� CompleteList�listr r �compDatar
r
r
r r , s. �� �
�r c
C s( t � } z | �tjdd � � W n" tjy1 } ztd| � | �� t�d� W Y d }~nd }~ww t �
� t t _ | d rk| d �
� tv rYtd| d � t| d �
� | � d S t�d| d d�t�� �f � d S | d rut| � d S | d rt| � d S | d r�t| � d S | �� t�d� d S )
N� z ERROR: %sr z&Generating public/private %s key pair.z"Key type was %s, must be one of %sz, r' r) r. )r � parseOptions�sys�argvr �
UsageError�print�opt_help�exitr � discardLogs� handleError�deferr�lowerr �joinr �printFingerprint�changePassPhrase�displayPublicKey)�options�ur
r
r �runR s8 ���� rP c C sL | d dkrt jj| d< | S | d dkr t jj| d< | S t �d�| d ���)Nr# zmd5-hexr% z"Unsupported fingerprint format: {})r �FingerprintFormats�MD5_HEX�
SHA256_BASE64�BadFingerPrintFormatr# )rN r
r
r �enumrepresentationp s �rU c C s da t�t�� � � )N� )�
exitStatusr �errr �Failurer
r
r
r rG } s rG �rsac C sZ ddl m} ddlm} | d sd| d< |jt| d �d|� d�}t�|�}t|| � d S )Nr ��default_backend)rZ r � i )�key_size�public_exponent�backend) � cryptography.hazmat.backendsr\ �)cryptography.hazmat.primitives.asymmetricrZ �generate_private_key�intr �Key�_saveKey)rN r\ rZ � keyPrimitive�keyr
r
r �generateRSAkey� s
�
ri �dsac C sX ddl m} ddlm} | d sd| d< |jt| d �|� d�}t�|�}t|| � d S )Nr r[ )rj r r] )r^ r` ) ra r\ rb rj rc rd r re rf )rN r\ rj rg rh r
r
r �generateDSAkey� s
�
rk �ecdsac C sl ddl m} ddlm} | d sd| d< dt| d ��d� }|jtj| |� d�}t� |�}t
|| � d S ) Nr r[ )�ecr � s ecdsa-sha2-nistp�ascii)�curver` ) ra r\ rb rm �str�encoderc r � _curveTablere rf )rN r\ rm rp rg rh r
r
r �generateECDSAkey� s �
rt �ed25519c C s. ddl m} |j�� }t�|�}t|| � d S )Nr )ru )rb ru �Ed25519PrivateKey�generater re rf )rN ru rg rh r
r
r �generateEd25519key� s
rx c C s | dkrdS dS )a
Return a reasonable default private key subtype for a given key type.
@type keyType: L{str}
@param keyType: A key type, as returned by
L{twisted.conch.ssh.keys.Key.type}.
@rtype: L{str}
@return: A private OpenSSH key subtype (C{'PEM'} or C{'v1'}).
�Ed25519r1 r0 r
r r
r
r �_defaultPrivateKeySubtype� s rz c C s� | d st j�d�}td| �| d< t j�| d d �r$| d d7 < t| �} z!tj�| d �}t d|�
� |� | d �t j� | d �f � W d S tj
yY t�d� Y d S w )Nr �
~/.ssh/id_rsa�%Enter file in which the key is (%s): �.pubz%s %s %sr# zbad key)�os�path�
expanduser�input�existsrU r re �fromFilerC �sizer' �basename� BadKeyErrorr@ rE )rN r rh r
r
r rK � s& ��
��rK c
C sT | d st j�d�}td| �| d< z
tj�| d �}W nh tjyj | �d�s0t � d�| d< ztjj| d | d d�}W n) tj
yM t � d� Y n tjyg } zt � d|� �� W Y d }~nd }~ww Y n tj
y� } zt � d|� �� W Y d }~nd }~ww | �d �s� t � d �}t � d �}||kr�nt
d
� q�|| d < | �d�d u r�t|�� �| d< z|jd| d | d d�}W n ty� } zt � d|� �� W Y d }~nd }~ww z tjj|| d d� W n tjtj
f�y } zt � d|� �� W Y d }~nd }~ww t| d d��
}|�|� W d � n 1 �s w Y t
d� d S )Nr r{ r| r! zEnter old passphrase: ��
passphrasez1Could not change passphrase: old passphrase errorz Could not change passphrase: r r> z0Enter new passphrase (empty for no passphrase): � Enter same passphrase again: �%Passphrases do not match. Try again.r& �openssh��subtyper� �wbz;Your identification has been saved with the new passphrase.)r~ r r� r� r re r� �EncryptedKeyError�get�getpassr� r@ rE rC rz r �toString� Exception�
fromString�open�write)rN r rh �e�p1�p2�
newkeydata�fdr
r
r rL � sh
��� ��
�
� �� �� � rL c C s� | d st j�d�}td| �| d< z
tj�| d �}W n" tjy> | �d�s0t � d�| d< tjj| d | d d�}Y nw |�
� � d�� d�}t
|� d S ) Nr r{ r| r! zEnter passphrase: r� r� ro )r~ r r� r� r re r� r� r� r� �publicr� �decoderC )rN r rh �
displayKeyr
r
r rM s
� rM �prompt�returnc C s t | �S )zv
Ask the user where to save the key.
This needs to be a separate function so the unit test can patch it.
)r� )r� r
r
r �_inputSaveFile% s r� c
C s� ddddd�}|| � � }|d s*tj�d|� ��}td|� d ��}|�� p'||d< tj�|d �rKtd
�|d �� t d �}|d �
� d
krKt � � |�
d�rUd|d< n |d sr t�d�}t�d�}||krintd� qZ||d< |�
d�du r�t| � � �|d< t�� � dt�� � �} t�|d ��| jd|d |d d�� t�|d d� t�|d d ��| �� jd| d �� t|�}td �|d �� td �|d �� td �|d �� t| �|d �� dS )!z�
Persist a SSH key on local filesystem.
@param key: Key which is persisted on local filesystem.
@type key: C{keys.Key} implementation.
@param options:
@type options: L{dict}
rl ru rZ rj )�ECry �RSA�DSAr z
~/.ssh/id_z%Enter file in which to save the key (z): z{} already exists.zOverwrite (y/n)? r r/ r- � r! r> z,Enter passphrase (empty for no passphrase): r� r� r& N�@r� r� i�� r} )r z(Your identification has been saved in {}z(Your public key has been saved in {}.pubz The key fingerprint in {} is:r# )r r~ r r� r� �stripr� rC r# r� rI r@ rE r� r� rz �getuser�socket� gethostnamer �FilePath�
setContentr� �chmodr� rU r' )
rh rN �KeyTypeMapping� keyTypeName� defaultPath�newPath�ynr� r� r r
r
r rf . sV
�
����rf �__main__)+�__doc__r� r~ r� r@ � functoolsr �impr �twisted.conch.sshr �twisted.pythonr r r r � unix_getpassr � tcgetattr� tcsetattr� ImportError�AttributeError�modules�dictr r �Optionsr rP rU rG ri rk rt rx rz rK rL rM rq r� rf r3 r
r
r
r �<module> sP
� &
3 >
�