REF: https://bugzilla.tianocore.org/show_bug.cgi?id=2531
AllocatePool may fail and BinData may be invalid, check it before use. Cc: Michael D Kinney <[email protected]> Cc: Liming Gao <[email protected]> Signed-off-by: GuoMinJ <[email protected]> --- MdePkg/Test/UnitTest/Library/BaseLib/Base64UnitTest.c | 3 +++ 1 file changed, 3 insertions(+) diff --git a/MdePkg/Test/UnitTest/Library/BaseLib/Base64UnitTest.c b/MdePkg/Test/UnitTest/Library/BaseLib/Base64UnitTest.c index 6f7c31cab4..933d2b40e0 100644 --- a/MdePkg/Test/UnitTest/Library/BaseLib/Base64UnitTest.c +++ b/MdePkg/Test/UnitTest/Library/BaseLib/Base64UnitTest.c @@ -252,6 +252,9 @@ RfcDecodeTest( BinSize = AsciiStrnLenS (binString, MAX_TEST_STRING_SIZE); BinData = AllocatePool (BinSize); + if (BinData == NULL) { + return UNIT_TEST_ERROR_TEST_FAILED; + } Btc->BufferToFree = BinData; ReturnSize = BinSize; -- 2.17.1 -=-=-=-=-=-=-=-=-=-=-=- Groups.io Links: You receive all messages sent to this group. View/Reply Online (#55475): https://edk2.groups.io/g/devel/message/55475 Mute This Topic: https://groups.io/mt/71744132/21656 Group Owner: [email protected] Unsubscribe: https://edk2.groups.io/g/devel/unsub [[email protected]] -=-=-=-=-=-=-=-=-=-=-=-
