3
l�_J � @ s� d dl mZmZmZ d dlZd dlZd dlmZ ejej �G dd� de
��Z ejej �G dd� de
��Z ejej �G dd � d e
��Z
ejej �G d
d � d e
��Zejej �G d d
� d
e
��Zdd� Zdd� Zdd� Zeje �eje �G dd� de
���Zeje �eje
�G dd� de
���Zeje �G dd� de
��Zeje �eje �G dd� de
���Zeje �eje �G d d � d e
���Zeje �eje �G d d � d e
���Zeje �eje�G d d!� d!e
���Zeje �eje �eje�G d"d#� d#e
����ZdS )$� )�absolute_import�division�print_functionN)�utilsc @ s( e Zd Zejdd� �Zejdd� �ZdS )�Modec C s dS )z@
A string naming this mode (e.g. "ECB", "CBC").
N� )�selfr r � /usr/lib64/python3.6/modes.py�name s z Mode.namec C s dS )zq
Checks that all the necessary invariants of this (mode, algorithm)
combination are met.
Nr )r � algorithmr r r �validate_for_algorithm s zMode.validate_for_algorithmN)�__name__�
__module__� __qualname__�abc�abstractpropertyr
�abstractmethodr r r r r r s r c @ s e Zd Zejdd� �ZdS )� ModeWithInitializationVectorc C s dS )zP
The value of the initialization vector for this mode as bytes.
Nr )r r r r �initialization_vector s z2ModeWithInitializationVector.initialization_vectorN)r
r r r r r r r r r r s r c @ s e Zd Zejdd� �ZdS )�
ModeWithTweakc C s dS )z@
The value of the tweak for this mode as bytes.
Nr )r r r r �tweak) s zModeWithTweak.tweakN)r
r r r r r r r r r r ' s r c @ s e Zd Zejdd� �ZdS )�
ModeWithNoncec C s dS )z@
The value of the nonce for this mode as bytes.
Nr )r r r r �nonce2 s zModeWithNonce.nonceN)r
r r r r r r r r r r 0 s r c @ s e Zd Zejdd� �ZdS )�ModeWithAuthenticationTagc C s dS )zP
The value of the tag supplied to the constructor of this mode.
Nr )r r r r �tag; s z ModeWithAuthenticationTag.tagN)r
r r r r r r r r r r 9 s r c C s |j dkr |jdkr td��d S )N� ZAESz=Only 128, 192, and 256 bit keys are allowed for this AES mode)�key_sizer
�
ValueError)r r r r r �_check_aes_key_lengthB s r c C s0 t | j�d |jkr,tdjt | j�| j���d S )N� z Invalid IV size ({}) for {}.)�lenr �
block_sizer �formatr
)r r r r r �_check_iv_lengthI s r# c C s t | |� t| |� d S )N)r r# )r r r r r �_check_iv_and_key_lengthR s
r$ c @ s& e Zd Zd Zdd� Zejd�ZeZ dS )�CBCc C s t jd|� || _d S )Nr )r �_check_byteslike�_initialization_vector)r r r r r �__init__\ s z CBC.__init__r' N)
r
r r r
r( r �read_only_propertyr r$ r r r r r r% W s
r% c @ s* e Zd Zd Zdd� Zejd�Zdd� ZdS )�XTSc C s* t jd|� t|�dkr td��|| _d S )Nr � z!tweak must be 128-bits (16 bytes))r r& r r �_tweak)r r r r r r( i s z XTS.__init__r, c C s |j dkrtd��d S )Nr � z\The XTS specification requires a 256-bit key for AES-128-XTS and 512-bit key for AES-256-XTS)r r- )r r )r r r r r r s s
zXTS.validate_for_algorithmN) r
r r r
r( r r) r r r r r r r* d s
r* c @ s e Zd Zd ZeZdS )�ECBN)r
r r r
r r r r r r r. { s r. c @ s& e Zd Zd Zdd� Zejd�ZeZ dS )�OFBc C s t jd|� || _d S )Nr )r r& r' )r r r r r r( � s z OFB.__init__r' N)
r
r r r
r( r r) r r$ r r r r r r/ � s
r/ c @ s& e Zd Zd Zdd� Zejd�ZeZ dS )�CFBc C s t jd|� || _d S )Nr )r r& r' )r r r r r r( � s z CFB.__init__r' N)
r
r r r
r( r r) r r$ r r r r r r0 � s
r0 c @ s& e Zd Zd Zdd� Zejd�ZeZ dS )�CFB8c C s t jd|� || _d S )Nr )r r& r' )r r r r r r( � s z
CFB8.__init__r' N)
r
r r r
r( r r) r r$ r r r r r r1 � s
r1 c @ s* e Zd Zd Zdd� Zejd�Zdd� ZdS )�CTRc C s t jd|� || _d S )Nr )r r& �_nonce)r r r r r r( � s z CTR.__init__r3 c C s: t | |� t| j�d |jkr6tdjt| j�| j���d S )Nr z Invalid nonce size ({}) for {}.)r r r r! r r" r
)r r r r r r � s
zCTR.validate_for_algorithmN) r
r r r
r( r r) r r r r r r r2 � s
r2 c @ s> e Zd Zd ZdZdZddd �Zejd
�Z ejd �Z
d d
� Z dS )�GCM� �' r r �@ Nr+ c C st t jd|� t|�dkr td��|| _|d k rdt jd|� |dk rJtd��t|�|k rdtdj|���|| _|| _d S )Nr r z-initialization_vector must be at least 1 byter � zmin_tag_length must be >= 4z.Authentication tag must be {} bytes or longer.) r r& r r r' � _check_bytesr" �_tagZ_min_tag_length)r r r Zmin_tag_lengthr r r r( � s z GCM.__init__r: r' c C s t | |� d S )N)r )r r r r r r � s zGCM.validate_for_algorithml l ��l ��? l l )Nr+ ) r
r r r
Z_MAX_ENCRYPTED_BYTESZ_MAX_AAD_BYTESr( r r) r r r r r r r r4 � s
r4 ) Z
__future__r r r r ZsixZ cryptographyr Z
add_metaclass�ABCMeta�objectr r r r r r r# r$ Zregister_interfacer% r* r. r/ r0 r1 r2 r4 r r r r � sP