MdePkg/UnitTestBaseLib: Add check for pointer BinData

[mirror_edk2.git] / MdePkg / Test / UnitTest / Library / BaseLib / Base64UnitTest.c

diff --git a/MdePkg/Test/UnitTest/Library/BaseLib/Base64UnitTest.c b/MdePkg/Test/UnitTest/Library/BaseLib/Base64UnitTest.c
index 6f7c31cab417c50b01261d279d64c02f320773a6..8952f9da6c3a0894c75ff21e541def1872e7afe4 100644 (file)
--- a/MdePkg/Test/UnitTest/Library/BaseLib/Base64UnitTest.c
+++ b/MdePkg/Test/UnitTest/Library/BaseLib/Base64UnitTest.c
@@ -252,9 +252,11 @@ RfcDecodeTest(
   BinSize = AsciiStrnLenS (binString, MAX_TEST_STRING_SIZE);\r
 \r
   BinData = AllocatePool (BinSize);\r
-  Btc->BufferToFree = BinData;\r
+  UT_ASSERT_NOT_NULL(BinData);\r
 \r
+  Btc->BufferToFree = BinData;\r
   ReturnSize = BinSize;\r
+\r
   Status = Base64Decode (b64String, b64StringLen, BinData, &ReturnSize);\r
 \r
   UT_ASSERT_STATUS_EQUAL (Status, Btc->ExpectedStatus);\r