3 uQh&@sdZddlZddlZddlZddlZddlZddlZddlZeej j 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. N)returnc Cstjddd}|jdddd|jdd dd d|jd d dd d|jtjdd\}}t|dkrv|jtdyt |d}WnJt k r}z.|jt d|dtj dtd|WYdd}~XnXt d|tj dt j|\}}|jr>t d|jtj d|j|jd}t|jd}|j|WdQRX|j|jd}|jrt d|jtj dt|jd}|j|WdQRXnt dtj dtjjj|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.)typehelpz-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 PEMPEMDER)rchoicesdefaultNrzNot a valid number: %s)filezGenerating %i-bit keyzWriting public key to %s)formatwbzWriting private key to %szWriting private key to stdout)rr )optparse OptionParser add_option parse_argssysargvlen print_help SystemExitint ValueErrorprintstderrrsaZnewkeysZpuboutZ save_pkcs1formopenwriteoutstdoutbuffer) parserclicli_argsZkeysizeexpub_keypriv_keydataoutfiler,&/tmp/pip-build-2nz6shyl/rsa/rsa/cli.pykeygen!sR r.c@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.r TN)rcCs4|j|jj|_|j|jj|_|j|jj|_dS)N)r __class____dict__ input_help output_help)selfr,r,r-__init__wszCryptoOperation.__init__)indatakeyr&rcCsdS)zPerforms the program's operation. Implement in a subclass. :returns: the data to write to the output. Nr,)r9r;r<r&r,r,r-perform_operation|sz!CryptoOperation.perform_operationcCsd|j\}}|j|d|j}|j|j}t|jjtj d|j |||}|j r`|j ||j dS)zRuns the program.r)r N) parse_cliread_keykeyform read_infileinputroperation_progressivetitlerrr= has_output write_outfileoutput)r9r%r&r<r;outdatar,r,r-__call__s  zCryptoOperation.__call__cCstj|j|jd}|jddd|jd|jr@|jddd|jd|jdd |jdd d |j t j d d\}}t ||j kr|jtd ||fS)zFParse the CLI options :returns: (cli_opts, cli_args) )rrz-iz--inputr)rrz-oz--outputz --keyformz&Key format of the %s key - default PEMrr )rr r r N)rr )rrrrrr7rEr8keynamerrrrexpected_cli_argsrr)r9r$r%r&r,r,r-r>szCryptoOperation.parse_cli)filenamer@rc CsDtd|j|ftjdt|d}|j}WdQRX|jj||S)zReads a public or private key.zReading %s key from %s)r rbN)rrJrrrread key_classZ load_pkcs1)r9rLr@keyfileZkeydatar,r,r-r?s zCryptoOperation.read_key)innamerc CsJ|r0td|tjdt|d }|jSQRXtdtjdtjjjS)zRead the input filezReading input from %s)r rMNzReading input from stdin)rrrrrNstdinr#)r9rQinfiler,r,r-rAs  zCryptoOperation.read_infile)rHoutnamerc CsX|r8td|tjdt|d}|j|WdQRXntdtjdtjjj|dS)zWrite the output filezWriting output to %s)r rNzWriting output to stdout)rrrrr r"r#)r9rHrTr+r,r,r-rFs  zCryptoOperation.write_outfile)%__name__ __module__ __qualname____doc__rJrr operationoperation_pastrCr7r8rKrEr PublicKeyrOr:abcabstractmethodbytesr< AbstractKey IndexabletypingZAnyr=rITuplerValuesListstrr>r?rArFr,r,r,r-r/ds*    r/) metaclassc@s@eZdZdZdZdZdZdZdZffe e j j e e ddd Zd S) EncryptOperationzEncrypts a file.r0zWEncrypts a file. The file must be shorter than the key length in order to be encrypted.encryptZ encryptedZ encrypting)r;r(r&rcCst|tjjsttj||S)zEncrypts files.) isinstancerr<r[AssertionErrorrh)r9r;r(r&r,r,r-r=sz"EncryptOperation.perform_operationN)rUrVrWrXrJrrYrZrCr^rr<r_r`r=r,r,r,r-rgsrgc@sFeZdZdZdZdZdZdZdZe j Z ffe e j jee ddd Zd S) DecryptOperationzDecrypts a file.privatezgDecrypts a file. The original file must be shorter than the key length in order to have been encrypted.r2r3r4)r;r)r&rcCst|tjjsttj||S)zDecrypts files.)rirr< PrivateKeyrjr2)r9r;r)r&r,r,r-r=sz"DecryptOperation.perform_operationN)rUrVrWrXrJrrYrZrCrrmrOr^r<r_r`r=r,r,r,r-rksrkc@sXeZdZdZdZdZddjeZdZ dZ dZ e j Zd Zd Zee jjeed d d ZdS) SignOperationz Signs a file.rlz/usage: %%prog [options] private_key hash_methodzCSigns a file, outputs the signature. Choose the hash method from %sz, sign signatureZSigningz\Name of the file to write the signature to. Written to stdout if this option is not present.)r;r)r&rcCsBt|tjjst|d}|tkr4tddjttj|||S)z Signs files.r z%Invalid hash method, choose one of %sz, ) rirr<rmrj HASH_METHODSrjoinro)r9r;r)r&Z hash_methodr,r,r-r=s zSignOperation.perform_operationN)rUrVrWrXrJrrsrrrrYrZrCrrmrOrKr8r^r<r_r`r=r,r,r,r-rns  rnc@sNeZdZdZdZdZdZdZdZdZ e j Z dZ d Zee jjed d d d Zd S)VerifyOperationzVerify a signature.r0z1usage: %%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 VerifyingrqFN)r;r(r&rcCst|tjjst|d}t|d}|j}WdQRXytj|||Wn.tjk rx}zt d|WYdd}~XnXt dt j ddS)zVerifies files.r rMNzVerification failed.zVerification OK)r ) rirr<r[rjrrNruZVerificationErrorrrrr)r9r;r(r&Zsignature_fileZsigfilerpr'r,r,r-r=+s z!VerifyOperation.perform_operation)rUrVrWrXrJrrrYrZrCrr[rOrKrEr^r<r_r`r=r,r,r,r-rts rt)rXr\rrarrZrsa.keyZ rsa.pkcs1sortedZpkcs1rrkeysUnionrbrdrer`r.ABCMetar/rgrkrnrtrhr2rorur,r,r,r-s&Cn!#