Hey
I don't think MdeModulePkg can depend on ShellPkg.

That is why we do not use ShellLib in the first version.

Do we change the package dependency rule?

Thank you
Yao Jiewen

> -----Original Message-----
> From: edk2-devel [mailto:edk2-devel-boun...@lists.01.org] On Behalf Of
> Chen A Chen
> Sent: Friday, January 25, 2019 2:14 PM
> To: edk2-devel@lists.01.org
> Cc: Wu, Hao A <hao.a...@intel.com>; Zhang, Chao B
> <chao.b.zh...@intel.com>
> Subject: [edk2] [PATCH 1/3] MdeModulePkg/CapsuleApp: Refine code logic
> of parsing parameter.
> 
> BZ:https://bugzilla.tianocore.org/show_bug.cgi?id=1482
> 
> No change functionality, use ShellLib to parsing command line.
> 
> Cc: Jian J Wang <jian.j.w...@intel.com>
> Cc: Hao Wu <hao.a...@intel.com>
> Cc: Zhang Chao B <chao.b.zh...@intel.com>
> Contributed-under: TianoCore Contribution Agreement 1.1
> Signed-off-by: Chen A Chen <chen.a.c...@intel.com>
> ---
>  MdeModulePkg/Application/CapsuleApp/CapsuleApp.c   | 433
> +++++++++++----------
>  MdeModulePkg/Application/CapsuleApp/CapsuleApp.inf |   2 +
>  2 files changed, 236 insertions(+), 199 deletions(-)
> 
> diff --git a/MdeModulePkg/Application/CapsuleApp/CapsuleApp.c
> b/MdeModulePkg/Application/CapsuleApp/CapsuleApp.c
> index 4d907242f3..acae0fe261 100644
> --- a/MdeModulePkg/Application/CapsuleApp/CapsuleApp.c
> +++ b/MdeModulePkg/Application/CapsuleApp/CapsuleApp.c
> @@ -27,6 +27,7 @@
>  #include <Guid/SystemResourceTable.h>
>  #include <Guid/FmpCapsule.h>
>  #include <IndustryStandard/WindowsUxCapsule.h>
> +#include <Library/ShellLib.h>
> 
>  #define CAPSULE_HEADER_SIZE  0x20
> 
> @@ -39,15 +40,27 @@
> 
>  #define MAX_CAPSULE_NUM 10
> 
> -extern UINTN  Argc;
> -extern CHAR16 **Argv;
> -
>  //
>  // Define how many block descriptors we want to test with.
>  //
>  UINTN  NumberOfDescriptors = 1;
> -UINTN  CapsuleFirstIndex;
> -UINTN  CapsuleLastIndex;
> +
> +STATIC CONST SHELL_PARAM_ITEM ParamList[] = {
> +  {L"-C", TypeFlag},
> +  {L"-E", TypeFlag},
> +  {L"-S", TypeFlag},
> +
> +  {L"-NR", TypeFlag},
> +
> +  {L"-G", TypeValue},
> +  {L"-O", TypeValue},
> +  {L"-N", TypeValue},
> +  {L"-D", TypeValue},
> +  {L"-P", TypeValue},
> +  {L"-I", TypeValue},
> +
> +  {NULL, TypeMax}
> +  };
> 
>  /**
>    Dump capsule information
> @@ -161,13 +174,12 @@ GetArg (
>  **/
>  EFI_STATUS
>  CreateBmpFmp (
> -  VOID
> +  IN CHAR16                                     *BmpName,
> +  IN CHAR16
> *OutputCapsuleName
>    )
>  {
> -  CHAR16
> *OutputCapsuleName;
>    VOID                                          *BmpBuffer;
>    UINTN                                         FileSize;
> -  CHAR16                                        *BmpName;
>    UINT8
> *FullCapsuleBuffer;
>    UINTN
> FullCapsuleBufferSize;
>    EFI_DISPLAY_CAPSULE                           *DisplayCapsule;
> @@ -191,22 +203,10 @@ CreateBmpFmp (
>    // HorizontalResolution >= BMP_IMAGE_HEADER.PixelWidth
>    // VerticalResolution   >= BMP_IMAGE_HEADER.PixelHeight
> 
> -  if (Argc != 5) {
> -    Print(L"CapsuleApp: Incorrect parameter count.\n");
> -    return EFI_UNSUPPORTED;
> -  }
> -
> -  if (StrCmp(Argv[3], L"-O") != 0) {
> -    Print(L"CapsuleApp: NO output capsule name.\n");
> -    return EFI_UNSUPPORTED;
> -  }
> -  OutputCapsuleName = Argv[4];
> -
>    BmpBuffer = NULL;
>    FileSize = 0;
>    FullCapsuleBuffer = NULL;
> 
> -  BmpName = Argv[2];
>    Status = ReadFileToBuffer(BmpName, &FileSize, &BmpBuffer);
>    if (EFI_ERROR(Status)) {
>      Print(L"CapsuleApp: BMP image (%s) is not found.\n", BmpName);
> @@ -425,13 +425,12 @@ IsFmpCapsuleGuid (
>  **/
>  EFI_STATUS
>  CreateNestedFmp (
> -  VOID
> +  IN CHAR16                                     *CapsuleName,
> +  IN CHAR16
> *OutputCapsuleName
>    )
>  {
> -  CHAR16
> *OutputCapsuleName;
>    VOID                                          *CapsuleBuffer;
>    UINTN                                         FileSize;
> -  CHAR16                                        *CapsuleName;
>    UINT8
> *FullCapsuleBuffer;
>    UINTN
> FullCapsuleBufferSize;
>    EFI_CAPSULE_HEADER
> *NestedCapsuleHeader;
> @@ -439,22 +438,10 @@ CreateNestedFmp (
>    UINT32                                        FwType;
>    EFI_STATUS                                    Status;
> 
> -  if (Argc != 5) {
> -    Print(L"CapsuleApp: Incorrect parameter count.\n");
> -    return EFI_UNSUPPORTED;
> -  }
> -
> -  if (StrCmp(Argv[3], L"-O") != 0) {
> -    Print(L"CapsuleApp: NO output capsule name.\n");
> -    return EFI_UNSUPPORTED;
> -  }
> -  OutputCapsuleName = Argv[4];
> -
>    CapsuleBuffer = NULL;
>    FileSize = 0;
>    FullCapsuleBuffer = NULL;
> 
> -  CapsuleName = Argv[2];
>    Status = ReadFileToBuffer(CapsuleName, &FileSize, &CapsuleBuffer);
>    if (EFI_ERROR(Status)) {
>      Print(L"CapsuleApp: Capsule image (%s) is not found.\n",
> CapsuleName);
> @@ -807,7 +794,7 @@ PrintUsage (
>    Print(L"  CapsuleApp -G <BMP> -O <Capsule>\n");
>    Print(L"  CapsuleApp -N <Capsule> -O <NestedCapsule>\n");
>    Print(L"  CapsuleApp -D <Capsule>\n");
> -  Print(L"  CapsuleApp -P GET <ImageTypeId> <Index> -O <FileName>\n");
> +  Print(L"  CapsuleApp -P <ImageTypeId> -I <Index> -O <FileName>\n");
>    Print(L"Parameter:\n");
>    Print(L"  -NR: No reset will be triggered for the capsule with\n");
>    Print(L"       CAPSULE_FLAGS_PERSIST_ACROSS_RESET and
> without\n");
> @@ -817,8 +804,7 @@ PrintUsage (
>    Print(L"  -C:  Clear capsule report variable
> (EFI_CAPSULE_REPORT_GUID),\n");
>    Print(L"       which is defined in UEFI specification.\n");
>    Print(L"  -P:  Dump UEFI FMP protocol info, or get image with
> specified\n");
> -  Print(L"       ImageTypeId and Index (decimal format) to a file if
> 'GET'\n");
> -  Print(L"       option is used.\n");
> +  Print(L"       ImageTypeId and Index (decimal format) to a file\n");
>    Print(L"  -E:  Dump UEFI ESRT table info.\n");
>    Print(L"  -G:  Convert a BMP file to be an UX capsule,\n");
>    Print(L"       according to Windows Firmware Update document\n");
> @@ -851,206 +837,255 @@ UefiMain (
>  {
>    EFI_STATUS                    Status;
>    RETURN_STATUS                 RStatus;
> -  UINTN                         FileSize[MAX_CAPSULE_NUM];
>    VOID
> *CapsuleBuffer[MAX_CAPSULE_NUM];
> -  EFI_CAPSULE_BLOCK_DESCRIPTOR  *BlockDescriptors;
>    EFI_CAPSULE_HEADER
> *CapsuleHeaderArray[MAX_CAPSULE_NUM + 1];
> -  UINT64                        MaxCapsuleSize;
> +  EFI_CAPSULE_BLOCK_DESCRIPTOR  *BlockDescriptors;
>    EFI_RESET_TYPE                ResetType;
> +  UINT64                        MaxCapsuleSize;
> +  LIST_ENTRY                    *Package;
> +  CHAR16                        *ProblemParam;
> +  CHAR16                        *BmpFileName;
> +  CHAR16                        *CapsuleFileName;
> +  CHAR16                        *OutputFileName;
> +  CHAR16                        *GuidStr;
> +  CHAR16                        *IndexStr;
> +  EFI_GUID                      ImageTypeGuid;
> +  BOOLEAN                       NoResetFlag;
>    BOOLEAN                       NeedReset;
> -  BOOLEAN                       NoReset;
> -  CHAR16                        *CapsuleName;
> +  UINTN
> CapsuleBufferSize[MAX_CAPSULE_NUM];
>    UINTN                         CapsuleNum;
> -  UINTN                         Index;
> -  EFI_GUID                      ImageTypeId;
>    UINTN                         ImageIndex;
> +  UINTN                         FlagCount;
> +  UINTN                         Index;
> 
> -  Status = GetArg();
> -  if (EFI_ERROR(Status)) {
> -    Print(L"Please use UEFI SHELL to run this application!\n", Status);
> -    return Status;
> -  }
> -  if (Argc < 2) {
> -    PrintUsage();
> -    return EFI_UNSUPPORTED;
> +  Status      = EFI_SUCCESS;
> +  NoResetFlag = FALSE;
> +  FlagCount   = 0;
> +
> +  ZeroMem (&CapsuleBuffer, sizeof (CapsuleBuffer));
> +  ZeroMem (&CapsuleBufferSize, sizeof (CapsuleBufferSize));
> +
> +  Status = ShellCommandLineParse (ParamList, &Package, &ProblemParam,
> TRUE);
> +  if (EFI_ERROR (Status)) {
> +    PrintUsage ();
> +    Status = EFI_UNSUPPORTED;
> +    goto Done;
>    }
> -  if (StrCmp(Argv[1], L"-D") == 0) {
> -    if (Argc != 3) {
> -      Print(L"CapsuleApp: Incorrect parameter count.\n");
> -      return EFI_UNSUPPORTED;
> +
> +  for (Index = 0; ParamList[Index].Type != TypeMax; Index++) {
> +    if (ShellCommandLineGetFlag (Package, ParamList[Index].Name)) {
> +      FlagCount++;
>      }
> -    Status = DumpCapsule(Argv[2]);
> -    return Status;
> -  }
> -  if (StrCmp(Argv[1], L"-G") == 0) {
> -    Status = CreateBmpFmp();
> -    return Status;
> -  }
> -  if (StrCmp(Argv[1], L"-N") == 0) {
> -    Status = CreateNestedFmp();
> -    return Status;
>    }
> -  if (StrCmp(Argv[1], L"-S") == 0) {
> -    Status = DumpCapsuleStatusVariable();
> -    return EFI_SUCCESS;
> -  }
> -  if (StrCmp(Argv[1], L"-C") == 0) {
> -    Status = ClearCapsuleStatusVariable();
> -    return Status;
> -  }
> -  if (StrCmp(Argv[1], L"-P") == 0) {
> -    if (Argc == 2) {
> -      DumpFmpData();
> -    }
> -    if (Argc >= 3) {
> -      if (StrCmp(Argv[2], L"GET") != 0) {
> -        Print(L"CapsuleApp: Unrecognized option(%s).\n", Argv[2]);
> -        return EFI_UNSUPPORTED;
> -      } else {
> -        if (Argc != 7) {
> -          Print(L"CapsuleApp: Incorrect parameter count.\n");
> -          return EFI_UNSUPPORTED;
> -        }
> 
> -        //
> -        // FMP->GetImage()
> -        //
> -        RStatus = StrToGuid (Argv[3], &ImageTypeId);
> -        if (RETURN_ERROR (RStatus) ||
> (Argv[3][GUID_STRING_LENGTH] != L'\0')) {
> -          Print (L"Invalid ImageTypeId - %s\n", Argv[3]);
> -          return EFI_INVALID_PARAMETER;
> -        }
> -        ImageIndex = StrDecimalToUintn(Argv[4]);
> -        if (StrCmp(Argv[5], L"-O") != 0) {
> -          Print(L"CapsuleApp: NO output file name.\n");
> -          return EFI_UNSUPPORTED;
> +  CapsuleNum = ShellCommandLineGetCount (Package) - 1;
> +
> +  if (CapsuleNum == 0) {
> +    if (FlagCount == 1) {
> +      //
> +      //  CapsuleApp -C
> +      //
> +      if (ShellCommandLineGetFlag (Package, L"-C")) {
> +        Status = ClearCapsuleStatusVariable();
> +      }
> +
> +      //
> +      //  CapsuleApp -D <CapsuleFile>
> +      //
> +      if (ShellCommandLineGetFlag (Package, L"-D")) {
> +        CapsuleFileName = (CHAR16 *)ShellCommandLineGetValue
> (Package, L"-D");
> +        if (CapsuleFileName != NULL) {
> +          Status = DumpCapsule (CapsuleFileName);
> +        } else {
> +          Print (L"CapsuleApp: Incorrect Parameter Count.\n");
> +          Status = EFI_UNSUPPORTED;
>          }
> -        DumpFmpImage(&ImageTypeId, ImageIndex, Argv[6]);
>        }
> -    }
> -    return EFI_SUCCESS;
> -  }
> 
> -  if (StrCmp(Argv[1], L"-E") == 0) {
> -    DumpEsrtData();
> -    return EFI_SUCCESS;
> -  }
> +      //
> +      //  CapsuleApp -E
> +      //
> +      if (ShellCommandLineGetFlag (Package, L"-E")) {
> +        DumpEsrtData ();
> +      }
> 
> -  if (Argv[1][0] == L'-') {
> -    Print(L"CapsuleApp: Unrecognized option(%s).\n", Argv[1]);
> -    return EFI_UNSUPPORTED;
> -  }
> +      //
> +      //  CapsuleApp -P
> +      //
> +      if (ShellCommandLineGetFlag (Package, L"-P")) {
> +        DumpFmpData ();
> +      }
> 
> -  CapsuleFirstIndex = 1;
> -  NoReset = FALSE;
> -  if ((Argc > 1) && (StrCmp(Argv[Argc - 1], L"-NR") == 0)) {
> -    NoReset = TRUE;
> -    CapsuleLastIndex = Argc - 2;
> -  } else {
> -    CapsuleLastIndex = Argc - 1;
> -  }
> -  CapsuleNum = CapsuleLastIndex - CapsuleFirstIndex + 1;
> +      //
> +      //  CapsuleApp -S
> +      //
> +      if (ShellCommandLineGetFlag (Package, L"-S")) {
> +        Status = DumpCapsuleStatusVariable ();
> +      }
> +    }
> 
> -  if (CapsuleFirstIndex > CapsuleLastIndex) {
> -    Print(L"CapsuleApp: NO capsule image.\n");
> -    return EFI_UNSUPPORTED;
> -  }
> -  if (CapsuleNum > MAX_CAPSULE_NUM) {
> -    Print(L"CapsuleApp: Too many capsule images.\n");
> -    return EFI_UNSUPPORTED;
> -  }
> +    if (FlagCount == 2) {
> +      //
> +      //  CapsuleApp -G <BMP> -O <OutputFileName>
> +      //
> +      if (ShellCommandLineGetFlag (Package, L"-G") &&
> ShellCommandLineGetFlag (Package, L"-O")) {
> +        BmpFileName     = (CHAR16 *)ShellCommandLineGetValue
> (Package, L"-G");
> +        OutputFileName  = (CHAR16 *)ShellCommandLineGetValue
> (Package, L"-O");
> +        if (BmpFileName != NULL && OutputFileName != NULL) {
> +          Status = CreateBmpFmp (BmpFileName, OutputFileName);
> +        } else {
> +          Print (L"CapsuleApp: Incorrect Parameter Count.\n");
> +          Status = EFI_UNSUPPORTED;
> +        }
> +      }
> 
> -  ZeroMem(&CapsuleBuffer, sizeof(CapsuleBuffer));
> -  ZeroMem(&FileSize, sizeof(FileSize));
> -  BlockDescriptors = NULL;
> +      //
> +      //  CapsuleApp -N <CapsuleFile> -O <OutputFileName>
> +      //
> +      if (ShellCommandLineGetFlag (Package, L"-N") &&
> ShellCommandLineGetFlag (Package ,L"-O")) {
> +        CapsuleFileName = (CHAR16 *)ShellCommandLineGetValue
> (Package, L"-N");
> +        OutputFileName  = (CHAR16 *)ShellCommandLineGetValue
> (Package, L"-O");
> +        if (CapsuleFileName != NULL && OutputFileName != NULL) {
> +          Status = CreateNestedFmp (CapsuleFileName,
> OutputFileName);
> +        } else {
> +          Print (L"CapsuleApp: Incorrect Parameter Count.\n");
> +          Status = EFI_UNSUPPORTED;
> +        }
> +      }
> +    }
> 
> -  for (Index = 0; Index < CapsuleNum; Index++) {
> -    CapsuleName = Argv[CapsuleFirstIndex + Index];
> -    Status = ReadFileToBuffer(CapsuleName, &FileSize[Index],
> &CapsuleBuffer[Index]);
> -    if (EFI_ERROR(Status)) {
> -      Print(L"CapsuleApp: capsule image (%s) is not found.\n",
> CapsuleName);
> +    if (FlagCount == 3) {
> +      //
> +      //  CapsuleApp -P <ImageTypeGuid> -I <ImageIndex> -O
> <OutputFile>
> +      //
> +      if ( ShellCommandLineGetFlag (Package, L"-P") &&
> +           ShellCommandLineGetFlag (Package, L"-I") &&
> +           ShellCommandLineGetFlag (Package, L"-O")
> +         ) {
> +        GuidStr         = (CHAR16 *)ShellCommandLineGetValue
> (Package, L"-P");
> +        IndexStr        = (CHAR16 *)ShellCommandLineGetValue
> (Package, L"-I");
> +        OutputFileName  = (CHAR16 *)ShellCommandLineGetValue
> (Package, L"-O");
> +        if (GuidStr != NULL && IndexStr != NULL && OutputFileName !=
> NULL) {
> +          RStatus = StrToGuid (GuidStr, &ImageTypeGuid);
> +          if (EFI_ERROR (RStatus) || GuidStr[GUID_STRING_LENGTH] !=
> L'\0') {
> +            Print (L"Invalid ImageTypeId - %s\n", GuidStr);
> +            Status = EFI_INVALID_PARAMETER;
> +            goto Done;
> +          }
> +          ImageIndex = StrDecimalToUintn (IndexStr);
> +          DumpFmpImage (&ImageTypeGuid, ImageIndex,
> OutputFileName);
> +        } else {
> +          Print (L"CapsuleApp: Incorrect Parameter Count.\n");
> +          Status = EFI_UNSUPPORTED;
> +        }
> +      }
> +    }
> +  } else {
> +    if (CapsuleNum > MAX_CAPSULE_NUM) {
> +      Print(L"CapsuleApp: Too many capsule images.\n");
> +      Status = EFI_UNSUPPORTED;
>        goto Done;
>      }
> -    if (!IsValidCapsuleHeader (CapsuleBuffer[Index], FileSize[Index])) {
> -      Print(L"CapsuleApp: Capsule image (%s) is not a valid capsule.\n",
> CapsuleName);
> -      return EFI_INVALID_PARAMETER;
> +
> +    if (ShellCommandLineGetFlag (Package, L"-NR")) {
> +      NoResetFlag = TRUE;
>      }
> -  }
> 
> -  //
> -  // Every capsule use 2 descriptor 1 for data 1 for end
> -  //
> -  Status = BuildGatherList(CapsuleBuffer, FileSize, CapsuleNum,
> &BlockDescriptors);
> -  if (EFI_ERROR(Status)) {
> -    goto Done;
> -  }
> +    for (Index = 0; Index < CapsuleNum; Index++) {
> +      CapsuleFileName = (CHAR16 *)ShellCommandLineGetRawValue
> (Package, Index + 1);
> +      Status = ReadFileToBuffer (CapsuleFileName,
> &CapsuleBufferSize[Index], &CapsuleBuffer[Index]);
> +      if (EFI_ERROR (Status)) {
> +        Print (L"CapsuleApp: capsule image (%s) is not found.\n",
> CapsuleFileName);
> +        goto Done;
> +      }
> +      if (!IsValidCapsuleHeader (CapsuleBuffer[Index],
> CapsuleBufferSize[Index])) {
> +        Print (L"CapsuleApp: Capsule image (%s) is not a valid capsule.\n",
> CapsuleFileName);
> +        Status = EFI_INVALID_PARAMETER;
> +        goto Done;
> +      }
> +    }
> 
> -  //
> -  // Call the runtime service capsule.
> -  //
> -  NeedReset = FALSE;
> -  for (Index = 0; Index < CapsuleNum; Index++) {
> -    CapsuleHeaderArray[Index] = (EFI_CAPSULE_HEADER *)
> CapsuleBuffer[Index];
> -    if ((CapsuleHeaderArray[Index]->Flags &
> CAPSULE_FLAGS_PERSIST_ACROSS_RESET) != 0) {
> -      NeedReset = TRUE;
> +    //
> +    // Every capsule use 2 descriptor 1 for data 1 for end
> +    //
> +    BlockDescriptors = NULL;
> +    Status = BuildGatherList (CapsuleBuffer, CapsuleBufferSize,
> CapsuleNum, &BlockDescriptors);
> +    if (EFI_ERROR (Status)) {
> +      Print (L"Build Gather List Fail, %r\n", Status);
> +      goto Done;
>      }
> -  }
> -  CapsuleHeaderArray[CapsuleNum] = NULL;
> 
> -  //
> -  // Inquire platform capability of UpdateCapsule.
> -  //
> -  Status = gRT->QueryCapsuleCapabilities (CapsuleHeaderArray,
> CapsuleNum, &MaxCapsuleSize, &ResetType);
> -  if (EFI_ERROR(Status)) {
> -    Print (L"CapsuleApp: failed to query capsule capability - %r\n", Status);
> -    goto Done;
> -  }
> +    //
> +    // Call the runtime service capsule.
> +    //
> +    NeedReset = FALSE;
> +    for (Index = 0; Index < CapsuleNum; Index++) {
> +      CapsuleHeaderArray[Index] = (EFI_CAPSULE_HEADER *)
> CapsuleBuffer[Index];
> +      if ((CapsuleHeaderArray[Index]->Flags &
> CAPSULE_FLAGS_PERSIST_ACROSS_RESET) != 0) {
> +        NeedReset = TRUE;
> +      }
> +    }
> +    CapsuleHeaderArray[CapsuleNum] = NULL;
> 
> -  for (Index = 0; Index < CapsuleNum; Index++) {
> -    if (FileSize[Index] > MaxCapsuleSize) {
> -      Print (L"CapsuleApp: capsule is too large to update, %ld is
> allowed\n", MaxCapsuleSize);
> -      Status = EFI_UNSUPPORTED;
> +    //
> +    // Inquire platform capability of UpdateCapsule.
> +    //
> +    Status = gRT->QueryCapsuleCapabilities (CapsuleHeaderArray,
> CapsuleNum, &MaxCapsuleSize, &ResetType);
> +    if (EFI_ERROR (Status)) {
> +      Print (L"CapsuleApp: failed to query capsule capability - %r\n",
> Status);
>        goto Done;
>      }
> -  }
> 
> -  //
> -  // Check whether the input capsule image has the flag of persist across
> system reset.
> -  //
> -  if (NeedReset) {
> -    Status =
> gRT->UpdateCapsule(CapsuleHeaderArray,CapsuleNum,(UINTN)
> BlockDescriptors);
> -    if (Status != EFI_SUCCESS) {
> -      Print (L"CapsuleApp: failed to update capsule - %r\n", Status);
> -      goto Done;
> +    for (Index = 0; Index < CapsuleNum; Index++) {
> +      if (CapsuleBufferSize[Index] > MaxCapsuleSize) {
> +        Print (L"CapsuleApp: capsule is too large to update, %ld is
> allowed\n", MaxCapsuleSize);
> +        Status = EFI_UNSUPPORTED;
> +        goto Done;
> +      }
>      }
> +
>      //
> -    // For capsule with CAPSULE_FLAGS_PERSIST_ACROSS_RESET +
> CAPSULE_FLAGS_INITIATE_RESET,
> -    // a system reset should have been triggered by gRT->UpdateCapsule()
> calling above.
> -    //
> -    // For capsule with CAPSULE_FLAGS_PERSIST_ACROSS_RESET and
> without CAPSULE_FLAGS_INITIATE_RESET,
> -    // check if -NR (no-reset) has been specified or not.
> +    // Check whether the input capsule image has the flag of persist across
> system reset.
>      //
> -    if (!NoReset) {
> +    if (NeedReset) {
> +      Status = gRT->UpdateCapsule
> (CapsuleHeaderArray,CapsuleNum,(UINTN) BlockDescriptors);
> +      if (Status != EFI_SUCCESS) {
> +        Print (L"CapsuleApp: failed to update capsule - %r\n", Status);
> +        goto Done;
> +      }
>        //
> -      // For capsule who has reset flag and no -NR (no-reset) has been
> specified, after calling UpdateCapsule service,
> -      // trigger a system reset to process capsule persist across a system
> reset.
> +      // For capsule with CAPSULE_FLAGS_PERSIST_ACROSS_RESET +
> CAPSULE_FLAGS_INITIATE_RESET,
> +      // a system reset should have been triggered by
> gRT->UpdateCapsule() calling above.
>        //
> -      gRT->ResetSystem (ResetType, EFI_SUCCESS, 0, NULL);
> -    }
> -  } else {
> -    //
> -    // For capsule who has no reset flag, only call UpdateCapsule Service
> without a
> -    // system reset. The service will process the capsule immediately.
> -    //
> -    Status = gRT->UpdateCapsule
> (CapsuleHeaderArray,CapsuleNum,(UINTN) BlockDescriptors);
> -    if (Status != EFI_SUCCESS) {
> -      Print (L"CapsuleApp: failed to update capsule - %r\n", Status);
> +      // For capsule with CAPSULE_FLAGS_PERSIST_ACROSS_RESET and
> without CAPSULE_FLAGS_INITIATE_RESET,
> +      // check if -NR (no-reset) has been specified or not.
> +      //
> +      if (!NoResetFlag) {
> +        //
> +        // For capsule who has reset flag and no -NR (no-reset) has been
> specified, after calling UpdateCapsule service,
> +        // trigger a system reset to process capsule persist across a
> system reset.
> +        //
> +        gRT->ResetSystem (ResetType, EFI_SUCCESS, 0, NULL);
> +      }
> +    } else {
> +      //
> +      // For capsule who has no reset flag, only call UpdateCapsule
> Service without a
> +      // system reset. The service will process the capsule immediately.
> +      //
> +      Status = gRT->UpdateCapsule
> (CapsuleHeaderArray,CapsuleNum,(UINTN) BlockDescriptors);
> +      if (Status != EFI_SUCCESS) {
> +        Print (L"CapsuleApp: failed to update capsule - %r\n", Status);
> +      }
>      }
> -  }
> 
> -  Status = EFI_SUCCESS;
> +    Status = EFI_SUCCESS;
> +  }
> 
>  Done:
> +  if (Package != NULL) {
> +    ShellCommandLineFreeVarList (Package);
> +  }
> +
>    for (Index = 0; Index < CapsuleNum; Index++) {
>      if (CapsuleBuffer[Index] != NULL) {
>        FreePool (CapsuleBuffer[Index]);
> diff --git a/MdeModulePkg/Application/CapsuleApp/CapsuleApp.inf
> b/MdeModulePkg/Application/CapsuleApp/CapsuleApp.inf
> index 8a21875286..d4edc539cc 100644
> --- a/MdeModulePkg/Application/CapsuleApp/CapsuleApp.inf
> +++ b/MdeModulePkg/Application/CapsuleApp/CapsuleApp.inf
> @@ -38,6 +38,7 @@
>  [Packages]
>    MdePkg/MdePkg.dec
>    MdeModulePkg/MdeModulePkg.dec
> +  ShellPkg/ShellPkg.dec
> 
>  [Guids]
>    gEfiCapsuleReportGuid                  ## CONSUMES   ## GUID
> @@ -61,6 +62,7 @@
>    UefiLib
>    PrintLib
>    BmpSupportLib
> +  ShellLib
> 
>  [UserExtensions.TianoCore."ExtraFiles"]
>    CapsuleAppExtra.uni
> --
> 2.16.2.windows.1
> 
> _______________________________________________
> edk2-devel mailing list
> edk2-devel@lists.01.org
> https://lists.01.org/mailman/listinfo/edk2-devel
_______________________________________________
edk2-devel mailing list
edk2-devel@lists.01.org
https://lists.01.org/mailman/listinfo/edk2-devel

Reply via email to