ShellPkg/Bcfg: Add support for 'addp' command.

Until now the 'addp' command has been handled in the same way as 'add'.
Just copy the DevicePath starting from the Hard Drive node when 'addp'
is used.

Contributed-under: TianoCore Contribution Agreement 1.0
Signed-off-by: Marvin Haeuser <Marvin.Haeuser@outlook.com>
Reviewed-by: Qiu Shumin <shumin.qiu@intel.com>

1 files changed, 5 insertions, 13 deletions

diff --git a/ShellPkg/Library/UefiShellBcfgCommandLib/UefiShellBcfgCommandLib.c b/ShellPkg/Library/UefiShellBcfgCommandLib/UefiShellBcfgCommandLib.c
index b407608d3..bf35a35b9 100644
--- a/ShellPkg/Library/UefiShellBcfgCommandLib/UefiShellBcfgCommandLib.c
+++ b/ShellPkg/Library/UefiShellBcfgCommandLib/UefiShellBcfgCommandLib.c
@@ -310,6 +310,7 @@ BcfgAdd(
 {

   EFI_STATUS                Status;

   EFI_DEVICE_PATH_PROTOCOL  *DevicePath;

+  EFI_DEVICE_PATH_PROTOCOL  *DevPath;

   EFI_DEVICE_PATH_PROTOCOL  *FilePath;

   CHAR16                    *Str;

   UINT8                     *TempByteBuffer;

@@ -462,9 +463,9 @@ BcfgAdd(
           ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_BCFG_FILE_DP), gShellBcfgHiiHandle, L"bcfg", Arg->FullName);  

           ShellStatus = SHELL_UNSUPPORTED;

         } else {

-/*

           if (UsePath) {

-            DevPath = DevicePath;

+            DevPath     = DevicePath;

+            ShellStatus = SHELL_INVALID_PARAMETER;

             while (!IsDevicePathEnd(DevPath)) {

               if ((DevicePathType(DevPath) == MEDIA_DEVICE_PATH) &&

                 (DevicePathSubType(DevPath) == MEDIA_HARDDRIVE_DP)) {

@@ -472,24 +473,15 @@ BcfgAdd(
                 //

                 // If we find it use it instead

                 //

-                DevicePath = DevPath;

+                ShellStatus = SHELL_SUCCESS;

+                FilePath    = DuplicateDevicePath (DevPath);

                 break;

               }

               DevPath = NextDevicePathNode(DevPath);

             }

-            //

-            // append the file

-            //

-            for(StringWalker=Arg->FullName; *StringWalker != CHAR_NULL && *StringWalker != ':'; StringWalker++);

-            FileNode = FileDevicePath(NULL, StringWalker+1);

-            FilePath = AppendDevicePath(DevicePath, FileNode);

-            FreePool(FileNode);

           } else {

-*/

             FilePath = DuplicateDevicePath(DevicePath);

-/*

           }

-*/

           FreePool(DevicePath);

         }

       }