Use previous digest size as a default value in BLAKE2

1 files changed, 2 insertions, 2 deletions

diff --git a/blake2.cpp b/blake2.cpp
index a9bea136..c3d7b4b0 100644
--- a/blake2.cpp
+++ b/blake2.cpp
@@ -389,7 +389,7 @@ void BLAKE2s::UncheckedSetKey(const byte *key, unsigned int length, const Crypto
         m_keyLength = 0;

     }

 

-    m_digestSize = params.GetIntValueWithDefault(Name::DigestSize(), DIGESTSIZE);

+    m_digestSize = params.GetIntValueWithDefault(Name::DigestSize(), m_digestSize);

 

     m_state.Reset();

     m_block.Reset(m_digestSize, m_keyLength);

@@ -420,7 +420,7 @@ void BLAKE2b::UncheckedSetKey(const byte *key, unsigned int length, const Crypto
         m_keyLength = 0;

     }

 

-    m_digestSize = params.GetIntValueWithDefault(Name::DigestSize(), DIGESTSIZE);

+    m_digestSize = params.GetIntValueWithDefault(Name::DigestSize(), m_digestSize);

 

     m_state.Reset();

     m_block.Reset(m_digestSize, m_keyLength);