File: //lib/python3/dist-packages/twisted/conch/scripts/__pycache__/ckeygen.cpython-38.pyc
U
W[^' � @ sh d Z ddlmZ ddlZddlZddlZddlZddlmZ ddl m
Z
ejej kr�zddl Z e j
e jf W n* eefk
r� dejd< e
e� Y nX ddlmZ ddlmZmZmZmZ dd lmZmZ e � Z d
d � Z G d d
� d
ej �Z dd� Z!dd� Z"dd� Z#e d�dd� �Z$e d�dd� �Z%e d�dd � �Z&d d � Z'd d � Z(d!d"� Z)d#d$� Z*e+d%k�rde!� dS )&z2
Implementation module for the `ckeygen` command.
� )�print_functionN)�wraps)�reload�termios)�keys)�failure�filepath�log�usage)� raw_input�_PY3c s � fdd�}|S )Nc s t � �� fdd��}|t�< |S )Nc s
� | |�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 ZdZdZddddgdddd gd
d dd gd
dddgddddgddddgddddggZddd gd d d gd d!d"gd#dd$gd%d&d'ggZejd
e� e
e � � ��id(�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�fzFilename of the key file.�type�tz Specify type of key to create.�comment�CzProvide new comment.�newpass�NzProvide new passphrase.�pass�PzProvide old passphrase.�format�o�
sha256-base64z Fingerprint format of key file.� fingerprint�lz Show fingerprint of key file.�
changepass�pz&Change passphrase of private key file.�quiet�qzQuiet.�
no-passphrasez"Create the key with no passphrase.�showpub�yz+Read private key file and print public key.)Z
optActions)�__name__�
__module__� __qualname__ZsynopsisZlongdescZ
optParametersZoptFlagsr
Z CompletionsZ CompleteList�listr r ZcompDatar
r
r
r r ) s&
���r c
C s t � } z| �tjdd � � W nB tjk
r` } z"td| � | �� t�d� W 5 d }~X Y nX t �
� t t _ | d r�| d �
� tkr�td| d � t| d �
� | � n t�d| d d�t�� �f � nJ| d r�t| � n8| d r�t| � n&| d �rt| � n| �� 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+ r0 )r Z parseOptions�sys�argvr
Z
UsageError�printZopt_help�exitr Z discardLogs� handleErrorZdeferr�lowerr �joinr �printFingerprint�changePassPhrase�displayPublicKey)�options�ur
r
r �runB s4 ��
rC c C sP | d dkr t jj| d< | S | d dkr8t jj| d< | S t �d| d f ��d S )Nr& zmd5-hexr( z"Unsupported fingerprint format: %s)r ZFingerprintFormatsZMD5_HEXZ
SHA256_BASE64ZBadFingerPrintFormat)rA r
r
r �enumrepresentation_ s �rD c C s da t�t�� � � d S )N� )Z
exitStatusr �errr ZFailurer
r
r
r r; l s r; �rsac C sZ ddl m} ddlm} | d s(d| d< |jt| d �d|� d�}t�|�}t|| � d S )Nr ��default_backend)rG r � i )�key_sizeZpublic_exponent�backend) � cryptography.hazmat.backendsrI �)cryptography.hazmat.primitives.asymmetricrG �generate_private_key�intr �Key�_saveKey)rA rI rG � keyPrimitive�keyr
r
r �generateRSAkeys s
�
rU �dsac C sX ddl m} ddlm} | d s(d| d< |jt| d �|� d�}t�|�}t|| � d S )Nr rH )rV r rJ )rK rL ) rM rI rN rV rO rP r rQ rR )rA rI rV rS rT r
r
r �generateDSAkey� s
�
rW �ecdsac C sl ddl m} ddlm} | d s(d| d< dt| d ��d� }|jtj| |� d�}t� |�}t
|| � d S ) Nr rH )�ecr � s ecdsa-sha2-nistp�ascii)�curverL ) rM rI rN rY �str�encoderO r Z _curveTablerQ rR )rA rI rY r\ rS rT r
r
r �generateECDSAkey� s �
r_ c C s� | d s$t j�d�}td| �| d< t j�| d d �rH| d d7 < t| �} z@tj�| d �}t d|�
� |� | d �t j� | d �f � W n tj
k
r� t�d� Y nX d S )Nr �
~/.ssh/id_rsa�%Enter file in which the key is (%s): �.pubz%s %s %sr& zbad key)�os�path�
expanduserr �existsrD r rQ �fromFiler9 �sizer) �basename� BadKeyErrorr7 r: )rA r rT r
r
r r> � s � r> c
C s | d s$t j�d�}td| �| d< ztj�| d �}W n� tjk
r� | �d�sbt � d�| d< z tjj| d | d d�}W nR tj
k
r� t � d� Y n4 tjk
r� } zt � d|f � W 5 d }~X Y nX Y n6 tj
k
�r
} zt � d|f � W 5 d }~X Y nX | �d ��sNt � d
�}t � d �}||k�r:�qFt
d � �q|| d < z|jd
| d d�}W n4 tk
�r� } zt � d|f � W 5 d }~X Y nX ztjj|| d d� W n< tjtj
fk
�r� } zt � d|f � W 5 d }~X Y nX t| d d��}|�|� W 5 Q R X t
d� d S )Nr r` ra r$ zEnter old passphrase: �Z
passphrasez1Could not change passphrase: old passphrase errorz Could not change passphrase: %sr" z0Enter new passphrase (empty for no passphrase): � Enter same passphrase again: �%Passphrases do not match. Try again.�openssh)Zextra�wbz;Your identification has been saved with the new passphrase.)rc rd re r r rQ rg �EncryptedKeyError�get�getpassrj r7 r: r9 �toString� ExceptionZ
fromString�open�write)rA r rT �e�p1�p2Z
newkeydata�fdr
r
r r? � sR �
�
&" �
""r? c C s� | d s$t j�d�}td| �| d< ztj�| d �}W nF tjk
r~ | �d�sbt � d�| d< tjj| d | d d�}Y nX |�
� � d�}t r�|�
d�}t|� d S ) Nr r` ra r$ zEnter passphrase: rk rn r[ )rc rd re r r rQ rg rp rq rr �publicrs r �decoder9 )rA r rT Z
displayKeyr
r
r r@ � s
�
r@ c
C s� dddd�}|| � � }|d sPtj�d|f �}td|f �}|�� pJ||d< tj�|d �r�td|d f � td �}|d
�� d kr�t �
� |� d �r�d
|d< n8|d s�t � d�}t � d�}||kr�q�td� q�||d< dt �
� t�� f } t�|d ��| �d|d �� t�|d d� t�|d d ��| �� �d| �� t|�}td|d f � td|d f � td|d f � 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}
rX rG rV )ZECZRSAZDSAr z ~/.ssh/id_%sz*Enter file in which to save the key (%s): z%s already exists.zOverwrite (y/n)? r r1 r/ � r$ z,Enter passphrase (empty for no passphrase): rl rm z%s@%srn i�� rb z(Your identification has been saved in %sz(Your public key has been saved in %s.pubz The key fingerprint in %s is:r& N)r rc rd re r �striprf r9 r< r7 r: rq rr Zgetuser�socketZ gethostnamer ZFilePathZ
setContentrs �chmodr{ rD r) )
rT rA ZKeyTypeMappingZ keyTypeNameZ defaultPathZnewPathZynrx ry r r
r
r rR � sF
�
��rR �__main__),�__doc__Z
__future__r r7 rc rr r � functoolsr �impr Z unix_getpassr Z tcgetattrZ tcsetattr� ImportError�AttributeError�modulesZtwisted.conch.sshr Ztwisted.pythonr r r r
Ztwisted.python.compatr r �dictr r ZOptionsr rC rD r; rU rW r_ r> r? r@ rR r2 r
r
r
r �<module> s@
05