o j@snddlmZddlmZmZmZmZmZmZm Z m Z ddl m Z eddZ GdddZd d d Zd Zd ZdS)) _copy_bytes)load_pycryptodome_raw_libcreate_string_bufferget_raw_buffer VoidPointer SmartPointerc_size_t c_uint8_ptris_writeable_buffer)get_random_byteszCryptodome.Cipher._Salsa20a int Salsa20_stream_init(uint8_t *key, size_t keylen, uint8_t *nonce, size_t nonce_len, void **pSalsaState); int Salsa20_stream_destroy(void *salsaState); int Salsa20_stream_encrypt(void *salsaState, const uint8_t in[], uint8_t out[], size_t len); c@s,eZdZdZddZd ddZd ddZdS) Salsa20CipherzSalsa20 cipher object. Do not create it directly. Use :py:func:`new` instead. :var nonce: The nonce with length 8 :vartype nonce: byte string cCst|tvrtdt|t|dkrtdt|tdd||_t|_tt |t t|t |t t||j }|rEtdt |j tj|_d|_t||_dS)zQInitialize a Salsa20 cipher object See also `new()` at the module level.z+Incorrect key length for Salsa20 (%d bytes)z-Incorrect nonce length for Salsa20 (%d bytes)Nz'Error %d instantiating a Salsa20 cipher)lenkey_size ValueErrorrnoncer_state_raw_salsa20_libSalsa20_stream_initr r address_ofrgetSalsa20_stream_destroy block_size)selfkeyrresultr/root/aizidognhua/tmp/workspace/projects/ec89d86c-575f-41c9-af57-ac45cbdbf775/venv/lib/python3.10/site-packages/Cryptodome/Cipher/Salsa20.py__init__6s,     zSalsa20Cipher.__init__NcCs|dur tt|}n|}t|stdt|t|kr%tdt|t|jt |t |t t|}|r@td||durHt |SdS)aEncrypt a piece of data. Args: plaintext(bytes/bytearray/memoryview): The data to encrypt, of any size. Keyword Args: output(bytes/bytearray/memoryview): The location where the ciphertext is written to. If ``None``, the ciphertext is returned. Returns: If ``output`` is ``None``, the ciphertext is returned as ``bytes``. Otherwise, ``None``. Nz4output must be a bytearray or a writeable memoryviewz9output must have the same length as the input (%d bytes)z&Error %d while encrypting with Salsa20) rrr TypeErrorrrSalsa20_stream_encryptrrr rr)r plaintextoutput ciphertextrrrrencryptSs(   zSalsa20Cipher.encryptc Cs>z|j||dWSty}z tt|ddd}~ww)aDecrypt a piece of data. Args: ciphertext(bytes/bytearray/memoryview): The data to decrypt, of any size. Keyword Args: output(bytes/bytearray/memoryview): The location where the plaintext is written to. If ``None``, the plaintext is returned. Returns: If ``output`` is ``None``, the plaintext is returned as ``bytes``. Otherwise, ``None``. )r#encdecN)r%rstrreplace)rr$r#errrdecryptys  zSalsa20Cipher.decryptN)__name__ __module__ __qualname____doc__rr%r+rrrrr .s  &r NcCs|durtd}t||S)a)Create a new Salsa20 cipher :keyword key: The secret key to use. It must be 16 or 32 bytes long. :type key: bytes/bytearray/memoryview :keyword nonce: A value that must never be reused for any other encryption done with this key. It must be 8 bytes long. If not provided, a random byte string will be generated (you can read it back via the ``nonce`` attribute of the returned object). :type nonce: bytes/bytearray/memoryview :Return: a :class:`Cryptodome.Cipher.Salsa20.Salsa20Cipher` object Nr )r r )rrrrrnews r1r) r,)Cryptodome.Util.py3compatrCryptodome.Util._raw_apirrrrrrr r Cryptodome.Randomr rr r1rrrrrrs (  ^