Add asserts to CFB_ModePolicy::SetFeedbackSize

weidai11 · Sep 30, 2023 · 8d3e357 · 8d3e357
1 parent 2e23f64
commit 8d3e357
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/modes.cpp b/modes.cpp
@@ -81,6 +81,8 @@ void CFB_ModePolicy::CipherResynchronize(const byte *iv, size_t length)
 
 void CFB_ModePolicy::SetFeedbackSize(unsigned int feedbackSize)
 {
+	CRYPTOPP_ASSERT(feedbackSize >= 0);
+	CRYPTOPP_ASSERT(feedbackSize <= BlockSize());
 	if (feedbackSize > BlockSize())
 		throw InvalidArgument("CFB_Mode: invalid feedback size");
 	m_feedbackSize = feedbackSize ? feedbackSize : BlockSize();