U ʹh&@sdZddlZddlZddlZddlZddlZddlZddlZeej j Z ej ej ejefZddddZGdddejdZGd d d eZGd d d eZGd ddeZGdddeZeZeZeZeZdS)zWCommandline scripts. These scripts are called by the executables defined in setup.py. Nreturnc Cstjddd}|jdddd|jdd dd d|jd d d dd|tjdd\}}t|dkrv|tdzt |d}WnJt k r}z,|t d|dtj dtd|W5d}~XYnXt d|tj dt |\}}|jr>t d|jtj d|j|jd}t|jd}||W5QRX|j|jd}|jrt d|jtj dt|jd}||W5QRXnt dtj dtjj|dS)zKey generator.zusage: %prog [options] keysizez/Generates a new RSA key pair of "keysize" bits.usage descriptionz--puboutstringzOutput 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.typehelp-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 PEMPEMZDERr r choicesdefaultNrzNot a valid number: %sfilezGenerating %i-bit keyzWriting public key to %s)formatwbzWriting private key to %szWriting private key to stdout)optparse OptionParser add_option parse_argssysargvlen print_help SystemExitint ValueErrorprintstderrrsaZnewkeysZpuboutZ save_pkcs1Zformopenwriteoutstdoutbuffer) parserclicli_argsZkeysizeexpub_keypriv_keydataoutfiler1-./venv/lib/python3.8/site-packages/rsa/cli.pykeygen!sZ r3c@seZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z ejZd d ddZejeejjeejdddZd d ddZejejeje fd ddZ!e e ejjdddZ"e edddZ#ee d dddZ$d S) CryptoOperationz9CLI callable that operates with input, output, and a key.publicz'usage: %%prog [options] %(keyname)s_keydecrypt 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.rTNrcCs4|j|jj|_|j|jj|_|j|jj|_dS)N)r __class____dict__ input_help output_help)selfr1r1r2__init__wszCryptoOperation.__init__)indatakeyr+rcCsdS)zPerforms the program's operation. Implement in a subclass. :returns: the data to write to the output. Nr1)r>r@rAr+r1r1r2perform_operation|sz!CryptoOperation.perform_operationcCsd|\}}||d|j}||j}t|jtj d| |||}|j r`| ||j dS)zRuns the program.rrN) parse_cliread_keykeyform read_infileinputr!operation_progressivetitlerr"rB has_output write_outfileoutput)r>r*r+rAr@outdatar1r1r2__call__s  zCryptoOperation.__call__cCstj|j|jd}|jddd|jd|jr@|jddd|jd|jdd |jd d d | t j d d\}}t ||j kr|td ||fS)zFParse the CLI options :returns: (cli_opts, cli_args) rz-iz--inputrrr z--outputz --keyformz&Key format of the %s key - default PEMr r rrN)rrrrrr<rJr=keynamerrrrexpected_cli_argsrr)r>r)r*r+r1r1r2rCszCryptoOperation.parse_cli)filenamerErc CsDtd|j|ftjdt|d}|}W5QRX|j||S)zReads a public or private key.zReading %s key from %srrb)r!rOrr"r$read key_classZ load_pkcs1)r>rQrEZkeyfileZkeydatar1r1r2rDs zCryptoOperation.read_key)innamerc CsV|rrUZinfiler1r1r2rFs  zCryptoOperation.read_infile)rMoutnamerc CsX|r8td|tjdt|d}||W5QRXntdtjdtjj|dS)zWrite the output filezWriting output to %srrzWriting output to stdoutN)r!rr"r$r%r'r()r>rMrWr0r1r1r2rKs  zCryptoOperation.write_outfile)%__name__ __module__ __qualname____doc__rOrr operationoperation_pastrHr<r=rPrJr# PublicKeyrTr?abcabstractmethodbytesrA AbstractKey IndexabletypingZAnyrBrNTuplerZValuesListstrrCrDrFrKr1r1r1r2r4ds2    r4) metaclassc@s>eZdZdZdZdZdZdZdZd e e j j e e dd d Zd S) EncryptOperationzEncrypts a file.r5zWEncrypts a file. The file must be shorter than the key length in order to be encrypted.encryptZ encryptedZ encryptingr1r@r-r+rcCst|tjjstt||S)zEncrypts files.) isinstancer#rAr^AssertionErrorrj)r>r@r-r+r1r1r2rBsz"EncryptOperation.perform_operationN)r1)rXrYrZr[rOrr\r]rHrar#rArbrcrBr1r1r1r2risric@sDeZdZdZdZdZdZdZdZe j Z d e e j jee dd 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.r7r8r9r1r@r.r+rcCst|tjjstt||S)zDecrypts files.)rlr#rA PrivateKeyrmr7)r>r@r.r+r1r1r2rBsz"DecryptOperation.perform_operationN)r1)rXrYrZr[rOrr\r]rHr#rqrTrarArbrcrBr1r1r1r2rnsrnc@sXeZdZdZdZdZddeZdZ dZ dZ e j Zd Zd Zee jjeed d d ZdS) SignOperationz Signs a file.roz/usage: %%prog [options] private_key hash_methodzCSigns a file, outputs the signature. Choose the hash method from %s, sign signatureZSigningz\Name of the file to write the signature to. Written to stdout if this option is not present.rpcCsBt|tjjst|d}|tkr4tddtt|||S)z Signs files.rz%Invalid hash method, choose one of %srs) rlr#rArqrm HASH_METHODSrjoinrt)r>r@r.r+Z hash_methodr1r1r2rBs zSignOperation.perform_operationN)rXrYrZr[rOrrxrwrr\r]rHr#rqrTrPr=rarArbrcrBr1r1r1r2rrs&rrc@sNeZdZdZdZdZdZdZdZdZ e j Z dZ d Zee jjed d d d Zd S)VerifyOperationzVerify a signature.r5z1usage: %%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 VerifyingrvFNrkc Cst|tjjst|d}t|d}|}W5QRXzt|||Wn.tjk rx}zt d|W5d}~XYnXt dt j ddS)zVerifies files.rrRzVerification failed.NzVerification OKr) rlr#rAr^rmr$rSrzZVerificationErrorrr!rr")r>r@r-r+Zsignature_fileZsigfilerur,r1r1r2rB+s z!VerifyOperation.perform_operation)rXrYrZr[rOrrr\r]rHr#r^rTrPrJrarArbrcrBr1r1r1r2rys ry)r[r_rrdrr#Zrsa.keyZ rsa.pkcs1sortedZpkcs1rwkeysZUnionrerfrgrcr3ABCMetar4rirnrrryrjr7rtrzr1r1r1r2s&Cn!#