Add check whether input parameter is NULL before call freepool API defined in MDE.

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

1 files changed, 7 insertions, 3 deletions

diff --git a/EdkModulePkg/Universal/Console/ConSplitter/Dxe/ConSplitterGraphics.c b/EdkModulePkg/Universal/Console/ConSplitter/Dxe/ConSplitterGraphics.c
index 9faeb56d12..2f60e11c40 100644
--- a/EdkModulePkg/Universal/Console/ConSplitter/Dxe/ConSplitterGraphics.c
+++ b/EdkModulePkg/Universal/Console/ConSplitter/Dxe/ConSplitterGraphics.c
@@ -1314,8 +1314,10 @@ DevNullTextOutSetMode (
     Private->TextOutMode.Mode = (INT32) ModeNumber;

     Private->DevNullColumns   = Column;

     Private->DevNullRows      = Row;

-

-    FreePool (Private->DevNullScreen);

+    

+    if (Private->DevNullScreen != NULL) {

+      FreePool (Private->DevNullScreen);

+    }

 

     Size                    = (Row * (Column + 1)) * sizeof (CHAR16);

     Private->DevNullScreen  = AllocateZeroPool (Size);

@@ -1323,7 +1325,9 @@ DevNullTextOutSetMode (
       return EFI_OUT_OF_RESOURCES;

     }

 

-    FreePool (Private->DevNullAttributes);

+    if (Private->DevNullAttributes != NULL) {

+      FreePool (Private->DevNullAttributes);

+    }

 

     Size                        = Row * Column * sizeof (INT32);

     Private->DevNullAttributes  = AllocateZeroPool (Size);