Virtualization
/
edk2
mirror of https://gitlab.com/qemu-project/edk2.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

UefiCpuPkg/MicrocodeUpdate: Fix GetImage API 

Current GetImage API forgets to return data to caller.

Add code to fix it.

Cc: Jeff Fan <jeff.fan@intel.com>
Contributed-under: TianoCore Contribution Agreement 1.0
Signed-off-by: Jiewen Yao <jiewen.yao@intel.com>
Reviewed-by: Jeff Fan <jeff.fan@intel.com>
This commit is contained in:
Jiewen Yao 2016-11-29 15:54:57 +08:00
parent e50f951286
commit bcc6a38f4c
1 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
UefiCpuPkg/Feature/Capsule/MicrocodeUpdateDxe/MicrocodeUpdate.c
View File

@ -255,6 +255,16 @@ MicrocodeRead (
TotalSize = MicrocodeEntryPoint->TotalSize; TotalSize = MicrocodeEntryPoint->TotalSize;
} }
if (ImageIndex == Count + 1) {
if (*ImageSize < TotalSize) {
*ImageSize = TotalSize;
return EFI_BUFFER_TOO_SMALL;
}
*ImageSize = TotalSize;
CopyMem (Image, MicrocodeEntryPoint, TotalSize);
return EFI_SUCCESS;
}
} else { } else {
// //
// It is the padding data between the microcode patches for microcode patches alignment. // It is the padding data between the microcode patches for microcode patches alignment.