File: //opt/cppython/lib/python3.8/site-packages/rsa/__pycache__/cli.cpython-38.pyc
U
JD�g�' � @ 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��zt |d �}W nJ t
k
r� } z,| �� t d|d tj d� td�|�W 5 d}~X Y nX t d| tj d� t
�|�\}}|j�r>t d|j tj d� |j|jd�}t|jd��}|�|� W 5 Q R X |j|jd�}|j�r�t d|j tj d� t|jd��}|�|� W 5 Q 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�help�-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��PEMZDERr
�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� r2 �4/opt/cppython/lib/python3.8/site-packages/rsa/cli.py�keygen! sZ ���� r4 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 TNr c C s4 | j | jj | _ | j| jj | _| j| jj | _d S )N)r � __class__�__dict__�
input_help� output_help)�selfr2 r2 r3 �__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.
Nr2 )r? rA rB r, r2 r2 r3 �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" rC �
has_output�
write_outfile�output)r? r+ r, rB rA �outdatar2 r2 r3 �__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 z-iz--inputr r r z--outputz --keyformz&Key format of the %s key - default PEMr r
r r N)r r r r r r= rK r> �keynamer r r r �expected_cli_argsr r )r? r* r+ r, r2 r2 r3 rD � s �zCryptoOperation.parse_cli)�filenamerF r c C sD t d| j|f tjd� t|d��}|�� }W 5 Q R X | j�||�S )z Reads a public or private key.zReading %s key from %sr �rb)r! rP r r" r% �read� key_classZ
load_pkcs1)r? rR rF �keyfileZkeydatar2 r2 r3 rE � s zCryptoOperation.read_key)�innamer c
C sV |r<t d| tjd� t|d��}|�� W 5 Q R � S Q R X t dtjd� tjj�� S )zRead the input filezReading input from %sr rS zReading input from stdin)r! r r" r% rT �stdinr) )r? rW �infiler2 r2 r3 rG � s zCryptoOperation.read_infile)rN �outnamer c C sX |r8t d| tjd� t|d��}|�|� W 5 Q R X n t dtjd� tjj�|� dS )zWrite the output filezWriting output to %sr r zWriting output to stdoutN)r! r r" r% r& r( r) )r? rN rZ r1 r2 r2 r3 rL � s z CryptoOperation.write_outfile)%�__name__�
__module__� __qualname__�__doc__rP r r � operation�operation_pastrI r= r> rQ rK r# � PublicKeyrU r@ �abc�abstractmethod�bytesrB � AbstractKey� Indexable�typing�AnyrC rO �Tupler �Values�List�strrD rE rG rL r2 r2 r2 r3 r5 d s2 � �
r5 )� metaclassc @ s> e Zd ZdZdZdZdZdZdZd e e
j j e
e d�d d
�Zd S )
�EncryptOperationzEncrypts a file.r6 zWEncrypts a file. The file must be shorter than the key length in order to be encrypted.�encryptZ encryptedZ
encryptingr2 �rA r. r, r c C s t |tjj�st�t�||�S )zEncrypts files.)�
isinstancer# rB ra �AssertionErrorro )r? rA r. r, r2 r2 r3 rC � s z"EncryptOperation.perform_operationN)r2 )r[ r\ r] r^ rP r r_ r` rI rd r# rB re rf rC r2 r2 r2 r3 rn � s � � �rn 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.r8 r9 r: r2 �rA r/ r, r c C s t |tjj�st�t�||�S )zDecrypts files.)rq r# rB �
PrivateKeyrr r8 )r? rA r/ r, r2 r2 r3 rC � s z"DecryptOperation.perform_operationN)r2 )r[ r\ r] r^ rP r r_ r` rI r# rv rU rd rB re rf rC r2 r2 r2 r3 rs � s � � �rs c @ sX e Zd ZdZdZdZdd�e� ZdZ dZ
dZ e j
Zd Zd
Zee jjeed �d d
�ZdS )�
SignOperationz
Signs a file.rt z/usage: %%prog [options] private_key hash_methodzCSigns a file, outputs the signature. Choose the hash method from %s�, �sign� signatureZSigning� z\Name of the file to write the signature to. Written to stdout if this option is not present.ru c C sB t |tjj�st�|d }|tkr4tdd�t� ��t�|||�S )z Signs files.r z%Invalid hash method, choose one of %srx ) rq r# rB rv rr � HASH_METHODSr �joinry )r? rA r/ r, Z hash_methodr2 r2 r3 rC s
z SignOperation.perform_operationN)r[ r\ r] r^ rP r r} r| r r_ r` rI r# rv rU rQ r> rd rB re rf rC r2 r2 r2 r3 rw � s& ��� �rw c @ sN e Zd ZdZdZdZdZdZdZdZ e
j Z dZ
d Zee
jjed
d �d d
�Zd
S )�VerifyOperationzVerify a signature.r6 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 Verifyingr{ FNrp c
C s� t |tjj�st�|d }t|d��}|�� }W 5 Q R X zt�|||� W n. tjk
rx } zt d�|�W 5 d}~X Y nX t
dt j d� dS )zVerifies files.r rS zVerification failed.NzVerification OKr )
rq r# rB ra rr r% rT r ZVerificationErrorr r! r r" )r? rA r. r, Zsignature_fileZsigfilerz r- r2 r2 r3 rC + s z!VerifyOperation.perform_operation)r[ r\ r] r^ rP r r r_ r` rI r# ra rU rQ rK rd rB re rf rC r2 r2 r2 r3 r~ s � �r~ ) r^ rb r rg r r# Zrsa.keyZ rsa.pkcs1�sortedZpkcs1r| �keys�Unionri rk rl rf r4 �ABCMetar5 rn rs rw r~ ro r8 ry r r2 r2 r2 r3 �<module> s& Cn!#