From de380305376bd48f78e44cdaf77635ba6dc09ac8 Mon Sep 17 00:00:00 2001
From: George Arama <gearama@microsoft.com>
Date: Fri, 31 May 2024 08:12:27 -0700
Subject: [PATCH] dassa

---
 .../test/ut/mocked_transport_adapter_test.hpp  | 18 +++++++++++++-----
 1 file changed, 13 insertions(+), 5 deletions(-)

diff --git a/sdk/keyvault/azure-security-keyvault-keys/test/ut/mocked_transport_adapter_test.hpp b/sdk/keyvault/azure-security-keyvault-keys/test/ut/mocked_transport_adapter_test.hpp
index 4ef4ca134..0f1639997 100644
--- a/sdk/keyvault/azure-security-keyvault-keys/test/ut/mocked_transport_adapter_test.hpp
+++ b/sdk/keyvault/azure-security-keyvault-keys/test/ut/mocked_transport_adapter_test.hpp
@@ -61,8 +61,8 @@ namespace Azure { namespace Security { namespace KeyVault { namespace Keys { nam
 
     const char* UpdateFakeKey(const char fakeKey[], std::string header)
     {
-      std::string result;
-      std::string keyType;
+      char* result;
+      std::string keyType = "RSA";
       // cspell: disable-next-line
       if (header.find("CreateKeyRSAHSM") != std::string::npos)
       {
@@ -90,11 +90,19 @@ namespace Azure { namespace Security { namespace KeyVault { namespace Keys { nam
       {
         keyType = "oct";
       }
+      size_t bufferSize = std::string(fakeKey).size() + keyType.size();
+      result = new char[bufferSize];
 
-      result += fakeKey;
-      result += keyType.c_str();
+#if defined(_MSC_VER)
+#pragma warning(push)
+#pragma warning(disable : 4996)
+#endif
+      std::snprintf(result, bufferSize, fakeKey, keyType.c_str());
+#if defined(_MSC_VER)
+#pragma warning(pop)
+#endif
 
-      return result.c_str();
+      return result;
     }
   };