X-Git-Url: http://git.bitcoin.ninja/index.cgi?a=blobdiff_plain;f=lightning%2Fsrc%2Futil%2Fchacha20poly1305rfc.rs;h=1dbd91e65e0488168f59d439cdd4565f8d56803f;hb=bbb590b5518eb6e76ddd642187163c487c67dd5c;hp=befe5d19f5f18f06454fd9140caf3fe026e9e8ec;hpb=daeb5a62914fef04d9d51e8d30ced9d6c1103b42;p=rust-lightning

diff --git a/lightning/src/util/chacha20poly1305rfc.rs b/lightning/src/util/chacha20poly1305rfc.rs
index befe5d19..1dbd91e6 100644
--- a/lightning/src/util/chacha20poly1305rfc.rs
+++ b/lightning/src/util/chacha20poly1305rfc.rs
@@ -286,10 +286,10 @@ mod fuzzy_chachapoly {
 
 		pub(super) fn encrypt_in_place(&mut self, _input_output: &mut [u8]) {
 			assert!(self.finished == false);
-			self.finished = true;
 		}
 
 		pub(super) fn finish_and_get_tag(&mut self, out_tag: &mut [u8]) {
+			assert!(self.finished == false);
 			out_tag.copy_from_slice(&self.tag);
 			self.finished = true;
 		}