diff options
author | David Reid <dreid@dreid.org> | 2013-11-12 16:41:13 -0800 |
---|---|---|
committer | David Reid <dreid@dreid.org> | 2013-11-12 16:41:13 -0800 |
commit | 152d6bec8c34c52d75373dd8d99a3d159baa9550 (patch) | |
tree | c8bee3c932c0173f6fa025ac0eb5e990fdf8cc51 /cryptography | |
parent | 4faa094526f49c83a75d21a6e796546d4d539c6c (diff) | |
download | cryptography-152d6bec8c34c52d75373dd8d99a3d159baa9550.tar.gz cryptography-152d6bec8c34c52d75373dd8d99a3d159baa9550.tar.bz2 cryptography-152d6bec8c34c52d75373dd8d99a3d159baa9550.zip |
raise an exception if you try to use a HashContext after finalize is called.
Diffstat (limited to 'cryptography')
-rw-r--r-- | cryptography/exceptions.py | 4 | ||||
-rw-r--r-- | cryptography/hazmat/primitives/hashes.py | 9 |
2 files changed, 12 insertions, 1 deletions
diff --git a/cryptography/exceptions.py b/cryptography/exceptions.py index 391bed82..c2e71493 100644 --- a/cryptography/exceptions.py +++ b/cryptography/exceptions.py @@ -14,3 +14,7 @@ class UnsupportedAlgorithm(Exception): pass + + +class AlreadyFinalized(Exception): + pass diff --git a/cryptography/hazmat/primitives/hashes.py b/cryptography/hazmat/primitives/hashes.py index 6ae622cd..f85d36a0 100644 --- a/cryptography/hazmat/primitives/hashes.py +++ b/cryptography/hazmat/primitives/hashes.py @@ -15,6 +15,7 @@ from __future__ import absolute_import, division, print_function import six +from cryptography import exceptions from cryptography.hazmat.primitives import interfaces @@ -37,17 +38,23 @@ class Hash(object): self._ctx = ctx def update(self, data): + if self._ctx is None: + raise exceptions.AlreadyFinalized() if isinstance(data, six.text_type): raise TypeError("Unicode-objects must be encoded before hashing") self._ctx.update(data) def copy(self): + if self._ctx is None: + raise exceptions.AlreadyFinalized() return Hash( self.algorithm, backend=self._backend, ctx=self._ctx.copy() ) def finalize(self): - return self._ctx.finalize() + digest = self._ctx.finalize() + self._ctx = None + return digest @interfaces.register(interfaces.HashAlgorithm) |