Sync up ArmPkg with patch from mailing list. Changed name of BdsLib.h to BdsUnixLib...

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

/** @file\r
*\r
*  Copyright (c) 2011, 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
#include <Library/DxeServicesTableLib.h>\r
#include <Library/HobLib.h>\r
\r
EFI_STATUS\r
ShutdownUefiBootServices( VOID )\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
  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
      // 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
    }\r
  } while (EFI_ERROR (Status));\r
\r
  return Status;\r
}\r
\r
EFI_STATUS\r
BdsConnectAllDrivers( VOID ) {\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
STATIC EFI_STATUS InsertSystemMemoryResources(LIST_ENTRY *ResourceList, EFI_HOB_RESOURCE_DESCRIPTOR *ResHob) {\r
    BDS_SYSTEM_MEMORY_RESOURCE  NewResource;\r
    LIST_ENTRY                  *Link;\r
    BDS_SYSTEM_MEMORY_RESOURCE  *Resource;\r
\r
    //DEBUG ((EFI_D_ERROR, "** InsertSystemMemoryResources(0x%X,0x%X)\n",(UINT32)ResHob->PhysicalStart,(UINT32)ResHob->ResourceLength));\r
\r
    if (IsListEmpty (ResourceList)) {\r
        ZeroMem(&NewResource,sizeof(BDS_SYSTEM_MEMORY_RESOURCE));\r
        NewResource.PhysicalStart = ResHob->PhysicalStart;\r
        NewResource.ResourceLength = ResHob->ResourceLength;\r
        InsertTailList (ResourceList, &NewResource.Link);\r
        return EFI_SUCCESS;\r
    }\r
\r
    //for (Link = GetFirstNode (ResourceList); !IsNull (ResourceList,Link); Link = GetNextNode (ResourceList,Link)) {\r
    Link = ResourceList->ForwardLink;\r
    while (Link != NULL && Link != ResourceList) {\r
        Resource = (BDS_SYSTEM_MEMORY_RESOURCE*)Link;\r
        //DEBUG ((EFI_D_ERROR, "   - (0x%X,0x%X)\n",(UINT32)Resource->PhysicalStart,(UINT32)Resource->ResourceLength));\r
\r
        // Sanity Check. The resources should not overlapped.\r
        ASSERT(!((ResHob->PhysicalStart >= Resource->PhysicalStart) && (ResHob->PhysicalStart < (Resource->PhysicalStart + Resource->ResourceLength))));\r
        ASSERT(!((ResHob->PhysicalStart + ResHob->ResourceLength >= Resource->PhysicalStart) &&\r
            ((ResHob->PhysicalStart + ResHob->ResourceLength) < (Resource->PhysicalStart + Resource->ResourceLength))));\r
\r
        // The new resource is attached after this resource descriptor\r
        if (ResHob->PhysicalStart == Resource->PhysicalStart + Resource->ResourceLength) {\r
            Resource->ResourceLength =  Resource->ResourceLength + ResHob->ResourceLength;\r
            //DEBUG ((EFI_D_ERROR, "** Attached new Length:0x%X\n",(UINT32)Resource->ResourceLength));\r
            return EFI_SUCCESS;\r
        }\r
        // The new resource is attached before this resource descriptor\r
        else if (ResHob->PhysicalStart + ResHob->ResourceLength == Resource->PhysicalStart) {\r
            Resource->PhysicalStart = ResHob->PhysicalStart;\r
            Resource->ResourceLength =  Resource->ResourceLength + ResHob->ResourceLength;\r
            //DEBUG ((EFI_D_ERROR, "** Attached2 new Length:0x%X\n",(UINT32)Resource->ResourceLength));\r
            return EFI_SUCCESS;\r
        }\r
        Link = Link->ForwardLink;\r
    }\r
\r
    // None of the Resource of the list is attached to this ResHob. Create a new entry for it\r
    ZeroMem(&NewResource,sizeof(BDS_SYSTEM_MEMORY_RESOURCE));\r
    NewResource.PhysicalStart = ResHob->PhysicalStart;\r
    NewResource.ResourceLength = ResHob->ResourceLength;\r
    InsertTailList (ResourceList, &NewResource.Link);\r
    return EFI_SUCCESS;\r
}\r
\r
EFI_STATUS GetSystemMemoryResources(LIST_ENTRY *ResourceList) {\r
    EFI_HOB_RESOURCE_DESCRIPTOR *ResHob;\r
\r
    InitializeListHead (ResourceList);\r
    \r
    // Find the first System Memory Resource Descriptor\r
    ResHob = (EFI_HOB_RESOURCE_DESCRIPTOR *)GetFirstHob (EFI_HOB_TYPE_RESOURCE_DESCRIPTOR);\r
    while ((ResHob != NULL) && (ResHob->ResourceType != EFI_RESOURCE_SYSTEM_MEMORY)) {\r
        ResHob = (EFI_HOB_RESOURCE_DESCRIPTOR *)GetNextHob (EFI_HOB_TYPE_RESOURCE_DESCRIPTOR,(VOID *)((UINTN)ResHob + ResHob->Header.HobLength)); \r
    }\r
\r
    // Did not find any\r
    if (ResHob == NULL) {\r
        return EFI_NOT_FOUND;\r
    } else {\r
        InsertSystemMemoryResources(ResourceList, ResHob);\r
    }\r
\r
    ResHob = (EFI_HOB_RESOURCE_DESCRIPTOR *)GetNextHob (EFI_HOB_TYPE_RESOURCE_DESCRIPTOR,(VOID *)((UINTN)ResHob + ResHob->Header.HobLength)); \r
    while (ResHob != NULL) {\r
        if (ResHob->ResourceType == EFI_RESOURCE_SYSTEM_MEMORY) {\r
            InsertSystemMemoryResources(ResourceList, ResHob);\r
        }\r
        ResHob = (EFI_HOB_RESOURCE_DESCRIPTOR *)GetNextHob (EFI_HOB_TYPE_RESOURCE_DESCRIPTOR,(VOID *)((UINTN)ResHob + ResHob->Header.HobLength)); \r
    }\r
\r
    return EFI_SUCCESS;\r
}\r