U
    ÈZjˆ*  ã                   @   sp   d Z dgZddlmZ ddlmZmZmZmZm	Z	m
Z
mZmZ ddlmZ eddƒZG dd„ deƒZd	d
„ ZdS )z'
Ciphertext Block Chaining (CBC) mode.
ÚCbcModeé    )Ú_copy_bytes)Úload_pycryptodome_raw_libÚVoidPointerÚcreate_string_bufferÚget_raw_bufferÚSmartPointerÚc_size_tÚc_uint8_ptrÚis_writeable_buffer)Úget_random_byteszCrypto.Cipher._raw_cbca®  
                int CBC_start_operation(void *cipher,
                                        const uint8_t iv[],
                                        size_t iv_len,
                                        void **pResult);
                int CBC_encrypt(void *cbcState,
                                const uint8_t *in,
                                uint8_t *out,
                                size_t data_len);
                int CBC_decrypt(void *cbcState,
                                const uint8_t *in,
                                uint8_t *out,
                                size_t data_len);
                int CBC_stop_operation(void *state);
                c                   @   s,   e Zd ZdZdd„ Zd	dd„Zd
dd„ZdS )r   a[  *Cipher-Block Chaining (CBC)*.

    Each of the ciphertext blocks depends on the current
    and all previous plaintext blocks.

    An Initialization Vector (*IV*) is required.

    See `NIST SP800-38A`_ , Section 6.2 .

    .. _`NIST SP800-38A` : http://csrc.nist.gov/publications/nistpubs/800-38a/sp800-38a.pdf

    :undocumented: __init__
    c                 C   sˆ   t ƒ | _t | ¡ t|ƒtt|ƒƒ| j ¡ ¡}|r>t	d| ƒ‚t
| j ¡ tjƒ| _| ¡  t|ƒ| _tdd|ƒ| _| j| _ddg| _dS )a  Create a new block cipher, configured in CBC mode.

        :Parameters:
          block_cipher : C pointer
            A smart pointer to the low-level block cipher instance.

          iv : bytes/bytearray/memoryview
            The initialization vector to use for encryption or decryption.
            It is as long as the cipher block.

            **The IV must be unpredictable**. Ideally it is picked randomly.

            Reusing the *IV* for encryptions performed with the same key
            compromises confidentiality.
        z)Error %d while instantiating the CBC modeNÚencryptÚdecrypt)r   Ú_stateÚraw_cbc_libZCBC_start_operationÚgetr
   r	   ÚlenZ
address_ofÚ
ValueErrorr   ZCBC_stop_operationÚreleaseÚ
block_sizer   ÚivÚIVÚ_next)ÚselfZblock_cipherr   Úresult© r   úZ/var/www/html/TRUCKING_PROJECT/venv/lib/python3.8/site-packages/Crypto/Cipher/_mode_cbc.pyÚ__init__N   s$    

ýÿ
ÿ
zCbcMode.__init__Nc                 C   sÈ   d| j krtdƒ‚dg| _ |dkr0tt|ƒƒ}n4|}t|ƒsDtdƒ‚t|ƒt|ƒkrdtdt|ƒ ƒ‚t | j 	¡ t
|ƒt
|ƒtt|ƒƒ¡}|r°|dkr¤td| j ƒ‚td| ƒ‚|dkrÀt|ƒS dS dS )	aŽ  Encrypt data with the key and the parameters set at initialization.

        A cipher object is stateful: once you have encrypted a message
        you cannot encrypt (or decrypt) another message using the same
        object.

        The data to encrypt can be broken up in two or
        more pieces and `encrypt` can be called multiple times.

        That is, the statement:

            >>> c.encrypt(a) + c.encrypt(b)

        is equivalent to:

             >>> c.encrypt(a+b)

        That also means that you cannot reuse an object for encrypting
        or decrypting other data with the same key.

        This function does not add any padding to the plaintext.

        :Parameters:
          plaintext : bytes/bytearray/memoryview
            The piece of data to encrypt.
            Its lenght must be multiple of the cipher block size.
        :Keywords:
          output : bytearray/memoryview
            The location where the ciphertext must be written to.
            If ``None``, the ciphertext is returned.
        :Return:
          If ``output`` is ``None``, the ciphertext is returned as ``bytes``.
          Otherwise, ``None``.
        r   z*encrypt() cannot be called after decrypt()Nú4output must be a bytearray or a writeable memoryviewú9output must have the same length as the input  (%d bytes)é   ú3Data must be padded to %d byte boundary in CBC modez%Error %d while encrypting in CBC mode)r   Ú	TypeErrorr   r   r   r   r   ZCBC_encryptr   r   r
   r	   r   r   )r   Ú	plaintextÚoutputÚ
ciphertextr   r   r   r   r   }   s0    $
ÿ
ýzCbcMode.encryptc                 C   sÈ   d| j krtdƒ‚dg| _ |dkr0tt|ƒƒ}n4|}t|ƒsDtdƒ‚t|ƒt|ƒkrdtdt|ƒ ƒ‚t | j 	¡ t
|ƒt
|ƒtt|ƒƒ¡}|r°|dkr¤td| j ƒ‚td| ƒ‚|dkrÀt|ƒS dS dS )	a  Decrypt data with the key and the parameters set at initialization.

        A cipher object is stateful: once you have decrypted a message
        you cannot decrypt (or encrypt) another message with the same
        object.

        The data to decrypt can be broken up in two or
        more pieces and `decrypt` can be called multiple times.

        That is, the statement:

            >>> c.decrypt(a) + c.decrypt(b)

        is equivalent to:

             >>> c.decrypt(a+b)

        This function does not remove any padding from the plaintext.

        :Parameters:
          ciphertext : bytes/bytearray/memoryview
            The piece of data to decrypt.
            Its length must be multiple of the cipher block size.
        :Keywords:
          output : bytearray/memoryview
            The location where the plaintext must be written to.
            If ``None``, the plaintext is returned.
        :Return:
          If ``output`` is ``None``, the plaintext is returned as ``bytes``.
          Otherwise, ``None``.
        r   z*decrypt() cannot be called after encrypt()Nr   r   r    r!   z%Error %d while decrypting in CBC mode)r   r"   r   r   r   r   r   ZCBC_decryptr   r   r
   r	   r   r   )r   r%   r$   r#   r   r   r   r   r   ¿   s0    !
ÿ
ýzCbcMode.decrypt)N)N)Ú__name__Ú
__module__Ú__qualname__Ú__doc__r   r   r   r   r   r   r   r   ?   s   /
Bc                 K   s�   |   |¡}| dd¡}| dd¡}d||fkr8t| jƒ}|dk	rR|dk	rVtdƒ‚n|}t|ƒ| jkrrtd| j ƒ‚|r†tdt|ƒ ƒ‚t||ƒS )aó  Instantiate a cipher object that performs CBC encryption/decryption.

    :Parameters:
      factory : module
        The underlying block cipher, a module from ``Crypto.Cipher``.

    :Keywords:
      iv : bytes/bytearray/memoryview
        The IV to use for CBC.

      IV : bytes/bytearray/memoryview
        Alias for ``iv``.

    Any other keyword will be passed to the underlying block cipher.
    See the relevant documentation for details (at least ``key`` will need
    to be present).
    r   Nr   )NNz*You must either use 'iv' or 'IV', not bothz.Incorrect IV length (it must be %d bytes long)zUnknown parameters for CBC: %s)	Z_create_base_cipherÚpopr   r   r"   r   r   Ústrr   )ÚfactoryÚkwargsZcipher_stater   r   r   r   r   Ú_create_cbc_cipherÿ   s     


ÿr.   N)r)   Ú__all__ZCrypto.Util.py3compatr   ZCrypto.Util._raw_apir   r   r   r   r   r	   r
   r   ZCrypto.Randomr   r   Úobjectr   r.   r   r   r   r   Ú<module>   s   (
 A