reset watchdog in any session messages.

Signed-off-by: Qi Zhang <[email protected]>
LeeTroy · Feb 11, 2022 · 9e73271 · 9e73271
1 parent c7c4aaa
commit 9e73271
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 8 deletions.
diff --git a/library/spdm_responder_lib/libspdm_rsp_heartbeat.c b/library/spdm_responder_lib/libspdm_rsp_heartbeat.c
@@ -5,7 +5,6 @@
  **/
 
 #include "internal/libspdm_responder_lib.h"
-#include "hal/library/platform_lib.h"
 
 /**
  * Process the SPDM HEARTBEAT request and return the response.
@@ -35,7 +34,6 @@ return_status spdm_get_response_heartbeat(IN void *context,
     spdm_context_t *spdm_context;
     spdm_session_info_t *session_info;
     libspdm_session_state_t session_state;
-    bool result;
 
     spdm_context = context;
     spdm_request = request;
@@ -105,10 +103,5 @@ return_status spdm_get_response_heartbeat(IN void *context,
     spdm_response->header.param1 = 0;
     spdm_response->header.param2 = 0;
 
-    result = libspdm_reset_watchdog(spdm_context->last_spdm_request_session_id);
-    if (!result) {
-        return RETURN_DEVICE_ERROR;
-    }
-
     return RETURN_SUCCESS;
 }
diff --git a/library/spdm_responder_lib/libspdm_rsp_receive_send.c b/library/spdm_responder_lib/libspdm_rsp_receive_send.c
@@ -491,7 +491,11 @@ return_status libspdm_build_response(IN void *context, IN uint32_t *session_id,
             libspdm_free_session_id(spdm_context, *session_id);
             break;
         default:
-            /* No session state update needed */
+            /* reset watchdog in any session messages. */
+            result = libspdm_reset_watchdog(*session_id);
+            if (!result) {
+                return RETURN_DEVICE_ERROR;
+            }
             break;
         }
     } else {