fix CTR mode not allowing NULL as IV

1 files changed, 3 insertions, 1 deletions

diff --git a/modes.cpp b/modes.cpp
index 789fafb2..46332284 100644
--- a/modes.cpp
+++ b/modes.cpp
@@ -56,7 +56,8 @@ void CFB_ModePolicy::TransformRegister()
 
 void CFB_ModePolicy::CipherResynchronize(const byte *iv, size_t length)
 {
-	memcpy_s(m_register, m_register.size(), iv, BlockSize());
+	assert(length == BlockSize());
+	CopyOrZero(m_register, iv, length);
 	TransformRegister();
 }
 
@@ -85,6 +86,7 @@ void OFB_ModePolicy::WriteKeystream(byte *keystreamBuffer, size_t iterationCount
 
 void OFB_ModePolicy::CipherResynchronize(byte *keystreamBuffer, const byte *iv, size_t length)
 {
+	assert(length == BlockSize());
 	CopyOrZero(m_register, iv, length);
 }