summaryrefslogtreecommitdiffstats
path: root/MdeModulePkg/Application
diff options
context:
space:
mode:
authorStar Zeng <star.zeng@intel.com>2018-07-11 11:45:50 +0800
committerStar Zeng <star.zeng@intel.com>2018-07-13 10:32:16 +0800
commit1c6dd45b9b3359d5c9d4d2a09ec05f81ad9ccaef (patch)
treeaef4cd879781cce1969866b57fb997e3451cbaf5 /MdeModulePkg/Application
parent79b10d4ce4f08aab4b9548fabc4542ca78a96247 (diff)
downloadedk2-1c6dd45b9b3359d5c9d4d2a09ec05f81ad9ccaef.tar.gz
edk2-1c6dd45b9b3359d5c9d4d2a09ec05f81ad9ccaef.tar.bz2
edk2-1c6dd45b9b3359d5c9d4d2a09ec05f81ad9ccaef.zip
MdeModulePkg CapsuleApp: Check Arg count for -D option
Cc: Michael D Kinney <michael.d.kinney@intel.com> Cc: Jiewen Yao <jiewen.yao@intel.com> Cc: Yonghong Zhu <yonghong.zhu@intel.com> Contributed-under: TianoCore Contribution Agreement 1.1 Signed-off-by: Star Zeng <star.zeng@intel.com> Reviewed-by: Jiewen Yao <jiewen.yao@intel.com>
Diffstat (limited to 'MdeModulePkg/Application')
-rw-r--r--MdeModulePkg/Application/CapsuleApp/CapsuleApp.c4
1 files changed, 4 insertions, 0 deletions
diff --git a/MdeModulePkg/Application/CapsuleApp/CapsuleApp.c b/MdeModulePkg/Application/CapsuleApp/CapsuleApp.c
index 3a1a0a1cb2..90f1a02d9a 100644
--- a/MdeModulePkg/Application/CapsuleApp/CapsuleApp.c
+++ b/MdeModulePkg/Application/CapsuleApp/CapsuleApp.c
@@ -792,6 +792,10 @@ UefiMain (
return EFI_INVALID_PARAMETER;
}
if (StrCmp(Argv[1], L"-D") == 0) {
+ if (Argc < 3) {
+ Print(L"CapsuleApp: NO input capsule name.\n");
+ return EFI_INVALID_PARAMETER;
+ }
Status = DumpCapsule(Argv[2]);
return Status;
}