1 files changed, 74 insertions, 0 deletions
diff --git a/Library/CommonLib/EfiMem.c b/Library/CommonLib/EfiMem.c
new file mode 100644
index 0000000..d9386c0
--- /dev/null
+++ b/Library/CommonLib/EfiMem.c
@@ -0,0 +1,74 @@
+/** @file
+EFI memory helpers routines/wrappers
+
+Copyright (c) 2016. Disk Cryptography Services for EFI (DCS), Alex Kolotnikov
+Copyright (c) 2016. VeraCrypt, Mounir IDRASSI 
+
+This program and the accompanying materials are licensed and made available
+under the terms and conditions of the GNU Lesser General Public License, version 3.0 (LGPL-3.0).
+
+The full text of the license may be found at
+https://opensource.org/licenses/LGPL-3.0
+**/
+
+#include <Uefi.h>
+#include <Library/MemoryAllocationLib.h>
+#include <Library/UefiBootServicesTableLib.h>
+#include <Library/BaseMemoryLib.h>
+
+#include "Library/CommonLib.h"
+
+
+//////////////////////////////////////////////////////////////////////////
+// Memory procedures wrappers
+//////////////////////////////////////////////////////////////////////////
+
+VOID*
+MemAlloc(
+   IN UINTN size
+   ) {
+   return AllocateZeroPool(size);
+}
+
+VOID*
+MemRealloc(
+	IN UINTN  OldSize,
+	IN UINTN  NewSize,
+	IN VOID   *OldBuffer  OPTIONAL
+	) {
+	return ReallocatePool(OldSize, NewSize, OldBuffer);
+}
+
+VOID
+MemFree(
+   IN VOID* ptr
+   ) {
+	if(ptr != NULL) FreePool(ptr);
+}
+
+//////////////////////////////////////////////////////////////////////////
+// Memory mapped IO
+//////////////////////////////////////////////////////////////////////////
+
+EFI_STATUS
+PrepareMemory(
+   IN UINTN    address,
+   IN UINTN    len,
+   OUT VOID**  mem)
+{
+   EFI_STATUS              status;
+   EFI_PHYSICAL_ADDRESS    ptr;
+   VOID*                   buf;
+   UINTN                   pages;
+   pages = ((len & ~0x0FFF) + 0x1000) >> 12;
+   ptr = address & ~0x0FFF;
+//	OUT_PRINT(L"mem try %0x, %0x\n", pages, (UINTN)ptr);
+   status = gBS->AllocatePages(AllocateAddress, EfiMemoryMappedIO, pages, &ptr);
+   if (EFI_ERROR(status)) {
+      return status;
+   }
+   buf = (void*)(UINTN)ptr;
+   SetMem(buf, pages << 12, 0);
+   *mem = buf;
+   return status;
+}