Fix incorrect order of encrypt/decrypt and nonce incremental

stratum-mining · Feb 2, 2024 · 7f61e7a · 7f61e7a
1 parent 5b26146
commit 7f61e7a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/protocols/v2/noise-sv2/src/cipher_state.rs b/protocols/v2/noise-sv2/src/cipher_state.rs
@@ -42,8 +42,8 @@ where
         ad: &[u8],
         data: &mut T,
     ) -> Result<(), aes_gcm::Error> {
-        self.set_n(self.get_n() + 1);
         let n = self.nonce_to_bytes();
+        self.set_n(self.get_n() + 1);
         if let Some(c) = self.get_cipher() {
             match c.encrypt(&n, ad, data) {
                 Ok(_) => Ok(()),
@@ -63,8 +63,8 @@ where
         ad: &[u8],
         data: &mut T,
     ) -> Result<(), aes_gcm::Error> {
-        self.set_n(self.get_n() + 1);
         let n = self.nonce_to_bytes();
+        self.set_n(self.get_n() + 1);
         if let Some(c) = self.get_cipher() {
             match c.decrypt(&n, ad, data) {
                 Ok(_) => Ok(()),