a db/@sdZddlmZmZddlZddlZddlmZddlZddl Zddl Ze ej j Z ddZGdddeZGd d d eZGd d d eZGd ddeZGdddeZGdddeZGdddeZGdddeZeZeZeZeZeZeZdS)zWCommandline scripts. These scripts are called by the executables defined in setup.py. )with_statementprint_functionN) OptionParsercCstddd}|jdddd|jdd dd d|jd d d dd|tjdd\}}t|dkrt|tdzt|d}Wn8t y|t d|dtj dtdYn0t d|tj dt |\}}|jr@t d|jtj d|j|jd}t|jd}||Wdn1s60Y|j|jd}|jrt d|jtj dt|jd}||Wdn1s0Ynt dtj dtj|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 PEMPEMZDERrr choicesdefaultNrzNot a valid number: %sfilezGenerating %i-bit keyzWriting public key to %s)formatwbzWriting private key to %szWriting private key to stdout)r add_option parse_argssysargvlen print_help SystemExitint ValueErrorprintstderrrsaZnewkeysZpuboutZ save_pkcs1formopenwriteoutstdout)parserclicli_argsZkeysizepub_keypriv_keydataoutfiler/X/home/tom/ab/renpy-build/tmp/install.linux-x86_64/lib/python3.9/site-packages/rsa/cli.pykeygen#sJ   *,r1c@seZdZdZejZdZdZdZ dZ dZ dZ dZ d Zd Zd ZejZd d ZejdddZddZddZddZddZddZdS)CryptoOperationz9CLI callable that operates with input, output, and a key.publicz'usage: %%prog [options] %(keyname)s_keyNdecrypt 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.rTcCs4|j|jj|_|j|jj|_|j|jj|_dSN)r __class____dict__ input_help output_helpselfr/r/r0__init__nszCryptoOperation.__init__cCsdS)zPerforms the program's operation. Implement in a subclass. :returns: the data to write to the output. Nr/)r=indatakeyr*r/r/r0perform_operationssz!CryptoOperation.perform_operationcCsd|\}}||d|j}||j}t|jtj d| |||}|j r`| ||j dSzRuns the program.rrN) parse_cliread_keykeyform read_infileinputr operation_progressivetitlerr!rA has_output write_outfileoutput)r=r)r*r@r?outdatar/r/r0__call__|s  zCryptoOperation.__call__cCst|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--inputrr r z--outputz --keyformz&Key format of the %s key - default PEMr rrrN)rrrrr:rJr;keynamerrrrexpected_cli_argsrr)r=r(r)r*r/r/r0rCszCryptoOperation.parse_clicCsXtd|j|ftjdt|d}|}Wdn1s@0Y|j||S)zReads a public or private key.zReading %s key from %srrbN)r rOrr!r$read key_classZ load_pkcs1)r=filenamerEkeyfileZkeydatar/r/r0rDs &zCryptoOperation.read_keycCs`|rHtd|tjdt|d}|WdS1s>0YtdtjdtjS)zRead the input fileReading input from %srrQNReading input from stdin)r rr!r$rRstdin)r=innameinfiler/r/r0rFs  &zCryptoOperation.read_infilecCsj|rLtd|tjdt|d}||Wdqf1s@0Yntdtjdtj|dS)zWrite the output filezWriting output to %srrNzWriting output to stdout)r rr!r$r%r')r=rMoutnamer.r/r/r0rKs  *zCryptoOperation.write_outfile)N)__name__ __module__ __qualname____doc__abcABCMeta __metaclass__rOrr operationoperation_pastrHr:r;rPrJr" PublicKeyrSr>abstractmethodrArNrCrDrFrKr/r/r/r0r2Zs*   r2c@s.eZdZdZdZdZdZdZdZd dd Z dS) EncryptOperationzEncrypts a file.r3zEncrypts a file. The file must be shorter than the key length in order to be encrypted. For larger files, use the pyrsa-encrypt-bigfile command.encrypt encrypted encryptingNcCs t||S)zEncrypts files.)r"rh)r=r?r+r*r/r/r0rAsz"EncryptOperation.perform_operation)N r\r]r^r_rOrrcrdrHrAr/r/r/r0rgsrgc@s4eZdZdZdZdZdZdZdZe j Z d dd Z dS) DecryptOperationzDecrypts a file.privatezDecrypts a file. The original file must be shorter than the key length in order to have been encrypted. For larger files, use the pyrsa-decrypt-bigfile command.r4r5r6NcCs t||S)zDecrypts files.)r"r4)r=r?r,r*r/r/r0rAsz"DecryptOperation.perform_operation)N r\r]r^r_rOrrcrdrHr" PrivateKeyrSrAr/r/r/r0rlsrlc@sHeZdZdZdZdZddeZdZ dZ dZ e j Zd Zd Zd d Zd S) SignOperationz Signs a file.rmz/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.cCs0|d}|tvr"tddtt|||S)z Signs files.rz%Invalid hash method, choose one of %srq) HASH_METHODSrjoinr"rr)r=r?r,r*Z hash_methodr/r/r0rAs zSignOperation.perform_operationN)r\r]r^r_rOrrvrurrcrdrHr"rorSrPr;rAr/r/r/r0rpsrpc@s>eZdZdZdZdZdZdZdZdZ e j Z dZ d Zd d Zd S) 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 VerifyingrtFcCs||d}t|d}|}Wdn1s00Yzt|||WntjyhtdYn0tdtjddS)zVerifies files.rrQNzVerification failed.zVerification OKr) r$rRr"rxZVerificationErrorrr rr!)r=r?r+r*Zsignature_fileZsigfilersr/r/r0rAs &z!VerifyOperation.perform_operationN)r\r]r^r_rOrrrcrdrHr"rerSrPrJrAr/r/r/r0rwsrwc@s8eZdZdZddZddZddZdd Zd d Zd S) BigfileOperationz>CryptoOperation that doesn't read the entire file into memory.cCst|g|_dSr7)r2r> file_objectsr<r/r/r0r>#s zBigfileOperation.__init__cCs|jD] }|qdS)zCloses any open file handles.N)rzclose)r=fobjr/r/r0__del__(s zBigfileOperation.__del__cCs^|\}}||d|j}||j}||j}t|j t j d| ||||dSrB) rCrDrE get_infilerG get_outfilerLr rHrIrr!rA)r=r)r*r@rZr.r/r/r0rN.s    zBigfileOperation.__call__cCsF|r.td|tjdt|d}|j|ntdtjdtj}|S)zReturns the input file objectrVrrQrW)r rr!r$rzappendrX)r=rYr|r/r/r0r~=s zBigfileOperation.get_infilecCsF|r.td|tjdt|d}|j|ntdtjdtj}|S)zReturns the output file objectzWill write output to %srrzWill write output to stdout)r rr!r$rzrr')r=r[r|r/r/r0rJs zBigfileOperation.get_outfileN) r\r]r^r_r>r}rNr~rr/r/r/r0ry s  ryc@s.eZdZdZdZdZdZdZdZd dd Z dS) EncryptBigfileOperationz#Encrypts a file to VARBLOCK format.r3zEncrypts a file to an encrypted VARBLOCK file. The file can be larger than the key length, but the output file is only compatible with Python-RSA.rhrirjNcCstj|||S)zEncrypts files to VARBLOCK.)r"bigfileencrypt_bigfile)r=rZr.r+r*r/r/r0rAcsz)EncryptBigfileOperation.perform_operation)Nrkr/r/r/r0rXsrc@s4eZdZdZdZdZdZdZdZe j Z d dd Z dS) DecryptBigfileOperationz#Decrypts a file in VARBLOCK format.rmzQDecrypts an encrypted VARBLOCK file that was encrypted with pyrsa-encrypt-bigfiler4r5r6NcCstj|||S)zDecrypts a VARBLOCK file.)r"rdecrypt_bigfile)r=rZr.r,r*r/r/r0rAtsz)DecryptBigfileOperation.perform_operation)Nrnr/r/r/r0risr)r_ __future__rrr`roptparserr"Z rsa.bigfileZ rsa.pkcs1sortedZpkcs1rukeysr1objectr2rgrlrprwryrrrhr4rrrxrrr/r/r/r0s. 7j8