Validate some fields in PE image to make sure not access violation for later code.

[mirror_edk2.git] / SecurityPkg / Library / DxeImageVerificationLib / DxeImageVerificationLib.h

diff --git a/SecurityPkg/Library/DxeImageVerificationLib/DxeImageVerificationLib.h b/SecurityPkg/Library/DxeImageVerificationLib/DxeImageVerificationLib.h
index 881e4e5c35861b2100869dbc8da05d7a279fab5a..55371e90bfc1773521ddb27d8cf31c5f6f1e5280 100644 (file)
--- a/SecurityPkg/Library/DxeImageVerificationLib/DxeImageVerificationLib.h
+++ b/SecurityPkg/Library/DxeImageVerificationLib/DxeImageVerificationLib.h
@@ -28,6 +28,7 @@ WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
 #include <Library/PcdLib.h>\r
 #include <Library/DevicePathLib.h>\r
 #include <Library/SecurityManagementLib.h>\r
+#include <Library/PeCoffLib.h>\r
 #include <Protocol/FirmwareVolume2.h>\r
 #include <Protocol/DevicePath.h>\r
 #include <Protocol/BlockIo.h>\r