← Back
Editing: 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
Save File
Cancel