o
    �)%aõ  ã                   @   sÜ   d dl mZ d dlmZmZ d dlmZ dedefdd„ZG dd	„ d	eeƒZ	G d
d„ deeƒZ
G dd„ deeƒZG dd„ deeƒZG dd„ deeƒZG dd„ deƒZG dd„ deƒZG dd„ deeƒZG dd„ deeƒZdS )é    )Úutils)ÚBlockCipherAlgorithmÚCipherAlgorithm)ÚModeWithNonceÚ	algorithmÚkeyc                 C   s<   t  d|¡ t|ƒd | jvrtd t|ƒd | j¡ƒ‚|S )Nr   é   zInvalid key size ({}) for {}.)r   Ú_check_byteslikeÚlenÚ	key_sizesÚ
ValueErrorÚformatÚname)r   r   © r   úS/usr/lib/python3/dist-packages/cryptography/hazmat/primitives/ciphers/algorithms.pyÚ_verify_key_size   s   ÿÿr   c                   @   s@   e Zd Zd ZdZeg d¢ƒZdefdd„Ze	de
fdd„ƒZd	S )
ÚAESé€   )r   éÀ   é   i   r   c                 C   ó   t | |ƒ| _d S ©N©r   r   ©Úselfr   r   r   r   Ú__init__"   ó   zAES.__init__Úreturnc                 C   ó   t | jƒd S ©Nr   ©r
   r   ©r   r   r   r   Úkey_size%   ó   zAES.key_sizeN©Ú__name__Ú
__module__Ú__qualname__r   Ú
block_sizeÚ	frozensetr   Úbytesr   ÚpropertyÚintr"   r   r   r   r   r      s    r   c                   @   ó@   e Zd ZdZdZeg d¢ƒZdefdd„Ze	de
fdd	„ƒZd
S )ÚCamelliaÚcamelliar   )r   r   r   r   c                 C   r   r   r   r   r   r   r   r   /   r   zCamellia.__init__r   c                 C   r   r   r    r!   r   r   r   r"   2   r#   zCamellia.key_sizeNr$   r   r   r   r   r.   *   s    r.   c                   @   r-   )Ú	TripleDESÚ3DESé@   )r2   r   r   r   c                 C   sF   t |ƒdkr||| 7 }nt |ƒdkr||d d… 7 }t| |ƒ| _d S )Nr   é   )r
   r   r   r   r   r   r   r   <   s
   zTripleDES.__init__r   c                 C   r   r   r    r!   r   r   r   r"   C   r#   zTripleDES.key_sizeNr$   r   r   r   r   r0   7   s    r0   c                   @   óD   e Zd Zd ZdZeedddƒƒZdefdd„Z	e
defd	d
„ƒZdS )ÚBlowfishr2   é    iÁ  r   r   c                 C   r   r   r   r   r   r   r   r   M   r   zBlowfish.__init__r   c                 C   r   r   r    r!   r   r   r   r"   P   r#   zBlowfish.key_sizeN©r%   r&   r'   r   r(   r)   Úranger   r*   r   r+   r,   r"   r   r   r   r   r5   H   ó    r5   c                   @   r4   )ÚCAST5r2   é(   é�   r   r   c                 C   r   r   r   r   r   r   r   r   Z   r   zCAST5.__init__r   c                 C   r   r   r    r!   r   r   r   r"   ]   r#   zCAST5.key_sizeNr7   r   r   r   r   r:   U   r9   r:   c                   @   s<   e Zd ZdZeg d¢ƒZdefdd„Zede	fdd„ƒZ
d	S )
ÚARC4ÚRC4)r;   é8   r2   éP   r   é    r   r   r   c                 C   r   r   r   r   r   r   r   r   f   r   zARC4.__init__r   c                 C   r   r   r    r!   r   r   r   r"   i   r#   zARC4.key_sizeN)r%   r&   r'   r   r)   r   r*   r   r+   r,   r"   r   r   r   r   r=   b   s    r=   c                   @   s>   e Zd Zd ZdZedgƒZdefdd„Ze	de
fdd„ƒZd	S )
ÚIDEAr2   r   r   c                 C   r   r   r   r   r   r   r   r   s   r   zIDEA.__init__r   c                 C   r   r   r    r!   r   r   r   r"   v   r#   zIDEA.key_sizeNr$   r   r   r   r   rB   n   ó    
rB   c                   @   s>   e Zd Zd ZdZedgƒZdefdd„Ze	de
fdd„ƒZdS )	ÚSEEDr   r   c                 C   r   r   r   r   r   r   r   r   €   r   zSEED.__init__r   c                 C   r   r   r    r!   r   r   r   r"   ƒ   r#   zSEED.key_sizeNr$   r   r   r   r   rD   {   rC   rD   c                   @   sH   e Zd Zd ZedgƒZdedefdd„Ze 	d¡Z
edefdd	„ƒZd
S )ÚChaCha20r   r   Únoncec                 C   s6   t | |ƒ| _t d|¡ t|ƒdkrtdƒ‚|| _d S )NrF   r3   z!nonce must be 128-bits (16 bytes))r   r   r   r	   r
   r   Ú_nonce)r   r   rF   r   r   r   r   Œ   s
   
zChaCha20.__init__rG   r   c                 C   r   r   r    r!   r   r   r   r"   —   r#   zChaCha20.key_sizeN)r%   r&   r'   r   r)   r   r*   r   r   Úread_only_propertyrF   r+   r,   r"   r   r   r   r   rE   ˆ   s    

	rE   N)Úcryptographyr   Ú&cryptography.hazmat.primitives.ciphersr   r   Ú,cryptography.hazmat.primitives.ciphers.modesr   r*   r   r   r.   r0   r5   r:   r=   rB   rD   rE   r   r   r   r   Ú<module>   s   