File: //lib/python3.6/site-packages/rsa/__pycache__/cli.cpython-36.opt-1.pyc
3
<�a�' � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZeej j
j � �Z
ej ej
eje f Zdd�dd�ZG dd� dejd�ZG d d
� d
e�ZG d d � d e�ZG d
d� de�ZG dd� de�Ze� Ze� Ze� Ze� ZdS )zWCommandline scripts.
These scripts are called by the executables defined in setup.py.
� N)�returnc C s� t jddd�} | jdddd� | jdd dd
d� | jd d dd
d� | jtjdd� �\}}t|�dkrv| j� td��yt |d �}W nJ t
k
r� } z.| j� t d|d tj d� td�|�W Y dd}~X nX t d| tj d� t
j|�\}}|j�r>t d|j tj d� |j|jd�}t|jd��}|j|� W dQ R X |j|jd�}|j�r�t d|j tj d� t|jd��}|j|� W dQ R X n t dtj d� tjjj|� dS ) zKey generator.z usage: %prog [options] keysizez/Generates a new RSA key pair of "keysize" bits.)�usage� descriptionz--pubout�stringz�Output filename for the public key. The public key is not saved if this option is not present. You can use pyrsa-priv2pub to create the public key file later.)�type�helpz-oz--outz`Output filename for the private key. The key is written to stdout if this option is not present.z--formz7key format of the private and public keys - default PEM�PEM�DER)r �choices�default� Nr zNot a valid number: %s)�filezGenerating %i-bit keyzWriting public key to %s)�format�wbzWriting private key to %sz Writing private key to stdout)r r )�optparse� OptionParser�
add_option�
parse_args�sys�argv�len�
print_help�
SystemExit�int�
ValueError�print�stderr�rsaZnewkeysZpuboutZ
save_pkcs1Zform�open�write�out�stdout�buffer) �parser�cli�cli_argsZkeysize�ex�pub_key�priv_key�data�outfile� r+ �/usr/lib/python3.6/cli.py�keygen! sR r- c @ s� e Zd ZdZdZdZdZdZdZdZ dZ
d Z d
Z d Z
ejZd d
�dd�Zejeejjeejd�dd��Zd d
�dd�Zej e j ej e f d
�dd�Z!e e ejjd�dd�Z"e ed�dd �Z#ee d d �d d �Z$d S ) �CryptoOperationz9CLI callable that operates with input, output, and a key.�publicz'usage: %%prog [options] %(keyname)s_key� �decrypt� decrypted�
decryptingzEName of the file to %(operation)s. Reads from stdin if not specified.zjName of the file to write the %(operation_past)s file to. Written to stdout if this option is not present.r TN)r c C s4 | j | jj | _ | j| jj | _| j| jj | _d S )N)r � __class__�__dict__�
input_help� output_help)�selfr+ r+ r, �__init__w s zCryptoOperation.__init__)�indata�keyr% r c C s dS )z�Performs the program's operation.
Implement in a subclass.
:returns: the data to write to the output.
Nr+ )r8 r: r; r% r+ r+ r, �perform_operation| s z!CryptoOperation.perform_operationc C sd | j � \}}| j|d |j�}| j|j�}t| jj� tj d� | j
|||�}| j r`| j ||j
� dS )zRuns the program.r )r
N)� parse_cli�read_key�keyform� read_infile�inputr �operation_progressive�titler r r< �
has_output�
write_outfile�output)r8 r$ r% r; r: �outdatar+ r+ r, �__call__� s zCryptoOperation.__call__c C s� t j| j| jd�}|jddd| jd� | jr@|jddd| jd� |jdd | j dd
d � |j t
j d
d� �\}}t |�| j
kr�|j� td
��||fS )zFParse the CLI options
:returns: (cli_opts, cli_args)
)r r z-iz--inputr )r r z-oz--outputz --keyformz&Key format of the %s key - default PEMr r )r r
r r N)r r )r r r r r r6 rD r7 �keynamer r r r �expected_cli_argsr r )r8 r# r$ r% r+ r+ r, r= � s zCryptoOperation.parse_cli)�filenamer? r c
C sD t d| j|f tjd� t|d��}|j� }W dQ R X | jj||�S )z Reads a public or private key.zReading %s key from %s)r
�rbN)r rI r r r �read� key_classZ
load_pkcs1)r8 rK r? ZkeyfileZkeydatar+ r+ r, r> � s zCryptoOperation.read_key)�innamer c C sJ |r0t d| tjd� t|d��
}|j� S Q R X t dtjd� tjjj� S )zRead the input filezReading input from %s)r
rL NzReading input from stdin)r r r r rM �stdinr" )r8 rO Zinfiler+ r+ r, r@ � s zCryptoOperation.read_infile)rG �outnamer c C sX |r8t d| tjd� t|d��}|j|� W dQ R X n t dtjd� tjjj|� dS )zWrite the output filezWriting output to %s)r
r NzWriting output to stdout)r r r r r r! r" )r8 rG rQ r* r+ r+ r, rE � s z CryptoOperation.write_outfile)%�__name__�
__module__� __qualname__�__doc__rI r r � operation�operation_pastrB r6 r7 rJ rD r � PublicKeyrN r9 �abc�abstractmethod�bytesr; � AbstractKey� Indexable�typingZAnyr< rH �Tupler ZValues�List�strr= r> r@ rE r+ r+ r+ r, r. d s*
r. )� metaclassc @ s@ e Zd ZdZdZdZdZdZdZf fe e
j j e
e d�dd �Zd
S ) �EncryptOperationzEncrypts a file.r/ zWEncrypts a file. The file must be shorter than the key length in order to be encrypted.�encryptZ encryptedZ
encrypting)r: r' r% r c C s t j||�S )zEncrypts files.)r rd )r8 r: r' r% r+ r+ r, r< � s z"EncryptOperation.perform_operationN)rR rS rT rU rI r rV rW rB r[ r r; r\ r] r< r+ r+ r+ r, rc � s rc c @ sF e Zd ZdZdZdZdZdZdZe j
Z f fe e j
jee d�dd �Zd
S ) �DecryptOperationzDecrypts a file.�privatezgDecrypts a file. The original file must be shorter than the key length in order to have been encrypted.r1 r2 r3 )r: r( r% r c C s t j||�S )zDecrypts files.)r r1 )r8 r: r( r% r+ r+ r, r< � s z"DecryptOperation.perform_operationN)rR rS rT rU rI r rV rW rB r �
PrivateKeyrN r[ r; r\ r] r< r+ r+ r+ r, re � s re c @ sX e Zd ZdZdZdZddje� ZdZ dZ
dZ e j
Zd Zd
Zee jjeed �d d
�ZdS )�
SignOperationz
Signs a file.rf z/usage: %%prog [options] private_key hash_methodzCSigns a file, outputs the signature. Choose the hash method from %sz, �sign� signatureZSigning� z\Name of the file to write the signature to. Written to stdout if this option is not present.)r: r( r% r c C s0 |d }|t kr"tddjt � ��tj|||�S )z Signs files.r z%Invalid hash method, choose one of %sz, )� HASH_METHODSr �joinr ri )r8 r: r( r% Z hash_methodr+ r+ r, r< s z SignOperation.perform_operationN)rR rS rT rU rI r rm rl r rV rW rB r rg rN rJ r7 r[ r; r\ r] r< r+ r+ r+ r, rh � s
rh c @ sN e Zd ZdZdZdZdZdZdZdZ e
j Z dZ
d Zee
jjed
d �d d
�Zd
S )�VerifyOperationzVerify a signature.r/ z1usage: %%prog [options] public_key signature_filezsVerifies a signature, exits with status 0 upon success, prints an error message and exits with status 1 upon error.�verifyZverifiedZ Verifyingrk FN)r: r' r% r c C sz |d }t |d��}|j� }W dQ R X ytj|||� W n. tjk
rf } ztd�|�W Y dd}~X nX tdtjd� dS )zVerifies files.r rL NzVerification failed.zVerification OK)r
) r rM r ro ZVerificationErrorr r r r )r8 r: r' r% Zsignature_fileZsigfilerj r& r+ r+ r, r< + s z!VerifyOperation.perform_operation)rR rS rT rU rI r r rV rW rB r rX rN rJ rD r[ r; r\ r] r< r+ r+ r+ r, rn s
rn ) rU rY r r^ r r Zrsa.keyZ rsa.pkcs1�sortedZpkcs1rl �keysZUnionr_ r` ra r] r- �ABCMetar. rc re rh rn rd r1 ri ro r+ r+ r+ r, �<module> s& Cn!#