Changed old event definitions reference to these new event definitions. git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@2729 6f19259b-4bc3-4df7-8a09-765794883524
		
			
				
	
	
		
			733 lines
		
	
	
		
			16 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			733 lines
		
	
	
		
			16 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
/** @file
 | 
						|
PCD DXE driver
 | 
						|
 | 
						|
Copyright (c) 2006 - 2007, Intel Corporation
 | 
						|
All rights reserved. This program and the accompanying materials
 | 
						|
are licensed and made available under the terms and conditions of the BSD License
 | 
						|
which accompanies this distribution.  The full text of the license may be found at
 | 
						|
http://opensource.org/licenses/bsd-license.php
 | 
						|
 | 
						|
THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,
 | 
						|
WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
 | 
						|
 | 
						|
 | 
						|
Module Name: Pcd.c
 | 
						|
 | 
						|
**/
 | 
						|
 | 
						|
#include "Service.h"
 | 
						|
 | 
						|
EFI_LOCK mPcdDatabaseLock = EFI_INITIALIZE_LOCK_VARIABLE(TPL_CALLBACK);
 | 
						|
 | 
						|
PCD_PROTOCOL mPcdInstance = {
 | 
						|
  DxePcdSetSku,
 | 
						|
 | 
						|
  DxePcdGet8,
 | 
						|
  DxePcdGet16,
 | 
						|
  DxePcdGet32,
 | 
						|
  DxePcdGet64,
 | 
						|
  DxePcdGetPtr,
 | 
						|
  DxePcdGetBool,
 | 
						|
  DxePcdGetSize,
 | 
						|
 | 
						|
  DxePcdGet8Ex,
 | 
						|
  DxePcdGet16Ex,
 | 
						|
  DxePcdGet32Ex,
 | 
						|
  DxePcdGet64Ex,
 | 
						|
  DxePcdGetPtrEx,
 | 
						|
  DxePcdGetBoolEx,
 | 
						|
  DxePcdGetSizeEx,
 | 
						|
 | 
						|
  DxePcdSet8,
 | 
						|
  DxePcdSet16,
 | 
						|
  DxePcdSet32,
 | 
						|
  DxePcdSet64,
 | 
						|
  DxePcdSetPtr,
 | 
						|
  DxePcdSetBool,
 | 
						|
 | 
						|
  DxePcdSet8Ex,
 | 
						|
  DxePcdSet16Ex,
 | 
						|
  DxePcdSet32Ex,
 | 
						|
  DxePcdSet64Ex,
 | 
						|
  DxePcdSetPtrEx,
 | 
						|
  DxePcdSetBoolEx,
 | 
						|
 | 
						|
  DxeRegisterCallBackOnSet,
 | 
						|
  DxeUnRegisterCallBackOnSet,
 | 
						|
  DxePcdGetNextToken,
 | 
						|
  DxePcdGetNextTokenSpace
 | 
						|
};
 | 
						|
 | 
						|
 | 
						|
//
 | 
						|
// Static global to reduce the code size
 | 
						|
//
 | 
						|
static EFI_HANDLE mNewHandle = NULL;
 | 
						|
 | 
						|
EFI_STATUS
 | 
						|
EFIAPI
 | 
						|
PcdDxeInit (
 | 
						|
  IN EFI_HANDLE           ImageHandle,
 | 
						|
  IN EFI_SYSTEM_TABLE     *SystemTable
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS          Status;
 | 
						|
 | 
						|
  //
 | 
						|
  // Make sure the Pcd Protocol is not already installed in the system
 | 
						|
  //
 | 
						|
 | 
						|
  ASSERT_PROTOCOL_ALREADY_INSTALLED (NULL, &gPcdProtocolGuid);
 | 
						|
 | 
						|
  BuildPcdDxeDataBase ();
 | 
						|
 | 
						|
  Status = gBS->InstallProtocolInterface (
 | 
						|
                  &mNewHandle,
 | 
						|
                  &gPcdProtocolGuid,
 | 
						|
                  EFI_NATIVE_INTERFACE,
 | 
						|
                  &mPcdInstance
 | 
						|
                  );
 | 
						|
 | 
						|
  ASSERT_EFI_ERROR (Status);
 | 
						|
 | 
						|
  return EFI_SUCCESS;
 | 
						|
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
VOID
 | 
						|
EFIAPI
 | 
						|
DxePcdSetSku (
 | 
						|
  IN  UINTN         SkuId
 | 
						|
  )
 | 
						|
{
 | 
						|
  mPcdDatabase->PeiDb.Init.SystemSkuId = (SKU_ID) SkuId;
 | 
						|
  
 | 
						|
  return;
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
 | 
						|
UINT8
 | 
						|
EFIAPI
 | 
						|
DxePcdGet8 (
 | 
						|
  IN UINTN                    TokenNumber
 | 
						|
  )
 | 
						|
{
 | 
						|
  return *((UINT8 *) GetWorker (TokenNumber, sizeof (UINT8)));
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
 | 
						|
UINT16
 | 
						|
EFIAPI
 | 
						|
DxePcdGet16 (
 | 
						|
  IN UINTN                    TokenNumber
 | 
						|
  )
 | 
						|
{
 | 
						|
  return ReadUnaligned16 (GetWorker (TokenNumber, sizeof (UINT16)));
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
 | 
						|
UINT32
 | 
						|
EFIAPI
 | 
						|
DxePcdGet32 (
 | 
						|
  IN UINTN                    TokenNumber
 | 
						|
  )
 | 
						|
{
 | 
						|
  return ReadUnaligned32 (GetWorker (TokenNumber, sizeof (UINT32)));
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
 | 
						|
UINT64
 | 
						|
EFIAPI
 | 
						|
DxePcdGet64 (
 | 
						|
  IN UINTN                     TokenNumber
 | 
						|
  )
 | 
						|
{
 | 
						|
  return ReadUnaligned64(GetWorker (TokenNumber, sizeof (UINT64)));
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
 | 
						|
VOID *
 | 
						|
EFIAPI
 | 
						|
DxePcdGetPtr (
 | 
						|
  IN UINTN                     TokenNumber
 | 
						|
  )
 | 
						|
{
 | 
						|
  return GetWorker (TokenNumber, 0);
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
 | 
						|
BOOLEAN
 | 
						|
EFIAPI
 | 
						|
DxePcdGetBool (
 | 
						|
  IN UINTN                     TokenNumber
 | 
						|
  )
 | 
						|
{
 | 
						|
  return *((BOOLEAN *) GetWorker (TokenNumber, sizeof (BOOLEAN)));
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
 | 
						|
UINTN
 | 
						|
EFIAPI
 | 
						|
DxePcdGetSize (
 | 
						|
  IN UINTN                     TokenNumber
 | 
						|
  )
 | 
						|
{
 | 
						|
  UINTN   Size;
 | 
						|
  UINT32  *LocalTokenNumberTable;
 | 
						|
  BOOLEAN IsPeiDb;
 | 
						|
  UINTN   MaxSize;
 | 
						|
  UINTN   TmpTokenNumber;
 | 
						|
  //
 | 
						|
  // TokenNumber Zero is reserved as PCD_INVALID_TOKEN_NUMBER.
 | 
						|
  // We have to decrement TokenNumber by 1 to make it usable
 | 
						|
  // as the array index.
 | 
						|
  //
 | 
						|
  TokenNumber--;
 | 
						|
 | 
						|
  //
 | 
						|
  // Backup the TokenNumber passed in as GetPtrTypeSize need the original TokenNumber
 | 
						|
  // 
 | 
						|
  TmpTokenNumber = TokenNumber;
 | 
						|
 | 
						|
  // EBC compiler is very choosy. It may report warning about comparison
 | 
						|
  // between UINTN and 0 . So we add 1 in each size of the 
 | 
						|
  // comparison.
 | 
						|
  ASSERT (TokenNumber + 1 < PCD_TOTAL_TOKEN_NUMBER + 1);
 | 
						|
 | 
						|
  // EBC compiler is very choosy. It may report warning about comparison
 | 
						|
  // between UINTN and 0 . So we add 1 in each size of the 
 | 
						|
  // comparison.
 | 
						|
  IsPeiDb = (BOOLEAN) (TokenNumber + 1 < PEI_LOCAL_TOKEN_NUMBER + 1);
 | 
						|
  
 | 
						|
  TokenNumber = IsPeiDb ? TokenNumber : 
 | 
						|
                          (TokenNumber - PEI_LOCAL_TOKEN_NUMBER);
 | 
						|
 | 
						|
  LocalTokenNumberTable = IsPeiDb ? mPcdDatabase->PeiDb.Init.LocalTokenNumberTable 
 | 
						|
                                  : mPcdDatabase->DxeDb.Init.LocalTokenNumberTable;
 | 
						|
 | 
						|
  Size = (LocalTokenNumberTable[TokenNumber] & PCD_DATUM_TYPE_ALL_SET) >> PCD_DATUM_TYPE_SHIFT;
 | 
						|
 | 
						|
  if (Size == 0) {
 | 
						|
    //
 | 
						|
    // For pointer type, we need to scan the SIZE_TABLE to get the current size.
 | 
						|
    //
 | 
						|
    return GetPtrTypeSize (TmpTokenNumber, &MaxSize);
 | 
						|
  } else {
 | 
						|
    return Size;
 | 
						|
  }
 | 
						|
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
 | 
						|
UINT8
 | 
						|
EFIAPI
 | 
						|
DxePcdGet8Ex (
 | 
						|
  IN CONST EFI_GUID         *Guid,
 | 
						|
  IN UINTN                 ExTokenNumber
 | 
						|
  )
 | 
						|
{
 | 
						|
  return *((UINT8 *) ExGetWorker (Guid, ExTokenNumber, sizeof(UINT8)));
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
 | 
						|
UINT16
 | 
						|
EFIAPI
 | 
						|
DxePcdGet16Ex (
 | 
						|
  IN CONST EFI_GUID        *Guid,
 | 
						|
  IN UINTN                ExTokenNumber
 | 
						|
  )
 | 
						|
{
 | 
						|
  return ReadUnaligned16 (ExGetWorker (Guid, ExTokenNumber, sizeof(UINT16)));
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
 | 
						|
UINT32
 | 
						|
EFIAPI
 | 
						|
DxePcdGet32Ex (
 | 
						|
  IN CONST EFI_GUID        *Guid,
 | 
						|
  IN UINTN                 ExTokenNumber
 | 
						|
  )
 | 
						|
{
 | 
						|
  return ReadUnaligned32 (ExGetWorker (Guid, ExTokenNumber, sizeof(UINT32)));
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
 | 
						|
UINT64
 | 
						|
EFIAPI
 | 
						|
DxePcdGet64Ex (
 | 
						|
  IN CONST EFI_GUID        *Guid,
 | 
						|
  IN UINTN                 ExTokenNumber
 | 
						|
  )
 | 
						|
{
 | 
						|
  return ReadUnaligned64 (ExGetWorker (Guid, ExTokenNumber, sizeof(UINT64)));
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
 | 
						|
VOID *
 | 
						|
EFIAPI
 | 
						|
DxePcdGetPtrEx (
 | 
						|
  IN CONST EFI_GUID        *Guid,
 | 
						|
  IN UINTN                 ExTokenNumber
 | 
						|
  )
 | 
						|
{
 | 
						|
  return  ExGetWorker (Guid, ExTokenNumber, 0);
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
 | 
						|
BOOLEAN
 | 
						|
EFIAPI
 | 
						|
DxePcdGetBoolEx (
 | 
						|
  IN CONST EFI_GUID        *Guid,
 | 
						|
  IN UINTN                 ExTokenNumber
 | 
						|
  )
 | 
						|
{
 | 
						|
  return *((BOOLEAN *) ExGetWorker (Guid, ExTokenNumber, sizeof(BOOLEAN)));
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
 | 
						|
UINTN
 | 
						|
EFIAPI
 | 
						|
DxePcdGetSizeEx (
 | 
						|
  IN CONST EFI_GUID        *Guid,
 | 
						|
  IN UINTN                 ExTokenNumber
 | 
						|
  )
 | 
						|
{
 | 
						|
  return DxePcdGetSize(GetExPcdTokenNumber (Guid, (UINT32) ExTokenNumber));
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
 | 
						|
EFI_STATUS
 | 
						|
EFIAPI
 | 
						|
DxePcdSet8 (
 | 
						|
  IN UINTN              TokenNumber,
 | 
						|
  IN UINT8              Value
 | 
						|
  )
 | 
						|
{
 | 
						|
  return SetValueWorker (TokenNumber, &Value, sizeof (Value));
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
 | 
						|
EFI_STATUS
 | 
						|
EFIAPI
 | 
						|
DxePcdSet16 (
 | 
						|
  IN UINTN              TokenNumber,
 | 
						|
  IN UINT16             Value
 | 
						|
  )
 | 
						|
{
 | 
						|
  return SetValueWorker (TokenNumber, &Value, sizeof (Value));
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
 | 
						|
EFI_STATUS
 | 
						|
EFIAPI
 | 
						|
DxePcdSet32 (
 | 
						|
  IN UINTN              TokenNumber,
 | 
						|
  IN UINT32             Value
 | 
						|
  )
 | 
						|
{
 | 
						|
  return SetValueWorker (TokenNumber, &Value, sizeof (Value));
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
 | 
						|
EFI_STATUS
 | 
						|
EFIAPI
 | 
						|
DxePcdSet64 (
 | 
						|
  IN UINTN              TokenNumber,
 | 
						|
  IN UINT64             Value
 | 
						|
  )
 | 
						|
{
 | 
						|
  return SetValueWorker (TokenNumber, &Value, sizeof (Value));
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
 | 
						|
EFI_STATUS
 | 
						|
EFIAPI
 | 
						|
DxePcdSetPtr (
 | 
						|
  IN          UINTN              TokenNumber,
 | 
						|
  IN OUT      UINTN              *SizeOfBuffer,
 | 
						|
  IN          VOID               *Buffer
 | 
						|
  )
 | 
						|
{
 | 
						|
  return SetWorker (TokenNumber, Buffer, SizeOfBuffer, TRUE);
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
 | 
						|
EFI_STATUS
 | 
						|
EFIAPI
 | 
						|
DxePcdSetBool (
 | 
						|
  IN UINTN              TokenNumber,
 | 
						|
  IN BOOLEAN            Value
 | 
						|
  )
 | 
						|
{
 | 
						|
  return SetValueWorker (TokenNumber, &Value, sizeof (Value));
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
 | 
						|
EFI_STATUS
 | 
						|
EFIAPI
 | 
						|
DxePcdSet8Ex (
 | 
						|
  IN CONST EFI_GUID         *Guid,
 | 
						|
  IN UINTN                  ExTokenNumber,
 | 
						|
  IN UINT8                  Value
 | 
						|
  )
 | 
						|
{
 | 
						|
  return  ExSetValueWorker (ExTokenNumber, Guid, &Value, sizeof (Value));
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
 | 
						|
EFI_STATUS
 | 
						|
EFIAPI
 | 
						|
DxePcdSet16Ex (
 | 
						|
  IN CONST EFI_GUID    *Guid,
 | 
						|
  IN UINTN             ExTokenNumber,
 | 
						|
  IN UINT16            Value
 | 
						|
  )
 | 
						|
{
 | 
						|
  return  ExSetValueWorker (ExTokenNumber, Guid, &Value, sizeof (Value));
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
 | 
						|
EFI_STATUS
 | 
						|
EFIAPI
 | 
						|
DxePcdSet32Ex (
 | 
						|
  IN CONST EFI_GUID     *Guid,
 | 
						|
  IN UINTN              ExTokenNumber,
 | 
						|
  IN UINT32             Value
 | 
						|
  )
 | 
						|
{
 | 
						|
  return  ExSetValueWorker (ExTokenNumber, Guid, &Value, sizeof (Value));
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
 | 
						|
EFI_STATUS
 | 
						|
EFIAPI
 | 
						|
DxePcdSet64Ex (
 | 
						|
  IN CONST EFI_GUID    *Guid,
 | 
						|
  IN UINTN             ExTokenNumber,
 | 
						|
  IN UINT64            Value
 | 
						|
  )
 | 
						|
{
 | 
						|
  return  ExSetValueWorker (ExTokenNumber, Guid, &Value, sizeof (Value));
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
 | 
						|
EFI_STATUS
 | 
						|
EFIAPI
 | 
						|
DxePcdSetPtrEx (
 | 
						|
  IN            CONST EFI_GUID         *Guid,
 | 
						|
  IN            UINTN                  ExTokenNumber,
 | 
						|
  IN OUT        UINTN                  *SizeOfBuffer,
 | 
						|
  IN            VOID                   *Buffer
 | 
						|
  )
 | 
						|
{
 | 
						|
  return  ExSetWorker(ExTokenNumber, Guid, Buffer, SizeOfBuffer, TRUE);
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
 | 
						|
EFI_STATUS
 | 
						|
EFIAPI
 | 
						|
DxePcdSetBoolEx (
 | 
						|
  IN CONST EFI_GUID    *Guid,
 | 
						|
  IN UINTN             ExTokenNumber,
 | 
						|
  IN BOOLEAN           Value
 | 
						|
  )
 | 
						|
{
 | 
						|
  return  ExSetValueWorker (ExTokenNumber, Guid, &Value, sizeof (Value));
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
 | 
						|
 | 
						|
EFI_STATUS
 | 
						|
EFIAPI
 | 
						|
DxeRegisterCallBackOnSet (
 | 
						|
  IN  CONST EFI_GUID          *Guid, OPTIONAL
 | 
						|
  IN  UINTN                   TokenNumber,
 | 
						|
  IN  PCD_PROTOCOL_CALLBACK   CallBackFunction
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS Status;
 | 
						|
  
 | 
						|
  ASSERT (CallBackFunction != NULL);
 | 
						|
  
 | 
						|
  //
 | 
						|
  // Aquire lock to prevent reentrance from TPL_CALLBACK level
 | 
						|
  //
 | 
						|
  EfiAcquireLock (&mPcdDatabaseLock);
 | 
						|
 | 
						|
  Status = DxeRegisterCallBackWorker (TokenNumber, Guid, CallBackFunction);
 | 
						|
 | 
						|
  EfiReleaseLock (&mPcdDatabaseLock);
 | 
						|
  
 | 
						|
  return Status;
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
 | 
						|
EFI_STATUS
 | 
						|
EFIAPI
 | 
						|
DxeUnRegisterCallBackOnSet (
 | 
						|
  IN  CONST EFI_GUID          *Guid, OPTIONAL
 | 
						|
  IN  UINTN                   TokenNumber,
 | 
						|
  IN  PCD_PROTOCOL_CALLBACK   CallBackFunction
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS Status;
 | 
						|
  
 | 
						|
  ASSERT (CallBackFunction != NULL);
 | 
						|
 | 
						|
  //
 | 
						|
  // Aquire lock to prevent reentrance from TPL_CALLBACK level
 | 
						|
  //
 | 
						|
  EfiAcquireLock (&mPcdDatabaseLock);
 | 
						|
  
 | 
						|
  Status = DxeUnRegisterCallBackWorker (TokenNumber, Guid, CallBackFunction);
 | 
						|
 | 
						|
  EfiReleaseLock (&mPcdDatabaseLock);
 | 
						|
  
 | 
						|
  return Status;
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
 | 
						|
EFI_STATUS
 | 
						|
EFIAPI
 | 
						|
DxePcdGetNextToken (
 | 
						|
  IN CONST EFI_GUID         *Guid, OPTIONAL
 | 
						|
  IN OUT   UINTN            *TokenNumber
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS          Status;
 | 
						|
  BOOLEAN             PeiExMapTableEmpty;
 | 
						|
  BOOLEAN             DxeExMapTableEmpty;
 | 
						|
 | 
						|
  if (!FeaturePcdGet (PcdDxePcdDatabaseTraverseEnabled)) {
 | 
						|
    return EFI_UNSUPPORTED;
 | 
						|
  }
 | 
						|
 | 
						|
  Status = EFI_NOT_FOUND;
 | 
						|
  PeiExMapTableEmpty = PEI_EXMAP_TABLE_EMPTY;
 | 
						|
  DxeExMapTableEmpty = DXE_EXMAP_TABLE_EMPTY;
 | 
						|
 | 
						|
  //
 | 
						|
  // Scan the local token space
 | 
						|
  //
 | 
						|
  if (Guid == NULL) {
 | 
						|
    // EBC compiler is very choosy. It may report warning about comparison
 | 
						|
    // between UINTN and 0 . So we add 1 in each size of the 
 | 
						|
    // comparison.
 | 
						|
    if (((*TokenNumber + 1 > PEI_NEX_TOKEN_NUMBER + 1) && (*TokenNumber + 1 < PEI_LOCAL_TOKEN_NUMBER + 1)) ||
 | 
						|
        ((*TokenNumber + 1 > (PEI_LOCAL_TOKEN_NUMBER + DXE_NEX_TOKEN_NUMBER + 1)))) {
 | 
						|
        return EFI_NOT_FOUND;
 | 
						|
    }
 | 
						|
    
 | 
						|
    (*TokenNumber)++;
 | 
						|
    if ((*TokenNumber + 1 > PEI_NEX_TOKEN_NUMBER + 1) &&
 | 
						|
        (*TokenNumber <= PEI_LOCAL_TOKEN_NUMBER)) {
 | 
						|
      //
 | 
						|
      // The first Non-Ex type Token Number for DXE PCD 
 | 
						|
      // database is PEI_LOCAL_TOKEN_NUMBER
 | 
						|
      //
 | 
						|
      *TokenNumber = PEI_LOCAL_TOKEN_NUMBER;
 | 
						|
    } else if (*TokenNumber + 1 > DXE_NEX_TOKEN_NUMBER + PEI_LOCAL_TOKEN_NUMBER + 1) {
 | 
						|
      *TokenNumber = PCD_INVALID_TOKEN_NUMBER;
 | 
						|
    }
 | 
						|
    return EFI_SUCCESS;
 | 
						|
  }
 | 
						|
 | 
						|
  if (PeiExMapTableEmpty && DxeExMapTableEmpty) {
 | 
						|
    *TokenNumber = PCD_INVALID_TOKEN_NUMBER;
 | 
						|
    return EFI_NOT_FOUND;
 | 
						|
  }
 | 
						|
 | 
						|
  if (!PeiExMapTableEmpty) {
 | 
						|
    Status = ExGetNextTokeNumber (
 | 
						|
                        Guid,
 | 
						|
                        TokenNumber,
 | 
						|
                        mPcdDatabase->PeiDb.Init.GuidTable,
 | 
						|
                        sizeof(mPcdDatabase->PeiDb.Init.GuidTable),
 | 
						|
                        mPcdDatabase->PeiDb.Init.ExMapTable,
 | 
						|
                        sizeof(mPcdDatabase->PeiDb.Init.ExMapTable)
 | 
						|
                        );
 | 
						|
  }
 | 
						|
 | 
						|
  if (Status == EFI_SUCCESS) {
 | 
						|
    return Status;
 | 
						|
  }
 | 
						|
 | 
						|
  if (!DxeExMapTableEmpty) {
 | 
						|
    Status = ExGetNextTokeNumber (
 | 
						|
                        Guid,
 | 
						|
                        TokenNumber,
 | 
						|
                        mPcdDatabase->DxeDb.Init.GuidTable,
 | 
						|
                        sizeof(mPcdDatabase->DxeDb.Init.GuidTable),
 | 
						|
                        mPcdDatabase->DxeDb.Init.ExMapTable,
 | 
						|
                        sizeof(mPcdDatabase->DxeDb.Init.ExMapTable)
 | 
						|
                        );
 | 
						|
  }
 | 
						|
 | 
						|
  return Status;
 | 
						|
}
 | 
						|
 | 
						|
STATIC
 | 
						|
EFI_GUID **
 | 
						|
GetDistinctTokenSpace (
 | 
						|
  IN OUT    UINTN             *ExMapTableSize,
 | 
						|
  IN        DYNAMICEX_MAPPING *ExMapTable,
 | 
						|
  IN        EFI_GUID          *GuidTable
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_GUID  **DistinctTokenSpace;
 | 
						|
  UINTN     OldGuidIndex;
 | 
						|
  UINTN     TsIdx;
 | 
						|
  UINTN     Idx;
 | 
						|
 | 
						|
 | 
						|
  DistinctTokenSpace = AllocateZeroPool (*ExMapTableSize * sizeof (EFI_GUID *));
 | 
						|
  ASSERT (DistinctTokenSpace != NULL);
 | 
						|
 | 
						|
  TsIdx = 0;
 | 
						|
  OldGuidIndex = ExMapTable[0].ExGuidIndex;
 | 
						|
  DistinctTokenSpace[TsIdx] = &GuidTable[OldGuidIndex];
 | 
						|
  for (Idx = 1; Idx < *ExMapTableSize; Idx++) {
 | 
						|
    if (ExMapTable[Idx].ExGuidIndex != OldGuidIndex) {
 | 
						|
      OldGuidIndex = ExMapTable[Idx].ExGuidIndex;
 | 
						|
      DistinctTokenSpace[++TsIdx] = &GuidTable[OldGuidIndex];
 | 
						|
    }
 | 
						|
  }
 | 
						|
 | 
						|
  //
 | 
						|
  // The total number of Distinct Token Space
 | 
						|
  // is TsIdx + 1 because we use TsIdx as a index
 | 
						|
  // to the DistinctTokenSpace[]
 | 
						|
  //
 | 
						|
  *ExMapTableSize = TsIdx + 1;
 | 
						|
  return DistinctTokenSpace;
 | 
						|
    
 | 
						|
}
 | 
						|
  
 | 
						|
//
 | 
						|
// Just pre-allocate a memory buffer that is big enough to
 | 
						|
// host all distinct TokenSpace guid in both
 | 
						|
// PEI ExMap and DXE ExMap.
 | 
						|
//
 | 
						|
STATIC EFI_GUID *TmpTokenSpaceBuffer[PEI_EXMAPPING_TABLE_SIZE + DXE_EXMAPPING_TABLE_SIZE] = { 0 };
 | 
						|
 | 
						|
EFI_STATUS
 | 
						|
EFIAPI
 | 
						|
DxePcdGetNextTokenSpace (
 | 
						|
  IN OUT CONST EFI_GUID               **Guid
 | 
						|
  )
 | 
						|
{
 | 
						|
  UINTN               Idx;
 | 
						|
  UINTN               Idx2;
 | 
						|
  UINTN               Idx3;
 | 
						|
  UINTN               PeiTokenSpaceTableSize;
 | 
						|
  UINTN               DxeTokenSpaceTableSize;
 | 
						|
  EFI_GUID            **PeiTokenSpaceTable;
 | 
						|
  EFI_GUID            **DxeTokenSpaceTable;
 | 
						|
  BOOLEAN             Match;
 | 
						|
  BOOLEAN             PeiExMapTableEmpty;
 | 
						|
  BOOLEAN             DxeExMapTableEmpty;
 | 
						|
 | 
						|
  if (!FeaturePcdGet (PcdDxePcdDatabaseTraverseEnabled)) {
 | 
						|
    return EFI_UNSUPPORTED;
 | 
						|
  }
 | 
						|
 | 
						|
  ASSERT (Guid != NULL);
 | 
						|
  
 | 
						|
  PeiExMapTableEmpty = PEI_EXMAP_TABLE_EMPTY;
 | 
						|
  DxeExMapTableEmpty = DXE_EXMAP_TABLE_EMPTY;
 | 
						|
 | 
						|
  if (PeiExMapTableEmpty && DxeExMapTableEmpty) {
 | 
						|
    if (*Guid != NULL) {
 | 
						|
      return EFI_NOT_FOUND;
 | 
						|
    } else {
 | 
						|
      return EFI_SUCCESS;
 | 
						|
    }
 | 
						|
  }
 | 
						|
  
 | 
						|
  
 | 
						|
  if (TmpTokenSpaceBuffer[0] == NULL) {
 | 
						|
    PeiTokenSpaceTableSize = 0;
 | 
						|
 | 
						|
    if (!PeiExMapTableEmpty) {
 | 
						|
      PeiTokenSpaceTableSize = PEI_EXMAPPING_TABLE_SIZE;
 | 
						|
      PeiTokenSpaceTable = GetDistinctTokenSpace (&PeiTokenSpaceTableSize,
 | 
						|
                            mPcdDatabase->PeiDb.Init.ExMapTable,
 | 
						|
                            mPcdDatabase->PeiDb.Init.GuidTable
 | 
						|
                            );
 | 
						|
      CopyMem (TmpTokenSpaceBuffer, PeiTokenSpaceTable, sizeof (EFI_GUID*) * PeiTokenSpaceTableSize);
 | 
						|
    }
 | 
						|
 | 
						|
    if (!DxeExMapTableEmpty) {
 | 
						|
      DxeTokenSpaceTableSize = DXE_EXMAPPING_TABLE_SIZE;
 | 
						|
      DxeTokenSpaceTable = GetDistinctTokenSpace (&DxeTokenSpaceTableSize,
 | 
						|
                            mPcdDatabase->DxeDb.Init.ExMapTable,
 | 
						|
                            mPcdDatabase->DxeDb.Init.GuidTable
 | 
						|
                            );
 | 
						|
 | 
						|
      //
 | 
						|
      // Make sure EFI_GUID in DxeTokenSpaceTable does not exist in PeiTokenSpaceTable
 | 
						|
      //
 | 
						|
      for (Idx2 = 0, Idx3 = PeiTokenSpaceTableSize; Idx2 < DxeTokenSpaceTableSize; Idx2++) {
 | 
						|
        Match = FALSE;
 | 
						|
        for (Idx = 0; Idx < PeiTokenSpaceTableSize; Idx++) {
 | 
						|
          if (CompareGuid (TmpTokenSpaceBuffer[Idx], DxeTokenSpaceTable[Idx2])) {
 | 
						|
            Match = TRUE;
 | 
						|
            break;
 | 
						|
          }
 | 
						|
        }
 | 
						|
        if (!Match) {
 | 
						|
          TmpTokenSpaceBuffer[Idx3++] = DxeTokenSpaceTable[Idx2];
 | 
						|
        }
 | 
						|
      }
 | 
						|
    }
 | 
						|
  }
 | 
						|
 | 
						|
  if (*Guid == NULL) {
 | 
						|
    *Guid = TmpTokenSpaceBuffer[0];
 | 
						|
    return EFI_SUCCESS;
 | 
						|
  }
 | 
						|
  
 | 
						|
  for (Idx = 0; Idx < (PEI_EXMAPPING_TABLE_SIZE + DXE_EXMAPPING_TABLE_SIZE); Idx++) {
 | 
						|
    if(CompareGuid (*Guid, TmpTokenSpaceBuffer[Idx])) {
 | 
						|
      Idx++;
 | 
						|
      *Guid = TmpTokenSpaceBuffer[Idx];
 | 
						|
      return EFI_SUCCESS;
 | 
						|
    }
 | 
						|
  }
 | 
						|
 | 
						|
  return EFI_NOT_FOUND;
 | 
						|
 | 
						|
}
 | 
						|
 | 
						|
 |