o
    j                     @   s0   d dl mZ d dlmZmZ G dd deZdS )    )cipher)symmetricBASE
cipherMODEc                   @   s4   e Zd ZejdfddZdd Zdd Zdd	 ZdS )
TDESNc                 C   s   t | ||| d S N)r   __init__)selfkeymodeIV r   /root/aizidognhua/tmp/workspace/projects/ec89d86c-575f-41c9-af57-ac45cbdbf775/venv/lib/python3.10/site-packages/unicrypto/backends/mbedtls/TDES.pyr      s   zTDES.__init__c                 C   sV   | j tjkrtj| jtjd| _d S | j tj	kr'tj| jtj
| j| _d S td)N    zUnknown cipher mode!)r
   r   ECB
mbedcipherDES3newr	   MODE_ECB_cipherCBCMODE_CBCr   	Exception)r   r   r   r   setup_cipher	   s
   zTDES.setup_cipherc                 C      | j |S r   )r   encryptr   datar   r   r   r         zTDES.encryptc                 C   r   r   )r   decryptr   r   r   r   r      r   zTDES.decrypt)	__name__
__module____qualname__r   r   r   r   r   r   r   r   r   r   r      s
    r   N)mbedtlsr   r   unicrypto.symmetricr   r   r   r   r   r   r   <module>   s   