File: //opt/alt/python37/lib/python3.7/site-packages/rsa/__pycache__/cli.cpython-37.pyc
B
��f�' � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZeej 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 d
dd� | �tjdd� �\}}t|�dkrv| �� td��yt |d �}W nJ t
k
r� } z,| �� t d|d tj d� td�|�W dd}~X Y nX t d| tj d� t
�|�\}}|j�r>t d|j tj d� |j|jd�}t|jd��}|�|� W dQ R X |j|jd�}|j�r�t d|j tj d� t|jd��}|�|� W dQ R X n t dtj d� tjj�|� 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�DERr )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)�optparse� OptionParser�
add_option�
parse_args�sys�argv�len�
print_help�
SystemExit�int�
ValueError�print�stderr�rsaZnewkeysZpuboutZ
save_pkcs1�form�open�write�out�stdout�buffer) �parser�cli�cli_argsZkeysize�ex�pub_key�priv_key�data�outfile� r, �8/opt/alt/python37/lib/python3.7/site-packages/rsa/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, )r9 r; r< r& r, r, r- �perform_operation| s z!CryptoOperation.perform_operationc C sd | � � \}}| �|d |j�}| �|j�}t| j�� tj d� | �
|||�}| j r`| � ||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)r9 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 d
d d � |� t
j d
d� �\}}t |�| j
kr�|�� 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 PEM)r r r )r r
r r N)r r r r r r7 rE r8 �keynamer r r r �expected_cli_argsr r )r9 r$ r% r& r, r, r- r> � s zCryptoOperation.parse_cli)�filenamer@ r c C sD t d| j|f tjd� t|d��}|�� }W dQ R X | j�||�S )z Reads a public or private key.zReading %s key from %s)r
�rbN)r rJ r r r �read� key_classZ
load_pkcs1)r9 rL r@ �keyfileZkeydatar, r, r- r? � s zCryptoOperation.read_key)�innamer c C sJ |r0t d| tjd� t|d��
}|�� S Q R X t dtjd� tjj�� S )zRead the input filezReading input from %s)r
rM NzReading input from stdin)r r r r rN �stdinr# )r9 rQ �infiler, r, r- rA � s zCryptoOperation.read_infile)rH �outnamer c C sX |r8t d| tjd� t|d��}|�|� W dQ R X n t dtjd� tjj�|� dS )zWrite the output filezWriting output to %s)r
r NzWriting output to stdout)r r r r r r" r# )r9 rH rT r+ r, r, r- rF � s z CryptoOperation.write_outfile)%�__name__�
__module__� __qualname__�__doc__rJ r r � operation�operation_pastrC r7 r8 rK rE r � PublicKeyrO r: �abc�abstractmethod�bytesr< � AbstractKey� Indexable�typing�Anyr= rI �Tupler �Values�List�strr> r? rA rF r, r, r, r- r/ d s*
r/ )� metaclassc @ s> e Zd ZdZdZdZdZdZdZd e e
j j e
e d�d d
�Zd S )
�EncryptOperationzEncrypts a file.r0 zWEncrypts a file. The file must be shorter than the key length in order to be encrypted.�encryptZ encryptedZ
encryptingr, )r; r( r& r c C s t |tjj�st�t�||�S )zEncrypts files.)�
isinstancer r< r[ �AssertionErrorri )r9 r; r( r& r, r, r- r= � s z"EncryptOperation.perform_operationN)r, )rU rV rW rX rJ r rY rZ rC r^ r r< r_ r` r= r, r, r, r- rh � s rh c @ sD e Zd ZdZdZdZdZdZdZe j
Z d e e j
jee d�d d
�Zd S )
�DecryptOperationzDecrypts a file.�privatezgDecrypts a file. The original file must be shorter than the key length in order to have been encrypted.r2 r3 r4 r, )r; r) r& r c C s t |tjj�st�t�||�S )zDecrypts files.)rj r r< �
PrivateKeyrk r2 )r9 r; r) r& r, r, r- r= � s z"DecryptOperation.perform_operationN)r, )rU rV rW rX rJ r rY rZ rC r rn rO r^ r< r_ r` r= r, r, r, r- rl � s rl c @ sX e Zd ZdZdZdZdd�e� ZdZ dZ
dZ e j
Zd Zd
Zee jjeed �d d
�ZdS )�
SignOperationz
Signs a file.rm 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 sB t |tjj�st�|d }|tkr4tdd�t� ��t�|||�S )z Signs files.r z%Invalid hash method, choose one of %sz, ) rj r r< rn rk � HASH_METHODSr �joinrp )r9 r; r) r& Z hash_methodr, r, r- r= s
z SignOperation.perform_operationN)rU rV rW rX rJ r rt rs r rY rZ rC r rn rO rK r8 r^ r< r_ r` r= r, r, r, r- ro � s
ro c @ sN e Zd ZdZdZdZdZdZdZdZ e
j Z dZ
d Zee
jjed
d �d d
�Zd
S )�VerifyOperationzVerify a signature.r0 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 Verifyingrr FN)r; r( r& r c
C s� t |tjj�st�|d }t|d��}|�� }W dQ R X yt�|||� W n. tjk
rx } zt d�|�W dd}~X Y nX t
dt j d� dS )zVerifies files.r rM NzVerification failed.zVerification OK)r
)
rj r r< r[ rk r rN rv ZVerificationErrorr r r r )r9 r; r( r& Zsignature_fileZsigfilerq r' r, r, r- r= + s z!VerifyOperation.perform_operation)rU rV rW rX rJ r r rY rZ rC r r[ rO rK rE r^ r< r_ r` r= r, r, r, r- ru s
ru ) rX r\ r ra r r Zrsa.keyZ rsa.pkcs1�sortedZpkcs1rs �keys�Unionrc re rf r` r. �ABCMetar/ rh rl ro ru ri r2 rp rv r, r, r, r- �<module> s& Cn!#