Update HiiDataBase to fix the SCT hang issues by the invalid device path.

[mirror_edk2.git] / MdeModulePkg / Universal / DriverSampleDxe / DriverSample.c

diff --git a/MdeModulePkg/Universal/DriverSampleDxe/DriverSample.c b/MdeModulePkg/Universal/DriverSampleDxe/DriverSample.c
index 784cd28d7a795ea1f58d6f4a5bf6059de9567dd3..f856083de4b952988dfdf82bec74faddff07b249 100644 (file)
--- a/MdeModulePkg/Universal/DriverSampleDxe/DriverSample.c
+++ b/MdeModulePkg/Universal/DriverSampleDxe/DriverSample.c
@@ -345,7 +345,10 @@ ExtractConfig (
   EFI_STRING                       ConfigRequestHdr;\r
   EFI_STRING                       ConfigRequest;\r
   UINTN                            Size;\r
-\r
+  \r
+  if (Progress == NULL || Results == NULL) {\r
+    return EFI_INVALID_PARAMETER;\r
+  }\r
   //\r
   // Initialize the local variables.\r
   //\r
@@ -833,7 +836,6 @@ DriverSampleInit (
   // Initialize the local variables.\r
   //\r
   ConfigRequestHdr = NULL;\r
-\r
   //\r
   // Initialize screen dimensions for SendForm().\r
   // Remove 3 characters from top and bottom\r