Avoid memory overread in last block of CFB encryption.

1 files changed, 1 insertions, 1 deletions

diff --git a/src/lib/modes/cfb/cfb.cpp b/src/lib/modes/cfb/cfb.cpp
index 7721e1487..7b4aed3f0 100644
--- a/src/lib/modes/cfb/cfb.cpp
+++ b/src/lib/modes/cfb/cfb.cpp
@@ -99,7 +99,7 @@ void CFB_Encryption::update(secure_vector<byte>& buffer, size_t offset)
 
       // Assumes feedback-sized block except for last input
       copy_mem(&state[0], &state[shift], BS - shift);
-      copy_mem(&state[BS-shift], &buf[0], shift);
+      copy_mem(&state[BS-shift], &buf[0], took);
       cipher().encrypt(state, keystream_buf());
 
       buf += took;