From dcc3f668c590bd0da898f331b968ed56d46936cf Mon Sep 17 00:00:00 2001 From: Alex Gaynor Date: Thu, 7 Nov 2013 14:28:16 -0800 Subject: Fixes for the module renaming --- cryptography/fernet.py | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/cryptography/fernet.py b/cryptography/fernet.py index 668958be..2a9f6a96 100644 --- a/cryptography/fernet.py +++ b/cryptography/fernet.py @@ -23,7 +23,7 @@ import six from cryptography.hazmat.primitives import padding, hashes from cryptography.hazmat.primitives.hmac import HMAC -from cryptography.hazmat.primitives.block import BlockCipher, ciphers, modes +from cryptography.hazmat.primitives.ciphers import Cipher, algorithms, modes class InvalidToken(Exception): @@ -71,10 +71,10 @@ class Fernet(object): "Unicode-objects must be encoded before encryption" ) - padder = padding.PKCS7(ciphers.AES.block_size).padder() + padder = padding.PKCS7(algorithms.AES.block_size).padder() padded_data = padder.update(data) + padder.finalize() - encryptor = BlockCipher( - ciphers.AES(self.encryption_key), modes.CBC(iv), self.backend + encryptor = Cipher( + algorithms.AES(self.encryption_key), modes.CBC(iv), self.backend ).encryptor() ciphertext = encryptor.update(padded_data) + encryptor.finalize() @@ -116,11 +116,11 @@ class Fernet(object): if not lib.constant_time_compare(hmac, len(hmac), data[-32:], 32): raise InvalidToken - decryptor = BlockCipher( - ciphers.AES(self.encryption_key), modes.CBC(iv), self.backend + decryptor = Cipher( + algorithms.AES(self.encryption_key), modes.CBC(iv), self.backend ).decryptor() plaintext_padded = decryptor.update(ciphertext) + decryptor.finalize() - unpadder = padding.PKCS7(ciphers.AES.block_size).unpadder() + unpadder = padding.PKCS7(algorithms.AES.block_size).unpadder() unpadded = unpadder.update(plaintext_padded) try: -- cgit v1.2.3