Merge pull request stratum-mining#729 from jakubtrnka/jakubt/handshak…

…e-fix Fix noise-handshake operations order
xyephy · Feb 2, 2024 · a8966e4 · a8966e4
2 parents cd3ab95 + 729cdec
commit a8966e4
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(()),