3
ꅭh� � @ s| d dl mZmZmZ d dlmZ d dlmZmZm Z d dl
m Z d dl m
Z
d dlmZmZ ejej�G dd� de��Zd S )
� )�absolute_import�division�print_function)�utils)�AlreadyFinalized�UnsupportedAlgorithm�_Reasons)� _get_backend)� CMACBackend)�ciphers�macc @ s6 e Zd Zd dd�Zdd� Zdd� Zdd � Zd
d � ZdS )
�CMACNc C sb t |�}t|t�s tdtj��t|tj�s2td��|| _ || _
|d krX| j
j | j �| _ n|| _ d S )Nz.Backend object does not implement CMACBackend.z*Expected instance of BlockCipherAlgorithm.)
r �
isinstancer
r r ZBACKEND_MISSING_INTERFACEr ZBlockCipherAlgorithm� TypeError�
_algorithm�_backendZcreate_cmac_ctx�_ctx)�self� algorithm�backend�ctx� r � /usr/lib64/python3.6/cmac.py�__init__ s
z
CMAC.__init__c C s. | j d krtd��tjd|� | j j|� d S )Nz Context was already finalized.�data)r r r � _check_bytes�update)r r r r r r & s
z CMAC.updatec C s& | j d krtd��| j j� }d | _ |S )Nz Context was already finalized.)r r �finalize)r Zdigestr r r r - s
z
CMAC.finalizec C s: t jd|� | jd kr td��| jd }| _|j|� d S )N� signaturez Context was already finalized.)r r r r �verify)r r r r r r r 4 s
z CMAC.verifyc C s* | j d krtd��t| j| j| j j� d�S )Nz Context was already finalized.)r r )r r r
r r �copy)r r r r r < s
z CMAC.copy)NN)�__name__�
__module__� __qualname__r r r r r r r r r r
s
r
N)Z
__future__r r r Z cryptographyr Zcryptography.exceptionsr r r Z cryptography.hazmat.backendsr Z'cryptography.hazmat.backends.interfacesr
Z cryptography.hazmat.primitivesr r Zregister_interfaceZ
MACContext�objectr
r r r r � s