X-Git-Url: https://git.proxmox.com/?a=blobdiff_plain;ds=sidebyside;f=InOsEmuPkg%2FLibrary%2FThunkProtocolList%2FThunkProtocolList.c;h=5aed594b6ed6f912d18a4583cc4c47d70a0a747a;hb=b8800fc547440d8648e49c189a60547bfc4a1615;hp=3c53716765a237b1b56d6cd4f4c0e8a0b70b4a95;hpb=7af94c1d0af9b8595ef536c8a2f7ff930b54a43a;p=mirror_edk2.git

diff --git a/InOsEmuPkg/Library/ThunkProtocolList/ThunkProtocolList.c b/InOsEmuPkg/Library/ThunkProtocolList/ThunkProtocolList.c
index 3c53716765..5aed594b6e 100644
--- 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
 
   Copyright (c) 2008 - 2011, Apple Inc. All rights reserved.<BR>
+  Copyright (c) 2011, Intel Corporation. All rights reserved.<BR>
   
   This program and the accompanying materials
   are licensed and made available under the terms and conditions of the BSD License
@@ -17,11 +18,10 @@
 #include <Library/BaseLib.h>
 #include <Library/BaseMemoryLib.h>
 #include <Library/DebugLib.h>
+#include <Library/MemoryAllocationLib.h>
 
 #include <Protocol/EmuIoThunk.h>
 
-#include <stdlib.h>
-
 
 #define EMU_IO_THUNK_PROTOCOL_DATA_SIGNATURE SIGNATURE_32('E','m','u','T')
 
@@ -53,7 +53,7 @@ AddThunkProtocol (
   }
   
   Instance = 0;
-  StartString = malloc (StrSize (ConfigString));
+  StartString = AllocatePool (StrSize (ConfigString));
   StrCpy (StartString, ConfigString);
   while (*StartString != '\0') {
 
@@ -74,7 +74,7 @@ AddThunkProtocol (
       SubString++;
     }
 
-    Private = malloc (sizeof (EMU_IO_THUNK_PROTOCOL_DATA));
+    Private = AllocatePool (sizeof (EMU_IO_THUNK_PROTOCOL_DATA));
     if (Private == NULL) {
       return EFI_OUT_OF_RESOURCES;
     }