1 files changed, 78 insertions, 0 deletions
diff --git a/MdePkg/Include/Library/StackCheckLib.h b/MdePkg/Include/Library/StackCheckLib.h
new file mode 100644
index 0000000000..5773caafa5
--- /dev/null
+++ b/MdePkg/Include/Library/StackCheckLib.h
@@ -0,0 +1,78 @@
+/** @file
+  This library provides stack cookie checking functions for symbols inserted by the compiler. This header
+  is not intended to be used directly by modules, but rather defines the expected interfaces to each supported
+  compiler, so that if the compiler interface is updated it is easier to track.
+
+  Copyright (c) Microsoft Corporation.
+  SPDX-License-Identifier: BSD-2-Clause-Patent
+**/
+
+#ifndef STACK_CHECK_LIB_H_
+#define STACK_CHECK_LIB_H_
+
+#include <Base.h>
+
+#if defined (__GNUC__) || defined (__clang__)
+
+// The __stack_chk_guard is a random value placed on the stack between the stack variables
+// and the return address so that continuously writing past the stack variables will cause
+// the stack cookie to be overwritten. Before the function returns, the stack cookie value
+// will be checked and if there is a mismatch then StackCheckLib handles the failure.
+extern VOID  *__stack_chk_guard;
+
+/**
+  Called when a stack cookie check fails. The return address is the failing address.
+
+**/
+VOID
+EFIAPI
+__stack_chk_fail (
+  VOID
+  );
+
+#elif defined (_MSC_VER)
+
+// The __security_cookie is a random value placed on the stack between the stack variables
+// and the return address so that continuously writing past the stack variables will cause
+// the stack cookie to be overwritten. Before the function returns, the stack cookie value
+// will be checked and if there is a mismatch then StackCheckLib handles the failure.
+extern VOID  *__security_cookie;
+
+/**
+  Called when a buffer check fails. This functionality is dependent on MSVC
+  C runtime libraries and so is unsupported in UEFI.
+
+**/
+VOID
+EFIAPI
+__report_rangecheckfailure (
+  VOID
+  );
+
+/**
+   The GS handler is for checking the stack cookie during SEH or
+   EH exceptions and is unsupported in UEFI.
+
+**/
+VOID
+EFIAPI
+__GSHandlerCheck (
+  VOID
+  );
+
+/**
+   Checks the stack cookie value against __security_cookie and calls the
+   stack cookie failure handler if there is a mismatch.
+
+   @param UINTN  CheckValue The value to check against __security_cookie
+
+**/
+VOID
+EFIAPI
+__security_check_cookie (
+  UINTN  CheckValue
+  );
+
+#endif // Compiler type
+
+#endif // STACK_CHECK_LIB_H_