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
Output Feedback (CFB) mode.
ÚOfbModeé    )Ú_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_ofba  
                        int OFB_start_operation(void *cipher,
                                                const uint8_t iv[],
                                                size_t iv_len,
                                                void **pResult);
                        int OFB_encrypt(void *ofbState,
                                        const uint8_t *in,
                                        uint8_t *out,
                                        size_t data_len);
                        int OFB_decrypt(void *ofbState,
                                        const uint8_t *in,
                                        uint8_t *out,
                                        size_t data_len);
                        int OFB_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¼  *Output FeedBack (OFB)*.

    This mode is very similar to CBC, but it
    transforms the underlying block cipher into a stream cipher.

    The keystream is the iterated block encryption of the
    previous ciphertext block.

    An Initialization Vector (*IV*) is required.

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

    .. _`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 )aS  Create a new block cipher, configured in OFB 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 a nonce, to to be reused for any other
            message**. It shall be a nonce or a random value.

            Reusing the *IV* for encryptions performed with the same key
            compromises confidentiality.
        z)Error %d while instantiating the OFB modeNÚencryptÚdecrypt)r   Ú_stateÚraw_ofb_libZOFB_start_operationÚgetr
   r	   ÚlenZ
address_ofÚ
ValueErrorr   ZOFB_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_ofb.pyÚ__init__I   s$    

ýÿ
ÿ
zOfbMode.__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š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)

        This function does not add any padding to the plaintext.

        :Parameters:
          plaintext : bytes/bytearray/memoryview
            The piece of data to encrypt.
            It can be of any length.
        :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)z%Error %d while encrypting in OFB mode)r   Ú	TypeErrorr   r   r   r   r   ZOFB_encryptr   r   r
   r	   r   )r   Ú	plaintextÚoutputÚ
ciphertextr   r   r   r   r   y   s,    !
ÿ
ýzOfbMode.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š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.
            It can be of any length.
        :Keywords:
          output : bytearray/memoryview
            The location where the plaintext is 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   z%Error %d while decrypting in OFB mode)r   r    r   r   r   r   r   ZOFB_decryptr   r   r
   r	   r   )r   r#   r"   r!   r   r   r   r   r   ¶   s,    !
ÿ
ýzOfbMode.decrypt)N)N)Ú__name__Ú
__module__Ú__qualname__Ú__doc__r   r   r   r   r   r   r   r   7   s   0
=c                 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 OFB encryption/decryption.

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

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

      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 OFB: %s)	Z_create_base_cipherÚpopr   r   r    r   r   Ústrr   )ÚfactoryÚkwargsZcipher_stater   r   r   r   r   Ú_create_ofb_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   (
 >