ArmPkg/BdsLib: Remove Linux loader from BdsLib

[mirror_edk2.git] / ArmPkg / Library / BdsLib / BdsHelper.c

/** @file\r
*\r
*  Copyright (c) 2011-2015, ARM Limited. All rights reserved.\r
*\r
*  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
**/\r
\r
#include "BdsInternal.h"\r
\r
EFI_STATUS\r
ShutdownUefiBootServices (\r
  VOID\r
  )\r
{\r
  EFI_STATUS              Status;\r
  UINTN                   MemoryMapSize;\r
  EFI_MEMORY_DESCRIPTOR   *MemoryMap;\r
  UINTN                   MapKey;\r
  UINTN                   DescriptorSize;\r
  UINT32                  DescriptorVersion;\r
  UINTN                   Pages;\r
\r
  MemoryMap = NULL;\r
  MemoryMapSize = 0;\r
  Pages = 0;\r
\r
  do {\r
    Status = gBS->GetMemoryMap (\r
                    &MemoryMapSize,\r
                    MemoryMap,\r
                    &MapKey,\r
                    &DescriptorSize,\r
                    &DescriptorVersion\r
                    );\r
    if (Status == EFI_BUFFER_TOO_SMALL) {\r
\r
      Pages = EFI_SIZE_TO_PAGES (MemoryMapSize) + 1;\r
      MemoryMap = AllocatePages (Pages);\r
\r
      //\r
      // Get System MemoryMap\r
      //\r
      Status = gBS->GetMemoryMap (\r
                      &MemoryMapSize,\r
                      MemoryMap,\r
                      &MapKey,\r
                      &DescriptorSize,\r
                      &DescriptorVersion\r
                      );\r
    }\r
\r
    // Don't do anything between the GetMemoryMap() and ExitBootServices()\r
    if (!EFI_ERROR(Status)) {\r
      Status = gBS->ExitBootServices (gImageHandle, MapKey);\r
      if (EFI_ERROR(Status)) {\r
        FreePages (MemoryMap, Pages);\r
        MemoryMap = NULL;\r
        MemoryMapSize = 0;\r
      }\r
    }\r
  } while (EFI_ERROR(Status));\r
\r
  return Status;\r
}\r
\r
/**\r
  Connect all DXE drivers\r
\r
  @retval EFI_SUCCESS           All drivers have been connected\r
  @retval EFI_NOT_FOUND         No handles match the search.\r
  @retval EFI_OUT_OF_RESOURCES  There is not resource pool memory to store the matching results.\r
\r
**/\r
EFI_STATUS\r
BdsConnectAllDrivers (\r
  VOID\r
  )\r
{\r
  UINTN                     HandleCount, Index;\r
  EFI_HANDLE                *HandleBuffer;\r
  EFI_STATUS                Status;\r
\r
  do {\r
    // Locate all the driver handles\r
    Status = gBS->LocateHandleBuffer (\r
                AllHandles,\r
                NULL,\r
                NULL,\r
                &HandleCount,\r
                &HandleBuffer\r
                );\r
    if (EFI_ERROR (Status)) {\r
      break;\r
    }\r
\r
    // Connect every handles\r
    for (Index = 0; Index < HandleCount; Index++) {\r
      gBS->ConnectController (HandleBuffer[Index], NULL, NULL, TRUE);\r
    }\r
\r
    if (HandleBuffer != NULL) {\r
      FreePool (HandleBuffer);\r
    }\r
\r
    // Check if new handles have been created after the start of the previous handles\r
    Status = gDS->Dispatch ();\r
  } while (!EFI_ERROR(Status));\r
\r
  return EFI_SUCCESS;\r
}\r
\r
EFI_STATUS\r
GetGlobalEnvironmentVariable (\r
  IN     CONST CHAR16*   VariableName,\r
  IN     VOID*           DefaultValue,\r
  IN OUT UINTN*          Size,\r
  OUT    VOID**          Value\r
  )\r
{\r
  return GetEnvironmentVariable (VariableName, &gEfiGlobalVariableGuid,\r
           DefaultValue, Size, Value);\r
}\r
\r
EFI_STATUS\r
GetEnvironmentVariable (\r
  IN     CONST CHAR16*   VariableName,\r
  IN     EFI_GUID*       VendorGuid,\r
  IN     VOID*           DefaultValue,\r
  IN OUT UINTN*          Size,\r
  OUT    VOID**          Value\r
  )\r
{\r
  EFI_STATUS  Status;\r
  UINTN       VariableSize;\r
\r
  // Try to get the variable size.\r
  *Value = NULL;\r
  VariableSize = 0;\r
  Status = gRT->GetVariable ((CHAR16 *) VariableName, VendorGuid, NULL, &VariableSize, *Value);\r
  if (Status == EFI_NOT_FOUND) {\r
    if ((DefaultValue != NULL) && (Size != NULL) && (*Size != 0)) {\r
      // If the environment variable does not exist yet then set it with the default value\r
      Status = gRT->SetVariable (\r
                    (CHAR16*)VariableName,\r
                    VendorGuid,\r
                    EFI_VARIABLE_NON_VOLATILE | EFI_VARIABLE_BOOTSERVICE_ACCESS | EFI_VARIABLE_RUNTIME_ACCESS,\r
                    *Size,\r
                    DefaultValue\r
                    );\r
      *Value = AllocateCopyPool (*Size, DefaultValue);\r
    } else {\r
      return EFI_NOT_FOUND;\r
    }\r
  } else if (Status == EFI_BUFFER_TOO_SMALL) {\r
    // Get the environment variable value\r
    *Value = AllocatePool (VariableSize);\r
    if (*Value == NULL) {\r
      return EFI_OUT_OF_RESOURCES;\r
    }\r
\r
    Status = gRT->GetVariable ((CHAR16 *)VariableName, VendorGuid, NULL, &VariableSize, *Value);\r
    if (EFI_ERROR (Status)) {\r
      FreePool(*Value);\r
      return EFI_INVALID_PARAMETER;\r
    }\r
\r
    if (Size) {\r
      *Size = VariableSize;\r
    }\r
  } else {\r
    *Value = AllocateCopyPool (*Size, DefaultValue);\r
    return Status;\r
  }\r
\r
  return EFI_SUCCESS;\r
}\r