InOsEmuPkg/Unix/Sec: Add BaseMemoryLib implementation to thunk layer

[mirror_edk2.git] / InOsEmuPkg / Library / ThunkProtocolList / ThunkProtocolList.c

diff --git a/InOsEmuPkg/Library/ThunkProtocolList/ThunkProtocolList.c b/InOsEmuPkg/Library/ThunkProtocolList/ThunkProtocolList.c
index 3c53716765a237b1b56d6cd4f4c0e8a0b70b4a95..5aed594b6ed6f912d18a4583cc4c47d70a0a747a 100644 (file)
--- a/InOsEmuPkg/Library/ThunkProtocolList/ThunkProtocolList.c
+++ b/InOsEmuPkg/Library/ThunkProtocolList/ThunkProtocolList.c
@@ -2,6 +2,7 @@
   Emulator Thunk to abstract OS services from pure EFI code\r
 \r
   Copyright (c) 2008 - 2011, Apple Inc. All rights reserved.<BR>\r
   Emulator Thunk to abstract OS services from pure EFI code\r
 \r
   Copyright (c) 2008 - 2011, Apple Inc. All rights reserved.<BR>\r

+  Copyright (c) 2011, Intel Corporation. All rights reserved.<BR>\r

   \r
   This program and the accompanying materials\r
   are licensed and made available under the terms and conditions of the BSD License\r
   \r
   This program and the accompanying materials\r
   are licensed and made available under the terms and conditions of the BSD License\r


@@ -17,11 +18,10 @@
 #include <Library/BaseLib.h>\r
 #include <Library/BaseMemoryLib.h>\r
 #include <Library/DebugLib.h>\r
 #include <Library/BaseLib.h>\r
 #include <Library/BaseMemoryLib.h>\r
 #include <Library/DebugLib.h>\r

+#include <Library/MemoryAllocationLib.h>\r

 \r
 #include <Protocol/EmuIoThunk.h>\r
 \r
 \r
 #include <Protocol/EmuIoThunk.h>\r
 \r

-#include <stdlib.h>\r
-\r

 \r
 #define EMU_IO_THUNK_PROTOCOL_DATA_SIGNATURE SIGNATURE_32('E','m','u','T')\r
 \r
 \r
 #define EMU_IO_THUNK_PROTOCOL_DATA_SIGNATURE SIGNATURE_32('E','m','u','T')\r
 \r


@@ -53,7 +53,7 @@ AddThunkProtocol (
   }\r
   \r
   Instance = 0;\r
   }\r
   \r
   Instance = 0;\r

-  StartString = malloc (StrSize (ConfigString));\r
+  StartString = AllocatePool (StrSize (ConfigString));\r

   StrCpy (StartString, ConfigString);\r
   while (*StartString != '\0') {\r
 \r
   StrCpy (StartString, ConfigString);\r
   while (*StartString != '\0') {\r
 \r


@@ -74,7 +74,7 @@ AddThunkProtocol (
       SubString++;\r
     }\r
 \r
       SubString++;\r
     }\r
 \r

-    Private = malloc (sizeof (EMU_IO_THUNK_PROTOCOL_DATA));\r
+    Private = AllocatePool (sizeof (EMU_IO_THUNK_PROTOCOL_DATA));\r

     if (Private == NULL) {\r
       return EFI_OUT_OF_RESOURCES;\r
     }\r
     if (Private == NULL) {\r
       return EFI_OUT_OF_RESOURCES;\r
     }\r