Fix coding style problem in RngDxe driver.

[mirror_edk2.git] / SecurityPkg / Application / RngTest / RngTest.c

diff --git a/SecurityPkg/Application/RngTest/RngTest.c b/SecurityPkg/Application/RngTest/RngTest.c
index 25edecefe3d27fbab9b24e327a991b5b8bda92bc..f501f806e93ce016d542e96e088fe74c55c82c18 100644 (file)
--- a/SecurityPkg/Application/RngTest/RngTest.c
+++ b/SecurityPkg/Application/RngTest/RngTest.c
@@ -103,6 +103,9 @@ UefiMain (
   //\r
   RandSize = 32;\r
   Rand     = AllocatePool (RandSize);\r
+  if (Rand == NULL) {\r
+    goto Exit;\r
+  }\r
   \r
   //\r
   // RNG with default algorithm\r