Initial import.

[mirror_edk2.git] / MdePkg / Library / UefiDevicePathLib / UefiDevicePathLib.c

/** @file\r
  Device Path services. The thing to remember is device paths are built out of\r
  nodes. The device path is terminated by an end node that is length\r
  sizeof(EFI_DEVICE_PATH_PROTOCOL). That would be why there is sizeof(EFI_DEVICE_PATH_PROTOCOL)\r
  all over this file.\r
\r
  The only place where multi-instance device paths are supported is in\r
  environment varibles. Multi-instance device paths should never be placed\r
  on a Handle.\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:  UefiDevicePathLib.c\r
\r
**/\r
\r
/**\r
  This function returns the size, in bytes, \r
  of the device path data structure specified by DevicePath.\r
  If DevicePath is NULL, then 0 is returned.\r
\r
  @param  DevicePath A pointer to a device path data structure.\r
\r
  @return The size of a device path in bytes.\r
\r
**/\r
UINTN\r
EFIAPI\r
GetDevicePathSize (\r
  IN CONST EFI_DEVICE_PATH_PROTOCOL  *DevicePath\r
  )\r
{\r
  CONST EFI_DEVICE_PATH_PROTOCOL  *Start;\r
\r
  if (DevicePath == NULL) {\r
    return 0;\r
  }\r
\r
  //\r
  // Search for the end of the device path structure\r
  //\r
  Start = DevicePath;\r
  while (!EfiIsDevicePathEnd (DevicePath)) {\r
    DevicePath = EfiNextDevicePathNode (DevicePath);\r
  }\r
\r
  //\r
  // Compute the size and add back in the size of the end device path structure\r
  //\r
  return ((UINTN) DevicePath - (UINTN) Start) + sizeof (EFI_DEVICE_PATH_PROTOCOL);\r
}\r
\r
/**\r
  This function allocates space for a new copy of the device path\r
  specified by DevicePath.\r
\r
  @param  DevicePath A pointer to a device path data structure.\r
\r
  @return The duplicated device path.\r
\r
**/\r
EFI_DEVICE_PATH_PROTOCOL *\r
EFIAPI\r
DuplicateDevicePath (\r
  IN CONST EFI_DEVICE_PATH_PROTOCOL   *DevicePath\r
  )\r
{\r
  EFI_DEVICE_PATH_PROTOCOL  *NewDevicePath;\r
  UINTN                     Size;\r
\r
  //\r
  // Compute the size\r
  //\r
  Size = GetDevicePathSize (DevicePath);\r
  if (Size == 0) {\r
    return NULL;\r
  }\r
\r
  //\r
  // Allocate space for duplicate device path\r
  //\r
  NewDevicePath = AllocateCopyPool (Size, DevicePath);\r
\r
  return NewDevicePath;\r
}\r
\r
/**\r
  This function appends the device path SecondDevicePath\r
  to every device path instance in FirstDevicePath. \r
\r
  @param  FirstDevicePath A pointer to a device path data structure.\r
  \r
  @param  SecondDevicePath A pointer to a device path data structure.\r
\r
  @return A pointer to the new device path is returned.\r
  NULL is returned if space for the new device path could not be allocated from pool.\r
  It is up to the caller to free the memory used by FirstDevicePath and SecondDevicePath\r
  if they are no longer needed.\r
\r
**/\r
EFI_DEVICE_PATH_PROTOCOL *\r
EFIAPI\r
AppendDevicePath (\r
  IN CONST EFI_DEVICE_PATH_PROTOCOL  *FirstDevicePath,\r
  IN CONST EFI_DEVICE_PATH_PROTOCOL  *SecondDevicePath\r
  )\r
{\r
  UINTN                     Size;\r
  UINTN                     Size1;\r
  UINTN                     Size2;\r
  EFI_DEVICE_PATH_PROTOCOL  *NewDevicePath;\r
  EFI_DEVICE_PATH_PROTOCOL  *DevicePath2;\r
\r
  //\r
  // If there's only 1 path, just duplicate it\r
  //\r
  if (FirstDevicePath == NULL) {\r
    return DuplicateDevicePath (SecondDevicePath);\r
  }\r
\r
  if (SecondDevicePath == NULL) {\r
    return DuplicateDevicePath (FirstDevicePath);\r
  }\r
\r
  //\r
  // Allocate space for the combined device path. It only has one end node of\r
  // length EFI_DEVICE_PATH_PROTOCOL\r
  //\r
  Size1         = GetDevicePathSize (FirstDevicePath);\r
  Size2         = GetDevicePathSize (SecondDevicePath);\r
  Size          = Size1 + Size2 - sizeof (EFI_DEVICE_PATH_PROTOCOL);\r
\r
  NewDevicePath = AllocatePool (Size);\r
\r
  if (NewDevicePath != NULL) {\r
    NewDevicePath = CopyMem (NewDevicePath, FirstDevicePath, Size1);\r
    //\r
    // Over write Src1 EndNode and do the copy\r
    //\r
    DevicePath2 = (EFI_DEVICE_PATH_PROTOCOL *) ((CHAR8 *) NewDevicePath + (Size1 - sizeof (EFI_DEVICE_PATH_PROTOCOL)));\r
    CopyMem (DevicePath2, SecondDevicePath, Size2);\r
  }\r
\r
  return NewDevicePath;\r
}\r
\r
/**\r
  This function appends the device path node SecondDevicePath\r
  to every device path instance in FirstDevicePath.\r
\r
  @param  FirstDevicePath A pointer to a device path data structure.\r
  \r
  @param  SecondDevicePath A pointer to a single device path node.\r
\r
  @return A pointer to the new device path.\r
  If there is not enough temporary pool memory available to complete this function,\r
  then NULL is returned.\r
\r
**/\r
EFI_DEVICE_PATH_PROTOCOL *\r
EFIAPI\r
AppendDevicePathNode (\r
  IN CONST EFI_DEVICE_PATH_PROTOCOL  *FirstDevicePath,\r
  IN CONST EFI_DEVICE_PATH_PROTOCOL  *SecondDevicePath\r
  )\r
{\r
  EFI_DEVICE_PATH_PROTOCOL  *NextNode;\r
  EFI_DEVICE_PATH_PROTOCOL  *NewDevicePath;\r
  UINTN                     NodeLength;\r
  UINTN                     Size1;\r
\r
  //\r
  // Build a Node that has a terminator on it\r
  //\r
  NodeLength  = DevicePathNodeLength (SecondDevicePath);\r
  Size1       = GetDevicePathSize (FirstDevicePath);\r
  \r
  NewDevicePath = AllocatePool (NodeLength + Size1);\r
  if (NewDevicePath != NULL) {\r
    //\r
    // Copy the first device path to the new device path\r
    //\r
    NewDevicePath = CopyMem (NewDevicePath, FirstDevicePath, Size1);\r
\r
    //\r
    // Copy the device path node to the new device path\r
    //\r
    NextNode      = (EFI_DEVICE_PATH_PROTOCOL *) ((CHAR8 *) NewDevicePath + (Size1 - sizeof (EFI_DEVICE_PATH_PROTOCOL)));\r
    NextNode      = CopyMem (NextNode, SecondDevicePath, NodeLength);\r
\r
    //\r
    // Terminate the whole device path\r
    //\r
    NextNode      = NextDevicePathNode (NextNode);\r
    SetDevicePathEndNode (NextNode);\r
  }\r
  return NewDevicePath;\r
}\r
\r
/**\r
  This function appends the device path instance Instance to the device path Source.\r
  If Source is NULL, then a new device path with one instance is created.  \r
\r
  @param  Source A pointer to a device path data structure.\r
  @param  Instance A pointer to a device path instance.\r
\r
  @return A pointer to the new device path.\r
  If there is not enough temporary pool memory available to complete this function,\r
  then NULL is returned.\r
\r
**/\r
EFI_DEVICE_PATH_PROTOCOL *\r
EFIAPI\r
AppendDevicePathInstance (\r
  IN CONST EFI_DEVICE_PATH_PROTOCOL  *Source,\r
  IN CONST EFI_DEVICE_PATH_PROTOCOL  *Instance\r
  )\r
{\r
  EFI_DEVICE_PATH_PROTOCOL  *NewDevicePath;\r
  EFI_DEVICE_PATH_PROTOCOL  *DevicePath;\r
  UINTN                     SrcSize;\r
  UINTN                     InstanceSize;\r
\r
  if (Source == NULL) {\r
    return DuplicateDevicePath (Instance);\r
  }\r
\r
  SrcSize       = GetDevicePathSize (Source);\r
  InstanceSize  = GetDevicePathSize (Instance);\r
\r
  NewDevicePath = AllocatePool (SrcSize + InstanceSize);\r
  if (NewDevicePath != NULL) {\r
    \r
    DevicePath = CopyMem (NewDevicePath, Source, SrcSize);;\r
 \r
    while (!IsDevicePathEnd (DevicePath)) {\r
      DevicePath = NextDevicePathNode (DevicePath);\r
    }\r
    \r
    DevicePath->SubType  = END_INSTANCE_DEVICE_PATH_SUBTYPE;\r
\r
    DevicePath           = NextDevicePathNode (DevicePath);\r
    CopyMem (DevicePath, Instance, InstanceSize);\r
  }\r
\r
  return NewDevicePath;\r
}\r
\r
/**\r
  Function retrieves the next device path instance from a device path data structure.\r
\r
  @param  DevicePath A pointer to a device path data structure.\r
  \r
  @param  Size A pointer to the size of a device path instance in bytes.\r
\r
  @return This function returns a pointer to the current device path instance.\r
  In addition, it returns the size in bytes of the current device path instance in Size,\r
  and a pointer to the next device path instance in DevicePath.\r
  If there are no more device path instances in DevicePath, then DevicePath will be set to NULL.\r
\r
**/\r
EFI_DEVICE_PATH_PROTOCOL *\r
EFIAPI\r
GetNextDevicePathInstance (\r
  IN OUT EFI_DEVICE_PATH_PROTOCOL  **DevicePath,\r
     OUT UINTN                     *Size\r
  )\r
{\r
  EFI_DEVICE_PATH_PROTOCOL  *DevPath;\r
  EFI_DEVICE_PATH_PROTOCOL  *ReturnValue;\r
  UINT8                     Temp;\r
\r
  ASSERT (DevicePath != NULL);\r
  ASSERT (Size != NULL);\r
  if (*DevicePath == NULL) {\r
    *Size = 0;\r
    return NULL;\r
  }\r
\r
  //\r
  // Find the end of the device path instance\r
  //\r
  DevPath = *DevicePath;\r
  while (!IsDevicePathEndType (DevPath)) {\r
    DevPath = NextDevicePathNode (DevPath);\r
  }\r
\r
  //\r
  // Compute the size of the device path instance\r
  //\r
  *Size = ((UINTN) DevPath - (UINTN) (*DevicePath)) + sizeof (EFI_DEVICE_PATH_PROTOCOL);\r
 \r
  //\r
  // Make a copy and return the device path instance\r
  //\r
  Temp              = DevPath->SubType;\r
  DevPath->SubType  = END_ENTIRE_DEVICE_PATH_SUBTYPE;\r
  ReturnValue       = DuplicateDevicePath (*DevicePath);\r
  DevPath->SubType  = Temp;\r
\r
  //\r
  // If DevPath is the end of an entire device path, then another instance\r
  // does not follow, so *DevicePath is set to NULL.\r
  //\r
  if (DevicePathSubType (DevPath) == END_ENTIRE_DEVICE_PATH_SUBTYPE) {\r
    *DevicePath = NULL;\r
  } else {\r
    *DevicePath = NextDevicePathNode (DevPath);\r
  }\r
\r
  return ReturnValue;\r
}\r
\r
/**\r
  Return TRUE is this is a multi instance device path.\r
\r
  @param  DevicePath A pointer to a device path data structure.\r
\r
  @retval  TRUE If DevicePath is multi-instance.\r
  @retval  FALSE If DevicePath is not multi-instance or DevicePath is NULL.\r
\r
**/\r
BOOLEAN\r
EFIAPI\r
IsDevicePathMultiInstance (\r
  IN CONST EFI_DEVICE_PATH_PROTOCOL  *DevicePath\r
  )\r
{\r
  CONST EFI_DEVICE_PATH_PROTOCOL  *Node;\r
\r
  if (DevicePath == NULL) {\r
    return FALSE;\r
  }\r
\r
  Node = DevicePath;\r
  while (!EfiIsDevicePathEnd (Node)) {\r
    if (EfiIsDevicePathEndInstance (Node)) {\r
      return TRUE;\r
    }\r
\r
    Node = EfiNextDevicePathNode (Node);\r
  }\r
\r
  return FALSE;\r
}\r
\r
/**\r
  This function retrieves the device path protocol from a handle.\r
\r
  @param  Handle The handle from which to retrieve the device path protocol.\r
\r
  @return This function returns the device path protocol from the handle specified by Handle.\r
  If Handle is NULL or Handle does not contain a device path protocol, then NULL is returned.\r
\r
**/\r
EFI_DEVICE_PATH_PROTOCOL *\r
EFIAPI\r
DevicePathFromHandle (\r
  IN EFI_HANDLE  Handle\r
  )\r
{\r
  EFI_DEVICE_PATH_PROTOCOL  *DevicePath;\r
  EFI_STATUS                Status;\r
\r
  Status = gBS->HandleProtocol (\r
                  Handle,\r
                  &gEfiDevicePathProtocolGuid,\r
                  (VOID *) &DevicePath\r
                  );\r
  if (EFI_ERROR (Status)) {\r
    DevicePath = NULL;\r
  }\r
  return DevicePath;\r
}\r
\r
/**\r
  This function allocates a device path for a file and appends it to an existing device path.\r
\r
  @param  Device A pointer to a device handle.  This parameter is optional and may be NULL.\r
  @param  FileName A pointer to a Null-terminated Unicode string.\r
\r
  @return If Device is a valid device handle that contains a device path protocol,\r
  then a device path for the file specified by FileName is allocated\r
  and appended to the device path associated with the handle Device. The allocated device path is returned.\r
  If Device is NULL or Device is a handle that does not support the device path protocol,\r
  then a device path containing a single device path node for the file specified by FileName\r
  is allocated and returned.\r
\r
**/\r
EFI_DEVICE_PATH_PROTOCOL *\r
EFIAPI\r
FileDevicePath (\r
  IN EFI_HANDLE    Device,     OPTIONAL\r
  IN CONST CHAR16  *FileName\r
  )\r
{\r
  UINTN                     FileNameSize;\r
  UINTN                     FilePathNodeSize;\r
  FILEPATH_DEVICE_PATH      *FilePathNode;\r
  EFI_DEVICE_PATH_PROTOCOL  *DevicePath;\r
\r
  DevicePath        = NULL;\r
\r
  FileNameSize      = StrSize (FileName);\r
  FilePathNodeSize  = FileNameSize + SIZE_OF_FILEPATH_DEVICE_PATH;\r
  FilePathNode      = AllocatePool (FilePathNodeSize);\r
  if (FilePathNode != NULL) {\r
    //\r
    // Build a file path node\r
    //\r
    FilePathNode->Header.Type     = MEDIA_DEVICE_PATH;\r
    FilePathNode->Header.SubType  = MEDIA_FILEPATH_DP;\r
    SetDevicePathNodeLength (&FilePathNode->Header, FilePathNodeSize);\r
    CopyMem (FilePathNode->PathName, FileName, FileNameSize);\r
 \r
    //\r
    // Append file path node to device's device path\r
    //\r
    if (Device != NULL) {\r
      DevicePath = DevicePathFromHandle (Device);\r
    }\r
    DevicePath = AppendDevicePathNode (DevicePath, (EFI_DEVICE_PATH_PROTOCOL *) FilePathNode);\r
    FreePool (FilePathNode);\r
  }\r
  return DevicePath;\r
}\r
\r