o 9F9`&@sdZddlZddlZddlZddlZddlZddlZddlZeej j Z ej ej ejefZdddZGdddejdZGd d d eZGd d d eZGd ddeZGdddeZeZeZeZeZdS)zWCommandline scripts. These scripts are called by the executables defined in setup.py. NreturncCstjddd}|jdddd|jdd dd d|jd d d dd|tjdd\}}t|dkr;|tdzt |d}Wnt y]|t d|dtj dtdwt d|tj dt |\}}|jrt d|jtj d|j|jd}t|jd }||Wdn1swY|j|jd}|jrt d|jtj dt|jd}||WddS1swYdSt dtj dtjj|dS)zKey generator.zusage: %prog [options] keysizez.Generates a new RSA keypair 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_argsZkeysizepub_keypriv_keydataoutfiler/'lib/python3.10/site-packages/rsa/cli.pykeygen!sP     "r1c @seZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z ejZd$ddZejdedejjded ejfddZd$ddZd ejejeje ffddZ!de de d ejjfddZ"de d efddZ#d ed!e d d fd"d#Z$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.rTrNcCs4|j|jj|_|j|jj|_|j|jj|_dS)N)r __class____dict__ input_help output_help)selfr/r/r0__init__jszCryptoOperation.__init__indatakeyr*cCsdS)zPerforms the program's operation. Implement in a subclass. :returns: the data to write to the output. Nr/)r<r>r?r*r/r/r0perform_operationosz!CryptoOperation.perform_operationcCsh|\}}||d|j}||j}t|jtj d| |||}|j r2| ||j dSdS)zRuns the program.rrN) parse_cliread_keykeyform read_infileinputr operation_progressivetitlerr!r@ has_output write_outfileoutput)r<r)r*r?r>outdatar/r/r0__call__ys  zCryptoOperation.__call__cCstj|j|jd}|jddd|jd|jr |jddd|jd|jdd |jd d d | t j d d\}}t ||j krG|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 r rN)rrrrrr:rHr;keynamerrrrexpected_cli_argsrr)r<r(r)r*r/r/r0rAszCryptoOperation.parse_clifilenamerCcCsXtd|j|ftjdt|d }|}Wdn1s wY|j||S)zReads a public or private key.zReading %s key from %srrbN)r rMrr!r#read key_classZ load_pkcs1)r<rOrCZkeyfileZkeydatar/r/r0rBs   zCryptoOperation.read_keyinnamecCsb|r$td|tjdt|d }|WdS1swYtdtjdtjjS)zRead the input filezReading input from %srrPNzReading input from stdin)r rr!r#rQstdinr')r<rSZinfiler/r/r0rDs   zCryptoOperation.read_infilerKoutnamecCsp|r(td|tjdt|d}||WddS1s!wYdStdtjdtjj|dS)zWrite the output filezWriting output to %srrNzWriting output to stdout)r rr!r#r$r&r')r<rKrUr.r/r/r0rIs  "zCryptoOperation.write_outfilerN)%__name__ __module__ __qualname____doc__rMrr operationoperation_pastrFr:r;rNrHr" PublicKeyrRr=abcabstractmethodbytesr? AbstractKey IndexabletypingZAnyr@rLTuplerZValuesListstrrArBrDrIr/r/r/r0r2Xs2    r2) metaclassc @sFeZdZdZdZdZdZdZdZ dde d e j j d e d e fd d ZdS)EncryptOperationzEncrypts a file.r3zWEncrypts a file. The file must be shorter than the key length in order to be encrypted.encryptZ encryptedZ encryptingr/r>r+r*rcCt|tjjs Jt||S)zEncrypts files.) isinstancer"r?r]ri)r<r>r+r*r/r/r0r@ z"EncryptOperation.perform_operationNr/)rWrXrYrZrMrr[r\rFr`r"r?rarbr@r/r/r/r0rhsrhc @sLeZdZdZdZdZdZdZdZe j Z dde d e j jd ed e fd d ZdS)DecryptOperationzDecrypts a file.privatezgDecrypts a file. The original file must be shorter than the key length in order to have been encrypted.r5r6r7r/r>r,r*rcCrj)zDecrypts files.)rkr"r? PrivateKeyr5)r<r>r,r*r/r/r0r@rlz"DecryptOperation.perform_operationNrm)rWrXrYrZrMrr[r\rFr"rprRr`r?rarbr@r/r/r/r0rnsrnc@s^eZdZdZdZdZddeZdZ dZ dZ e j Zd Zd Zd ed e jjd edefddZdS) 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.r>r,r*rcCsBt|tjjs J|d}|tvrtddtt|||S)z Signs files.rz%Invalid hash method, choose one of %srr)rkr"r?rp HASH_METHODSrjoinrs)r<r>r,r*Z hash_methodr/r/r0r@szSignOperation.perform_operationN)rWrXrYrZrMrrwrvrr[r\rFr"rprRrNr;r`r?rarbr@r/r/r/r0rqs$rqc@sTeZdZdZdZdZdZdZdZdZ e j Z dZ d Zd ed e jjd ed dfddZdS)VerifyOperationzVerify a signature.r3z1usage: %%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 VerifyingruFr>r+r*rNcCst|tjjs J|d}t|d }|}Wdn1s!wYz t|||Wn tjy;tdwt dt j ddS)zVerifies files.rrPNzVerification failed.zVerification OKr) rkr"r?r]r#rQryZVerificationErrorrr rr!)r<r>r+r*Zsignature_fileZsigfilertr/r/r0r@s  z!VerifyOperation.perform_operation)rWrXrYrZrMrrr[r\rFr"r]rRrNrHr`r?rarbr@r/r/r/r0rxs rxrV)rZr^rrcrr"Zrsa.keyZ rsa.pkcs1sortedZpkcs1rvkeysZUnionrdrerfrbr1ABCMetar2rhrnrqrxrir5rsryr/r/r/r0s( 7i