vendor/edk2/edk2-stable202411

1 files changed, 40 insertions, 0 deletions

diff --git a/MdePkg/Library/StackCheckLib/StackCheckLibCommonMsvc.c b/MdePkg/Library/StackCheckLib/StackCheckLibCommonMsvc.c
new file mode 100644
index 000000000000..6018e3790990
--- /dev/null
+++ b/MdePkg/Library/StackCheckLib/StackCheckLibCommonMsvc.c
@@ -0,0 +1,40 @@
+/** @file
+  Provides the required functionality for handling stack
+  cookie check failures for MSVC.
+
+  Copyright (c) Microsoft Corporation.
+  SPDX-License-Identifier: BSD-2-Clause-Patent
+**/
+
+#include <Base.h>
+
+#include <Library/DebugLib.h>
+#include <Library/BaseLib.h>
+#include <Library/StackCheckFailureHookLib.h>
+
+/**
+  Triggers an interrupt using the vector specified by PcdStackCookieExceptionVector
+**/
+VOID
+TriggerStackCookieInterrupt (
+  VOID
+  );
+
+VOID  *__security_cookie = (VOID *)(UINTN)STACK_COOKIE_VALUE;
+
+/**
+  This function gets called when an MSVC generated stack cookie fails. This implementation calls into a platform
+  failure hook lib and then triggers the stack cookie interrupt.
+
+  @param[in] ActualCookieValue  The value that was written onto the stack, corrupting the stack cookie.
+
+**/
+VOID
+StackCheckFailure (
+  VOID  *ActualCookieValue
+  )
+{
+  DEBUG ((DEBUG_ERROR, "Stack cookie check failed at address 0x%llx!\n", RETURN_ADDRESS (0)));
+  StackCheckFailureHook (RETURN_ADDRESS (0));
+  TriggerStackCookieInterrupt ();
+}