fixed bug with chacha20 encryption of short messages

2018-12-03 13:36:17 -05:00
parent 8fc3a1f9c9
commit abc4f6c70b
2 changed files with 3 additions and 2 deletions
--- a/libi2pd/ChaCha20.cpp
+++ b/libi2pd/ChaCha20.cpp
@@ -111,7 +111,8 @@ void Chacha20Encrypt (Chacha20State& state, uint8_t * buf, size_t sz)
 			buf[i] ^= state.block.data[state.offset + i];
 		buf += s;
 		sz -= s;
-		state.offset = 0;
+		state.offset += s;
+		if (state.offset >= chacha::blocksize) state.offset = 0;	
 	}
 	for (size_t i = 0; i < sz; i += chacha::blocksize) 
 	{