Add some robustness check in DriverSample driver.

git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@10209 6f19259b-4bc3-4df7-8a09-765794883524

diff --git a/MdeModulePkg/Universal/DriverSampleDxe/DriverSample.c b/MdeModulePkg/Universal/DriverSampleDxe/DriverSample.c
index a1e5a1b6d4ab23361f4c15062f7899efa2c10160..9a6cfc8d7db3a60722726de63d65b8fd48421991 100644 (file)
--- a/MdeModulePkg/Universal/DriverSampleDxe/DriverSample.c
+++ b/MdeModulePkg/Universal/DriverSampleDxe/DriverSample.c
@@ -425,6 +425,7 @@ ExtractConfig (
     ConfigRequestHdr = HiiConstructConfigHdr (&mFormSetGuid, VariableName, PrivateData->DriverHandle[0]);\r
     Size = (StrLen (ConfigRequestHdr) + 32 + 1) * sizeof (CHAR16);\r
     ConfigRequest = AllocateZeroPool (Size);\r
+    ASSERT (ConfigRequest != NULL);\r
     AllocatedRequest = TRUE;\r
     UnicodeSPrint (ConfigRequest, Size, L"%s&OFFSET=0&WIDTH=%016LX", ConfigRequestHdr, (UINT64)BufferSize);\r
     FreePool (ConfigRequestHdr);\r
@@ -454,6 +455,7 @@ ExtractConfig (
       if (StrStr (StrPointer, L"&") == NULL) {\r
         Size = (StrLen (Request) + 32 + 1) * sizeof (CHAR16);\r
         ConfigRequest    = AllocateZeroPool (Size);\r
+        ASSERT (ConfigRequest != NULL);\r
         AllocatedRequest = TRUE;\r
         UnicodeSPrint (ConfigRequest, Size, L"%s&OFFSET=0&WIDTH=%016LX", Request, (UINT64)BufferSize);\r
       }\r