MdePkg/SmiHandlerProfile: Add Context support in Unregister

[mirror_edk2.git] / MdePkg / Library / BaseLib / AArch64 / DisableInterrupts.S

diff --git a/MdePkg/Library/BaseLib/AArch64/DisableInterrupts.S b/MdePkg/Library/BaseLib/AArch64/DisableInterrupts.S
index b80a7b41d7db8e568aad95f4cbcd60b91f1de6fa..9985f1cf68a2adbb234a6cb206f6cee3ea133260 100644 (file)
--- a/MdePkg/Library/BaseLib/AArch64/DisableInterrupts.S
+++ b/MdePkg/Library/BaseLib/AArch64/DisableInterrupts.S
@@ -17,7 +17,9 @@
 \r
 .text\r
 .p2align 2\r
-ASM_GLOBAL ASM_PFX(DisableInterrupts)\r
+GCC_ASM_EXPORT(DisableInterrupts)\r
+\r
+.set DAIF_WR_IRQ_BIT,   (1 << 1)\r
 \r
 #/**\r
 #  Disables CPU interrupts.\r
@@ -30,5 +32,5 @@ ASM_GLOBAL ASM_PFX(DisableInterrupts)
 #  );\r
 #\r
 ASM_PFX(DisableInterrupts):\r
-   msr  daifset, #2\r
+   msr  daifset, #DAIF_WR_IRQ_BIT\r
    ret\r