Adding Simple Pointer, GOP, SimpleTextInEx, and Networking protocols to the emulator...

[mirror_edk2.git] / UnixPkg / CpuRuntimeDxe / Cpu.c

diff --git a/UnixPkg/CpuRuntimeDxe/Cpu.c b/UnixPkg/CpuRuntimeDxe/Cpu.c
index 0cc913abfb1247c7695c3f197b5131dae11d47ee..d4926c3707b5b884d9f1812e6791fcf5cf339fa2 100644 (file)
--- a/UnixPkg/CpuRuntimeDxe/Cpu.c
+++ b/UnixPkg/CpuRuntimeDxe/Cpu.c
@@ -28,6 +28,7 @@ Abstract:
 #include <Guid/DataHubRecords.h>\r
 #include <Protocol/CpuIo2.h>\r
 #include <Protocol/FrameworkHii.h>\r
+#include <Protocol/UnixThunk.h>\r
 \r
 #include <Library/BaseLib.h>\r
 #include <Library/DebugLib.h>\r
@@ -37,6 +38,8 @@ Abstract:
 #include <Library/BaseMemoryLib.h>\r
 #include <Library/MemoryAllocationLib.h>\r
 #include <Library/UefiBootServicesTableLib.h>\r
+#include <Library/UnixLib.h>\r
+\r
 #include "CpuDriver.h"\r
 #include "UnixDxe.h"\r
 #include <Protocol/UnixIo.h>\r
@@ -151,6 +154,7 @@ Returns:
 \r
   Private                 = CPU_ARCH_PROTOCOL_PRIVATE_DATA_FROM_THIS (This);\r
   Private->InterruptState = TRUE;\r
+  gUnix->EnableInterrupt ();\r
   return EFI_SUCCESS;\r
 }\r
 \r
@@ -184,6 +188,7 @@ Returns:
 \r
   Private                 = CPU_ARCH_PROTOCOL_PRIVATE_DATA_FROM_THIS (This);\r
   Private->InterruptState = FALSE;\r
+  gUnix->DisableInterrupt ();\r
   return EFI_SUCCESS;\r
 }\r
 \r