o
    Ìn~b  ã                   @   s’   d Z ddlZddlmZ ddlmZmZmZmZ ddl	m
Z
mZmZmZ e
ddƒZdd	„ Zd
d„ Zdd„ ZdZdZdZdZdZdZdZdZdZdS )aÇ  
Module's constants for the modes of operation supported with Triple DES:

:var MODE_ECB: :ref:`Electronic Code Book (ECB) <ecb_mode>`
:var MODE_CBC: :ref:`Cipher-Block Chaining (CBC) <cbc_mode>`
:var MODE_CFB: :ref:`Cipher FeedBack (CFB) <cfb_mode>`
:var MODE_OFB: :ref:`Output FeedBack (OFB) <ofb_mode>`
:var MODE_CTR: :ref:`CounTer Mode (CTR) <ctr_mode>`
:var MODE_OPENPGP:  :ref:`OpenPGP Mode <openpgp_mode>`
:var MODE_EAX: :ref:`EAX Mode <eax_mode>`
é    N)Ú_create_cipher)Úbyte_stringÚbchrÚbordÚbstr)Úload_pycryptodome_raw_libÚVoidPointerÚSmartPointerÚc_size_tzCryptodome.Cipher._raw_des3aÀ  
                    int DES3_start_operation(const uint8_t key[],
                                             size_t key_len,
                                             void **pResult);
                    int DES3_encrypt(const void *state,
                                     const uint8_t *in,
                                     uint8_t *out,
                                     size_t data_len);
                    int DES3_decrypt(const void *state,
                                     const uint8_t *in,
                                     uint8_t *out,
                                     size_t data_len);
                    int DES3_stop_operation(void *state);
                    c                    sp   dd„ ‰ t | ƒtvrtdƒ‚d ‡ fdd„| D ƒ¡}|dd… |dd	… ks2|d
d… |dd… kr6tdƒ‚|S )am  Set the parity bits in a TDES key.

    :param key_in: the TDES key whose bits need to be adjusted
    :type key_in: byte string

    :returns: a copy of ``key_in``, with the parity bits correctly set
    :rtype: byte string

    :raises ValueError: if the TDES key is not 16 or 24 bytes long
    :raises ValueError: if the TDES key degenerates into Single DES
    c                 S   s0   d}t ddƒD ]
}|| |? d@ N }q| d@ |B S )Né   é   éþ   )Úrange)Zkey_byteZparityÚi© r   úA/usr/local/lib/python3.10/dist-packages/Cryptodome/Cipher/DES3.pyÚparity_byteI   s   z&adjust_key_parity.<locals>.parity_bytezNot a valid TDES keyó    c                    s   g | ]
}t ˆ t|ƒƒƒ‘qS r   )r   r   )Ú.0Úx©r   r   r   Ú
<listcomp>R   s    z%adjust_key_parity.<locals>.<listcomp>Nr   é   iðÿÿÿiøÿÿÿz(Triple DES key degenerates to single DES)ÚlenÚkey_sizeÚ
ValueErrorÚjoin)Úkey_inZkey_outr   r   r   Úadjust_key_parity<   s   0r   c                 C   sz   z|   d¡}W n ty   tdƒ‚w tt|ƒƒ}tj}tj}tƒ }||t	t
|ƒƒ| ¡ ƒ}|r6td| ƒ‚t| ¡ |ƒS )z}This method instantiates and returns a handle to a low-level base cipher.
    It will absorb named parameters in the process.ÚkeyzMissing 'key' parameterz,Error %X while instantiating the TDES cipher)ÚpopÚKeyErrorÚ	TypeErrorr   r   Ú_raw_des3_libZDES3_start_operationZDES3_stop_operationr   r
   r   Z
address_ofr   r	   Úget)Zdict_parametersr   r   Zstart_operationZstop_operationÚcipherÚresultr   r   r   Ú_create_base_cipherZ   s$   ÿ
þÿr'   c                 O   s    t tjt | |g|¢R i |¤ŽS )a	  Create a new Triple DES cipher.

    :param key:
        The secret key to use in the symmetric cipher.
        It must be 16 or 24 byte long. The parity bits will be ignored.
    :type key: bytes/bytearray/memoryview

    :param mode:
        The chaining mode to use for encryption or decryption.
    :type mode: One of the supported ``MODE_*`` constants

    :Keyword Arguments:
        *   **iv** (*bytes*, *bytearray*, *memoryview*) --
            (Only applicable for ``MODE_CBC``, ``MODE_CFB``, ``MODE_OFB``,
            and ``MODE_OPENPGP`` modes).

            The initialization vector to use for encryption or decryption.

            For ``MODE_CBC``, ``MODE_CFB``, and ``MODE_OFB`` it must be 8 bytes long.

            For ``MODE_OPENPGP`` mode only,
            it must be 8 bytes long for encryption
            and 10 bytes for decryption (in the latter case, it is
            actually the *encrypted* IV which was prefixed to the ciphertext).

            If not provided, a random byte string is generated (you must then
            read its value with the :attr:`iv` attribute).

        *   **nonce** (*bytes*, *bytearray*, *memoryview*) --
            (Only applicable for ``MODE_EAX`` and ``MODE_CTR``).

            A value that must never be reused for any other encryption done
            with this key.

            For ``MODE_EAX`` there are no
            restrictions on its length (recommended: **16** bytes).

            For ``MODE_CTR``, its length must be in the range **[0..7]**.

            If not provided for ``MODE_EAX``, a random byte string is generated (you
            can read it back via the ``nonce`` attribute).

        *   **segment_size** (*integer*) --
            (Only ``MODE_CFB``).The number of **bits** the plaintext and ciphertext
            are segmented in. It must be a multiple of 8.
            If not specified, it will be assumed to be 8.

        *   **mac_len** : (*integer*) --
            (Only ``MODE_EAX``)
            Length of the authentication tag, in bytes.
            It must be no longer than 8 (default).

        *   **initial_value** : (*integer*) --
            (Only ``MODE_CTR``). The initial value for the counter within
            the counter block. By default it is **0**.

    :Return: a Triple DES object, of the applicable mode.
    )r   ÚsysÚmodulesÚ__name__)r   ÚmodeÚargsÚkwargsr   r   r   Únewr   s    <r.   r   é   é   é   é   é   é	   r   )r   é   )Ú__doc__r(   ZCryptodome.Cipherr   ZCryptodome.Util.py3compatr   r   r   r   ZCryptodome.Util._raw_apir   r   r	   r
   r#   r   r'   r.   ZMODE_ECBZMODE_CBCZMODE_CFBZMODE_OFBZMODE_CTRZMODE_OPENPGPZMODE_EAXÚ
block_sizer   r   r   r   r   Ú<module>   s*   þ>