Fix some bugs in FSVariable and BdsPlatform for duet platform.

[mirror_edk2.git] / DuetPkg / FSVariable / MemStorage.c

/*++\r
\r
Copyright (c) 2006 - 2007, Intel Corporation\r
All rights reserved. This program and the accompanying materials\r
are licensed and made available under the terms and conditions of the BSD License\r
which accompanies this distribution.  The full text of the license may be found at\r
http://opensource.org/licenses/bsd-license.php\r
\r
THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
\r
Module Name:\r
\r
    MemStorage.c\r
\r
Abstract:\r
\r
    handles variable store/reads with emulated memory\r
\r
Revision History\r
\r
--*/\r
#include "FSVariable.h"\r
\r
STATIC\r
VOID\r
EFIAPI\r
OnVirtualAddressChange (\r
  IN EFI_EVENT            Event,\r
  IN VOID                 *Context\r
  );\r
\r
STATIC\r
EFI_STATUS\r
EFIAPI\r
MemEraseStore(\r
  IN VARIABLE_STORAGE   *This\r
  );\r
\r
STATIC\r
EFI_STATUS\r
EFIAPI\r
MemWriteStore (\r
  IN VARIABLE_STORAGE   *This,\r
  IN UINTN                Offset,\r
  IN UINTN                BufferSize,\r
  IN VOID                 *Buffer\r
  );\r
\r
EFI_STATUS\r
MemStorageConstructor (\r
  OUT VARIABLE_STORAGE          **VarStore,\r
  OUT EFI_EVENT_NOTIFY          *GoVirtualEvent,\r
  IN  UINTN                     Size\r
  )\r
{\r
  EFI_STATUS                  Status;\r
  VS_DEV                      *Dev;\r
\r
  Status = gBS->AllocatePool (EfiRuntimeServicesData, sizeof(VS_DEV), &Dev);\r
  ASSERT_EFI_ERROR (Status);\r
\r
  ZeroMem (Dev, sizeof(VS_DEV));\r
\r
  Dev->Signature   = VARIABLE_STORE_SIGNATURE;\r
  Dev->Size        = Size;\r
\r
  Dev->VarStore.Erase    = MemEraseStore;\r
  Dev->VarStore.Write    = MemWriteStore;\r
\r
  Status = gBS->AllocatePool (EfiRuntimeServicesData, Size, &VAR_DATA_PTR (Dev));\r
  ASSERT_EFI_ERROR (Status);\r
\r
  DEBUG ((EFI_D_ERROR, "VStorage: Size = 0x%x\n", Size));\r
  \r
  *VarStore       = &Dev->VarStore;\r
  *GoVirtualEvent = OnVirtualAddressChange;\r
\r
  return EFI_SUCCESS;\r
}\r
\r
STATIC\r
VOID\r
EFIAPI\r
OnVirtualAddressChange (\r
  IN EFI_EVENT            Event,\r
  IN VOID                 *Context\r
  )\r
{\r
  VS_DEV                  *Dev;\r
\r
  Dev = DEV_FROM_THIS (Context);\r
\r
  EfiConvertPointer (0, &VAR_DATA_PTR (Dev));\r
  EfiConvertPointer (0, (VOID**)&Dev->VarStore.Erase);\r
  EfiConvertPointer (0, (VOID**)&Dev->VarStore.Write);\r
}\r
\r
STATIC\r
EFI_STATUS\r
EFIAPI\r
MemEraseStore(\r
  IN VARIABLE_STORAGE   *This\r
  )\r
{\r
  VS_DEV              *Dev;\r
\r
  Dev = DEV_FROM_THIS(This);\r
  SetMem (VAR_DATA_PTR (Dev), Dev->Size, VAR_DEFAULT_VALUE);\r
  return EFI_SUCCESS;\r
}\r
\r
\r
STATIC\r
EFI_STATUS\r
EFIAPI\r
MemWriteStore (\r
  IN VARIABLE_STORAGE     *This,\r
  IN UINTN                Offset,\r
  IN UINTN                BufferSize,\r
  IN VOID                 *UserBuffer\r
  )\r
{\r
  VS_DEV              *Dev;\r
\r
  Dev = DEV_FROM_THIS(This);\r
\r
  ASSERT (Offset + BufferSize < Dev->Size);\r
\r
  // For better performance\r
  if (VAR_DATA_PTR (Dev) + Offset != UserBuffer) {\r
    CopyMem (VAR_DATA_PTR (Dev) + Offset, UserBuffer, BufferSize);\r
  }\r
  return EFI_SUCCESS;\r
}\r