Add check of buffer value returned from AllocateZeroPool()

[mirror_edk2.git] / MdeModulePkg / Universal / Console / ConSplitterDxe / ConSplitterGraphics.c

diff --git a/MdeModulePkg/Universal/Console/ConSplitterDxe/ConSplitterGraphics.c b/MdeModulePkg/Universal/Console/ConSplitterDxe/ConSplitterGraphics.c
index c3e15d4ac9bfafe911b0e36b06793760e9d16cc2..f70b0765ea8c1f17a9f0822a2810d25151b81b38 100644 (file)
--- a/MdeModulePkg/Universal/Console/ConSplitterDxe/ConSplitterGraphics.c
+++ b/MdeModulePkg/Universal/Console/ConSplitterDxe/ConSplitterGraphics.c
@@ -1130,6 +1130,9 @@ DevNullSyncGopStdOut (
   MaxColumn     = Private->DevNullColumns;\r
 \r
   Buffer        = AllocateZeroPool ((MaxColumn + 1) * sizeof (CHAR16));\r
+  if (Buffer == NULL) {\r
+    return ReturnStatus;\r
+  }\r
 \r
   for (Row = 0; Row < Private->DevNullRows; Row++, Screen += (MaxColumn + 1), Attributes += MaxColumn) {\r
 \r