From b1f1d757664195253ad6dac743be2af3190a2fed Mon Sep 17 00:00:00 2001
From: J Wyman <jwyman@nvidia.com>
Date: Mon, 8 Dec 2025 18:12:27 -0500
Subject: [PATCH] fix: Correctly Guard List Access

This change adds missing mutex guards when accessing a shared list.
---
 src/sagemaker_server.cc | 11 +++++++----
 1 file changed, 7 insertions(+), 4 deletions(-)

diff --git a/src/sagemaker_server.cc b/src/sagemaker_server.cc
index 0e6dce5ce2..8922ff655e 100644
--- a/src/sagemaker_server.cc
+++ b/src/sagemaker_server.cc
@@ -182,10 +182,13 @@ SagemakerAPIServer::Handle(evhtp_request_t* req)
         if (action == "/invoke") {
           LOG_VERBOSE(1) << "SageMaker request: INVOKE MODEL";
 
-          if (sagemaker_models_list_.find(multi_model_name.c_str()) ==
-              sagemaker_models_list_.end()) {
-            evhtp_send_reply(req, EVHTP_RES_NOTFOUND); /* 404*/
-            return;
+          {
+            std::lock_guard<std::mutex> lock(models_list_mutex_);
+            if (sagemaker_models_list_.find(multi_model_name.c_str()) ==
+                sagemaker_models_list_.end()) {
+              evhtp_send_reply(req, EVHTP_RES_NOTFOUND); /* 404*/
+              return;
+            }
           }
           LOG_VERBOSE(1) << "SageMaker MME Custom Invoke Model Path";