diff options
-rw-r--r-- | EdkModulePkg/Core/Dxe/DxeMain/DxeMain.c | 39 |
1 files changed, 37 insertions, 2 deletions
diff --git a/EdkModulePkg/Core/Dxe/DxeMain/DxeMain.c b/EdkModulePkg/Core/Dxe/DxeMain/DxeMain.c index fce61bd45e..427e6b8ca5 100644 --- a/EdkModulePkg/Core/Dxe/DxeMain/DxeMain.c +++ b/EdkModulePkg/Core/Dxe/DxeMain/DxeMain.c @@ -848,6 +848,11 @@ DxeMainUefiDecompressGetInfo ( OUT UINT32 *ScratchSize
)
{
+ if (Source == NULL
+ || DestinationSize == NULL
+ || ScratchSize == NULL) {
+ return EFI_INVALID_PARAMETER;
+ }
return UefiDecompressGetInfo (Source, SourceSize, DestinationSize, ScratchSize);
}
@@ -866,7 +871,13 @@ DxeMainUefiDecompress ( EFI_STATUS Status;
UINT32 TestDestinationSize;
UINT32 TestScratchSize;
-
+
+ if (Source == NULL
+ || Destination== NULL
+ || Scratch == NULL) {
+ return EFI_INVALID_PARAMETER;
+ }
+
Status = UefiDecompressGetInfo (Source, SourceSize, &TestDestinationSize, &TestScratchSize);
if (EFI_ERROR (Status)) {
return Status;
@@ -888,6 +899,12 @@ DxeMainTianoDecompressGetInfo ( OUT UINT32 *ScratchSize
)
{
+ if (Source == NULL
+ || DestinationSize == NULL
+ || ScratchSize == NULL) {
+ return EFI_INVALID_PARAMETER;
+ }
+
return TianoDecompressGetInfo (Source, SourceSize, DestinationSize, ScratchSize);
}
@@ -906,7 +923,13 @@ DxeMainTianoDecompress ( EFI_STATUS Status;
UINT32 TestDestinationSize;
UINT32 TestScratchSize;
-
+
+ if (Source == NULL
+ || Destination== NULL
+ || Scratch == NULL) {
+ return EFI_INVALID_PARAMETER;
+ }
+
Status = TianoDecompressGetInfo (Source, SourceSize, &TestDestinationSize, &TestScratchSize);
if (EFI_ERROR (Status)) {
return Status;
@@ -928,6 +951,12 @@ DxeMainCustomDecompressGetInfo ( OUT UINT32 *ScratchSize
)
{
+ if (Source == NULL
+ || DestinationSize == NULL
+ || ScratchSize == NULL) {
+ return EFI_INVALID_PARAMETER;
+ }
+
return CustomDecompressGetInfo (Source, SourceSize, DestinationSize, ScratchSize);
}
@@ -947,6 +976,12 @@ DxeMainCustomDecompress ( UINT32 TestDestinationSize;
UINT32 TestScratchSize;
+ if (Source == NULL
+ || Destination== NULL
+ || Scratch == NULL) {
+ return EFI_INVALID_PARAMETER;
+ }
+
Status = CustomDecompressGetInfo (Source, SourceSize, &TestDestinationSize, &TestScratchSize);
if (EFI_ERROR (Status)) {
return Status;
|