MdeModulePkg UhciPei: Also check TempPtr against NULL to return error

Cc: Hao Wu <hao.a.wu@intel.com>
Cc: Jiewen Yao <jiewen.yao@intel.com>
Contributed-under: TianoCore Contribution Agreement 1.1
Signed-off-by: Star Zeng <star.zeng@intel.com>
Reviewed-by: Hao Wu <hao.a.wu@intel.com>
This commit is contained in:
Star Zeng 2017-11-23 09:02:43 +08:00
parent d003e6b98f
commit 56fb9faa60
1 changed files with 2 additions and 2 deletions

View File

@ -2863,8 +2863,8 @@ CreateMemoryBlock (
&MappedAddr,
&Mapping
);
if (EFI_ERROR (Status)) {
return Status;
if (EFI_ERROR (Status) || (TempPtr == NULL)) {
return EFI_OUT_OF_RESOURCES;
}
Ptr = TempPtr;