diff options
Diffstat (limited to 'cryptography/primitives/block/modes.py')
-rw-r--r-- | cryptography/primitives/block/modes.py | 9 |
1 files changed, 4 insertions, 5 deletions
diff --git a/cryptography/primitives/block/modes.py b/cryptography/primitives/block/modes.py index e4fc886e..1e9b14b7 100644 --- a/cryptography/primitives/block/modes.py +++ b/cryptography/primitives/block/modes.py @@ -13,6 +13,8 @@ from __future__ import absolute_import, division, print_function +from cryptography.primitives.abc.block import modes + class CBC(object): name = "CBC" @@ -21,12 +23,9 @@ class CBC(object): super(CBC, self).__init__() self.initialization_vector = initialization_vector - def get_iv_or_nonce(self, api): - return self.initialization_vector - class ECB(object): name = "ECB" - def get_iv_or_nonce(self, api): - return api.get_iv_for_ecb() + +modes.ModeWithInitializationVector.register(CBC) |