3
l_o @ s d dl mZmZmZ d dlZd dlZd dlmZ d dlm Z d dl
m Z ej ej
G dd deZdd Zd
d Zd d
Zdd Zdd ZG dd deZejeG dd deZejeG dd deZG dd deZejeG dd deZejeG d d d eZdS ) )absolute_importdivisionprint_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 )selfdatar r /usr/lib64/python3.6/padding.pyupdate s zPaddingContext.updatec C s dS )z6
Finalize the padding, returns bytes.
Nr )r
r r r finalize s zPaddingContext.finalizeN)__name__
__module__ __qualname__abcabstractmethodr
r r r r r r s r c C s8 d| kodkn s t d| d dkr4t dd 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 dtjd| | t|7 } t| |d }| d ||d }| ||d d } | |fS )Nz Context was already finalized.r r )r r _check_byteslikebyteslen)buffer_r r finished_blocksresultr 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 paddingfnpad_sizer r r _byte_padding_pad7 s r c C sr | d krt dtjd| | 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 dt| |d kr(td|| |d }|sBtdtj| 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 | jS )N)_PKCS7PaddingContextr )r
r r r padderd s z PKCS7.padderc C s
t | jS )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 | jS )N)_ANSIX923PaddingContextr )r
r r r r* s zANSIX923.padderc C s
t | jS )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_metaclassABCMetaobjectr r r r r# r& r' Zregister_interfacer) r+ r2 r3 r4 r r r r s,