3
l�_o � @ s d dl mZmZmZ d dlZd dlZd dlmZ d dlm Z d dl
m Z ej ej
�G dd� de��Zdd � Zd
d � Zd d
� Zdd� Zdd� ZG dd� de�Zeje�G dd� de��Zeje�G dd� de��ZG dd� de�Zeje�G dd� de��Zeje�G d d � d e��ZdS ) � )�absolute_import�division�print_functionN)�utils)�AlreadyFinalized)�libc @ s( e Zd Zejdd� �Zejdd� �ZdS )�PaddingContextc C s dS )zR
Pads the provided bytes and returns any available data as bytes.
N� )�self�datar r � /usr/lib64/python3.6/padding.py�update s zPaddingContext.updatec C s dS )z6
Finalize the padding, returns bytes.
Nr )r
r r r �finalize s zPaddingContext.finalizeN)�__name__�
__module__� __qualname__�abc�abstractmethodr
r r r r r r s r c C s8 d| kodkn s t d��| d dkr4t d��d S )Nr i� z%block_size must be in range(0, 2041).� z#block_size must be a multiple of 8.)�
ValueError)�
block_sizer r r �_byte_padding_check s r c C sh | d krt d��tjd|� | t|�7 } t| �|d }| d ||d � }| ||d d � } | |fS )Nz Context was already finalized.r r )r r �_check_byteslike�bytes�len)�buffer_r r �finished_blocks�resultr r r �_byte_padding_update' s r c C s, | d krt d��|d t| � }| ||� S )Nz Context was already finalized.r )r r )r r Z paddingfn�pad_sizer r r �_byte_padding_pad7 s r c C sr | d krt d��tjd|� | t|�7 } tt| �|d d d�}| d ||d � }| ||d d � } | |fS )Nz Context was already finalized.r r � r )r r r r �maxr )r r r r r r r r �_byte_unpadding_update? s r# c C s\ | d krt d��t| �|d kr(td��|| |d �}|sBtd��tj| d�}| d | � S )Nz Context was already finalized.r zInvalid padding bytes.r! �����)r r r �sixZ
indexbytes)r r ZcheckfnZvalidr r r r �_byte_unpadding_checkO s r& c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�PKCS7c C s t |� || _d S )N)r r )r
r r r r �__init__` s zPKCS7.__init__c C s
t | j�S )N)�_PKCS7PaddingContextr )r
r r r �padderd s z PKCS7.padderc C s
t | j�S )N)�_PKCS7UnpaddingContextr )r
r r r �unpadderg s zPKCS7.unpadderN)r r r r( r* r, r r r r r' _ s r' c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
r) c C s || _ d| _d S )N� )r �_buffer)r
r r r r r( m s z _PKCS7PaddingContext.__init__c C s t | j|| j�\| _}|S )N)r r. r )r
r r r r r r
r s z_PKCS7PaddingContext.updatec C s t j|�| S )N)r% �int2byte)r
�sizer r r �_paddingx s z _PKCS7PaddingContext._paddingc C s t | j| j| j�}d | _|S )N)r r. r r1 )r
r r r r r { s z _PKCS7PaddingContext.finalizeN)r r r r( r
r1 r r r r r r) k s r) c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )r+ c C s || _ d| _d S )Nr- )r r. )r
r r r r r( � s z _PKCS7UnpaddingContext.__init__c C s t | j|| j�\| _}|S )N)r# r. r )r
r r r r r r
� s z _PKCS7UnpaddingContext.updatec C s t | j| jtj�}d | _|S )N)r& r. r r Z Cryptography_check_pkcs7_padding)r
r r r r r � s z _PKCS7UnpaddingContext.finalizeN)r r r r( r
r r r r r r+ � s r+ c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�ANSIX923c C s t |� || _d S )N)r r )r
r r r r r( � s zANSIX923.__init__c C s
t | j�S )N)�_ANSIX923PaddingContextr )r
r r r r* � s zANSIX923.padderc C s
t | j�S )N)�_ANSIX923UnpaddingContextr )r
r r r r, � s zANSIX923.unpadderN)r r r r( r* r, r r r r r2 � s r2 c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
r3 c C s || _ d| _d S )Nr- )r r. )r
r r r r r( � s z _ANSIX923PaddingContext.__init__c C s t | j|| j�\| _}|S )N)r r. r )r
r r r r r r
� s z _ANSIX923PaddingContext.updatec C s t jd�|d t j|� S )Nr r! )r% r/ )r
r0 r r r r1 � s z _ANSIX923PaddingContext._paddingc C s t | j| j| j�}d | _|S )N)r r. r r1 )r
r r r r r � s z _ANSIX923PaddingContext.finalizeN)r r r r( r
r1 r r r r r r3 � s r3 c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )r4 c C s || _ d| _d S )Nr- )r r. )r
r r r r r( � s z"_ANSIX923UnpaddingContext.__init__c C s t | j|| j�\| _}|S )N)r# r. r )r
r r r r r r
� s z _ANSIX923UnpaddingContext.updatec C s t | j| jtj�}d | _|S )N)r& r. r r Z#Cryptography_check_ansix923_padding)r
r r r r r � s z"_ANSIX923UnpaddingContext.finalizeN)r r r r( r
r r r r r r4 � s r4 ) Z
__future__r r r r r% Z cryptographyr Zcryptography.exceptionsr Z%cryptography.hazmat.bindings._paddingr Z
add_metaclass�ABCMeta�objectr r r r r# r& r' Zregister_interfacer) r+ r2 r3 r4 r r r r � s,