increase sequence_number on receiving encrypted spdm message.

fix #178 Signed-off-by: Yang, Longlong <[email protected]>
intel · Dec 14, 2023 · 37b87e7 · 37b87e7
1 parent e14d45a
commit 37b87e7
Showing 1 changed file with 12 additions and 12 deletions.
diff --git a/spdmlib/src/common/session.rs b/spdmlib/src/common/session.rs
@@ -883,19 +883,19 @@ impl SpdmSession {
                         app_buffer,
                         &self.handshake_secret.request_direction,
                     );
-                    if r.is_ok() {
-                        self.handshake_secret.request_direction.sequence_number += 1
-                    };
+                    if r != Err(SPDM_STATUS_SEQUENCE_NUMBER_OVERFLOW) {
+                        self.handshake_secret.request_direction.sequence_number += 1;
+                    }
                     r
                 } else {
                     let r = self.decode_msg(
                         secured_buffer,
                         app_buffer,
                         &self.handshake_secret.response_direction,
                     );
-                    if r.is_ok() {
-                        self.handshake_secret.response_direction.sequence_number += 1
-                    };
+                    if r != Err(SPDM_STATUS_SEQUENCE_NUMBER_OVERFLOW) {
+                        self.handshake_secret.response_direction.sequence_number += 1;
+                    }
                     r
                 }
             }
@@ -906,19 +906,19 @@ impl SpdmSession {
                         app_buffer,
                         &self.application_secret.request_direction,
                     );
-                    if r.is_ok() {
-                        self.application_secret.request_direction.sequence_number += 1
-                    };
+                    if r != Err(SPDM_STATUS_SEQUENCE_NUMBER_OVERFLOW) {
+                        self.application_secret.request_direction.sequence_number += 1;
+                    }
                     r
                 } else {
                     let r = self.decode_msg(
                         secured_buffer,
                         app_buffer,
                         &self.application_secret.response_direction,
                     );
-                    if r.is_ok() {
-                        self.application_secret.response_direction.sequence_number += 1
-                    };
+                    if r != Err(SPDM_STATUS_SEQUENCE_NUMBER_OVERFLOW) {
+                        self.application_secret.response_direction.sequence_number += 1;
+                    }
                     r
                 }
             }