From: "Gao, Liming" <liming.gao@intel.com>
To: "Zhang, Chao B" <chao.b.zhang@intel.com>,
"Yao, Jiewen" <jiewen.yao@intel.com>,
"edk2-devel@lists.01.org" <edk2-devel@lists.01.org>
Cc: "Wu, Hao A" <hao.a.wu@intel.com>,
"Chen, Chen A" <chen.a.chen@intel.com>,
"Gao, Liming" <liming.gao@intel.com>
Subject: Re: [PATCH 1/3] MdeModulePkg/CapsuleApp: Refine code logic of parsing parameter.
Date: Fri, 25 Jan 2019 09:24:46 +0000 [thread overview]
Message-ID: <4A89E2EF3DFEDB4C8BFDE51014F606A14E3D0750@SHSMSX104.ccr.corp.intel.com> (raw)
In-Reply-To: <FF72C7E4248F3C4E9BDF19D4918E90F24DE5E930@shsmsx102.ccr.corp.intel.com>
Chao:
CapsuleApp can base on gEfiShellProtocolGuid and gEfiShellParametersProtocolGuid to do it. Current CapsuleApp bases on these two protocols to parse the parameters.
Thanks
Liming
From: Zhang, Chao B
Sent: Friday, January 25, 2019 3:24 PM
To: Yao, Jiewen <jiewen.yao@intel.com>; edk2-devel@lists.01.org; Gao, Liming <liming.gao@intel.com>
Cc: Wu, Hao A <hao.a.wu@intel.com>; Chen, Chen A <chen.a.chen@intel.com>
Subject: RE: [edk2] [PATCH 1/3] MdeModulePkg/CapsuleApp: Refine code logic of parsing parameter.
Jiewen & Liming:
It is a problem. Unlike UiApp. CapsuleApp is supposed to run in Shell. ShellLib provides standard parameter parsing support.
Any suggestion on this?
From: Yao, Jiewen
Sent: Friday, January 25, 2019 2:25 PM
To: Chen, Chen A <chen.a.chen@intel.com<mailto:chen.a.chen@intel.com>>; edk2-devel@lists.01.org<mailto:edk2-devel@lists.01.org>
Cc: Wu, Hao A <hao.a.wu@intel.com<mailto:hao.a.wu@intel.com>>; Zhang, Chao B <chao.b.zhang@intel.com<mailto:chao.b.zhang@intel.com>>
Subject: RE: [edk2] [PATCH 1/3] MdeModulePkg/CapsuleApp: Refine code logic of parsing parameter.
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-bounces@lists.01.org] On Behalf Of
> Chen A Chen
> Sent: Friday, January 25, 2019 2:14 PM
> To: edk2-devel@lists.01.org<mailto:edk2-devel@lists.01.org>
> Cc: Wu, Hao A <hao.a.wu@intel.com<mailto:hao.a.wu@intel.com>>; Zhang, Chao B
> <chao.b.zhang@intel.com<mailto:chao.b.zhang@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.wang@intel.com<mailto:jian.j.wang@intel.com>>
> Cc: Hao Wu <hao.a.wu@intel.com<mailto:hao.a.wu@intel.com>>
> Cc: Zhang Chao B <chao.b.zhang@intel.com<mailto:chao.b.zhang@intel.com>>
> Contributed-under: TianoCore Contribution Agreement 1.1
> Signed-off-by: Chen A Chen <chen.a.chen@intel.com<mailto:chen.a.chen@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<mailto:edk2-devel@lists.01.org>
> https://lists.01.org/mailman/listinfo/edk2-devel
next prev parent reply other threads:[~2019-01-25 9:24 UTC|newest]
Thread overview: 6+ messages / expand[flat|nested] mbox.gz Atom feed top
2019-01-25 6:13 [PATCH 1/3] MdeModulePkg/CapsuleApp: Refine code logic of parsing parameter Chen A Chen
2019-01-25 6:24 ` Yao, Jiewen
2019-01-25 7:23 ` Zhang, Chao B
2019-01-25 9:24 ` Gao, Liming [this message]
2019-01-25 15:36 ` Carsey, Jaben
2019-01-25 22:50 ` Yao, Jiewen
Reply instructions:
You may reply publicly to this message via plain-text email
using any one of the following methods:
* Save the following mbox file, import it into your mail client,
and reply-to-list from there: mbox
Avoid top-posting and favor interleaved quoting:
https://en.wikipedia.org/wiki/Posting_style#Interleaved_style
* Reply using the --to, --cc, and --in-reply-to
switches of git-send-email(1):
git send-email \
--in-reply-to=4A89E2EF3DFEDB4C8BFDE51014F606A14E3D0750@SHSMSX104.ccr.corp.intel.com \
--to=devel@edk2.groups.io \
/path/to/YOUR_REPLY
https://kernel.org/pub/software/scm/git/docs/git-send-email.html
* If your mail client supports setting the In-Reply-To header
via mailto: links, try the mailto: link
Be sure your reply has a Subject: header at the top and a blank line
before the message body.
This is a public inbox, see mirroring instructions
for how to clone and mirror all data and code used for this inbox