[mirror_edk2.git] / EdkModulePkg / Core / Dxe / Hand / handle.c

/*++\r
\r
Copyright (c) 2006, 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
  handle.c\r
\r
Abstract:\r
\r
  EFI handle & protocol handling\r
\r
\r
\r
Revision History\r
\r
--*/\r
\r
#include <DxeMain.h>\r
\r
\r
//\r
// mProtocolDatabase     - A list of all protocols in the system.  (simple list for now)\r
// gHandleList           - A list of all the handles in the system\r
// gProtocolDatabaseLock - Lock to protect the mProtocolDatabase\r
// gHandleDatabaseKey    -  The Key to show that the handle has been created/modified\r
//\r
static LIST_ENTRY      mProtocolDatabase     = INITIALIZE_LIST_HEAD_VARIABLE (mProtocolDatabase);\r
LIST_ENTRY             gHandleList           = INITIALIZE_LIST_HEAD_VARIABLE (gHandleList);\r
EFI_LOCK               gProtocolDatabaseLock = EFI_INITIALIZE_LOCK_VARIABLE (EFI_TPL_NOTIFY);\r
UINT64                 gHandleDatabaseKey    = 0;\r
\r
\r
VOID\r
CoreAcquireProtocolLock (\r
  VOID\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  Acquire lock on gProtocolDatabaseLock.\r
  \r
Arguments:\r
\r
  None\r
  \r
Returns:\r
\r
  None\r
\r
--*/\r
{\r
  CoreAcquireLock (&gProtocolDatabaseLock);\r
}\r
\r
\r
VOID\r
CoreReleaseProtocolLock (\r
  VOID\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  Release lock on gProtocolDatabaseLock.\r
  \r
Arguments:\r
\r
  None\r
  \r
Returns:\r
\r
  None\r
\r
--*/\r
{\r
  CoreReleaseLock (&gProtocolDatabaseLock);\r
}\r
\r
\r
EFI_STATUS\r
CoreValidateHandle (\r
  IN  EFI_HANDLE                UserHandle\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  Check whether a handle is a valid EFI_HANDLE\r
  \r
Arguments:\r
\r
  UserHandle    - The handle to check\r
  \r
Returns:\r
\r
  EFI_INVALID_PARAMETER   - The handle is NULL or not a valid EFI_HANDLE.\r
\r
  EFI_SUCCESS             - The handle is valid EFI_HANDLE.\r
\r
--*/\r
{\r
  IHANDLE             *Handle;\r
\r
  Handle = (IHANDLE *)UserHandle;\r
  if (Handle == NULL) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
  if (Handle->Signature != EFI_HANDLE_SIGNATURE) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
  return EFI_SUCCESS;\r
}\r
\r
\r
PROTOCOL_ENTRY  *\r
CoreFindProtocolEntry (\r
  IN EFI_GUID   *Protocol,\r
  IN BOOLEAN    Create\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  Finds the protocol entry for the requested protocol.\r
  \r
  The gProtocolDatabaseLock must be owned\r
\r
Arguments:\r
  \r
  Protocol  - The ID of the protocol \r
\r
  Create    - Create a new entry if not found\r
\r
Returns:\r
\r
  Protocol entry\r
\r
--*/\r
{\r
  LIST_ENTRY          *Link;\r
  PROTOCOL_ENTRY      *Item;\r
  PROTOCOL_ENTRY      *ProtEntry;\r
\r
  ASSERT_LOCKED(&gProtocolDatabaseLock);\r
\r
  //\r
  // Search the database for the matching GUID\r
  //\r
\r
  ProtEntry = NULL;\r
  for (Link = mProtocolDatabase.ForwardLink; \r
       Link != &mProtocolDatabase; \r
       Link = Link->ForwardLink) {\r
\r
    Item = CR(Link, PROTOCOL_ENTRY, AllEntries, PROTOCOL_ENTRY_SIGNATURE);\r
    if (CompareGuid (&Item->ProtocolID, Protocol)) {\r
\r
      //\r
      // This is the protocol entry\r
      //\r
\r
      ProtEntry = Item;\r
      break;\r
    }\r
  }\r
\r
  //\r
  // If the protocol entry was not found and Create is TRUE, then \r
  // allocate a new entry\r
  //     \r
  if ((ProtEntry == NULL) && Create) {\r
    ProtEntry = CoreAllocateBootServicesPool (sizeof(PROTOCOL_ENTRY));\r
    \r
    if (ProtEntry != NULL) {\r
      //\r
      // Initialize new protocol entry structure\r
      //\r
      ProtEntry->Signature = PROTOCOL_ENTRY_SIGNATURE;\r
      CopyMem ((VOID *)&ProtEntry->ProtocolID, Protocol, sizeof (EFI_GUID));\r
      InitializeListHead (&ProtEntry->Protocols);\r
      InitializeListHead (&ProtEntry->Notify);\r
\r
      //\r
      // Add it to protocol database\r
      //\r
      InsertTailList (&mProtocolDatabase, &ProtEntry->AllEntries);\r
    }\r
  }\r
\r
  return ProtEntry;\r
}\r
\r
\r
PROTOCOL_INTERFACE *\r
CoreFindProtocolInterface (\r
  IN IHANDLE        *Handle,\r
  IN EFI_GUID       *Protocol,\r
  IN VOID           *Interface\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  Finds the protocol instance for the requested handle and protocol.\r
  \r
  Note: This function doesn't do parameters checking, it's caller's responsibility \r
        to pass in valid parameters.\r
  \r
Arguments:\r
  \r
  Handle    - The handle to search the protocol on\r
  \r
  Protocol  - GUID of the protocol\r
\r
  Interface - The interface for the protocol being searched\r
\r
Returns:\r
\r
  Protocol instance (NULL: Not found)\r
\r
--*/  \r
{\r
  PROTOCOL_INTERFACE  *Prot;\r
  PROTOCOL_ENTRY      *ProtEntry;\r
  LIST_ENTRY          *Link;\r
\r
  ASSERT_LOCKED(&gProtocolDatabaseLock);\r
  Prot = NULL;\r
\r
  //\r
  // Lookup the protocol entry for this protocol ID\r
  //\r
\r
  ProtEntry = CoreFindProtocolEntry (Protocol, FALSE);\r
  if (ProtEntry != NULL) {\r
\r
    //\r
    // Look at each protocol interface for any matches\r
    //\r
    for (Link = Handle->Protocols.ForwardLink; Link != &Handle->Protocols; Link=Link->ForwardLink) {\r
\r
      //\r
      // If this protocol interface matches, remove it\r
      //\r
      Prot = CR(Link, PROTOCOL_INTERFACE, Link, PROTOCOL_INTERFACE_SIGNATURE);\r
      if (Prot->Interface == Interface && Prot->Protocol == ProtEntry) {\r
        break;\r
      }\r
\r
      Prot = NULL;\r
    }\r
  }\r
\r
  return Prot;\r
}\r
\r
STATIC\r
EFI_STATUS\r
CoreUnregisterProtocolNotifyEvent (\r
  IN EFI_EVENT      Event\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  Removes an event from a register protocol notify list on a protocol.\r
\r
Arguments:\r
  \r
  Event   - The event to search for in the protocol database.\r
\r
Returns:\r
\r
  EFI_SUCCESS   if the event was found and removed.\r
  EFI_NOT_FOUND if the event was not found in the protocl database.\r
\r
--*/\r
{\r
  LIST_ENTRY         *Link;\r
  PROTOCOL_ENTRY     *ProtEntry;\r
  LIST_ENTRY         *NotifyLink;\r
  PROTOCOL_NOTIFY    *ProtNotify;\r
\r
  CoreAcquireProtocolLock ();\r
\r
  for ( Link =  mProtocolDatabase.ForwardLink; \r
        Link != &mProtocolDatabase; \r
        Link =  Link->ForwardLink) {\r
\r
    ProtEntry = CR(Link, PROTOCOL_ENTRY, AllEntries, PROTOCOL_ENTRY_SIGNATURE);\r
\r
    for ( NotifyLink =  ProtEntry->Notify.ForwardLink; \r
          NotifyLink != &ProtEntry->Notify; \r
          NotifyLink =  NotifyLink->ForwardLink) {\r
\r
      ProtNotify = CR(NotifyLink, PROTOCOL_NOTIFY, Link, PROTOCOL_NOTIFY_SIGNATURE);\r
\r
      if (ProtNotify->Event == Event) {\r
        RemoveEntryList(&ProtNotify->Link);\r
        CoreFreePool(ProtNotify);\r
        CoreReleaseProtocolLock ();\r
        return EFI_SUCCESS;\r
      }\r
    }\r
  }\r
\r
  CoreReleaseProtocolLock ();\r
  return EFI_NOT_FOUND;\r
}\r
\r
\r
EFI_STATUS\r
CoreUnregisterProtocolNotify (\r
  IN EFI_EVENT      Event\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  Removes all the events in the protocol database that match Event.\r
\r
Arguments:\r
  \r
  Event   - The event to search for in the protocol database.\r
\r
Returns:\r
\r
  EFI_SUCCESS when done searching the entire database.\r
\r
--*/\r
{\r
  EFI_STATUS       Status;\r
\r
  do {\r
    Status = CoreUnregisterProtocolNotifyEvent (Event);\r
  } while (!EFI_ERROR (Status));\r
\r
  return EFI_SUCCESS;\r
}\r
\r
\r
\r
EFI_STATUS\r
EFIAPI\r
CoreInstallProtocolInterface (\r
  IN OUT EFI_HANDLE     *UserHandle,\r
  IN EFI_GUID           *Protocol,\r
  IN EFI_INTERFACE_TYPE InterfaceType,\r
  IN VOID               *Interface\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  Wrapper function to CoreInstallProtocolInterfaceNotify.  This is the public API which\r
  Calls the private one which contains a BOOLEAN parameter for notifications\r
\r
Arguments:\r
\r
  UserHandle     - The handle to install the protocol handler on,\r
                    or NULL if a new handle is to be allocated\r
\r
  Protocol       - The protocol to add to the handle\r
\r
  InterfaceType  - Indicates whether Interface is supplied in native form.\r
\r
  Interface      - The interface for the protocol being added\r
\r
Returns:\r
\r
  Status code    \r
\r
--*/\r
{\r
  return CoreInstallProtocolInterfaceNotify (\r
            UserHandle, \r
            Protocol, \r
            InterfaceType, \r
            Interface, \r
            TRUE\r
            );\r
}\r
\r
EFI_STATUS\r
CoreInstallProtocolInterfaceNotify (\r
  IN OUT EFI_HANDLE     *UserHandle,\r
  IN EFI_GUID           *Protocol,\r
  IN EFI_INTERFACE_TYPE InterfaceType,\r
  IN VOID               *Interface,\r
  IN BOOLEAN            Notify\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  Installs a protocol interface into the boot services environment.\r
\r
Arguments:\r
\r
  UserHandle     - The handle to install the protocol handler on,\r
                   or NULL if a new handle is to be allocated\r
\r
  Protocol       - The protocol to add to the handle\r
\r
  InterfaceType  - Indicates whether Interface is supplied in native form.\r
\r
  Interface      - The interface for the protocol being added\r
\r
  Notify         - indicates whether notify the notification list \r
                   for this protocol\r
\r
Returns:\r
\r
  EFI_INVALID_PARAMETER     - Invalid parameter\r
  \r
  EFI_OUT_OF_RESOURCES       - No enough buffer to allocate\r
  \r
  EFI_SUCCESS               - Protocol interface successfully installed\r
\r
--*/\r
{\r
  PROTOCOL_INTERFACE  *Prot;\r
  PROTOCOL_ENTRY      *ProtEntry;\r
  IHANDLE             *Handle;\r
  EFI_STATUS          Status;\r
  VOID                *ExistingInterface;\r
\r
  //\r
  // returns EFI_INVALID_PARAMETER if InterfaceType is invalid.\r
  // Also added check for invalid UserHandle and Protocol pointers.\r
  //\r
  if (UserHandle == NULL || Protocol == NULL) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  if (InterfaceType != EFI_NATIVE_INTERFACE) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  //\r
  // Print debug message\r
  //\r
  DEBUG((EFI_D_ERROR | EFI_D_INFO, "InstallProtocolInterface: %g %x\n", Protocol, Interface));\r
\r
  Status = EFI_OUT_OF_RESOURCES;\r
  Prot = NULL;\r
  Handle = NULL;\r
\r
  ASSERT (NULL != gBS);\r
\r
  if (*UserHandle != NULL_HANDLE) {\r
    Status = CoreHandleProtocol (*UserHandle, Protocol, (VOID **)&ExistingInterface);\r
    if (!EFI_ERROR (Status)) {\r
      return EFI_INVALID_PARAMETER;\r
    }\r
  }\r
\r
  //\r
  // Lock the protocol database \r
  //\r
  CoreAcquireProtocolLock ();\r
\r
  //\r
  // Lookup the Protocol Entry for the requested protocol\r
  //\r
  ProtEntry = CoreFindProtocolEntry (Protocol, TRUE);\r
  if (ProtEntry == NULL) {\r
    goto Done;\r
  }\r
\r
  //\r
  // Allocate a new protocol interface structure\r
  //\r
  Prot = CoreAllocateZeroBootServicesPool (sizeof(PROTOCOL_INTERFACE));\r
  if (Prot == NULL) {\r
    Status = EFI_OUT_OF_RESOURCES;\r
    goto Done;\r
  }\r
\r
  //\r
  // If caller didn't supply a handle, allocate a new one\r
  //\r
  Handle = (IHANDLE *)*UserHandle;\r
  if (Handle == NULL) {\r
    Handle = CoreAllocateZeroBootServicesPool (sizeof(IHANDLE));\r
    if (Handle == NULL) {\r
      Status = EFI_OUT_OF_RESOURCES;\r
      goto Done;\r
    }\r
\r
    //\r
    // Initialize new handler structure\r
    //\r
    Handle->Signature = EFI_HANDLE_SIGNATURE;\r
    InitializeListHead (&Handle->Protocols);\r
\r
    //\r
    // Initialize the Key to show that the handle has been created/modified\r
    //\r
    gHandleDatabaseKey++;\r
    Handle->Key = gHandleDatabaseKey;\r
\r
    //\r
    // Add this handle to the list global list of all handles\r
    // in the system\r
    //\r
    InsertTailList (&gHandleList, &Handle->AllHandles);\r
  } \r
\r
  Status = CoreValidateHandle (Handle);\r
  if (EFI_ERROR (Status)) {\r
    goto Done;\r
  }\r
\r
  //\r
  // Each interface that is added must be unique\r
  //\r
  ASSERT (CoreFindProtocolInterface (Handle, Protocol, Interface) == NULL);\r
\r
  //\r
  // Initialize the protocol interface structure\r
  //\r
  Prot->Signature = PROTOCOL_INTERFACE_SIGNATURE;\r
  Prot->Handle = Handle;\r
  Prot->Protocol = ProtEntry;\r
  Prot->Interface = Interface;\r
\r
  //\r
  // Initalize OpenProtocol Data base\r
  //\r
  InitializeListHead (&Prot->OpenList);\r
  Prot->OpenListCount = 0;\r
\r
  //\r
  // Add this protocol interface to the head of the supported \r
  // protocol list for this handle\r
  //\r
  InsertHeadList (&Handle->Protocols, &Prot->Link);\r
\r
  //\r
  // Add this protocol interface to the tail of the \r
  // protocol entry\r
  // \r
  InsertTailList (&ProtEntry->Protocols, &Prot->ByProtocol);\r
\r
  //\r
  // Notify the notification list for this protocol \r
  //\r
  if (Notify) {\r
    CoreNotifyProtocolEntry (ProtEntry);\r
  }\r
  Status = EFI_SUCCESS;\r
\r
Done:\r
  //\r
  // Done, unlock the database and return\r
  //\r
  CoreReleaseProtocolLock ();\r
  if (!EFI_ERROR (Status)) {\r
    //\r
    // Return the new handle back to the caller\r
    //\r
    *UserHandle = Handle;\r
  } else {\r
    //\r
    // There was an error, clean up\r
    //\r
    if (Prot != NULL) {\r
      CoreFreePool (Prot);\r
    }\r
  }\r
\r
  return Status;\r
}\r
\r
\r
\r
EFI_STATUS\r
EFIAPI\r
CoreInstallMultipleProtocolInterfaces (\r
  IN OUT EFI_HANDLE           *Handle,\r
  ...\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  Installs a list of protocol interface into the boot services environment.\r
  This function calls InstallProtocolInterface() in a loop. If any error\r
  occures all the protocols added by this function are removed. This is \r
  basically a lib function to save space.\r
\r
Arguments:\r
\r
  Handle      - The handle to install the protocol handlers on,\r
                or NULL if a new handle is to be allocated\r
  ...         - EFI_GUID followed by protocol instance. A NULL terminates the \r
                list. The pairs are the arguments to InstallProtocolInterface().\r
                All the protocols are added to Handle.\r
\r
Returns:\r
\r
  EFI_INVALID_PARAMETER       - Handle is NULL.\r
  \r
  EFI_SUCCESS                 - Protocol interfaces successfully installed.\r
\r
--*/\r
{\r
  VA_LIST                   args;\r
  EFI_STATUS                Status;\r
  EFI_GUID                  *Protocol;\r
  VOID                      *Interface;\r
  EFI_TPL                   OldTpl;\r
  UINTN                     Index;\r
  EFI_HANDLE                OldHandle;\r
  EFI_HANDLE                DeviceHandle;\r
  EFI_DEVICE_PATH_PROTOCOL  *DevicePath;\r
  \r
  if (Handle == NULL) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
  \r
  //\r
  // Syncronize with notifcations. \r
  // \r
  OldTpl = CoreRaiseTpl (EFI_TPL_NOTIFY);\r
  OldHandle = *Handle;\r
\r
  //\r
  // Check for duplicate device path and install the protocol interfaces\r
  //\r
  VA_START (args, Handle);\r
  for (Index = 0, Status = EFI_SUCCESS; !EFI_ERROR (Status); Index++) {\r
    //\r
    // If protocol is NULL, then it's the end of the list\r
    //\r
    Protocol = VA_ARG (args, EFI_GUID *);\r
    if (Protocol == NULL) {\r
      break;\r
    }\r
\r
    Interface = VA_ARG (args, VOID *);\r
\r
    //\r
    // Make sure you are installing on top a device path that has already been added.\r
    //\r
    if (CompareGuid (Protocol, &gEfiDevicePathProtocolGuid)) {\r
      DeviceHandle = NULL;\r
      DevicePath   = Interface;\r
      Status = CoreLocateDevicePath (&gEfiDevicePathProtocolGuid, &DevicePath, &DeviceHandle);\r
      if (!EFI_ERROR (Status) && (DeviceHandle != NULL_HANDLE) && IsDevicePathEnd(DevicePath)) {\r
        Status = EFI_ALREADY_STARTED;\r
        continue;\r
      }\r
    }\r
  \r
    //\r
    // Install it\r
    //\r
    Status = CoreInstallProtocolInterface (Handle, Protocol, EFI_NATIVE_INTERFACE, Interface);\r
  }\r
  \r
  //\r
  // If there was an error, remove all the interfaces that were installed without any errors\r
  //\r
  if (EFI_ERROR (Status)) {\r
    //\r
    // Reset the va_arg back to the first argument.\r
    //\r
    VA_START (args, Handle);\r
    for (; Index > 1; Index--) {\r
      Protocol = VA_ARG (args, EFI_GUID *);\r
      Interface = VA_ARG (args, VOID *);\r
      CoreUninstallProtocolInterface (*Handle, Protocol, Interface);\r
    }        \r
    *Handle = OldHandle;\r
  }\r
\r
  //\r
  // Done\r
  //\r
  CoreRestoreTpl (OldTpl);\r
  return Status;\r
}\r
\r
EFI_STATUS\r
CoreDisconnectControllersUsingProtocolInterface (\r
  IN EFI_HANDLE           UserHandle,\r
  IN PROTOCOL_INTERFACE   *Prot\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  Attempts to disconnect all drivers that are using the protocol interface being queried.\r
  If failed, reconnect all drivers disconnected.\r
  \r
  Note: This function doesn't do parameters checking, it's caller's responsibility \r
        to pass in valid parameters.\r
\r
Arguments:\r
\r
  UserHandle  - The handle on which the protocol is installed \r
  Prot        - The protocol to disconnect drivers from\r
\r
Returns:\r
\r
  EFI_SUCCESS       - Drivers using the protocol interface are all disconnected\r
  EFI_ACCESS_DENIED - Failed to disconnect one or all of the drivers\r
\r
--*/\r
{\r
  EFI_STATUS            Status;\r
  BOOLEAN               ItemFound;\r
  LIST_ENTRY            *Link;\r
  OPEN_PROTOCOL_DATA    *OpenData;\r
\r
  Status = EFI_SUCCESS;\r
  \r
  //\r
  // Attempt to disconnect all drivers from this protocol interface\r
  //\r
  do {\r
    ItemFound = FALSE;\r
    for ( Link = Prot->OpenList.ForwardLink;\r
          (Link != &Prot->OpenList) && !ItemFound;\r
          Link = Link->ForwardLink ) {\r
      OpenData = CR (Link, OPEN_PROTOCOL_DATA, Link, OPEN_PROTOCOL_DATA_SIGNATURE);\r
      if (OpenData->Attributes & EFI_OPEN_PROTOCOL_BY_DRIVER) {\r
        ItemFound = TRUE;\r
        CoreReleaseProtocolLock ();\r
        Status = CoreDisconnectController (UserHandle, OpenData->AgentHandle, NULL);\r
        CoreAcquireProtocolLock ();\r
        if (EFI_ERROR (Status)) {\r
           ItemFound = FALSE;\r
           break;\r
        }\r
      }\r
    }\r
  } while (ItemFound);\r
\r
  if (!EFI_ERROR (Status)) {\r
    //\r
    // Attempt to remove BY_HANDLE_PROTOOCL and GET_PROTOCOL and TEST_PROTOCOL Open List items\r
    //\r
    do {\r
      ItemFound = FALSE;\r
      for ( Link = Prot->OpenList.ForwardLink;\r
            (Link != &Prot->OpenList) && !ItemFound;\r
            Link = Link->ForwardLink ) {\r
        OpenData = CR (Link, OPEN_PROTOCOL_DATA, Link, OPEN_PROTOCOL_DATA_SIGNATURE);\r
        if (OpenData->Attributes & \r
            (EFI_OPEN_PROTOCOL_BY_HANDLE_PROTOCOL | EFI_OPEN_PROTOCOL_GET_PROTOCOL | EFI_OPEN_PROTOCOL_TEST_PROTOCOL)) {\r
          ItemFound = TRUE;\r
          RemoveEntryList (&OpenData->Link);  \r
          Prot->OpenListCount--;\r
          CoreFreePool (OpenData);\r
        }\r
      }\r
    } while (ItemFound);\r
  }\r
\r
  //\r
  // If there are errors or still has open items in the list, then reconnect all the drivers and return an error\r
  //\r
  if (EFI_ERROR (Status) || (Prot->OpenListCount > 0)) {\r
    CoreReleaseProtocolLock ();\r
    CoreConnectController (UserHandle, NULL, NULL, TRUE);\r
    CoreAcquireProtocolLock ();\r
    Status = EFI_ACCESS_DENIED;\r
  }\r
\r
  return Status;\r
}\r
\r
\r
EFI_STATUS\r
EFIAPI\r
CoreUninstallProtocolInterface (\r
  IN EFI_HANDLE       UserHandle,\r
  IN EFI_GUID         *Protocol,\r
  IN VOID             *Interface\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  Uninstalls all instances of a protocol:interfacer from a handle. \r
  If the last protocol interface is remove from the handle, the \r
  handle is freed.\r
\r
Arguments:\r
\r
  UserHandle      - The handle to remove the protocol handler from\r
\r
  Protocol        - The protocol, of protocol:interface, to remove\r
\r
  Interface       - The interface, of protocol:interface, to remove\r
\r
Returns:\r
\r
  EFI_INVALID_PARAMETER       - Protocol is NULL.\r
  \r
  EFI_SUCCESS                 - Protocol interface successfully uninstalled.\r
\r
--*/\r
{\r
  EFI_STATUS            Status;\r
  IHANDLE               *Handle;\r
  PROTOCOL_INTERFACE    *Prot;\r
\r
  //\r
  // Check that Protocol is valid\r
  //\r
  if (Protocol == NULL) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  //\r
  // Check that UserHandle is a valid handle\r
  //\r
  Status = CoreValidateHandle (UserHandle);\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
\r
  //\r
  // Lock the protocol database\r
  //\r
  CoreAcquireProtocolLock ();\r
\r
  //\r
  // Check that Protocol exists on UserHandle, and Interface matches the interface in the database\r
  //\r
  Prot = CoreFindProtocolInterface (UserHandle, Protocol, Interface);\r
  if (Prot == NULL) {\r
    Status = EFI_NOT_FOUND;\r
    goto Done;\r
  }\r
\r
  //\r
  // Attempt to disconnect all drivers that are using the protocol interface that is about to be removed\r
  //\r
  Status = CoreDisconnectControllersUsingProtocolInterface (\r
             UserHandle,\r
             Prot\r
             );\r
  if (EFI_ERROR (Status)) {\r
    //\r
    // One or more drivers refused to release, so return the error\r
    //\r
    goto Done;\r
  }\r
\r
  //\r
  // Remove the protocol interface from the protocol\r
  //\r
  Status = EFI_NOT_FOUND;\r
  Handle = (IHANDLE *)UserHandle;\r
  Prot   = CoreRemoveInterfaceFromProtocol (Handle, Protocol, Interface);\r
\r
  if (Prot != NULL) {\r
    //\r
    // Update the Key to show that the handle has been created/modified\r
    //\r
    gHandleDatabaseKey++;\r
    Handle->Key = gHandleDatabaseKey;\r
    \r
    //\r
    // Remove the protocol interface from the handle\r
    //\r
    RemoveEntryList (&Prot->Link);\r
\r
    //\r
    // Free the memory\r
    //\r
    Prot->Signature = 0;\r
    CoreFreePool (Prot);\r
    Status = EFI_SUCCESS;\r
  }\r
\r
  //\r
  // If there are no more handlers for the handle, free the handle\r
  //\r
  if (IsListEmpty (&Handle->Protocols)) {\r
    Handle->Signature = 0;\r
    RemoveEntryList (&Handle->AllHandles);\r
    CoreFreePool (Handle);\r
  }\r
\r
Done:  \r
  //\r
  // Done, unlock the database and return\r
  //\r
  CoreReleaseProtocolLock ();\r
  return Status;\r
}\r
\r
\r
\r
EFI_STATUS\r
EFIAPI\r
CoreUninstallMultipleProtocolInterfaces (\r
  IN EFI_HANDLE           Handle,\r
  ...\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  Uninstalls a list of protocol interface in the boot services environment. \r
  This function calls UnisatllProtocolInterface() in a loop. This is \r
  basically a lib function to save space.\r
\r
Arguments:\r
\r
  Handle      - The handle to uninstall the protocol\r
\r
  ...         - EFI_GUID followed by protocol instance. A NULL terminates the \r
                list. The pairs are the arguments to UninstallProtocolInterface().\r
                All the protocols are added to Handle.\r
\r
Returns:\r
\r
  Status code    \r
\r
--*/\r
{\r
  EFI_STATUS      Status;\r
  VA_LIST         args;\r
  EFI_GUID        *Protocol;\r
  VOID            *Interface;\r
  UINTN           Index;\r
\r
  VA_START (args, Handle);\r
  for (Index = 0, Status = EFI_SUCCESS; !EFI_ERROR (Status); Index++) {\r
    //\r
    // If protocol is NULL, then it's the end of the list\r
    //\r
    Protocol = VA_ARG (args, EFI_GUID *);\r
    if (Protocol == NULL) {\r
      break;\r
    }\r
\r
    Interface = VA_ARG (args, VOID *);\r
\r
    //\r
    // Uninstall it\r
    //\r
    Status = CoreUninstallProtocolInterface (Handle, Protocol, Interface);\r
  }\r
\r
  //\r
  // If there was an error, add all the interfaces that were\r
  // uninstalled without any errors\r
  //\r
  if (EFI_ERROR (Status)) {\r
    //\r
    // Reset the va_arg back to the first argument.\r
    //\r
    VA_START (args, Handle);\r
    for (; Index > 1; Index--) {\r
      Protocol = VA_ARG(args, EFI_GUID *);\r
      Interface = VA_ARG(args, VOID *);\r
      CoreInstallProtocolInterface (&Handle, Protocol, EFI_NATIVE_INTERFACE, Interface);\r
    }        \r
  }\r
\r
  return Status;\r
}    \r
\r
PROTOCOL_INTERFACE  *\r
CoreGetProtocolInterface (\r
  IN  EFI_HANDLE                UserHandle,\r
  IN  EFI_GUID                  *Protocol\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  Locate a certain GUID protocol interface in a Handle's protocols.\r
\r
Arguments:\r
\r
  UserHandle  - The handle to obtain the protocol interface on\r
\r
  Protocol    - The GUID of the protocol \r
\r
Returns:\r
\r
  The requested protocol interface for the handle\r
  \r
--*/  \r
{\r
  EFI_STATUS          Status;\r
  PROTOCOL_ENTRY      *ProtEntry;\r
  PROTOCOL_INTERFACE  *Prot;\r
  IHANDLE             *Handle;\r
  LIST_ENTRY          *Link;\r
\r
  Status = CoreValidateHandle (UserHandle);\r
  if (EFI_ERROR (Status)) {\r
    return NULL;\r
  }\r
  \r
  Handle = (IHANDLE *)UserHandle;\r
\r
  //\r
  // Look at each protocol interface for a match\r
  //\r
  for (Link = Handle->Protocols.ForwardLink; Link != &Handle->Protocols; Link = Link->ForwardLink) {\r
    Prot = CR(Link, PROTOCOL_INTERFACE, Link, PROTOCOL_INTERFACE_SIGNATURE);\r
    ProtEntry = Prot->Protocol;\r
    if (CompareGuid (&ProtEntry->ProtocolID, Protocol)) {\r
      return Prot;\r
    }\r
  }\r
  return NULL;\r
}\r
\r
\r
EFI_STATUS\r
EFIAPI\r
CoreHandleProtocol (\r
  IN EFI_HANDLE       UserHandle,\r
  IN EFI_GUID         *Protocol,\r
  OUT VOID            **Interface\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  Queries a handle to determine if it supports a specified protocol.\r
\r
Arguments:\r
\r
  UserHandle  - The handle being queried.\r
\r
  Protocol    - The published unique identifier of the protocol.\r
\r
  Interface   - Supplies the address where a pointer to the corresponding Protocol\r
               Interface is returned.\r
\r
Returns:\r
\r
  The requested protocol interface for the handle\r
  \r
--*/  \r
{\r
  return CoreOpenProtocol (\r
          UserHandle,     \r
          Protocol, \r
          Interface, \r
          gDxeCoreImageHandle, \r
          NULL,     \r
          EFI_OPEN_PROTOCOL_BY_HANDLE_PROTOCOL\r
          );\r
}\r
\r
\r
EFI_STATUS\r
EFIAPI\r
CoreOpenProtocol (\r
  IN  EFI_HANDLE                UserHandle,\r
  IN  EFI_GUID                  *Protocol,\r
  OUT VOID                      **Interface OPTIONAL,\r
  IN  EFI_HANDLE                ImageHandle,\r
  IN  EFI_HANDLE                ControllerHandle,\r
  IN  UINT32                    Attributes\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  Locates the installed protocol handler for the handle, and\r
  invokes it to obtain the protocol interface. Usage information\r
  is registered in the protocol data base.\r
\r
Arguments:\r
\r
  UserHandle     - The handle to obtain the protocol interface on\r
\r
  Protocol       - The ID of the protocol \r
\r
  Interface      - The location to return the protocol interface\r
\r
  ImageHandle       - The handle of the Image that is opening the protocol interface\r
                    specified by Protocol and Interface.\r
  \r
  ControllerHandle  - The controller handle that is requiring this interface.\r
\r
  Attributes     - The open mode of the protocol interface specified by Handle\r
                    and Protocol.\r
\r
Returns:\r
\r
  EFI_INVALID_PARAMETER       - Protocol is NULL.\r
  \r
  EFI_SUCCESS                 - Get the protocol interface.\r
  \r
--*/\r
{\r
  EFI_STATUS          Status;\r
  PROTOCOL_INTERFACE  *Prot;\r
  LIST_ENTRY          *Link;\r
  OPEN_PROTOCOL_DATA  *OpenData;\r
  BOOLEAN             ByDriver;\r
  BOOLEAN             Exclusive;\r
  BOOLEAN             Disconnect;\r
  BOOLEAN             ExactMatch;\r
\r
  //\r
  // Check for invalid Protocol\r
  //\r
  if (Protocol == NULL) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  //\r
  // Check for invalid Interface\r
  //\r
  if (Attributes != EFI_OPEN_PROTOCOL_TEST_PROTOCOL) {\r
    if (Interface == NULL) {\r
      return EFI_INVALID_PARAMETER;\r
    } else {\r
      *Interface = NULL;\r
    }\r
  }\r
  \r
  //\r
  // Check for invalid UserHandle\r
  //\r
  Status = CoreValidateHandle (UserHandle);\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
\r
  //\r
  // Check for invalid Attributes\r
  //\r
  switch (Attributes) {\r
  case EFI_OPEN_PROTOCOL_BY_CHILD_CONTROLLER :\r
    Status = CoreValidateHandle (ImageHandle);\r
    if (EFI_ERROR (Status)) {\r
      return Status;\r
    }\r
    Status = CoreValidateHandle (ControllerHandle);\r
    if (EFI_ERROR (Status)) {\r
      return Status;\r
    }\r
    if (UserHandle == ControllerHandle) {\r
      return EFI_INVALID_PARAMETER;\r
    }\r
    break;\r
  case EFI_OPEN_PROTOCOL_BY_DRIVER :\r
  case EFI_OPEN_PROTOCOL_BY_DRIVER | EFI_OPEN_PROTOCOL_EXCLUSIVE :\r
    Status = CoreValidateHandle (ImageHandle);\r
    if (EFI_ERROR (Status)) {\r
      return Status;\r
    }\r
    Status = CoreValidateHandle (ControllerHandle);\r
    if (EFI_ERROR (Status)) {\r
      return Status;\r
    }\r
    break;\r
  case EFI_OPEN_PROTOCOL_EXCLUSIVE :\r
    Status = CoreValidateHandle (ImageHandle);\r
    if (EFI_ERROR (Status)) {\r
      return Status;\r
    }\r
    break;\r
  case EFI_OPEN_PROTOCOL_BY_HANDLE_PROTOCOL :\r
  case EFI_OPEN_PROTOCOL_GET_PROTOCOL :\r
  case EFI_OPEN_PROTOCOL_TEST_PROTOCOL :\r
    break;\r
  default:\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  //\r
  // Lock the protocol database\r
  //\r
  CoreAcquireProtocolLock ();\r
\r
  //\r
  // Look at each protocol interface for a match\r
  //\r
  Prot = CoreGetProtocolInterface (UserHandle, Protocol);\r
  if (Prot == NULL) {\r
    Status = EFI_UNSUPPORTED;\r
    goto Done;\r
  }\r
\r
  //\r
  // This is the protocol interface entry for this protocol\r
  //    \r
  if (Attributes != EFI_OPEN_PROTOCOL_TEST_PROTOCOL) {\r
    *Interface = Prot->Interface;\r
  }\r
  Status = EFI_SUCCESS;\r
\r
  ByDriver        = FALSE;\r
  Exclusive       = FALSE;\r
  for ( Link = Prot->OpenList.ForwardLink; Link != &Prot->OpenList; Link = Link->ForwardLink) {\r
    OpenData = CR (Link, OPEN_PROTOCOL_DATA, Link, OPEN_PROTOCOL_DATA_SIGNATURE);\r
    ExactMatch =  (BOOLEAN)((OpenData->AgentHandle == ImageHandle) && \r
                            (OpenData->Attributes == Attributes)  &&\r
                            (OpenData->ControllerHandle == ControllerHandle));\r
    if (OpenData->Attributes & EFI_OPEN_PROTOCOL_BY_DRIVER) {\r
      ByDriver = TRUE;\r
      if (ExactMatch) {\r
        Status = EFI_ALREADY_STARTED;\r
        goto Done;\r
      }\r
    }\r
    if (OpenData->Attributes & EFI_OPEN_PROTOCOL_EXCLUSIVE) {\r
      Exclusive = TRUE;\r
    } else if (ExactMatch) {\r
      OpenData->OpenCount++;\r
      Status = EFI_SUCCESS;\r
      goto Done;\r
    }\r
  }\r
\r
  //\r
  // ByDriver  TRUE  -> A driver is managing (UserHandle, Protocol)\r
  // ByDriver  FALSE -> There are no drivers managing (UserHandle, Protocol)\r
  // Exclusive TRUE  -> Something has exclusive access to (UserHandle, Protocol)\r
  // Exclusive FALSE -> Nothing has exclusive access to (UserHandle, Protocol)\r
  //\r
\r
  switch (Attributes) {\r
  case EFI_OPEN_PROTOCOL_BY_DRIVER :\r
    if (Exclusive || ByDriver) {\r
      Status = EFI_ACCESS_DENIED;\r
      goto Done;\r
    }\r
    break;\r
  case EFI_OPEN_PROTOCOL_BY_DRIVER | EFI_OPEN_PROTOCOL_EXCLUSIVE :\r
  case EFI_OPEN_PROTOCOL_EXCLUSIVE :\r
    if (Exclusive) {\r
      Status = EFI_ACCESS_DENIED;\r
      goto Done;\r
    }\r
    if (ByDriver) {\r
      do {\r
        Disconnect = FALSE;\r
        for ( Link = Prot->OpenList.ForwardLink; (Link != &Prot->OpenList) && (!Disconnect); Link = Link->ForwardLink) {\r
          OpenData = CR (Link, OPEN_PROTOCOL_DATA, Link, OPEN_PROTOCOL_DATA_SIGNATURE);\r
          if (OpenData->Attributes & EFI_OPEN_PROTOCOL_BY_DRIVER) {\r
            Disconnect = TRUE;\r
            CoreReleaseProtocolLock ();\r
            Status = CoreDisconnectController (UserHandle, OpenData->AgentHandle, NULL);\r
            CoreAcquireProtocolLock ();\r
            if (EFI_ERROR (Status)) {\r
              Status = EFI_ACCESS_DENIED;\r
              goto Done;\r
            }\r
          }\r
        }\r
      } while (Disconnect);\r
    } \r
    break;\r
  case EFI_OPEN_PROTOCOL_BY_CHILD_CONTROLLER :\r
  case EFI_OPEN_PROTOCOL_BY_HANDLE_PROTOCOL :\r
  case EFI_OPEN_PROTOCOL_GET_PROTOCOL :\r
  case EFI_OPEN_PROTOCOL_TEST_PROTOCOL :\r
    break;\r
  }\r
\r
  if (ImageHandle == NULL) {\r
    Status = EFI_SUCCESS;\r
    goto Done;\r
  }\r
  //\r
  // Create new entry\r
  //\r
  OpenData = CoreAllocateBootServicesPool (sizeof(OPEN_PROTOCOL_DATA));\r
  if (OpenData == NULL) {\r
    Status = EFI_OUT_OF_RESOURCES;\r
  } else {\r
    OpenData->Signature         = OPEN_PROTOCOL_DATA_SIGNATURE;\r
    OpenData->AgentHandle       = ImageHandle;\r
    OpenData->ControllerHandle  = ControllerHandle;\r
    OpenData->Attributes        = Attributes;\r
    OpenData->OpenCount         = 1;\r
    InsertTailList (&Prot->OpenList, &OpenData->Link);\r
    Prot->OpenListCount++;\r
    Status = EFI_SUCCESS;\r
  }\r
\r
Done:\r
  //\r
  // Done. Release the database lock are return\r
  //\r
  CoreReleaseProtocolLock ();\r
  return Status;\r
}\r
\r
\r
EFI_STATUS\r
EFIAPI\r
CoreCloseProtocol (\r
  IN  EFI_HANDLE                UserHandle,\r
  IN  EFI_GUID                  *Protocol,\r
  IN  EFI_HANDLE                AgentHandle,\r
  IN  EFI_HANDLE                ControllerHandle  \r
  )\r
/*++\r
\r
Routine Description:\r
\r
  Closes a protocol on a handle that was opened using OpenProtocol().\r
\r
Arguments:\r
\r
  UserHandle       -  The handle for the protocol interface that was previously opened\r
                      with OpenProtocol(), and is now being closed.\r
  Protocol         -  The published unique identifier of the protocol. It is the caller's\r
                      responsibility to pass in a valid GUID.\r
  AgentHandle      -  The handle of the agent that is closing the protocol interface.\r
  ControllerHandle -  If the agent that opened a protocol is a driver that follows the\r
                      EFI Driver Model, then this parameter is the controller handle\r
                      that required the protocol interface. If the agent does not follow\r
                      the EFI Driver Model, then this parameter is optional and may be NULL.\r
\r
Returns:\r
\r
  EFI_SUCCESS             - The protocol instance was closed.\r
  EFI_INVALID_PARAMETER   - Handle, AgentHandle or ControllerHandle is not a valid EFI_HANDLE. \r
  EFI_NOT_FOUND           - Can not find the specified protocol or AgentHandle.\r
  \r
--*/\r
{\r
  EFI_STATUS          Status;\r
  PROTOCOL_INTERFACE  *ProtocolInterface;\r
  LIST_ENTRY          *Link;\r
  OPEN_PROTOCOL_DATA  *OpenData;\r
\r
  //\r
  // Check for invalid parameters\r
  //\r
  Status = CoreValidateHandle (UserHandle);\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
  Status = CoreValidateHandle (AgentHandle);\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
  if (ControllerHandle != NULL_HANDLE) {\r
    Status = CoreValidateHandle (ControllerHandle);\r
    if (EFI_ERROR (Status)) {\r
      return Status;\r
    }\r
  }\r
  if (Protocol == NULL) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  //\r
  // Lock the protocol database\r
  //\r
  CoreAcquireProtocolLock ();\r
\r
  //\r
  // Look at each protocol interface for a match\r
  //\r
  Status = EFI_NOT_FOUND;\r
  ProtocolInterface = CoreGetProtocolInterface (UserHandle, Protocol);\r
  if (ProtocolInterface == NULL) {\r
    goto Done;\r
  }\r
\r
  //\r
  // Walk the Open data base looking for AgentHandle\r
  //\r
  Link = ProtocolInterface->OpenList.ForwardLink;\r
  while (Link != &ProtocolInterface->OpenList) {\r
    OpenData = CR (Link, OPEN_PROTOCOL_DATA, Link, OPEN_PROTOCOL_DATA_SIGNATURE);\r
    Link = Link->ForwardLink;\r
    if ((OpenData->AgentHandle == AgentHandle) && (OpenData->ControllerHandle == ControllerHandle)) {\r
        RemoveEntryList (&OpenData->Link);  \r
        ProtocolInterface->OpenListCount--;\r
        CoreFreePool (OpenData);\r
        Status = EFI_SUCCESS;\r
    }\r
  }\r
\r
Done:\r
  //\r
  // Done. Release the database lock and return.\r
  //\r
  CoreReleaseProtocolLock ();\r
  return Status;\r
}\r
\r
\r
\r
EFI_STATUS\r
EFIAPI\r
CoreOpenProtocolInformation (\r
  IN  EFI_HANDLE                          UserHandle,\r
  IN  EFI_GUID                            *Protocol,\r
  OUT EFI_OPEN_PROTOCOL_INFORMATION_ENTRY **EntryBuffer,\r
  OUT UINTN                               *EntryCount\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  Return information about Opened protocols in the system\r
\r
Arguments:\r
\r
  UserHandle  - The handle to close the protocol interface on\r
\r
  Protocol    - The ID of the protocol \r
\r
  EntryBuffer - A pointer to a buffer of open protocol information in the form of\r
                EFI_OPEN_PROTOCOL_INFORMATION_ENTRY structures.\r
\r
  EntryCount  - Number of EntryBuffer entries\r
\r
Returns:\r
\r
  \r
--*/\r
{\r
  EFI_STATUS                          Status;\r
  PROTOCOL_INTERFACE                  *ProtocolInterface;\r
  LIST_ENTRY                          *Link;\r
  OPEN_PROTOCOL_DATA                  *OpenData;\r
  EFI_OPEN_PROTOCOL_INFORMATION_ENTRY *Buffer;\r
  UINTN                               Count;\r
  UINTN                               Size;\r
\r
  *EntryBuffer = NULL;\r
  *EntryCount = 0;\r
\r
  //\r
  // Lock the protocol database\r
  //\r
  CoreAcquireProtocolLock ();\r
\r
  //\r
  // Look at each protocol interface for a match\r
  //\r
  Status = EFI_NOT_FOUND;\r
  ProtocolInterface = CoreGetProtocolInterface (UserHandle, Protocol);\r
  if (ProtocolInterface == NULL) {\r
    goto Done;\r
  }\r
\r
  //\r
  // Count the number of Open Entries\r
  //\r
  for ( Link = ProtocolInterface->OpenList.ForwardLink, Count = 0; \r
        (Link != &ProtocolInterface->OpenList) ;\r
        Link = Link->ForwardLink  ) {\r
    Count++;\r
  } \r
\r
  ASSERT (Count == ProtocolInterface->OpenListCount);\r
\r
  if (Count == 0) {\r
    Size = sizeof(EFI_OPEN_PROTOCOL_INFORMATION_ENTRY);\r
  } else {\r
    Size = Count * sizeof(EFI_OPEN_PROTOCOL_INFORMATION_ENTRY);\r
  }\r
\r
  Buffer = CoreAllocateBootServicesPool (Size);\r
  if (Buffer == NULL) {\r
    Status = EFI_OUT_OF_RESOURCES;\r
    goto Done;\r
  }\r
\r
  Status = EFI_SUCCESS;\r
  for ( Link = ProtocolInterface->OpenList.ForwardLink, Count = 0; \r
        (Link != &ProtocolInterface->OpenList);\r
        Link = Link->ForwardLink, Count++  ) {\r
    OpenData = CR (Link, OPEN_PROTOCOL_DATA, Link, OPEN_PROTOCOL_DATA_SIGNATURE);\r
\r
    Buffer[Count].AgentHandle      = OpenData->AgentHandle;\r
    Buffer[Count].ControllerHandle = OpenData->ControllerHandle;\r
    Buffer[Count].Attributes       = OpenData->Attributes;\r
    Buffer[Count].OpenCount        = OpenData->OpenCount;\r
  } \r
\r
  *EntryBuffer = Buffer;\r
  *EntryCount = Count;\r
        \r
Done:\r
  //\r
  // Done. Release the database lock are return\r
  //\r
  CoreReleaseProtocolLock ();\r
  return Status;\r
}\r
\r
\r
\r
EFI_STATUS\r
EFIAPI\r
CoreProtocolsPerHandle (\r
  IN EFI_HANDLE       UserHandle,\r
  OUT EFI_GUID        ***ProtocolBuffer,\r
  OUT UINTN           *ProtocolBufferCount\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  Retrieves the list of protocol interface GUIDs that are installed on a handle in a buffer allocated\r
 from pool.\r
\r
Arguments:\r
\r
  UserHandle           - The handle from which to retrieve the list of protocol interface\r
                          GUIDs.\r
\r
  ProtocolBuffer       - A pointer to the list of protocol interface GUID pointers that are\r
                          installed on Handle.\r
\r
  ProtocolBufferCount  - A pointer to the number of GUID pointers present in\r
                          ProtocolBuffer.\r
\r
Returns:\r
  EFI_SUCCESS   -  The list of protocol interface GUIDs installed on Handle was returned in\r
                   ProtocolBuffer. The number of protocol interface GUIDs was\r
                   returned in ProtocolBufferCount.\r
  EFI_INVALID_PARAMETER   -  Handle is NULL.\r
  EFI_INVALID_PARAMETER   -  Handle is not a valid EFI_HANDLE.\r
  EFI_INVALID_PARAMETER   -  ProtocolBuffer is NULL.\r
  EFI_INVALID_PARAMETER   -  ProtocolBufferCount is NULL.\r
  EFI_OUT_OF_RESOURCES    -  There is not enough pool memory to store the results.\r
  \r
--*/\r
{\r
  EFI_STATUS                          Status;\r
  IHANDLE                             *Handle;\r
  PROTOCOL_INTERFACE                  *Prot;\r
  LIST_ENTRY                          *Link;\r
  UINTN                               ProtocolCount;\r
  EFI_GUID                            **Buffer;\r
\r
  Status = CoreValidateHandle (UserHandle);\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
\r
  Handle = (IHANDLE *)UserHandle;\r
\r
  if (ProtocolBuffer == NULL) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  if (ProtocolBufferCount == NULL) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  *ProtocolBufferCount = 0;\r
\r
  ProtocolCount = 0;\r
\r
  CoreAcquireProtocolLock ();\r
  \r
  for (Link = Handle->Protocols.ForwardLink; Link != &Handle->Protocols; Link = Link->ForwardLink) {\r
    ProtocolCount++;\r
  }\r
\r
  //\r
  // If there are no protocol interfaces installed on Handle, then Handle is not a valid EFI_HANDLE\r
  //\r
  if (ProtocolCount == 0) {\r
    Status = EFI_INVALID_PARAMETER;\r
    goto Done;\r
  }\r
\r
  Buffer = CoreAllocateBootServicesPool (sizeof (EFI_GUID *) * ProtocolCount);\r
  if (Buffer == NULL) {\r
    Status = EFI_OUT_OF_RESOURCES;\r
    goto Done;\r
  }\r
\r
  *ProtocolBuffer = Buffer;\r
  *ProtocolBufferCount = ProtocolCount;\r
\r
  for ( Link = Handle->Protocols.ForwardLink, ProtocolCount = 0;\r
        Link != &Handle->Protocols; \r
        Link = Link->ForwardLink, ProtocolCount++) {\r
    Prot = CR(Link, PROTOCOL_INTERFACE, Link, PROTOCOL_INTERFACE_SIGNATURE);\r
    Buffer[ProtocolCount] = &(Prot->Protocol->ProtocolID);\r
  }\r
  Status = EFI_SUCCESS;\r
\r
Done:\r
  CoreReleaseProtocolLock ();\r
  return Status;\r
}\r
\r
\r
UINT64\r
CoreGetHandleDatabaseKey (\r
  VOID\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  return handle database key.\r
\r
Arguments:\r
\r
  None\r
  \r
Returns:\r
  \r
  Handle database key.\r
  \r
--*/\r
{\r
  return gHandleDatabaseKey;\r
}\r
\r
\r
VOID\r
CoreConnectHandlesByKey (\r
  UINT64  Key\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  Go connect any handles that were created or modified while a image executed.\r
\r
Arguments:\r
\r
  Key  -  The Key to show that the handle has been created/modified\r
\r
Returns:\r
  \r
  None\r
--*/\r
{\r
  UINTN           Count;\r
  LIST_ENTRY      *Link;\r
  EFI_HANDLE      *HandleBuffer;\r
  IHANDLE         *Handle;\r
  UINTN           Index;\r
\r
  //\r
  // Lock the protocol database\r
  //\r
  CoreAcquireProtocolLock ();\r
\r
  for (Link = gHandleList.ForwardLink, Count = 0; Link != &gHandleList; Link = Link->ForwardLink) {\r
    Handle = CR (Link, IHANDLE, AllHandles, EFI_HANDLE_SIGNATURE);\r
    if (Handle->Key > Key) {\r
      Count++;\r
    }\r
  }\r
\r
  HandleBuffer = CoreAllocateBootServicesPool (Count * sizeof (EFI_HANDLE));\r
  if (HandleBuffer == NULL) {\r
    CoreReleaseProtocolLock ();\r
    return;\r
  }\r
  \r
  for (Link = gHandleList.ForwardLink, Count = 0; Link != &gHandleList; Link = Link->ForwardLink) {\r
    Handle = CR (Link, IHANDLE, AllHandles, EFI_HANDLE_SIGNATURE);\r
    if (Handle->Key > Key) {\r
      HandleBuffer[Count++] = Handle;\r
    }\r
  }\r
\r
  //\r
  // Unlock the protocol database\r
  //\r
  CoreReleaseProtocolLock ();\r
\r
  //\r
  // Connect all handles whose Key value is greater than Key\r
  //\r
  for (Index = 0; Index < Count; Index++) {\r
    CoreConnectController (HandleBuffer[Index], NULL, NULL, TRUE);\r
  }\r
  \r
  CoreFreePool(HandleBuffer);\r
}\r