ShellPkg/DpDynamicCommand: Add ResetEnd support in DP command

[mirror_edk2.git] / ShellPkg / DynamicCommand / DpDynamicCommand / DpDynamicCommand.c

diff --git a/ShellPkg/DynamicCommand/DpDynamicCommand/DpDynamicCommand.c b/ShellPkg/DynamicCommand/DpDynamicCommand/DpDynamicCommand.c
index 6f3997fff4063f52aa5865d12a35dd74af943f02..e4ee4cc5ec3b422fa900994981552a3335ad8f6f 100644 (file)
--- a/ShellPkg/DynamicCommand/DpDynamicCommand/DpDynamicCommand.c
+++ b/ShellPkg/DynamicCommand/DpDynamicCommand/DpDynamicCommand.c
@@ -3,13 +3,7 @@
 \r
   Copyright (c) 2017, Intel Corporation. All rights reserved. <BR>\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
+  SPDX-License-Identifier: BSD-2-Clause-Patent\r
 \r
 **/\r
 #include "Dp.h"\r
@@ -25,19 +19,20 @@
   @param[in] Shell                  The instance of the shell protocol used in the context\r
                                     of processing this command.\r
 \r
-  @return EFI_SUCCESS               the operation was sucessful\r
+  @return EFI_SUCCESS               the operation was successful\r
   @return other                     the operation failed.\r
 **/\r
 SHELL_STATUS\r
 EFIAPI\r
 DpCommandHandler (\r
-  IN EFI_SHELL_DYNAMIC_COMMAND_PROTOCOL    *This,\r
-  IN EFI_SYSTEM_TABLE                      *SystemTable,\r
-  IN EFI_SHELL_PARAMETERS_PROTOCOL         *ShellParameters,\r
-  IN EFI_SHELL_PROTOCOL                    *Shell\r
+  IN EFI_SHELL_DYNAMIC_COMMAND_PROTOCOL  *This,\r
+  IN EFI_SYSTEM_TABLE                    *SystemTable,\r
+  IN EFI_SHELL_PARAMETERS_PROTOCOL       *ShellParameters,\r
+  IN EFI_SHELL_PROTOCOL                  *Shell\r
   )\r
 {\r
   gEfiShellParametersProtocol = ShellParameters;\r
+  gEfiShellProtocol           = Shell;\r
   return RunDp (gImageHandle, SystemTable);\r
 }\r
 \r
@@ -54,14 +49,14 @@ DpCommandHandler (
 CHAR16 *\r
 EFIAPI\r
 DpCommandGetHelp (\r
-  IN EFI_SHELL_DYNAMIC_COMMAND_PROTOCOL    *This,\r
-  IN CONST CHAR8                           *Language\r
+  IN EFI_SHELL_DYNAMIC_COMMAND_PROTOCOL  *This,\r
+  IN CONST CHAR8                         *Language\r
   )\r
 {\r
   return HiiGetString (mDpHiiHandle, STRING_TOKEN (STR_GET_HELP_DP), Language);\r
 }\r
 \r
-EFI_SHELL_DYNAMIC_COMMAND_PROTOCOL mDpDynamicCommand = {\r
+EFI_SHELL_DYNAMIC_COMMAND_PROTOCOL  mDpDynamicCommand = {\r
   L"dp",\r
   DpCommandHandler,\r
   DpCommandGetHelp\r
@@ -75,18 +70,19 @@ EFI_SHELL_DYNAMIC_COMMAND_PROTOCOL mDpDynamicCommand = {
   @param ImageHandle            The image handle of the process.\r
   @param SystemTable            The EFI System Table pointer.\r
 \r
-  @retval EFI_SUCCESS           Tftp command is executed sucessfully.\r
+  @retval EFI_SUCCESS           Tftp command is executed successfully.\r
   @retval EFI_ABORTED           HII package was failed to initialize.\r
   @retval others                Other errors when executing tftp command.\r
 **/\r
 EFI_STATUS\r
 EFIAPI\r
 DpCommandInitialize (\r
-  IN EFI_HANDLE               ImageHandle,\r
-  IN EFI_SYSTEM_TABLE         *SystemTable\r
+  IN EFI_HANDLE        ImageHandle,\r
+  IN EFI_SYSTEM_TABLE  *SystemTable\r
   )\r
 {\r
-  EFI_STATUS                  Status;\r
+  EFI_STATUS  Status;\r
+\r
   mDpHiiHandle = InitializeHiiPackage (ImageHandle);\r
   if (mDpHiiHandle == NULL) {\r
     return EFI_ABORTED;\r
@@ -113,10 +109,11 @@ DpCommandInitialize (
 EFI_STATUS\r
 EFIAPI\r
 DpUnload (\r
-  IN EFI_HANDLE               ImageHandle\r
-)\r
+  IN EFI_HANDLE  ImageHandle\r
+  )\r
 {\r
-  EFI_STATUS                  Status;\r
+  EFI_STATUS  Status;\r
+\r
   Status = gBS->UninstallProtocolInterface (\r
                   ImageHandle,\r
                   &gEfiShellDynamicCommandProtocolGuid,\r
@@ -125,6 +122,7 @@ DpUnload (
   if (EFI_ERROR (Status)) {\r
     return Status;\r
   }\r
+\r
   HiiRemovePackages (mDpHiiHandle);\r
   return EFI_SUCCESS;\r
 }\r