U ڲg'@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_pkcs1formopenwriteoutstdoutbuffer) parserclicli_argsZkeysizeexpub_keypriv_keydataoutfiler2+/tmp/pip-unpacked-wheel-1z1vrokf/rsa/cli.pykeygen!sZ r4c@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)selfr2r2r3__init__wszCryptoOperation.__init__)indatakeyr,rcCsdS)zPerforms the program's operation. Implement in a subclass. :returns: the data to write to the output. Nr2)r?rArBr,r2r2r3perform_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"rC has_output write_outfileoutput)r?r+r,rBrAoutdatar2r2r3__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=rKr>keynamerrrrexpected_cli_argsrr)r?r*r+r,r2r2r3rDszCryptoOperation.parse_cli)filenamerFrc CsDtd|j|ftjdt|d}|}W5QRX|j||S)zReads a public or private key.zReading %s key from %srrb)r!rPrr"r%read key_classZ load_pkcs1)r?rRrFkeyfileZkeydatar2r2r3rEs zCryptoOperation.read_key)innamerc CsV|rrQrKr# PublicKeyrUr@abcabstractmethodbytesrB AbstractKey IndexabletypingAnyrCrOTuplerValuesListstrrDrErGrLr2r2r2r3r5ds2    r5) metaclassc@s>eZdZdZdZdZdZdZdZd e e j j e e dd d Zd S) EncryptOperationzEncrypts a file.r6zWEncrypts a file. The file must be shorter than the key length in order to be encrypted.encryptZ encryptedZ encryptingr2rAr.r,rcCst|tjjstt||S)zEncrypts files.) isinstancer#rBraAssertionErrorro)r?rAr.r,r2r2r3rCsz"EncryptOperation.perform_operationN)r2)r[r\r]r^rPrr_r`rIrdr#rBrerfrCr2r2r2r3rnsrnc@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.r8r9r:r2rAr/r,rcCst|tjjstt||S)zDecrypts files.)rqr#rB PrivateKeyrrr8)r?rAr/r,r2r2r3rCsz"DecryptOperation.perform_operationN)r2)r[r\r]r^rPrr_r`rIr#rvrUrdrBrerfrCr2r2r2r3rssrsc@sXeZdZdZdZdZddeZdZ dZ dZ e j Zd Zd Zee jjeed d d ZdS) SignOperationz Signs a file.rtz/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.rucCsBt|tjjst|d}|tkr4tddtt|||S)z Signs files.rz%Invalid hash method, choose one of %srx) rqr#rBrvrr HASH_METHODSrjoinry)r?rAr/r,Z hash_methodr2r2r3rCs zSignOperation.perform_operationN)r[r\r]r^rPrr}r|rr_r`rIr#rvrUrQr>rdrBrerfrCr2r2r2r3rws&rwc@sNeZdZdZdZdZdZdZdZdZ e j Z dZ d Zee jjed d d d Zd S)VerifyOperationzVerify a signature.r6z1usage: %%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{FNrpc Cst|tjjst|d}t|d}|}W5QRXzt|||Wn.tjk rx}zt d|W5d}~XYnXt dt j ddS)zVerifies files.rrSzVerification failed.NzVerification OKr) rqr#rBrarrr%rTrZVerificationErrorrr!rr")r?rAr.r,Zsignature_fileZsigfilerzr-r2r2r3rC+s z!VerifyOperation.perform_operation)r[r\r]r^rPrrr_r`rIr#rarUrQrKrdrBrerfrCr2r2r2r3r~s r~)r^rbrrgrr#Zrsa.keyZ rsa.pkcs1sortedZpkcs1r|keysUnionrirkrlrfr4ABCMetar5rnrsrwr~ror8ryrr2r2r2r3s&Cn!#