U g@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|tkrtdt|t|dkr8tdt|tdd||_t|_tt |t t|t |t t||j }|rtdt |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_libZSalsa20_stream_initr rZ address_ofrgetZSalsa20_stream_destroy block_size)selfkeyrresultr=/tmp/pip-unpacked-wheel-_q8s9isk/Cryptodome/Cipher/Salsa20.py__init__6s,     zSalsa20Cipher.__init__NcCs|dkrtt|}n4|}t|s*tdt|t|krJtdt|t|jt |t |t t|}|rtd||dkrt |SdSdS)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 TypeErrorrrZSalsa20_stream_encryptrrr rr)r plaintextoutput ciphertextrrrrencryptSs(   zSalsa20Cipher.encryptc CsLz|j||dWStk rF}ztt|ddW5d}~XYnXdS)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``. )rencdecN)r!rstrreplace)rr rerrrdecryptys zSalsa20Cipher.decrypt)N)N)__name__ __module__ __qualname____doc__rr!r'rrrrr .s &r NcCs|dkrtd}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 )rrrrrnewsr,r) )N)ZCryptodome.Util.py3compatrZCryptodome.Util._raw_apirrrrrrr r ZCryptodome.Randomr rr r,rrrrrrs (  ^